CustomServiceController/ServiceController.cpp at main · gokulkrishna-k/CustomServiceController · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
#include "ServiceController.h"


LONG ServiceController::createSvc(CString csBinaryPathName, CString csDisplayName, CString csServiceName) {

	lpszBinaryPathName = (LPCTSTR)csBinaryPathName;
	lpszDisplayName = (LPCTSTR)csDisplayName;
	lpszServiceName = (LPCTSTR)csServiceName;

	schService = CreateService(
		schSCManager,
		lpszServiceName,
		lpszDisplayName,
		SERVICE_ALL_ACCESS,
		SERVICE_WIN32_OWN_PROCESS,
		SERVICE_DEMAND_START,
		SERVICE_ERROR_NORMAL,
		lpszBinaryPathName,
		NULL,
		NULL,
		NULL,
		NULL,
		NULL
	);
	if (schService == NULL) {
		return GetLastError();
	}
}

LONG ServiceController::startSvc() {


	if (!StartService(schService, 0, NULL)) {
		return GetLastError();
	}

	if (!QueryServiceStatusEx(
		schService,
		SC_STATUS_PROCESS_INFO,
		(LPBYTE)&ssStatus,
		sizeof(SERVICE_STATUS_PROCESS),
		&dwBytesNeeded))
	{
		return GetLastError();
	}

	if (ssStatus.dwCurrentState == SERVICE_RUNNING)
	{
		return SERVICE_RUNNING;
	}
	else
	{
		return GetLastError();
	}
}

LONG ServiceController::stopSvc() {

	if (!ControlService(
		schService,
		SERVICE_CONTROL_STOP,
		(LPSERVICE_STATUS)&ssStatus))
	{
		return GetLastError();
	}

	while (ssStatus.dwCurrentState == SERVICE_STOP_PENDING) {
		if (QueryServiceStatusEx(
			schService,
			SC_STATUS_PROCESS_INFO,
			(LPBYTE)&ssStatus,
			sizeof(SERVICE_STATUS_PROCESS),
			&dwBytesNeeded)) {

			if (ssStatus.dwCurrentState == SERVICE_STOPPED) {
				return SERVICE_STOPPED;
			}
			Sleep(1000);
		}
	}

	return GetLastError();

}

LONG ServiceController::sendCustomCtrlMsg(DWORD ctrlMsg) {

	DWORD dwBytesNeeded;

	if (!ControlService(
		schService,
		ctrlMsg,
		(LPSERVICE_STATUS)&ssStatus))
	{
		return GetLastError();
	}
	else {
		return 0;
	}
}