comctl32/listview: Add a TRACE(..) to LISTVIEW_CreateHeader.
[wine] / dlls / mstask / mstask_main.c
1 /*
2  * Copyright (C) 2008 Google (Roy Shea)
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., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
17  */
18
19 #include <stdio.h>
20
21 #include "mstask_private.h"
22 #include "winreg.h"
23 #include "advpub.h"
24
25 #include "wine/debug.h"
26
27
28 WINE_DEFAULT_DEBUG_CHANNEL(mstask);
29
30 static HINSTANCE hInst;
31 LONG dll_ref = 0;
32
33 BOOL WINAPI DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved)
34 {
35     TRACE("(%p, %d, %p)\n", hinstDLL, fdwReason, lpvReserved);
36
37     switch (fdwReason)
38     {
39         case DLL_WINE_PREATTACH:
40             return FALSE;
41         case DLL_PROCESS_ATTACH:
42             DisableThreadLibraryCalls(hinstDLL);
43             hInst = hinstDLL;
44             break;
45         case DLL_PROCESS_DETACH:
46             break;
47     }
48
49     return TRUE;
50 }
51
52 HRESULT WINAPI DllGetClassObject(REFCLSID rclsid, REFIID iid, LPVOID *ppv)
53 {
54     TRACE("(%s %s %p)\n", debugstr_guid(rclsid), debugstr_guid(iid), ppv);
55
56     if (IsEqualGUID(rclsid, &CLSID_CTaskScheduler)) {
57         return IClassFactory_QueryInterface((LPCLASSFACTORY)&MSTASK_ClassFactory, iid, ppv);
58     }
59
60     FIXME("Not supported class: %s\n", debugstr_guid(rclsid));
61     return CLASS_E_CLASSNOTAVAILABLE;
62 }
63
64 HRESULT WINAPI DllCanUnloadNow(void)
65 {
66     return dll_ref != 0 ? S_FALSE : S_OK;
67 }
68
69 static inline char *mstask_strdup(const char *s)
70 {
71     size_t n = strlen(s) + 1;
72     char *d = HeapAlloc(GetProcessHeap(), 0, n);
73     return d ? memcpy(d, s, n) : NULL;
74 }
75
76 static HRESULT init_register_strtable(STRTABLEA *strtable)
77 {
78 #define CLSID_EXPANSION_ENTRY(id) { "CLSID_" #id, &CLSID_ ## id }
79     static const struct
80     {
81         const char *name;
82         const CLSID *clsid;
83     }
84     expns[] =
85     {
86         CLSID_EXPANSION_ENTRY(CTaskScheduler),
87         CLSID_EXPANSION_ENTRY(CTask)
88     };
89 #undef CLSID_EXPANSION_ENTRY
90     static STRENTRYA pse[sizeof expns / sizeof expns[0]];
91     unsigned int i;
92
93     strtable->cEntries = sizeof pse / sizeof pse[0];
94     strtable->pse = pse;
95     for (i = 0; i < strtable->cEntries; i++)
96     {
97         static const char dummy_sample[] =
98                 "{12345678-1234-1234-1234-123456789012}";
99         const CLSID *clsid = expns[i].clsid;
100         pse[i].pszName = mstask_strdup(expns[i].name);
101         pse[i].pszValue = HeapAlloc(GetProcessHeap(), 0, sizeof dummy_sample);
102         if (!pse[i].pszName || !pse[i].pszValue)
103             return E_OUTOFMEMORY;
104         sprintf(pse[i].pszValue,
105                 "{%08X-%04X-%04X-%02X%02X-%02X%02X%02X%02X%02X%02X}",
106                 clsid->Data1, clsid->Data2, clsid->Data3, clsid->Data4[0],
107                 clsid->Data4[1], clsid->Data4[2], clsid->Data4[3],
108                 clsid->Data4[4], clsid->Data4[5], clsid->Data4[6],
109                 clsid->Data4[7]);
110     }
111
112     return S_OK;
113 }
114
115 static void cleanup_register_strtable(STRTABLEA *strtable)
116 {
117     unsigned int i;
118     for (i = 0; i < strtable->cEntries; i++)
119     {
120         HeapFree(GetProcessHeap(), 0, strtable->pse[i].pszName);
121         HeapFree(GetProcessHeap(), 0, strtable->pse[i].pszValue);
122         if (!strtable->pse[i].pszName || !strtable->pse[i].pszValue)
123             return;
124     }
125 }
126
127 static HRESULT register_mstask(BOOL do_register)
128 {
129     HRESULT hr;
130     STRTABLEA strtable;
131     HMODULE hAdvpack;
132     HRESULT (WINAPI *pRegInstall)(HMODULE hm,
133             LPCSTR pszSection, const STRTABLEA* pstTable);
134     static const WCHAR wszAdvpack[] =
135             {'a','d','v','p','a','c','k','.','d','l','l',0};
136
137     TRACE("(%x)\n", do_register);
138
139     hAdvpack = LoadLibraryW(wszAdvpack);
140     pRegInstall = (void *)GetProcAddress(hAdvpack, "RegInstall");
141
142     hr = init_register_strtable(&strtable);
143     if (SUCCEEDED(hr))
144         hr = pRegInstall(hInst, do_register ? "RegisterDll" : "UnregisterDll",
145                 &strtable);
146     cleanup_register_strtable(&strtable);
147
148     if (FAILED(hr))
149         WINE_ERR("RegInstall failed: %08x\n", hr);
150
151     return hr;
152 }
153
154 HRESULT WINAPI DllRegisterServer(void)
155 {
156     return register_mstask(TRUE);
157 }
158
159 HRESULT WINAPI DllUnregisterServer(void)
160 {
161     return register_mstask(FALSE);
162 }