dlls/comcat/register.c

   1 /*
   2  *      ComCatMgr ICatRegister implementation for comcat.dll
   3  *
   4  * Copyright (C) 2002 John K. Hohm
   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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19  */
  20
  21 #include "comcat_private.h"
  22
  23 #include "wine/debug.h"
  24
  25 WINE_DEFAULT_DEBUG_CHANNEL(ole);
  26
  27 /**********************************************************************
  28  * File-scope string constants
  29  */
  30 static const WCHAR comcat_keyname[21] = {
  31     'C', 'o', 'm', 'p', 'o', 'n', 'e', 'n', 't', ' ', 'C', 'a',
  32     't', 'e', 'g', 'o', 'r', 'i', 'e', 's', 0 };
  33 static const WCHAR impl_keyname[23] = {
  34     'I', 'm', 'p', 'l', 'e', 'm', 'e', 'n',
  35     't', 'e', 'd', ' ', 'C', 'a', 't', 'e',
  36     'g', 'o', 'r', 'i', 'e', 's', 0 };
  37 static const WCHAR req_keyname[20] = {
  38     'R', 'e', 'q', 'u', 'i', 'r', 'e', 'd',
  39     ' ', 'C', 'a', 't', 'e', 'g', 'o', 'r',
  40     'i', 'e', 's', 0 };
  41
  42 static HRESULT COMCAT_RegisterClassCategories(
  43     REFCLSID rclsid, LPCWSTR type,
  44     ULONG cCategories, CATID *rgcatid);
  45 static HRESULT COMCAT_UnRegisterClassCategories(
  46     REFCLSID rclsid, LPCWSTR type,
  47     ULONG cCategories, CATID *rgcatid);
  48
  49 /**********************************************************************
  50  * COMCAT_ICatRegister_QueryInterface
  51  */
  52 static HRESULT WINAPI COMCAT_ICatRegister_QueryInterface(
  53     LPCATREGISTER iface,
  54     REFIID riid,
  55     LPVOID *ppvObj)
  56 {
  57     ICOM_THIS_MULTI(ComCatMgrImpl, regVtbl, iface);
  58     TRACE("\n\tIID:\t%s\n",debugstr_guid(riid));
  59
  60     if (This == NULL || ppvObj == NULL) return E_POINTER;
  61
  62     return IUnknown_QueryInterface((LPUNKNOWN)&This->unkVtbl, riid, ppvObj);
  63 }
  64
  65 /**********************************************************************
  66  * COMCAT_ICatRegister_AddRef
  67  */
  68 static ULONG WINAPI COMCAT_ICatRegister_AddRef(LPCATREGISTER iface)
  69 {
  70     ICOM_THIS_MULTI(ComCatMgrImpl, regVtbl, iface);
  71     TRACE("\n");
  72
  73     if (This == NULL) return E_POINTER;
  74
  75     return IUnknown_AddRef((LPUNKNOWN)&This->unkVtbl);
  76 }
  77
  78 /**********************************************************************
  79  * COMCAT_ICatRegister_Release
  80  */
  81 static ULONG WINAPI COMCAT_ICatRegister_Release(LPCATREGISTER iface)
  82 {
  83     ICOM_THIS_MULTI(ComCatMgrImpl, regVtbl, iface);
  84     TRACE("\n");
  85
  86     if (This == NULL) return E_POINTER;
  87
  88     return IUnknown_Release((LPUNKNOWN)&This->unkVtbl);
  89 }
  90
  91 /**********************************************************************
  92  * COMCAT_ICatRegister_RegisterCategories
  93  */
  94 static HRESULT WINAPI COMCAT_ICatRegister_RegisterCategories(
  95     LPCATREGISTER iface,
  96     ULONG cCategories,
  97     CATEGORYINFO *rgci)
  98 {
  99 /*     ICOM_THIS_MULTI(ComCatMgrImpl, regVtbl, iface); */
 100     HKEY comcat_key;
 101     HRESULT res;
 102
 103     TRACE("\n");
 104
 105     if (iface == NULL || (cCategories && rgci == NULL))
 106         return E_POINTER;
 107
 108     /* Create (or open) the component categories key. */
 109     res = RegCreateKeyExW(HKEY_CLASSES_ROOT, comcat_keyname, 0, NULL, 0,
 110                           KEY_READ | KEY_WRITE, NULL, &comcat_key, NULL);
 111     if (res != ERROR_SUCCESS) return E_FAIL;
 112
 113     for (; cCategories; --cCategories, ++rgci) {
 114         WCHAR fmt[4] = { '%', 'l', 'X', 0 };
 115         WCHAR keyname[39];
 116         WCHAR valname[9];
 117         HKEY cat_key;
 118
 119         /* Create (or open) the key for this category. */
 120         if (!StringFromGUID2(&rgci->catid, keyname, 39)) continue;
 121         res = RegCreateKeyExW(comcat_key, keyname, 0, NULL, 0,
 122                               KEY_READ | KEY_WRITE, NULL, &cat_key, NULL);
 123         if (res != ERROR_SUCCESS) continue;
 124
 125         /* Set the value for this locale's description. */
 126         wsprintfW(valname, fmt, rgci->lcid);
 127         RegSetValueExW(cat_key, valname, 0, REG_SZ,
 128                        (CONST BYTE*)(rgci->szDescription),
 129                        (lstrlenW(rgci->szDescription) + 1) * sizeof(WCHAR));
 130
 131         RegCloseKey(cat_key);
 132     }
 133
 134     RegCloseKey(comcat_key);
 135     return S_OK;
 136 }
 137
 138 /**********************************************************************
 139  * COMCAT_ICatRegister_UnRegisterCategories
 140  */
 141 static HRESULT WINAPI COMCAT_ICatRegister_UnRegisterCategories(
 142     LPCATREGISTER iface,
 143     ULONG cCategories,
 144     CATID *rgcatid)
 145 {
 146 /*     ICOM_THIS_MULTI(ComCatMgrImpl, regVtbl, iface); */
 147     HKEY comcat_key;
 148     HRESULT res;
 149
 150     TRACE("\n");
 151
 152     if (iface == NULL || (cCategories && rgcatid == NULL))
 153         return E_POINTER;
 154
 155     /* Open the component categories key. */
 156     res = RegOpenKeyExW(HKEY_CLASSES_ROOT, comcat_keyname, 0,
 157                         KEY_READ | KEY_WRITE, &comcat_key);
 158     if (res != ERROR_SUCCESS) return E_FAIL;
 159
 160     for (; cCategories; --cCategories, ++rgcatid) {
 161         WCHAR keyname[39];
 162
 163         /* Delete the key for this category. */
 164         if (!StringFromGUID2(rgcatid, keyname, 39)) continue;
 165         RegDeleteKeyW(comcat_key, keyname);
 166     }
 167
 168     RegCloseKey(comcat_key);
 169     return S_OK;
 170 }
 171
 172 /**********************************************************************
 173  * COMCAT_ICatRegister_RegisterClassImplCategories
 174  */
 175 static HRESULT WINAPI COMCAT_ICatRegister_RegisterClassImplCategories(
 176     LPCATREGISTER iface,
 177     REFCLSID rclsid,
 178     ULONG cCategories,
 179     CATID *rgcatid)
 180 {
 181 /*     ICOM_THIS_MULTI(ComCatMgrImpl, regVtbl, iface); */
 182     TRACE("\n");
 183
 184     return COMCAT_RegisterClassCategories(
 185         rclsid, impl_keyname, cCategories, rgcatid);
 186 }
 187
 188 /**********************************************************************
 189  * COMCAT_ICatRegister_UnRegisterClassImplCategories
 190  */
 191 static HRESULT WINAPI COMCAT_ICatRegister_UnRegisterClassImplCategories(
 192     LPCATREGISTER iface,
 193     REFCLSID rclsid,
 194     ULONG cCategories,
 195     CATID *rgcatid)
 196 {
 197 /*     ICOM_THIS_MULTI(ComCatMgrImpl, regVtbl, iface); */
 198     TRACE("\n");
 199
 200     return COMCAT_UnRegisterClassCategories(
 201         rclsid, impl_keyname, cCategories, rgcatid);
 202 }
 203
 204 /**********************************************************************
 205  * COMCAT_ICatRegister_RegisterClassReqCategories
 206  */
 207 static HRESULT WINAPI COMCAT_ICatRegister_RegisterClassReqCategories(
 208     LPCATREGISTER iface,
 209     REFCLSID rclsid,
 210     ULONG cCategories,
 211     CATID *rgcatid)
 212 {
 213 /*     ICOM_THIS_MULTI(ComCatMgrImpl, regVtbl, iface); */
 214     TRACE("\n");
 215
 216     return COMCAT_RegisterClassCategories(
 217         rclsid, req_keyname, cCategories, rgcatid);
 218 }
 219
 220 /**********************************************************************
 221  * COMCAT_ICatRegister_UnRegisterClassReqCategories
 222  */
 223 static HRESULT WINAPI COMCAT_ICatRegister_UnRegisterClassReqCategories(
 224     LPCATREGISTER iface,
 225     REFCLSID rclsid,
 226     ULONG cCategories,
 227     CATID *rgcatid)
 228 {
 229 /*     ICOM_THIS_MULTI(ComCatMgrImpl, regVtbl, iface); */
 230     TRACE("\n");
 231
 232     return COMCAT_UnRegisterClassCategories(
 233         rclsid, req_keyname, cCategories, rgcatid);
 234 }
 235
 236 /**********************************************************************
 237  * COMCAT_ICatRegister_Vtbl
 238  */
 239 ICatRegisterVtbl COMCAT_ICatRegister_Vtbl =
 240 {
 241     COMCAT_ICatRegister_QueryInterface,
 242     COMCAT_ICatRegister_AddRef,
 243     COMCAT_ICatRegister_Release,
 244     COMCAT_ICatRegister_RegisterCategories,
 245     COMCAT_ICatRegister_UnRegisterCategories,
 246     COMCAT_ICatRegister_RegisterClassImplCategories,
 247     COMCAT_ICatRegister_UnRegisterClassImplCategories,
 248     COMCAT_ICatRegister_RegisterClassReqCategories,
 249     COMCAT_ICatRegister_UnRegisterClassReqCategories
 250 };
 251
 252 /**********************************************************************
 253  * COMCAT_RegisterClassCategories
 254  */
 255 static HRESULT COMCAT_RegisterClassCategories(
 256     REFCLSID rclsid,
 257     LPCWSTR type,
 258     ULONG cCategories,
 259     CATID *rgcatid)
 260 {
 261     WCHAR keyname[39];
 262     HRESULT res;
 263     HKEY clsid_key, class_key, type_key;
 264
 265     if (cCategories && rgcatid == NULL) return E_POINTER;
 266
 267     /* Format the class key name. */
 268     res = StringFromGUID2(rclsid, keyname, 39);
 269     if (FAILED(res)) return res;
 270
 271     /* Create (or open) the CLSID key. */
 272     res = RegCreateKeyExW(HKEY_CLASSES_ROOT, clsid_keyname, 0, NULL, 0,
 273                           KEY_READ | KEY_WRITE, NULL, &clsid_key, NULL);
 274     if (res != ERROR_SUCCESS) return E_FAIL;
 275
 276     /* Create (or open) the class key. */
 277     res = RegCreateKeyExW(clsid_key, keyname, 0, NULL, 0,
 278                           KEY_READ | KEY_WRITE, NULL, &class_key, NULL);
 279     if (res == ERROR_SUCCESS) {
 280         /* Create (or open) the category type key. */
 281         res = RegCreateKeyExW(class_key, type, 0, NULL, 0,
 282                               KEY_READ | KEY_WRITE, NULL, &type_key, NULL);
 283         if (res == ERROR_SUCCESS) {
 284             for (; cCategories; --cCategories, ++rgcatid) {
 285                 HKEY key;
 286
 287                 /* Format the category key name. */
 288                 res = StringFromGUID2(rgcatid, keyname, 39);
 289                 if (FAILED(res)) continue;
 290
 291                 /* Do the register. */
 292                 res = RegCreateKeyExW(type_key, keyname, 0, NULL, 0,
 293                                       KEY_READ | KEY_WRITE, NULL, &key, NULL);
 294                 if (res == ERROR_SUCCESS) RegCloseKey(key);
 295             }
 296             res = S_OK;
 297         } else res = E_FAIL;
 298         RegCloseKey(class_key);
 299     } else res = E_FAIL;
 300     RegCloseKey(clsid_key);
 301
 302     return res;
 303 }
 304
 305 /**********************************************************************
 306  * COMCAT_UnRegisterClassCategories
 307  */
 308 static HRESULT COMCAT_UnRegisterClassCategories(
 309     REFCLSID rclsid,
 310     LPCWSTR type,
 311     ULONG cCategories,
 312     CATID *rgcatid)
 313 {
 314     WCHAR keyname[68] = { 'C', 'L', 'S', 'I', 'D', '\\' };
 315     HRESULT res;
 316     HKEY type_key;
 317
 318     if (cCategories && rgcatid == NULL) return E_POINTER;
 319
 320     /* Format the class category type key name. */
 321     res = StringFromGUID2(rclsid, keyname + 6, 39);
 322     if (FAILED(res)) return res;
 323     keyname[44] = '\\';
 324     lstrcpyW(keyname + 45, type);
 325
 326     /* Open the class category type key. */
 327     res = RegOpenKeyExW(HKEY_CLASSES_ROOT, keyname, 0,
 328                         KEY_READ | KEY_WRITE, &type_key);
 329     if (res != ERROR_SUCCESS) return E_FAIL;
 330
 331     for (; cCategories; --cCategories, ++rgcatid) {
 332         /* Format the category key name. */
 333         res = StringFromGUID2(rgcatid, keyname, 39);
 334         if (FAILED(res)) continue;
 335
 336         /* Do the unregister. */
 337         RegDeleteKeyW(type_key, keyname);
 338     }
 339     RegCloseKey(type_key);
 340
 341     return S_OK;
 342 }