reactos/base/ctf/msctf/categorymgr.cpp
Katayama Hirofumi MZ a18a5734ac
[MSCTF][CICERO][DOC] Fork MSCTF from Wine (#8245)
JIRA issue: CORE-19361
- Delete msctf.c and add msctf.cpp.
- Modify precomp.h and enable precompiled header.
- Modify media/doc/WINESYNC.txt.
- Add cicMemReCalloc function to cicero.
2025-07-13 18:30:42 +09:00

428 lines
11 KiB
C++

/*
* PROJECT: ReactOS CTF
* LICENSE: LGPL-2.0-or-later (https://spdx.org/licenses/LGPL-2.0-or-later)
* PURPOSE: ITfCategoryMgr implementation
* COPYRIGHT: Copyright 2009 Aric Stewart, CodeWeavers
* Copyright 2025 Katayama Hirofumi MZ <katayama.hirofumi.mz@gmail.com>
*/
#include "precomp.h"
#include <wine/debug.h>
WINE_DEFAULT_DEBUG_CHANNEL(msctf);
////////////////////////////////////////////////////////////////////////////
class CCategoryMgr
: public ITfCategoryMgr
{
public:
CCategoryMgr();
virtual ~CCategoryMgr();
// ** IUnknown methods **
STDMETHODIMP QueryInterface(REFIID riid, void **ppvObj) override;
STDMETHODIMP_(ULONG) AddRef() override;
STDMETHODIMP_(ULONG) Release() override;
// ** ITfCategoryMgr methods **
STDMETHODIMP RegisterCategory(
_In_ REFCLSID rclsid,
_In_ REFGUID rcatid,
_In_ REFGUID rguid) override;
STDMETHODIMP UnregisterCategory(
_In_ REFCLSID rclsid,
_In_ REFGUID rcatid,
_In_ REFGUID rguid) override;
STDMETHODIMP EnumCategoriesInItem(
_In_ REFGUID rguid,
_Out_ IEnumGUID **ppEnum) override;
STDMETHODIMP EnumItemsInCategory(
_In_ REFGUID rcatid,
_Out_ IEnumGUID **ppEnum) override;
STDMETHODIMP FindClosestCategory(
_In_ REFGUID rguid,
_Out_ GUID *pcatid,
_In_ const GUID **ppcatidList,
_In_ ULONG ulCount) override;
STDMETHODIMP RegisterGUIDDescription(
_In_ REFCLSID rclsid,
_In_ REFGUID rguid,
_In_ const WCHAR *pchDesc,
_In_ ULONG cch) override;
STDMETHODIMP UnregisterGUIDDescription(
_In_ REFCLSID rclsid,
_In_ REFGUID rguid) override;
STDMETHODIMP GetGUIDDescription(
_In_ REFGUID rguid,
_Out_ BSTR *pbstrDesc) override;
STDMETHODIMP RegisterGUIDDWORD(
_In_ REFCLSID rclsid,
_In_ REFGUID rguid,
_In_ DWORD dw) override;
STDMETHODIMP UnregisterGUIDDWORD(
_In_ REFCLSID rclsid,
_In_ REFGUID rguid) override;
STDMETHODIMP GetGUIDDWORD(
_In_ REFGUID rguid,
_Out_ DWORD *pdw) override;
STDMETHODIMP RegisterGUID(
_In_ REFGUID rguid,
_Out_ TfGuidAtom *pguidatom) override;
STDMETHODIMP GetGUID(
_In_ TfGuidAtom guidatom,
_Out_ GUID *pguid) override;
STDMETHODIMP IsEqualTfGuidAtom(
_In_ TfGuidAtom guidatom,
_In_ REFGUID rguid,
_Out_ BOOL *pfEqual) override;
protected:
LONG m_cRefs;
};
////////////////////////////////////////////////////////////////////////////
CCategoryMgr::CCategoryMgr()
: m_cRefs(1)
{
}
CCategoryMgr::~CCategoryMgr()
{
TRACE("destroying %p\n", this);
}
STDMETHODIMP CCategoryMgr::QueryInterface(REFIID riid, void **ppvObj)
{
if (!ppvObj)
return E_INVALIDARG;
*ppvObj = NULL;
if (riid == IID_IUnknown || riid == IID_ITfCategoryMgr)
*ppvObj = this;
if (*ppvObj)
{
AddRef();
return S_OK;
}
WARN("unsupported interface: %s\n", debugstr_guid(&riid));
return E_NOINTERFACE;
}
STDMETHODIMP_(ULONG) CCategoryMgr::AddRef()
{
return ::InterlockedIncrement(&m_cRefs);
}
STDMETHODIMP_(ULONG) CCategoryMgr::Release()
{
ULONG ret = ::InterlockedDecrement(&m_cRefs);
if (!ret)
delete this;
return ret;
}
STDMETHODIMP CCategoryMgr::RegisterCategory(
_In_ REFCLSID rclsid,
_In_ REFGUID rcatid,
_In_ REFGUID rguid)
{
WCHAR szFullKey[110], szClsid[39], szCatid[39], szGuid[39];
HKEY hTipKey = NULL, hCatKey = NULL, hItemKey = NULL;
LSTATUS error;
HRESULT hr = E_FAIL;
TRACE("%p -> (%s, %s, %s)\n", this, debugstr_guid(&rclsid), debugstr_guid(&rcatid),
debugstr_guid(&rguid));
StringFromGUID2(rclsid, szClsid, _countof(szClsid));
StringFromGUID2(rcatid, szCatid, _countof(szCatid));
StringFromGUID2(rguid, szGuid, _countof(szGuid));
StringCchPrintfW(szFullKey, _countof(szFullKey), L"%s\\%s", szwSystemTIPKey, szClsid);
error = RegOpenKeyExW(HKEY_LOCAL_MACHINE, szFullKey, 0, KEY_READ | KEY_WRITE, &hTipKey);
if (error != ERROR_SUCCESS)
return E_FAIL;
StringCchPrintfW(szFullKey, _countof(szFullKey), L"Category\\Category\\%s\\%s", szCatid, szGuid);
error = RegCreateKeyExW(hTipKey, szFullKey, 0, NULL, 0, KEY_READ | KEY_WRITE, NULL,
&hCatKey, NULL);
if (error == ERROR_SUCCESS)
{
RegCloseKey(hCatKey);
StringCchPrintfW(szFullKey, _countof(szFullKey), L"Category\\Item\\%s\\%s", szGuid, szCatid);
error = RegCreateKeyExW(hTipKey, szFullKey, 0, NULL, 0, KEY_READ | KEY_WRITE, NULL,
&hItemKey, NULL);
if (error == ERROR_SUCCESS)
{
RegCloseKey(hItemKey);
hr = S_OK;
}
}
RegCloseKey(hTipKey);
return hr;
}
STDMETHODIMP CCategoryMgr::UnregisterCategory(
_In_ REFCLSID rclsid,
_In_ REFGUID rcatid,
_In_ REFGUID rguid)
{
WCHAR szFullKey[110], szClsid[39], szCatid[39], szGuid[39];
HKEY hTipKey = NULL;
LSTATUS error;
TRACE("%p -> (%s %s %s)\n", this, debugstr_guid(&rclsid), debugstr_guid(&rcatid),
debugstr_guid(&rguid));
StringFromGUID2(rclsid, szClsid, _countof(szClsid));
StringFromGUID2(rcatid, szCatid, _countof(szCatid));
StringFromGUID2(rguid, szGuid, _countof(szGuid));
StringCchPrintfW(szFullKey, _countof(szFullKey), L"%s\\%s", szwSystemTIPKey, szClsid);
error = RegOpenKeyExW(HKEY_LOCAL_MACHINE, szFullKey, 0, KEY_READ | KEY_WRITE, &hTipKey);
if (error != ERROR_SUCCESS)
return E_FAIL;
StringCchPrintfW(szFullKey, _countof(szFullKey), L"Category\\Category\\%s\\%s", szCatid, szGuid);
RegDeleteTreeW(hTipKey, szFullKey);
StringCchPrintfW(szFullKey, _countof(szFullKey), L"Category\\Item\\%s\\%s", szGuid, szCatid);
RegDeleteTreeW(hTipKey, szFullKey);
RegCloseKey(hTipKey);
return S_OK;
}
STDMETHODIMP CCategoryMgr::EnumCategoriesInItem(
_In_ REFGUID rguid,
_Out_ IEnumGUID **ppEnum)
{
FIXME("STUB:(%p)\n", this);
return E_NOTIMPL;
}
STDMETHODIMP CCategoryMgr::EnumItemsInCategory(
_In_ REFGUID rcatid,
_Out_ IEnumGUID **ppEnum)
{
FIXME("STUB:(%p)\n", this);
return E_NOTIMPL;
}
STDMETHODIMP CCategoryMgr::FindClosestCategory(
_In_ REFGUID rguid,
_Out_ GUID *pcatid,
_In_ const GUID **ppcatidList,
_In_ ULONG ulCount)
{
WCHAR szFullKey[120], szGuid[39];
HKEY hKey = NULL;
HRESULT hr = S_FALSE;
DWORD dwIndex = 0;
LSTATUS error;
TRACE("(%p)\n", this);
if (!pcatid || (ulCount && !ppcatidList))
return E_INVALIDARG;
StringFromGUID2(rguid, szGuid, _countof(szGuid));
StringCchPrintfW(szFullKey, _countof(szFullKey), L"%s\\%s\\Category\\Item\\%s", szwSystemTIPKey, szGuid, szGuid);
*pcatid = GUID_NULL;
error = RegOpenKeyExW(HKEY_LOCAL_MACHINE, szFullKey, 0, KEY_READ, &hKey);
if (error != ERROR_SUCCESS)
return S_FALSE;
// Enumerate subkeys to find the closest category
while (TRUE)
{
WCHAR szCatidName[39];
DWORD cchCatidName = _countof(szCatidName);
GUID currentCatid;
error = RegEnumKeyExW(hKey, dwIndex, szCatidName, &cchCatidName, NULL, NULL, NULL, NULL);
if (error != ERROR_SUCCESS || error == ERROR_NO_MORE_ITEMS)
break;
dwIndex++;
HRESULT hr2 = CLSIDFromString(szCatidName, &currentCatid);
if (FAILED(hr2))
continue; // Skip invalid GUID strings
if (ulCount <= 0)
{
*pcatid = currentCatid;
hr = S_OK; // Found a category
break;
}
// If a list of categories is provided, check if the current one is in the list
BOOL bFound = FALSE;
for (ULONG j = 0; j < ulCount; j++)
{
if (currentCatid == *ppcatidList[j])
{
bFound = TRUE;
*pcatid = currentCatid;
hr = S_OK; // Found a matching category
break;
}
}
if (bFound)
break; // Found and matched, so stop searching
}
RegCloseKey(hKey);
return hr;
}
STDMETHODIMP CCategoryMgr::RegisterGUIDDescription(
_In_ REFCLSID rclsid,
_In_ REFGUID rguid,
_In_ const WCHAR *pchDesc,
_In_ ULONG cch)
{
FIXME("STUB:(%p)\n", this);
return E_NOTIMPL;
}
STDMETHODIMP CCategoryMgr::UnregisterGUIDDescription(
_In_ REFCLSID rclsid,
_In_ REFGUID rguid)
{
FIXME("STUB:(%p)\n", this);
return E_NOTIMPL;
}
STDMETHODIMP CCategoryMgr::GetGUIDDescription(
_In_ REFGUID rguid,
_Out_ BSTR *pbstrDesc)
{
FIXME("STUB:(%p)\n", this);
return E_NOTIMPL;
}
STDMETHODIMP CCategoryMgr::RegisterGUIDDWORD(
_In_ REFCLSID rclsid,
_In_ REFGUID rguid,
_In_ DWORD dw)
{
FIXME("STUB:(%p)\n", this);
return E_NOTIMPL;
}
STDMETHODIMP CCategoryMgr::UnregisterGUIDDWORD(
_In_ REFCLSID rclsid,
_In_ REFGUID rguid)
{
FIXME("STUB:(%p)\n", this);
return E_NOTIMPL;
}
STDMETHODIMP CCategoryMgr::GetGUIDDWORD(
_In_ REFGUID rguid,
_Out_ DWORD *pdw)
{
FIXME("STUB:(%p)\n", this);
return E_NOTIMPL;
}
STDMETHODIMP CCategoryMgr::RegisterGUID(
_In_ REFGUID rguid,
_Out_ TfGuidAtom *pguidatom)
{
TRACE("%p -> (%s, %p)\n", this, debugstr_guid(&rguid), pguidatom);
if (!pguidatom)
return E_INVALIDARG;
DWORD dwCookieId = 0, dwEnumIndex = 0;
do
{
dwCookieId = enumerate_Cookie(COOKIE_MAGIC_GUIDATOM, &dwEnumIndex);
if (dwCookieId && rguid == *(const GUID *)get_Cookie_data(dwCookieId))
{
*pguidatom = dwCookieId;
return S_OK;
}
} while (dwCookieId != 0);
GUID *pNewGuid = (GUID *)cicMemAlloc(sizeof(GUID));
if (!pNewGuid)
return E_OUTOFMEMORY;
*pNewGuid = rguid;
dwCookieId = generate_Cookie(COOKIE_MAGIC_GUIDATOM, pNewGuid);
if (dwCookieId == 0)
{
cicMemFree(pNewGuid);
return E_FAIL;
}
*pguidatom = dwCookieId;
return S_OK;
}
STDMETHODIMP CCategoryMgr::GetGUID(
_In_ TfGuidAtom guidatom,
_Out_ GUID *pguid)
{
TRACE("%p -> (%d, %p)\n", this, guidatom, pguid);
if (!pguid)
return E_INVALIDARG;
*pguid = GUID_NULL;
if (get_Cookie_magic(guidatom) == COOKIE_MAGIC_GUIDATOM)
*pguid = *(const GUID *)get_Cookie_data(guidatom);
return S_OK;
}
STDMETHODIMP CCategoryMgr::IsEqualTfGuidAtom(
_In_ TfGuidAtom guidatom,
_In_ REFGUID rguid,
_Out_ BOOL *pfEqual)
{
TRACE("%p -> (%d %s %p)\n", this, guidatom, debugstr_guid(&rguid), pfEqual);
if (!pfEqual)
return E_INVALIDARG;
*pfEqual = FALSE;
if (get_Cookie_magic(guidatom) == COOKIE_MAGIC_GUIDATOM)
{
if (rguid == *(const GUID *)get_Cookie_data(guidatom))
*pfEqual = TRUE;
}
return S_OK;
}
////////////////////////////////////////////////////////////////////////////
EXTERN_C
HRESULT CategoryMgr_Constructor(IUnknown *pUnkOuter, IUnknown **ppOut)
{
if (pUnkOuter)
return CLASS_E_NOAGGREGATION;
CCategoryMgr *This = new(cicNoThrow) CCategoryMgr();
if (!This)
return E_OUTOFMEMORY;
*ppOut = static_cast<ITfCategoryMgr *>(This);
TRACE("returning %p\n", *ppOut);
return S_OK;
}