1 /*
2  * Unit tests for advpack.dll
3  *
4  * Copyright (C) 2005 Robert Reif
5  * Copyright (C) 2005 Sami Aario
6  *
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.
11  *
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.
16  *
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
20  */
21 
22 #include <stdio.h>
23 #include <stdarg.h>
24 #include <windows.h>
25 #include <advpub.h>
26 #include <assert.h>
27 #include "wine/test.h"
28 
29 /* defines for the TranslateInfString/Ex tests */
30 #define TEST_STRING1 "\\Application Name"
31 #define TEST_STRING2 "%49001%\\Application Name"
32 
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)
37 
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);
46 
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;
52 
53 static void get_progfiles_dir(void)
54 {
55     HKEY hkey;
56     DWORD size = MAX_PATH;
57 
58     RegOpenKeyA(HKEY_LOCAL_MACHINE, "Software\\Microsoft\\Windows\\CurrentVersion", &hkey);
59     RegQueryValueExA(hkey, "ProgramFilesDir", NULL, NULL, (LPBYTE)PROG_FILES_ROOT, &size);
60     RegCloseKey(hkey);
61 
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;
66 }
67 
68 static BOOL init_function_pointers(void)
69 {
70     hAdvPack = LoadLibraryA("advpack.dll");
71 
72     if (!hAdvPack)
73         return FALSE;
74 
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");
82 
83     if (!pCloseINFEngine || !pDelNode || !pGetVersionFromFile ||
84         !pOpenINFEngine || !pSetPerUserSecValues || !pTranslateInfString)
85     {
86         win_skip("Needed functions are not available\n");
87         FreeLibrary(hAdvPack);
88         return FALSE;
89     }
90 
91     return TRUE;
92 }
93 
94 static void version_test(void)
95 {
96     HRESULT hr;
97     DWORD major, minor;
98 
99     major = minor = 0;
100     hr = pGetVersionFromFile("kernel32.dll", &major, &minor, FALSE);
101     ok (hr == S_OK, "GetVersionFromFileEx(kernel32.dll) failed, returned "
102         "0x%08x\n", hr);
103     trace("kernel32.dll Language ID: 0x%08x, Codepage ID: 0x%08x\n",
104            major, minor);
105 
106     major = minor = 0;
107     hr = pGetVersionFromFile("kernel32.dll", &major, &minor, TRUE);
108     ok (hr == S_OK, "GetVersionFromFileEx(kernel32.dll) failed, returned "
109         "0x%08x\n", hr);
110     trace("kernel32.dll version: %d.%d.%d.%d\n", HIWORD(major), LOWORD(major),
111           HIWORD(minor), LOWORD(minor));
112 
113     major = minor = 0;
114     hr = pGetVersionFromFile("advpack.dll", &major, &minor, FALSE);
115     ok (hr == S_OK, "GetVersionFromFileEx(advpack.dll) failed, returned "
116         "0x%08x\n", hr);
117     trace("advpack.dll Language ID: 0x%08x, Codepage ID: 0x%08x\n",
118            major, minor);
119 
120     major = minor = 0;
121     hr = pGetVersionFromFile("advpack.dll", &major, &minor, TRUE);
122     ok (hr == S_OK, "GetVersionFromFileEx(advpack.dll) failed, returned "
123         "0x%08x\n", hr);
124     trace("advpack.dll version: %d.%d.%d.%d\n", HIWORD(major), LOWORD(major),
125           HIWORD(minor), LOWORD(minor));
126 }
127 
128 static void delnode_test(void)
129 {
130     HRESULT hr;
131     HANDLE hn;
132     CHAR currDir[MAX_PATH];
133     UINT currDirLen;
134 
135     /* Native DelNode apparently does not support relative paths, so we use
136        absolute paths for testing */
137     currDirLen = GetCurrentDirectoryA(ARRAY_SIZE(currDir), currDir);
138     assert(currDirLen > 0 && currDirLen < ARRAY_SIZE(currDir));
139 
140     if(currDir[currDirLen - 1] == '\\')
141         currDir[--currDirLen] = 0;
142 
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");
148 
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);
153     CloseHandle(hn);
154     hr = pDelNode(lstrcatA(currDir, "\\DelNodeTestFile1"), 0);
155     ok (hr == S_OK, "DelNode failed deleting a single file\n");
156     currDir[currDirLen] = '\0';
157 
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';
163 
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);
169     CloseHandle(hn);
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';
173 
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);
179     CloseHandle(hn);
180     hn = CreateFileA("DelNodeTestDir\\DelNodeTestFile2", GENERIC_WRITE, 0, NULL,
181         CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
182     assert(hn != INVALID_HANDLE_VALUE);
183     CloseHandle(hn);
184     hn = CreateFileA("DelNodeTestDir\\DelNodeTestFile3", GENERIC_WRITE, 0, NULL,
185         CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
186     assert(hn != INVALID_HANDLE_VALUE);
187     CloseHandle(hn);
188     hr = pDelNode(lstrcatA(currDir, "\\DelNodeTestDir"), 0);
189     ok (hr == S_OK, "DelNode failed deleting a directory containing multiple files\n");
190 }
191 
192 static void append_str(char **str, const char *data, ...)
193 {
194     va_list valist;
195 
196     va_start(valist, data);
197     vsprintf(*str, data, valist);
198     *str += strlen(*str);
199     va_end(valist);
200 }
201 
202 static void create_inf_file(void)
203 {
204     char data[1024];
205     char *ptr = data;
206     DWORD dwNumberOfBytesWritten;
207     HANDLE hf = CreateFileA(inf_file, GENERIC_WRITE, 0, NULL,
208                             CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
209 
210     append_str(&ptr, "[Version]\n");
211     append_str(&ptr, "Signature=\"$Chicago$\"\n");
212     append_str(&ptr, "[CustInstDestSection]\n");
213     append_str(&ptr, "49001=ProgramFilesDir\n");
214     append_str(&ptr, "49010=DestA,1\n");
215     append_str(&ptr, "49020=DestB\n");
216     append_str(&ptr, "49030=DestC\n");
217     append_str(&ptr, "[ProgramFilesDir]\n");
218     append_str(&ptr, "HKLM,\"Software\\Microsoft\\Windows\\CurrentVersion\",");
219     append_str(&ptr, "\"ProgramFilesDir\",,\"%%24%%\\%%LProgramF%%\"\n");
220     append_str(&ptr, "[section]\n");
221     append_str(&ptr, "NotACustomDestination=Version\n");
222     append_str(&ptr, "CustomDestination=CustInstDestSection\n");
223     append_str(&ptr, "[Options.NTx86]\n");
224     append_str(&ptr, "49001=ProgramFilesDir\n");
225     append_str(&ptr, "InstallDir=%%49001%%\\%%DefaultAppPath%%\n");
226     append_str(&ptr, "Result1=%%49010%%\n");
227     append_str(&ptr, "Result2=%%49020%%\n");
228     append_str(&ptr, "Result3=%%49030%%\n");
229     append_str(&ptr, "CustomHDestination=CustInstDestSection\n");
230     append_str(&ptr, "[Strings]\n");
231     append_str(&ptr, "DefaultAppPath=\"Application Name\"\n");
232     append_str(&ptr, "LProgramF=\"%s\"\n", PROG_FILES);
233     append_str(&ptr, "[DestA]\n");
234     append_str(&ptr, "HKLM,\"Software\\Garbage\",\"ProgramFilesDir\",,'%%24%%\\%%LProgramF%%'\n");
235     append_str(&ptr, "[DestB]\n");
236     append_str(&ptr, "'HKLM','Software\\Microsoft\\Windows\\CurrentVersion',");
237     append_str(&ptr, "'ProgramFilesDir',,\"%%24%%\"\n");
238     append_str(&ptr, "[DestC]\n");
239     append_str(&ptr, "HKLM,\"Software\\Garbage\",\"ProgramFilesDir\",,'%%24%%'\n");
240 
241     WriteFile(hf, data, ptr - data, &dwNumberOfBytesWritten, NULL);
242     CloseHandle(hf);
243 }
244 
245 static void translateinfstring_test(void)
246 {
247     HRESULT hr;
248     char buffer[MAX_PATH];
249     DWORD dwSize;
250 
251     create_inf_file();
252 
253     /* pass in a couple invalid parameters */
254     hr = pTranslateInfString(NULL, NULL, NULL, NULL, buffer, MAX_PATH, &dwSize, NULL);
255     ok(hr == E_INVALIDARG, "Expected E_INVALIDARG, got 0x%08x\n", (UINT)hr);
256 
257     /* try to open an inf file that doesn't exist */
258     hr = pTranslateInfString("c:\\a.inf", "Options.NTx86", "Options.NTx86",
259                              "InstallDir", buffer, MAX_PATH, &dwSize, NULL);
260     ok(hr == HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND) || hr == E_INVALIDARG ||
261        hr == HRESULT_FROM_WIN32(ERROR_MOD_NOT_FOUND),
262        "Expected E_INVALIDARG, 0x80070002 or 0x8007007e, got 0x%08x\n", (UINT)hr);
263 
264     if(hr == HRESULT_FROM_WIN32(ERROR_MOD_NOT_FOUND))
265     {
266         win_skip("WinNT 3.51 detected. Skipping tests for TranslateInfString()\n");
267         return;
268     }
269 
270     /* try a nonexistent section */
271     buffer[0] = 0;
272     hr = pTranslateInfString(inf_file, "idontexist", "Options.NTx86",
273                              "InstallDir", buffer, MAX_PATH, &dwSize, NULL);
274     if (hr == E_ACCESSDENIED)
275     {
276         skip("TranslateInfString is broken\n");
277         return;
278     }
279     ok(hr == S_OK, "Expected S_OK, got 0x%08x\n", (UINT)hr);
280     ok(!strcmp(buffer, TEST_STRING2), "Expected %s, got %s\n", TEST_STRING2, buffer);
281     ok(dwSize == 25, "Expected size 25, got %d\n", dwSize);
282 
283     buffer[0] = 0;
284     /* try other nonexistent section */
285     hr = pTranslateInfString(inf_file, "Options.NTx86", "idontexist",
286                              "InstallDir", buffer, MAX_PATH, &dwSize, NULL);
287     ok(hr == SPAPI_E_LINE_NOT_FOUND || hr == E_INVALIDARG,
288        "Expected SPAPI_E_LINE_NOT_FOUND or E_INVALIDARG, got 0x%08x\n", (UINT)hr);
289 
290     buffer[0] = 0;
291     /* try nonexistent key */
292     hr = pTranslateInfString(inf_file, "Options.NTx86", "Options.NTx86",
293                              "notvalid", buffer, MAX_PATH, &dwSize, NULL);
294     ok(hr == SPAPI_E_LINE_NOT_FOUND || hr == E_INVALIDARG,
295        "Expected SPAPI_E_LINE_NOT_FOUND or E_INVALIDARG, got 0x%08x\n", (UINT)hr);
296 
297     buffer[0] = 0;
298     /* test the behavior of pszInstallSection */
299     hr = pTranslateInfString(inf_file, "section", "Options.NTx86",
300                              "InstallDir", buffer, MAX_PATH, &dwSize, NULL);
301     ok(hr == ERROR_SUCCESS || hr == E_FAIL,
302        "Expected ERROR_SUCCESS or E_FAIL, got 0x%08x\n", (UINT)hr);
303 
304     if(hr == ERROR_SUCCESS)
305     {
306         ok(!strcmp(buffer, APP_PATH), "Expected '%s', got '%s'\n", APP_PATH, buffer);
307         ok(dwSize == APP_PATH_LEN, "Expected size %d, got %d\n", APP_PATH_LEN, dwSize);
308     }
309 
310     buffer[0] = 0;
311     /* try without a pszInstallSection */
312     hr = pTranslateInfString(inf_file, NULL, "Options.NTx86",
313                              "InstallDir", buffer, MAX_PATH, &dwSize, NULL);
314     ok(hr == S_OK, "Expected S_OK, got 0x%08x\n", (UINT)hr);
315     todo_wine
316     {
317         ok(!strcmp(buffer, TEST_STRING2), "Expected %s, got %s\n", TEST_STRING2, buffer);
318         ok(dwSize == 25, "Expected size 25, got %d\n", dwSize);
319     }
320 
321     DeleteFileA("c:\\a.inf");
322     DeleteFileA(inf_file);
323 }
324 
325 static void translateinfstringex_test(void)
326 {
327     HINF hinf;
328     HRESULT hr;
329     char buffer[MAX_PATH];
330     DWORD size = MAX_PATH;
331 
332     hr = pOpenINFEngine(inf_file, NULL, 0, &hinf, NULL);
333     if (hr == E_UNEXPECTED)
334     {
335         win_skip("Skipping tests on win9x because of brokenness\n");
336         return;
337     }
338 
339     create_inf_file();
340 
341     /* need to see if there are any flags */
342 
343     /* try a NULL filename */
344     hr = pOpenINFEngine(NULL, "Options.NTx86", 0, &hinf, NULL);
345     ok(hr == E_INVALIDARG, "Expected E_INVALIDARG, got %08x\n", hr);
346 
347     /* try an empty filename */
348     hr = pOpenINFEngine("", "Options.NTx86", 0, &hinf, NULL);
349     ok(hr == HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND) /* NT+ */ ||
350        hr == HRESULT_FROM_WIN32(E_UNEXPECTED) /* 9x */,
351         "Expected HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND or E_UNEXPECTED), got %08x\n", hr);
352 
353     /* try a NULL hinf */
354     hr = pOpenINFEngine(inf_file, "Options.NTx86", 0, NULL, NULL);
355     ok(hr == E_INVALIDARG, "Expected E_INVALIDARG, got %08x\n", hr);
356 
357     /* open the INF without the Install section specified */
358     hr = pOpenINFEngine(inf_file, NULL, 0, &hinf, NULL);
359     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
360 
361     /* try a NULL hinf */
362     hr = pTranslateInfStringEx(NULL, inf_file, "Options.NTx86", "InstallDir",
363                               buffer, size, &size, NULL);
364     ok(hr == E_INVALIDARG, "Expected E_INVALIDARG, got %08x\n", hr);
365 
366     /* try a NULL filename */
367     hr = pTranslateInfStringEx(hinf, NULL, "Options.NTx86", "InstallDir",
368                               buffer, size, &size, NULL);
369     ok(hr == E_INVALIDARG, "Expected E_INVALIDARG, got %08x\n", hr);
370 
371     /* try an empty filename */
372     memset(buffer, 'a', 25);
373     buffer[24] = '\0';
374     size = MAX_PATH;
375     hr = pTranslateInfStringEx(hinf, "", "Options.NTx86", "InstallDir",
376                               buffer, size, &size, NULL);
377     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
378     todo_wine
379     {
380         ok(!strcmp(buffer, TEST_STRING2), "Expected %s, got %s\n", TEST_STRING2, buffer);
381         ok(size == 25, "Expected size 25, got %d\n", size);
382     }
383 
384     /* try a NULL translate section */
385     hr = pTranslateInfStringEx(hinf, inf_file, NULL, "InstallDir",
386                               buffer, size, &size, NULL);
387     ok(hr == E_INVALIDARG, "Expected E_INVALIDARG, got %08x\n", hr);
388 
389     /* try an empty translate section */
390     hr = pTranslateInfStringEx(hinf, inf_file, "", "InstallDir",
391                               buffer, size, &size, NULL);
392     ok(hr == SPAPI_E_LINE_NOT_FOUND, "Expected SPAPI_E_LINE_NOT_FOUND, got %08x\n", hr);
393 
394     /* try a NULL translate key */
395     hr = pTranslateInfStringEx(hinf, inf_file, "Options.NTx86", NULL,
396                               buffer, size, &size, NULL);
397     ok(hr == E_INVALIDARG, "Expected E_INVALIDARG, got %08x\n", hr);
398 
399     /* try an empty translate key */
400     hr = pTranslateInfStringEx(hinf, inf_file, "Options.NTx86", "",
401                               buffer, size, &size, NULL);
402     ok(hr == SPAPI_E_LINE_NOT_FOUND, "Expected SPAPI_E_LINE_NOT_FOUND, got %08x\n", hr);
403 
404     /* successfully translate the string */
405     memset(buffer, 'a', 25);
406     buffer[24] = '\0';
407     size = MAX_PATH;
408     hr = pTranslateInfStringEx(hinf, inf_file, "Options.NTx86", "InstallDir",
409                               buffer, size, &size, NULL);
410     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
411     todo_wine
412     {
413         ok(!strcmp(buffer, TEST_STRING2), "Expected %s, got %s\n", TEST_STRING2, buffer);
414         ok(size == 25, "Expected size 25, got %d\n", size);
415     }
416 
417     /* try a NULL hinf */
418     hr = pCloseINFEngine(NULL);
419     ok(hr == E_INVALIDARG, "Expected E_INVALIDARG, got %08x\n", hr);
420 
421     /* successfully close the hinf */
422     hr = pCloseINFEngine(hinf);
423     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
424 
425     /* open the inf with the install section */
426     hr = pOpenINFEngine(inf_file, "section", 0, &hinf, NULL);
427     if (hr == E_FAIL)
428     {
429         skip("can't open engine with install section (needs admin rights)\n");
430         DeleteFileA(inf_file);
431         return;
432     }
433     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
434 
435     /* translate the string with the install section specified */
436     memset(buffer, 'a', APP_PATH_LEN);
437     buffer[APP_PATH_LEN - 1] = '\0';
438     size = MAX_PATH;
439     hr = pTranslateInfStringEx(hinf, inf_file, "Options.NTx86", "InstallDir",
440                               buffer, size, &size, NULL);
441     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
442     ok(!strcmp(buffer, APP_PATH), "Expected %s, got %s\n", APP_PATH, buffer);
443     ok(size == APP_PATH_LEN, "Expected size %d, got %d\n", APP_PATH_LEN, size);
444 
445     /* Single quote test (Note size includes null on return from call) */
446     memset(buffer, 'a', APP_PATH_LEN);
447     buffer[APP_PATH_LEN - 1] = '\0';
448     size = MAX_PATH;
449     hr = pTranslateInfStringEx(hinf, inf_file, "Options.NTx86", "Result1",
450                               buffer, size, &size, NULL);
451     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
452     ok(!lstrcmpiA(buffer, PROG_FILES_ROOT),
453            "Expected %s, got %s\n", PROG_FILES_ROOT, buffer);
454     ok(size == strlen(PROG_FILES_ROOT)+1, "Expected size %d, got %d\n",
455            lstrlenA(PROG_FILES_ROOT)+1, size);
456 
457     memset(buffer, 'a', APP_PATH_LEN);
458     buffer[APP_PATH_LEN - 1] = '\0';
459     size = MAX_PATH;
460     hr = pTranslateInfStringEx(hinf, inf_file, "Options.NTx86", "Result2",
461                               buffer, size, &size, NULL);
462     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
463     ok(!lstrcmpiA(buffer, PROG_FILES_ROOT),
464            "Expected %s, got %s\n", PROG_FILES_ROOT, buffer);
465     ok(size == strlen(PROG_FILES_ROOT)+1, "Expected size %d, got %d\n",
466            lstrlenA(PROG_FILES_ROOT)+1, size);
467 
468     {
469         char drive[MAX_PATH];
470         lstrcpyA(drive, PROG_FILES_ROOT);
471         drive[3] = 0x00; /* Just keep the system drive plus '\' */
472 
473         memset(buffer, 'a', APP_PATH_LEN);
474         buffer[APP_PATH_LEN - 1] = '\0';
475         size = MAX_PATH;
476         hr = pTranslateInfStringEx(hinf, inf_file, "Options.NTx86", "Result3",
477                                   buffer, size, &size, NULL);
478         ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
479         ok(!lstrcmpiA(buffer, drive),
480                "Expected %s, got %s\n", drive, buffer);
481         ok(size == strlen(drive)+1, "Expected size %d, got %d\n",
482                lstrlenA(drive)+1, size);
483     }
484 
485     /* close the INF again */
486     hr = pCloseINFEngine(hinf);
487     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
488 
489     DeleteFileA(inf_file);
490 
491     /* Create another .inf file which is just here to trigger a wine bug */
492     {
493         char data[1024];
494         char *ptr = data;
495         DWORD dwNumberOfBytesWritten;
496         HANDLE hf = CreateFileA(inf_file, GENERIC_WRITE, 0, NULL,
497                            CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
498 
499         append_str(&ptr, "[Version]\n");
500         append_str(&ptr, "Signature=\"$Chicago$\"\n");
501         append_str(&ptr, "[section]\n");
502         append_str(&ptr, "NotACustomDestination=Version\n");
503         append_str(&ptr, "CustomDestination=CustInstDestSection\n");
504         append_str(&ptr, "[CustInstDestSection]\n");
505         append_str(&ptr, "49010=DestA,1\n");
506         append_str(&ptr, "49020=DestB\n");
507         append_str(&ptr, "49030=DestC\n");
508         append_str(&ptr, "49040=DestD\n");
509         append_str(&ptr, "[Options.NTx86]\n");
510         append_str(&ptr, "Result2=%%49030%%\n");
511         append_str(&ptr, "[DestA]\n");
512         append_str(&ptr, "HKLM,\"Software\\Garbage\",\"ProgramFilesDir\",,'%%24%%'\n");
513         /* The point of this test is to have HKCU just before the quoted HKLM */
514         append_str(&ptr, "[DestB]\n");
515         append_str(&ptr, "HKCU,\"Software\\Garbage\",\"ProgramFilesDir\",,'%%24%%'\n");
516         append_str(&ptr, "[DestC]\n");
517         append_str(&ptr, "'HKLM','Software\\Microsoft\\Windows\\CurrentVersion',");
518         append_str(&ptr, "'ProgramFilesDir',,\"%%24%%\"\n");
519         append_str(&ptr, "[DestD]\n");
520         append_str(&ptr, "HKLM,\"Software\\Garbage\",\"ProgramFilesDir\",,'%%24%%'\n");
521 
522         WriteFile(hf, data, ptr - data, &dwNumberOfBytesWritten, NULL);
523         CloseHandle(hf);
524     }
525 
526     /* open the inf with the install section */
527     hr = pOpenINFEngine(inf_file, "section", 0, &hinf, NULL);
528     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
529 
530     /* Single quote test (Note size includes null on return from call) */
531     memset(buffer, 'a', APP_PATH_LEN);
532     buffer[APP_PATH_LEN - 1] = '\0';
533     size = MAX_PATH;
534     hr = pTranslateInfStringEx(hinf, inf_file, "Options.NTx86", "Result2",
535                               buffer, size, &size, NULL);
536     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
537     ok(!lstrcmpiA(buffer, PROG_FILES_ROOT),
538            "Expected %s, got %s\n", PROG_FILES_ROOT, buffer);
539     ok(size == strlen(PROG_FILES_ROOT)+1, "Expected size %d, got %d\n",
540            lstrlenA(PROG_FILES_ROOT)+1, size);
541 
542     /* close the INF again */
543     hr = pCloseINFEngine(hinf);
544     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
545 
546     DeleteFileA(inf_file);
547 }
548 
549 static BOOL check_reg_str(HKEY hkey, LPCSTR name, LPCSTR value)
550 {
551     DWORD size = MAX_PATH;
552     char check[MAX_PATH];
553 
554     if (RegQueryValueExA(hkey, name, NULL, NULL, (LPBYTE)check, &size))
555         return FALSE;
556 
557     return !lstrcmpA(check, value);
558 }
559 
560 static BOOL check_reg_dword(HKEY hkey, LPCSTR name, DWORD value)
561 {
562     DWORD size = sizeof(DWORD);
563     DWORD check;
564 
565     if (RegQueryValueExA(hkey, name, NULL, NULL, (LPBYTE)&check, &size))
566         return FALSE;
567 
568     return (check == value);
569 }
570 
571 static void setperusersecvalues_test(void)
572 {
573     PERUSERSECTIONA peruser;
574     HRESULT hr;
575     HKEY guid;
576 
577     lstrcpyA(peruser.szDispName, "displayname");
578     lstrcpyA(peruser.szLocale, "locale");
579     lstrcpyA(peruser.szStub, "stub");
580     lstrcpyA(peruser.szVersion, "1,1,1,1");
581     lstrcpyA(peruser.szCompID, "compid");
582     peruser.dwIsInstalled = 1;
583     peruser.bRollback = FALSE;
584 
585     /* try a NULL pPerUser */
586     if (0)
587     {
588         /* This crashes on systems with IE7 */
589         hr = pSetPerUserSecValues(NULL);
590         todo_wine
591         ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
592         ok(!OPEN_GUID_KEY(), "Expected guid key to not exist\n");
593     }
594 
595     /* at the very least, szGUID must be valid */
596     peruser.szGUID[0] = '\0';
597     hr = pSetPerUserSecValues(&peruser);
598     ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
599     ok(!OPEN_GUID_KEY(), "Expected guid key to not exist\n");
600 
601     /* set initial values */
602     lstrcpyA(peruser.szGUID, "guid");
603     hr = pSetPerUserSecValues(&peruser);
604     if (hr == E_FAIL)
605     {
606         skip("SetPerUserSecValues is broken\n");
607         return;
608     }
609     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
610     ok(OPEN_GUID_KEY(), "Expected guid key to exist\n");
611     ok(check_reg_str(guid, NULL, "displayname"), "Expected displayname\n");
612     ok(check_reg_str(guid, "ComponentID", "compid"), "Expected compid\n");
613     ok(check_reg_str(guid, "Locale", "locale"), "Expected locale\n");
614     ok(check_reg_str(guid, "StubPath", "stub"), "Expected stub\n");
615     ok(check_reg_str(guid, "Version", "1,1,1,1"), "Expected 1,1,1,1\n");
616     ok(check_reg_dword(guid, "IsInstalled", 1), "Expected 1\n");
617     ok(!REG_VAL_EXISTS(guid, "OldDisplayName"), "Expected OldDisplayName to not exist\n");
618     ok(!REG_VAL_EXISTS(guid, "OldLocale"), "Expected OldLocale to not exist\n");
619     ok(!REG_VAL_EXISTS(guid, "OldStubPath"), "Expected OldStubPath to not exist\n");
620     ok(!REG_VAL_EXISTS(guid, "OldVersion"), "Expected OldVersion to not exist\n");
621     ok(!REG_VAL_EXISTS(guid, "RealStubPath"), "Expected RealStubPath to not exist\n");
622 
623     /* raise the version, but bRollback is FALSE, so vals not saved */
624     lstrcpyA(peruser.szVersion, "2,1,1,1");
625     hr = pSetPerUserSecValues(&peruser);
626     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
627     ok(check_reg_str(guid, NULL, "displayname"), "Expected displayname\n");
628     ok(check_reg_str(guid, "ComponentID", "compid"), "Expected compid\n");
629     ok(check_reg_str(guid, "Locale", "locale"), "Expected locale\n");
630     ok(check_reg_str(guid, "StubPath", "stub"), "Expected stub\n");
631     ok(check_reg_str(guid, "Version", "2,1,1,1"), "Expected 2,1,1,1\n");
632     ok(check_reg_dword(guid, "IsInstalled", 1), "Expected 1\n");
633     ok(!REG_VAL_EXISTS(guid, "OldDisplayName"), "Expected OldDisplayName to not exist\n");
634     ok(!REG_VAL_EXISTS(guid, "OldLocale"), "Expected OldLocale to not exist\n");
635     ok(!REG_VAL_EXISTS(guid, "OldStubPath"), "Expected OldStubPath to not exist\n");
636     ok(!REG_VAL_EXISTS(guid, "OldVersion"), "Expected OldVersion to not exist\n");
637     ok(!REG_VAL_EXISTS(guid, "RealStubPath"), "Expected RealStubPath to not exist\n");
638 
639     /* raise the version again, bRollback is TRUE so vals are saved */
640     peruser.bRollback = TRUE;
641     lstrcpyA(peruser.szVersion, "3,1,1,1");
642     hr = pSetPerUserSecValues(&peruser);
643     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
644     ok(check_reg_str(guid, NULL, "displayname"), "Expected displayname\n");
645     ok(check_reg_str(guid, "ComponentID", "compid"), "Expected compid\n");
646     ok(check_reg_str(guid, "Locale", "locale"), "Expected locale\n");
647     ok(check_reg_dword(guid, "IsInstalled", 1), "Expected 1\n");
648     ok(check_reg_str(guid, "Version", "3,1,1,1"), "Expected 3,1,1,1\n");
649     todo_wine
650     {
651         ok(check_reg_str(guid, "OldDisplayName", "displayname"), "Expected displayname\n");
652         ok(check_reg_str(guid, "OldLocale", "locale"), "Expected locale\n");
653         ok(check_reg_str(guid, "RealStubPath", "stub"), "Expected stub\n");
654         ok(check_reg_str(guid, "OldStubPath", "stub"), "Expected stub\n");
655         ok(check_reg_str(guid, "OldVersion", "2,1,1,1"), "Expected 2,1,1,1\n");
656         ok(check_reg_str(guid, "StubPath",
657            "rundll32.exe advpack.dll,UserInstStubWrapper guid"),
658            "Expected real stub\n");
659     }
660 
661     RegDeleteKeyA(HKEY_LOCAL_MACHINE, GUID_KEY);
662 }
663 
664 START_TEST(advpack)
665 {
666     if (!init_function_pointers())
667         return;
668 
669     /* Make sure we create the temporary file in a directory
670      * where we have adequate rights
671      */
672     GetTempPathA(MAX_PATH, inf_file);
673     lstrcatA(inf_file,"test.inf");
674 
675     get_progfiles_dir();
676 
677     version_test();
678     delnode_test();
679     setperusersecvalues_test();
680     translateinfstring_test();
681     translateinfstringex_test();
682 
683     FreeLibrary(hAdvPack);
684 }
685