Improve swedish translations and add new translations to swedish.
[reactos.git] / base / applications / mscutils / servman / control.c
1 /*
2 * PROJECT: ReactOS Services
3 * LICENSE: GPL - See COPYING in the top level directory
4 * FILE: base/applications/mscutils/servman/control.c
5 * PURPOSE: Pauses and resumes a service
6 * COPYRIGHT: Copyright 2006-2015 Ged Murphy <gedmurphy@reactos.org>
7 *
8 */
9
10 #include "precomp.h"
11
12 #define MAX_WAIT_TIME 30000
13
14 BOOL
15 DoControlService(LPWSTR ServiceName,
16 HWND hProgress,
17 DWORD Control)
18 {
19 SC_HANDLE hSCManager;
20 SC_HANDLE hService;
21 SERVICE_STATUS_PROCESS ServiceStatus = {0};
22 SERVICE_STATUS Status;
23 DWORD BytesNeeded = 0;
24 DWORD StartTickCount;
25 DWORD OldCheckPoint;
26 DWORD WaitTime;
27 DWORD MaxWait;
28 DWORD ReqState, i;
29 BOOL Result;
30
31 /* Set the state we're interested in */
32 switch (Control)
33 {
34 case SERVICE_CONTROL_PAUSE:
35 ReqState = SERVICE_PAUSED;
36 break;
37 case SERVICE_CONTROL_CONTINUE:
38 ReqState = SERVICE_RUNNING;
39 break;
40 default:
41 /* Unhandled control code */
42 return FALSE;
43 }
44
45 hSCManager = OpenSCManagerW(NULL,
46 NULL,
47 SC_MANAGER_CONNECT);
48 if (!hSCManager) return FALSE;
49
50 hService = OpenServiceW(hSCManager,
51 ServiceName,
52 SERVICE_PAUSE_CONTINUE | SERVICE_INTERROGATE | SERVICE_QUERY_STATUS);
53 if (!hService)
54 {
55 CloseServiceHandle(hSCManager);
56 return FALSE;
57 }
58
59 /* Send the control message to the service */
60 Result = ControlService(hService,
61 Control,
62 &Status);
63 if (Result)
64 {
65 if (hProgress)
66 {
67 /* Increment the progress bar */
68 IncrementProgressBar(hProgress, DEFAULT_STEP);
69 }
70
71 /* Get the service status */
72 Result = QueryServiceStatusEx(hService,
73 SC_STATUS_PROCESS_INFO,
74 (LPBYTE)&ServiceStatus,
75 sizeof(SERVICE_STATUS_PROCESS),
76 &BytesNeeded);
77 if (Result)
78 {
79 Result = FALSE;
80 MaxWait = MAX_WAIT_TIME;
81 OldCheckPoint = ServiceStatus.dwCheckPoint;
82 StartTickCount = GetTickCount();
83
84 /* Loop until it's at the correct state */
85 while (ServiceStatus.dwCurrentState != ReqState)
86 {
87 /* Fixup the wait time */
88 WaitTime = ServiceStatus.dwWaitHint / 10;
89
90 if (WaitTime < 1000) WaitTime = 1000;
91 else if (WaitTime > 10000) WaitTime = 10000;
92
93 /* We don't wanna wait for up to 10 secs without incrementing */
94 for (i = WaitTime / 1000; i > 0; i--)
95 {
96 Sleep(1000);
97 if (hProgress)
98 {
99 /* Increment the progress bar */
100 IncrementProgressBar(hProgress, DEFAULT_STEP);
101 }
102 }
103
104 /* Get the latest status info */
105 if (!QueryServiceStatusEx(hService,
106 SC_STATUS_PROCESS_INFO,
107 (LPBYTE)&ServiceStatus,
108 sizeof(SERVICE_STATUS_PROCESS),
109 &BytesNeeded))
110 {
111 /* Something went wrong... */
112 break;
113 }
114
115 /* Is the service making progress? */
116 if (ServiceStatus.dwCheckPoint > OldCheckPoint)
117 {
118 /* It is, get the latest tickcount to reset the max wait time */
119 StartTickCount = GetTickCount();
120 OldCheckPoint = ServiceStatus.dwCheckPoint;
121 IncrementProgressBar(hProgress, DEFAULT_STEP);
122 }
123 else
124 {
125 /* It's not, make sure we haven't exceeded our wait time */
126 if(GetTickCount() >= StartTickCount + MaxWait)
127 {
128 /* We have, give up */
129 break;
130 }
131 }
132 }
133 }
134
135 if (ServiceStatus.dwCurrentState == ReqState)
136 {
137 Result = TRUE;
138 }
139 }
140
141 CloseServiceHandle(hService);
142 CloseServiceHandle(hSCManager);
143
144 return Result;
145 }