d3d9: Pass count=0 when clearing the whole surface.
[wine] / dlls / qmgr / qmgr_main.c
CommitLineData
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 39WINE_DEFAULT_DEBUG_CHANNEL(qmgr);
2b0255f6 40
313a903a
RS
41/* Handle to the base address of this DLL */
42static HINSTANCE hInst;
43
44/* Other GUIDs used by this module */
45DEFINE_GUID(CLSID_BackgroundCopyQMgr, 0x69AD4AEE, 0x51BE, 0x439b, 0xA9,0x2C, 0x86,0xAE,0x49,0x0E,0x8B,0x30);
46
b655da64 47/* Entry point for DLL */
2b0255f6
RS
48BOOL 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
67static 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
99static 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
110static 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 */
140static 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
171HRESULT WINAPI DllRegisterServer(void)
172{
173 return register_server(TRUE);
174}
175
176HRESULT WINAPI DllUnregisterServer(void)
177{
178 return register_server(FALSE);
179}