dlls/atl/registrar.c

   1 /*
   2  * Copyright 2005 Jacek Caban
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2.1 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with this library; if not, write to the Free Software
  16  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  17  */
  18
  19
  20 #include <stdarg.h>
  21
  22 #define COBJMACROS
  23
  24 #include "windef.h"
  25 #include "winbase.h"
  26 #include "winuser.h"
  27 #include "winreg.h"
  28 #include "objbase.h"
  29 #include "oaidl.h"
  30 #include "shlwapi.h"
  31
  32 #define ATL_INITGUID
  33 #include "atliface.h"
  34 #include "atlbase.h"
  35
  36 #include "wine/debug.h"
  37 #include "wine/unicode.h"
  38
  39 WINE_DEFAULT_DEBUG_CHANNEL(atl);
  40
  41 LONG dll_count = 0;
  42
  43 /**************************************************************
  44  * ATLRegistrar implementation
  45  */
  46
  47 static struct {
  48     WCHAR name[22];
  49     HKEY  key;
  50 } root_keys[] = {
  51     {{'H','K','E','Y','_','C','L','A','S','S','E','S','_','R','O','O','T',0},
  52                     HKEY_CLASSES_ROOT},
  53     {{'H','K','E','Y','_','C','U','R','R','E','N','T','_','U','S','E','R',0},
  54                     HKEY_CURRENT_USER},
  55     {{'H','K','E','Y','_','L','O','C','A','L','_','M','A','C','H','I','N','E',0},
  56                     HKEY_LOCAL_MACHINE},
  57     {{'H','K','E','Y','_','U','S','E','R','S',0},
  58                     HKEY_USERS},
  59     {{'H','K','E','Y','_','P','E','R','F','O','R','M','A','N','C','E','_','D','A','T','A',0},
  60                     HKEY_PERFORMANCE_DATA},
  61     {{'H','K','E','Y','_','D','Y','N','_','D','A','T','A',0},
  62                     HKEY_DYN_DATA},
  63     {{'H','K','E','Y','_','C','U','R','R','E','N','T','_','C','O','N','F','I','G',0},
  64                     HKEY_CURRENT_CONFIG},
  65     {{'H','K','C','R',0}, HKEY_CLASSES_ROOT},
  66     {{'H','K','C','U',0}, HKEY_CURRENT_USER},
  67     {{'H','K','L','M',0}, HKEY_LOCAL_MACHINE},
  68     {{'H','K','U',0},     HKEY_USERS},
  69     {{'H','K','P','D',0}, HKEY_PERFORMANCE_DATA},
  70     {{'H','K','D','D',0}, HKEY_DYN_DATA},
  71     {{'H','K','C','C',0}, HKEY_CURRENT_CONFIG}
  72 };
  73
  74 typedef struct rep_list_str {
  75     LPOLESTR key;
  76     LPOLESTR item;
  77     int key_len;
  78     struct rep_list_str *next;
  79 } rep_list;
  80
  81 typedef struct {
  82     const IRegistrarVtbl *lpVtbl;
  83     LONG ref;
  84     rep_list *rep;
  85 } Registrar;
  86
  87 typedef struct {
  88     LPOLESTR str;
  89     DWORD alloc;
  90     DWORD len;
  91 } strbuf;
  92
  93 static void strbuf_init(strbuf *buf)
  94 {
  95     buf->str = HeapAlloc(GetProcessHeap(), 0, 128*sizeof(WCHAR));
  96     buf->alloc = 128;
  97     buf->len = 0;
  98 }
  99
 100 static void strbuf_write(LPCOLESTR str, strbuf *buf, int len)
 101 {
 102     if(len == -1)
 103         len = lstrlenW(str);
 104     if(buf->len+len+1 >= buf->alloc) {
 105         buf->alloc = (buf->len+len)<<1;
 106         buf->str = HeapReAlloc(GetProcessHeap(), 0, buf->str, buf->alloc*sizeof(WCHAR));
 107     }
 108     memcpy(buf->str+buf->len, str, len*sizeof(OLECHAR));
 109     buf->len += len;
 110     buf->str[buf->len] = '\0';
 111 }
 112
 113 static HRESULT get_word(LPCOLESTR *str, strbuf *buf)
 114 {
 115     LPCOLESTR iter, iter2 = *str;
 116
 117     buf->len = 0;
 118     buf->str[0] = '\0';
 119
 120     while(isspaceW(*iter2))
 121         iter2++;
 122     iter = iter2;
 123     if(!*iter) {
 124         *str = iter;
 125         return S_OK;
 126     }
 127
 128     if(*iter == '{' || *iter == '}' || *iter == '=') {
 129         strbuf_write(iter++, buf, 1);
 130     }else if(*iter == '\'') {
 131         iter2 = ++iter;
 132         iter = strchrW(iter, '\'');
 133         if(!iter) {
 134             WARN("Unexpected end of script\n");
 135             *str = iter;
 136             return DISP_E_EXCEPTION;
 137         }
 138         strbuf_write(iter2, buf, iter-iter2);
 139         iter++;
 140     }else {
 141         while(*iter && !isspaceW(*iter))
 142             iter++;
 143         strbuf_write(iter2, buf, iter-iter2);
 144     }
 145
 146     while(isspaceW(*iter))
 147         iter++;
 148     *str = iter;
 149     return S_OK;
 150 }
 151
 152 static HRESULT do_preprocess(Registrar *This, LPCOLESTR data, strbuf *buf)
 153 {
 154     LPCOLESTR iter, iter2 = data;
 155     rep_list *rep_iter;
 156     static const WCHAR wstr[] = {'%',0};
 157
 158     iter = strchrW(data, '%');
 159     while(iter) {
 160         strbuf_write(iter2, buf, iter-iter2);
 161
 162         iter2 = ++iter;
 163         if(!*iter2)
 164             return DISP_E_EXCEPTION;
 165         iter = strchrW(iter2, '%');
 166         if(!iter)
 167             return DISP_E_EXCEPTION;
 168
 169         if(iter == iter2) {
 170             strbuf_write(wstr, buf, 1);
 171         }else {
 172             for(rep_iter = This->rep; rep_iter; rep_iter = rep_iter->next) {
 173                 if(rep_iter->key_len == iter-iter2
 174                         && !memicmpW(iter2, rep_iter->key, rep_iter->key_len))
 175                     break;
 176             }
 177             if(!rep_iter) {
 178                 WARN("Could not find replacement: %s\n", debugstr_wn(iter2, iter-iter2));
 179                 return DISP_E_EXCEPTION;
 180             }
 181
 182             strbuf_write(rep_iter->item, buf, -1);
 183         }
 184
 185         iter2 = ++iter;
 186         iter = strchrW(iter, '%');
 187     }
 188
 189     strbuf_write(iter2, buf, -1);
 190     TRACE("%s\n", debugstr_w(buf->str));
 191
 192     return S_OK;
 193 }
 194
 195 static HRESULT do_process_key(LPCOLESTR *pstr, HKEY parent_key, strbuf *buf, BOOL do_register)
 196 {
 197     LPCOLESTR iter = *pstr;
 198     HRESULT hres;
 199     LONG lres;
 200     HKEY hkey = 0;
 201     strbuf name;
 202
 203     enum {
 204         NORMAL,
 205         NO_REMOVE,
 206         IS_VAL,
 207         FORCE_REMOVE,
 208         DO_DELETE
 209     } key_type = NORMAL;
 210
 211     static const WCHAR wstrNoRemove[] = {'N','o','R','e','m','o','v','e',0};
 212     static const WCHAR wstrForceRemove[] = {'F','o','r','c','e','R','e','m','o','v','e',0};
 213     static const WCHAR wstrDelete[] = {'D','e','l','e','t','e',0};
 214     static const WCHAR wstrval[] = {'v','a','l',0};
 215
 216     iter = *pstr;
 217     hres = get_word(&iter, buf);
 218     if(FAILED(hres))
 219         return hres;
 220     strbuf_init(&name);
 221
 222     while(buf->str[1] || buf->str[0] != '}') {
 223         key_type = NORMAL;
 224         if(!lstrcmpiW(buf->str, wstrNoRemove))
 225             key_type = NO_REMOVE;
 226         else if(!lstrcmpiW(buf->str, wstrForceRemove))
 227             key_type = FORCE_REMOVE;
 228         else if(!lstrcmpiW(buf->str, wstrval))
 229             key_type = IS_VAL;
 230         else if(!lstrcmpiW(buf->str, wstrDelete))
 231             key_type = DO_DELETE;
 232
 233         if(key_type != NORMAL) {
 234             hres = get_word(&iter, buf);
 235             if(FAILED(hres))
 236                 break;
 237         }
 238         TRACE("name = %s\n", debugstr_w(buf->str));
 239
 240         if(do_register) {
 241             if(key_type == IS_VAL) {
 242                 hkey = parent_key;
 243                 strbuf_write(buf->str, &name, -1);
 244             }else if(key_type == DO_DELETE) {
 245                 TRACE("Deleting %s\n", debugstr_w(buf->str));
 246                 lres = SHDeleteKeyW(parent_key, buf->str);
 247             }else {
 248                 if(key_type == FORCE_REMOVE)
 249                     SHDeleteKeyW(parent_key, buf->str);
 250                 lres = RegCreateKeyW(parent_key, buf->str, &hkey);
 251                 if(lres != ERROR_SUCCESS) {
 252                     WARN("Could not create(open) key: %08lx\n", lres);
 253                     hres = HRESULT_FROM_WIN32(lres);
 254                     break;
 255                 }
 256             }
 257         }else if(key_type != IS_VAL && key_type != DO_DELETE) {
 258             strbuf_write(buf->str, &name, -1);
 259             lres = RegOpenKeyW(parent_key, buf->str, &hkey);
 260               if(lres != ERROR_SUCCESS)
 261                 WARN("Could not open key %s: %08lx\n", debugstr_w(name.str), lres);
 262         }
 263
 264         if(key_type != DO_DELETE && *iter == '=') {
 265             iter++;
 266             hres = get_word(&iter, buf);
 267             if(FAILED(hres))
 268                 break;
 269             if(buf->len != 1) {
 270                 WARN("Wrong registry type: %s\n", debugstr_w(buf->str));
 271                 hres = DISP_E_EXCEPTION;
 272                 break;
 273             }
 274             if(do_register) {
 275                 switch(buf->str[0]) {
 276                 case 's':
 277                     hres = get_word(&iter, buf);
 278                     if(FAILED(hres))
 279                         break;
 280                     lres = RegSetValueExW(hkey, name.len ? name.str :  NULL, 0, REG_SZ, (PBYTE)buf->str,
 281                             (lstrlenW(buf->str)+1)*sizeof(WCHAR));
 282                     if(lres != ERROR_SUCCESS) {
 283                         WARN("Could set value of key: %08lx\n", lres);
 284                         hres = HRESULT_FROM_WIN32(lres);
 285                         break;
 286                     }
 287                     break;
 288                 case 'd': {
 289                     DWORD dw;
 290                     if(*iter == '0' && iter[1] == 'x') {
 291                         iter += 2;
 292                         dw = strtolW(iter, (WCHAR**)&iter, 16);
 293                     }else {
 294                         dw = strtolW(iter, (WCHAR**)&iter, 10);
 295                     }
 296                     lres = RegSetValueExW(hkey, name.len ? name.str :  NULL, 0, REG_DWORD,
 297                             (PBYTE)&dw, sizeof(dw));
 298                     if(lres != ERROR_SUCCESS) {
 299                         WARN("Could set value of key: %08lx\n", lres);
 300                         hres = HRESULT_FROM_WIN32(lres);
 301                         break;
 302                     }
 303                     break;
 304                 }
 305                 default:
 306                     WARN("Wrong resource type: %s\n", debugstr_w(buf->str));
 307                     hres = DISP_E_EXCEPTION;
 308                 };
 309                 if(FAILED(hres))
 310                     break;
 311             }else {
 312                 if(*iter == '-')
 313                     iter++;
 314                 hres = get_word(&iter, buf);
 315                 if(FAILED(hres))
 316                     break;
 317             }
 318         }else if(key_type == IS_VAL) {
 319             WARN("value not set!\n");
 320             hres = DISP_E_EXCEPTION;
 321             break;
 322         }
 323
 324         if(key_type != IS_VAL && key_type != DO_DELETE && *iter == '{') {
 325             hres = get_word(&iter, buf);
 326             if(FAILED(hres))
 327                 break;
 328             hres = do_process_key(&iter, hkey, buf, do_register);
 329             if(FAILED(hres))
 330                 break;
 331         }
 332
 333         TRACE("%x %x\n", do_register, key_type);
 334         if(!do_register && (key_type == NORMAL || key_type == FORCE_REMOVE)) {
 335             TRACE("Deleting %s\n", debugstr_w(name.str));
 336             RegDeleteKeyW(parent_key, name.str);
 337         }
 338
 339         if(hkey && key_type != IS_VAL)
 340             RegCloseKey(hkey);
 341         hkey = 0;
 342         name.len = 0;
 343
 344         hres = get_word(&iter, buf);
 345         if(FAILED(hres))
 346             break;
 347     }
 348
 349     HeapFree(GetProcessHeap(), 0, name.str);
 350     if(hkey && key_type != IS_VAL)
 351         RegCloseKey(hkey);
 352     *pstr = iter;
 353     return hres;
 354 }
 355
 356 static HRESULT do_process_root_key(LPCOLESTR data, BOOL do_register)
 357 {
 358     LPCOLESTR iter = data;
 359     strbuf buf;
 360     HRESULT hres = S_OK;
 361     int i;
 362
 363     strbuf_init(&buf);
 364     hres = get_word(&iter, &buf);
 365     if(FAILED(hres))
 366         return hres;
 367
 368     while(*iter) {
 369         if(!buf.len) {
 370             WARN("ward.len == 0, failed\n");
 371             hres = DISP_E_EXCEPTION;
 372             break;
 373         }
 374         for(i=0; i<sizeof(root_keys)/sizeof(root_keys[0]); i++) {
 375             if(!lstrcmpiW(buf.str, root_keys[i].name))
 376                 break;
 377         }
 378         if(i == sizeof(root_keys)/sizeof(root_keys[0])) {
 379             WARN("Wrong root key name: %s\n", debugstr_w(buf.str));
 380             hres = DISP_E_EXCEPTION;
 381             break;
 382         }
 383         hres = get_word(&iter, &buf);
 384         if(FAILED(hres))
 385             break;
 386         if(buf.str[1] || buf.str[0] != '{') {
 387             WARN("Failed, expected '{', got %s\n", debugstr_w(buf.str));
 388             hres = DISP_E_EXCEPTION;
 389             break;
 390         }
 391         hres = do_process_key(&iter, root_keys[i].key, &buf, do_register);
 392         if(FAILED(hres)) {
 393             WARN("Processing key failed: %08lx\n", hres);
 394             break;
 395         }
 396         hres = get_word(&iter, &buf);
 397         if(FAILED(hres))
 398             break;
 399     }
 400     HeapFree(GetProcessHeap(), 0, buf.str);
 401     return hres;
 402 }
 403
 404 static HRESULT string_register(Registrar *This, LPCOLESTR data, BOOL do_register)
 405 {
 406     strbuf buf;
 407     HRESULT hres;
 408
 409     TRACE("(%p %s %x)\n", This, debugstr_w(data), do_register);
 410
 411     strbuf_init(&buf);
 412     hres = do_preprocess(This, data, &buf);
 413     if(FAILED(hres)) {
 414         WARN("preprocessing failed!\n");
 415         HeapFree(GetProcessHeap(), 0, buf.str);
 416         return hres;
 417     }
 418
 419     hres = do_process_root_key(buf.str, do_register);
 420     if(FAILED(hres) && do_register)
 421         do_process_root_key(buf.str, FALSE);
 422
 423     HeapFree(GetProcessHeap(), 0, buf.str);
 424     return hres;
 425 }
 426
 427 static HRESULT resource_register(Registrar *This, LPCOLESTR resFileName,
 428                         LPCOLESTR szID, LPCOLESTR szType, BOOL do_register)
 429 {
 430     HINSTANCE hins;
 431     HRSRC src;
 432     LPSTR regstra;
 433     LPWSTR regstrw;
 434     DWORD len, reslen;
 435     HRESULT hres;
 436
 437     hins = LoadLibraryExW(resFileName, NULL, LOAD_LIBRARY_AS_DATAFILE);
 438     if(hins) {
 439         src = FindResourceW(hins, szID, szType);
 440         if(src) {
 441             regstra = (LPSTR)LoadResource(hins, src);
 442             reslen = SizeofResource(hins, src);
 443             if(regstra) {
 444                 len = MultiByteToWideChar(CP_ACP, 0, regstra, reslen, NULL, 0)+1;
 445                 regstrw = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, len*sizeof(WCHAR));
 446                 MultiByteToWideChar(CP_ACP, 0, regstra, reslen, regstrw, -1);
 447                 regstrw[len-1] = '\0';
 448
 449                 hres = string_register(This, regstrw, do_register);
 450
 451                 HeapFree(GetProcessHeap(), 0, regstrw);
 452             }else {
 453                 WARN("could not load resource\n");
 454                 hres = HRESULT_FROM_WIN32(GetLastError());
 455             }
 456             HeapFree(GetProcessHeap(), 0, regstra);
 457         }else {
 458             WARN("Could not find source\n");
 459             hres = HRESULT_FROM_WIN32(GetLastError());
 460         }
 461         FreeLibrary(hins);
 462     }else {
 463         WARN("Could not load resource file\n");
 464         hres = HRESULT_FROM_WIN32(GetLastError());
 465     }
 466
 467     return hres;
 468 }
 469
 470 static HRESULT file_register(Registrar *This, LPCOLESTR fileName, BOOL do_register)
 471 {
 472     HANDLE file;
 473     DWORD filelen, len;
 474     LPWSTR regstrw;
 475     LPSTR regstra;
 476     LRESULT lres;
 477     HRESULT hres;
 478
 479     file = CreateFileW(fileName, GENERIC_READ, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_READONLY, NULL);
 480     if(file) {
 481         filelen = GetFileSize(file, NULL);
 482         regstra = HeapAlloc(GetProcessHeap(), 0, filelen);
 483         lres = ReadFile(file, regstra, filelen, NULL, NULL);
 484         if(lres == ERROR_SUCCESS) {
 485             len = MultiByteToWideChar(CP_ACP, 0, regstra, filelen, NULL, 0)+1;
 486             regstrw = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, len*sizeof(WCHAR));
 487             MultiByteToWideChar(CP_ACP, 0, regstra, filelen, regstrw, -1);
 488             regstrw[len-1] = '\0';
 489
 490             hres = string_register(This, regstrw, do_register);
 491
 492             HeapFree(GetProcessHeap(), 0, regstrw);
 493         }else {
 494             WARN("Failed to read faile\n");
 495             hres = HRESULT_FROM_WIN32(lres);
 496         }
 497         HeapFree(GetProcessHeap(), 0, regstra);
 498         CloseHandle(file);
 499     }else {
 500         WARN("Could not open file\n");
 501         hres = HRESULT_FROM_WIN32(GetLastError());
 502     }
 503
 504     return hres;
 505 }
 506
 507 static HRESULT WINAPI Registrar_QueryInterface(IRegistrar *iface, REFIID riid, void **ppvObject)
 508 {
 509     TRACE("(%p)->(%s %p\n", iface, debugstr_guid(riid), ppvObject);
 510
 511     if(IsEqualGUID(&IID_IUnknown, riid) || IsEqualGUID(&IID_IRegistrar, riid)) {
 512         IRegistrar_AddRef(iface);
 513         *ppvObject = iface;
 514         return S_OK;
 515     }
 516     return E_NOINTERFACE;
 517 }
 518
 519 static ULONG WINAPI Registrar_AddRef(IRegistrar *iface)
 520 {
 521     Registrar *This = (Registrar*)iface;
 522     ULONG ref = InterlockedIncrement(&This->ref);
 523     TRACE("(%p) ->%ld\n", This, ref);
 524     return ref;
 525 }
 526
 527 static ULONG WINAPI Registrar_Release(IRegistrar *iface)
 528 {
 529     Registrar *This = (Registrar*)iface;
 530     ULONG ref = InterlockedDecrement(&This->ref);
 531
 532     TRACE("(%p) ->%ld\n", This, ref);
 533     if(!ref) {
 534         IRegistrar_ClearReplacements(iface);
 535         HeapFree(GetProcessHeap(), 0, This);
 536         InterlockedDecrement(&dll_count);
 537     }
 538     return ref;
 539 }
 540
 541 static HRESULT WINAPI Registrar_AddReplacement(IRegistrar *iface, LPCOLESTR Key, LPCOLESTR item)
 542 {
 543     Registrar *This = (Registrar*)iface;
 544     int len;
 545     rep_list *new_rep;
 546
 547     TRACE("(%p)->(%s %s)\n", This, debugstr_w(Key), debugstr_w(item));
 548
 549     new_rep = HeapAlloc(GetProcessHeap(), 0, sizeof(rep_list));
 550
 551     new_rep->key_len  = lstrlenW(Key);
 552     new_rep->key = HeapAlloc(GetProcessHeap(), 0, new_rep->key_len*sizeof(OLECHAR)+1);
 553     memcpy(new_rep->key, Key, (new_rep->key_len+1)*sizeof(OLECHAR));
 554
 555     len = lstrlenW(item)+1;
 556     new_rep->item = HeapAlloc(GetProcessHeap(), 0, len*sizeof(OLECHAR));
 557     memcpy(new_rep->item, item, len*sizeof(OLECHAR));
 558
 559     new_rep->next = This->rep;
 560     This->rep = new_rep;
 561
 562     return S_OK;
 563 }
 564
 565 static HRESULT WINAPI Registrar_ClearReplacements(IRegistrar *iface)
 566 {
 567     Registrar *This = (Registrar*)iface;
 568     rep_list *iter, *iter2;
 569
 570     TRACE("(%p)\n", This);
 571
 572     if(!This->rep)
 573         return S_OK;
 574
 575     iter = This->rep;
 576     while(iter) {
 577         iter2 = iter->next;
 578         HeapFree(GetProcessHeap(), 0, iter->key);
 579         HeapFree(GetProcessHeap(), 0, iter->item);
 580         HeapFree(GetProcessHeap(), 0, iter);
 581         iter = iter2;
 582     }
 583
 584     This->rep = NULL;
 585     return S_OK;
 586 }
 587
 588 static HRESULT WINAPI Registrar_ResourceRegisterSz(IRegistrar* iface, LPCOLESTR resFileName,
 589                 LPCOLESTR szID, LPCOLESTR szType)
 590 {
 591     Registrar *This = (Registrar*)iface;
 592     TRACE("(%p)->(%s %s %s)\n", This, debugstr_w(resFileName), debugstr_w(szID), debugstr_w(szType));
 593     return resource_register(This, resFileName, szID, szType, TRUE);
 594 }
 595
 596 static HRESULT WINAPI Registrar_ResourceUnregisterSz(IRegistrar* iface, LPCOLESTR resFileName,
 597                 LPCOLESTR szID, LPCOLESTR szType)
 598 {
 599     Registrar *This = (Registrar*)iface;
 600     TRACE("(%p)->(%s %s %s)\n", This, debugstr_w(resFileName), debugstr_w(szID), debugstr_w(szType));
 601     return resource_register(This, resFileName, szID, szType, FALSE);
 602 }
 603
 604 static HRESULT WINAPI Registrar_FileRegister(IRegistrar* iface, LPCOLESTR fileName)
 605 {
 606     Registrar *This = (Registrar*)iface;
 607     TRACE("(%p)->(%s)\n", This, debugstr_w(fileName));
 608     return file_register(This, fileName, TRUE);
 609 }
 610
 611 static HRESULT WINAPI Registrar_FileUnregister(IRegistrar* iface, LPCOLESTR fileName)
 612 {
 613     Registrar *This = (Registrar*)iface;
 614     FIXME("(%p)->(%s)\n", This, debugstr_w(fileName));
 615     return file_register(This, fileName, FALSE);
 616 }
 617
 618 static HRESULT WINAPI Registrar_StringRegister(IRegistrar* iface, LPCOLESTR data)
 619 {
 620     Registrar *This = (Registrar*)iface;
 621     TRACE("(%p)->(%s)\n", This, debugstr_w(data));
 622     return string_register(This, data, TRUE);
 623 }
 624
 625 static HRESULT WINAPI Registrar_StringUnregister(IRegistrar* iface, LPCOLESTR data)
 626 {
 627     Registrar *This = (Registrar*)iface;
 628     TRACE("(%p)->(%s)\n", This, debugstr_w(data));
 629     return string_register(This, data, FALSE);
 630 }
 631
 632 static HRESULT WINAPI Registrar_ResourceRegister(IRegistrar* iface, LPCOLESTR resFileName,
 633                 UINT nID, LPCOLESTR szType)
 634 {
 635     Registrar *This = (Registrar*)iface;
 636     TRACE("(%p)->(%s %d %s)\n", iface, debugstr_w(resFileName), nID, debugstr_w(szType));
 637     return resource_register(This, resFileName, MAKEINTRESOURCEW(nID), szType, TRUE);
 638 }
 639
 640 static HRESULT WINAPI Registrar_ResourceUnregister(IRegistrar* iface, LPCOLESTR resFileName,
 641                 UINT nID, LPCOLESTR szType)
 642 {
 643     Registrar *This = (Registrar*)iface;
 644     TRACE("(%p)->(%s %d %s)\n", This, debugstr_w(resFileName), nID, debugstr_w(szType));
 645     return resource_register(This, resFileName, MAKEINTRESOURCEW(nID), szType, FALSE);
 646 }
 647
 648 static const IRegistrarVtbl RegistrarVtbl = {
 649     Registrar_QueryInterface,
 650     Registrar_AddRef,
 651     Registrar_Release,
 652     Registrar_AddReplacement,
 653     Registrar_ClearReplacements,
 654     Registrar_ResourceRegisterSz,
 655     Registrar_ResourceUnregisterSz,
 656     Registrar_FileRegister,
 657     Registrar_FileUnregister,
 658     Registrar_StringRegister,
 659     Registrar_StringUnregister,
 660     Registrar_ResourceRegister,
 661     Registrar_ResourceUnregister,
 662 };
 663
 664 static HRESULT Registrar_create(LPUNKNOWN pUnkOuter, REFIID riid, void **ppvObject)
 665 {
 666     Registrar *ret;
 667
 668     if(!IsEqualGUID(&IID_IUnknown, riid) && !IsEqualGUID(&IID_IRegistrar, riid))
 669         return E_NOINTERFACE;
 670
 671     ret = HeapAlloc(GetProcessHeap(), 0, sizeof(Registrar));
 672     ret->lpVtbl = &RegistrarVtbl;
 673     ret->ref = 1;
 674     ret->rep = NULL;
 675     *ppvObject = ret;
 676
 677     InterlockedIncrement(&dll_count);
 678
 679     return S_OK;
 680 }
 681
 682 /**************************************************************
 683  * ClassFactory implementation
 684  */
 685
 686 static HRESULT WINAPI RegistrarCF_QueryInterface(IClassFactory *iface, REFIID riid, void **ppvObject)
 687 {
 688     TRACE("(%p)->(%s %p)\n", iface, debugstr_guid(riid), ppvObject);
 689
 690     if(IsEqualGUID(&IID_IUnknown, riid) || IsEqualGUID(&IID_IRegistrar, riid)) {
 691         *ppvObject = iface;
 692         return S_OK;
 693     }
 694
 695     return E_NOINTERFACE;
 696 }
 697
 698 static ULONG WINAPI RegistrarCF_AddRef(IClassFactory *iface)
 699 {
 700     InterlockedIncrement(&dll_count);
 701     return 2;
 702 }
 703
 704 static ULONG WINAPI RegistrarCF_Release(IClassFactory *iface)
 705 {
 706     InterlockedDecrement(&dll_count);
 707     return 1;
 708 }
 709
 710 static HRESULT WINAPI RegistrarCF_CreateInstance(IClassFactory *iface, LPUNKNOWN pUnkOuter,
 711                                                 REFIID riid, void **ppvObject)
 712 {
 713     TRACE("(%p)->(%s %p)\n", iface, debugstr_guid(riid), ppvObject);
 714     return Registrar_create(pUnkOuter, riid, ppvObject);
 715 }
 716
 717 static HRESULT WINAPI RegistrarCF_LockServer(IClassFactory *iface, BOOL lock)
 718 {
 719     TRACE("(%p)->(%x)\n", iface, lock);
 720
 721     if(lock)
 722         InterlockedIncrement(&dll_count);
 723     else
 724         InterlockedDecrement(&dll_count);
 725
 726     return S_OK;
 727 }
 728
 729 static const IClassFactoryVtbl IRegistrarCFVtbl = {
 730     RegistrarCF_QueryInterface,
 731     RegistrarCF_AddRef,
 732     RegistrarCF_Release,
 733     RegistrarCF_CreateInstance,
 734     RegistrarCF_LockServer
 735 };
 736
 737 static IClassFactory RegistrarCF = { &IRegistrarCFVtbl };
 738
 739 /**************************************************************
 740  * DllGetClassObject (ATL.2)
 741  */
 742 HRESULT WINAPI DllGetClassObject(REFCLSID clsid, REFIID riid, LPVOID *ppvObject)
 743 {
 744     TRACE("(%s %s %p)", debugstr_guid(clsid), debugstr_guid(riid), ppvObject);
 745
 746     if(IsEqualGUID(&CLSID_ATLRegistrar, clsid)) {
 747         *ppvObject = &RegistrarCF;
 748         return S_OK;
 749     }
 750
 751     FIXME("Not supported class %s\n", debugstr_guid(clsid));
 752     return CLASS_E_CLASSNOTAVAILABLE;
 753 }
 754
 755 extern HINSTANCE hInst;
 756
 757 static HRESULT do_register_dll_server(LPCOLESTR wszDll, LPCOLESTR wszId, BOOL do_register)
 758 {
 759     WCHAR buf[MAX_PATH];
 760     HRESULT hres;
 761     IRegistrar *pRegistrar;
 762
 763     static const WCHAR wszModule[] = {'M','O','D','U','L','E',0};
 764     static const WCHAR wszRegistry[] = {'R','E','G','I','S','T','R','Y',0};
 765     static const WCHAR wszCLSID_ATLRegistrar[] =
 766             {'C','L','S','I','D','_','A','T','L','R','e','g','i','s','t','r','a','r',0};
 767
 768     Registrar_create(NULL, &IID_IRegistrar, (void**)&pRegistrar);
 769     IRegistrar_AddReplacement(pRegistrar, wszModule, wszDll);
 770
 771     StringFromGUID2(&CLSID_ATLRegistrar, buf, sizeof(buf)/sizeof(buf[0]));
 772     IRegistrar_AddReplacement(pRegistrar, wszCLSID_ATLRegistrar, buf);
 773
 774     if(do_register)
 775         hres = IRegistrar_ResourceRegisterSz(pRegistrar, wszDll, wszId, wszRegistry);
 776     else
 777         hres = IRegistrar_ResourceUnregisterSz(pRegistrar, wszDll, wszId, wszRegistry);
 778
 779     IRegistrar_Release(pRegistrar);
 780     return hres;
 781 }
 782
 783 static HRESULT do_register_server(BOOL do_register)
 784 {
 785     static const WCHAR wszDll[] = {'a','t','l','.','d','l','l',0};
 786     UINT nID = 101;
 787     return do_register_dll_server(wszDll, (LPCOLESTR) nID, do_register);
 788 }
 789
 790 /***********************************************************************
 791  *           AtlModuleUpdateRegistryFromResourceD         [ATL.@]
 792  *
 793  */
 794 HRESULT WINAPI AtlModuleUpdateRegistryFromResourceD(_ATL_MODULEW* pM, LPCOLESTR lpszRes,
 795                 BOOL bRegister, struct _ATL_REGMAP_ENTRY* pMapEntries, IRegistrar* pReg)
 796 {
 797     HINSTANCE lhInst = pM->m_hInst;
 798     /* everything inside this function below this point
 799      * should go into atl71.AtlUpdateRegistryFromResourceD
 800      */
 801     WCHAR module_name[MAX_PATH];
 802
 803     if(pMapEntries || pReg) {
 804         FIXME("MapEntries and Registrar parameter not supported\n");
 805         return E_FAIL;
 806     }
 807
 808     if(!GetModuleFileNameW(lhInst, module_name, MAX_PATH)) {
 809         FIXME("hinst %p: did not get module name\n",
 810         lhInst);
 811         return E_FAIL;
 812     }
 813
 814     TRACE("%p (%s), %s, %d, %p, %p\n", hInst, debugstr_w(module_name),
 815         debugstr_w(lpszRes), bRegister, pMapEntries, pReg);
 816
 817     return do_register_dll_server(module_name, lpszRes, bRegister);
 818 }
 819
 820 /***********************************************************************
 821  *              DllRegisterServer (ATL.@)
 822  */
 823 HRESULT WINAPI DllRegisterServer(void)
 824 {
 825     TRACE("\n");
 826     return do_register_server(TRUE);
 827 }
 828
 829 /***********************************************************************
 830  *              DllRegisterServer (ATL.@)
 831  */
 832 HRESULT WINAPI DllUnregisterServer(void)
 833 {
 834     TRACE("\n");
 835     return do_register_server(FALSE);
 836 }
 837
 838 /***********************************************************************
 839  *              DllCanUnloadNow (ATL.@)
 840  */
 841 HRESULT WINAPI DllCanUnloadNow(void)
 842 {
 843     TRACE("dll_count = %lu\n", dll_count);
 844     return dll_count ? S_FALSE : S_OK;
 845 }