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 DMSCRIPT_DllRegisterServer(void) {
580 hr = register_coclasses(coclass_list);
582 hr = register_interfaces(interface_list);
586 /***********************************************************************
587 * DllUnregisterServer (DMSCRIPT.4)
589 HRESULT WINAPI DMSCRIPT_DllUnregisterServer(void) {
594 hr = unregister_coclasses(coclass_list);
596 hr = unregister_interfaces(interface_list);