user32: Delete the static critical section when unloading the dll.
[wine] / dlls / wiaservc / service.c
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.IClassFactory_iface,
99                                CLSCTX_LOCAL_SERVER, REGCLS_MULTIPLEUSE, &dwReg);
100     if (FAILED(hr))
101         return FALSE;
102
103     return TRUE;
104 }
105
106 /* Service entry point */
107 VOID WINAPI
108 ServiceMain(DWORD dwArgc, LPWSTR *lpszArgv)
109 {
110     static const WCHAR stisvc_nameW[] = {'s','t','i','s','v','c',0};
111     TRACE("(%d, %p)\n", dwArgc, lpszArgv);
112
113     stop_event = CreateEventW(NULL, TRUE, FALSE, NULL);
114     if (!stop_event) {
115         ERR("failed to create stop_event\n");
116         return;
117     }
118
119     status_handle = RegisterServiceCtrlHandlerExW(stisvc_nameW, ServiceHandler, NULL);
120     if (!status_handle) {
121         ERR("failed to register handler: %u\n", GetLastError());
122         return;
123     }
124
125     UpdateStatus(SERVICE_START_PENDING, NO_ERROR, 3000);
126     if (!StartCount()) {
127         ERR("failed starting service thread\n");
128         UpdateStatus(SERVICE_STOPPED, NO_ERROR, 0);
129         return;
130     }
131
132     UpdateStatus(SERVICE_RUNNING, NO_ERROR, 0);
133
134     WaitForSingleObject(stop_event, INFINITE);
135
136     CoRevokeClassObject(dwReg);
137     UpdateStatus(SERVICE_STOPPED, NO_ERROR, 0);
138     CloseHandle(stop_event);
139     TRACE("service stopped\n");
140
141     CoUninitialize();
142 }