ReactOS  0.4.13-dev-79-gcd489d8
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 
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 }
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
TRUE
#define TRUE
Definition: types.h:120
create_inf_file
static void create_inf_file(LPCSTR filename, const char *data)
Definition: SetupDiInstallClassExA.c:34
CloseHandle
#define CloseHandle
Definition: compat.h:398
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:46
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
START_TEST
START_TEST(SetupDiInstallClassExA)
Definition: SetupDiInstallClassExA.c:276
RegDeleteKeyW
LONG WINAPI RegDeleteKeyW(_In_ HKEY hKey, _In_ LPCWSTR lpSubKey)
Definition: reg.c:1240
INVALID_HANDLE_VALUE
#define INVALID_HANDLE_VALUE
Definition: compat.h:391
assert
#define assert(x)
Definition: debug.h:53
GetLastError
DWORD WINAPI GetLastError(VOID)
Definition: except.c:1059
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:30
sprintf
#define sprintf(buf, format,...)
Definition: sprintf.c:55
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:87
BOOL
unsigned int BOOL
Definition: ntddk_ex.h:94
GENERIC_WRITE
#define GENERIC_WRITE
Definition: nt_native.h:90
ok
#define ok(value,...)
Definition: CAtlFileMapping.cpp:50
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
MAX_PATH
#define MAX_PATH
Definition: compat.h:26
DWORD
unsigned long DWORD
Definition: ntddk_ex.h:95
SetLastError
#define SetLastError(x)
Definition: compat.h:409
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:126
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:29
del
static VOID del(LPHIST_ENTRY item)
Definition: history.c:199
CREATE_ALWAYS
#define CREATE_ALWAYS
Definition: disk.h:72
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
GetCurrentDirectoryA
DWORD WINAPI GetCurrentDirectoryA(IN DWORD nBufferLength, OUT LPSTR lpBuffer)
Definition: path.c:2145
path
Definition: services.c:325
skip
#define skip(...)
Definition: CAtlFileMapping.cpp:55
OpenServiceA
SC_HANDLE WINAPI OpenServiceA(SC_HANDLE hSCManager, LPCSTR lpServiceName, DWORD dwDesiredAccess)
Definition: scm.c:2055
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:3381
CreateFileA
#define CreateFileA(a, b, c, d, e, f, g)
Definition: compat.h:399
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