ReactOS  0.4.15-dev-1206-g731eddf
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 
30 static const char inffile[] = "test.inf";
31 static char CURR_DIR[MAX_PATH];
32 
33 /* Helpers */
34 
35 static 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 
47 static 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 
277 START_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 }
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
GENERIC_ALL
#define GENERIC_ALL
Definition: nt_native.h:92
ERROR_SECTION_NOT_FOUND
#define ERROR_SECTION_NOT_FOUND
Definition: setupapi.h:288
create_inf_file
static void create_inf_file(LPCSTR filename, const char *data)
Definition: SetupDiInstallClassExA.c:35
CloseHandle
#define CloseHandle
Definition: compat.h:598
INF_STYLE_WIN4
#define INF_STYLE_WIN4
Definition: infsupp.h:41
ERROR_SUCCESS
#define ERROR_SUCCESS
Definition: deptool.c:10
test_SetupDiInstallClassExA
static void test_SetupDiInstallClassExA(void)
Definition: SetupDiInstallClassExA.c:47
strcat
char * strcat(char *DstString, const char *SrcString)
Definition: utclib.c:568
strlen
ACPI_SIZE strlen(const char *String)
Definition: utclib.c:269
scm_handle
static SC_HANDLE scm_handle
Definition: ServiceArgs.c:20
TRUE
#define TRUE
Definition: types.h:120
START_TEST
START_TEST(SetupDiInstallClassExA)
Definition: SetupDiInstallClassExA.c:277
RegDeleteKeyW
LONG WINAPI RegDeleteKeyW(_In_ HKEY hKey, _In_ LPCWSTR lpSubKey)
Definition: reg.c:1237
INVALID_HANDLE_VALUE
#define INVALID_HANDLE_VALUE
Definition: compat.h:590
assert
#define assert(x)
Definition: debug.h:53
GetLastError
DWORD WINAPI GetLastError(VOID)
Definition: except.c:1044
DeleteService
BOOL WINAPI DeleteService(SC_HANDLE hService)
Definition: scm.c:916
filename
const char * filename
Definition: ioapi.h:135
SetupCloseInfFile
VOID WINAPI SetupCloseInfFile(IN HINF InfHandle)
Definition: infsupp.c:43
CURR_DIR
static char CURR_DIR[MAX_PATH]
Definition: SetupDiInstallClassExA.c:31
OpenSCManagerA
SC_HANDLE WINAPI OpenSCManagerA(LPCSTR lpMachineName, LPCSTR lpDatabaseName, DWORD dwDesiredAccess)
Definition: scm.c:1967
ERROR_ACCESS_DENIED
#define ERROR_ACCESS_DENIED
Definition: compat.h:97
BOOL
unsigned int BOOL
Definition: ntddk_ex.h:94
GENERIC_WRITE
#define GENERIC_WRITE
Definition: nt_native.h:90
CloseServiceHandle
BOOL WINAPI CloseServiceHandle(SC_HANDLE hSCObject)
Definition: scm.c:577
NULL
smooth NULL
Definition: ftsmooth.c:416
DeleteFileA
BOOL WINAPI DeleteFileA(IN LPCSTR lpFileName)
Definition: delete.c:24
RegHandle
_In_opt_ PETWENABLECALLBACK _In_opt_ PVOID _Out_ PREGHANDLE RegHandle
Definition: wmifuncs.h:76
LPCSTR
const char * LPCSTR
Definition: xmlstorage.h:183
ERROR_BAD_SERVICE_INSTALLSECT
#define ERROR_BAD_SERVICE_INSTALLSECT
Definition: setupapi.h:314
StringCbPrintfA
STRSAFEAPI StringCbPrintfA(STRSAFE_LPSTR pszDest, size_t cbDest, STRSAFE_LPCSTR pszFormat,...)
Definition: strsafe.h:547
MAX_PATH
#define MAX_PATH
Definition: compat.h:34
DWORD
unsigned long DWORD
Definition: ntddk_ex.h:95
SetLastError
#define SetLastError(x)
Definition: compat.h:611
DI_QUIETINSTALL
#define DI_QUIETINSTALL
Definition: setupapi.h:68
data
GLint GLenum GLsizei GLsizei GLsizei GLint GLsizei const GLvoid * data
Definition: gl.h:1950
ret
int ret
Definition: wcstombs-tests.c:31
FILE_ATTRIBUTE_NORMAL
#define FILE_ATTRIBUTE_NORMAL
Definition: compat.h:137
L
static const WCHAR L[]
Definition: oid.c:1250
SetCurrentDirectoryA
BOOL WINAPI SetCurrentDirectoryA(IN LPCSTR lpPathName)
Definition: path.c:2205
len
GLenum GLsizei len
Definition: glext.h:6722
SetupOpenInfFileA
HINF WINAPI SetupOpenInfFileA(PCSTR name, PCSTR class, DWORD style, UINT *error)
Definition: parser.c:1139
inffile
static const char inffile[]
Definition: SetupDiInstallClassExA.c:30
SIZE_T
ULONG_PTR SIZE_T
Definition: typedefs.h:80
del
static VOID del(LPHIST_ENTRY item)
Definition: history.c:199
CREATE_ALWAYS
#define CREATE_ALWAYS
Definition: disk.h:72
lstrlenA
int WINAPI lstrlenA(LPCSTR lpString)
Definition: lstring.c:145
GetTempPathA
DWORD WINAPI GetTempPathA(IN DWORD nBufferLength, OUT LPSTR lpBuffer)
Definition: path.c:2053
KEY_ALL_ACCESS
#define KEY_ALL_ACCESS
Definition: nt_native.h:1041
ERROR_INVALID_CLASS
#define ERROR_INVALID_CLASS
Definition: setupapi.h:297
ok
#define ok(value,...)
Definition: atltest.h:57
GetCurrentDirectoryA
DWORD WINAPI GetCurrentDirectoryA(IN DWORD nBufferLength, OUT LPSTR lpBuffer)
Definition: path.c:2145
OpenServiceA
SC_HANDLE WINAPI OpenServiceA(SC_HANDLE hSCManager, LPCSTR lpServiceName, DWORD dwDesiredAccess)
Definition: scm.c:2055
skip
#define skip(...)
Definition: atltest.h:64
res
GLuint res
Definition: glext.h:9613
strcpy
char * strcpy(char *DstString, const char *SrcString)
Definition: utclib.c:388
ULONG
unsigned int ULONG
Definition: retypes.h:1
RegOpenKeyExW
LONG WINAPI RegOpenKeyExW(HKEY hKey, LPCWSTR lpSubKey, DWORD ulOptions, REGSAM samDesired, PHKEY phkResult)
Definition: reg.c:3366
CreateFileA
#define CreateFileA(a, b, c, d, e, f, g)
Definition: compat.h:599
DELETE
#define DELETE
Definition: nt_native.h:57
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:3488
HKEY_LOCAL_MACHINE
#define HKEY_LOCAL_MACHINE
Definition: winreg.h:12
temp_path
char temp_path[MAX_PATH]
Definition: mspatcha.c:123