34c0546ecc58c86e2693453f8486eddf0781671d
[reactos.git] / modules / rostests / apitests / setupapi / SetupDiInstallClassExA.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 static void test_SetupDiInstallClassExA(void)
47 {
48 char inf[2048];
49 char path[MAX_PATH];
50 BOOL ret;
51 ULONG del;
52 HKEY RegHandle;
53 HINF infhandle;
54 SC_HANDLE scm_handle, svc_handle;
55
56 /* [Version]:Signature */
57 strcpy(inf, "[Version]\nSignature=\"$Chicago$\"\n");
58 create_inf_file(inffile, inf);
59 sprintf(path, "%s\\%s", CURR_DIR, inffile);
60
61 SetLastError(0xdeadbeef);
62 ret = SetupDiInstallClassExA(NULL, path, DI_QUIETINSTALL, NULL, NULL, NULL,NULL);
63 ok(!ret, "Expected failure\n");
64 ok(GetLastError() == ERROR_INVALID_CLASS,
65 "Expected ERROR_INVALID_CLASS, got %08x\n", (int)GetLastError());
66 DeleteFileA(inffile);
67
68 /* [Version]:Signature+Class */
69 strcat(inf, "Class=MySampleClass\n");
70 create_inf_file(inffile, inf);
71 sprintf(path, "%s\\%s", CURR_DIR, inffile);
72
73 SetLastError(0xdeadbeef);
74 ret = SetupDiInstallClassExA(NULL, path, DI_QUIETINSTALL, NULL, NULL, NULL,NULL);
75 ok(!ret, "Expected failure\n");
76 ok(GetLastError() == ERROR_INVALID_CLASS,
77 "Expected ERROR_INVALID_CLASS, got %08x\n", (int)GetLastError());
78 DeleteFileA(inffile);
79
80 /* [Version]:Signature+Class+ClassGUID */
81 strcat(inf, "ClassGuid={3b409830-5f9d-432a-abf5-7d2e4e102467}\n");
82 create_inf_file(inffile, inf);
83 sprintf(path, "%s\\%s", CURR_DIR, inffile);
84
85 SetLastError(0xdeadbeef);
86 ret = SetupDiInstallClassExA(NULL, path, DI_QUIETINSTALL, NULL, NULL, NULL,NULL);
87 ok(!ret, "Expected failure\n");
88 ok(GetLastError() == ERROR_SECTION_NOT_FOUND,
89 "Expected ERROR_SECTION_NOT_FOUND, got %08x\n", (int)GetLastError());
90 DeleteFileA(inffile);
91
92 /* [Version]Signature+Class+ClassGUID;[ClassInstall32.NT]Empty */
93 strcat(inf, "[ClassInstall32.NT]\n");
94 create_inf_file(inffile, inf);
95 sprintf(path, "%s\\%s", CURR_DIR, inffile);
96
97 SetLastError(0xdeadbeef);
98 ret = SetupDiInstallClassExA(NULL, path, DI_QUIETINSTALL, NULL, NULL, NULL,NULL);
99 ok(ret, "Expected success\n");
100 ok(!GetLastError(),
101 "Expected no error, got %08x\n", (int)GetLastError());
102 DeleteFileA(inffile);
103 if (ret == TRUE)
104 {
105 RegOpenKeyExW(HKEY_LOCAL_MACHINE, L"SYSTEM", 0, KEY_ALL_ACCESS, &RegHandle);
106 del = RegDeleteKeyW(RegHandle, L"CurrentControlSet\\Control\\Class\\{3B409830-5F9D-432A-ABF5-7D2E4E102467}");
107 ok(del == ERROR_SUCCESS, "Expected success \n");
108 }
109
110 /* [Version]Signature+Class+ClassGUID;[ClassInstall32.NT]AddReg */
111 strcat(inf, "AddReg=SampleClassAddReg\n");
112 create_inf_file(inffile, inf);
113 sprintf(path, "%s\\%s", CURR_DIR, inffile);
114
115 SetLastError(0xdeadbeef);
116 ret = SetupDiInstallClassExA(NULL, path, DI_QUIETINSTALL, NULL, NULL, NULL,NULL);
117 ok(ret, "Expected success\n");
118 ok(!GetLastError(),
119 "Expected no error, got %08x\n", (int)GetLastError());
120 DeleteFileA(inffile);
121 if (ret == TRUE)
122 {
123 RegOpenKeyExW(HKEY_LOCAL_MACHINE, L"SYSTEM", 0, KEY_ALL_ACCESS, &RegHandle);
124 del = RegDeleteKeyW(RegHandle, L"CurrentControlSet\\Control\\Class\\{3B409830-5F9D-432A-ABF5-7D2E4E102467}");
125 ok(del == ERROR_SUCCESS, "Expected success \n");
126 }
127
128 /* [Version]Signature+Class+ClassGUID;[ClassInstall32.NT]AddReg; [SampleClassAddReg];*/
129 strcat(inf, "[SampleClassAddReg]\n");
130 create_inf_file(inffile, inf);
131 sprintf(path, "%s\\%s", CURR_DIR, inffile);
132
133 SetLastError(0xdeadbeef);
134 ret = SetupDiInstallClassExA(NULL, path, DI_QUIETINSTALL, NULL, NULL, NULL,NULL);
135 ok(ret, "Expected success\n");
136 ok(!GetLastError(),
137 "Expected no error, got %08x\n", (int)GetLastError());
138 DeleteFileA(inffile);
139 if (ret == TRUE)
140 {
141 RegOpenKeyExW(HKEY_LOCAL_MACHINE, L"SYSTEM", 0, KEY_ALL_ACCESS, &RegHandle);
142 del = RegDeleteKeyW(RegHandle, L"CurrentControlSet\\Control\\Class\\{3B409830-5F9D-432A-ABF5-7D2E4E102467}");
143 ok(del == ERROR_SUCCESS, "Expected success \n");
144 }
145
146 /* [Version]Signature+Class+ClassGUID;[ClassInstall32.NT]AddReg; [SampleClassAddReg]HKR;*/
147 strcat(inf, "HKR,,,,\"ReactOS Test SetupDiInstallClassExA\"\n");
148 create_inf_file(inffile, inf);
149 sprintf(path, "%s\\%s", CURR_DIR, inffile);
150
151 SetLastError(0xdeadbeef);
152 ret = SetupDiInstallClassExA(NULL, path, DI_QUIETINSTALL, NULL, NULL, NULL,NULL);
153 ok(ret, "Expected success\n");
154 ok(!GetLastError(),
155 "Expected no error, got %08x\n", (int)GetLastError());
156 DeleteFileA(inffile);
157 if (ret == TRUE)
158 {
159 RegOpenKeyExW(HKEY_LOCAL_MACHINE, L"SYSTEM", 0, KEY_ALL_ACCESS, &RegHandle);
160 del = RegDeleteKeyW(RegHandle, L"CurrentControlSet\\Control\\Class\\{3B409830-5F9D-432A-ABF5-7D2E4E102467}");
161 ok(del == ERROR_SUCCESS, "Expected success \n");
162 }
163
164 /*[Version]Signature+Class+ClassGUID;[ClassInstall32.NT]AddReg;[SampleClassAddReg]HKR;[ClassInstall32.NT.Services]*/
165 strcat(inf, "[ClassInstall32.NT.Services]\n");
166 create_inf_file(inffile, inf);
167 sprintf(path, "%s\\%s", CURR_DIR, inffile);
168
169 SetLastError(0xdeadbeef);
170 ret = SetupDiInstallClassExA(NULL, path, DI_QUIETINSTALL, NULL, NULL, NULL,NULL);
171 ok(ret, "Expected success\n");
172 ok(!GetLastError(),
173 "Expected no error, got %08x\n", (int)GetLastError());
174 DeleteFileA(inffile);
175 if (ret == TRUE)
176 {
177 RegOpenKeyExW(HKEY_LOCAL_MACHINE, L"SYSTEM", 0, KEY_ALL_ACCESS, &RegHandle);
178 del = RegDeleteKeyW(RegHandle, L"CurrentControlSet\\Control\\Class\\{3B409830-5F9D-432A-ABF5-7D2E4E102467}");
179 ok(del == ERROR_SUCCESS, "Expected success\n");
180 }
181
182 /* Add a reference */
183 strcat(inf, "AddService=Reactostest,,Reactostest.Service\n");
184 create_inf_file(inffile, inf);
185 infhandle = SetupOpenInfFileA(path, NULL, INF_STYLE_WIN4, NULL);
186 SetLastError(0xdeadbeef);
187 ret = SetupDiInstallClassExA(NULL, path, DI_QUIETINSTALL, NULL, NULL, NULL,NULL);
188 ok(!ret, "Expected failure\n");
189 ok(GetLastError() == ERROR_BAD_SERVICE_INSTALLSECT,
190 "Expected ERROR_BAD_SERVICE_INSTALLSECT, got %08x\n", (int)GetLastError());
191 SetupCloseInfFile(infhandle);
192 DeleteFileA(inffile);
193
194 /* Add the section */
195 strcat(inf, "[Reactostest.Service]\n");
196 create_inf_file(inffile, inf);
197 infhandle = SetupOpenInfFileA(path, NULL, INF_STYLE_WIN4, NULL);
198 SetLastError(0xdeadbeef);
199 ret = SetupDiInstallClassExA(NULL, path, DI_QUIETINSTALL, NULL, NULL, NULL,NULL);
200 ok(!ret, "Expected failure\n");
201 ok(GetLastError() == ERROR_BAD_SERVICE_INSTALLSECT,
202 "Expected ERROR_BAD_SERVICE_INSTALLSECT, got %08x\n", (int)GetLastError());
203 SetupCloseInfFile(infhandle);
204 DeleteFileA(inffile);
205
206 /* Just the ServiceBinary */
207 strcat(inf, "ServiceBinary=%12%\\reactostest.sys\n");
208 create_inf_file(inffile, inf);
209 infhandle = SetupOpenInfFileA(path, NULL, INF_STYLE_WIN4, NULL);
210 SetLastError(0xdeadbeef);
211 ret = SetupDiInstallClassExA(NULL, path, DI_QUIETINSTALL, NULL, NULL, NULL,NULL);
212 ok(!ret, "Expected failure\n");
213 ok(GetLastError() == ERROR_BAD_SERVICE_INSTALLSECT,
214 "Expected ERROR_BAD_SERVICE_INSTALLSECT, got %08x\n", (int)GetLastError());
215 SetupCloseInfFile(infhandle);
216 DeleteFileA(inffile);
217
218 /* Add the ServiceType */
219 strcat(inf, "ServiceType=1\n");
220 create_inf_file(inffile, inf);
221 infhandle = SetupOpenInfFileA(path, NULL, INF_STYLE_WIN4, NULL);
222 SetLastError(0xdeadbeef);
223 ret = SetupDiInstallClassExA(NULL, path, DI_QUIETINSTALL, NULL, NULL, NULL,NULL);
224 ok(!ret, "Expected failure\n");
225 ok(GetLastError() == ERROR_BAD_SERVICE_INSTALLSECT,
226 "Expected ERROR_BAD_SERVICE_INSTALLSECT, got %08x\n", (int)GetLastError());
227 SetupCloseInfFile(infhandle);
228 DeleteFileA(inffile);
229
230 /* Add the StartType */
231 strcat(inf, "StartType=4\n");
232 create_inf_file(inffile, inf);
233 infhandle = SetupOpenInfFileA(path, NULL, INF_STYLE_WIN4, NULL);
234 SetLastError(0xdeadbeef);
235 ret = SetupDiInstallClassExA(NULL, path, DI_QUIETINSTALL, NULL, NULL, NULL,NULL);
236 ok(!ret, "Expected failure\n");
237 ok(GetLastError() == ERROR_BAD_SERVICE_INSTALLSECT,
238 "Expected ERROR_BAD_SERVICE_INSTALLSECT, got %08x\n", (int)GetLastError());
239 SetupCloseInfFile(infhandle);
240 DeleteFileA(inffile);
241
242 /* This should be it, the minimal entries to install a service */
243 strcat(inf, "ErrorControl=1");
244 create_inf_file(inffile, inf);
245 infhandle = SetupOpenInfFileA(path, NULL, INF_STYLE_WIN4, NULL);
246 SetLastError(0xdeadbeef);
247 ret = SetupDiInstallClassExA(NULL, path, DI_QUIETINSTALL, NULL, NULL, NULL,NULL);
248 if (!ret && GetLastError() == ERROR_ACCESS_DENIED)
249 {
250 skip("Not enough rights to install the service\n");
251 SetupCloseInfFile(infhandle);
252 DeleteFileA(inffile);
253 return;
254 }
255 ok(ret, "Expected success\n");
256 ok(GetLastError() == ERROR_SUCCESS,
257 "Expected ERROR_SUCCESS, got %08x\n", (int)GetLastError());
258 SetupCloseInfFile(infhandle);
259 DeleteFileA(inffile);
260
261 scm_handle = OpenSCManagerA(NULL, NULL, GENERIC_ALL);
262
263 /* Open the service to see if it's really there */
264 svc_handle = OpenServiceA(scm_handle, "Reactostest", DELETE);
265 ok(svc_handle != NULL, "Service was not created\n");
266
267 SetLastError(0xdeadbeef);
268 ret = DeleteService(svc_handle);
269 ok(ret, "Service could not be deleted : %d\n", (int)GetLastError());
270
271 CloseServiceHandle(svc_handle);
272 CloseServiceHandle(scm_handle);
273 }
274
275
276 START_TEST(SetupDiInstallClassExA)
277 {
278 char temp_path[MAX_PATH], prev_path[MAX_PATH];
279 DWORD len;
280
281 GetCurrentDirectoryA(MAX_PATH, prev_path);
282 GetTempPathA(MAX_PATH, temp_path);
283 SetCurrentDirectoryA(temp_path);
284
285 strcpy(CURR_DIR, temp_path);
286 len = strlen(CURR_DIR);
287 if (len && (CURR_DIR[len - 1] == '\\'))
288 CURR_DIR[len - 1] = 0;
289
290 test_SetupDiInstallClassExA();
291 SetCurrentDirectoryA(prev_path);
292
293 }