2 * self-registerable dll functions for dmscript.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
21 #include "dmscript_private.h"
23 WINE_DEFAULT_DEBUG_CHANNEL(dmscript
);
26 * Near the bottom of this file are the exported DllRegisterServer and
27 * DllUnregisterServer, which make all this worthwhile.
30 /***********************************************************************
31 * interface for self-registering
33 struct regsvr_interface
{
34 IID
const *iid
; /* NULL for end of list */
35 LPCSTR name
; /* can be NULL to omit */
36 IID
const *base_iid
; /* can be NULL to omit */
37 int num_methods
; /* can be <0 to omit */
38 CLSID
const *ps_clsid
; /* can be NULL to omit */
39 CLSID
const *ps_clsid32
; /* can be NULL to omit */
42 static HRESULT
register_interfaces(struct regsvr_interface
const *list
);
43 static HRESULT
unregister_interfaces(struct regsvr_interface
const *list
);
45 struct regsvr_coclass
{
46 CLSID
const *clsid
; /* NULL for end of list */
47 LPCSTR name
; /* can be NULL to omit */
48 LPCSTR ips
; /* can be NULL to omit */
49 LPCSTR ips32
; /* can be NULL to omit */
50 LPCSTR ips32_tmodel
; /* can be NULL to omit */
51 LPCSTR progid
; /* can be NULL to omit */
52 LPCSTR viprogid
; /* can be NULL to omit */
53 LPCSTR progid_extra
; /* can be NULL to omit */
56 static HRESULT
register_coclasses(struct regsvr_coclass
const *list
);
57 static HRESULT
unregister_coclasses(struct regsvr_coclass
const *list
);
59 /***********************************************************************
60 * static string constants
62 static WCHAR
const interface_keyname
[10] = {
63 'I', 'n', 't', 'e', 'r', 'f', 'a', 'c', 'e', 0 };
64 static WCHAR
const base_ifa_keyname
[14] = {
65 'B', 'a', 's', 'e', 'I', 'n', 't', 'e', 'r', 'f', 'a', 'c',
67 static WCHAR
const num_methods_keyname
[11] = {
68 'N', 'u', 'm', 'M', 'e', 't', 'h', 'o', 'd', 's', 0 };
69 static WCHAR
const ps_clsid_keyname
[15] = {
70 'P', 'r', 'o', 'x', 'y', 'S', 't', 'u', 'b', 'C', 'l', 's',
72 static WCHAR
const ps_clsid32_keyname
[17] = {
73 'P', 'r', 'o', 'x', 'y', 'S', 't', 'u', 'b', 'C', 'l', 's',
74 'i', 'd', '3', '2', 0 };
75 static WCHAR
const clsid_keyname
[6] = {
76 'C', 'L', 'S', 'I', 'D', 0 };
77 static WCHAR
const curver_keyname
[7] = {
78 'C', 'u', 'r', 'V', 'e', 'r', 0 };
79 static WCHAR
const ips_keyname
[13] = {
80 'I', 'n', 'P', 'r', 'o', 'c', 'S', 'e', 'r', 'v', 'e', 'r',
82 static WCHAR
const ips32_keyname
[15] = {
83 'I', 'n', 'P', 'r', 'o', 'c', 'S', 'e', 'r', 'v', 'e', 'r',
85 static WCHAR
const progid_keyname
[7] = {
86 'P', 'r', 'o', 'g', 'I', 'D', 0 };
87 static WCHAR
const viprogid_keyname
[25] = {
88 'V', 'e', 'r', 's', 'i', 'o', 'n', 'I', 'n', 'd', 'e', 'p',
89 'e', 'n', 'd', 'e', 'n', 't', 'P', 'r', 'o', 'g', 'I', 'D',
91 static char const tmodel_valuename
[] = "ThreadingModel";
93 /***********************************************************************
94 * static helper functions
96 static LONG
register_key_guid(HKEY base
, WCHAR
const *name
, GUID
const *guid
);
97 static LONG
register_key_defvalueW(HKEY base
, WCHAR
const *name
,
99 static LONG
register_key_defvalueA(HKEY base
, WCHAR
const *name
,
101 static LONG
register_progid(WCHAR
const *clsid
,
102 char const *progid
, char const *curver_progid
,
103 char const *name
, char const *extra
);
104 static LONG
recursive_delete_key(HKEY key
);
105 static LONG
recursive_delete_keyA(HKEY base
, char const *name
);
106 static LONG
recursive_delete_keyW(HKEY base
, WCHAR
const *name
);
108 /***********************************************************************
109 * register_interfaces
111 static HRESULT
register_interfaces(struct regsvr_interface
const *list
) {
112 LONG res
= ERROR_SUCCESS
;
115 res
= RegCreateKeyExW(HKEY_CLASSES_ROOT
, interface_keyname
, 0, NULL
, 0,
116 KEY_READ
| KEY_WRITE
, NULL
, &interface_key
, NULL
);
117 if (res
!= ERROR_SUCCESS
) goto error_return
;
119 for (; res
== ERROR_SUCCESS
&& list
->iid
; ++list
) {
123 StringFromGUID2(list
->iid
, buf
, 39);
124 res
= RegCreateKeyExW(interface_key
, buf
, 0, NULL
, 0,
125 KEY_READ
| KEY_WRITE
, NULL
, &iid_key
, NULL
);
126 if (res
!= ERROR_SUCCESS
) goto error_close_interface_key
;
129 res
= RegSetValueExA(iid_key
, NULL
, 0, REG_SZ
,
130 (CONST BYTE
*)(list
->name
),
131 strlen(list
->name
) + 1);
132 if (res
!= ERROR_SUCCESS
) goto error_close_iid_key
;
135 if (list
->base_iid
) {
136 register_key_guid(iid_key
, base_ifa_keyname
, list
->base_iid
);
137 if (res
!= ERROR_SUCCESS
) goto error_close_iid_key
;
140 if (0 <= list
->num_methods
) {
141 static WCHAR
const fmt
[3] = { '%', 'd', 0 };
144 res
= RegCreateKeyExW(iid_key
, num_methods_keyname
, 0, NULL
, 0,
145 KEY_READ
| KEY_WRITE
, NULL
, &key
, NULL
);
146 if (res
!= ERROR_SUCCESS
) goto error_close_iid_key
;
148 wsprintfW(buf
, fmt
, list
->num_methods
);
149 res
= RegSetValueExW(key
, NULL
, 0, REG_SZ
,
151 (lstrlenW(buf
) + 1) * sizeof(WCHAR
));
154 if (res
!= ERROR_SUCCESS
) goto error_close_iid_key
;
157 if (list
->ps_clsid
) {
158 register_key_guid(iid_key
, ps_clsid_keyname
, list
->ps_clsid
);
159 if (res
!= ERROR_SUCCESS
) goto error_close_iid_key
;
162 if (list
->ps_clsid32
) {
163 register_key_guid(iid_key
, ps_clsid32_keyname
, list
->ps_clsid32
);
164 if (res
!= ERROR_SUCCESS
) goto error_close_iid_key
;
168 RegCloseKey(iid_key
);
171 error_close_interface_key
:
172 RegCloseKey(interface_key
);
174 return res
!= ERROR_SUCCESS
? HRESULT_FROM_WIN32(res
) : S_OK
;
177 /***********************************************************************
178 * unregister_interfaces
180 static HRESULT
unregister_interfaces(struct regsvr_interface
const *list
) {
181 LONG res
= ERROR_SUCCESS
;
184 res
= RegOpenKeyExW(HKEY_CLASSES_ROOT
, interface_keyname
, 0,
185 KEY_READ
| KEY_WRITE
, &interface_key
);
186 if (res
== ERROR_FILE_NOT_FOUND
) return S_OK
;
187 if (res
!= ERROR_SUCCESS
) goto error_return
;
189 for (; res
== ERROR_SUCCESS
&& list
->iid
; ++list
) {
192 StringFromGUID2(list
->iid
, buf
, 39);
193 res
= recursive_delete_keyW(interface_key
, buf
);
196 RegCloseKey(interface_key
);
198 return res
!= ERROR_SUCCESS
? HRESULT_FROM_WIN32(res
) : S_OK
;
201 /***********************************************************************
204 static HRESULT
register_coclasses(struct regsvr_coclass
const *list
) {
205 LONG res
= ERROR_SUCCESS
;
208 res
= RegCreateKeyExW(HKEY_CLASSES_ROOT
, clsid_keyname
, 0, NULL
, 0,
209 KEY_READ
| KEY_WRITE
, NULL
, &coclass_key
, NULL
);
210 if (res
!= ERROR_SUCCESS
) goto error_return
;
212 for (; res
== ERROR_SUCCESS
&& list
->clsid
; ++list
) {
216 StringFromGUID2(list
->clsid
, buf
, 39);
217 res
= RegCreateKeyExW(coclass_key
, buf
, 0, NULL
, 0,
218 KEY_READ
| KEY_WRITE
, NULL
, &clsid_key
, NULL
);
219 if (res
!= ERROR_SUCCESS
) goto error_close_coclass_key
;
222 res
= RegSetValueExA(clsid_key
, NULL
, 0, REG_SZ
,
223 (CONST BYTE
*)(list
->name
),
224 strlen(list
->name
) + 1);
225 if (res
!= ERROR_SUCCESS
) goto error_close_clsid_key
;
229 res
= register_key_defvalueA(clsid_key
, ips_keyname
, list
->ips
);
230 if (res
!= ERROR_SUCCESS
) goto error_close_clsid_key
;
236 res
= RegCreateKeyExW(clsid_key
, ips32_keyname
, 0, NULL
, 0,
237 KEY_READ
| KEY_WRITE
, NULL
,
239 if (res
!= ERROR_SUCCESS
) goto error_close_clsid_key
;
241 res
= RegSetValueExA(ips32_key
, NULL
, 0, REG_SZ
,
242 (CONST BYTE
*)list
->ips32
,
243 lstrlenA(list
->ips32
) + 1);
244 if (res
== ERROR_SUCCESS
&& list
->ips32_tmodel
)
245 res
= RegSetValueExA(ips32_key
, tmodel_valuename
, 0, REG_SZ
,
246 (CONST BYTE
*)list
->ips32_tmodel
,
247 strlen(list
->ips32_tmodel
) + 1);
248 RegCloseKey(ips32_key
);
249 if (res
!= ERROR_SUCCESS
) goto error_close_clsid_key
;
253 res
= register_key_defvalueA(clsid_key
, progid_keyname
,
255 if (res
!= ERROR_SUCCESS
) goto error_close_clsid_key
;
257 res
= register_progid(buf
, list
->progid
, NULL
,
258 list
->name
, list
->progid_extra
);
259 if (res
!= ERROR_SUCCESS
) goto error_close_clsid_key
;
262 if (list
->viprogid
) {
263 res
= register_key_defvalueA(clsid_key
, viprogid_keyname
,
265 if (res
!= ERROR_SUCCESS
) goto error_close_clsid_key
;
267 res
= register_progid(buf
, list
->viprogid
, list
->progid
,
268 list
->name
, list
->progid_extra
);
269 if (res
!= ERROR_SUCCESS
) goto error_close_clsid_key
;
272 error_close_clsid_key
:
273 RegCloseKey(clsid_key
);
276 error_close_coclass_key
:
277 RegCloseKey(coclass_key
);
279 return res
!= ERROR_SUCCESS
? HRESULT_FROM_WIN32(res
) : S_OK
;
282 /***********************************************************************
283 * unregister_coclasses
285 static HRESULT
unregister_coclasses(struct regsvr_coclass
const *list
) {
286 LONG res
= ERROR_SUCCESS
;
289 res
= RegOpenKeyExW(HKEY_CLASSES_ROOT
, clsid_keyname
, 0,
290 KEY_READ
| KEY_WRITE
, &coclass_key
);
291 if (res
== ERROR_FILE_NOT_FOUND
) return S_OK
;
292 if (res
!= ERROR_SUCCESS
) goto error_return
;
294 for (; res
== ERROR_SUCCESS
&& list
->clsid
; ++list
) {
297 StringFromGUID2(list
->clsid
, buf
, 39);
298 res
= recursive_delete_keyW(coclass_key
, buf
);
299 if (res
!= ERROR_SUCCESS
) goto error_close_coclass_key
;
302 res
= recursive_delete_keyA(HKEY_CLASSES_ROOT
, list
->progid
);
303 if (res
!= ERROR_SUCCESS
) goto error_close_coclass_key
;
306 if (list
->viprogid
) {
307 res
= recursive_delete_keyA(HKEY_CLASSES_ROOT
, list
->viprogid
);
308 if (res
!= ERROR_SUCCESS
) goto error_close_coclass_key
;
312 error_close_coclass_key
:
313 RegCloseKey(coclass_key
);
315 return res
!= ERROR_SUCCESS
? HRESULT_FROM_WIN32(res
) : S_OK
;
318 /***********************************************************************
321 static LONG
register_key_guid(HKEY base
, WCHAR
const *name
, GUID
const *guid
) {
324 StringFromGUID2(guid
, buf
, 39);
325 return register_key_defvalueW(base
, name
, buf
);
328 /***********************************************************************
329 * regsvr_key_defvalueW
331 static LONG
register_key_defvalueW(
334 WCHAR
const *value
) {
338 res
= RegCreateKeyExW(base
, name
, 0, NULL
, 0,
339 KEY_READ
| KEY_WRITE
, NULL
, &key
, NULL
);
340 if (res
!= ERROR_SUCCESS
) return res
;
341 res
= RegSetValueExW(key
, NULL
, 0, REG_SZ
, (CONST BYTE
*)value
,
342 (lstrlenW(value
) + 1) * sizeof(WCHAR
));
347 /***********************************************************************
348 * regsvr_key_defvalueA
350 static LONG
register_key_defvalueA(
357 res
= RegCreateKeyExW(base
, name
, 0, NULL
, 0,
358 KEY_READ
| KEY_WRITE
, NULL
, &key
, NULL
);
359 if (res
!= ERROR_SUCCESS
) return res
;
360 res
= RegSetValueExA(key
, NULL
, 0, REG_SZ
, (CONST BYTE
*)value
,
361 lstrlenA(value
) + 1);
366 /***********************************************************************
369 static LONG
register_progid(
372 char const *curver_progid
,
378 res
= RegCreateKeyExA(HKEY_CLASSES_ROOT
, progid
, 0,
379 NULL
, 0, KEY_READ
| KEY_WRITE
, NULL
,
381 if (res
!= ERROR_SUCCESS
) return res
;
384 res
= RegSetValueExA(progid_key
, NULL
, 0, REG_SZ
,
385 (CONST BYTE
*)name
, strlen(name
) + 1);
386 if (res
!= ERROR_SUCCESS
) goto error_close_progid_key
;
390 res
= register_key_defvalueW(progid_key
, clsid_keyname
, clsid
);
391 if (res
!= ERROR_SUCCESS
) goto error_close_progid_key
;
395 res
= register_key_defvalueA(progid_key
, curver_keyname
,
397 if (res
!= ERROR_SUCCESS
) goto error_close_progid_key
;
403 res
= RegCreateKeyExA(progid_key
, extra
, 0,
404 NULL
, 0, KEY_READ
| KEY_WRITE
, NULL
,
406 if (res
== ERROR_SUCCESS
)
407 RegCloseKey(extra_key
);
410 error_close_progid_key
:
411 RegCloseKey(progid_key
);
415 /***********************************************************************
416 * recursive_delete_key
418 static LONG
recursive_delete_key(HKEY key
) {
420 WCHAR subkey_name
[MAX_PATH
];
425 cName
= sizeof(subkey_name
) / sizeof(WCHAR
);
426 res
= RegEnumKeyExW(key
, 0, subkey_name
, &cName
,
427 NULL
, NULL
, NULL
, NULL
);
428 if (res
!= ERROR_SUCCESS
&& res
!= ERROR_MORE_DATA
) {
429 res
= ERROR_SUCCESS
; /* presumably we're done enumerating */
432 res
= RegOpenKeyExW(key
, subkey_name
, 0,
433 KEY_READ
| KEY_WRITE
, &subkey
);
434 if (res
== ERROR_FILE_NOT_FOUND
) continue;
435 if (res
!= ERROR_SUCCESS
) break;
437 res
= recursive_delete_key(subkey
);
439 if (res
!= ERROR_SUCCESS
) break;
442 if (res
== ERROR_SUCCESS
) res
= RegDeleteKeyW(key
, 0);
446 /***********************************************************************
447 * recursive_delete_keyA
449 static LONG
recursive_delete_keyA(HKEY base
, char const *name
) {
453 res
= RegOpenKeyExA(base
, name
, 0, KEY_READ
| KEY_WRITE
, &key
);
454 if (res
== ERROR_FILE_NOT_FOUND
) return ERROR_SUCCESS
;
455 if (res
!= ERROR_SUCCESS
) return res
;
456 res
= recursive_delete_key(key
);
461 /***********************************************************************
462 * recursive_delete_keyW
464 static LONG
recursive_delete_keyW(HKEY base
, WCHAR
const *name
) {
468 res
= RegOpenKeyExW(base
, name
, 0, KEY_READ
| KEY_WRITE
, &key
);
469 if (res
== ERROR_FILE_NOT_FOUND
) return ERROR_SUCCESS
;
470 if (res
!= ERROR_SUCCESS
) return res
;
471 res
= recursive_delete_key(key
);
476 /***********************************************************************
479 static struct regsvr_coclass
const coclass_list
[] = {
480 { &CLSID_DirectMusicScriptAutoImpSegment
,
481 "DirectMusic Script AutoImp Segment",
485 "Microsoft.DirectMusicScriptAutoImpSegment.1",
486 "Microsoft.DirectMusicScriptAutoImpSegment"
488 { &CLSID_DirectMusicScriptTrack
,
489 "DirectMusicScriptTrack",
493 "Microsoft.DirectMusicScriptTrack.1",
494 "Microsoft.DirectMusicScriptTrack"
496 { &CLSID_AudioVBScript
,
497 "DirectMusic Audio VB Script Language",
505 { &CLSID_DirectMusicScript
,
506 "DirectMusic Script Object",
510 "Microsoft.DirectMusicScript.1",
511 "Microsoft.DirectMusicScript"
513 { &CLSID_DirectMusicScriptAutoImpPerformance
,
514 "DirectMusic Script AutoImp Performance",
518 "Microsoft.DirectMusicScriptAutoImpPerformance.1",
519 "Microsoft.DirectMusicScriptAutoImpPerformance"
521 { &CLSID_DirectMusicScriptSourceCodeLoader
,
522 "DirectMusic Script Source Code Loader",
526 "Microsoft.DirectMusicScripSourceCodeLoader.1", /* [sic] */
527 "Microsoft.DirectMusicScripSourceCodeLoader"
529 { &CLSID_DirectMusicScriptAutoImpSegmentState
,
530 "DirectMusic Script AutoImp SegmentState",
534 "Microsoft.DirectMusicScriptAutoImpSegmentState.1",
535 "Microsoft.DirectMusicScriptAutoImpSegmentState"
537 { &CLSID_DirectMusicScriptAutoImpAudioPathConfig
,
538 "DirectMusic Script AutoImp AudioPathConfig",
542 "Microsoft.DirectMusicScriptAutoImpAudioPathConfig.1",
543 "Microsoft.DirectMusicScriptAutoImpAudioPathConfig"
545 { &CLSID_DirectMusicScriptAutoImpAudioPath
,
546 "DirectMusic Script AutoImp AudioPath",
550 "Microsoft.DirectMusicScriptAutoImpAudioPath.1",
551 "Microsoft.DirectMusicScriptAutoImpAudioPath"
553 { &CLSID_DirectMusicScriptAutoImpSong
,
554 "DirectMusic Script AutoImp Song",
558 "Microsoft.DirectMusicScriptAutoImpSong.1",
559 "Microsoft.DirectMusicScriptAutoImpSong"
561 { NULL
} /* list terminator */
564 /***********************************************************************
568 static struct regsvr_interface
const interface_list
[] = {
569 { NULL
} /* list terminator */
572 /***********************************************************************
573 * DllRegisterServer (DMSCRIPT.3)
575 HRESULT WINAPI
DllRegisterServer(void)
581 hr
= register_coclasses(coclass_list
);
583 hr
= register_interfaces(interface_list
);
587 /***********************************************************************
588 * DllUnregisterServer (DMSCRIPT.4)
590 HRESULT WINAPI
DllUnregisterServer(void)
596 hr
= unregister_coclasses(coclass_list
);
598 hr
= unregister_interfaces(interface_list
);