Roy Shea : qmgr: Add ServiceMain.
Alexandre Julliard
julliard at winehq.org
Tue Feb 19 13:59:13 CST 2008
Module: wine
Branch: master
Commit: 5c51ad2c3f78cf801b7458a810c4490f85421ed1
URL: http://source.winehq.org/git/wine.git/?a=commit;h=5c51ad2c3f78cf801b7458a810c4490f85421ed1
Author: Roy Shea <roy at cs.hmc.edu>
Date: Mon Feb 11 11:09:34 2008 -0800
qmgr: Add ServiceMain.
---
dlls/qmgr/Makefile.in | 5 +-
dlls/qmgr/qmgr.spec | 1 +
dlls/qmgr/service.c | 138 +++++++++++++++++++++++++++++++++++++++++++++++++
3 files changed, 142 insertions(+), 2 deletions(-)
diff --git a/dlls/qmgr/Makefile.in b/dlls/qmgr/Makefile.in
index 65812c3..a6c40ac 100644
--- a/dlls/qmgr/Makefile.in
+++ b/dlls/qmgr/Makefile.in
@@ -3,13 +3,14 @@ TOPOBJDIR = ../..
SRCDIR = @srcdir@
VPATH = @srcdir@
MODULE = qmgr.dll
-IMPORTS = advpack advapi32 kernel32
+IMPORTS = advpack ole32 advapi32 kernel32
EXTRALIBS = -luuid
C_SRCS = \
factory.c \
qmgr.c \
- qmgr_main.c
+ qmgr_main.c \
+ service.c
RC_SRCS = rsrc.rc
diff --git a/dlls/qmgr/qmgr.spec b/dlls/qmgr/qmgr.spec
index 0f9b073..2431c63 100644
--- a/dlls/qmgr/qmgr.spec
+++ b/dlls/qmgr/qmgr.spec
@@ -1,2 +1,3 @@
@ stdcall -private DllRegisterServer()
@ stdcall -private DllUnregisterServer()
+@ stdcall -private ServiceMain(long ptr)
diff --git a/dlls/qmgr/service.c b/dlls/qmgr/service.c
new file mode 100644
index 0000000..5e31b8f
--- /dev/null
+++ b/dlls/qmgr/service.c
@@ -0,0 +1,138 @@
+/*
+ * ServiceMain function for qmgr running within svchost
+ *
+ * Copyright 2007 (C) 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
+ */
+
+#include "windef.h"
+#include "objbase.h"
+#include "winsvc.h"
+#include "bits.h"
+
+#include "qmgr.h"
+#include "wine/debug.h"
+
+WINE_DEFAULT_DEBUG_CHANNEL(qmgr);
+
+static WCHAR qmgr_nameW[] = {'B','I','T','S',0};
+static SERVICE_STATUS_HANDLE status_handle;
+static SERVICE_STATUS status;
+static HANDLE stop_event = NULL;
+
+static VOID
+UpdateStatus(DWORD dwCurrentState, DWORD dwWin32ExitCode, DWORD dwWaitHint)
+{
+ status.dwServiceType = SERVICE_WIN32_OWN_PROCESS;
+ status.dwCurrentState = dwCurrentState;
+ if (dwCurrentState == SERVICE_START_PENDING)
+ status.dwControlsAccepted = 0;
+ else
+ status.dwControlsAccepted
+ = (SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_PAUSE_CONTINUE
+ | SERVICE_ACCEPT_SHUTDOWN);
+ status.dwWin32ExitCode = 0;
+ status.dwServiceSpecificExitCode = 0;
+ status.dwCheckPoint = 0;
+ status.dwWaitHint = dwWaitHint;
+
+ if (!SetServiceStatus(status_handle, &status)) {
+ ERR("failed to set service status\n");
+ SetEvent(stop_event);
+ }
+}
+
+/* Handle incoming ControlService signals */
+static DWORD WINAPI
+ServiceHandler(DWORD ctrl, DWORD event_type, LPVOID event_data, LPVOID context)
+{
+ switch (ctrl) {
+ case SERVICE_CONTROL_STOP:
+ case SERVICE_CONTROL_SHUTDOWN:
+ TRACE("shutting down service\n");
+ UpdateStatus(SERVICE_STOP_PENDING, NO_ERROR, 0);
+ SetEvent(stop_event);
+ break;
+ default:
+ FIXME("ignoring handle service ctrl %x\n", ctrl);
+ UpdateStatus(status.dwCurrentState, NO_ERROR, 0);
+ break;
+ }
+
+ return NO_ERROR;
+}
+
+/* Main thread of the service */
+static BOOL
+StartCount(void)
+{
+ HRESULT hr;
+ DWORD dwReg;
+
+ TRACE("\n");
+
+ hr = CoInitializeEx(NULL, COINIT_MULTITHREADED);
+ if (!SUCCEEDED(hr))
+ return FALSE;
+
+ hr = CoInitializeSecurity(NULL, -1, NULL, NULL, RPC_C_AUTHN_LEVEL_NONE,
+ RPC_C_IMP_LEVEL_IMPERSONATE, NULL, EOAC_NONE,
+ NULL);
+ if (!SUCCEEDED(hr))
+ return FALSE;
+
+ hr = CoRegisterClassObject(&CLSID_BackgroundCopyManager,
+ (IUnknown *) &BITS_ClassFactory,
+ CLSCTX_LOCAL_SERVER, REGCLS_MULTIPLEUSE,
+ &dwReg);
+ if (!SUCCEEDED(hr))
+ return FALSE;
+
+ return TRUE;
+}
+
+/* Service entry point */
+VOID WINAPI
+ServiceMain(DWORD dwArgc, LPWSTR *lpszArgv)
+{
+ TRACE("\n");
+
+ stop_event = CreateEventW(NULL, TRUE, FALSE, NULL);
+ if (!stop_event) {
+ ERR("failed to create stop_event\n");
+ return;
+ }
+
+ status_handle = RegisterServiceCtrlHandlerExW(qmgr_nameW, ServiceHandler, NULL);
+ if (!status_handle) {
+ ERR("failed to register handler: %u\n", GetLastError());
+ return;
+ }
+
+ UpdateStatus(SERVICE_START_PENDING, NO_ERROR, 3000);
+ if (!StartCount()) {
+ ERR("failed starting service thread\n");
+ UpdateStatus(SERVICE_STOPPED, NO_ERROR, 0);
+ return;
+ }
+
+ UpdateStatus(SERVICE_RUNNING, NO_ERROR, 0);
+
+ WaitForSingleObject(stop_event, INFINITE);
+ UpdateStatus(SERVICE_STOPPED, NO_ERROR, 0);
+ CloseHandle(stop_event);
+ TRACE("service stoped\n");
+}
More information about the wine-cvs
mailing list