huwcampbell.com
Reactos
1#include "DriverTester.h"
2
3
4BOOL
5RegisterDriver(LPCWSTR lpDriverName,
6 LPCWSTR lpPathName)
7{
8 SC_HANDLE hSCManager;
9 SC_HANDLE hService;
10
11 hSCManager = OpenSCManagerW(NULL,
12 NULL,
13 SC_MANAGER_ALL_ACCESS);
14 if (!hSCManager)
15 return FALSE;
16
17retry:
18 hService = CreateServiceW(hSCManager,
19 lpDriverName,
20 lpDriverName,
21 SERVICE_ALL_ACCESS,
22 SERVICE_KERNEL_DRIVER,
23 SERVICE_DEMAND_START,
24 SERVICE_ERROR_NORMAL,
25 lpPathName,
26 NULL,
27 NULL,
28 NULL,
29 NULL,
30 NULL);
31
32 if (hService)
33 {
34 CloseServiceHandle(hService);
35 CloseServiceHandle(hSCManager);
36 return TRUE;
37 }
38 else
39 {
40 DWORD err = GetLastError();
41
42 if (err == ERROR_SERVICE_MARKED_FOR_DELETE)
43 {
44 StopDriver(DRIVER_NAME);
45 goto retry;
46 }
47
48 CloseServiceHandle(hSCManager);
49
50 // return TRUE if the driver is already registered
51 return (err == ERROR_SERVICE_EXISTS);
52 }
53}
54
55BOOL
56StartDriver(LPCWSTR lpDriverName)
57{
58 SC_HANDLE hSCManager;
59 SC_HANDLE hService;
60 BOOL bRet;
61
62 hSCManager = OpenSCManagerW(NULL,
63 NULL,
64 SC_MANAGER_ALL_ACCESS);
65 if (!hSCManager)
66 return FALSE;
67
68 hService = OpenServiceW(hSCManager,
69 lpDriverName,
70 SERVICE_ALL_ACCESS);
71 if (!hService)
72 {
73 CloseServiceHandle(hSCManager);
74 return FALSE;
75 }
76
77 bRet = StartServiceW(hService, 0, NULL);
78 if (!bRet)
79 {
80 if (GetLastError() == ERROR_SERVICE_ALREADY_RUNNING)
81 {
82 wprintf(L"%s.sys already running\n", DRIVER_NAME);
83 bRet = TRUE;
84 }
85 }
86
87 CloseServiceHandle(hService);
88 CloseServiceHandle(hSCManager);
89
90 return bRet;
91}
92
93BOOL
94StopDriver(LPCWSTR lpDriverName)
95{
96 SC_HANDLE hSCManager;
97 SC_HANDLE hService;
98 SERVICE_STATUS serviceStatus;
99 BOOL bRet;
100
101 hSCManager = OpenSCManagerW(NULL,
102 NULL,
103 SC_MANAGER_ALL_ACCESS);
104 if (!hSCManager)
105 return FALSE;
106
107 hService = OpenServiceW(hSCManager,
108 lpDriverName,
109 SERVICE_ALL_ACCESS);
110 if (!hService)
111 {
112 CloseServiceHandle(hSCManager);
113 return FALSE;
114 }
115
116 bRet = ControlService(hService,
117 SERVICE_CONTROL_STOP,
118 &serviceStatus);
119 if (!bRet)
120 {
121 if (GetLastError() == ERROR_SERVICE_NOT_ACTIVE)
122 {
123 wprintf(L"%s.sys wasn't running\n", DRIVER_NAME);
124 bRet = TRUE;
125 }
126 }
127
128 CloseServiceHandle(hService);
129 CloseServiceHandle(hSCManager);
130
131 return bRet;
132}
133
134BOOL
135UnregisterDriver(LPCWSTR lpDriverName)
136{
137 SC_HANDLE hService;
138 SC_HANDLE hSCManager;
139 BOOL bRet;
140
141 hSCManager = OpenSCManagerW(NULL,
142 NULL,
143 SC_MANAGER_ALL_ACCESS);
144 if (!hSCManager)
145 return FALSE;
146
147 hService = OpenServiceW(hSCManager,
148 lpDriverName,
149 SERVICE_ALL_ACCESS);
150 if (!hService)
151 {
152 CloseServiceHandle(hSCManager);
153 return FALSE;
154 }
155
156 bRet = DeleteService(hService);
157
158 CloseServiceHandle(hService);
159 CloseServiceHandle(hSCManager);
160
161 return bRet;
162}