2 * Unit tests for advpack.dll
4 * Copyright (C) 2005 Robert Reif
5 * Copyright (C) 2005 Sami Aario
7 * This library is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
12 * This library is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with this library; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
27 #include "wine/test.h"
29 /* defines for the TranslateInfString/Ex tests */
30 #define TEST_STRING1 "\\Application Name"
31 #define TEST_STRING2 "%49001%\\Application Name"
33 /* defines for the SetPerUserSecValues tests */
34 #define GUID_KEY "SOFTWARE\\Microsoft\\Active Setup\\Installed Components\\guid"
35 #define REG_VAL_EXISTS(key, value) !RegQueryValueExA(key, value, NULL, NULL, NULL, NULL)
36 #define OPEN_GUID_KEY() !RegOpenKeyA(HKEY_LOCAL_MACHINE, GUID_KEY, &guid)
38 static HMODULE hAdvPack
;
39 static HRESULT (WINAPI
*pCloseINFEngine
)(HINF
);
40 static HRESULT (WINAPI
*pDelNode
)(LPCSTR
,DWORD
);
41 static HRESULT (WINAPI
*pGetVersionFromFile
)(LPCSTR
,LPDWORD
,LPDWORD
,BOOL
);
42 static HRESULT (WINAPI
*pOpenINFEngine
)(PCSTR
,PCSTR
,DWORD
,HINF
*,PVOID
);
43 static HRESULT (WINAPI
*pSetPerUserSecValues
)(PPERUSERSECTIONA pPerUser
);
44 static HRESULT (WINAPI
*pTranslateInfString
)(LPCSTR
,LPCSTR
,LPCSTR
,LPCSTR
,LPSTR
,DWORD
,LPDWORD
,LPVOID
);
45 static HRESULT (WINAPI
*pTranslateInfStringEx
)(HINF
,PCSTR
,PCSTR
,PCSTR
,PSTR
,DWORD
,PDWORD
,PVOID
);
47 static CHAR inf_file
[MAX_PATH
];
48 static CHAR PROG_FILES_ROOT
[MAX_PATH
];
49 static CHAR PROG_FILES
[MAX_PATH
];
50 static CHAR APP_PATH
[MAX_PATH
];
51 static DWORD APP_PATH_LEN
;
53 static void get_progfiles_dir(void)
56 DWORD size
= MAX_PATH
;
58 RegOpenKeyA(HKEY_LOCAL_MACHINE
, "Software\\Microsoft\\Windows\\CurrentVersion", &hkey
);
59 RegQueryValueExA(hkey
, "ProgramFilesDir", NULL
, NULL
, (LPBYTE
)PROG_FILES_ROOT
, &size
);
62 lstrcpyA(PROG_FILES
, PROG_FILES_ROOT
+ 3); /* skip C:\ */
63 lstrcpyA(APP_PATH
, PROG_FILES_ROOT
);
64 lstrcatA(APP_PATH
, TEST_STRING1
);
65 APP_PATH_LEN
= lstrlenA(APP_PATH
) + 1;
68 static BOOL
init_function_pointers(void)
70 hAdvPack
= LoadLibraryA("advpack.dll");
75 pCloseINFEngine
= (void*)GetProcAddress(hAdvPack
, "CloseINFEngine");
76 pDelNode
= (void *)GetProcAddress(hAdvPack
, "DelNode");
77 pGetVersionFromFile
= (void *)GetProcAddress(hAdvPack
, "GetVersionFromFile");
78 pOpenINFEngine
= (void*)GetProcAddress(hAdvPack
, "OpenINFEngine");
79 pSetPerUserSecValues
= (void*)GetProcAddress(hAdvPack
, "SetPerUserSecValues");
80 pTranslateInfString
= (void *)GetProcAddress(hAdvPack
, "TranslateInfString");
81 pTranslateInfStringEx
= (void*)GetProcAddress(hAdvPack
, "TranslateInfStringEx");
83 if (!pCloseINFEngine
|| !pDelNode
|| !pGetVersionFromFile
||
84 !pOpenINFEngine
|| !pSetPerUserSecValues
|| !pTranslateInfString
)
86 win_skip("Needed functions are not available\n");
87 FreeLibrary(hAdvPack
);
94 static void version_test(void)
100 hr
= pGetVersionFromFile("kernel32.dll", &major
, &minor
, FALSE
);
101 ok (hr
== S_OK
, "GetVersionFromFileEx(kernel32.dll) failed, returned "
103 trace("kernel32.dll Language ID: 0x%08x, Codepage ID: 0x%08x\n",
107 hr
= pGetVersionFromFile("kernel32.dll", &major
, &minor
, TRUE
);
108 ok (hr
== S_OK
, "GetVersionFromFileEx(kernel32.dll) failed, returned "
110 trace("kernel32.dll version: %d.%d.%d.%d\n", HIWORD(major
), LOWORD(major
),
111 HIWORD(minor
), LOWORD(minor
));
114 hr
= pGetVersionFromFile("advpack.dll", &major
, &minor
, FALSE
);
115 ok (hr
== S_OK
, "GetVersionFromFileEx(advpack.dll) failed, returned "
117 trace("advpack.dll Language ID: 0x%08x, Codepage ID: 0x%08x\n",
121 hr
= pGetVersionFromFile("advpack.dll", &major
, &minor
, TRUE
);
122 ok (hr
== S_OK
, "GetVersionFromFileEx(advpack.dll) failed, returned "
124 trace("advpack.dll version: %d.%d.%d.%d\n", HIWORD(major
), LOWORD(major
),
125 HIWORD(minor
), LOWORD(minor
));
128 static void delnode_test(void)
132 CHAR currDir
[MAX_PATH
];
135 /* Native DelNode apparently does not support relative paths, so we use
136 absolute paths for testing */
137 currDirLen
= GetCurrentDirectoryA(sizeof(currDir
) / sizeof(CHAR
), currDir
);
138 assert(currDirLen
> 0 && currDirLen
< sizeof(currDir
) / sizeof(CHAR
));
140 if(currDir
[currDirLen
- 1] == '\\')
141 currDir
[--currDirLen
] = 0;
143 /* Simple tests; these should fail. */
144 hr
= pDelNode(NULL
, 0);
145 ok (hr
== E_FAIL
, "DelNode called with NULL pathname should return E_FAIL\n");
146 hr
= pDelNode("", 0);
147 ok (hr
== E_FAIL
, "DelNode called with empty pathname should return E_FAIL\n");
149 /* Test deletion of a file. */
150 hn
= CreateFileA("DelNodeTestFile1", GENERIC_WRITE
, 0, NULL
,
151 CREATE_ALWAYS
, FILE_ATTRIBUTE_NORMAL
, NULL
);
152 assert(hn
!= INVALID_HANDLE_VALUE
);
154 hr
= pDelNode(lstrcatA(currDir
, "\\DelNodeTestFile1"), 0);
155 ok (hr
== S_OK
, "DelNode failed deleting a single file\n");
156 currDir
[currDirLen
] = '\0';
158 /* Test deletion of an empty directory. */
159 CreateDirectoryA("DelNodeTestDir", NULL
);
160 hr
= pDelNode(lstrcatA(currDir
, "\\DelNodeTestDir"), 0);
161 ok (hr
== S_OK
, "DelNode failed deleting an empty directory\n");
162 currDir
[currDirLen
] = '\0';
164 /* Test deletion of a directory containing one file. */
165 CreateDirectoryA("DelNodeTestDir", NULL
);
166 hn
= CreateFileA("DelNodeTestDir\\DelNodeTestFile1", GENERIC_WRITE
, 0, NULL
,
167 CREATE_ALWAYS
, FILE_ATTRIBUTE_NORMAL
, NULL
);
168 assert(hn
!= INVALID_HANDLE_VALUE
);
170 hr
= pDelNode(lstrcatA(currDir
, "\\DelNodeTestDir"), 0);
171 ok (hr
== S_OK
, "DelNode failed deleting a directory containing one file\n");
172 currDir
[currDirLen
] = '\0';
174 /* Test deletion of a directory containing multiple files. */
175 CreateDirectoryA("DelNodeTestDir", NULL
);
176 hn
= CreateFileA("DelNodeTestDir\\DelNodeTestFile1", GENERIC_WRITE
, 0, NULL
,
177 CREATE_ALWAYS
, FILE_ATTRIBUTE_NORMAL
, NULL
);
178 assert(hn
!= INVALID_HANDLE_VALUE
);
180 hn
= CreateFileA("DelNodeTestDir\\DelNodeTestFile2", GENERIC_WRITE
, 0, NULL
,
181 CREATE_ALWAYS
, FILE_ATTRIBUTE_NORMAL
, NULL
);
182 assert(hn
!= INVALID_HANDLE_VALUE
);
184 hn
= CreateFileA("DelNodeTestDir\\DelNodeTestFile3", GENERIC_WRITE
, 0, NULL
,
185 CREATE_ALWAYS
, FILE_ATTRIBUTE_NORMAL
, NULL
);
186 assert(hn
!= INVALID_HANDLE_VALUE
);
188 hr
= pDelNode(lstrcatA(currDir
, "\\DelNodeTestDir"), 0);
189 ok (hr
== S_OK
, "DelNode failed deleting a directory containing multiple files\n");
190 currDir
[currDirLen
] = '\0';
193 static void append_str(char **str
, const char *data
, ...)
197 va_start(valist
, data
);
198 vsprintf(*str
, data
, valist
);
199 *str
+= strlen(*str
);
203 static void create_inf_file(void)
207 DWORD dwNumberOfBytesWritten
;
208 HANDLE hf
= CreateFileA(inf_file
, GENERIC_WRITE
, 0, NULL
,
209 CREATE_ALWAYS
, FILE_ATTRIBUTE_NORMAL
, NULL
);
211 append_str(&ptr
, "[Version]\n");
212 append_str(&ptr
, "Signature=\"$Chicago$\"\n");
213 append_str(&ptr
, "[CustInstDestSection]\n");
214 append_str(&ptr
, "49001=ProgramFilesDir\n");
215 append_str(&ptr
, "49010=DestA,1\n");
216 append_str(&ptr
, "49020=DestB\n");
217 append_str(&ptr
, "49030=DestC\n");
218 append_str(&ptr
, "[ProgramFilesDir]\n");
219 append_str(&ptr
, "HKLM,\"Software\\Microsoft\\Windows\\CurrentVersion\",");
220 append_str(&ptr
, "\"ProgramFilesDir\",,\"%%24%%\\%%LProgramF%%\"\n");
221 append_str(&ptr
, "[section]\n");
222 append_str(&ptr
, "NotACustomDestination=Version\n");
223 append_str(&ptr
, "CustomDestination=CustInstDestSection\n");
224 append_str(&ptr
, "[Options.NTx86]\n");
225 append_str(&ptr
, "49001=ProgramFilesDir\n");
226 append_str(&ptr
, "InstallDir=%%49001%%\\%%DefaultAppPath%%\n");
227 append_str(&ptr
, "Result1=%%49010%%\n");
228 append_str(&ptr
, "Result2=%%49020%%\n");
229 append_str(&ptr
, "Result3=%%49030%%\n");
230 append_str(&ptr
, "CustomHDestination=CustInstDestSection\n");
231 append_str(&ptr
, "[Strings]\n");
232 append_str(&ptr
, "DefaultAppPath=\"Application Name\"\n");
233 append_str(&ptr
, "LProgramF=\"%s\"\n", PROG_FILES
);
234 append_str(&ptr
, "[DestA]\n");
235 append_str(&ptr
, "HKLM,\"Software\\Garbage\",\"ProgramFilesDir\",,'%%24%%\\%%LProgramF%%'\n");
236 append_str(&ptr
, "[DestB]\n");
237 append_str(&ptr
, "'HKLM','Software\\Microsoft\\Windows\\CurrentVersion',");
238 append_str(&ptr
, "'ProgramFilesDir',,\"%%24%%\"\n");
239 append_str(&ptr
, "[DestC]\n");
240 append_str(&ptr
, "HKLM,\"Software\\Garbage\",\"ProgramFilesDir\",,'%%24%%'\n");
242 WriteFile(hf
, data
, ptr
- data
, &dwNumberOfBytesWritten
, NULL
);
246 static void translateinfstring_test(void)
249 char buffer
[MAX_PATH
];
254 /* pass in a couple invalid parameters */
255 hr
= pTranslateInfString(NULL
, NULL
, NULL
, NULL
, buffer
, MAX_PATH
, &dwSize
, NULL
);
256 ok(hr
== E_INVALIDARG
, "Expected E_INVALIDARG, got 0x%08x\n", (UINT
)hr
);
258 /* try to open an inf file that doesn't exist */
259 hr
= pTranslateInfString("c:\\a.inf", "Options.NTx86", "Options.NTx86",
260 "InstallDir", buffer
, MAX_PATH
, &dwSize
, NULL
);
261 ok(hr
== HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND
) || hr
== E_INVALIDARG
||
262 hr
== HRESULT_FROM_WIN32(ERROR_MOD_NOT_FOUND
),
263 "Expected E_INVALIDARG, 0x80070002 or 0x8007007e, got 0x%08x\n", (UINT
)hr
);
265 if(hr
== HRESULT_FROM_WIN32(ERROR_MOD_NOT_FOUND
))
267 win_skip("WinNT 3.51 detected. Skipping tests for TranslateInfString()\n");
271 /* try a nonexistent section */
273 hr
= pTranslateInfString(inf_file
, "idontexist", "Options.NTx86",
274 "InstallDir", buffer
, MAX_PATH
, &dwSize
, NULL
);
275 if (hr
== E_ACCESSDENIED
)
277 skip("TranslateInfString is broken\n");
280 ok(hr
== S_OK
, "Expected S_OK, got 0x%08x\n", (UINT
)hr
);
281 ok(!strcmp(buffer
, TEST_STRING2
), "Expected %s, got %s\n", TEST_STRING2
, buffer
);
282 ok(dwSize
== 25, "Expected size 25, got %d\n", dwSize
);
285 /* try other nonexistent section */
286 hr
= pTranslateInfString(inf_file
, "Options.NTx86", "idontexist",
287 "InstallDir", buffer
, MAX_PATH
, &dwSize
, NULL
);
288 ok(hr
== SPAPI_E_LINE_NOT_FOUND
|| hr
== E_INVALIDARG
,
289 "Expected SPAPI_E_LINE_NOT_FOUND or E_INVALIDARG, got 0x%08x\n", (UINT
)hr
);
292 /* try nonexistent key */
293 hr
= pTranslateInfString(inf_file
, "Options.NTx86", "Options.NTx86",
294 "notvalid", buffer
, MAX_PATH
, &dwSize
, NULL
);
295 ok(hr
== SPAPI_E_LINE_NOT_FOUND
|| hr
== E_INVALIDARG
,
296 "Expected SPAPI_E_LINE_NOT_FOUND or E_INVALIDARG, got 0x%08x\n", (UINT
)hr
);
299 /* test the behavior of pszInstallSection */
300 hr
= pTranslateInfString(inf_file
, "section", "Options.NTx86",
301 "InstallDir", buffer
, MAX_PATH
, &dwSize
, NULL
);
302 ok(hr
== ERROR_SUCCESS
|| hr
== E_FAIL
,
303 "Expected ERROR_SUCCESS or E_FAIL, got 0x%08x\n", (UINT
)hr
);
305 if(hr
== ERROR_SUCCESS
)
307 ok(!strcmp(buffer
, APP_PATH
), "Expected '%s', got '%s'\n", APP_PATH
, buffer
);
308 ok(dwSize
== APP_PATH_LEN
, "Expected size %d, got %d\n", APP_PATH_LEN
, dwSize
);
312 /* try without a pszInstallSection */
313 hr
= pTranslateInfString(inf_file
, NULL
, "Options.NTx86",
314 "InstallDir", buffer
, MAX_PATH
, &dwSize
, NULL
);
315 ok(hr
== S_OK
, "Expected S_OK, got 0x%08x\n", (UINT
)hr
);
318 ok(!strcmp(buffer
, TEST_STRING2
), "Expected %s, got %s\n", TEST_STRING2
, buffer
);
319 ok(dwSize
== 25, "Expected size 25, got %d\n", dwSize
);
322 DeleteFileA("c:\\a.inf");
323 DeleteFileA(inf_file
);
326 static void translateinfstringex_test(void)
330 char buffer
[MAX_PATH
];
331 DWORD size
= MAX_PATH
;
333 hr
= pOpenINFEngine(inf_file
, NULL
, 0, &hinf
, NULL
);
334 if (hr
== E_UNEXPECTED
)
336 win_skip("Skipping tests on win9x because of brokenness\n");
342 /* need to see if there are any flags */
344 /* try a NULL filename */
345 hr
= pOpenINFEngine(NULL
, "Options.NTx86", 0, &hinf
, NULL
);
346 ok(hr
== E_INVALIDARG
, "Expected E_INVALIDARG, got %08x\n", hr
);
348 /* try an empty filename */
349 hr
= pOpenINFEngine("", "Options.NTx86", 0, &hinf
, NULL
);
350 ok(hr
== HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND
) /* NT+ */ ||
351 hr
== HRESULT_FROM_WIN32(E_UNEXPECTED
) /* 9x */,
352 "Expected HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND or E_UNEXPECTED), got %08x\n", hr
);
354 /* try a NULL hinf */
355 hr
= pOpenINFEngine(inf_file
, "Options.NTx86", 0, NULL
, NULL
);
356 ok(hr
== E_INVALIDARG
, "Expected E_INVALIDARG, got %08x\n", hr
);
358 /* open the INF without the Install section specified */
359 hr
= pOpenINFEngine(inf_file
, NULL
, 0, &hinf
, NULL
);
360 ok(hr
== S_OK
, "Expected S_OK, got %08x\n", hr
);
362 /* try a NULL hinf */
363 hr
= pTranslateInfStringEx(NULL
, inf_file
, "Options.NTx86", "InstallDir",
364 buffer
, size
, &size
, NULL
);
365 ok(hr
== E_INVALIDARG
, "Expected E_INVALIDARG, got %08x\n", hr
);
367 /* try a NULL filename */
368 hr
= pTranslateInfStringEx(hinf
, NULL
, "Options.NTx86", "InstallDir",
369 buffer
, size
, &size
, NULL
);
370 ok(hr
== E_INVALIDARG
, "Expected E_INVALIDARG, got %08x\n", hr
);
372 /* try an empty filename */
373 memset(buffer
, 'a', 25);
376 hr
= pTranslateInfStringEx(hinf
, "", "Options.NTx86", "InstallDir",
377 buffer
, size
, &size
, NULL
);
378 ok(hr
== S_OK
, "Expected S_OK, got %08x\n", hr
);
381 ok(!strcmp(buffer
, TEST_STRING2
), "Expected %s, got %s\n", TEST_STRING2
, buffer
);
382 ok(size
== 25, "Expected size 25, got %d\n", size
);
385 /* try a NULL translate section */
386 hr
= pTranslateInfStringEx(hinf
, inf_file
, NULL
, "InstallDir",
387 buffer
, size
, &size
, NULL
);
388 ok(hr
== E_INVALIDARG
, "Expected E_INVALIDARG, got %08x\n", hr
);
390 /* try an empty translate section */
391 hr
= pTranslateInfStringEx(hinf
, inf_file
, "", "InstallDir",
392 buffer
, size
, &size
, NULL
);
393 ok(hr
== SPAPI_E_LINE_NOT_FOUND
, "Expected SPAPI_E_LINE_NOT_FOUND, got %08x\n", hr
);
395 /* try a NULL translate key */
396 hr
= pTranslateInfStringEx(hinf
, inf_file
, "Options.NTx86", NULL
,
397 buffer
, size
, &size
, NULL
);
398 ok(hr
== E_INVALIDARG
, "Expected E_INVALIDARG, got %08x\n", hr
);
400 /* try an empty translate key */
401 hr
= pTranslateInfStringEx(hinf
, inf_file
, "Options.NTx86", "",
402 buffer
, size
, &size
, NULL
);
403 ok(hr
== SPAPI_E_LINE_NOT_FOUND
, "Expected SPAPI_E_LINE_NOT_FOUND, got %08x\n", hr
);
405 /* successfully translate the string */
406 memset(buffer
, 'a', 25);
409 hr
= pTranslateInfStringEx(hinf
, inf_file
, "Options.NTx86", "InstallDir",
410 buffer
, size
, &size
, NULL
);
411 ok(hr
== S_OK
, "Expected S_OK, got %08x\n", hr
);
414 ok(!strcmp(buffer
, TEST_STRING2
), "Expected %s, got %s\n", TEST_STRING2
, buffer
);
415 ok(size
== 25, "Expected size 25, got %d\n", size
);
418 /* try a NULL hinf */
419 hr
= pCloseINFEngine(NULL
);
420 ok(hr
== E_INVALIDARG
, "Expected E_INVALIDARG, got %08x\n", hr
);
422 /* successfully close the hinf */
423 hr
= pCloseINFEngine(hinf
);
424 ok(hr
== S_OK
, "Expected S_OK, got %08x\n", hr
);
426 /* open the inf with the install section */
427 hr
= pOpenINFEngine(inf_file
, "section", 0, &hinf
, NULL
);
428 ok(hr
== S_OK
, "Expected S_OK, got %08x\n", hr
);
430 /* translate the string with the install section specified */
431 memset(buffer
, 'a', APP_PATH_LEN
);
432 buffer
[APP_PATH_LEN
- 1] = '\0';
434 hr
= pTranslateInfStringEx(hinf
, inf_file
, "Options.NTx86", "InstallDir",
435 buffer
, size
, &size
, NULL
);
436 ok(hr
== S_OK
, "Expected S_OK, got %08x\n", hr
);
437 ok(!strcmp(buffer
, APP_PATH
), "Expected %s, got %s\n", APP_PATH
, buffer
);
438 ok(size
== APP_PATH_LEN
, "Expected size %d, got %d\n", APP_PATH_LEN
, size
);
440 /* Single quote test (Note size includes null on return from call) */
441 memset(buffer
, 'a', APP_PATH_LEN
);
442 buffer
[APP_PATH_LEN
- 1] = '\0';
444 hr
= pTranslateInfStringEx(hinf
, inf_file
, "Options.NTx86", "Result1",
445 buffer
, size
, &size
, NULL
);
446 ok(hr
== S_OK
, "Expected S_OK, got %08x\n", hr
);
447 ok(!lstrcmpiA(buffer
, PROG_FILES_ROOT
),
448 "Expected %s, got %s\n", PROG_FILES_ROOT
, buffer
);
449 ok(size
== strlen(PROG_FILES_ROOT
)+1, "Expected size %d, got %d\n",
450 lstrlenA(PROG_FILES_ROOT
)+1, size
);
452 memset(buffer
, 'a', APP_PATH_LEN
);
453 buffer
[APP_PATH_LEN
- 1] = '\0';
455 hr
= pTranslateInfStringEx(hinf
, inf_file
, "Options.NTx86", "Result2",
456 buffer
, size
, &size
, NULL
);
457 ok(hr
== S_OK
, "Expected S_OK, got %08x\n", hr
);
458 ok(!lstrcmpiA(buffer
, PROG_FILES_ROOT
),
459 "Expected %s, got %s\n", PROG_FILES_ROOT
, buffer
);
460 ok(size
== strlen(PROG_FILES_ROOT
)+1, "Expected size %d, got %d\n",
461 lstrlenA(PROG_FILES_ROOT
)+1, size
);
464 char drive
[MAX_PATH
];
465 lstrcpyA(drive
, PROG_FILES_ROOT
);
466 drive
[3] = 0x00; /* Just keep the system drive plus '\' */
468 memset(buffer
, 'a', APP_PATH_LEN
);
469 buffer
[APP_PATH_LEN
- 1] = '\0';
471 hr
= pTranslateInfStringEx(hinf
, inf_file
, "Options.NTx86", "Result3",
472 buffer
, size
, &size
, NULL
);
473 ok(hr
== S_OK
, "Expected S_OK, got %08x\n", hr
);
474 ok(!lstrcmpiA(buffer
, drive
),
475 "Expected %s, got %s\n", drive
, buffer
);
476 ok(size
== strlen(drive
)+1, "Expected size %d, got %d\n",
477 lstrlenA(drive
)+1, size
);
480 /* close the INF again */
481 hr
= pCloseINFEngine(hinf
);
482 ok(hr
== S_OK
, "Expected S_OK, got %08x\n", hr
);
484 DeleteFileA(inf_file
);
486 /* Create another .inf file which is just here to trigger a wine bug */
490 DWORD dwNumberOfBytesWritten
;
491 HANDLE hf
= CreateFileA(inf_file
, GENERIC_WRITE
, 0, NULL
,
492 CREATE_ALWAYS
, FILE_ATTRIBUTE_NORMAL
, NULL
);
494 append_str(&ptr
, "[Version]\n");
495 append_str(&ptr
, "Signature=\"$Chicago$\"\n");
496 append_str(&ptr
, "[section]\n");
497 append_str(&ptr
, "NotACustomDestination=Version\n");
498 append_str(&ptr
, "CustomDestination=CustInstDestSection\n");
499 append_str(&ptr
, "[CustInstDestSection]\n");
500 append_str(&ptr
, "49010=DestA,1\n");
501 append_str(&ptr
, "49020=DestB\n");
502 append_str(&ptr
, "49030=DestC\n");
503 append_str(&ptr
, "49040=DestD\n");
504 append_str(&ptr
, "[Options.NTx86]\n");
505 append_str(&ptr
, "Result2=%%49030%%\n");
506 append_str(&ptr
, "[DestA]\n");
507 append_str(&ptr
, "HKLM,\"Software\\Garbage\",\"ProgramFilesDir\",,'%%24%%'\n");
508 /* The point of this test is to have HKCU just before the quoted HKLM */
509 append_str(&ptr
, "[DestB]\n");
510 append_str(&ptr
, "HKCU,\"Software\\Garbage\",\"ProgramFilesDir\",,'%%24%%'\n");
511 append_str(&ptr
, "[DestC]\n");
512 append_str(&ptr
, "'HKLM','Software\\Microsoft\\Windows\\CurrentVersion',");
513 append_str(&ptr
, "'ProgramFilesDir',,\"%%24%%\"\n");
514 append_str(&ptr
, "[DestD]\n");
515 append_str(&ptr
, "HKLM,\"Software\\Garbage\",\"ProgramFilesDir\",,'%%24%%'\n");
517 WriteFile(hf
, data
, ptr
- data
, &dwNumberOfBytesWritten
, NULL
);
521 /* open the inf with the install section */
522 hr
= pOpenINFEngine(inf_file
, "section", 0, &hinf
, NULL
);
523 ok(hr
== S_OK
, "Expected S_OK, got %08x\n", hr
);
525 /* Single quote test (Note size includes null on return from call) */
526 memset(buffer
, 'a', APP_PATH_LEN
);
527 buffer
[APP_PATH_LEN
- 1] = '\0';
529 hr
= pTranslateInfStringEx(hinf
, inf_file
, "Options.NTx86", "Result2",
530 buffer
, size
, &size
, NULL
);
531 ok(hr
== S_OK
, "Expected S_OK, got %08x\n", hr
);
532 ok(!lstrcmpiA(buffer
, PROG_FILES_ROOT
),
533 "Expected %s, got %s\n", PROG_FILES_ROOT
, buffer
);
534 ok(size
== strlen(PROG_FILES_ROOT
)+1, "Expected size %d, got %d\n",
535 lstrlenA(PROG_FILES_ROOT
)+1, size
);
537 /* close the INF again */
538 hr
= pCloseINFEngine(hinf
);
539 ok(hr
== S_OK
, "Expected S_OK, got %08x\n", hr
);
541 DeleteFileA(inf_file
);
544 static BOOL
check_reg_str(HKEY hkey
, LPCSTR name
, LPCSTR value
)
546 DWORD size
= MAX_PATH
;
547 char check
[MAX_PATH
];
549 if (RegQueryValueExA(hkey
, name
, NULL
, NULL
, (LPBYTE
)check
, &size
))
552 return !lstrcmpA(check
, value
);
555 static BOOL
check_reg_dword(HKEY hkey
, LPCSTR name
, DWORD value
)
557 DWORD size
= sizeof(DWORD
);
560 if (RegQueryValueExA(hkey
, name
, NULL
, NULL
, (LPBYTE
)&check
, &size
))
563 return (check
== value
);
566 static void setperusersecvalues_test(void)
568 PERUSERSECTIONA peruser
;
572 lstrcpyA(peruser
.szDispName
, "displayname");
573 lstrcpyA(peruser
.szLocale
, "locale");
574 lstrcpyA(peruser
.szStub
, "stub");
575 lstrcpyA(peruser
.szVersion
, "1,1,1,1");
576 lstrcpyA(peruser
.szCompID
, "compid");
577 peruser
.dwIsInstalled
= 1;
578 peruser
.bRollback
= FALSE
;
580 /* try a NULL pPerUser */
583 /* This crashes on systems with IE7 */
584 hr
= pSetPerUserSecValues(NULL
);
586 ok(hr
== S_OK
, "Expected S_OK, got %d\n", hr
);
587 ok(!OPEN_GUID_KEY(), "Expected guid key to not exist\n");
590 /* at the very least, szGUID must be valid */
591 peruser
.szGUID
[0] = '\0';
592 hr
= pSetPerUserSecValues(&peruser
);
593 ok(hr
== S_OK
, "Expected S_OK, got %d\n", hr
);
594 ok(!OPEN_GUID_KEY(), "Expected guid key to not exist\n");
596 /* set initial values */
597 lstrcpyA(peruser
.szGUID
, "guid");
598 hr
= pSetPerUserSecValues(&peruser
);
601 skip("SetPerUserSecValues is broken\n");
604 ok(hr
== S_OK
, "Expected S_OK, got %08x\n", hr
);
605 ok(OPEN_GUID_KEY(), "Expected guid key to exist\n");
606 ok(check_reg_str(guid
, NULL
, "displayname"), "Expected displayname\n");
607 ok(check_reg_str(guid
, "ComponentID", "compid"), "Expected compid\n");
608 ok(check_reg_str(guid
, "Locale", "locale"), "Expected locale\n");
609 ok(check_reg_str(guid
, "StubPath", "stub"), "Expected stub\n");
610 ok(check_reg_str(guid
, "Version", "1,1,1,1"), "Expected 1,1,1,1\n");
611 ok(check_reg_dword(guid
, "IsInstalled", 1), "Expected 1\n");
612 ok(!REG_VAL_EXISTS(guid
, "OldDisplayName"), "Expected OldDisplayName to not exist\n");
613 ok(!REG_VAL_EXISTS(guid
, "OldLocale"), "Expected OldLocale to not exist\n");
614 ok(!REG_VAL_EXISTS(guid
, "OldStubPath"), "Expected OldStubPath to not exist\n");
615 ok(!REG_VAL_EXISTS(guid
, "OldVersion"), "Expected OldVersion to not exist\n");
616 ok(!REG_VAL_EXISTS(guid
, "RealStubPath"), "Expected RealStubPath to not exist\n");
618 /* raise the version, but bRollback is FALSE, so vals not saved */
619 lstrcpyA(peruser
.szVersion
, "2,1,1,1");
620 hr
= pSetPerUserSecValues(&peruser
);
621 ok(hr
== S_OK
, "Expected S_OK, got %08x\n", hr
);
622 ok(check_reg_str(guid
, NULL
, "displayname"), "Expected displayname\n");
623 ok(check_reg_str(guid
, "ComponentID", "compid"), "Expected compid\n");
624 ok(check_reg_str(guid
, "Locale", "locale"), "Expected locale\n");
625 ok(check_reg_str(guid
, "StubPath", "stub"), "Expected stub\n");
626 ok(check_reg_str(guid
, "Version", "2,1,1,1"), "Expected 2,1,1,1\n");
627 ok(check_reg_dword(guid
, "IsInstalled", 1), "Expected 1\n");
628 ok(!REG_VAL_EXISTS(guid
, "OldDisplayName"), "Expected OldDisplayName to not exist\n");
629 ok(!REG_VAL_EXISTS(guid
, "OldLocale"), "Expected OldLocale to not exist\n");
630 ok(!REG_VAL_EXISTS(guid
, "OldStubPath"), "Expected OldStubPath to not exist\n");
631 ok(!REG_VAL_EXISTS(guid
, "OldVersion"), "Expected OldVersion to not exist\n");
632 ok(!REG_VAL_EXISTS(guid
, "RealStubPath"), "Expected RealStubPath to not exist\n");
634 /* raise the version again, bRollback is TRUE so vals are saved */
635 peruser
.bRollback
= TRUE
;
636 lstrcpyA(peruser
.szVersion
, "3,1,1,1");
637 hr
= pSetPerUserSecValues(&peruser
);
638 ok(hr
== S_OK
, "Expected S_OK, got %08x\n", hr
);
639 ok(check_reg_str(guid
, NULL
, "displayname"), "Expected displayname\n");
640 ok(check_reg_str(guid
, "ComponentID", "compid"), "Expected compid\n");
641 ok(check_reg_str(guid
, "Locale", "locale"), "Expected locale\n");
642 ok(check_reg_dword(guid
, "IsInstalled", 1), "Expected 1\n");
643 ok(check_reg_str(guid
, "Version", "3,1,1,1"), "Expected 3,1,1,1\n");
646 ok(check_reg_str(guid
, "OldDisplayName", "displayname"), "Expected displayname\n");
647 ok(check_reg_str(guid
, "OldLocale", "locale"), "Expected locale\n");
648 ok(check_reg_str(guid
, "RealStubPath", "stub"), "Expected stub\n");
649 ok(check_reg_str(guid
, "OldStubPath", "stub"), "Expected stub\n");
650 ok(check_reg_str(guid
, "OldVersion", "2,1,1,1"), "Expected 2,1,1,1\n");
651 ok(check_reg_str(guid
, "StubPath",
652 "rundll32.exe advpack.dll,UserInstStubWrapper guid"),
653 "Expected real stub\n");
656 RegDeleteKeyA(HKEY_LOCAL_MACHINE
, GUID_KEY
);
661 if (!init_function_pointers())
664 /* Make sure we create the temporary file in a directory
665 * were we have enough rights
667 GetTempPathA(MAX_PATH
, inf_file
);
668 lstrcatA(inf_file
,"test.inf");
674 setperusersecvalues_test();
675 translateinfstring_test();
676 translateinfstringex_test();
678 FreeLibrary(hAdvPack
);