ReactOS  0.4.11-dev-946-g431643b
wmisvc.c
Go to the documentation of this file.
1 /*
2  * ReactOS Services
3  * Copyright (C) 2015 ReactOS Team
4  *
5  * This program is free software; you can redistribute it and/or modify
6  * it under the terms of the GNU General Public License as published by
7  * the Free Software Foundation; either version 2 of the License, or
8  * (at your option) any later version.
9  *
10  * This program is distributed in the hope that it will be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13  * GNU General Public License for more details.
14  *
15  * You should have received a copy of the GNU General Public License along
16  * with this program; if not, write to the Free Software Foundation, Inc.,
17  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
18  */
19 /*
20  * COPYRIGHT: See COPYING in the top level directory
21  * PROJECT: ReactOS Services
22  * FILE: base/services/wmisvc/wmisvc.c
23  * PURPOSE: WMI service
24  * PROGRAMMER: Pierre Schweitzer
25  */
26 
27 /* INCLUDES *****************************************************************/
28 
29 #define WIN32_NO_STATUS
30 #define _INC_WINDOWS
31 #define COM_NO_WINDOWS_H
32 #include <stdarg.h>
33 #include <windef.h>
34 #include <winbase.h>
35 #include <winreg.h>
36 #include <winsvc.h>
37 
38 #define NDEBUG
39 #include <debug.h>
40 
41 /* GLOBALS ******************************************************************/
42 
43 static WCHAR ServiceName[] = L"winmgmt";
44 
47 
48 /* FUNCTIONS *****************************************************************/
49 
50 static VOID
52 {
54  ServiceStatus.dwCurrentState = dwState;
55  ServiceStatus.dwControlsAccepted = 0;
56  ServiceStatus.dwWin32ExitCode = 0;
57  ServiceStatus.dwServiceSpecificExitCode = 0;
58  ServiceStatus.dwCheckPoint = 0;
59 
60  if (dwState == SERVICE_START_PENDING ||
61  dwState == SERVICE_STOP_PENDING ||
62  dwState == SERVICE_PAUSE_PENDING ||
63  dwState == SERVICE_CONTINUE_PENDING)
64  ServiceStatus.dwWaitHint = 10000;
65  else
66  ServiceStatus.dwWaitHint = 0;
67 
69  &ServiceStatus);
70 }
71 
72 static DWORD WINAPI
74  DWORD dwEventType,
75  LPVOID lpEventData,
76  LPVOID lpContext)
77 {
78  DPRINT1("ServiceControlHandler() called\n");
79 
80  switch (dwControl)
81  {
83  DPRINT1(" SERVICE_CONTROL_STOP received\n");
85  return ERROR_SUCCESS;
86 
88  DPRINT1(" SERVICE_CONTROL_PAUSE received\n");
90  return ERROR_SUCCESS;
91 
93  DPRINT1(" SERVICE_CONTROL_CONTINUE received\n");
95  return ERROR_SUCCESS;
96 
98  DPRINT1(" SERVICE_CONTROL_INTERROGATE received\n");
100  &ServiceStatus);
101  return ERROR_SUCCESS;
102 
104  DPRINT1(" SERVICE_CONTROL_SHUTDOWN received\n");
106  return ERROR_SUCCESS;
107 
108  default :
109  DPRINT1(" Control %lu received\n");
111  }
112 }
113 
114 VOID WINAPI
116 {
119 
120  DPRINT("ServiceMain() called\n");
121 
124  NULL);
125  if (!ServiceStatusHandle)
126  {
127  DPRINT1("RegisterServiceCtrlHandlerExW() failed! (Error %lu)\n", GetLastError());
128  return;
129  }
130 
132 
133  do
134  {
135  Sleep(1);
136  } while (1);
137 
139 }
140 
141 BOOL WINAPI
143  DWORD fdwReason,
145 {
146  switch (fdwReason)
147  {
148  case DLL_PROCESS_ATTACH:
149  DisableThreadLibraryCalls(hinstDLL);
150  break;
151 
152  case DLL_PROCESS_DETACH:
153  break;
154  }
155 
156  return TRUE;
157 }
static int argc
Definition: ServiceArgs.c:12
#define TRUE
Definition: types.h:120
VOID WINAPI DECLSPEC_HOTPATCH Sleep(IN DWORD dwMilliseconds)
Definition: synch.c:736
#define ERROR_SUCCESS
Definition: deptool.c:10
DWORD dwCurrentState
Definition: winsvc.h:100
#define SERVICE_PAUSE_PENDING
Definition: winsvc.h:26
#define UNREFERENCED_PARAMETER(P)
Definition: ntbasedef.h:323
__wchar_t WCHAR
Definition: xmlstorage.h:180
#define SERVICE_PAUSED
Definition: winsvc.h:27
static VOID UpdateServiceStatus(DWORD dwState)
Definition: wmisvc.c:51
BOOL WINAPI DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved)
Definition: wmisvc.c:142
BOOL WINAPI DisableThreadLibraryCalls(IN HMODULE hLibModule)
Definition: loader.c:85
DWORD WINAPI GetLastError(VOID)
Definition: except.c:1059
#define SERVICE_START_PENDING
Definition: winsvc.h:22
DWORD dwServiceSpecificExitCode
Definition: winsvc.h:103
#define DLL_PROCESS_ATTACH
Definition: compat.h:120
static char ** argv
Definition: ServiceArgs.c:11
CHAR * LPTSTR
Definition: xmlstorage.h:192
static IN DWORD IN LPVOID lpvReserved
BOOL WINAPI SetServiceStatus(SERVICE_STATUS_HANDLE hServiceStatus, LPSERVICE_STATUS lpServiceStatus)
Definition: sctrl.c:957
DWORD dwCheckPoint
Definition: winsvc.h:104
#define SERVICE_STOPPED
Definition: winsvc.h:21
#define SERVICE_RUNNING
Definition: winsvc.h:24
static SERVICE_STATUS_HANDLE ServiceStatusHandle
Definition: wmisvc.c:45
smooth NULL
Definition: ftsmooth.c:416
static DWORD WINAPI ServiceControlHandler(DWORD dwControl, DWORD dwEventType, LPVOID lpEventData, LPVOID lpContext)
Definition: wmisvc.c:73
void DPRINT(...)
Definition: polytest.cpp:61
#define SERVICE_CONTROL_INTERROGATE
Definition: winsvc.h:39
#define SERVICE_WIN32_OWN_PROCESS
Definition: cmtypes.h:960
#define DLL_PROCESS_DETACH
Definition: compat.h:119
static SERVICE_STATUS_HANDLE(WINAPI *pRegisterServiceCtrlHandlerExA)(LPCSTR
unsigned int BOOL
Definition: ntddk_ex.h:94
DWORD dwWaitHint
Definition: winsvc.h:105
DWORD dwWin32ExitCode
Definition: winsvc.h:102
unsigned long DWORD
Definition: ntddk_ex.h:95
VOID WINAPI ServiceMain(DWORD argc, LPTSTR *argv)
Definition: wmisvc.c:115
DWORD dwServiceType
Definition: winsvc.h:99
#define SERVICE_CONTROL_PAUSE
Definition: winsvc.h:37
static const WCHAR L[]
Definition: oid.c:1087
static SERVICE_STATUS ServiceStatus
Definition: wmisvc.c:46
SERVICE_STATUS_HANDLE WINAPI RegisterServiceCtrlHandlerExW(LPCWSTR lpServiceName, LPHANDLER_FUNCTION_EX lpHandlerProc, LPVOID lpContext)
Definition: sctrl.c:773
#define WINAPI
Definition: msvc.h:20
DWORD dwControlsAccepted
Definition: winsvc.h:101
#define SERVICE_CONTINUE_PENDING
Definition: winsvc.h:25
#define SERVICE_CONTROL_CONTINUE
Definition: winsvc.h:38
#define DPRINT1
Definition: precomp.h:8
static WCHAR ServiceName[]
Definition: wmisvc.c:43
#define ERROR_CALL_NOT_IMPLEMENTED
Definition: compat.h:92
#define SERVICE_CONTROL_STOP
Definition: winsvc.h:36
#define SERVICE_STOP_PENDING
Definition: winsvc.h:23
#define SERVICE_CONTROL_SHUTDOWN
Definition: winsvc.h:40