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.h"
  25
  26 #include "wine/debug.h"
  27
  28 WINE_DEFAULT_DEBUG_CHANNEL(wia);
  29
  30 static HANDLE stop_event = NULL;
  31
  32 static SERVICE_STATUS_HANDLE status_handle;
  33 static SERVICE_STATUS status;
  34 static DWORD dwReg;
  35
  36 static VOID
  37 UpdateStatus(DWORD dwCurrentState, DWORD dwWin32ExitCode, DWORD dwWaitHint)
  38 {
  39     status.dwServiceType = SERVICE_WIN32_OWN_PROCESS;
  40     status.dwCurrentState = dwCurrentState;
  41     if (dwCurrentState == SERVICE_START_PENDING)
  42         status.dwControlsAccepted = 0;
  43     else
  44         status.dwControlsAccepted
  45             = (SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_PAUSE_CONTINUE
  46                | SERVICE_ACCEPT_SHUTDOWN);
  47     status.dwWin32ExitCode = 0;
  48     status.dwServiceSpecificExitCode = 0;
  49     status.dwCheckPoint = 0;
  50     status.dwWaitHint = dwWaitHint;
  51
  52     if (!SetServiceStatus(status_handle, &status)) {
  53         ERR("failed to set service status\n");
  54         SetEvent(stop_event);
  55     }
  56 }
  57
  58 /* Handle incoming ControlService signals */
  59 static DWORD WINAPI
  60 ServiceHandler(DWORD ctrl, DWORD event_type, LPVOID event_data, LPVOID context)
  61 {
  62     switch (ctrl) {
  63     case SERVICE_CONTROL_STOP:
  64     case SERVICE_CONTROL_SHUTDOWN:
  65         TRACE("shutting down service\n");
  66         UpdateStatus(SERVICE_STOP_PENDING, NO_ERROR, 0);
  67         SetEvent(stop_event);
  68         break;
  69     default:
  70         FIXME("ignoring handle service ctrl %x\n", ctrl);
  71         UpdateStatus(status.dwCurrentState, NO_ERROR, 0);
  72         break;
  73     }
  74
  75     return NO_ERROR;
  76 }
  77
  78 static BOOL
  79 StartCount(void)
  80 {
  81     HRESULT hr;
  82
  83     TRACE("\n");
  84
  85     hr = CoInitializeEx(NULL, COINIT_MULTITHREADED);
  86     if (FAILED(hr))
  87         return FALSE;
  88
  89     hr = CoInitializeSecurity(NULL, -1, NULL, NULL, RPC_C_AUTHN_LEVEL_NONE,
  90                               RPC_C_IMP_LEVEL_IMPERSONATE, NULL, EOAC_NONE,
  91                               NULL);
  92     if (FAILED(hr))
  93         return FALSE;
  94
  95 /* FIXME: implement
  96     hr = CoRegisterClassObject(&CLSID_WiaDevMgr,
  97                                (IUnknown *) &WIASERVC_ClassFactory,
  98                                CLSCTX_LOCAL_SERVER, REGCLS_MULTIPLEUSE,
  99                                &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 }