4e775c302818ec7c72be3daafa34bdc290b994c2
[reactos.git] / modules / rostests / apitests / setupapi / SetupInstallServicesFromInfSectionEx.c
1 /*
2 * SetupAPI device class-related functions tests
3 *
4 * Copyright 2015 Víctor Martínez (victor.martinez@reactos.org)
5 *
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU General Public Licence as
8 * published by the Free Software Foundation; either version 2 of
9 * the License, or (at your option) any later version.
10 *
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * General Public License for more details.
15 *
16 * You should have received a copy of the GNU General Public License
17 * along with this library; if not, write to the Free Software Foundation,
18 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
19 */
20
21 #include <apitest.h>
22 #include <stdio.h>
23 #include <assert.h>
24 #include <winuser.h>
25 #include <winreg.h>
26 #include <winsvc.h>
27 #include <setupapi.h>
28
29 static const char inffile[] = "test.inf";
30 static char CURR_DIR[MAX_PATH];
31
32 /* Helpers */
33
34 static void create_inf_file(LPCSTR filename, const char *data)
35 {
36 DWORD res;
37 BOOL ret;
38 HANDLE handle = CreateFileA(filename, GENERIC_WRITE, 0, NULL,
39 CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
40 assert(handle != INVALID_HANDLE_VALUE);
41 ret = WriteFile(handle, data, strlen(data), &res, NULL);
42 assert(ret != 0);
43 CloseHandle(handle);
44 }
45
46 /* CBT hook to ensure a window (e.g., MessageBox) cannot be created */
47 static HHOOK hhook;
48 static LRESULT CALLBACK cbt_hook_proc(int nCode, WPARAM wParam, LPARAM lParam)
49 {
50 return nCode == HCBT_CREATEWND ? 1: CallNextHookEx(hhook, nCode, wParam, lParam);
51 }
52
53 static void test_SetupInstallServicesFromInfSectionExA(void)
54 {
55 char inf[2048];
56 char path[MAX_PATH];
57 HINF infhandle;
58 BOOL ret;
59 SC_HANDLE scm_handle, svc_handle;
60
61 /* Bail out if we are on win98 */
62 SetLastError(0xdeadbeef);
63 scm_handle = OpenSCManagerA(NULL, NULL, GENERIC_ALL);
64
65 if (!scm_handle && (GetLastError() == ERROR_CALL_NOT_IMPLEMENTED))
66 {
67 win_skip("OpenSCManagerA is not implemented, we are most likely on win9x\n");
68 return;
69 }
70 CloseServiceHandle(scm_handle);
71
72 /* Basic inf file to satisfy SetupOpenInfFileA */
73 strcpy(inf, "[Version]\nSignature=\"$Chicago$\"\n");
74 create_inf_file(inffile, inf);
75 sprintf(path, "%s\\%s", CURR_DIR, inffile);
76 infhandle = SetupOpenInfFileA(path, NULL, INF_STYLE_WIN4, NULL);
77
78 /* Nothing but the Version section */
79 SetLastError(0xdeadbeef);
80 ret = SetupInstallServicesFromInfSectionExA(infhandle, "trolololo", 0, NULL, NULL, NULL,NULL);
81 ok(!ret, "Expected failure\n");
82 ok(GetLastError() == ERROR_SECTION_NOT_FOUND,
83 "Expected ERROR_SECTION_NOT_FOUND, got %08d\n", (int)GetLastError());
84 SetupCloseInfFile(infhandle);
85 DeleteFileA(inffile);
86
87
88 /* Add the section */
89 strcat(inf, "[Winetest.Services]\n");
90 create_inf_file(inffile, inf);
91 infhandle = SetupOpenInfFileA(path, NULL, INF_STYLE_WIN4, NULL);
92 SetLastError(0xdeadbeef);
93 ret = SetupInstallServicesFromInfSectionExA(infhandle, "Winetest.Services", 0, NULL, NULL, NULL, NULL);
94 ok(!ret, "Expected failure\n");
95 ok(GetLastError() == ERROR_SECTION_NOT_FOUND,
96 "Expected ERROR_SECTION_NOT_FOUND, got %08d\n", (int)GetLastError());
97 SetupCloseInfFile(infhandle);
98 DeleteFileA(inffile);
99
100 /* Add a reference */
101 strcat(inf, "AddService=Winetest,,Winetest.Service\n");
102 create_inf_file(inffile, inf);
103 infhandle = SetupOpenInfFileA(path, NULL, INF_STYLE_WIN4, NULL);
104 SetLastError(0xdeadbeef);
105 ret = SetupInstallServicesFromInfSectionExA(infhandle, "Winetest.Services", 0, NULL, NULL, NULL, NULL);
106 ok(!ret, "Expected failure\n");
107 ok(GetLastError() == ERROR_BAD_SERVICE_INSTALLSECT,
108 "Expected ERROR_BAD_SERVICE_INSTALLSECT, got %08d\n", (int)GetLastError());
109 SetupCloseInfFile(infhandle);
110 DeleteFileA(inffile);
111
112 /* Add the section */
113 strcat(inf, "[Winetest.Service]\n");
114 create_inf_file(inffile, inf);
115 infhandle = SetupOpenInfFileA(path, NULL, INF_STYLE_WIN4, NULL);
116 SetLastError(0xdeadbeef);
117 ret = SetupInstallServicesFromInfSectionExA(infhandle, "Winetest.Services", 0, NULL, NULL, NULL, NULL);
118 ok(!ret, "Expected failure\n");
119 ok(GetLastError() == ERROR_BAD_SERVICE_INSTALLSECT,
120 "Expected ERROR_BAD_SERVICE_INSTALLSECT, got %08d\n", (int)GetLastError());
121 SetupCloseInfFile(infhandle);
122 DeleteFileA(inffile);
123
124 /* Just the ServiceBinary */
125 strcat(inf, "ServiceBinary=%12%\\winetest.sys\n");
126 create_inf_file(inffile, inf);
127 infhandle = SetupOpenInfFileA(path, NULL, INF_STYLE_WIN4, NULL);
128 SetLastError(0xdeadbeef);
129 ret = SetupInstallServicesFromInfSectionExA(infhandle, "Winetest.Services", 0, NULL, NULL, NULL, NULL);
130 ok(!ret, "Expected failure\n");
131 ok(GetLastError() == ERROR_BAD_SERVICE_INSTALLSECT,
132 "Expected ERROR_BAD_SERVICE_INSTALLSECT, got %08d\n", (int)GetLastError());
133 SetupCloseInfFile(infhandle);
134 DeleteFileA(inffile);
135
136 /* Add the ServiceType */
137 strcat(inf, "ServiceType=1\n");
138 create_inf_file(inffile, inf);
139 infhandle = SetupOpenInfFileA(path, NULL, INF_STYLE_WIN4, NULL);
140 SetLastError(0xdeadbeef);
141 ret = SetupInstallServicesFromInfSectionExA(infhandle, "Winetest.Services", 0, NULL, NULL, NULL, NULL);
142 ok(!ret, "Expected failure\n");
143 ok(GetLastError() == ERROR_BAD_SERVICE_INSTALLSECT,
144 "Expected ERROR_BAD_SERVICE_INSTALLSECT, got %08d\n", (int)GetLastError());
145 SetupCloseInfFile(infhandle);
146 DeleteFileA(inffile);
147
148 /* Add the StartType */
149 strcat(inf, "StartType=4\n");
150 create_inf_file(inffile, inf);
151 infhandle = SetupOpenInfFileA(path, NULL, INF_STYLE_WIN4, NULL);
152 SetLastError(0xdeadbeef);
153 ret = SetupInstallServicesFromInfSectionExA(infhandle, "Winetest.Services", 0, NULL, NULL, NULL, NULL);
154 ok(!ret, "Expected failure\n");
155 ok(GetLastError() == ERROR_BAD_SERVICE_INSTALLSECT,
156 "Expected ERROR_BAD_SERVICE_INSTALLSECT, got %08d\n", (int)GetLastError());
157 SetupCloseInfFile(infhandle);
158 DeleteFileA(inffile);
159
160 /* This should be it, the minimal entries to install a service */
161 strcat(inf, "ErrorControl=1");
162 create_inf_file(inffile, inf);
163 infhandle = SetupOpenInfFileA(path, NULL, INF_STYLE_WIN4, NULL);
164 SetLastError(0xdeadbeef);
165 ret = SetupInstallServicesFromInfSectionExA(infhandle, "Winetest.Services", 0, NULL, NULL, NULL, NULL);
166 if (!ret && GetLastError() == ERROR_ACCESS_DENIED)
167 {
168 skip("Not enough rights to install the service\n");
169 SetupCloseInfFile(infhandle);
170 DeleteFileA(inffile);
171 return;
172 }
173 ok(ret, "Expected success\n");
174 ok(GetLastError() == ERROR_SUCCESS,
175 "Expected ERROR_SUCCESS, got %08d\n", (int)GetLastError());
176 SetupCloseInfFile(infhandle);
177 DeleteFileA(inffile);
178
179 scm_handle = OpenSCManagerA(NULL, NULL, GENERIC_ALL);
180
181 /* Open the service to see if it's really there */
182 svc_handle = OpenServiceA(scm_handle, "Winetest", DELETE);
183 ok(svc_handle != NULL, "Service was not created\n");
184
185 SetLastError(0xdeadbeef);
186 ret = DeleteService(svc_handle);
187 ok(ret, "Service could not be deleted : %d\n", (int)GetLastError());
188
189 CloseServiceHandle(svc_handle);
190 CloseServiceHandle(scm_handle);
191
192 strcpy(inf, "[Version]\nSignature=\"$Chicago$\"\n");
193 strcat(inf, "[XSP.InstallPerVer]\n");
194 strcat(inf, "AddReg=AspEventlogMsg.Reg,Perf.Reg,AspVersions.Reg,FreeADO.Reg,IndexServer.Reg\n");
195 create_inf_file(inffile, inf);
196 sprintf(path, "%s\\%s", CURR_DIR, inffile);
197 infhandle = SetupOpenInfFileA(path, NULL, INF_STYLE_WIN4, NULL);
198
199 SetLastError(0xdeadbeef);
200 ret = SetupInstallServicesFromInfSectionExA(infhandle, "XSP.InstallPerVer", 0, NULL, NULL, NULL,NULL);
201 ok(ret, "Expected success\n");
202 ok(GetLastError() == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %08d\n", (int)GetLastError());
203 SetupCloseInfFile(infhandle);
204 DeleteFileA(inffile);
205
206 /* TODO: Test the Flags */
207 }
208 static void test_SetupInstallServicesFromInfSectionExW(void)
209 {
210 char inf[2048];
211 char path[MAX_PATH];
212 HINF infhandle;
213 BOOL ret;
214 SC_HANDLE scm_handle, svc_handle;
215
216 /* Bail out if we are on win98 */
217 SetLastError(0xdeadbeef);
218 scm_handle = OpenSCManagerW(NULL, NULL, GENERIC_ALL);
219
220 if (!scm_handle && (GetLastError() == ERROR_CALL_NOT_IMPLEMENTED))
221 {
222 win_skip("OpenSCManagerW is not implemented, we are most likely on win9x\n");
223 return;
224 }
225 CloseServiceHandle(scm_handle);
226
227 /* Basic inf file to satisfy SetupOpenInfFileA */
228 strcpy(inf, "[Version]\nSignature=\"$Chicago$\"\n");
229 create_inf_file(inffile, inf);
230 sprintf(path, "%s\\%s", CURR_DIR, inffile);
231 infhandle = SetupOpenInfFileA(path, NULL, INF_STYLE_WIN4, NULL);
232
233 /* Nothing but the Version section */
234 SetLastError(0xdeadbeef);
235 ret = SetupInstallServicesFromInfSectionExW(infhandle, L"trolololo", 0, NULL, NULL, NULL,NULL);
236 ok(!ret, "Expected failure\n");
237 ok(GetLastError() == ERROR_SECTION_NOT_FOUND,
238 "Expected ERROR_SECTION_NOT_FOUND, got %08d\n", (int)GetLastError());
239 SetupCloseInfFile(infhandle);
240 DeleteFileA(inffile);
241
242
243 /* Add the section */
244 strcat(inf, "[Winetest.Services]\n");
245 create_inf_file(inffile, inf);
246 infhandle = SetupOpenInfFileA(path, NULL, INF_STYLE_WIN4, NULL);
247 SetLastError(0xdeadbeef);
248 ret = SetupInstallServicesFromInfSectionExW(infhandle, L"Winetest.Services", 0, NULL, NULL, NULL, NULL);
249 ok(!ret, "Expected failure\n");
250 ok(GetLastError() == ERROR_SECTION_NOT_FOUND,
251 "Expected ERROR_SECTION_NOT_FOUND, got %08d\n", (int)GetLastError());
252 SetupCloseInfFile(infhandle);
253 DeleteFileA(inffile);
254
255 /* Add a reference */
256 strcat(inf, "AddService=Winetest,,Winetest.Service\n");
257 create_inf_file(inffile, inf);
258 infhandle = SetupOpenInfFileA(path, NULL, INF_STYLE_WIN4, NULL);
259 SetLastError(0xdeadbeef);
260 ret = SetupInstallServicesFromInfSectionExW(infhandle, L"Winetest.Services", 0, NULL, NULL, NULL, NULL);
261 ok(!ret, "Expected failure\n");
262 ok(GetLastError() == ERROR_BAD_SERVICE_INSTALLSECT,
263 "Expected ERROR_BAD_SERVICE_INSTALLSECT, got %08d\n", (int)GetLastError());
264 SetupCloseInfFile(infhandle);
265 DeleteFileA(inffile);
266
267 /* Add the section */
268 strcat(inf, "[Winetest.Service]\n");
269 create_inf_file(inffile, inf);
270 infhandle = SetupOpenInfFileA(path, NULL, INF_STYLE_WIN4, NULL);
271 SetLastError(0xdeadbeef);
272 ret = SetupInstallServicesFromInfSectionExW(infhandle, L"Winetest.Services", 0, NULL, NULL, NULL, NULL);
273 ok(!ret, "Expected failure\n");
274 ok(GetLastError() == ERROR_BAD_SERVICE_INSTALLSECT,
275 "Expected ERROR_BAD_SERVICE_INSTALLSECT, got %08d\n", (int)GetLastError());
276 SetupCloseInfFile(infhandle);
277 DeleteFileA(inffile);
278
279 /* Just the ServiceBinary */
280 strcat(inf, "ServiceBinary=%12%\\winetest.sys\n");
281 create_inf_file(inffile, inf);
282 infhandle = SetupOpenInfFileA(path, NULL, INF_STYLE_WIN4, NULL);
283 SetLastError(0xdeadbeef);
284 ret = SetupInstallServicesFromInfSectionExW(infhandle, L"Winetest.Services", 0, NULL, NULL, NULL, NULL);
285 ok(!ret, "Expected failure\n");
286 ok(GetLastError() == ERROR_BAD_SERVICE_INSTALLSECT,
287 "Expected ERROR_BAD_SERVICE_INSTALLSECT, got %08d\n", (int)GetLastError());
288 SetupCloseInfFile(infhandle);
289 DeleteFileA(inffile);
290
291 /* Add the ServiceType */
292 strcat(inf, "ServiceType=1\n");
293 create_inf_file(inffile, inf);
294 infhandle = SetupOpenInfFileA(path, NULL, INF_STYLE_WIN4, NULL);
295 SetLastError(0xdeadbeef);
296 ret = SetupInstallServicesFromInfSectionExW(infhandle, L"Winetest.Services", 0, NULL, NULL, NULL, NULL);
297 ok(!ret, "Expected failure\n");
298 ok(GetLastError() == ERROR_BAD_SERVICE_INSTALLSECT,
299 "Expected ERROR_BAD_SERVICE_INSTALLSECT, got %08d\n", (int)GetLastError());
300 SetupCloseInfFile(infhandle);
301 DeleteFileA(inffile);
302
303 /* Add the StartType */
304 strcat(inf, "StartType=4\n");
305 create_inf_file(inffile, inf);
306 infhandle = SetupOpenInfFileA(path, NULL, INF_STYLE_WIN4, NULL);
307 SetLastError(0xdeadbeef);
308 ret = SetupInstallServicesFromInfSectionExW(infhandle, L"Winetest.Services", 0, NULL, NULL, NULL, NULL);
309 ok(!ret, "Expected failure\n");
310 ok(GetLastError() == ERROR_BAD_SERVICE_INSTALLSECT,
311 "Expected ERROR_BAD_SERVICE_INSTALLSECT, got %08d\n", (int)GetLastError());
312 SetupCloseInfFile(infhandle);
313 DeleteFileA(inffile);
314
315 /* This should be it, the minimal entries to install a service */
316 strcat(inf, "ErrorControl=1");
317 create_inf_file(inffile, inf);
318 infhandle = SetupOpenInfFileA(path, NULL, INF_STYLE_WIN4, NULL);
319 SetLastError(0xdeadbeef);
320 ret = SetupInstallServicesFromInfSectionExW(infhandle, L"Winetest.Services", 0, NULL, NULL, NULL, NULL);
321 if (!ret && GetLastError() == ERROR_ACCESS_DENIED)
322 {
323 skip("Not enough rights to install the service\n");
324 SetupCloseInfFile(infhandle);
325 DeleteFileA(inffile);
326 return;
327 }
328 ok(ret, "Expected success\n");
329 ok(GetLastError() == ERROR_SUCCESS,
330 "Expected ERROR_SUCCESS, got %08d\n", (int)GetLastError());
331 SetupCloseInfFile(infhandle);
332 DeleteFileA(inffile);
333
334 scm_handle = OpenSCManagerW(NULL, NULL, GENERIC_ALL);
335
336 /* Open the service to see if it's really there */
337 svc_handle = OpenServiceW(scm_handle, L"Winetest", DELETE);
338 ok(svc_handle != NULL, "Service was not created\n");
339
340 SetLastError(0xdeadbeef);
341 ret = DeleteService(svc_handle);
342 ok(ret, "Service could not be deleted : %d\n", (int)GetLastError());
343
344 CloseServiceHandle(svc_handle);
345 CloseServiceHandle(scm_handle);
346
347 strcpy(inf, "[Version]\nSignature=\"$Chicago$\"\n");
348 strcat(inf, "[XSP.InstallPerVer]\n");
349 strcat(inf, "AddReg=AspEventlogMsg.Reg,Perf.Reg,AspVersions.Reg,FreeADO.Reg,IndexServer.Reg\n");
350 create_inf_file(inffile, inf);
351 sprintf(path, "%s\\%s", CURR_DIR, inffile);
352 infhandle = SetupOpenInfFileA(path, NULL, INF_STYLE_WIN4, NULL);
353
354 SetLastError(0xdeadbeef);
355 ret = SetupInstallServicesFromInfSectionExW(infhandle, L"XSP.InstallPerVer", 0, NULL, NULL, NULL, NULL);
356 ok(ret, "Expected success\n");
357 ok(GetLastError() == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %08d\n", (int)GetLastError());
358 SetupCloseInfFile(infhandle);
359 DeleteFileA(inffile);
360
361 /* TODO: Test the Flags */
362 }
363
364
365 START_TEST(SetupInstallServicesFromInfSectionEx)
366 {
367 char temp_path[MAX_PATH], prev_path[MAX_PATH];
368
369 GetCurrentDirectoryA(MAX_PATH, prev_path);
370 GetTempPathA(MAX_PATH, temp_path);
371 SetCurrentDirectoryA(temp_path);
372
373 strcpy(CURR_DIR, temp_path);
374
375 /* Set CBT hook to disallow MessageBox creation in current thread */
376 hhook = SetWindowsHookExA(WH_CBT, cbt_hook_proc, 0, GetCurrentThreadId());
377 assert(hhook != 0);
378
379 test_SetupInstallServicesFromInfSectionExA();
380 test_SetupInstallServicesFromInfSectionExW();
381
382 UnhookWindowsHookEx(hhook);
383 SetCurrentDirectoryA(prev_path);
384 }