dlls/advpack/tests/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 #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 WINAPIV append_str(char **str, const char *data, ...)
 193 {
 194     __ms_va_list valist;
 195
 196     __ms_va_start(valist, data);
 197     vsprintf(*str, data, valist);
 198     *str += strlen(*str);
 199     __ms_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 }