1 /*
2 * ServiceMain function for wiaservc running within svchost
3 *
4 * Copyright 2009 Damjan Jovanovic
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 "objbase.h"
23 #include "winsvc.h"
24 #include "wia_lh.h"
25
26 #include "wiaservc_private.h"
27
28 #include "wine/debug.h"
29
30 WINE_DEFAULT_DEBUG_CHANNEL(wia);
31
32 static HANDLE stop_event = NULL;
33
34 static SERVICE_STATUS_HANDLE status_handle;
35 static SERVICE_STATUS status;
36 static DWORD dwReg;
37
38 static VOID
39 UpdateStatus(DWORD dwCurrentState, DWORD dwWin32ExitCode, DWORD dwWaitHint)
40 {
41 status.dwServiceType = SERVICE_WIN32_OWN_PROCESS;
42 status.dwCurrentState = dwCurrentState;
43 if (dwCurrentState == SERVICE_START_PENDING)
44 status.dwControlsAccepted = 0;
45 else
46 status.dwControlsAccepted
47 = (SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_PAUSE_CONTINUE
48 | SERVICE_ACCEPT_SHUTDOWN);
49 status.dwWin32ExitCode = 0;
50 status.dwServiceSpecificExitCode = 0;
51 status.dwCheckPoint = 0;
52 status.dwWaitHint = dwWaitHint;
53
54 if (!SetServiceStatus(status_handle, &status)) {
55 ERR("failed to set service status\n");
56 SetEvent(stop_event);
57 }
58 }
59
60 /* Handle incoming ControlService signals */
61 static DWORD WINAPI
62 ServiceHandler(DWORD ctrl, DWORD event_type, LPVOID event_data, LPVOID context)
63 {
64 switch (ctrl) {
65 case SERVICE_CONTROL_STOP:
66 case SERVICE_CONTROL_SHUTDOWN:
67 TRACE("shutting down service\n");
68 UpdateStatus(SERVICE_STOP_PENDING, NO_ERROR, 0);
69 SetEvent(stop_event);
70 break;
71 default:
72 FIXME("ignoring handle service ctrl %x\n", ctrl);
73 UpdateStatus(status.dwCurrentState, NO_ERROR, 0);
74 break;
75 }
76
77 return NO_ERROR;
78 }
79
80 static BOOL
81 StartCount(void)
82 {
83 HRESULT hr;
84
85 TRACE("\n");
86
87 hr = CoInitializeEx(NULL, COINIT_MULTITHREADED);
88 if (FAILED(hr))
89 return FALSE;
90
91 hr = CoInitializeSecurity(NULL, -1, NULL, NULL, RPC_C_AUTHN_LEVEL_NONE,
92 RPC_C_IMP_LEVEL_IMPERSONATE, NULL, EOAC_NONE,
93 NULL);
94 if (FAILED(hr))
95 return FALSE;
96
97 hr = CoRegisterClassObject(&CLSID_WiaDevMgr,
98 (IUnknown *) &WIASERVC_ClassFactory,
99 CLSCTX_LOCAL_SERVER, REGCLS_MULTIPLEUSE,
100 &dwReg);
101 if (FAILED(hr))
102 return FALSE;
103
104 return TRUE;
105 }
106
107 /* Service entry point */
108 VOID WINAPI
109 ServiceMain(DWORD dwArgc, LPWSTR *lpszArgv)
110 {
111 static const WCHAR stisvc_nameW[] = {'s','t','i','s','v','c',0};
112 TRACE("(%d, %p)\n", dwArgc, lpszArgv);
113
114 stop_event = CreateEventW(NULL, TRUE, FALSE, NULL);
115 if (!stop_event) {
116 ERR("failed to create stop_event\n");
117 return;
118 }
119
120 status_handle = RegisterServiceCtrlHandlerExW(stisvc_nameW, ServiceHandler, NULL);
121 if (!status_handle) {
122 ERR("failed to register handler: %u\n", GetLastError());
123 return;
124 }
125
126 UpdateStatus(SERVICE_START_PENDING, NO_ERROR, 3000);
127 if (!StartCount()) {
128 ERR("failed starting service thread\n");
129 UpdateStatus(SERVICE_STOPPED, NO_ERROR, 0);
130 return;
131 }
132
133 UpdateStatus(SERVICE_RUNNING, NO_ERROR, 0);
134
135 WaitForSingleObject(stop_event, INFINITE);
136
137 CoRevokeClassObject(dwReg);
138 UpdateStatus(SERVICE_STOPPED, NO_ERROR, 0);
139 CloseHandle(stop_event);
140 TRACE("service stopped\n");
141
142 CoUninitialize();
143 }
144
This page was automatically generated by the
LXR engine.
Visit the LXR main site for more
information.