286 lines
7.5 KiB
C++
286 lines
7.5 KiB
C++
#include "JSAPI2_MediaCore.h"
|
|
#include "main.h"
|
|
#include "api.h"
|
|
#include "JSAPI.h"
|
|
#include "JSAPI2_Security.h"
|
|
#include "JSAPI2_CallbackManager.h"
|
|
|
|
JSAPI2::MediaCoreAPI::MediaCoreAPI(const wchar_t *_key, JSAPI::ifc_info *_info) : metadataGuard("MediaCoreAPI metadata Guard")
|
|
{
|
|
info = _info;
|
|
key = _key;
|
|
refCount = 1;
|
|
}
|
|
|
|
JSAPI2::MediaCoreAPI::~MediaCoreAPI()
|
|
{
|
|
JSAPI2::callbackManager.Deregister(this);
|
|
}
|
|
|
|
#define DISP_TABLE \
|
|
CHECK_ID(IsRegisteredExtension)\
|
|
CHECK_ID(GetMetadata)\
|
|
CHECK_ID(AddMetadataHook)\
|
|
CHECK_ID(RemoveMetadataHook)\
|
|
|
|
|
|
#define CHECK_ID(str) JSAPI_DISP_ENUMIFY(str),
|
|
enum {
|
|
DISP_TABLE
|
|
};
|
|
|
|
#undef CHECK_ID
|
|
#define CHECK_ID(str)\
|
|
if (CSTR_EQUAL == CompareStringW(lcid, NORM_IGNORECASE, rgszNames[i], -1, L## #str, -1))\
|
|
{ rgdispid[i] = JSAPI_DISP_ENUMIFY(str); continue; }
|
|
|
|
HRESULT JSAPI2::MediaCoreAPI::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++)
|
|
{
|
|
DISP_TABLE
|
|
|
|
rgdispid[i] = DISPID_UNKNOWN;
|
|
unknowns = true;
|
|
|
|
}
|
|
if (unknowns)
|
|
return DISP_E_UNKNOWNNAME;
|
|
else
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT JSAPI2::MediaCoreAPI::GetTypeInfo(unsigned int itinfo, LCID lcid, ITypeInfo FAR* FAR* pptinfo)
|
|
{
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
HRESULT JSAPI2::MediaCoreAPI::GetTypeInfoCount(unsigned int FAR * pctinfo)
|
|
{
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
HRESULT JSAPI2::MediaCoreAPI::IsRegisteredExtension(WORD wFlags, DISPPARAMS FAR *pdispparams, VARIANT FAR *pvarResult, unsigned int FAR *puArgErr)
|
|
{
|
|
JSAPI_VERIFY_METHOD(wFlags);
|
|
JSAPI_VERIFY_PARAMCOUNT(pdispparams, 1);
|
|
JSAPI_VERIFY_PARAMTYPE(pdispparams, 1, VT_BSTR, puArgErr);
|
|
|
|
JSAPI_INIT_RESULT(pvarResult, VT_BOOL);
|
|
|
|
const wchar_t *extension = JSAPI_PARAM(pdispparams, 1).bstrVal;
|
|
int start_offs=0;
|
|
wchar_t filename[MAX_PATH] = {0};
|
|
StringCbPrintfW(filename, sizeof(filename), L"test.%s", extension);
|
|
In_Module *i = in_setmod_noplay(filename, &start_offs);
|
|
if (i)
|
|
V_BOOL(pvarResult) = VARIANT_TRUE;
|
|
else
|
|
V_BOOL(pvarResult) = VARIANT_FALSE;
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT JSAPI2::MediaCoreAPI::GetMetadata(WORD wFlags, DISPPARAMS FAR *pdispparams, VARIANT FAR *pvarResult, unsigned int FAR *puArgErr)
|
|
{
|
|
JSAPI_VERIFY_METHOD(wFlags);
|
|
JSAPI_VERIFY_PARAMCOUNT(pdispparams, 2);
|
|
JSAPI_VERIFY_PARAMTYPE(pdispparams, 1, VT_BSTR, puArgErr);
|
|
JSAPI_VERIFY_PARAMTYPE(pdispparams, 2, VT_BSTR, puArgErr);
|
|
|
|
JSAPI_INIT_RESULT(pvarResult, VT_BSTR);
|
|
|
|
if (security.GetActionAuthorization(L"mediacore", L"metadata", key, info, JSAPI2::api_security::ACTION_PROMPT) == JSAPI2::api_security::ACTION_ALLOWED)
|
|
{
|
|
wchar_t buffer[4096] = {0};
|
|
extendedFileInfoStructW info;
|
|
|
|
info.filename = JSAPI_PARAM(pdispparams, 1).bstrVal;
|
|
info.metadata = JSAPI_PARAM(pdispparams, 2).bstrVal;
|
|
info.ret = buffer;
|
|
info.retlen = sizeof(buffer)/sizeof(*buffer);
|
|
|
|
if (NULL != info.filename &&
|
|
NULL != info.metadata)
|
|
{
|
|
if (0 == SendMessageW(hMainWindow, WM_WA_IPC, (WPARAM)&info, IPC_GET_EXTENDED_FILE_INFOW_HOOKABLE))
|
|
info.ret = NULL;
|
|
|
|
JSAPI_SET_RESULT(pvarResult, bstrVal, SysAllocString(info.ret));
|
|
}
|
|
else
|
|
JSAPI_EMPTY_RESULT(pvarResult);
|
|
}
|
|
else
|
|
{
|
|
JSAPI_EMPTY_RESULT(pvarResult);
|
|
}
|
|
|
|
return S_OK;
|
|
}
|
|
void JSAPI2::MediaCoreAPI::RemoveMetadataHook(const wchar_t *filename)
|
|
{
|
|
Nullsoft::Utility::AutoLock metadataLock(metadataGuard);
|
|
RemoveMetadataHook_again:
|
|
MetadataMap::iterator itr;
|
|
for (itr=metadataMap.begin();itr!=metadataMap.end();itr++)
|
|
{
|
|
if (!_wcsicmp(filename, itr->url.c_str()))
|
|
{
|
|
metadataMap.erase(itr);
|
|
goto RemoveMetadataHook_again;
|
|
}
|
|
}
|
|
}
|
|
|
|
void JSAPI2::MediaCoreAPI::RemoveMetadataHook(const wchar_t *filename, const wchar_t *tag)
|
|
{
|
|
Nullsoft::Utility::AutoLock metadataLock(metadataGuard);
|
|
RemoveMetadataHook_again2:
|
|
MetadataMap::iterator itr;
|
|
for (itr=metadataMap.begin();itr!=metadataMap.end();itr++)
|
|
{
|
|
if (!_wcsicmp(filename, itr->url.c_str())
|
|
&& !_wcsicmp(tag, itr->tag.c_str()))
|
|
{
|
|
metadataMap.erase(itr);
|
|
goto RemoveMetadataHook_again2;
|
|
}
|
|
}
|
|
}
|
|
|
|
HRESULT JSAPI2::MediaCoreAPI::AddMetadataHook(WORD wFlags, DISPPARAMS FAR *pdispparams, VARIANT FAR *pvarResult, unsigned int FAR *puArgErr)
|
|
{
|
|
JSAPI_VERIFY_METHOD(wFlags);
|
|
JSAPI_VERIFY_PARAMCOUNT(pdispparams, 3);
|
|
JSAPI_VERIFY_PARAMTYPE(pdispparams, 1, VT_BSTR, puArgErr);
|
|
JSAPI_VERIFY_PARAMTYPE(pdispparams, 2, VT_BSTR, puArgErr);
|
|
JSAPI_VERIFY_PARAMTYPE(pdispparams, 3, VT_BSTR, puArgErr);
|
|
|
|
JSAPI_INIT_RESULT(pvarResult, VT_BOOL);
|
|
|
|
if (security.GetActionAuthorization(L"mediacore", L"metadatahook", key, info, JSAPI2::api_security::ACTION_PROMPT) == JSAPI2::api_security::ACTION_ALLOWED)
|
|
{
|
|
const wchar_t *filename = JSAPI_PARAM(pdispparams, 1).bstrVal;
|
|
const wchar_t *tag = JSAPI_PARAM(pdispparams, 2).bstrVal;
|
|
const wchar_t *value = JSAPI_PARAM(pdispparams, 3).bstrVal;
|
|
|
|
if (NULL == filename || L'\0' == *filename ||
|
|
NULL == tag || L'\0' == *tag)
|
|
{
|
|
JSAPI_SET_RESULT(pvarResult, boolVal, VARIANT_FALSE);
|
|
return S_OK;
|
|
}
|
|
|
|
JSAPI2::callbackManager.Register(this);
|
|
metadata_info info;
|
|
info.url = filename;
|
|
info.tag = tag;
|
|
info.metadata= value;
|
|
|
|
Nullsoft::Utility::AutoLock metadataLock(metadataGuard);
|
|
|
|
RemoveMetadataHook(filename, tag);
|
|
|
|
metadataMap.push_back(info);
|
|
JSAPI_SET_RESULT(pvarResult, boolVal, VARIANT_TRUE);
|
|
}
|
|
else
|
|
{
|
|
JSAPI_SET_RESULT(pvarResult, boolVal, VARIANT_FALSE);
|
|
}
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT JSAPI2::MediaCoreAPI::RemoveMetadataHook(WORD wFlags, DISPPARAMS FAR *pdispparams, VARIANT FAR *pvarResult, unsigned int FAR *puArgErr)
|
|
{
|
|
JSAPI_VERIFY_METHOD(wFlags);
|
|
JSAPI_VERIFY_PARAMCOUNT_OPTIONAL(pdispparams, 1, 2);
|
|
JSAPI_VERIFY_PARAMTYPE(pdispparams, 1, VT_BSTR, puArgErr);
|
|
JSAPI_VERIFY_PARAMTYPE_OPTIONAL(pdispparams, 2, VT_BSTR, puArgErr);
|
|
|
|
JSAPI_INIT_RESULT(pvarResult, VT_BOOL);
|
|
|
|
const wchar_t *filename = JSAPI_PARAM(pdispparams, 1).bstrVal;
|
|
const wchar_t *tag = JSAPI_PARAM_OPTIONAL(pdispparams, 2, bstrVal, 0);
|
|
|
|
if (NULL == filename || L'\0' == *filename)
|
|
{
|
|
JSAPI_SET_RESULT(pvarResult, boolVal, VARIANT_FALSE);
|
|
return S_OK;
|
|
}
|
|
|
|
if (NULL != tag && L'\0' == *tag)
|
|
tag = NULL;
|
|
|
|
if (NULL != tag)
|
|
RemoveMetadataHook(filename, tag);
|
|
else
|
|
RemoveMetadataHook(filename);
|
|
|
|
|
|
JSAPI_SET_RESULT(pvarResult, boolVal, VARIANT_TRUE);
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
bool JSAPI2::MediaCoreAPI::OverrideMetadata(const wchar_t *filename, const wchar_t *tag, wchar_t *out, size_t outCch)
|
|
{
|
|
Nullsoft::Utility::AutoLock metadataLock(metadataGuard);
|
|
MetadataMap::iterator itr;
|
|
for (itr=metadataMap.begin();itr!=metadataMap.end();itr++)
|
|
{
|
|
if (!_wcsicmp(filename, itr->url.c_str()) && !_wcsicmp(tag, itr->tag.c_str()))
|
|
{
|
|
StringCchCopyW(out, outCch, itr->metadata.c_str());
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
#undef CHECK_ID
|
|
#define CHECK_ID(str) case JSAPI_DISP_ENUMIFY(str): return str(wFlags, pdispparams, pvarResult, puArgErr);
|
|
HRESULT JSAPI2::MediaCoreAPI::Invoke(DISPID dispid, REFIID riid, LCID lcid, WORD wFlags, DISPPARAMS FAR *pdispparams, VARIANT FAR *pvarResult, EXCEPINFO FAR * pexecinfo, unsigned int FAR *puArgErr)
|
|
{
|
|
switch (dispid)
|
|
{
|
|
DISP_TABLE
|
|
}
|
|
return DISP_E_MEMBERNOTFOUND;
|
|
}
|
|
|
|
STDMETHODIMP JSAPI2::MediaCoreAPI::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 JSAPI2::MediaCoreAPI::AddRef(void)
|
|
{
|
|
return InterlockedIncrement(&refCount);
|
|
}
|
|
|
|
|
|
ULONG JSAPI2::MediaCoreAPI::Release(void)
|
|
{
|
|
LONG lRef = InterlockedDecrement(&refCount);
|
|
if (lRef == 0) delete this;
|
|
return lRef;
|
|
}
|