2 * ServiceMain function for qmgr running within svchost
4 * Copyright 2007 (C) Google (Roy Shea)
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.
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.
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
27 #include "wine/debug.h"
29 WINE_DEFAULT_DEBUG_CHANNEL(qmgr
);
31 HANDLE stop_event
= NULL
;
33 static WCHAR qmgr_nameW
[] = {'B','I','T','S',0};
34 static SERVICE_STATUS_HANDLE status_handle
;
35 static SERVICE_STATUS status
;
38 UpdateStatus(DWORD dwCurrentState
, DWORD dwWin32ExitCode
, DWORD dwWaitHint
)
40 status
.dwServiceType
= SERVICE_WIN32_OWN_PROCESS
;
41 status
.dwCurrentState
= dwCurrentState
;
42 if (dwCurrentState
== SERVICE_START_PENDING
)
43 status
.dwControlsAccepted
= 0;
45 status
.dwControlsAccepted
46 = (SERVICE_ACCEPT_STOP
| SERVICE_ACCEPT_PAUSE_CONTINUE
47 | SERVICE_ACCEPT_SHUTDOWN
);
48 status
.dwWin32ExitCode
= 0;
49 status
.dwServiceSpecificExitCode
= 0;
50 status
.dwCheckPoint
= 0;
51 status
.dwWaitHint
= dwWaitHint
;
53 if (!SetServiceStatus(status_handle
, &status
)) {
54 ERR("failed to set service status\n");
59 /* Handle incoming ControlService signals */
61 ServiceHandler(DWORD ctrl
, DWORD event_type
, LPVOID event_data
, LPVOID context
)
64 case SERVICE_CONTROL_STOP
:
65 case SERVICE_CONTROL_SHUTDOWN
:
66 TRACE("shutting down service\n");
67 UpdateStatus(SERVICE_STOP_PENDING
, NO_ERROR
, 0);
71 FIXME("ignoring handle service ctrl %x\n", ctrl
);
72 UpdateStatus(status
.dwCurrentState
, NO_ERROR
, 0);
79 /* Main thread of the service */
88 hr
= CoInitializeEx(NULL
, COINIT_MULTITHREADED
);
92 hr
= CoInitializeSecurity(NULL
, -1, NULL
, NULL
, RPC_C_AUTHN_LEVEL_NONE
,
93 RPC_C_IMP_LEVEL_IMPERSONATE
, NULL
, EOAC_NONE
,
98 hr
= CoRegisterClassObject(&CLSID_BackgroundCopyManager
,
99 (IUnknown
*) &BITS_ClassFactory
,
100 CLSCTX_LOCAL_SERVER
, REGCLS_MULTIPLEUSE
,
108 /* Service entry point */
110 ServiceMain(DWORD dwArgc
, LPWSTR
*lpszArgv
)
116 stop_event
= CreateEventW(NULL
, TRUE
, FALSE
, NULL
);
118 ERR("failed to create stop_event\n");
122 status_handle
= RegisterServiceCtrlHandlerExW(qmgr_nameW
, ServiceHandler
, NULL
);
123 if (!status_handle
) {
124 ERR("failed to register handler: %u\n", GetLastError());
128 UpdateStatus(SERVICE_START_PENDING
, NO_ERROR
, 3000);
130 ERR("failed starting service thread\n");
131 UpdateStatus(SERVICE_STOPPED
, NO_ERROR
, 0);
135 globalMgr
.jobEvent
= CreateEventW(NULL
, TRUE
, FALSE
, NULL
);
136 if (!globalMgr
.jobEvent
) {
137 ERR("Couldn't create event: error %d\n", GetLastError());
138 UpdateStatus(SERVICE_STOPPED
, NO_ERROR
, 0);
142 fileTxThread
= CreateThread(NULL
, 0, fileTransfer
, NULL
, 0, &threadId
);
145 ERR("Failed starting file transfer thread\n");
146 UpdateStatus(SERVICE_STOPPED
, NO_ERROR
, 0);
150 UpdateStatus(SERVICE_RUNNING
, NO_ERROR
, 0);
152 WaitForSingleObject(fileTxThread
, INFINITE
);
153 UpdateStatus(SERVICE_STOPPED
, NO_ERROR
, 0);
154 CloseHandle(stop_event
);
155 TRACE("service stoped\n");