dlls/urlmon/regsvr.c

   1 /*
   2  *      self-registerable dll functions for urlmon.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 #define COM_NO_WINDOWS_H
  22 #include <stdarg.h>
  23 #include <stdio.h>
  24 #include <string.h>
  25
  26 #include "windef.h"
  27 #include "winbase.h"
  28 #include "winuser.h"
  29 #include "wingdi.h"
  30 #include "winreg.h"
  31 #include "winerror.h"
  32 #include "advpub.h"
  33
  34 #include "objbase.h"
  35
  36 #include "urlmon.h"
  37
  38 #include "wine/debug.h"
  39
  40 #include "urlmon_main.h"
  41
  42 WINE_DEFAULT_DEBUG_CHANNEL(urlmon);
  43
  44 /*
  45  * Near the bottom of this file are the exported DllRegisterServer and
  46  * DllUnregisterServer, which make all this worthwhile.
  47  */
  48
  49 /***********************************************************************
  50  *              interface for self-registering
  51  */
  52 struct regsvr_interface
  53 {
  54     IID const *iid;             /* NULL for end of list */
  55     LPCSTR name;                /* can be NULL to omit */
  56     IID const *base_iid;        /* can be NULL to omit */
  57     int num_methods;            /* can be <0 to omit */
  58     CLSID const *ps_clsid;      /* can be NULL to omit */
  59     CLSID const *ps_clsid32;    /* can be NULL to omit */
  60 };
  61
  62 static HRESULT register_interfaces(struct regsvr_interface const *list);
  63 static HRESULT unregister_interfaces(struct regsvr_interface const *list);
  64
  65 struct regsvr_coclass
  66 {
  67     CLSID const *clsid;         /* NULL for end of list */
  68     LPCSTR name;                /* can be NULL to omit */
  69     LPCSTR ips;                 /* can be NULL to omit */
  70     LPCSTR ips32;               /* can be NULL to omit */
  71     LPCSTR ips32_tmodel;        /* can be NULL to omit */
  72     LPCSTR progid;              /* can be NULL to omit */
  73     LPCSTR viprogid;            /* can be NULL to omit */
  74     LPCSTR progid_extra;        /* can be NULL to omit */
  75 };
  76
  77 static HRESULT register_coclasses(struct regsvr_coclass const *list);
  78 static HRESULT unregister_coclasses(struct regsvr_coclass const *list);
  79
  80 /***********************************************************************
  81  *              static string constants
  82  */
  83 static WCHAR const interface_keyname[10] = {
  84     'I', 'n', 't', 'e', 'r', 'f', 'a', 'c', 'e', 0 };
  85 static WCHAR const base_ifa_keyname[14] = {
  86     'B', 'a', 's', 'e', 'I', 'n', 't', 'e', 'r', 'f', 'a', 'c',
  87     'e', 0 };
  88 static WCHAR const num_methods_keyname[11] = {
  89     'N', 'u', 'm', 'M', 'e', 't', 'h', 'o', 'd', 's', 0 };
  90 static WCHAR const ps_clsid_keyname[15] = {
  91     'P', 'r', 'o', 'x', 'y', 'S', 't', 'u', 'b', 'C', 'l', 's',
  92     'i', 'd', 0 };
  93 static WCHAR const ps_clsid32_keyname[17] = {
  94     'P', 'r', 'o', 'x', 'y', 'S', 't', 'u', 'b', 'C', 'l', 's',
  95     'i', 'd', '3', '2', 0 };
  96 static WCHAR const clsid_keyname[6] = {
  97     'C', 'L', 'S', 'I', 'D', 0 };
  98 static WCHAR const curver_keyname[7] = {
  99     'C', 'u', 'r', 'V', 'e', 'r', 0 };
 100 static WCHAR const ips_keyname[13] = {
 101     'I', 'n', 'P', 'r', 'o', 'c', 'S', 'e', 'r', 'v', 'e', 'r',
 102     0 };
 103 static WCHAR const ips32_keyname[15] = {
 104     'I', 'n', 'P', 'r', 'o', 'c', 'S', 'e', 'r', 'v', 'e', 'r',
 105     '3', '2', 0 };
 106 static WCHAR const progid_keyname[7] = {
 107     'P', 'r', 'o', 'g', 'I', 'D', 0 };
 108 static WCHAR const viprogid_keyname[25] = {
 109     'V', 'e', 'r', 's', 'i', 'o', 'n', 'I', 'n', 'd', 'e', 'p',
 110     'e', 'n', 'd', 'e', 'n', 't', 'P', 'r', 'o', 'g', 'I', 'D',
 111     0 };
 112 static char const tmodel_valuename[] = "ThreadingModel";
 113
 114 /***********************************************************************
 115  *              static helper functions
 116  */
 117 static LONG register_key_guid(HKEY base, WCHAR const *name, GUID const *guid);
 118 static LONG register_key_defvalueW(HKEY base, WCHAR const *name,
 119                                    WCHAR const *value);
 120 static LONG register_key_defvalueA(HKEY base, WCHAR const *name,
 121                                    char const *value);
 122 static LONG register_progid(WCHAR const *clsid,
 123                             char const *progid, char const *curver_progid,
 124                             char const *name, char const *extra);
 125 static LONG recursive_delete_key(HKEY key);
 126 static LONG recursive_delete_keyA(HKEY base, char const *name);
 127 static LONG recursive_delete_keyW(HKEY base, WCHAR const *name);
 128
 129 /***********************************************************************
 130  *              register_interfaces
 131  */
 132 static HRESULT register_interfaces(struct regsvr_interface const *list)
 133 {
 134     LONG res = ERROR_SUCCESS;
 135     HKEY interface_key;
 136
 137     res = RegCreateKeyExW(HKEY_CLASSES_ROOT, interface_keyname, 0, NULL, 0,
 138                           KEY_READ | KEY_WRITE, NULL, &interface_key, NULL);
 139     if (res != ERROR_SUCCESS) goto error_return;
 140
 141     for (; res == ERROR_SUCCESS && list->iid; ++list) {
 142         WCHAR buf[39];
 143         HKEY iid_key;
 144
 145         StringFromGUID2(list->iid, buf, 39);
 146         res = RegCreateKeyExW(interface_key, buf, 0, NULL, 0,
 147                               KEY_READ | KEY_WRITE, NULL, &iid_key, NULL);
 148         if (res != ERROR_SUCCESS) goto error_close_interface_key;
 149
 150         if (list->name) {
 151             res = RegSetValueExA(iid_key, NULL, 0, REG_SZ,
 152                                  (CONST BYTE*)(list->name),
 153                                  strlen(list->name) + 1);
 154             if (res != ERROR_SUCCESS) goto error_close_iid_key;
 155         }
 156
 157         if (list->base_iid) {
 158             res = register_key_guid(iid_key, base_ifa_keyname, list->base_iid);
 159             if (res != ERROR_SUCCESS) goto error_close_iid_key;
 160         }
 161
 162         if (0 <= list->num_methods) {
 163             static WCHAR const fmt[3] = { '%', 'd', 0 };
 164             HKEY key;
 165
 166             res = RegCreateKeyExW(iid_key, num_methods_keyname, 0, NULL, 0,
 167                                   KEY_READ | KEY_WRITE, NULL, &key, NULL);
 168             if (res != ERROR_SUCCESS) goto error_close_iid_key;
 169
 170             wsprintfW(buf, fmt, list->num_methods);
 171             res = RegSetValueExW(key, NULL, 0, REG_SZ,
 172                                  (CONST BYTE*)buf,
 173                                  (lstrlenW(buf) + 1) * sizeof(WCHAR));
 174             RegCloseKey(key);
 175
 176             if (res != ERROR_SUCCESS) goto error_close_iid_key;
 177         }
 178
 179         if (list->ps_clsid) {
 180             res = register_key_guid(iid_key, ps_clsid_keyname, list->ps_clsid);
 181             if (res != ERROR_SUCCESS) goto error_close_iid_key;
 182         }
 183
 184         if (list->ps_clsid32) {
 185             res = register_key_guid(iid_key, ps_clsid32_keyname, list->ps_clsid32);
 186             if (res != ERROR_SUCCESS) goto error_close_iid_key;
 187         }
 188
 189     error_close_iid_key:
 190         RegCloseKey(iid_key);
 191     }
 192
 193 error_close_interface_key:
 194     RegCloseKey(interface_key);
 195 error_return:
 196     return res != ERROR_SUCCESS ? HRESULT_FROM_WIN32(res) : S_OK;
 197 }
 198
 199 /***********************************************************************
 200  *              unregister_interfaces
 201  */
 202 static HRESULT unregister_interfaces(struct regsvr_interface const *list)
 203 {
 204     LONG res = ERROR_SUCCESS;
 205     HKEY interface_key;
 206
 207     res = RegOpenKeyExW(HKEY_CLASSES_ROOT, interface_keyname, 0,
 208                         KEY_READ | KEY_WRITE, &interface_key);
 209     if (res == ERROR_FILE_NOT_FOUND) return S_OK;
 210     if (res != ERROR_SUCCESS) goto error_return;
 211
 212     for (; res == ERROR_SUCCESS && list->iid; ++list) {
 213         WCHAR buf[39];
 214
 215         StringFromGUID2(list->iid, buf, 39);
 216         res = recursive_delete_keyW(interface_key, buf);
 217     }
 218
 219     RegCloseKey(interface_key);
 220 error_return:
 221     return res != ERROR_SUCCESS ? HRESULT_FROM_WIN32(res) : S_OK;
 222 }
 223
 224 /***********************************************************************
 225  *              register_coclasses
 226  */
 227 static HRESULT register_coclasses(struct regsvr_coclass const *list)
 228 {
 229     LONG res = ERROR_SUCCESS;
 230     HKEY coclass_key;
 231
 232     res = RegCreateKeyExW(HKEY_CLASSES_ROOT, clsid_keyname, 0, NULL, 0,
 233                           KEY_READ | KEY_WRITE, NULL, &coclass_key, NULL);
 234     if (res != ERROR_SUCCESS) goto error_return;
 235
 236     for (; res == ERROR_SUCCESS && list->clsid; ++list) {
 237         WCHAR buf[39];
 238         HKEY clsid_key;
 239
 240         StringFromGUID2(list->clsid, buf, 39);
 241         res = RegCreateKeyExW(coclass_key, buf, 0, NULL, 0,
 242                               KEY_READ | KEY_WRITE, NULL, &clsid_key, NULL);
 243         if (res != ERROR_SUCCESS) goto error_close_coclass_key;
 244
 245         if (list->name) {
 246             res = RegSetValueExA(clsid_key, NULL, 0, REG_SZ,
 247                                  (CONST BYTE*)(list->name),
 248                                  strlen(list->name) + 1);
 249             if (res != ERROR_SUCCESS) goto error_close_clsid_key;
 250         }
 251
 252         if (list->ips) {
 253             res = register_key_defvalueA(clsid_key, ips_keyname, list->ips);
 254             if (res != ERROR_SUCCESS) goto error_close_clsid_key;
 255         }
 256
 257         if (list->ips32) {
 258             HKEY ips32_key;
 259
 260             res = RegCreateKeyExW(clsid_key, ips32_keyname, 0, NULL, 0,
 261                                   KEY_READ | KEY_WRITE, NULL,
 262                                   &ips32_key, NULL);
 263             if (res != ERROR_SUCCESS) goto error_close_clsid_key;
 264
 265             res = RegSetValueExA(ips32_key, NULL, 0, REG_SZ,
 266                                  (CONST BYTE*)list->ips32,
 267                                  lstrlenA(list->ips32) + 1);
 268             if (res == ERROR_SUCCESS && list->ips32_tmodel)
 269                 res = RegSetValueExA(ips32_key, tmodel_valuename, 0, REG_SZ,
 270                                      (CONST BYTE*)list->ips32_tmodel,
 271                                      strlen(list->ips32_tmodel) + 1);
 272             RegCloseKey(ips32_key);
 273             if (res != ERROR_SUCCESS) goto error_close_clsid_key;
 274         }
 275
 276         if (list->progid) {
 277             res = register_key_defvalueA(clsid_key, progid_keyname,
 278                                          list->progid);
 279             if (res != ERROR_SUCCESS) goto error_close_clsid_key;
 280
 281             res = register_progid(buf, list->progid, NULL,
 282                                   list->name, list->progid_extra);
 283             if (res != ERROR_SUCCESS) goto error_close_clsid_key;
 284         }
 285
 286         if (list->viprogid) {
 287             res = register_key_defvalueA(clsid_key, viprogid_keyname,
 288                                          list->viprogid);
 289             if (res != ERROR_SUCCESS) goto error_close_clsid_key;
 290
 291             res = register_progid(buf, list->viprogid, list->progid,
 292                                   list->name, list->progid_extra);
 293             if (res != ERROR_SUCCESS) goto error_close_clsid_key;
 294         }
 295
 296     error_close_clsid_key:
 297         RegCloseKey(clsid_key);
 298     }
 299
 300 error_close_coclass_key:
 301     RegCloseKey(coclass_key);
 302 error_return:
 303     return res != ERROR_SUCCESS ? HRESULT_FROM_WIN32(res) : S_OK;
 304 }
 305
 306 /***********************************************************************
 307  *              unregister_coclasses
 308  */
 309 static HRESULT unregister_coclasses(struct regsvr_coclass const *list)
 310 {
 311     LONG res = ERROR_SUCCESS;
 312     HKEY coclass_key;
 313
 314     res = RegOpenKeyExW(HKEY_CLASSES_ROOT, clsid_keyname, 0,
 315                         KEY_READ | KEY_WRITE, &coclass_key);
 316     if (res == ERROR_FILE_NOT_FOUND) return S_OK;
 317     if (res != ERROR_SUCCESS) goto error_return;
 318
 319     for (; res == ERROR_SUCCESS && list->clsid; ++list) {
 320         WCHAR buf[39];
 321
 322         StringFromGUID2(list->clsid, buf, 39);
 323         res = recursive_delete_keyW(coclass_key, buf);
 324         if (res != ERROR_SUCCESS) goto error_close_coclass_key;
 325
 326         if (list->progid) {
 327             res = recursive_delete_keyA(HKEY_CLASSES_ROOT, list->progid);
 328             if (res != ERROR_SUCCESS) goto error_close_coclass_key;
 329         }
 330
 331         if (list->viprogid) {
 332             res = recursive_delete_keyA(HKEY_CLASSES_ROOT, list->viprogid);
 333             if (res != ERROR_SUCCESS) goto error_close_coclass_key;
 334         }
 335     }
 336
 337 error_close_coclass_key:
 338     RegCloseKey(coclass_key);
 339 error_return:
 340     return res != ERROR_SUCCESS ? HRESULT_FROM_WIN32(res) : S_OK;
 341 }
 342
 343 /***********************************************************************
 344  *              regsvr_key_guid
 345  */
 346 static LONG register_key_guid(HKEY base, WCHAR const *name, GUID const *guid)
 347 {
 348     WCHAR buf[39];
 349
 350     StringFromGUID2(guid, buf, 39);
 351     return register_key_defvalueW(base, name, buf);
 352 }
 353
 354 /***********************************************************************
 355  *              regsvr_key_defvalueW
 356  */
 357 static LONG register_key_defvalueW(
 358     HKEY base,
 359     WCHAR const *name,
 360     WCHAR const *value)
 361 {
 362     LONG res;
 363     HKEY key;
 364
 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 = RegSetValueExW(key, NULL, 0, REG_SZ, (CONST BYTE*)value,
 369                          (lstrlenW(value) + 1) * sizeof(WCHAR));
 370     RegCloseKey(key);
 371     return res;
 372 }
 373
 374 /***********************************************************************
 375  *              regsvr_key_defvalueA
 376  */
 377 static LONG register_key_defvalueA(
 378     HKEY base,
 379     WCHAR const *name,
 380     char const *value)
 381 {
 382     LONG res;
 383     HKEY key;
 384
 385     res = RegCreateKeyExW(base, name, 0, NULL, 0,
 386                           KEY_READ | KEY_WRITE, NULL, &key, NULL);
 387     if (res != ERROR_SUCCESS) return res;
 388     res = RegSetValueExA(key, NULL, 0, REG_SZ, (CONST BYTE*)value,
 389                          lstrlenA(value) + 1);
 390     RegCloseKey(key);
 391     return res;
 392 }
 393
 394 /***********************************************************************
 395  *              regsvr_progid
 396  */
 397 static LONG register_progid(
 398     WCHAR const *clsid,
 399     char const *progid,
 400     char const *curver_progid,
 401     char const *name,
 402     char const *extra)
 403 {
 404     LONG res;
 405     HKEY progid_key;
 406
 407     res = RegCreateKeyExA(HKEY_CLASSES_ROOT, progid, 0,
 408                           NULL, 0, KEY_READ | KEY_WRITE, NULL,
 409                           &progid_key, NULL);
 410     if (res != ERROR_SUCCESS) return res;
 411
 412     if (name) {
 413         res = RegSetValueExA(progid_key, NULL, 0, REG_SZ,
 414                              (CONST BYTE*)name, strlen(name) + 1);
 415         if (res != ERROR_SUCCESS) goto error_close_progid_key;
 416     }
 417
 418     if (clsid) {
 419         res = register_key_defvalueW(progid_key, clsid_keyname, clsid);
 420         if (res != ERROR_SUCCESS) goto error_close_progid_key;
 421     }
 422
 423     if (curver_progid) {
 424         res = register_key_defvalueA(progid_key, curver_keyname,
 425                                      curver_progid);
 426         if (res != ERROR_SUCCESS) goto error_close_progid_key;
 427     }
 428
 429     if (extra) {
 430         HKEY extra_key;
 431
 432         res = RegCreateKeyExA(progid_key, extra, 0,
 433                               NULL, 0, KEY_READ | KEY_WRITE, NULL,
 434                               &extra_key, NULL);
 435         if (res == ERROR_SUCCESS)
 436             RegCloseKey(extra_key);
 437     }
 438
 439 error_close_progid_key:
 440     RegCloseKey(progid_key);
 441     return res;
 442 }
 443
 444 /***********************************************************************
 445  *              recursive_delete_key
 446  */
 447 static LONG recursive_delete_key(HKEY key)
 448 {
 449     LONG res;
 450     WCHAR subkey_name[MAX_PATH];
 451     DWORD cName;
 452     HKEY subkey;
 453
 454     for (;;) {
 455         cName = sizeof(subkey_name) / sizeof(WCHAR);
 456         res = RegEnumKeyExW(key, 0, subkey_name, &cName,
 457                             NULL, NULL, NULL, NULL);
 458         if (res != ERROR_SUCCESS && res != ERROR_MORE_DATA) {
 459             res = ERROR_SUCCESS; /* presumably we're done enumerating */
 460             break;
 461         }
 462         res = RegOpenKeyExW(key, subkey_name, 0,
 463                             KEY_READ | KEY_WRITE, &subkey);
 464         if (res == ERROR_FILE_NOT_FOUND) continue;
 465         if (res != ERROR_SUCCESS) break;
 466
 467         res = recursive_delete_key(subkey);
 468         RegCloseKey(subkey);
 469         if (res != ERROR_SUCCESS) break;
 470     }
 471
 472     if (res == ERROR_SUCCESS) res = RegDeleteKeyW(key, 0);
 473     return res;
 474 }
 475
 476 /***********************************************************************
 477  *              recursive_delete_keyA
 478  */
 479 static LONG recursive_delete_keyA(HKEY base, char const *name)
 480 {
 481     LONG res;
 482     HKEY key;
 483
 484     res = RegOpenKeyExA(base, name, 0, KEY_READ | KEY_WRITE, &key);
 485     if (res == ERROR_FILE_NOT_FOUND) return ERROR_SUCCESS;
 486     if (res != ERROR_SUCCESS) return res;
 487     res = recursive_delete_key(key);
 488     RegCloseKey(key);
 489     return res;
 490 }
 491
 492 /***********************************************************************
 493  *              recursive_delete_keyW
 494  */
 495 static LONG recursive_delete_keyW(HKEY base, WCHAR const *name)
 496 {
 497     LONG res;
 498     HKEY key;
 499
 500     res = RegOpenKeyExW(base, name, 0, KEY_READ | KEY_WRITE, &key);
 501     if (res == ERROR_FILE_NOT_FOUND) return ERROR_SUCCESS;
 502     if (res != ERROR_SUCCESS) return res;
 503     res = recursive_delete_key(key);
 504     RegCloseKey(key);
 505     return res;
 506 }
 507
 508 /***********************************************************************
 509  *              coclass list
 510  */
 511 static struct regsvr_coclass const coclass_list[] = {
 512     {   &CLSID_StdURLMoniker,
 513         "URL Moniker",
 514         NULL,
 515         "urlmon.dll",
 516         "Apartment"
 517     },
 518     {   &CLSID_CdlProtocol,
 519         "CDL: Asynchronous Pluggable Protocol Handler",
 520         NULL,
 521         "urlmon.dll",
 522         "Apartment"
 523     },
 524     {   &CLSID_FileProtocol,
 525         "file:, local: Asynchronous Pluggable Protocol Handler",
 526         NULL,
 527         "urlmon.dll",
 528         "Apartment"
 529     },
 530     {   &CLSID_FtpProtocol,
 531         "ftp: Asynchronous Pluggable Protocol Handler",
 532         NULL,
 533         "urlmon.dll",
 534         "Apartment"
 535     },
 536     {   &CLSID_GopherProtocol,
 537         "gopher: Asynchronous Pluggable Protocol Handler",
 538         NULL,
 539         "urlmon.dll",
 540         "Apartment"
 541     },
 542     {   &CLSID_HttpProtocol,
 543         "http: Asynchronous Pluggable Protocol Handler",
 544         NULL,
 545         "urlmon.dll",
 546         "Apartment"
 547     },
 548     {   &CLSID_HttpSProtocol,
 549         "https: Asynchronous Pluggable Protocol Handler",
 550         NULL,
 551         "urlmon.dll",
 552         "Apartment"
 553     },
 554     {   &CLSID_MkProtocol,
 555         "mk: Asynchronous Pluggable Protocol Handler",
 556         NULL,
 557         "urlmon.dll",
 558         "Apartment"
 559     },
 560     {   &CLSID_InternetSecurityManager,
 561         "Security Manager",
 562         NULL,
 563         "urlmon.dll",
 564         "Both"
 565     },
 566     {   &CLSID_InternetZoneManager,
 567         "URL Zone Manager",
 568         NULL,
 569         "urlmon.dll",
 570         "Both"
 571     },
 572     { NULL }                    /* list terminator */
 573 };
 574
 575 /***********************************************************************
 576  *              interface list
 577  */
 578
 579 static struct regsvr_interface const interface_list[] = {
 580     { NULL }                    /* list terminator */
 581 };
 582
 583 /***********************************************************************
 584  *              register_inf
 585  */
 586
 587 #define INF_SET_CLSID(clsid)                  \
 588     do                                        \
 589     {                                         \
 590         static CHAR name[] = "CLSID_" #clsid; \
 591                                               \
 592         pse[i].pszName = name;                \
 593         clsids[i++] = &CLSID_ ## clsid;       \
 594     } while (0)
 595
 596 static HRESULT register_inf(BOOL doregister)
 597 {
 598     HRESULT hres;
 599     HMODULE hAdvpack;
 600     typeof(RegInstallA) *pRegInstall;
 601     STRTABLEA strtable;
 602     STRENTRYA pse[7];
 603     static CLSID const *clsids[34];
 604     int i = 0;
 605
 606     static const WCHAR wszAdvpack[] = {'a','d','v','p','a','c','k','.','d','l','l',0};
 607
 608     INF_SET_CLSID(CdlProtocol);
 609     INF_SET_CLSID(FileProtocol);
 610     INF_SET_CLSID(FtpProtocol);
 611     INF_SET_CLSID(GopherProtocol);
 612     INF_SET_CLSID(HttpProtocol);
 613     INF_SET_CLSID(HttpSProtocol);
 614     INF_SET_CLSID(MkProtocol);
 615
 616     for(i = 0; i < sizeof(pse)/sizeof(pse[0]); i++) {
 617         pse[i].pszValue = HeapAlloc(GetProcessHeap(), 0, 39);
 618         sprintf(pse[i].pszValue, "{%08lX-%04X-%04X-%02X%02X-%02X%02X%02X%02X%02X%02X}",
 619                 clsids[i]->Data1, clsids[i]->Data2, clsids[i]->Data3, clsids[i]->Data4[0],
 620                 clsids[i]->Data4[1], clsids[i]->Data4[2], clsids[i]->Data4[3], clsids[i]->Data4[4],
 621                 clsids[i]->Data4[5], clsids[i]->Data4[6], clsids[i]->Data4[7]);
 622     }
 623
 624     strtable.cEntries = sizeof(pse)/sizeof(pse[0]);
 625     strtable.pse = pse;
 626
 627     hAdvpack = LoadLibraryW(wszAdvpack);
 628     pRegInstall = (typeof(RegInstallA)*)GetProcAddress(hAdvpack, "RegInstall");
 629
 630     hres = pRegInstall(URLMON_hInstance, doregister ? "RegisterDll" : "UnregisterDll", &strtable);
 631
 632     for(i=0; i < sizeof(pse)/sizeof(pse[0]); i++)
 633         HeapFree(GetProcessHeap(), 0, pse[i].pszValue);
 634
 635     return hres;
 636 }
 637
 638 #undef INF_SET_CLSID
 639
 640 /***********************************************************************
 641  *              DllRegisterServer (URLMON.@)
 642  */
 643 HRESULT WINAPI DllRegisterServer(void)
 644 {
 645     HRESULT hr;
 646
 647     TRACE("\n");
 648
 649     hr = register_coclasses(coclass_list);
 650     if (SUCCEEDED(hr))
 651         hr = register_interfaces(interface_list);
 652     if(FAILED(hr))
 653         return hr;
 654     return register_inf(TRUE);
 655 }
 656
 657 /***********************************************************************
 658  *              DllUnregisterServer (URLMON.@)
 659  */
 660 HRESULT WINAPI DllUnregisterServer(void)
 661 {
 662     HRESULT hr;
 663
 664     TRACE("\n");
 665
 666     hr = unregister_coclasses(coclass_list);
 667     if (SUCCEEDED(hr))
 668         hr = unregister_interfaces(interface_list);
 669     if(FAILED(hr))
 670         return hr;
 671     return register_inf(FALSE);
 672 }