ReactOS 0.4.16-dev-1489-g8fbbb41
SetupDiInstallClassExA.c
Go to the documentation of this file.
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#include <strsafe.h>
29
30static const char inffile[] = "test.inf";
31static char CURR_DIR[MAX_PATH];
32
33/* Helpers */
34
35static void create_inf_file(LPCSTR filename, const char *data)
36{
37 DWORD res;
38 BOOL ret;
39 HANDLE handle = CreateFileA(filename, GENERIC_WRITE, 0, NULL,
40 CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
41 assert(handle != INVALID_HANDLE_VALUE);
42 ret = WriteFile(handle, data, lstrlenA(data), &res, NULL);
43 assert(ret != 0);
44 CloseHandle(handle);
45}
46
47static void test_SetupDiInstallClassExA(void)
48{
49 char inf[2048];
50 char path[MAX_PATH];
51 BOOL ret;
52 ULONG del;
53 HKEY RegHandle;
54 HINF infhandle;
55 SC_HANDLE scm_handle, svc_handle;
56
57 /* [Version]:Signature */
58 strcpy(inf, "[Version]\nSignature=\"$Chicago$\"\n");
59 create_inf_file(inffile, inf);
60 StringCbPrintfA(path, sizeof(path), "%s\\%s", CURR_DIR, inffile);
61
62 SetLastError(0xdeadbeef);
63 ret = SetupDiInstallClassExA(NULL, path, DI_QUIETINSTALL, NULL, NULL, NULL,NULL);
64 ok(!ret, "Expected failure\n");
65 ok(GetLastError() == ERROR_INVALID_CLASS,
66 "Expected ERROR_INVALID_CLASS, got %08x\n", (int)GetLastError());
67 DeleteFileA(inffile);
68
69 /* [Version]:Signature+Class */
70 strcat(inf, "Class=MySampleClass\n");
71 create_inf_file(inffile, inf);
72 StringCbPrintfA(path, sizeof(path), "%s\\%s", CURR_DIR, inffile);
73
74 SetLastError(0xdeadbeef);
75 ret = SetupDiInstallClassExA(NULL, path, DI_QUIETINSTALL, NULL, NULL, NULL,NULL);
76 ok(!ret, "Expected failure\n");
77 ok(GetLastError() == ERROR_INVALID_CLASS,
78 "Expected ERROR_INVALID_CLASS, got %08x\n", (int)GetLastError());
79 DeleteFileA(inffile);
80
81 /* [Version]:Signature+Class+ClassGUID */
82 strcat(inf, "ClassGuid={3b409830-5f9d-432a-abf5-7d2e4e102467}\n");
83 create_inf_file(inffile, inf);
84 StringCbPrintfA(path, sizeof(path), "%s\\%s", CURR_DIR, inffile);
85
86 SetLastError(0xdeadbeef);
87 ret = SetupDiInstallClassExA(NULL, path, DI_QUIETINSTALL, NULL, NULL, NULL,NULL);
88 ok(!ret, "Expected failure\n");
89 ok(GetLastError() == ERROR_SECTION_NOT_FOUND,
90 "Expected ERROR_SECTION_NOT_FOUND, got %08x\n", (int)GetLastError());
91 DeleteFileA(inffile);
92
93 /* [Version]Signature+Class+ClassGUID;[ClassInstall32.NT]Empty */
94 strcat(inf, "[ClassInstall32.NT]\n");
95 create_inf_file(inffile, inf);
96 StringCbPrintfA(path, sizeof(path), "%s\\%s", CURR_DIR, inffile);
97
98 SetLastError(0xdeadbeef);
99 ret = SetupDiInstallClassExA(NULL, path, DI_QUIETINSTALL, NULL, NULL, NULL,NULL);
100 ok(ret, "Expected success\n");
101 ok(!GetLastError(),
102 "Expected no error, got %08x\n", (int)GetLastError());
103 DeleteFileA(inffile);
104 if (ret == TRUE)
105 {
106 RegOpenKeyExW(HKEY_LOCAL_MACHINE, L"SYSTEM", 0, KEY_ALL_ACCESS, &RegHandle);
107 del = RegDeleteKeyW(RegHandle, L"CurrentControlSet\\Control\\Class\\{3B409830-5F9D-432A-ABF5-7D2E4E102467}");
108 ok(del == ERROR_SUCCESS, "Expected success\n");
109 }
110
111 /* [Version]Signature+Class+ClassGUID;[ClassInstall32.NT]AddReg */
112 strcat(inf, "AddReg=SampleClassAddReg\n");
113 create_inf_file(inffile, inf);
114 StringCbPrintfA(path, sizeof(path), "%s\\%s", CURR_DIR, inffile);
115
116 SetLastError(0xdeadbeef);
117 ret = SetupDiInstallClassExA(NULL, path, DI_QUIETINSTALL, NULL, NULL, NULL,NULL);
118 ok(ret, "Expected success\n");
119 ok(!GetLastError(),
120 "Expected no error, got %08x\n", (int)GetLastError());
121 DeleteFileA(inffile);
122 if (ret == TRUE)
123 {
124 RegOpenKeyExW(HKEY_LOCAL_MACHINE, L"SYSTEM", 0, KEY_ALL_ACCESS, &RegHandle);
125 del = RegDeleteKeyW(RegHandle, L"CurrentControlSet\\Control\\Class\\{3B409830-5F9D-432A-ABF5-7D2E4E102467}");
126 ok(del == ERROR_SUCCESS, "Expected success\n");
127 }
128
129 /* [Version]Signature+Class+ClassGUID;[ClassInstall32.NT]AddReg; [SampleClassAddReg];*/
130 strcat(inf, "[SampleClassAddReg]\n");
131 create_inf_file(inffile, inf);
132 StringCbPrintfA(path, sizeof(path), "%s\\%s", CURR_DIR, inffile);
133
134 SetLastError(0xdeadbeef);
135 ret = SetupDiInstallClassExA(NULL, path, DI_QUIETINSTALL, NULL, NULL, NULL,NULL);
136 ok(ret, "Expected success\n");
137 ok(!GetLastError(),
138 "Expected no error, got %08x\n", (int)GetLastError());
139 DeleteFileA(inffile);
140 if (ret == TRUE)
141 {
142 RegOpenKeyExW(HKEY_LOCAL_MACHINE, L"SYSTEM", 0, KEY_ALL_ACCESS, &RegHandle);
143 del = RegDeleteKeyW(RegHandle, L"CurrentControlSet\\Control\\Class\\{3B409830-5F9D-432A-ABF5-7D2E4E102467}");
144 ok(del == ERROR_SUCCESS, "Expected success\n");
145 }
146
147 /* [Version]Signature+Class+ClassGUID;[ClassInstall32.NT]AddReg; [SampleClassAddReg]HKR;*/
148 strcat(inf, "HKR,,,,\"ReactOS Test SetupDiInstallClassExA\"\n");
149 create_inf_file(inffile, inf);
150 StringCbPrintfA(path, sizeof(path), "%s\\%s", CURR_DIR, inffile);
151
152 SetLastError(0xdeadbeef);
153 ret = SetupDiInstallClassExA(NULL, path, DI_QUIETINSTALL, NULL, NULL, NULL,NULL);
154 ok(ret, "Expected success\n");
155 ok(!GetLastError(),
156 "Expected no error, got %08x\n", (int)GetLastError());
157 DeleteFileA(inffile);
158 if (ret == TRUE)
159 {
160 RegOpenKeyExW(HKEY_LOCAL_MACHINE, L"SYSTEM", 0, KEY_ALL_ACCESS, &RegHandle);
161 del = RegDeleteKeyW(RegHandle, L"CurrentControlSet\\Control\\Class\\{3B409830-5F9D-432A-ABF5-7D2E4E102467}");
162 ok(del == ERROR_SUCCESS, "Expected success\n");
163 }
164
165 /*[Version]Signature+Class+ClassGUID;[ClassInstall32.NT]AddReg;[SampleClassAddReg]HKR;[ClassInstall32.NT.Services]*/
166 strcat(inf, "[ClassInstall32.NT.Services]\n");
167 create_inf_file(inffile, inf);
168 StringCbPrintfA(path, sizeof(path), "%s\\%s", CURR_DIR, inffile);
169
170 SetLastError(0xdeadbeef);
171 ret = SetupDiInstallClassExA(NULL, path, DI_QUIETINSTALL, NULL, NULL, NULL,NULL);
172 ok(ret, "Expected success\n");
173 ok(!GetLastError(),
174 "Expected no error, got %08x\n", (int)GetLastError());
175 DeleteFileA(inffile);
176 if (ret == TRUE)
177 {
178 RegOpenKeyExW(HKEY_LOCAL_MACHINE, L"SYSTEM", 0, KEY_ALL_ACCESS, &RegHandle);
179 del = RegDeleteKeyW(RegHandle, L"CurrentControlSet\\Control\\Class\\{3B409830-5F9D-432A-ABF5-7D2E4E102467}");
180 ok(del == ERROR_SUCCESS, "Expected success\n");
181 }
182
183 /* Add a reference */
184 strcat(inf, "AddService=Reactostest,,Reactostest.Service\n");
185 create_inf_file(inffile, inf);
186 infhandle = SetupOpenInfFileA(path, NULL, INF_STYLE_WIN4, NULL);
187 SetLastError(0xdeadbeef);
188 ret = SetupDiInstallClassExA(NULL, path, DI_QUIETINSTALL, NULL, NULL, NULL,NULL);
189 ok(!ret, "Expected failure\n");
190 ok(GetLastError() == ERROR_BAD_SERVICE_INSTALLSECT,
191 "Expected ERROR_BAD_SERVICE_INSTALLSECT, got %08x\n", (int)GetLastError());
192 SetupCloseInfFile(infhandle);
193 DeleteFileA(inffile);
194
195 /* Add the section */
196 strcat(inf, "[Reactostest.Service]\n");
197 create_inf_file(inffile, inf);
198 infhandle = SetupOpenInfFileA(path, NULL, INF_STYLE_WIN4, NULL);
199 SetLastError(0xdeadbeef);
200 ret = SetupDiInstallClassExA(NULL, path, DI_QUIETINSTALL, NULL, NULL, NULL,NULL);
201 ok(!ret, "Expected failure\n");
202 ok(GetLastError() == ERROR_BAD_SERVICE_INSTALLSECT,
203 "Expected ERROR_BAD_SERVICE_INSTALLSECT, got %08x\n", (int)GetLastError());
204 SetupCloseInfFile(infhandle);
205 DeleteFileA(inffile);
206
207 /* Just the ServiceBinary */
208 strcat(inf, "ServiceBinary=%12%\\reactostest.sys\n");
209 create_inf_file(inffile, inf);
210 infhandle = SetupOpenInfFileA(path, NULL, INF_STYLE_WIN4, NULL);
211 SetLastError(0xdeadbeef);
212 ret = SetupDiInstallClassExA(NULL, path, DI_QUIETINSTALL, NULL, NULL, NULL,NULL);
213 ok(!ret, "Expected failure\n");
214 ok(GetLastError() == ERROR_BAD_SERVICE_INSTALLSECT,
215 "Expected ERROR_BAD_SERVICE_INSTALLSECT, got %08x\n", (int)GetLastError());
216 SetupCloseInfFile(infhandle);
217 DeleteFileA(inffile);
218
219 /* Add the ServiceType */
220 strcat(inf, "ServiceType=1\n");
221 create_inf_file(inffile, inf);
222 infhandle = SetupOpenInfFileA(path, NULL, INF_STYLE_WIN4, NULL);
223 SetLastError(0xdeadbeef);
224 ret = SetupDiInstallClassExA(NULL, path, DI_QUIETINSTALL, NULL, NULL, NULL,NULL);
225 ok(!ret, "Expected failure\n");
226 ok(GetLastError() == ERROR_BAD_SERVICE_INSTALLSECT,
227 "Expected ERROR_BAD_SERVICE_INSTALLSECT, got %08x\n", (int)GetLastError());
228 SetupCloseInfFile(infhandle);
229 DeleteFileA(inffile);
230
231 /* Add the StartType */
232 strcat(inf, "StartType=4\n");
233 create_inf_file(inffile, inf);
234 infhandle = SetupOpenInfFileA(path, NULL, INF_STYLE_WIN4, NULL);
235 SetLastError(0xdeadbeef);
236 ret = SetupDiInstallClassExA(NULL, path, DI_QUIETINSTALL, NULL, NULL, NULL,NULL);
237 ok(!ret, "Expected failure\n");
238 ok(GetLastError() == ERROR_BAD_SERVICE_INSTALLSECT,
239 "Expected ERROR_BAD_SERVICE_INSTALLSECT, got %08x\n", (int)GetLastError());
240 SetupCloseInfFile(infhandle);
241 DeleteFileA(inffile);
242
243 /* This should be it, the minimal entries to install a service */
244 strcat(inf, "ErrorControl=1");
245 create_inf_file(inffile, inf);
246 infhandle = SetupOpenInfFileA(path, NULL, INF_STYLE_WIN4, NULL);
247 SetLastError(0xdeadbeef);
248 ret = SetupDiInstallClassExA(NULL, path, DI_QUIETINSTALL, NULL, NULL, NULL,NULL);
249 if (!ret && GetLastError() == ERROR_ACCESS_DENIED)
250 {
251 skip("Not enough rights to install the service\n");
252 SetupCloseInfFile(infhandle);
253 DeleteFileA(inffile);
254 return;
255 }
256 ok(ret, "Expected success\n");
257 ok(GetLastError() == ERROR_SUCCESS,
258 "Expected ERROR_SUCCESS, got %08x\n", (int)GetLastError());
259 SetupCloseInfFile(infhandle);
260 DeleteFileA(inffile);
261
262 scm_handle = OpenSCManagerA(NULL, NULL, GENERIC_ALL);
263
264 /* Open the service to see if it's really there */
265 svc_handle = OpenServiceA(scm_handle, "Reactostest", DELETE);
266 ok(svc_handle != NULL, "Service was not created\n");
267
268 SetLastError(0xdeadbeef);
269 ret = DeleteService(svc_handle);
270 ok(ret, "Service could not be deleted : %d\n", (int)GetLastError());
271
272 CloseServiceHandle(svc_handle);
273 CloseServiceHandle(scm_handle);
274}
275
276
277START_TEST(SetupDiInstallClassExA)
278{
279 char temp_path[MAX_PATH], prev_path[MAX_PATH];
280 SIZE_T len;
281
282 GetCurrentDirectoryA(MAX_PATH, prev_path);
283 GetTempPathA(MAX_PATH, temp_path);
284 SetCurrentDirectoryA(temp_path);
285
286 strcpy(CURR_DIR, temp_path);
287 len = strlen(CURR_DIR);
288 if (len && (CURR_DIR[len - 1] == '\\'))
289 CURR_DIR[len - 1] = 0;
290
291 test_SetupDiInstallClassExA();
292 SetCurrentDirectoryA(prev_path);
293
294}
scm_handle
static SC_HANDLE scm_handle
Definition: ServiceArgs.c:20
CURR_DIR
static char CURR_DIR[MAX_PATH]
Definition: SetupDiInstallClassExA.c:31
test_SetupDiInstallClassExA
static void test_SetupDiInstallClassExA(void)
Definition: SetupDiInstallClassExA.c:47
inffile
static const char inffile[]
Definition: SetupDiInstallClassExA.c:30
strlen
ACPI_SIZE strlen(const char *String)
Definition: utclib.c:269
ok
#define ok(value,...)
Definition: atltest.h:57
skip
#define skip(...)
Definition: atltest.h:64
START_TEST
#define START_TEST(x)
Definition: atltest.h:75
del
static VOID del(LPHIST_ENTRY item)
Definition: history.c:199
ERROR_SUCCESS
#define ERROR_SUCCESS
Definition: deptool.c:10
NULL
#define NULL
Definition: types.h:112
TRUE
#define TRUE
Definition: types.h:120
RegOpenKeyExW
LONG WINAPI RegOpenKeyExW(HKEY hKey, LPCWSTR lpSubKey, DWORD ulOptions, REGSAM samDesired, PHKEY phkResult)
Definition: reg.c:3333
RegDeleteKeyW
LONG WINAPI RegDeleteKeyW(_In_ HKEY hKey, _In_ LPCWSTR lpSubKey)
Definition: reg.c:1239
CloseHandle
#define CloseHandle
Definition: compat.h:739
SetLastError
#define SetLastError(x)
Definition: compat.h:752
INVALID_HANDLE_VALUE
#define INVALID_HANDLE_VALUE
Definition: compat.h:731
CreateFileA
#define CreateFileA(a, b, c, d, e, f, g)
Definition: compat.h:740
MAX_PATH
#define MAX_PATH
Definition: compat.h:34
FILE_ATTRIBUTE_NORMAL
#define FILE_ATTRIBUTE_NORMAL
Definition: compat.h:137
ERROR_ACCESS_DENIED
#define ERROR_ACCESS_DENIED
Definition: compat.h:97
DeleteFileA
BOOL WINAPI DeleteFileA(IN LPCSTR lpFileName)
Definition: delete.c:24
WriteFile
BOOL WINAPI WriteFile(IN HANDLE hFile, IN LPCVOID lpBuffer, IN DWORD nNumberOfBytesToWrite OPTIONAL, OUT LPDWORD lpNumberOfBytesWritten, IN LPOVERLAPPED lpOverlapped OPTIONAL)
Definition: rw.c:24
GetCurrentDirectoryA
DWORD WINAPI GetCurrentDirectoryA(IN DWORD nBufferLength, OUT LPSTR lpBuffer)
Definition: path.c:2146
SetCurrentDirectoryA
BOOL WINAPI SetCurrentDirectoryA(IN LPCSTR lpPathName)
Definition: path.c:2206
GetTempPathA
DWORD WINAPI GetTempPathA(IN DWORD nBufferLength, OUT LPSTR lpBuffer)
Definition: path.c:2054
SetupDiInstallClassExA
BOOL WINAPI SetupDiInstallClassExA(IN HWND hwndParent OPTIONAL, IN PCSTR InfFileName OPTIONAL, IN DWORD Flags, IN HSPFILEQ FileQueue OPTIONAL, IN CONST GUID *InterfaceClassGuid OPTIONAL, IN PVOID Reserved1, IN PVOID Reserved2)
Definition: devinst.c:3483
SetupOpenInfFileA
HINF WINAPI SetupOpenInfFileA(PCSTR name, PCSTR class, DWORD style, UINT *error)
Definition: parser.c:1139
assert
#define assert(x)
Definition: debug.h:53
ret
return ret
Definition: mutex.c:146
L
#define L(x)
Definition: resources.c:13
BOOL
unsigned int BOOL
Definition: ntddk_ex.h:94
DWORD
unsigned long DWORD
Definition: ntddk_ex.h:95
data
GLint GLenum GLsizei GLsizei GLsizei GLint GLsizei const GLvoid * data
Definition: gl.h:1950
res
GLuint res
Definition: glext.h:9613
len
GLenum GLsizei len
Definition: glext.h:6722
INF_STYLE_WIN4
#define INF_STYLE_WIN4
Definition: infsupp.h:43
filename
const char * filename
Definition: ioapi.h:137
lstrlenA
int WINAPI lstrlenA(LPCSTR lpString)
Definition: lstring.c:145
CREATE_ALWAYS
#define CREATE_ALWAYS
Definition: disk.h:72
create_inf_file
static void create_inf_file(void)
Definition: advpack.c:202
temp_path
char temp_path[MAX_PATH]
Definition: mspatcha.c:123
KEY_ALL_ACCESS
#define KEY_ALL_ACCESS
Definition: nt_native.h:1041
GENERIC_ALL
#define GENERIC_ALL
Definition: nt_native.h:92
DELETE
#define DELETE
Definition: nt_native.h:57
GENERIC_WRITE
#define GENERIC_WRITE
Definition: nt_native.h:90
DeleteService
BOOL WINAPI DeleteService(SC_HANDLE hService)
Definition: scm.c:921
OpenServiceA
SC_HANDLE WINAPI OpenServiceA(SC_HANDLE hSCManager, LPCSTR lpServiceName, DWORD dwDesiredAccess)
Definition: scm.c:2151
OpenSCManagerA
SC_HANDLE WINAPI OpenSCManagerA(LPCSTR lpMachineName, LPCSTR lpDatabaseName, DWORD dwDesiredAccess)
Definition: scm.c:2063
CloseServiceHandle
BOOL WINAPI CloseServiceHandle(SC_HANDLE hSCObject)
Definition: scm.c:580
strcat
strcat
Definition: string.h:92
strcpy
strcpy
Definition: string.h:131
ERROR_INVALID_CLASS
#define ERROR_INVALID_CLASS
Definition: setupapi.h:303
DI_QUIETINSTALL
#define DI_QUIETINSTALL
Definition: setupapi.h:68
ERROR_BAD_SERVICE_INSTALLSECT
#define ERROR_BAD_SERVICE_INSTALLSECT
Definition: setupapi.h:320
ERROR_SECTION_NOT_FOUND
#define ERROR_SECTION_NOT_FOUND
Definition: setupapi.h:294
StringCbPrintfA
STRSAFEAPI StringCbPrintfA(STRSAFE_LPSTR pszDest, size_t cbDest, STRSAFE_LPCSTR pszFormat,...)
Definition: strsafe.h:547
SIZE_T
ULONG_PTR SIZE_T
Definition: typedefs.h:80
ULONG
uint32_t ULONG
Definition: typedefs.h:59
SetupCloseInfFile
VOID WINAPI SetupCloseInfFile(IN HINF InfHandle)
Definition: infsupp.c:45
GetLastError
DWORD WINAPI GetLastError(void)
Definition: except.c:1042
HKEY_LOCAL_MACHINE
#define HKEY_LOCAL_MACHINE
Definition: winreg.h:12
RegHandle
_In_opt_ PETWENABLECALLBACK _In_opt_ PVOID _Out_ PREGHANDLE RegHandle
Definition: wmifuncs.h:78
LPCSTR
const char * LPCSTR
Definition: xmlstorage.h:183