rostests/winetests/advpack/advpack.c

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