2 * Unit tests for setupapi.dll query functions
4 * Copyright (C) 2006 James Hawkins
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
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 * Lesser General Public License for more details.
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
24 #include "wine/test.h"
26 CHAR CURR_DIR
[MAX_PATH
];
27 CHAR WIN_DIR
[MAX_PATH
];
29 static void get_directories(void)
33 GetCurrentDirectoryA(MAX_PATH
, CURR_DIR
);
34 len
= lstrlenA(CURR_DIR
);
36 if(len
&& (CURR_DIR
[len
-1] == '\\'))
39 GetWindowsDirectoryA(WIN_DIR
, MAX_PATH
);
40 len
= lstrlenA(WIN_DIR
);
42 if (len
&& (WIN_DIR
[len
-1] == '\\'))
46 static void append_str(char **str
, const char *data
)
52 static void create_inf_file(LPSTR filename
)
56 DWORD dwNumberOfBytesWritten
;
57 HANDLE hf
= CreateFile(filename
, GENERIC_WRITE
, 0, NULL
,
58 CREATE_ALWAYS
, FILE_ATTRIBUTE_NORMAL
, NULL
);
60 append_str(&ptr
, "[Version]\n");
61 append_str(&ptr
, "Signature=\"$Chicago$\"\n");
62 append_str(&ptr
, "AdvancedINF=2.5\n");
63 append_str(&ptr
, "[SourceDisksNames]\n");
64 append_str(&ptr
, "2 = %%SrcDiskName%%, LANCOM\\LANtools\\lanconf.cab\n");
65 append_str(&ptr
, "[SourceDisksFiles]\n");
66 append_str(&ptr
, "lanconf.exe = 2\n");
67 append_str(&ptr
, "[DestinationDirs]\n");
68 append_str(&ptr
, "DefaultDestDir = 24, %%DefaultDest%%\n");
69 append_str(&ptr
, "[Strings]\n");
70 append_str(&ptr
, "LangDir = english\n");
71 append_str(&ptr
, "DefaultDest = LANCOM\n");
72 append_str(&ptr
, "SrcDiskName = \"LANCOM Software CD\"\n");
74 WriteFile(hf
, data
, ptr
- data
, &dwNumberOfBytesWritten
, NULL
);
78 static void create_inf_file2(LPSTR filename
)
82 DWORD dwNumberOfBytesWritten
;
83 HANDLE hf
= CreateFile(filename
, GENERIC_WRITE
, 0, NULL
,
84 CREATE_ALWAYS
, FILE_ATTRIBUTE_NORMAL
, NULL
);
86 append_str(&ptr
, "[SourceFileInfo]\n");
87 append_str(&ptr
, "sp1qfe\\bitsinst.exe=250B3702C7CCD7C2F9E4DAA1555C933E,000600060A28062C,27136,SP1QFE\n");
88 append_str(&ptr
, "sp1qfe\\bitsprx2.dll=4EBEA67F4BB4EB402E725CA7CA2857AE,000600060A280621,7680,SP1QFE\n");
89 append_str(&ptr
, "sp1qfe\\bitsprx3.dll=C788A1D9330DA011EF25E95D3BC7BDE5,000600060A280621,7168,SP1QFE\n");
90 append_str(&ptr
, "sp1qfe\\qmgr.dll=696AC82FB290A03F205901442E0E9589,000600060A280621,361984,SP1QFE\n");
91 append_str(&ptr
, "sp1qfe\\qmgrprxy.dll=8B5848144829E1BC985EA4C3D8CA7E3F,000600060A280621,17408,SP1QFE\n");
92 append_str(&ptr
, "sp1qfe\\winhttp.dll=3EC6F518114606CA59D4160322077437,000500010A280615,331776,SP1QFE\n");
93 append_str(&ptr
, "sp1qfe\\xpob2res.dll=DB83156B9F496F20D1EA70E4ABEC0166,000500010A280622,158720,SP1QFE\n");
95 WriteFile(hf
, data
, ptr
- data
, &dwNumberOfBytesWritten
, NULL
);
99 static BOOL
check_info_filename(PSP_INF_INFORMATION info
, LPSTR test
)
105 if (!SetupQueryInfFileInformationA(info
, 0, NULL
, 0, &size
))
108 filename
= HeapAlloc(GetProcessHeap(), 0, size
);
112 SetupQueryInfFileInformationA(info
, 0, filename
, size
, &size
);
114 if (!lstrcmpiA(test
, filename
))
117 HeapFree(GetProcessHeap(), 0, filename
);
121 static PSP_INF_INFORMATION
alloc_inf_info(LPCSTR filename
, DWORD search
, PDWORD size
)
123 PSP_INF_INFORMATION info
;
126 ret
= SetupGetInfInformationA(filename
, search
, NULL
, 0, size
);
130 info
= HeapAlloc(GetProcessHeap(), 0, *size
);
134 static void test_SetupGetInfInformation(void)
136 PSP_INF_INFORMATION info
;
137 CHAR inf_filename
[MAX_PATH
];
138 CHAR inf_one
[MAX_PATH
], inf_two
[MAX_PATH
];
144 lstrcpyA(inf_filename
, CURR_DIR
);
145 lstrcatA(inf_filename
, "\\");
146 lstrcatA(inf_filename
, "test.inf");
148 /* try an invalid inf handle */
150 SetLastError(0xbeefcafe);
151 ret
= SetupGetInfInformationA(NULL
, INFINFO_INF_SPEC_IS_HINF
, NULL
, 0, &size
);
152 ok(ret
== FALSE
, "Expected SetupGetInfInformation to fail\n");
153 ok(GetLastError() == ERROR_INVALID_HANDLE
||
154 broken(GetLastError() == ERROR_BAD_PATHNAME
) || /* win95 */
155 broken(GetLastError() == ERROR_FILE_NOT_FOUND
) || /* win98 */
156 broken(GetLastError() == ERROR_PATH_NOT_FOUND
) || /* NT4 */
157 broken(GetLastError() == ERROR_INVALID_NAME
), /* win2k */
158 "Expected ERROR_INVALID_HANDLE, got %d\n", GetLastError());
159 ok(size
== 0xdeadbeef, "Expected size to remain unchanged\n");
161 /* try an invalid inf filename */
162 /* do not use NULL as absolute inf filename on win9x (crash) */
163 if ((GetLastError() != ERROR_BAD_PATHNAME
) && /* win95 */
164 (GetLastError() != ERROR_FILE_NOT_FOUND
)) /* win98 */
167 SetLastError(0xbeefcafe);
168 ret
= SetupGetInfInformationA(NULL
, INFINFO_INF_NAME_IS_ABSOLUTE
, NULL
, 0, &size
);
169 ok(ret
== FALSE
, "Expected SetupGetInfInformation to fail\n");
170 ok(GetLastError() == ERROR_INVALID_PARAMETER
,
171 "Expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
172 ok(size
== 0xdeadbeef, "Expected size to remain unchanged\n");
175 create_inf_file(inf_filename
);
177 /* try an invalid search flag */
179 SetLastError(0xbeefcafe);
180 ret
= SetupGetInfInformationA(inf_filename
, -1, NULL
, 0, &size
);
181 ok(ret
== FALSE
, "Expected SetupGetInfInformation to fail\n");
182 ok(GetLastError() == ERROR_INVALID_PARAMETER
,
183 "Expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
184 ok(size
== 0xdeadbeef, "Expected size to remain unchanged\n");
186 /* try a nonexistent inf file */
188 SetLastError(0xbeefcafe);
189 ret
= SetupGetInfInformationA("idontexist", INFINFO_INF_NAME_IS_ABSOLUTE
, NULL
, 0, &size
);
190 ok(ret
== FALSE
, "Expected SetupGetInfInformation to fail\n");
191 ok(GetLastError() == ERROR_FILE_NOT_FOUND
,
192 "Expected ERROR_FILE_NOT_FOUND, got %d\n", GetLastError());
193 ok(size
== 0xdeadbeef, "Expected size to remain unchanged\n");
195 /* successfully open the inf file */
197 ret
= SetupGetInfInformationA(inf_filename
, INFINFO_INF_NAME_IS_ABSOLUTE
, NULL
, 0, &size
);
198 ok(ret
== TRUE
, "Expected SetupGetInfInformation to succeed\n");
199 ok(size
!= 0xdeadbeef, "Expected a valid size on return\n");
201 /* set ReturnBuffer to NULL and ReturnBufferSize to non-zero value 'size' */
202 SetLastError(0xbeefcafe);
203 ret
= SetupGetInfInformationA(inf_filename
, INFINFO_INF_NAME_IS_ABSOLUTE
, NULL
, size
, &size
);
204 ok(ret
== FALSE
, "Expected SetupGetInfInformation to fail\n");
205 ok(GetLastError() == ERROR_INVALID_PARAMETER
,
206 "Expected ERROR_INVALID_PARAMETER, got %d\n", GetLastError());
208 /* set ReturnBuffer to NULL and ReturnBufferSize to non-zero value 'size-1' */
209 ret
= SetupGetInfInformationA(inf_filename
, INFINFO_INF_NAME_IS_ABSOLUTE
, NULL
, size
-1, &size
);
210 ok(ret
== TRUE
, "Expected SetupGetInfInformation to succeed\n");
212 /* some tests for behaviour with a NULL RequiredSize pointer */
213 ret
= SetupGetInfInformationA(inf_filename
, INFINFO_INF_NAME_IS_ABSOLUTE
, NULL
, 0, NULL
);
214 ok(ret
== TRUE
, "Expected SetupGetInfInformation to succeed\n");
215 ret
= SetupGetInfInformationA(inf_filename
, INFINFO_INF_NAME_IS_ABSOLUTE
, NULL
, size
- 1, NULL
);
216 ok(ret
== TRUE
, "Expected SetupGetInfInformation to succeed\n");
217 ret
= SetupGetInfInformationA(inf_filename
, INFINFO_INF_NAME_IS_ABSOLUTE
, NULL
, size
, NULL
);
218 ok(ret
== FALSE
, "Expected SetupGetInfInformation to fail\n");
220 info
= HeapAlloc(GetProcessHeap(), 0, size
);
222 /* try valid ReturnBuffer but too small size */
223 SetLastError(0xbeefcafe);
224 ret
= SetupGetInfInformationA(inf_filename
, INFINFO_INF_NAME_IS_ABSOLUTE
, info
, size
- 1, &size
);
225 ok(ret
== FALSE
, "Expected SetupGetInfInformation to fail\n");
226 ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER
,
227 "Expected ERROR_INSUFFICIENT_BUFFER, got %d\n", GetLastError());
229 /* successfully get the inf information */
230 ret
= SetupGetInfInformationA(inf_filename
, INFINFO_INF_NAME_IS_ABSOLUTE
, info
, size
, &size
);
231 ok(ret
== TRUE
, "Expected SetupGetInfInformation to succeed\n");
232 ok(check_info_filename(info
, inf_filename
), "Expected returned filename to be equal\n");
234 HeapFree(GetProcessHeap(), 0, info
);
236 /* try the INFINFO_INF_SPEC_IS_HINF search flag */
237 hinf
= SetupOpenInfFileA(inf_filename
, NULL
, INF_STYLE_WIN4
, NULL
);
238 info
= alloc_inf_info(hinf
, INFINFO_INF_SPEC_IS_HINF
, &size
);
239 ret
= SetupGetInfInformationA(hinf
, INFINFO_INF_SPEC_IS_HINF
, info
, size
, &size
);
240 ok(ret
== TRUE
, "Expected SetupGetInfInformation to succeed\n");
241 ok(check_info_filename(info
, inf_filename
), "Expected returned filename to be equal\n");
242 SetupCloseInfFile(hinf
);
244 lstrcpyA(inf_two
, WIN_DIR
);
245 lstrcatA(inf_two
, "\\system32\\");
246 lstrcatA(inf_two
, "test.inf");
247 create_inf_file(inf_two
);
249 HeapFree(GetProcessHeap(), 0, info
);
250 info
= alloc_inf_info("test.inf", INFINFO_DEFAULT_SEARCH
, &size
);
252 /* check if system32 is searched for inf */
253 ret
= SetupGetInfInformationA("test.inf", INFINFO_DEFAULT_SEARCH
, info
, size
, &size
);
254 if (!ret
&& GetLastError() == ERROR_FILE_NOT_FOUND
)
255 revfile
= inf_one
; /* Vista */
259 lstrcpyA(inf_one
, WIN_DIR
);
260 lstrcatA(inf_one
, "\\inf\\");
261 lstrcatA(inf_one
, "test.inf");
262 create_inf_file(inf_one
);
264 HeapFree(GetProcessHeap(), 0, info
);
265 info
= alloc_inf_info("test.inf", INFINFO_DEFAULT_SEARCH
, &size
);
267 /* test the INFINFO_DEFAULT_SEARCH search flag */
268 ret
= SetupGetInfInformationA("test.inf", INFINFO_DEFAULT_SEARCH
, info
, size
, &size
);
269 ok(ret
== TRUE
, "Expected SetupGetInfInformation to succeed: %d\n", GetLastError());
270 ok(check_info_filename(info
, inf_one
), "Expected returned filename to be equal\n");
272 HeapFree(GetProcessHeap(), 0, info
);
273 info
= alloc_inf_info("test.inf", INFINFO_REVERSE_DEFAULT_SEARCH
, &size
);
275 /* test the INFINFO_REVERSE_DEFAULT_SEARCH search flag */
276 ret
= SetupGetInfInformationA("test.inf", INFINFO_REVERSE_DEFAULT_SEARCH
, info
, size
, &size
);
277 ok(ret
== TRUE
, "Expected SetupGetInfInformation to succeed\n");
278 ok(check_info_filename(info
, revfile
), "Expected returned filename to be equal\n");
280 HeapFree(GetProcessHeap(), 0, info
);
282 DeleteFileA(inf_filename
);
283 DeleteFileA(inf_one
);
284 DeleteFileA(inf_two
);
287 static void test_SetupGetSourceFileLocation(void)
289 char buffer
[MAX_PATH
] = "not empty", inf_filename
[MAX_PATH
];
291 DWORD required
, error
;
295 lstrcpyA(inf_filename
, CURR_DIR
);
296 lstrcatA(inf_filename
, "\\");
297 lstrcatA(inf_filename
, "test.inf");
299 create_inf_file(inf_filename
);
301 hinf
= SetupOpenInfFileA(inf_filename
, NULL
, INF_STYLE_WIN4
, NULL
);
302 ok(hinf
!= INVALID_HANDLE_VALUE
, "could not open inf file\n");
307 ret
= SetupGetSourceFileLocationA(hinf
, NULL
, "lanconf.exe", &source_id
, buffer
, sizeof(buffer
), &required
);
308 ok(ret
, "SetupGetSourceFileLocation failed\n");
310 ok(required
== 1, "unexpected required size: %d\n", required
);
311 ok(source_id
== 2, "unexpected source id: %d\n", source_id
);
312 ok(!lstrcmpA("", buffer
), "unexpected result string: %s\n", buffer
);
314 SetupCloseInfFile(hinf
);
315 DeleteFileA(inf_filename
);
317 create_inf_file2(inf_filename
);
319 SetLastError(0xdeadbeef);
320 hinf
= SetupOpenInfFileA(inf_filename
, NULL
, INF_STYLE_WIN4
, NULL
);
321 error
= GetLastError();
322 ok(hinf
== INVALID_HANDLE_VALUE
, "could open inf file\n");
323 ok(error
== ERROR_WRONG_INF_STYLE
, "got wrong error: %d\n", error
);
325 hinf
= SetupOpenInfFileA(inf_filename
, NULL
, INF_STYLE_OLDNT
, NULL
);
326 ok(hinf
!= INVALID_HANDLE_VALUE
, "could not open inf file\n");
328 ret
= SetupGetSourceFileLocationA(hinf
, NULL
, "", &source_id
, buffer
, sizeof(buffer
), &required
);
329 ok(!ret
, "SetupGetSourceFileLocation succeeded\n");
331 SetupCloseInfFile(hinf
);
332 DeleteFileA(inf_filename
);
335 static void test_SetupGetSourceInfo(void)
337 char buffer
[MAX_PATH
], inf_filename
[MAX_PATH
];
342 lstrcpyA(inf_filename
, CURR_DIR
);
343 lstrcatA(inf_filename
, "\\");
344 lstrcatA(inf_filename
, "test.inf");
346 create_inf_file(inf_filename
);
348 hinf
= SetupOpenInfFileA(inf_filename
, NULL
, INF_STYLE_WIN4
, NULL
);
349 ok(hinf
!= INVALID_HANDLE_VALUE
, "could not open inf file\n");
353 ret
= SetupGetSourceInfoA(hinf
, 2, SRCINFO_PATH
, buffer
, sizeof(buffer
), &required
);
354 ok(ret
, "SetupGetSourceInfoA failed\n");
356 ok(required
== 1, "unexpected required size: %d\n", required
);
357 ok(!lstrcmpA("", buffer
), "unexpected result string: %s\n", buffer
);
362 ret
= SetupGetSourceInfoA(hinf
, 2, SRCINFO_TAGFILE
, buffer
, sizeof(buffer
), &required
);
363 ok(ret
, "SetupGetSourceInfoA failed\n");
365 ok(required
== 28, "unexpected required size: %d\n", required
);
366 ok(!lstrcmpA("LANCOM\\LANtools\\lanconf.cab", buffer
), "unexpected result string: %s\n", buffer
);
371 ret
= SetupGetSourceInfoA(hinf
, 2, SRCINFO_DESCRIPTION
, buffer
, sizeof(buffer
), &required
);
372 ok(ret
, "SetupGetSourceInfoA failed\n");
374 ok(required
== 19, "unexpected required size: %d\n", required
);
375 ok(!lstrcmpA("LANCOM Software CD", buffer
), "unexpected result string: %s\n", buffer
);
377 SetupCloseInfFile(hinf
);
378 DeleteFileA(inf_filename
);
381 static void test_SetupGetTargetPath(void)
383 char buffer
[MAX_PATH
], inf_filename
[MAX_PATH
];
389 lstrcpyA(inf_filename
, CURR_DIR
);
390 lstrcatA(inf_filename
, "\\");
391 lstrcatA(inf_filename
, "test.inf");
393 create_inf_file(inf_filename
);
395 hinf
= SetupOpenInfFileA(inf_filename
, NULL
, INF_STYLE_WIN4
, NULL
);
396 ok(hinf
!= INVALID_HANDLE_VALUE
, "could not open inf file\n");
399 ctx
.CurrentInf
= hinf
;
405 ret
= SetupGetTargetPathA(hinf
, &ctx
, NULL
, buffer
, sizeof(buffer
), &required
);
406 ok(ret
, "SetupGetTargetPathA failed\n");
408 ok(required
== 10, "unexpected required size: %d\n", required
);
409 ok(!lstrcmpiA("C:\\LANCOM", buffer
), "unexpected result string: %s\n", buffer
);
411 SetupCloseInfFile(hinf
);
412 DeleteFileA(inf_filename
);
419 test_SetupGetInfInformation();
420 test_SetupGetSourceFileLocation();
421 test_SetupGetSourceInfo();
422 test_SetupGetTargetPath();