ole32: Make ole2thk.dll into a stand-alone 16-bit module.
[wine] / dlls / ole32 / regsvr.c
1 /*
2  *      self-registerable dll functions for ole32.dll
3  *
4  * Copyright (C) 2003 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., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
19  */
20
21 #include "config.h"
22
23 #include <stdarg.h>
24 #include <string.h>
25
26 #include "windef.h"
27 #include "winbase.h"
28 #include "winuser.h"
29 #include "winreg.h"
30 #include "winerror.h"
31 #include "objbase.h"
32
33 #include "ole2.h"
34 #include "olectl.h"
35 #include "comcat.h"
36 #include "initguid.h"
37 #include "compobj_private.h"
38 #include "moniker.h"
39
40 #include "wine/debug.h"
41 #include "wine/unicode.h"
42
43 WINE_DEFAULT_DEBUG_CHANNEL(ole);
44
45 /*
46  * Near the bottom of this file are the exported DllRegisterServer and
47  * DllUnregisterServer, which make all this worthwhile.
48  */
49
50 /***********************************************************************
51  *              interface for self-registering
52  */
53 struct regsvr_interface
54 {
55     IID const *iid;             /* NULL for end of list */
56     LPCSTR name;                /* can be NULL to omit */
57     IID const *base_iid;        /* can be NULL to omit */
58     int num_methods;            /* can be <0 to omit */
59     CLSID const *ps_clsid;      /* can be NULL to omit */
60     CLSID const *ps_clsid32;    /* can be NULL to omit */
61 };
62
63 static HRESULT register_interfaces(struct regsvr_interface const *list);
64 static HRESULT unregister_interfaces(struct regsvr_interface const *list);
65
66 struct regsvr_coclass
67 {
68     CLSID const *clsid;         /* NULL for end of list */
69     LPCSTR name;                /* can be NULL to omit */
70     LPCSTR ips;                 /* can be NULL to omit */
71     LPCSTR ips32;               /* can be NULL to omit */
72     LPCSTR ips32_tmodel;        /* can be NULL to omit */
73     LPCSTR progid;              /* can be NULL to omit */
74 };
75
76 static HRESULT register_coclasses(struct regsvr_coclass const *list);
77 static HRESULT unregister_coclasses(struct regsvr_coclass const *list);
78
79 /***********************************************************************
80  *              static string constants
81  */
82 static WCHAR const interface_keyname[10] = {
83     'I', 'n', 't', 'e', 'r', 'f', 'a', 'c', 'e', 0 };
84 static WCHAR const base_ifa_keyname[14] = {
85     'B', 'a', 's', 'e', 'I', 'n', 't', 'e', 'r', 'f', 'a', 'c',
86     'e', 0 };
87 static WCHAR const num_methods_keyname[11] = {
88     'N', 'u', 'm', 'M', 'e', 't', 'h', 'o', 'd', 's', 0 };
89 static WCHAR const ps_clsid_keyname[15] = {
90     'P', 'r', 'o', 'x', 'y', 'S', 't', 'u', 'b', 'C', 'l', 's',
91     'i', 'd', 0 };
92 static WCHAR const ps_clsid32_keyname[17] = {
93     'P', 'r', 'o', 'x', 'y', 'S', 't', 'u', 'b', 'C', 'l', 's',
94     'i', 'd', '3', '2', 0 };
95 static WCHAR const clsid_keyname[6] = {
96     'C', 'L', 'S', 'I', 'D', 0 };
97 static WCHAR const ips_keyname[13] = {
98     'I', 'n', 'P', 'r', 'o', 'c', 'S', 'e', 'r', 'v', 'e', 'r',
99     0 };
100 static WCHAR const ips32_keyname[15] = {
101     'I', 'n', 'P', 'r', 'o', 'c', 'S', 'e', 'r', 'v', 'e', 'r',
102     '3', '2', 0 };
103 static WCHAR const progid_keyname[7] = {
104     'P', 'r', 'o', 'g', 'I', 'D', 0 };
105 static char const tmodel_valuename[] = "ThreadingModel";
106
107 /***********************************************************************
108  *              static helper functions
109  */
110 static LONG register_key_guid(HKEY base, WCHAR const *name, GUID const *guid);
111 static LONG register_key_defvalueW(HKEY base, WCHAR const *name,
112                                    WCHAR const *value);
113 static LONG register_key_defvalueA(HKEY base, WCHAR const *name,
114                                    char const *value);
115 static LONG register_progid(WCHAR const *clsid, char const *progid,
116                             char const *name);
117
118 /***********************************************************************
119  *              register_interfaces
120  */
121 static HRESULT register_interfaces(struct regsvr_interface const *list)
122 {
123     LONG res = ERROR_SUCCESS;
124     HKEY interface_key;
125
126     res = RegCreateKeyExW(HKEY_CLASSES_ROOT, interface_keyname, 0, NULL, 0,
127                           KEY_READ | KEY_WRITE, NULL, &interface_key, NULL);
128     if (res != ERROR_SUCCESS) goto error_return;
129
130     for (; res == ERROR_SUCCESS && list->iid; ++list) {
131         WCHAR buf[39];
132         HKEY iid_key;
133
134         StringFromGUID2(list->iid, buf, 39);
135         res = RegCreateKeyExW(interface_key, buf, 0, NULL, 0,
136                               KEY_READ | KEY_WRITE, NULL, &iid_key, NULL);
137         if (res != ERROR_SUCCESS) goto error_close_interface_key;
138
139         if (list->name) {
140             res = RegSetValueExA(iid_key, NULL, 0, REG_SZ,
141                                  (CONST BYTE*)(list->name),
142                                  strlen(list->name) + 1);
143             if (res != ERROR_SUCCESS) goto error_close_iid_key;
144         }
145
146         if (list->base_iid) {
147             res = register_key_guid(iid_key, base_ifa_keyname, list->base_iid);
148             if (res != ERROR_SUCCESS) goto error_close_iid_key;
149         }
150
151         if (0 <= list->num_methods) {
152             static WCHAR const fmt[3] = { '%', 'd', 0 };
153             HKEY key;
154
155             res = RegCreateKeyExW(iid_key, num_methods_keyname, 0, NULL, 0,
156                                   KEY_READ | KEY_WRITE, NULL, &key, NULL);
157             if (res != ERROR_SUCCESS) goto error_close_iid_key;
158
159             sprintfW(buf, fmt, list->num_methods);
160             res = RegSetValueExW(key, NULL, 0, REG_SZ,
161                                  (CONST BYTE*)buf,
162                                  (lstrlenW(buf) + 1) * sizeof(WCHAR));
163             RegCloseKey(key);
164
165             if (res != ERROR_SUCCESS) goto error_close_iid_key;
166         }
167
168         if (list->ps_clsid) {
169             res = register_key_guid(iid_key, ps_clsid_keyname, list->ps_clsid);
170             if (res != ERROR_SUCCESS) goto error_close_iid_key;
171         }
172
173         if (list->ps_clsid32) {
174             res = register_key_guid(iid_key, ps_clsid32_keyname, list->ps_clsid32);
175             if (res != ERROR_SUCCESS) goto error_close_iid_key;
176         }
177
178     error_close_iid_key:
179         RegCloseKey(iid_key);
180     }
181
182 error_close_interface_key:
183     RegCloseKey(interface_key);
184 error_return:
185     return res != ERROR_SUCCESS ? HRESULT_FROM_WIN32(res) : S_OK;
186 }
187
188 /***********************************************************************
189  *              unregister_interfaces
190  */
191 static HRESULT unregister_interfaces(struct regsvr_interface const *list)
192 {
193     LONG res = ERROR_SUCCESS;
194     HKEY interface_key;
195
196     res = RegOpenKeyExW(HKEY_CLASSES_ROOT, interface_keyname, 0,
197                         KEY_READ | KEY_WRITE, &interface_key);
198     if (res == ERROR_FILE_NOT_FOUND) return S_OK;
199     if (res != ERROR_SUCCESS) goto error_return;
200
201     for (; res == ERROR_SUCCESS && list->iid; ++list) {
202         WCHAR buf[39];
203
204         StringFromGUID2(list->iid, buf, 39);
205         res = RegDeleteTreeW(interface_key, buf);
206         if (res == ERROR_FILE_NOT_FOUND) res = ERROR_SUCCESS;
207     }
208
209     RegCloseKey(interface_key);
210 error_return:
211     return res != ERROR_SUCCESS ? HRESULT_FROM_WIN32(res) : S_OK;
212 }
213
214 /***********************************************************************
215  *              register_coclasses
216  */
217 static HRESULT register_coclasses(struct regsvr_coclass const *list)
218 {
219     LONG res = ERROR_SUCCESS;
220     HKEY coclass_key;
221
222     res = RegCreateKeyExW(HKEY_CLASSES_ROOT, clsid_keyname, 0, NULL, 0,
223                           KEY_READ | KEY_WRITE, NULL, &coclass_key, NULL);
224     if (res != ERROR_SUCCESS) goto error_return;
225
226     for (; res == ERROR_SUCCESS && list->clsid; ++list) {
227         WCHAR buf[39];
228         HKEY clsid_key;
229
230         StringFromGUID2(list->clsid, buf, 39);
231         res = RegCreateKeyExW(coclass_key, buf, 0, NULL, 0,
232                               KEY_READ | KEY_WRITE, NULL, &clsid_key, NULL);
233         if (res != ERROR_SUCCESS) goto error_close_coclass_key;
234
235         if (list->name) {
236             res = RegSetValueExA(clsid_key, NULL, 0, REG_SZ,
237                                  (CONST BYTE*)(list->name),
238                                  strlen(list->name) + 1);
239             if (res != ERROR_SUCCESS) goto error_close_clsid_key;
240         }
241
242         if (list->ips) {
243             res = register_key_defvalueA(clsid_key, ips_keyname, list->ips);
244             if (res != ERROR_SUCCESS) goto error_close_clsid_key;
245         }
246
247         if (list->ips32) {
248             HKEY ips32_key;
249
250             res = RegCreateKeyExW(clsid_key, ips32_keyname, 0, NULL, 0,
251                                   KEY_READ | KEY_WRITE, NULL,
252                                   &ips32_key, NULL);
253             if (res != ERROR_SUCCESS) goto error_close_clsid_key;
254
255             res = RegSetValueExA(ips32_key, NULL, 0, REG_SZ,
256                                  (CONST BYTE*)list->ips32,
257                                  lstrlenA(list->ips32) + 1);
258             if (res == ERROR_SUCCESS && list->ips32_tmodel)
259                 res = RegSetValueExA(ips32_key, tmodel_valuename, 0, REG_SZ,
260                                      (CONST BYTE*)list->ips32_tmodel,
261                                      strlen(list->ips32_tmodel) + 1);
262             RegCloseKey(ips32_key);
263             if (res != ERROR_SUCCESS) goto error_close_clsid_key;
264         }
265
266         if (list->progid) {
267             res = register_key_defvalueA(clsid_key, progid_keyname,
268                                          list->progid);
269             if (res != ERROR_SUCCESS) goto error_close_clsid_key;
270
271             res = register_progid(buf, list->progid, list->name);
272             if (res != ERROR_SUCCESS) goto error_close_clsid_key;
273         }
274
275     error_close_clsid_key:
276         RegCloseKey(clsid_key);
277     }
278
279 error_close_coclass_key:
280     RegCloseKey(coclass_key);
281 error_return:
282     return res != ERROR_SUCCESS ? HRESULT_FROM_WIN32(res) : S_OK;
283 }
284
285 /***********************************************************************
286  *              unregister_coclasses
287  */
288 static HRESULT unregister_coclasses(struct regsvr_coclass const *list)
289 {
290     LONG res = ERROR_SUCCESS;
291     HKEY coclass_key;
292
293     res = RegOpenKeyExW(HKEY_CLASSES_ROOT, clsid_keyname, 0,
294                         KEY_READ | KEY_WRITE, &coclass_key);
295     if (res == ERROR_FILE_NOT_FOUND) return S_OK;
296     if (res != ERROR_SUCCESS) goto error_return;
297
298     for (; res == ERROR_SUCCESS && list->clsid; ++list) {
299         WCHAR buf[39];
300
301         StringFromGUID2(list->clsid, buf, 39);
302         res = RegDeleteTreeW(coclass_key, buf);
303         if (res == ERROR_FILE_NOT_FOUND) res = ERROR_SUCCESS;
304         if (res != ERROR_SUCCESS) goto error_close_coclass_key;
305
306         if (list->progid) {
307             res = RegDeleteTreeA(HKEY_CLASSES_ROOT, list->progid);
308             if (res == ERROR_FILE_NOT_FOUND) res = ERROR_SUCCESS;
309             if (res != ERROR_SUCCESS) goto error_close_coclass_key;
310         }
311     }
312
313 error_close_coclass_key:
314     RegCloseKey(coclass_key);
315 error_return:
316     return res != ERROR_SUCCESS ? HRESULT_FROM_WIN32(res) : S_OK;
317 }
318
319 /***********************************************************************
320  *              regsvr_key_guid
321  */
322 static LONG register_key_guid(HKEY base, WCHAR const *name, GUID const *guid)
323 {
324     WCHAR buf[39];
325
326     StringFromGUID2(guid, buf, 39);
327     return register_key_defvalueW(base, name, buf);
328 }
329
330 /***********************************************************************
331  *              regsvr_key_defvalueW
332  */
333 static LONG register_key_defvalueW(
334     HKEY base,
335     WCHAR const *name,
336     WCHAR const *value)
337 {
338     LONG res;
339     HKEY key;
340
341     res = RegCreateKeyExW(base, name, 0, NULL, 0,
342                           KEY_READ | KEY_WRITE, NULL, &key, NULL);
343     if (res != ERROR_SUCCESS) return res;
344     res = RegSetValueExW(key, NULL, 0, REG_SZ, (CONST BYTE*)value,
345                          (lstrlenW(value) + 1) * sizeof(WCHAR));
346     RegCloseKey(key);
347     return res;
348 }
349
350 /***********************************************************************
351  *              regsvr_key_defvalueA
352  */
353 static LONG register_key_defvalueA(
354     HKEY base,
355     WCHAR const *name,
356     char const *value)
357 {
358     LONG res;
359     HKEY key;
360
361     res = RegCreateKeyExW(base, name, 0, NULL, 0,
362                           KEY_READ | KEY_WRITE, NULL, &key, NULL);
363     if (res != ERROR_SUCCESS) return res;
364     res = RegSetValueExA(key, NULL, 0, REG_SZ, (CONST BYTE*)value,
365                          lstrlenA(value) + 1);
366     RegCloseKey(key);
367     return res;
368 }
369
370 /***********************************************************************
371  *              regsvr_progid
372  */
373 static LONG register_progid(
374     WCHAR const *clsid,
375     char const *progid,
376     char const *name)
377 {
378     LONG res;
379     HKEY progid_key;
380
381     res = RegCreateKeyExA(HKEY_CLASSES_ROOT, progid, 0,
382                           NULL, 0, KEY_READ | KEY_WRITE, NULL,
383                           &progid_key, NULL);
384     if (res != ERROR_SUCCESS) return res;
385
386     if (name) {
387         res = RegSetValueExA(progid_key, NULL, 0, REG_SZ,
388                              (CONST BYTE*)name, strlen(name) + 1);
389         if (res != ERROR_SUCCESS) goto error_close_progid_key;
390     }
391
392     if (clsid) {
393         res = register_key_defvalueW(progid_key, clsid_keyname, clsid);
394         if (res != ERROR_SUCCESS) goto error_close_progid_key;
395     }
396
397 error_close_progid_key:
398     RegCloseKey(progid_key);
399     return res;
400 }
401
402 /***********************************************************************
403  *              coclass list
404  */
405 static GUID const CLSID_StdOleLink = {
406     0x00000300, 0x0000, 0x0000, {0xC0,0x00,0x00,0x00,0x00,0x00,0x00,0x46} };
407
408 static GUID const CLSID_PackagerMoniker = {
409     0x00000308, 0x0000, 0x0000, {0xC0,0x00,0x00,0x00,0x00,0x00,0x00,0x46} };
410
411 static GUID const CLSID_PSFactoryBuffer_actxprxy = {
412     0xB8DA6310, 0xE19B, 0x11D0, {0x93,0x3C,0x00,0xA0,0xC9,0x0D,0xCA,0xA9} };
413
414 extern GUID const CLSID_Picture_Metafile;
415 extern GUID const CLSID_Picture_Dib;
416
417 static struct regsvr_coclass const coclass_list[] = {
418     {   &CLSID_StdOleLink,
419         "StdOleLink",
420         NULL,
421         "ole32.dll",
422         NULL
423     },
424     {   &CLSID_FileMoniker,
425         "FileMoniker",
426         NULL,
427         "ole32.dll",
428         "Both",
429         "file"
430     },
431     {   &CLSID_ItemMoniker,
432         "ItemMoniker",
433         NULL,
434         "ole32.dll",
435         "Both"
436     },
437     {   &CLSID_AntiMoniker,
438         "AntiMoniker",
439         NULL,
440         "ole32.dll",
441         "Both"
442     },
443     {   &CLSID_PointerMoniker,
444         "PointerMoniker",
445         NULL,
446         "ole32.dll",
447         "Both"
448     },
449     {   &CLSID_PackagerMoniker,
450         "PackagerMoniker",
451         NULL,
452         "ole32.dll",
453         "Both"
454     },
455     {   &CLSID_CompositeMoniker,
456         "CompositeMoniker",
457         NULL,
458         "ole32.dll",
459         "Both"
460     },
461     {   &CLSID_DfMarshal,
462         "DfMarshal",
463         NULL,
464         "ole32.dll",
465         "Both"
466     },
467     {   &CLSID_Picture_Metafile,
468         "Picture (Metafile)",
469         NULL,
470         "ole32.dll",
471         NULL,
472         "StaticMetafile"
473     },
474     {   &CLSID_Picture_Dib,
475         "Picture (Device Independent Bitmap)",
476         NULL,
477         "ole32.dll",
478         NULL,
479         "StaticDib"
480     },
481     {   &CLSID_ClassMoniker,
482         "ClassMoniker",
483         NULL,
484         "ole32.dll",
485         "Both",
486         "CLSID"
487     },
488     {   &CLSID_PSFactoryBuffer,
489         "PSFactoryBuffer",
490         NULL,
491         "ole32.dll",
492         "Both"
493     },
494     {   &CLSID_StdGlobalInterfaceTable,
495         "StdGlobalInterfaceTable",
496         NULL,
497         "ole32.dll",
498         "Apartment"
499     },
500     {   &CLSID_StdComponentCategoriesMgr,
501         "Component Categories Manager",
502         NULL,
503         "ole32.dll",
504         "Both"
505     },
506     { NULL }                    /* list terminator */
507 };
508
509 /***********************************************************************
510  *              interface list
511  */
512
513 #define INTERFACE_ENTRY(interface, base, clsid32) { &IID_##interface, #interface, base, sizeof(interface##Vtbl)/sizeof(void*), NULL, clsid32 }
514 #define BAS_INTERFACE_ENTRY(interface, base) INTERFACE_ENTRY(interface, &IID_##base, NULL)
515 #define ACTX_INTERFACE_ENTRY(interface) INTERFACE_ENTRY(interface, NULL, &CLSID_PSFactoryBuffer_actxprxy)
516 #define LCL_INTERFACE_ENTRY(interface) INTERFACE_ENTRY(interface, NULL, NULL)
517
518 static const struct regsvr_interface interface_list[] = {
519     LCL_INTERFACE_ENTRY(IUnknown),
520     LCL_INTERFACE_ENTRY(IMalloc),
521     LCL_INTERFACE_ENTRY(IMarshal),
522     BAS_INTERFACE_ENTRY(IMoniker, IPersistStream),
523     LCL_INTERFACE_ENTRY(IMessageFilter),
524     LCL_INTERFACE_ENTRY(IStdMarshalInfo),
525     LCL_INTERFACE_ENTRY(IExternalConnection),
526     LCL_INTERFACE_ENTRY(IMallocSpy),
527     LCL_INTERFACE_ENTRY(IMultiQI),
528     BAS_INTERFACE_ENTRY(IPersistStream, IPersist),
529     BAS_INTERFACE_ENTRY(IPersistStorage, IPersist),
530     BAS_INTERFACE_ENTRY(IPersistFile, IPersist),
531     LCL_INTERFACE_ENTRY(IDataAdviseHolder),
532     LCL_INTERFACE_ENTRY(IOleAdviseHolder),
533     BAS_INTERFACE_ENTRY(IOleInPlaceObject, IOleWindow),
534     BAS_INTERFACE_ENTRY(IOleInPlaceUIWindow, IOleWindow),
535     BAS_INTERFACE_ENTRY(IOleInPlaceActiveObject, IOleWindow),
536     BAS_INTERFACE_ENTRY(IOleInPlaceSite, IOleWindow),
537     BAS_INTERFACE_ENTRY(IOleContainer, IParseDisplayName),
538     BAS_INTERFACE_ENTRY(IOleItemContainer, IOleContainer),
539     LCL_INTERFACE_ENTRY(IDropSource),
540     BAS_INTERFACE_ENTRY(IAdviseSink2, IAdviseSink),
541     BAS_INTERFACE_ENTRY(IViewObject2, IViewObject),
542     BAS_INTERFACE_ENTRY(IOleCache2, IOleCache),
543     LCL_INTERFACE_ENTRY(IClientSecurity),
544     LCL_INTERFACE_ENTRY(IServerSecurity),
545     ACTX_INTERFACE_ENTRY(IEnumGUID),
546     ACTX_INTERFACE_ENTRY(IEnumCATEGORYINFO),
547     ACTX_INTERFACE_ENTRY(ICatRegister),
548     ACTX_INTERFACE_ENTRY(ICatInformation),
549     { NULL }                    /* list terminator */
550 };
551
552 /***********************************************************************
553  *              DllRegisterServer (OLE32.@)
554  */
555 HRESULT WINAPI DllRegisterServer(void)
556 {
557     HRESULT hr;
558
559     TRACE("\n");
560
561     hr = OLE32_DllRegisterServer();
562     if (SUCCEEDED(hr))
563         hr = register_coclasses(coclass_list);
564     if (SUCCEEDED(hr))
565         hr = register_interfaces(interface_list);
566     return hr;
567 }
568
569 /***********************************************************************
570  *              DllUnregisterServer (OLE32.@)
571  */
572 HRESULT WINAPI DllUnregisterServer(void)
573 {
574     HRESULT hr;
575
576     TRACE("\n");
577
578     hr = unregister_coclasses(coclass_list);
579     if (SUCCEEDED(hr))
580         hr = unregister_interfaces(interface_list);
581     if (SUCCEEDED(hr))
582         hr = OLE32_DllUnregisterServer();
583     return hr;
584 }