commit d6db3732da14708210e30c03039f0eba51cf4bdc
author David Hedberg <david.hedberg@gmail.com> Sat Jul 31 20:02:50 2010 +0200
committer Alexandre Julliard <julliard@winehq.org> Mon Aug 02 15:31:33 2010 +0200
tree 67929e017cb1fb961f719c12bf610f4b97db376b
parent 87212d510582d5889122b05438075e287235a3a3

explorerframe: Add some infrastructure to register the NamespaceTreeControl CLSID.

dlls/explorerframe/explorerframe_main.c

diff --git a/dlls/explorerframe/explorerframe_main.c b/dlls/explorerframe/explorerframe_main.c
index fd958ef..dd3c0cf 100644
--- a/dlls/explorerframe/explorerframe_main.c
+++ b/dlls/explorerframe/explorerframe_main.c
@@ -19,6 +19,7 @@
  */
 
 #include <stdarg.h>
+#include <stdio.h>
 
 #define COBJMACROS
 #define NONAMELESSUNION
@@ -26,8 +27,11 @@
 
 #include "windef.h"
 #include "winbase.h"
+#include "winuser.h"
 #include "winreg.h"
 #include "shlwapi.h"
+#include "advpub.h"
+#include "shobjidl.h"
 
 #include "wine/debug.h"
 
@@ -203,3 +207,80 @@
 
     return CLASS_E_CLASSNOTAVAILABLE;
 }
+
+/*************************************************************************
+ *          Register/Unregister DLL, based on shdocvw/factory.c
+ */
+static HRESULT reg_install(LPCSTR section, STRTABLEA *strtable)
+{
+    HRESULT (WINAPI *pRegInstall)(HMODULE hm, LPCSTR pszSection, const STRTABLEA* pstTable);
+    HMODULE hadvpack;
+    HRESULT hres;
+
+    static const WCHAR advpackW[] = {'a','d','v','p','a','c','k','.','d','l','l',0};
+
+    hadvpack = LoadLibraryW(advpackW);
+    pRegInstall = (void *)GetProcAddress(hadvpack, "RegInstall");
+
+    hres = pRegInstall(explorerframe_hinstance, section, strtable);
+
+    FreeLibrary(hadvpack);
+    return hres;
+}
+
+#define INF_SET_CLSID(clsid)                  \
+    do                                        \
+    {                                         \
+        static CHAR name[] = "CLSID_" #clsid; \
+                                              \
+        pse[i].pszName = name;                \
+        clsids[i++] = &CLSID_ ## clsid;       \
+    } while (0)
+
+static HRESULT register_server(BOOL doregister)
+{
+    STRTABLEA strtable;
+    STRENTRYA pse[1];
+    static CLSID const *clsids[1];
+    unsigned int i = 0;
+    HRESULT hres;
+
+    INF_SET_CLSID(NamespaceTreeControl);
+
+    for(i = 0; i < sizeof(pse)/sizeof(pse[0]); i++)
+    {
+        pse[i].pszValue = HeapAlloc(GetProcessHeap(), 0, 39);
+        sprintf(pse[i].pszValue, "{%08X-%04X-%04X-%02X%02X-%02X%02X%02X%02X%02X%02X}",
+                clsids[i]->Data1, clsids[i]->Data2, clsids[i]->Data3, clsids[i]->Data4[0],
+                clsids[i]->Data4[1], clsids[i]->Data4[2], clsids[i]->Data4[3], clsids[i]->Data4[4],
+                clsids[i]->Data4[5], clsids[i]->Data4[6], clsids[i]->Data4[7]);
+    }
+
+    strtable.cEntries = sizeof(pse)/sizeof(pse[0]);
+    strtable.pse = pse;
+
+    hres = reg_install(doregister ? "RegisterDll" : "UnregisterDll", &strtable);
+
+    for(i=0; i < sizeof(pse)/sizeof(pse[0]); i++)
+        HeapFree(GetProcessHeap(), 0, pse[i].pszValue);
+
+    return hres;
+}
+
+#undef INF_SET_CLSID
+
+/*************************************************************************
+ *          DllRegisterServer (ExplorerFrame.@)
+ */
+HRESULT WINAPI DllRegisterServer(void)
+{
+    return register_server(TRUE);
+}
+
+/*************************************************************************
+ *          DllUnregisterServer (ExplorerFrame.@)
+ */
+HRESULT WINAPI DllUnregisterServer(void)
+{
+    return register_server(FALSE);
+}