[QMGR] Sync with Wine Staging 3.3. CORE-14434
[reactos.git] / dll / win32 / qmgr / service.c
1 /*
2 * ServiceMain function for qmgr running within svchost
3 *
4 * Copyright 2007 (C) Google (Roy Shea)
5 *
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
10 *
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
15 *
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
19 */
20
21 #include "windef.h"
22 #include "winsvc.h"
23 #include "qmgr.h"
24 #include "wine/debug.h"
25
26 WINE_DEFAULT_DEBUG_CHANNEL(qmgr);
27
28 HANDLE stop_event = NULL;
29
30 static SERVICE_STATUS_HANDLE status_handle;
31 static SERVICE_STATUS status;
32
33 static VOID
34 UpdateStatus(DWORD dwCurrentState, DWORD dwWin32ExitCode, DWORD dwWaitHint)
35 {
36 status.dwServiceType = SERVICE_WIN32_OWN_PROCESS;
37 status.dwCurrentState = dwCurrentState;
38 if (dwCurrentState == SERVICE_START_PENDING)
39 status.dwControlsAccepted = 0;
40 else
41 status.dwControlsAccepted
42 = (SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_PAUSE_CONTINUE
43 | SERVICE_ACCEPT_SHUTDOWN);
44 status.dwWin32ExitCode = 0;
45 status.dwServiceSpecificExitCode = 0;
46 status.dwCheckPoint = 0;
47 status.dwWaitHint = dwWaitHint;
48
49 if (!SetServiceStatus(status_handle, &status)) {
50 ERR("failed to set service status\n");
51 SetEvent(stop_event);
52 }
53 }
54
55 /* Handle incoming ControlService signals */
56 static DWORD WINAPI
57 ServiceHandler(DWORD ctrl, DWORD event_type, LPVOID event_data, LPVOID context)
58 {
59 switch (ctrl) {
60 case SERVICE_CONTROL_STOP:
61 case SERVICE_CONTROL_SHUTDOWN:
62 TRACE("shutting down service\n");
63 UpdateStatus(SERVICE_STOP_PENDING, NO_ERROR, 0);
64 SetEvent(stop_event);
65 break;
66 default:
67 FIXME("ignoring handle service ctrl %x\n", ctrl);
68 UpdateStatus(status.dwCurrentState, NO_ERROR, 0);
69 break;
70 }
71
72 return NO_ERROR;
73 }
74
75 /* Main thread of the service */
76 static BOOL
77 StartCount(void)
78 {
79 HRESULT hr;
80 DWORD dwReg;
81
82 TRACE("\n");
83
84 hr = CoInitializeEx(NULL, COINIT_MULTITHREADED);
85 if (FAILED(hr))
86 return FALSE;
87
88 hr = CoInitializeSecurity(NULL, -1, NULL, NULL, RPC_C_AUTHN_LEVEL_NONE,
89 RPC_C_IMP_LEVEL_IMPERSONATE, NULL, EOAC_NONE,
90 NULL);
91 if (FAILED(hr))
92 return FALSE;
93
94 hr = CoRegisterClassObject(&CLSID_BackgroundCopyManager,
95 (IUnknown *) &BITS_ClassFactory.IClassFactory_iface,
96 CLSCTX_LOCAL_SERVER, REGCLS_MULTIPLEUSE, &dwReg);
97 if (FAILED(hr))
98 return FALSE;
99
100 return TRUE;
101 }
102
103 /* Service entry point */
104 VOID WINAPI
105 ServiceMain(DWORD dwArgc, LPWSTR *lpszArgv)
106 {
107 HANDLE fileTxThread;
108 static const WCHAR qmgr_nameW[] = {'B','I','T','S',0};
109 DWORD threadId;
110 TRACE("\n");
111
112 stop_event = CreateEventW(NULL, TRUE, FALSE, NULL);
113 if (!stop_event) {
114 ERR("failed to create stop_event\n");
115 return;
116 }
117
118 status_handle = RegisterServiceCtrlHandlerExW(qmgr_nameW, ServiceHandler, NULL);
119 if (!status_handle) {
120 ERR("failed to register handler: %u\n", GetLastError());
121 return;
122 }
123
124 UpdateStatus(SERVICE_START_PENDING, NO_ERROR, 3000);
125 if (!StartCount()) {
126 ERR("failed starting service thread\n");
127 UpdateStatus(SERVICE_STOPPED, NO_ERROR, 0);
128 return;
129 }
130
131 globalMgr.jobEvent = CreateEventW(NULL, TRUE, FALSE, NULL);
132 if (!globalMgr.jobEvent) {
133 ERR("Couldn't create event: error %d\n", GetLastError());
134 UpdateStatus(SERVICE_STOPPED, NO_ERROR, 0);
135 return;
136 }
137
138 fileTxThread = CreateThread(NULL, 0, fileTransfer, NULL, 0, &threadId);
139 if (!fileTxThread)
140 {
141 ERR("Failed starting file transfer thread\n");
142 UpdateStatus(SERVICE_STOPPED, NO_ERROR, 0);
143 return;
144 }
145
146 UpdateStatus(SERVICE_RUNNING, NO_ERROR, 0);
147
148 WaitForSingleObject(fileTxThread, INFINITE);
149 UpdateStatus(SERVICE_STOPPED, NO_ERROR, 0);
150 CloseHandle(stop_event);
151 TRACE("service stopped\n");
152
153 CoUninitialize();
154 }