mshtml: Added an explicit CDECL attribute to imported functions.
[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,
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 }