fa64969c6e1b9a800311228b5bdfc5d49e279b10
[reactos.git] / reactos / base / services / schedsvc / schedsvc.c
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/schedsvc/schedsvc.c
23 * PURPOSE: Scheduling service
24 * PROGRAMMER: Eric Kohl
25 */
26
27 /* INCLUDES *****************************************************************/
28
29 #include "precomp.h"
30
31 WINE_DEFAULT_DEBUG_CHANNEL(schedsvc);
32
33 /* GLOBALS ******************************************************************/
34
35 static WCHAR ServiceName[] = L"Schedule";
36
37 static SERVICE_STATUS_HANDLE ServiceStatusHandle;
38 static SERVICE_STATUS ServiceStatus;
39
40 /* FUNCTIONS *****************************************************************/
41
42 static VOID
43 UpdateServiceStatus(DWORD dwState)
44 {
45 ServiceStatus.dwServiceType = SERVICE_WIN32_OWN_PROCESS;
46 ServiceStatus.dwCurrentState = dwState;
47 ServiceStatus.dwControlsAccepted = 0;
48 ServiceStatus.dwWin32ExitCode = 0;
49 ServiceStatus.dwServiceSpecificExitCode = 0;
50 ServiceStatus.dwCheckPoint = 0;
51
52 if (dwState == SERVICE_START_PENDING ||
53 dwState == SERVICE_STOP_PENDING ||
54 dwState == SERVICE_PAUSE_PENDING ||
55 dwState == SERVICE_CONTINUE_PENDING)
56 ServiceStatus.dwWaitHint = 10000;
57 else
58 ServiceStatus.dwWaitHint = 0;
59
60 SetServiceStatus(ServiceStatusHandle,
61 &ServiceStatus);
62 }
63
64
65 static DWORD WINAPI
66 ServiceControlHandler(DWORD dwControl,
67 DWORD dwEventType,
68 LPVOID lpEventData,
69 LPVOID lpContext)
70 {
71 TRACE("ServiceControlHandler() called\n");
72
73 switch (dwControl)
74 {
75 case SERVICE_CONTROL_STOP:
76 TRACE(" SERVICE_CONTROL_STOP received\n");
77 UpdateServiceStatus(SERVICE_STOP_PENDING);
78 /* Stop listening to incoming RPC messages */
79 RpcMgmtStopServerListening(NULL);
80 UpdateServiceStatus(SERVICE_STOPPED);
81 return ERROR_SUCCESS;
82
83 case SERVICE_CONTROL_PAUSE:
84 TRACE(" SERVICE_CONTROL_PAUSE received\n");
85 UpdateServiceStatus(SERVICE_PAUSED);
86 return ERROR_SUCCESS;
87
88 case SERVICE_CONTROL_CONTINUE:
89 TRACE(" SERVICE_CONTROL_CONTINUE received\n");
90 UpdateServiceStatus(SERVICE_RUNNING);
91 return ERROR_SUCCESS;
92
93 case SERVICE_CONTROL_INTERROGATE:
94 TRACE(" SERVICE_CONTROL_INTERROGATE received\n");
95 SetServiceStatus(ServiceStatusHandle,
96 &ServiceStatus);
97 return ERROR_SUCCESS;
98
99 case SERVICE_CONTROL_SHUTDOWN:
100 TRACE(" SERVICE_CONTROL_SHUTDOWN received\n");
101 UpdateServiceStatus(SERVICE_STOP_PENDING);
102 UpdateServiceStatus(SERVICE_STOPPED);
103 return ERROR_SUCCESS;
104
105 default :
106 TRACE(" Control %lu received\n", dwControl);
107 return ERROR_CALL_NOT_IMPLEMENTED;
108 }
109 }
110
111
112 static
113 DWORD
114 ServiceInit(VOID)
115 {
116 HANDLE hThread;
117
118 /* Initialize the job list */
119 InitializeListHead(&JobListHead);
120
121 /* Initialize the job list lock */
122 RtlInitializeResource(&JobListLock);
123
124 hThread = CreateThread(NULL,
125 0,
126 (LPTHREAD_START_ROUTINE)RpcThreadRoutine,
127 NULL,
128 0,
129 NULL);
130 if (!hThread)
131 {
132 ERR("Can't create PortThread\n");
133 return GetLastError();
134 }
135
136 CloseHandle(hThread);
137
138 return ERROR_SUCCESS;
139 }
140
141
142 VOID WINAPI
143 SchedServiceMain(DWORD argc, LPTSTR *argv)
144 {
145 DWORD dwError;
146
147 UNREFERENCED_PARAMETER(argc);
148 UNREFERENCED_PARAMETER(argv);
149
150 TRACE("SchedServiceMain() called\n");
151
152 ServiceStatusHandle = RegisterServiceCtrlHandlerExW(ServiceName,
153 ServiceControlHandler,
154 NULL);
155 if (!ServiceStatusHandle)
156 {
157 ERR("RegisterServiceCtrlHandlerExW() failed! (Error %lu)\n", GetLastError());
158 return;
159 }
160
161 UpdateServiceStatus(SERVICE_START_PENDING);
162
163 dwError = ServiceInit();
164 if (dwError != ERROR_SUCCESS)
165 {
166 ERR("Service stopped (dwError: %lu\n", dwError);
167 UpdateServiceStatus(SERVICE_STOPPED);
168 return;
169 }
170
171 UpdateServiceStatus(SERVICE_RUNNING);
172 }
173
174
175 BOOL WINAPI
176 DllMain(HINSTANCE hinstDLL,
177 DWORD fdwReason,
178 LPVOID lpvReserved)
179 {
180 switch (fdwReason)
181 {
182 case DLL_PROCESS_ATTACH:
183 DisableThreadLibraryCalls(hinstDLL);
184 break;
185
186 case DLL_PROCESS_DETACH:
187 break;
188 }
189
190 return TRUE;
191 }