2 * self-registerable dll functions for itss.dll
4 * Copyright (C) 2003 John K. Hohm
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.
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.
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
31 #include "wine/itss.h"
32 #include "wine/debug.h"
34 WINE_DEFAULT_DEBUG_CHANNEL(itss
);
37 * Near the bottom of this file are the exported DllRegisterServer and
38 * DllUnregisterServer, which make all this worthwhile.
41 /***********************************************************************
42 * interface for self-registering
44 struct regsvr_interface
46 IID
const *iid
; /* NULL for end of list */
47 LPCSTR name
; /* can be NULL to omit */
48 IID
const *base_iid
; /* can be NULL to omit */
49 int num_methods
; /* can be <0 to omit */
50 CLSID
const *ps_clsid
; /* can be NULL to omit */
51 CLSID
const *ps_clsid32
; /* can be NULL to omit */
54 static HRESULT
register_interfaces(struct regsvr_interface
const *list
);
55 static HRESULT
unregister_interfaces(struct regsvr_interface
const *list
);
59 CLSID
const *clsid
; /* NULL for end of list */
60 LPCSTR name
; /* can be NULL to omit */
61 LPCSTR ips
; /* can be NULL to omit */
62 LPCSTR ips32
; /* can be NULL to omit */
63 LPCSTR ips32_tmodel
; /* can be NULL to omit */
64 LPCSTR clsid_str
; /* can be NULL to omit */
65 LPCSTR progid
; /* can be NULL to omit */
68 static HRESULT
register_coclasses(struct regsvr_coclass
const *list
);
69 static HRESULT
unregister_coclasses(struct regsvr_coclass
const *list
);
71 /***********************************************************************
72 * static string constants
74 static WCHAR
const interface_keyname
[10] = {
75 'I', 'n', 't', 'e', 'r', 'f', 'a', 'c', 'e', 0 };
76 static WCHAR
const base_ifa_keyname
[14] = {
77 'B', 'a', 's', 'e', 'I', 'n', 't', 'e', 'r', 'f', 'a', 'c',
79 static WCHAR
const num_methods_keyname
[11] = {
80 'N', 'u', 'm', 'M', 'e', 't', 'h', 'o', 'd', 's', 0 };
81 static WCHAR
const ps_clsid_keyname
[15] = {
82 'P', 'r', 'o', 'x', 'y', 'S', 't', 'u', 'b', 'C', 'l', 's',
84 static WCHAR
const ps_clsid32_keyname
[17] = {
85 'P', 'r', 'o', 'x', 'y', 'S', 't', 'u', 'b', 'C', 'l', 's',
86 'i', 'd', '3', '2', 0 };
87 static WCHAR
const clsid_keyname
[6] = {
88 'C', 'L', 'S', 'I', 'D', 0 };
89 static WCHAR
const ips_keyname
[13] = {
90 'I', 'n', 'P', 'r', 'o', 'c', 'S', 'e', 'r', 'v', 'e', 'r',
92 static WCHAR
const ips32_keyname
[15] = {
93 'I', 'n', 'P', 'r', 'o', 'c', 'S', 'e', 'r', 'v', 'e', 'r',
95 static WCHAR
const progid_keyname
[7] = {
96 'P', 'r', 'o', 'g', 'I', 'D', 0 };
97 static char const tmodel_valuename
[] = "ThreadingModel";
99 /***********************************************************************
100 * static helper functions
102 static LONG
register_key_guid(HKEY base
, WCHAR
const *name
, GUID
const *guid
);
103 static LONG
register_key_defvalueW(HKEY base
, WCHAR
const *name
,
105 static LONG
register_key_defvalueA(HKEY base
, WCHAR
const *name
,
107 static LONG
recursive_delete_key(HKEY key
);
108 static LONG
recursive_delete_keyA(HKEY base
, char const *name
);
109 static LONG
recursive_delete_keyW(HKEY base
, WCHAR
const *name
);
111 /***********************************************************************
112 * register_interfaces
114 static HRESULT
register_interfaces(struct regsvr_interface
const *list
)
116 LONG res
= ERROR_SUCCESS
;
119 res
= RegCreateKeyExW(HKEY_CLASSES_ROOT
, interface_keyname
, 0, NULL
, 0,
120 KEY_READ
| KEY_WRITE
, NULL
, &interface_key
, NULL
);
121 if (res
!= ERROR_SUCCESS
) goto error_return
;
123 for (; res
== ERROR_SUCCESS
&& list
->iid
; ++list
) {
127 StringFromGUID2(list
->iid
, buf
, 39);
128 res
= RegCreateKeyExW(interface_key
, buf
, 0, NULL
, 0,
129 KEY_READ
| KEY_WRITE
, NULL
, &iid_key
, NULL
);
130 if (res
!= ERROR_SUCCESS
) goto error_close_interface_key
;
133 res
= RegSetValueExA(iid_key
, NULL
, 0, REG_SZ
,
134 (CONST BYTE
*)(list
->name
),
135 strlen(list
->name
) + 1);
136 if (res
!= ERROR_SUCCESS
) goto error_close_iid_key
;
139 if (list
->base_iid
) {
140 register_key_guid(iid_key
, base_ifa_keyname
, list
->base_iid
);
141 if (res
!= ERROR_SUCCESS
) goto error_close_iid_key
;
144 if (0 <= list
->num_methods
) {
145 static WCHAR
const fmt
[3] = { '%', 'd', 0 };
148 res
= RegCreateKeyExW(iid_key
, num_methods_keyname
, 0, NULL
, 0,
149 KEY_READ
| KEY_WRITE
, NULL
, &key
, NULL
);
150 if (res
!= ERROR_SUCCESS
) goto error_close_iid_key
;
152 wsprintfW(buf
, fmt
, list
->num_methods
);
153 res
= RegSetValueExW(key
, NULL
, 0, REG_SZ
,
155 (lstrlenW(buf
) + 1) * sizeof(WCHAR
));
158 if (res
!= ERROR_SUCCESS
) goto error_close_iid_key
;
161 if (list
->ps_clsid
) {
162 register_key_guid(iid_key
, ps_clsid_keyname
, list
->ps_clsid
);
163 if (res
!= ERROR_SUCCESS
) goto error_close_iid_key
;
166 if (list
->ps_clsid32
) {
167 register_key_guid(iid_key
, ps_clsid32_keyname
, list
->ps_clsid32
);
168 if (res
!= ERROR_SUCCESS
) goto error_close_iid_key
;
172 RegCloseKey(iid_key
);
175 error_close_interface_key
:
176 RegCloseKey(interface_key
);
178 return res
!= ERROR_SUCCESS
? HRESULT_FROM_WIN32(res
) : S_OK
;
181 /***********************************************************************
182 * unregister_interfaces
184 static HRESULT
unregister_interfaces(struct regsvr_interface
const *list
)
186 LONG res
= ERROR_SUCCESS
;
189 res
= RegOpenKeyExW(HKEY_CLASSES_ROOT
, interface_keyname
, 0,
190 KEY_READ
| KEY_WRITE
, &interface_key
);
191 if (res
== ERROR_FILE_NOT_FOUND
) return S_OK
;
192 if (res
!= ERROR_SUCCESS
) goto error_return
;
194 for (; res
== ERROR_SUCCESS
&& list
->iid
; ++list
) {
197 StringFromGUID2(list
->iid
, buf
, 39);
198 res
= recursive_delete_keyW(interface_key
, buf
);
201 RegCloseKey(interface_key
);
203 return res
!= ERROR_SUCCESS
? HRESULT_FROM_WIN32(res
) : S_OK
;
206 /***********************************************************************
209 static HRESULT
register_coclasses(struct regsvr_coclass
const *list
)
211 LONG res
= ERROR_SUCCESS
;
214 res
= RegCreateKeyExW(HKEY_CLASSES_ROOT
, clsid_keyname
, 0, NULL
, 0,
215 KEY_READ
| KEY_WRITE
, NULL
, &coclass_key
, NULL
);
216 if (res
!= ERROR_SUCCESS
) goto error_return
;
218 for (; res
== ERROR_SUCCESS
&& list
->clsid
; ++list
) {
222 StringFromGUID2(list
->clsid
, buf
, 39);
223 res
= RegCreateKeyExW(coclass_key
, buf
, 0, NULL
, 0,
224 KEY_READ
| KEY_WRITE
, NULL
, &clsid_key
, NULL
);
225 if (res
!= ERROR_SUCCESS
) goto error_close_coclass_key
;
228 res
= RegSetValueExA(clsid_key
, NULL
, 0, REG_SZ
,
229 (CONST BYTE
*)(list
->name
),
230 strlen(list
->name
) + 1);
231 if (res
!= ERROR_SUCCESS
) goto error_close_clsid_key
;
235 res
= register_key_defvalueA(clsid_key
, ips_keyname
, list
->ips
);
236 if (res
!= ERROR_SUCCESS
) goto error_close_clsid_key
;
242 res
= RegCreateKeyExW(clsid_key
, ips32_keyname
, 0, NULL
, 0,
243 KEY_READ
| KEY_WRITE
, NULL
,
245 if (res
!= ERROR_SUCCESS
) goto error_close_clsid_key
;
247 res
= RegSetValueExA(ips32_key
, NULL
, 0, REG_SZ
,
248 (CONST BYTE
*)list
->ips32
,
249 lstrlenA(list
->ips32
) + 1);
250 if (res
== ERROR_SUCCESS
&& list
->ips32_tmodel
)
251 res
= RegSetValueExA(ips32_key
, tmodel_valuename
, 0, REG_SZ
,
252 (CONST BYTE
*)list
->ips32_tmodel
,
253 strlen(list
->ips32_tmodel
) + 1);
254 RegCloseKey(ips32_key
);
255 if (res
!= ERROR_SUCCESS
) goto error_close_clsid_key
;
258 if (list
->clsid_str
) {
259 res
= register_key_defvalueA(clsid_key
, clsid_keyname
,
261 if (res
!= ERROR_SUCCESS
) goto error_close_clsid_key
;
267 res
= register_key_defvalueA(clsid_key
, progid_keyname
,
269 if (res
!= ERROR_SUCCESS
) goto error_close_clsid_key
;
271 res
= RegCreateKeyExA(HKEY_CLASSES_ROOT
, list
->progid
, 0,
272 NULL
, 0, KEY_READ
| KEY_WRITE
, NULL
,
274 if (res
!= ERROR_SUCCESS
) goto error_close_clsid_key
;
276 res
= register_key_defvalueW(progid_key
, clsid_keyname
, buf
);
277 RegCloseKey(progid_key
);
278 if (res
!= ERROR_SUCCESS
) goto error_close_clsid_key
;
281 error_close_clsid_key
:
282 RegCloseKey(clsid_key
);
285 error_close_coclass_key
:
286 RegCloseKey(coclass_key
);
288 return res
!= ERROR_SUCCESS
? HRESULT_FROM_WIN32(res
) : S_OK
;
291 /***********************************************************************
292 * unregister_coclasses
294 static HRESULT
unregister_coclasses(struct regsvr_coclass
const *list
)
296 LONG res
= ERROR_SUCCESS
;
299 res
= RegOpenKeyExW(HKEY_CLASSES_ROOT
, clsid_keyname
, 0,
300 KEY_READ
| KEY_WRITE
, &coclass_key
);
301 if (res
== ERROR_FILE_NOT_FOUND
) return S_OK
;
302 if (res
!= ERROR_SUCCESS
) goto error_return
;
304 for (; res
== ERROR_SUCCESS
&& list
->clsid
; ++list
) {
307 StringFromGUID2(list
->clsid
, buf
, 39);
308 res
= recursive_delete_keyW(coclass_key
, buf
);
309 if (res
!= ERROR_SUCCESS
) goto error_close_coclass_key
;
312 res
= recursive_delete_keyA(HKEY_CLASSES_ROOT
, list
->progid
);
313 if (res
!= ERROR_SUCCESS
) goto error_close_coclass_key
;
317 error_close_coclass_key
:
318 RegCloseKey(coclass_key
);
320 return res
!= ERROR_SUCCESS
? HRESULT_FROM_WIN32(res
) : S_OK
;
323 /***********************************************************************
326 static LONG
register_key_guid(HKEY base
, WCHAR
const *name
, GUID
const *guid
)
330 StringFromGUID2(guid
, buf
, 39);
331 return register_key_defvalueW(base
, name
, buf
);
334 /***********************************************************************
335 * regsvr_key_defvalueW
337 static LONG
register_key_defvalueW(
345 res
= RegCreateKeyExW(base
, name
, 0, NULL
, 0,
346 KEY_READ
| KEY_WRITE
, NULL
, &key
, NULL
);
347 if (res
!= ERROR_SUCCESS
) return res
;
348 res
= RegSetValueExW(key
, NULL
, 0, REG_SZ
, (CONST BYTE
*)value
,
349 (lstrlenW(value
) + 1) * sizeof(WCHAR
));
354 /***********************************************************************
355 * regsvr_key_defvalueA
357 static LONG
register_key_defvalueA(
365 res
= RegCreateKeyExW(base
, name
, 0, NULL
, 0,
366 KEY_READ
| KEY_WRITE
, NULL
, &key
, NULL
);
367 if (res
!= ERROR_SUCCESS
) return res
;
368 res
= RegSetValueExA(key
, NULL
, 0, REG_SZ
, (CONST BYTE
*)value
,
369 lstrlenA(value
) + 1);
374 /***********************************************************************
375 * recursive_delete_key
377 static LONG
recursive_delete_key(HKEY key
)
380 WCHAR subkey_name
[MAX_PATH
];
385 cName
= sizeof(subkey_name
) / sizeof(WCHAR
);
386 res
= RegEnumKeyExW(key
, 0, subkey_name
, &cName
,
387 NULL
, NULL
, NULL
, NULL
);
388 if (res
!= ERROR_SUCCESS
&& res
!= ERROR_MORE_DATA
) {
389 res
= ERROR_SUCCESS
; /* presumably we're done enumerating */
392 res
= RegOpenKeyExW(key
, subkey_name
, 0,
393 KEY_READ
| KEY_WRITE
, &subkey
);
394 if (res
== ERROR_FILE_NOT_FOUND
) continue;
395 if (res
!= ERROR_SUCCESS
) break;
397 res
= recursive_delete_key(subkey
);
399 if (res
!= ERROR_SUCCESS
) break;
402 if (res
== ERROR_SUCCESS
) res
= RegDeleteKeyW(key
, 0);
406 /***********************************************************************
407 * recursive_delete_keyA
409 static LONG
recursive_delete_keyA(HKEY base
, char const *name
)
414 res
= RegOpenKeyExA(base
, name
, 0, KEY_READ
| KEY_WRITE
, &key
);
415 if (res
== ERROR_FILE_NOT_FOUND
) return ERROR_SUCCESS
;
416 if (res
!= ERROR_SUCCESS
) return res
;
417 res
= recursive_delete_key(key
);
422 /***********************************************************************
423 * recursive_delete_keyW
425 static LONG
recursive_delete_keyW(HKEY base
, WCHAR
const *name
)
430 res
= RegOpenKeyExW(base
, name
, 0, KEY_READ
| KEY_WRITE
, &key
);
431 if (res
== ERROR_FILE_NOT_FOUND
) return ERROR_SUCCESS
;
432 if (res
!= ERROR_SUCCESS
) return res
;
433 res
= recursive_delete_key(key
);
438 /***********************************************************************
442 static struct regsvr_coclass
const coclass_list
[] = {
449 { NULL
} /* list terminator */
452 /***********************************************************************
456 static struct regsvr_interface
const interface_list
[] = {
457 { NULL
} /* list terminator */
460 /***********************************************************************
461 * DllRegisterServer (ITSS.@)
463 HRESULT WINAPI
DllRegisterServer(void)
469 hr
= register_coclasses(coclass_list
);
471 hr
= register_interfaces(interface_list
);
475 /***********************************************************************
476 * DllUnregisterServer (ITSS.@)
478 HRESULT WINAPI
DllUnregisterServer(void)
484 hr
= unregister_coclasses(coclass_list
);
486 hr
= unregister_interfaces(interface_list
);