dlls/msctf/categorymgr.c

   1 /*
   2  *  ITfCategoryMgr implementation
   3  *
   4  *  Copyright 2009 Aric Stewart, CodeWeavers
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #include "config.h"
  22
  23 #include <stdarg.h>
  24
  25 #define COBJMACROS
  26
  27 #include "wine/debug.h"
  28 #include "windef.h"
  29 #include "winbase.h"
  30 #include "winreg.h"
  31 #include "winuser.h"
  32 #include "shlwapi.h"
  33 #include "winerror.h"
  34 #include "objbase.h"
  35
  36 #include "wine/unicode.h"
  37
  38 #include "msctf.h"
  39 #include "msctf_internal.h"
  40
  41 WINE_DEFAULT_DEBUG_CHANNEL(msctf);
  42
  43 typedef struct tagCategoryMgr {
  44     ITfCategoryMgr ITfCategoryMgr_iface;
  45     LONG refCount;
  46 } CategoryMgr;
  47
  48 static inline CategoryMgr *impl_from_ITfCategoryMgr(ITfCategoryMgr *iface)
  49 {
  50     return CONTAINING_RECORD(iface, CategoryMgr, ITfCategoryMgr_iface);
  51 }
  52
  53 static void CategoryMgr_Destructor(CategoryMgr *This)
  54 {
  55     TRACE("destroying %p\n", This);
  56     HeapFree(GetProcessHeap(),0,This);
  57 }
  58
  59 static HRESULT WINAPI CategoryMgr_QueryInterface(ITfCategoryMgr *iface, REFIID iid, LPVOID *ppvOut)
  60 {
  61     CategoryMgr *This = impl_from_ITfCategoryMgr(iface);
  62     *ppvOut = NULL;
  63
  64     if (IsEqualIID(iid, &IID_IUnknown) || IsEqualIID(iid, &IID_ITfCategoryMgr))
  65     {
  66         *ppvOut = This;
  67     }
  68
  69     if (*ppvOut)
  70     {
  71         IUnknown_AddRef(iface);
  72         return S_OK;
  73     }
  74
  75     WARN("unsupported interface: %s\n", debugstr_guid(iid));
  76     return E_NOINTERFACE;
  77 }
  78
  79 static ULONG WINAPI CategoryMgr_AddRef(ITfCategoryMgr *iface)
  80 {
  81     CategoryMgr *This = impl_from_ITfCategoryMgr(iface);
  82     return InterlockedIncrement(&This->refCount);
  83 }
  84
  85 static ULONG WINAPI CategoryMgr_Release(ITfCategoryMgr *iface)
  86 {
  87     CategoryMgr *This = impl_from_ITfCategoryMgr(iface);
  88     ULONG ret;
  89
  90     ret = InterlockedDecrement(&This->refCount);
  91     if (ret == 0)
  92         CategoryMgr_Destructor(This);
  93     return ret;
  94 }
  95
  96 /*****************************************************
  97  * ITfCategoryMgr functions
  98  *****************************************************/
  99
 100 static HRESULT WINAPI CategoryMgr_RegisterCategory ( ITfCategoryMgr *iface,
 101         REFCLSID rclsid, REFGUID rcatid, REFGUID rguid)
 102 {
 103     WCHAR fullkey[110];
 104     WCHAR buf[39];
 105     WCHAR buf2[39];
 106     ULONG res;
 107     HKEY tipkey,catkey,itmkey;
 108     CategoryMgr *This = impl_from_ITfCategoryMgr(iface);
 109
 110     static const WCHAR ctg[] = {'C','a','t','e','g','o','r','y',0};
 111     static const WCHAR itm[] = {'I','t','e','m',0};
 112     static const WCHAR fmt[] = {'%','s','\\','%','s',0};
 113     static const WCHAR fmt2[] = {'%','s','\\','%','s','\\','%','s','\\','%','s',0};
 114
 115     TRACE("(%p) %s %s %s\n",This,debugstr_guid(rclsid), debugstr_guid(rcatid), debugstr_guid(rguid));
 116
 117     StringFromGUID2(rclsid, buf, 39);
 118     sprintfW(fullkey,fmt,szwSystemTIPKey,buf);
 119
 120     if (RegOpenKeyExW(HKEY_LOCAL_MACHINE,fullkey, 0, KEY_READ | KEY_WRITE,
 121                 &tipkey ) != ERROR_SUCCESS)
 122         return E_FAIL;
 123
 124     StringFromGUID2(rcatid, buf, 39);
 125     StringFromGUID2(rguid, buf2, 39);
 126     sprintfW(fullkey,fmt2,ctg,ctg,buf,buf2);
 127
 128     res = RegCreateKeyExW(tipkey, fullkey, 0, NULL, 0, KEY_READ | KEY_WRITE,
 129             NULL, &catkey, NULL);
 130     RegCloseKey(catkey);
 131
 132     if (!res)
 133     {
 134         sprintfW(fullkey,fmt2,ctg,itm,buf2,buf);
 135         res = RegCreateKeyExW(tipkey, fullkey, 0, NULL, 0, KEY_READ | KEY_WRITE,
 136                 NULL, &itmkey, NULL);
 137
 138         RegCloseKey(itmkey);
 139     }
 140
 141     RegCloseKey(tipkey);
 142
 143     if (!res)
 144         return S_OK;
 145     else
 146         return E_FAIL;
 147 }
 148
 149 static HRESULT WINAPI CategoryMgr_UnregisterCategory ( ITfCategoryMgr *iface,
 150         REFCLSID rclsid, REFGUID rcatid, REFGUID rguid)
 151 {
 152     WCHAR fullkey[110];
 153     WCHAR buf[39];
 154     WCHAR buf2[39];
 155     HKEY tipkey;
 156     CategoryMgr *This = impl_from_ITfCategoryMgr(iface);
 157
 158     static const WCHAR ctg[] = {'C','a','t','e','g','o','r','y',0};
 159     static const WCHAR itm[] = {'I','t','e','m',0};
 160     static const WCHAR fmt[] = {'%','s','\\','%','s',0};
 161     static const WCHAR fmt2[] = {'%','s','\\','%','s','\\','%','s','\\','%','s',0};
 162
 163     TRACE("(%p) %s %s %s\n",This,debugstr_guid(rclsid), debugstr_guid(rcatid), debugstr_guid(rguid));
 164
 165     StringFromGUID2(rclsid, buf, 39);
 166     sprintfW(fullkey,fmt,szwSystemTIPKey,buf);
 167
 168     if (RegOpenKeyExW(HKEY_LOCAL_MACHINE,fullkey, 0, KEY_READ | KEY_WRITE,
 169                 &tipkey ) != ERROR_SUCCESS)
 170         return E_FAIL;
 171
 172     StringFromGUID2(rcatid, buf, 39);
 173     StringFromGUID2(rguid, buf2, 39);
 174     sprintfW(fullkey,fmt2,ctg,ctg,buf,buf2);
 175
 176     sprintfW(fullkey,fmt2,ctg,itm,buf2,buf);
 177     RegDeleteTreeW(tipkey, fullkey);
 178     sprintfW(fullkey,fmt2,ctg,itm,buf2,buf);
 179     RegDeleteTreeW(tipkey, fullkey);
 180
 181     RegCloseKey(tipkey);
 182     return S_OK;
 183 }
 184
 185 static HRESULT WINAPI CategoryMgr_EnumCategoriesInItem ( ITfCategoryMgr *iface,
 186         REFGUID rguid, IEnumGUID **ppEnum)
 187 {
 188     CategoryMgr *This = impl_from_ITfCategoryMgr(iface);
 189     FIXME("STUB:(%p)\n",This);
 190     return E_NOTIMPL;
 191 }
 192
 193 static HRESULT WINAPI CategoryMgr_EnumItemsInCategory ( ITfCategoryMgr *iface,
 194         REFGUID rcatid, IEnumGUID **ppEnum)
 195 {
 196     CategoryMgr *This = impl_from_ITfCategoryMgr(iface);
 197     FIXME("STUB:(%p)\n",This);
 198     return E_NOTIMPL;
 199 }
 200
 201 static HRESULT WINAPI CategoryMgr_FindClosestCategory ( ITfCategoryMgr *iface,
 202         REFGUID rguid, GUID *pcatid, const GUID **ppcatidList, ULONG ulCount)
 203 {
 204     static const WCHAR fmt[] = { '%','s','\\','%','s','\\','C','a','t','e','g','o','r','y','\\','I','t','e','m','\\','%','s',0};
 205
 206     WCHAR fullkey[110];
 207     WCHAR buf[39];
 208     HKEY key;
 209     HRESULT hr = S_FALSE;
 210     INT index = 0;
 211     CategoryMgr *This = impl_from_ITfCategoryMgr(iface);
 212
 213     TRACE("(%p)\n",This);
 214
 215     if (!pcatid || (ulCount && ppcatidList == NULL))
 216         return E_INVALIDARG;
 217
 218     StringFromGUID2(rguid, buf, 39);
 219     sprintfW(fullkey,fmt,szwSystemTIPKey,buf,buf);
 220     *pcatid = GUID_NULL;
 221
 222     if (RegOpenKeyExW(HKEY_LOCAL_MACHINE,fullkey, 0, KEY_READ, &key ) !=
 223             ERROR_SUCCESS)
 224         return S_FALSE;
 225
 226     while (1)
 227     {
 228         HRESULT hr2;
 229         ULONG res;
 230         GUID guid;
 231         WCHAR catid[39];
 232         DWORD cName;
 233
 234         cName = 39;
 235         res = RegEnumKeyExW(key, index, catid, &cName, NULL, NULL, NULL, NULL);
 236         if (res != ERROR_SUCCESS && res != ERROR_MORE_DATA) break;
 237         index ++;
 238
 239         hr2 = CLSIDFromString(catid, &guid);
 240         if (FAILED(hr2)) continue;
 241
 242         if (ulCount)
 243         {
 244             int j;
 245             BOOL found = FALSE;
 246             for (j = 0; j < ulCount; j++)
 247                 if (IsEqualGUID(&guid, ppcatidList[j]))
 248                 {
 249                     found = TRUE;
 250                     *pcatid = guid;
 251                     hr = S_OK;
 252                     break;
 253                 }
 254             if (found) break;
 255         }
 256         else
 257         {
 258             *pcatid = guid;
 259             hr = S_OK;
 260             break;
 261         }
 262     }
 263
 264     return hr;
 265 }
 266
 267 static HRESULT WINAPI CategoryMgr_RegisterGUIDDescription (
 268         ITfCategoryMgr *iface, REFCLSID rclsid, REFGUID rguid,
 269         const WCHAR *pchDesc, ULONG cch)
 270 {
 271     CategoryMgr *This = impl_from_ITfCategoryMgr(iface);
 272     FIXME("STUB:(%p)\n",This);
 273     return E_NOTIMPL;
 274 }
 275
 276 static HRESULT WINAPI CategoryMgr_UnregisterGUIDDescription (
 277         ITfCategoryMgr *iface, REFCLSID rclsid, REFGUID rguid)
 278 {
 279     CategoryMgr *This = impl_from_ITfCategoryMgr(iface);
 280     FIXME("STUB:(%p)\n",This);
 281     return E_NOTIMPL;
 282 }
 283
 284 static HRESULT WINAPI CategoryMgr_GetGUIDDescription ( ITfCategoryMgr *iface,
 285         REFGUID rguid, BSTR *pbstrDesc)
 286 {
 287     CategoryMgr *This = impl_from_ITfCategoryMgr(iface);
 288     FIXME("STUB:(%p)\n",This);
 289     return E_NOTIMPL;
 290 }
 291
 292 static HRESULT WINAPI CategoryMgr_RegisterGUIDDWORD ( ITfCategoryMgr *iface,
 293         REFCLSID rclsid, REFGUID rguid, DWORD dw)
 294 {
 295     CategoryMgr *This = impl_from_ITfCategoryMgr(iface);
 296     FIXME("STUB:(%p)\n",This);
 297     return E_NOTIMPL;
 298 }
 299
 300 static HRESULT WINAPI CategoryMgr_UnregisterGUIDDWORD ( ITfCategoryMgr *iface,
 301         REFCLSID rclsid, REFGUID rguid)
 302 {
 303     CategoryMgr *This = impl_from_ITfCategoryMgr(iface);
 304     FIXME("STUB:(%p)\n",This);
 305     return E_NOTIMPL;
 306 }
 307
 308 static HRESULT WINAPI CategoryMgr_GetGUIDDWORD ( ITfCategoryMgr *iface,
 309         REFGUID rguid, DWORD *pdw)
 310 {
 311     CategoryMgr *This = impl_from_ITfCategoryMgr(iface);
 312     FIXME("STUB:(%p)\n",This);
 313     return E_NOTIMPL;
 314 }
 315
 316 static HRESULT WINAPI CategoryMgr_RegisterGUID ( ITfCategoryMgr *iface,
 317         REFGUID rguid, TfGuidAtom *pguidatom
 318 )
 319 {
 320     DWORD index;
 321     GUID *checkguid;
 322     DWORD id;
 323     CategoryMgr *This = impl_from_ITfCategoryMgr(iface);
 324
 325     TRACE("(%p) %s %p\n",This,debugstr_guid(rguid),pguidatom);
 326
 327     if (!pguidatom)
 328         return E_INVALIDARG;
 329
 330     index = 0;
 331     do {
 332         id = enumerate_Cookie(COOKIE_MAGIC_GUIDATOM,&index);
 333         if (id && IsEqualGUID(rguid,get_Cookie_data(id)))
 334         {
 335             *pguidatom = id;
 336             return S_OK;
 337         }
 338     } while(id);
 339
 340     checkguid = HeapAlloc(GetProcessHeap(),0,sizeof(GUID));
 341     *checkguid = *rguid;
 342     id = generate_Cookie(COOKIE_MAGIC_GUIDATOM,checkguid);
 343
 344     if (!id)
 345     {
 346         HeapFree(GetProcessHeap(),0,checkguid);
 347         return E_FAIL;
 348     }
 349
 350     *pguidatom = id;
 351
 352     return S_OK;
 353 }
 354
 355 static HRESULT WINAPI CategoryMgr_GetGUID ( ITfCategoryMgr *iface,
 356         TfGuidAtom guidatom, GUID *pguid)
 357 {
 358     CategoryMgr *This = impl_from_ITfCategoryMgr(iface);
 359
 360     TRACE("(%p) %i\n",This,guidatom);
 361
 362     if (!pguid)
 363         return E_INVALIDARG;
 364
 365     *pguid = GUID_NULL;
 366
 367     if (get_Cookie_magic(guidatom) == COOKIE_MAGIC_GUIDATOM)
 368         *pguid = *((REFGUID)get_Cookie_data(guidatom));
 369
 370     return S_OK;
 371 }
 372
 373 static HRESULT WINAPI CategoryMgr_IsEqualTfGuidAtom ( ITfCategoryMgr *iface,
 374         TfGuidAtom guidatom, REFGUID rguid, BOOL *pfEqual)
 375 {
 376     CategoryMgr *This = impl_from_ITfCategoryMgr(iface);
 377
 378     TRACE("(%p) %i %s %p\n",This,guidatom,debugstr_guid(rguid),pfEqual);
 379
 380     if (!pfEqual)
 381         return E_INVALIDARG;
 382
 383     *pfEqual = FALSE;
 384     if (get_Cookie_magic(guidatom) == COOKIE_MAGIC_GUIDATOM)
 385     {
 386         if (IsEqualGUID(rguid,get_Cookie_data(guidatom)))
 387             *pfEqual = TRUE;
 388     }
 389
 390     return S_OK;
 391 }
 392
 393
 394 static const ITfCategoryMgrVtbl CategoryMgr_CategoryMgrVtbl =
 395 {
 396     CategoryMgr_QueryInterface,
 397     CategoryMgr_AddRef,
 398     CategoryMgr_Release,
 399
 400     CategoryMgr_RegisterCategory,
 401     CategoryMgr_UnregisterCategory,
 402     CategoryMgr_EnumCategoriesInItem,
 403     CategoryMgr_EnumItemsInCategory,
 404     CategoryMgr_FindClosestCategory,
 405     CategoryMgr_RegisterGUIDDescription,
 406     CategoryMgr_UnregisterGUIDDescription,
 407     CategoryMgr_GetGUIDDescription,
 408     CategoryMgr_RegisterGUIDDWORD,
 409     CategoryMgr_UnregisterGUIDDWORD,
 410     CategoryMgr_GetGUIDDWORD,
 411     CategoryMgr_RegisterGUID,
 412     CategoryMgr_GetGUID,
 413     CategoryMgr_IsEqualTfGuidAtom
 414 };
 415
 416 HRESULT CategoryMgr_Constructor(IUnknown *pUnkOuter, IUnknown **ppOut)
 417 {
 418     CategoryMgr *This;
 419     if (pUnkOuter)
 420         return CLASS_E_NOAGGREGATION;
 421
 422     This = HeapAlloc(GetProcessHeap(),HEAP_ZERO_MEMORY,sizeof(CategoryMgr));
 423     if (This == NULL)
 424         return E_OUTOFMEMORY;
 425
 426     This->ITfCategoryMgr_iface.lpVtbl = &CategoryMgr_CategoryMgrVtbl;
 427     This->refCount = 1;
 428
 429     TRACE("returning %p\n", This);
 430     *ppOut = (IUnknown *)This;
 431     return S_OK;
 432 }