Commit | Line | Data |
---|---|---|
2b0255f6 | 1 | /* |
b655da64 | 2 | * Main DLL interface to Queue Manager (BITS) |
2b0255f6 RS |
3 | * |
4 | * Background Intelligent Transfer Service (BITS) interface. Dll is named | |
5 | * qmgr for backwards compatibility with early versions of BITS. | |
6 | * | |
7 | * Copyright 2007 Google (Roy Shea) | |
8 | * | |
9 | * This library is free software; you can redistribute it and/or | |
10 | * modify it under the terms of the GNU Lesser General Public | |
11 | * License as published by the Free Software Foundation; either | |
12 | * version 2.1 of the License, or (at your option) any later version. | |
13 | * | |
14 | * This library is distributed in the hope that it will be useful, | |
15 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
16 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
17 | * Lesser General Public License for more details. | |
18 | * | |
19 | * You should have received a copy of the GNU Lesser General Public | |
20 | * License along with this library; if not, write to the Free Software | |
21 | * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA | |
22 | */ | |
23 | ||
313a903a | 24 | #include <stdio.h> |
2b0255f6 | 25 | |
313a903a RS |
26 | #include "objbase.h" |
27 | #include "winuser.h" | |
28 | #include "winreg.h" | |
29 | #include "advpub.h" | |
30 | #include "olectl.h" | |
31 | #include "winsvc.h" | |
32 | ||
33 | #include "bits.h" | |
34 | #include "qmgr.h" | |
35 | #include "initguid.h" | |
b655da64 | 36 | |
2b0255f6 | 37 | #include "wine/debug.h" |
2b0255f6 | 38 | |
b655da64 | 39 | WINE_DEFAULT_DEBUG_CHANNEL(qmgr); |
2b0255f6 | 40 | |
313a903a RS |
41 | /* Handle to the base address of this DLL */ |
42 | static HINSTANCE hInst; | |
43 | ||
44 | /* Other GUIDs used by this module */ | |
45 | DEFINE_GUID(CLSID_BackgroundCopyQMgr, 0x69AD4AEE, 0x51BE, 0x439b, 0xA9,0x2C, 0x86,0xAE,0x49,0x0E,0x8B,0x30); | |
46 | ||
b655da64 | 47 | /* Entry point for DLL */ |
2b0255f6 RS |
48 | BOOL WINAPI DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved) |
49 | { | |
50 | TRACE("(%p, %d, %p)\n", hinstDLL, fdwReason, lpvReserved); | |
51 | ||
52 | switch (fdwReason) | |
53 | { | |
b655da64 RS |
54 | case DLL_WINE_PREATTACH: |
55 | return FALSE; /* prefer native version */ | |
56 | case DLL_PROCESS_ATTACH: | |
57 | DisableThreadLibraryCalls(hinstDLL); | |
313a903a | 58 | hInst = hinstDLL; |
b655da64 RS |
59 | break; |
60 | case DLL_PROCESS_DETACH: | |
61 | break; | |
2b0255f6 RS |
62 | } |
63 | ||
64 | return TRUE; | |
65 | } | |
313a903a RS |
66 | |
67 | static HRESULT init_register_strtable(STRTABLEA *strtable) | |
68 | { | |
69 | #define CLSID_EXPANSION_ENTRY(id) { "CLSID_" #id, &CLSID_ ## id } | |
70 | static const struct { | |
71 | const char *name; | |
72 | const CLSID *clsid; | |
73 | } expns[] = { | |
74 | CLSID_EXPANSION_ENTRY(BackgroundCopyQMgr), | |
75 | CLSID_EXPANSION_ENTRY(BackgroundCopyManager) | |
76 | }; | |
77 | #undef CLSID_EXPANSION_ENTRY | |
78 | static STRENTRYA pse[sizeof expns / sizeof expns[0]]; | |
201cbd3e | 79 | DWORD i; |
313a903a RS |
80 | |
81 | strtable->cEntries = sizeof pse / sizeof pse[0]; | |
82 | strtable->pse = pse; | |
83 | for (i = 0; i < strtable->cEntries; i++) { | |
84 | static const char dummy_sample[] = "{12345678-1234-1234-1234-123456789012}"; | |
85 | const CLSID *clsid = expns[i].clsid; | |
86 | pse[i].pszName = qmgr_strdup(expns[i].name); | |
87 | pse[i].pszValue = HeapAlloc(GetProcessHeap(), 0, sizeof dummy_sample); | |
88 | if (!pse[i].pszName || !pse[i].pszValue) | |
89 | return E_OUTOFMEMORY; | |
90 | sprintf(pse[i].pszValue, "{%08X-%04X-%04X-%02X%02X-%02X%02X%02X%02X%02X%02X}", | |
91 | clsid->Data1, clsid->Data2, clsid->Data3, clsid->Data4[0], | |
92 | clsid->Data4[1], clsid->Data4[2], clsid->Data4[3], clsid->Data4[4], | |
93 | clsid->Data4[5], clsid->Data4[6], clsid->Data4[7]); | |
94 | } | |
95 | ||
96 | return S_OK; | |
97 | } | |
98 | ||
99 | static void cleanup_register_strtable(STRTABLEA *strtable) | |
100 | { | |
201cbd3e | 101 | DWORD i; |
313a903a RS |
102 | for (i = 0; i < strtable->cEntries; i++) { |
103 | HeapFree(GetProcessHeap(), 0, strtable->pse[i].pszName); | |
104 | HeapFree(GetProcessHeap(), 0, strtable->pse[i].pszValue); | |
105 | if (!strtable->pse[i].pszName || !strtable->pse[i].pszValue) | |
106 | return; | |
107 | } | |
108 | } | |
109 | ||
110 | static HRESULT register_service(BOOL do_register) | |
111 | { | |
112 | static const WCHAR name[] = { 'B','I','T','S', 0 }; | |
113 | static const WCHAR path[] = { 's','v','c','h','o','s','t','.','e','x','e', | |
114 | ' ','-','k',' ','n','e','t','s','v','c','s', 0 }; | |
115 | SC_HANDLE scm, service; | |
313a903a RS |
116 | |
117 | scm = OpenSCManagerW(NULL, NULL, SC_MANAGER_ALL_ACCESS); | |
118 | if (!scm) | |
119 | return SELFREG_E_CLASS; | |
120 | ||
121 | if (do_register) | |
122 | service = CreateServiceW(scm, name, name, SERVICE_ALL_ACCESS, | |
123 | SERVICE_WIN32_OWN_PROCESS, | |
124 | SERVICE_DEMAND_START, SERVICE_ERROR_NORMAL, | |
125 | path, NULL, NULL, NULL, NULL, NULL); | |
126 | else | |
127 | service = OpenServiceW(scm, name, DELETE); | |
128 | ||
129 | ||
130 | CloseServiceHandle(scm); | |
84cf9caa AJ |
131 | if (service) |
132 | { | |
133 | if (!do_register) DeleteService(service); | |
134 | CloseServiceHandle(service); | |
135 | } | |
136 | return S_OK; | |
313a903a RS |
137 | } |
138 | ||
139 | /* Use an INF file to register or unregister the DLL */ | |
140 | static HRESULT register_server(BOOL do_register) | |
141 | { | |
142 | HRESULT hr; | |
143 | STRTABLEA strtable; | |
b2603687 RS |
144 | HMODULE hAdvpack; |
145 | HRESULT (WINAPI *pRegInstall)(HMODULE hm, LPCSTR pszSection, const STRTABLEA* pstTable); | |
146 | static const WCHAR wszAdvpack[] = {'a','d','v','p','a','c','k','.','d','l','l',0}; | |
313a903a RS |
147 | |
148 | TRACE("(%x)\n", do_register); | |
149 | ||
150 | hr = register_service(do_register); | |
151 | if (FAILED(hr)) { | |
152 | ERR("register_service failed: %d\n", GetLastError()); | |
153 | return hr; | |
154 | } | |
155 | ||
b2603687 RS |
156 | hAdvpack = LoadLibraryW(wszAdvpack); |
157 | pRegInstall = (void *)GetProcAddress(hAdvpack, "RegInstall"); | |
158 | ||
313a903a RS |
159 | hr = init_register_strtable(&strtable); |
160 | if (SUCCEEDED(hr)) | |
b2603687 | 161 | hr = pRegInstall(hInst, do_register ? "RegisterDll" : "UnregisterDll", |
313a903a RS |
162 | &strtable); |
163 | cleanup_register_strtable(&strtable); | |
164 | ||
165 | if (FAILED(hr)) | |
166 | ERR("RegInstall failed: %08x\n", hr); | |
167 | ||
168 | return hr; | |
169 | } | |
170 | ||
171 | HRESULT WINAPI DllRegisterServer(void) | |
172 | { | |
173 | return register_server(TRUE); | |
174 | } | |
175 | ||
176 | HRESULT WINAPI DllUnregisterServer(void) | |
177 | { | |
178 | return register_server(FALSE); | |
179 | } |