commit 6e5bfbfe17d71605973f16ed25d30934948ea1b6
author Roy Shea <royshea@gmail.com> Mon Jul 28 12:06:24 2008 -0700
committer Alexandre Julliard <julliard@winehq.org> Tue Jul 29 12:50:35 2008 +0200
tree 825850d972c2166459b6dc96b9cc2d082531af40
parent 170023e30d0b8e7e636524f22405508a8e6b1c95

mstask: Implemented DllRegisterServer.

dlls/mstask/mstask_main.c

diff --git a/dlls/mstask/mstask_main.c b/dlls/mstask/mstask_main.c
index 61e0a99..53c3449 100644
--- a/dlls/mstask/mstask_main.c
+++ b/dlls/mstask/mstask_main.c
@@ -16,11 +16,18 @@
  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  */
 
+#include <stdio.h>
+
 #include "mstask_private.h"
+#include "winreg.h"
+#include "advpub.h"
+
 #include "wine/debug.h"
 
+
 WINE_DEFAULT_DEBUG_CHANNEL(mstask);
 
+static HINSTANCE hInst;
 LONG dll_ref = 0;
 
 BOOL WINAPI DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved)
@@ -33,6 +40,7 @@
             return FALSE;
         case DLL_PROCESS_ATTACH:
             DisableThreadLibraryCalls(hinstDLL);
+            hInst = hinstDLL;
             break;
         case DLL_PROCESS_DETACH:
             break;
@@ -57,3 +65,93 @@
 {
     return dll_ref != 0 ? S_FALSE : S_OK;
 }
+
+static inline char *mstask_strdup(const char *s)
+{
+    size_t n = strlen(s) + 1;
+    char *d = HeapAlloc(GetProcessHeap(), 0, n);
+    return d ? memcpy(d, s, n) : NULL;
+}
+
+static HRESULT init_register_strtable(STRTABLEA *strtable)
+{
+#define CLSID_EXPANSION_ENTRY(id) { "CLSID_" #id, &CLSID_ ## id }
+    static const struct
+    {
+        const char *name;
+        const CLSID *clsid;
+    }
+    expns[] =
+    {
+        CLSID_EXPANSION_ENTRY(CTaskScheduler),
+        CLSID_EXPANSION_ENTRY(CTask)
+    };
+#undef CLSID_EXPANSION_ENTRY
+    static STRENTRYA pse[sizeof expns / sizeof expns[0]];
+    int i;
+
+    strtable->cEntries = sizeof pse / sizeof pse[0];
+    strtable->pse = pse;
+    for (i = 0; i < strtable->cEntries; i++)
+    {
+        static const char dummy_sample[] =
+                "{12345678-1234-1234-1234-123456789012}";
+        const CLSID *clsid = expns[i].clsid;
+        pse[i].pszName = mstask_strdup(expns[i].name);
+        pse[i].pszValue = HeapAlloc(GetProcessHeap(), 0, sizeof dummy_sample);
+        if (!pse[i].pszName || !pse[i].pszValue)
+            return E_OUTOFMEMORY;
+        sprintf(pse[i].pszValue,
+                "{%08X-%04X-%04X-%02X%02X-%02X%02X%02X%02X%02X%02X}",
+                clsid->Data1, clsid->Data2, clsid->Data3, clsid->Data4[0],
+                clsid->Data4[1], clsid->Data4[2], clsid->Data4[3],
+                clsid->Data4[4], clsid->Data4[5], clsid->Data4[6],
+                clsid->Data4[7]);
+    }
+
+    return S_OK;
+}
+
+static void cleanup_register_strtable(STRTABLEA *strtable)
+{
+    int i;
+    for (i = 0; i < strtable->cEntries; i++)
+    {
+        HeapFree(GetProcessHeap(), 0, strtable->pse[i].pszName);
+        HeapFree(GetProcessHeap(), 0, strtable->pse[i].pszValue);
+        if (!strtable->pse[i].pszName || !strtable->pse[i].pszValue)
+            return;
+    }
+}
+
+static HRESULT register_mstask(BOOL do_register)
+{
+    HRESULT hr;
+    STRTABLEA strtable;
+    HMODULE hAdvpack;
+    HRESULT (WINAPI *pRegInstall)(HMODULE hm,
+            LPCSTR pszSection, const STRTABLEA* pstTable);
+    static const WCHAR wszAdvpack[] =
+            {'a','d','v','p','a','c','k','.','d','l','l',0};
+
+    TRACE("(%x)\n", do_register);
+
+    hAdvpack = LoadLibraryW(wszAdvpack);
+    pRegInstall = (void *)GetProcAddress(hAdvpack, "RegInstall");
+
+    hr = init_register_strtable(&strtable);
+    if (SUCCEEDED(hr))
+        hr = pRegInstall(hInst, do_register ? "RegisterDll" : "UnregisterDll",
+                &strtable);
+    cleanup_register_strtable(&strtable);
+
+    if (FAILED(hr))
+        WINE_ERR("RegInstall failed: %08x\n", hr);
+
+    return hr;
+}
+
+HRESULT WINAPI DllRegisterServer(void)
+{
+    return register_mstask(TRUE);
+}