diff options
Diffstat (limited to 'Src/Plugins/Input/in_flv/AMFDispatch.cpp')
| -rw-r--r-- | Src/Plugins/Input/in_flv/AMFDispatch.cpp | 238 |
1 files changed, 238 insertions, 0 deletions
diff --git a/Src/Plugins/Input/in_flv/AMFDispatch.cpp b/Src/Plugins/Input/in_flv/AMFDispatch.cpp new file mode 100644 index 00000000..0bd8af76 --- /dev/null +++ b/Src/Plugins/Input/in_flv/AMFDispatch.cpp @@ -0,0 +1,238 @@ +#include "AMFDispatch.h" + +AMFDispatch::AMFDispatch(AMFMixedArray *array) +{ + object=array; + if (object) + object->AddRef(); + refCount=1; +} + +AMFDispatch::~AMFDispatch() +{ + if (object) + object->Release(); +} + +STDMETHODIMP AMFDispatch::QueryInterface(REFIID riid, PVOID *ppvObject) +{ + if (!ppvObject) + return E_POINTER; + + else if (IsEqualIID(riid, IID_IDispatch)) + *ppvObject = (IDispatch *)this; + else if (IsEqualIID(riid, IID_IUnknown)) + *ppvObject = this; + else + { + *ppvObject = NULL; + return E_NOINTERFACE; + } + + AddRef(); + return S_OK; +} + +ULONG AMFDispatch::AddRef(void) +{ + return InterlockedIncrement((volatile LONG *)&refCount); +} + +ULONG AMFDispatch::Release(void) +{ + ULONG count = InterlockedDecrement((volatile LONG *)&refCount); + if (count == 0) + delete this; + return count; +} + +enum +{ + DISP_AMF_DEBUGPRINT, + DISP_AMF_MAX, +}; + +#define CHECK_ID(str, id) if (wcscmp(rgszNames[i], L##str) == 0) { rgdispid[i] = id; continue; } +HRESULT AMFDispatch::GetIDsOfNames(REFIID riid, OLECHAR FAR* FAR* rgszNames, unsigned int cNames, LCID lcid, DISPID FAR* rgdispid) +{ + bool unknowns = false; + for (unsigned int i = 0;i != cNames;i++) + { + CHECK_ID("DebugPrint", DISP_AMF_DEBUGPRINT) + if (object) + { + //size_t index = object->array.getPosition(rgszNames[i]); + size_t index = 0; + for (auto it = object->array.begin(); it != object->array.end(); it++, index++) + { + if (wcscmp(it->first.c_str(), rgszNames[i]) == 0) + { + break; + } + } + + if (index != object->array.size()) + { + rgdispid[i] = (DISPID)index + DISP_AMF_MAX; + continue; + } + } + + rgdispid[i] = DISPID_UNKNOWN; + unknowns = true; + + } + if (unknowns) + return DISP_E_UNKNOWNNAME; + else + return S_OK; +} + +HRESULT AMFDispatch::GetTypeInfo(unsigned int itinfo, LCID lcid, ITypeInfo FAR* FAR* pptinfo) +{ + return E_NOTIMPL; +} + +HRESULT AMFDispatch::GetTypeInfoCount(unsigned int FAR * pctinfo) +{ + return E_NOTIMPL; +} + +static void AMFType_To_Variant(AMFType *obj, VARIANT *pvarResult) +{ + VariantInit(pvarResult); + switch(obj->type) + { + case AMFType::TYPE_DOUBLE: // double + { + AMFDouble *cast_obj = static_cast<AMFDouble *>(obj); + V_VT(pvarResult) = VT_R8; + V_R8(pvarResult) = cast_obj->val; + } + break; + case AMFType::TYPE_BOOL: // bool + { + AMFBoolean *cast_obj = static_cast<AMFBoolean *>(obj); + V_VT(pvarResult) = VT_BOOL; + V_BOOL(pvarResult) = cast_obj->boolean; + } + break; + case AMFType::TYPE_MOVIE: // movie (basically just a URL) + case AMFType::TYPE_STRING: // string + { + AMFString *cast_obj = static_cast<AMFString *>(obj); + V_VT(pvarResult) = VT_BSTR; + V_BSTR(pvarResult) = SysAllocString(cast_obj->str); + } + break; + case AMFType::TYPE_LONG_STRING: // string + { + AMFLongString *cast_obj = static_cast<AMFLongString *>(obj); + V_VT(pvarResult) = VT_BSTR; + V_BSTR(pvarResult) = SysAllocString(cast_obj->str); + } + break; + case AMFType::TYPE_MIXEDARRAY: + { + AMFMixedArray *cast_obj = static_cast<AMFMixedArray *>(obj); + V_VT(pvarResult) = VT_DISPATCH; + V_DISPATCH(pvarResult) = new AMFDispatch(cast_obj); + } + break; + case AMFType::TYPE_DATE: + { + AMFTime *cast_obj = static_cast<AMFTime *>(obj); + V_VT(pvarResult) = VT_DATE; + V_DATE(pvarResult) = cast_obj->val; + } + break; + case AMFType::TYPE_ARRAY: + { + AMFArray *cast_obj = static_cast<AMFArray *>(obj); + SAFEARRAYBOUND rgsabound[1]; + rgsabound[0].lLbound = 0; + rgsabound[0].cElements = (ULONG)cast_obj->array.size(); + SAFEARRAY *psa = SafeArrayCreate(VT_VARIANT, 1, rgsabound); + VARIANT **data; + SafeArrayAccessData(psa, (void **)&data); + for (size_t i=0;i!=cast_obj->array.size();i++) + { + AMFType_To_Variant(cast_obj->array[i], data[i]); + } + SafeArrayUnaccessData(psa); + V_VT(pvarResult) = VT_ARRAY; + V_ARRAY(pvarResult) = psa; + } + break; + } +} + +HRESULT AMFDispatch::Invoke(DISPID dispid, REFIID riid, LCID lcid, WORD wFlags, DISPPARAMS FAR *pdispparams, VARIANT FAR *pvarResult, EXCEPINFO FAR * pexecinfo, unsigned int FAR *puArgErr) +{ + if (pvarResult) + VariantInit(pvarResult); + + switch(dispid) + { + case DISP_AMF_DEBUGPRINT: + { + wchar_t debugstring[4096]=L""; + wchar_t *str = debugstring; + size_t len = 4096; + object->DebugPrint(1, str, len); + V_VT(pvarResult) = VT_BSTR; + V_BSTR(pvarResult) = SysAllocString(debugstring); + } + return S_OK; + } + size_t index = dispid - DISP_AMF_MAX; + if (index >= object->array.size()) + return DISP_E_MEMBERNOTFOUND; + + //AMFType *obj = object->array.at(index).second; + AMFType* obj = 0; + auto it = object->array.begin(); + while (index--) + { + it++; + } + if (it != object->array.end()) + { + obj = it->second; + } + + if (!obj) + return S_OK; + + switch(obj->type) + { + case AMFType::TYPE_DOUBLE: + case AMFType::TYPE_BOOL: + case AMFType::TYPE_STRING: + case AMFType::TYPE_MIXEDARRAY: + case AMFType::TYPE_ARRAY: + AMFType_To_Variant(obj, pvarResult); + return S_OK; + + case AMFType::TYPE_OBJECT: // object + // TODO + return DISP_E_TYPEMISMATCH; + case AMFType::TYPE_NULL: // null + return S_OK; + case AMFType::TYPE_REFERENCE: // reference + return DISP_E_TYPEMISMATCH; + case AMFType::TYPE_TERMINATOR: + // TODO? + return DISP_E_TYPEMISMATCH; + case AMFType::TYPE_DATE: // date + return DISP_E_TYPEMISMATCH; + case AMFType::TYPE_LONG_STRING: // long string + return DISP_E_TYPEMISMATCH; + case AMFType::TYPE_XML: // XML + return DISP_E_TYPEMISMATCH; + default: + return DISP_E_TYPEMISMATCH; + + } + return S_OK; +} |