qmgr [2/3]: Implement DLL server registration.
Dan Hipschman
dsh at linux.ucla.edu
Fri Jan 18 13:06:25 CST 2008
I'm just guessing on the dates here since Roy never sent these. The tarred
up directory is named 12_21_bits_patch_set, so I'll just put them all down
as Dec 21.
ChangeLog:
Implement DLL server registration for qmgr.
From: Roy Shea <roy at cs.hmc.edu>
Date: Thu Dec 21 18:38:31 CST 2007
---
dlls/qmgr/Makefile.in | 4 ++-
dlls/qmgr/qmgr.h | 11 +++++++
dlls/qmgr/qmgr.inf | 31 +++++++++++++++++++
dlls/qmgr/qmgr.spec | 4 +-
dlls/qmgr/qmgr_main.c | 81 +++++++++++++++++++++++++++++++++++++++++++++++--
dlls/qmgr/rsrc.rc | 20 ++++++++++++
6 files changed, 145 insertions(+), 6 deletions(-)
create mode 100644 dlls/qmgr/qmgr.inf
create mode 100644 dlls/qmgr/rsrc.rc
diff --git a/dlls/qmgr/Makefile.in b/dlls/qmgr/Makefile.in
index a19eea6..7159607 100644
--- a/dlls/qmgr/Makefile.in
+++ b/dlls/qmgr/Makefile.in
@@ -3,7 +3,7 @@ TOPOBJDIR = ../..
SRCDIR = @srcdir@
VPATH = @srcdir@
MODULE = qmgr.dll
-IMPORTS = kernel32
+IMPORTS = advpack kernel32
EXTRALIBS = -luuid
C_SRCS = \
@@ -11,6 +11,8 @@ C_SRCS = \
qmgr.c \
qmgr_main.c
+RC_SRCS = rsrc.rc
+
IDL_I_SRCS = qmgr_local.idl
@MAKE_DLL_RULES@
diff --git a/dlls/qmgr/qmgr.h b/dlls/qmgr/qmgr.h
index 9b708c8..3ad314d 100644
--- a/dlls/qmgr/qmgr.h
+++ b/dlls/qmgr/qmgr.h
@@ -27,6 +27,8 @@
#define COBJMACROS
#include "bits.h"
+#include <string.h>
+
/* Background copy manager vtbl and related data */
typedef struct
{
@@ -43,4 +45,13 @@ extern ClassFactoryImpl BITS_ClassFactory;
HRESULT BackgroundCopyManagerConstructor(IUnknown *pUnkOuter, LPVOID *ppObj);
+/* Little helper functions */
+static inline char *
+qmgr_strdup(const char *s)
+{
+ size_t n = strlen(s) + 1;
+ char *d = HeapAlloc(GetProcessHeap(), 0, n);
+ return d ? memcpy(d, s, n) : NULL;
+}
+
#endif /* __QMGR_H__ */
diff --git a/dlls/qmgr/qmgr.inf b/dlls/qmgr/qmgr.inf
new file mode 100644
index 0000000..0550f01
--- /dev/null
+++ b/dlls/qmgr/qmgr.inf
@@ -0,0 +1,31 @@
+[version]
+Signature="$CHICAGO$"
+
+
+[Strings]
+Services="System\CurrentControlSet\Services"
+
+
+[RegisterDll]
+AddReg = Qmgr.Reg
+
+
+[UnregisterDll]
+DelReg = Qmgr.Reg
+
+
+[Qmgr.Reg]
+
+HKCR,"AppID\BITS","AppID",,"%CLSID_BackgroundCopyQMgr%"
+HKCR,"AppID\%CLSID_BackgroundCopyQMgr%","LocalService",,"BITS"
+HKCR,"CLSID\%CLSID_BackgroundCopyManager%","AppID",,"%CLSID_BackgroundCopyQMgr%"
+
+HKLM,"Software\Microsoft\Windows NT\CurrentVersion\SvcHost","netsvcs",0x00010000,"BITS"
+
+HKLM,"%Services%\BITS","Type",0x00010001,00000110
+HKLM,"%Services%\BITS","Start",0x00010001,00000003
+HKLM,"%Services%\BITS","ErrorControl",0x00010001,00000001
+HKLM,"%Services%\BITS","Description",,"BITS Service"
+HKLM,"%Services%\BITS","DispalyName",,"BITS"
+HKLM,"%Services%\BITS","ImagePath",0x00020000,"svchost.exe -k netsvcs"
+HKLM,"%Services%\BITS\Parameters","ServiceDll",0x00020000,"qmgr.dll"
diff --git a/dlls/qmgr/qmgr.spec b/dlls/qmgr/qmgr.spec
index 1a63b32..0f9b073 100644
--- a/dlls/qmgr/qmgr.spec
+++ b/dlls/qmgr/qmgr.spec
@@ -1,2 +1,2 @@
-@ stub DllRegisterServer
-@ stub DllUnregisterServer
+@ stdcall -private DllRegisterServer()
+@ stdcall -private DllUnregisterServer()
diff --git a/dlls/qmgr/qmgr_main.c b/dlls/qmgr/qmgr_main.c
index d00ac2d..2b93bf3 100644
--- a/dlls/qmgr/qmgr_main.c
+++ b/dlls/qmgr/qmgr_main.c
@@ -21,15 +21,27 @@
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
*/
-#include <stdarg.h>
+#include <stdio.h>
-#include "windef.h"
-#include "winbase.h"
+#include "objbase.h"
+#include "winuser.h"
+#include "winreg.h"
+#include "advpub.h"
+
+#include "bits.h"
+#include "qmgr.h"
+#include "initguid.h"
#include "wine/debug.h"
WINE_DEFAULT_DEBUG_CHANNEL(qmgr);
+/* Handle to the base address of this DLL */
+static HINSTANCE hInst;
+
+/* Other GUIDs used by this module */
+DEFINE_GUID(CLSID_BackgroundCopyQMgr, 0x69AD4AEE, 0x51BE, 0x439b, 0xA9,0x2C, 0x86,0xAE,0x49,0x0E,0x8B,0x30);
+
/* Entry point for DLL */
BOOL WINAPI DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved)
{
@@ -41,6 +53,7 @@ BOOL WINAPI DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved)
return FALSE; /* prefer native version */
case DLL_PROCESS_ATTACH:
DisableThreadLibraryCalls(hinstDLL);
+ hInst = hinstDLL;
break;
case DLL_PROCESS_DETACH:
break;
@@ -48,3 +61,65 @@ BOOL WINAPI DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved)
return TRUE;
}
+
+/* Use an INF file to register or unregister the DLL */
+static HRESULT register_server(BOOL do_register)
+{
+ #define CLSID_EXPANSION_ENTRY(id) { "CLSID_" #id, &CLSID_ ## id }
+ static const struct {
+ const char *name;
+ const CLSID *clsid;
+ } expns[] = {
+ CLSID_EXPANSION_ENTRY(BackgroundCopyQMgr),
+ CLSID_EXPANSION_ENTRY(BackgroundCopyManager)
+ };
+ #undef CLSID_EXPANSION_ENTRY
+ HRESULT hr = S_OK;
+ STRTABLEA strtable;
+ STRENTRYA pse[sizeof expns / sizeof expns[0]];
+ int i;
+
+ TRACE("(%x)\n", do_register);
+
+ 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 = qmgr_strdup(expns[i].name);
+ pse[i].pszValue = HeapAlloc(GetProcessHeap(), 0, sizeof dummy_sample);
+ if (!pse[i].pszName || !pse[i].pszValue)
+ goto finished;
+ 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]);
+ }
+
+ hr = RegInstallA(hInst, do_register ? "RegisterDll" : "UnregisterDll", &strtable);
+
+ finished:
+ for (i = 0; i < sizeof pse / sizeof pse[0]; i++) {
+ HeapFree(GetProcessHeap(), 0, pse[i].pszName);
+ HeapFree(GetProcessHeap(), 0, pse[i].pszValue);
+ if (!pse[i].pszName || !pse[i].pszValue) {
+ hr = E_OUTOFMEMORY;
+ break;
+ }
+ }
+
+ if (FAILED(hr))
+ ERR("RegInstall failed: %08x\n", hr);
+
+ return hr;
+}
+
+HRESULT WINAPI DllRegisterServer()
+{
+ return register_server(TRUE);
+}
+
+HRESULT WINAPI DllUnregisterServer()
+{
+ return register_server(FALSE);
+}
diff --git a/dlls/qmgr/rsrc.rc b/dlls/qmgr/rsrc.rc
new file mode 100644
index 0000000..dfc64d0
--- /dev/null
+++ b/dlls/qmgr/rsrc.rc
@@ -0,0 +1,20 @@
+/*
+ * Copyright 2007 Google (Roy Shea)
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2.1 of the License, or (at your option) any later version.
+ *
+ * This library is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
+ */
+
+/* @makedep: qmgr.inf */
+REGINST REGINST qmgr.inf
More information about the wine-patches
mailing list