1 #include "DriverTester.h"
2
3 static BOOL
Initialize(LPCWSTR lpDriverPath)4 Initialize(LPCWSTR lpDriverPath)
5 {
6 if (!RegisterDriver(DRIVER_NAME, lpDriverPath))
7 {
8 wprintf(L"[%lu] Failed to install %s\n", GetLastError(), DRIVER_NAME);
9 return FALSE;
10 }
11
12 return TRUE;
13 }
14
15 static BOOL
Uninitialize(LPCWSTR lpDriverPath)16 Uninitialize(LPCWSTR lpDriverPath)
17 {
18 if (!UnregisterDriver(DRIVER_NAME))
19 {
20 wprintf(L"[%lu] Failed to unregister %s\n", GetLastError(), DRIVER_NAME);
21 return FALSE;
22 }
23
24 return TRUE;
25 }
26
27 static BOOL
UsermodeMethod(LPCWSTR lpDriverPath)28 UsermodeMethod(LPCWSTR lpDriverPath)
29 {
30 wprintf(L"\nStarting %s.sys via the SCM\n", DRIVER_NAME);
31
32 if (!StartDriver(DRIVER_NAME))
33 {
34 wprintf(L"[%lu] Failed to start %s\n", GetLastError(), DRIVER_NAME);
35 UnregisterDriver(DRIVER_NAME);
36 return FALSE;
37 }
38
39 wprintf(L"\tStarted\n");
40
41 wprintf(L"Stopping %s.sys via the SCM\n", DRIVER_NAME);
42
43 if (!StopDriver(DRIVER_NAME))
44 {
45 wprintf(L"[%lu] Failed to stop %s\n", GetLastError(), DRIVER_NAME);
46 UnregisterDriver(DRIVER_NAME);
47 return FALSE;
48 }
49
50 wprintf(L"\tStopped\n");
51
52 return TRUE;
53 }
54
55 static BOOL
UndocumentedMethod(LPCWSTR lpDriverPath)56 UndocumentedMethod(LPCWSTR lpDriverPath)
57 {
58 wprintf(L"\nStarting %s.sys via native API\n", DRIVER_NAME);
59
60 if (!NtStartDriver(DRIVER_NAME))
61 {
62 wprintf(L"[%lu] Failed to start %s\n", GetLastError(), DRIVER_NAME);
63 UnregisterDriver(DRIVER_NAME);
64 return FALSE;
65 }
66
67 wprintf(L"\tStarted\n");
68
69 wprintf(L"Stopping %s.sys via native API\n", DRIVER_NAME);
70
71 if (!NtStopDriver(DRIVER_NAME))
72 {
73 wprintf(L"[%lu] Failed to stop %s\n", GetLastError(), DRIVER_NAME);
74 UnregisterDriver(DRIVER_NAME);
75 return FALSE;
76 }
77
78 wprintf(L"\tStopped\n");
79
80 return TRUE;
81 }
82
83
84 static BOOL
SneakyUndocumentedMethods(LPCWSTR lpDriverPath)85 SneakyUndocumentedMethods(LPCWSTR lpDriverPath)
86 {
87 WCHAR szDevice[MAX_PATH];
88
89 if (ConvertPath(lpDriverPath, szDevice))
90 {
91 wprintf(L"\nStarting %s.sys via NtSetSystemInformation with SystemLoadGdiDriverInformation\n", DRIVER_NAME);
92 if (LoadVia_SystemLoadGdiDriverInformation(szDevice))
93 {
94 wprintf(L"\tStarted\n");
95
96 NtStopDriver(DRIVER_NAME);
97 }
98
99 wprintf(L"\nStarting %s.sys via NtSetSystemInformation with SystemExtendServiceTableInformation\n", DRIVER_NAME);
100 if (LoadVia_SystemExtendServiceTableInformation(szDevice))
101 {
102 wprintf(L"\tStarted\n");
103
104 NtStopDriver(DRIVER_NAME);
105 }
106
107 return TRUE;
108 }
109
110 return FALSE;
111 }
112
113
wmain(int argc,wchar_t * argv[])114 int __cdecl wmain(int argc, wchar_t *argv[])
115 {
116 WCHAR buf[MAX_PATH];
117
118 if (argc != 2)
119 {
120 wprintf(L"Usage: DriverTester.exe <path>");
121 return -1;
122 }
123
124 if (!SearchPathW(NULL,
125 argv[1],
126 L".sys",
127 MAX_PATH,
128 buf,
129 NULL))
130 {
131 wprintf(L"%s does not exist", argv[1]);
132 return -1;
133 }
134
135 if (Initialize(argv[1]))
136 {
137 //
138 // Load using conventional SCM methods
139 //
140 UsermodeMethod(argv[1]);
141
142 //
143 // Load using undocumented NtLoad/UnloadDriver
144 //
145 UndocumentedMethod(argv[1]);
146
147 //
148 // Load using hidden unknown methods
149 //
150 SneakyUndocumentedMethods(argv[1]);
151
152 Uninitialize(argv[1]);
153 }
154
155 return 0;
156 }
157
158