dlls/advpack/tests/advpack.c - wine - Git at Google

 /*
  * Unit tests for advpack.dll
  *
  * Copyright (C) 2005 Robert Reif
  * Copyright (C) 2005 Sami Aario
  *
  * This library is free software; you can redistribute it and/or
  * modify it under the terms of the GNU Lesser General Public
  * License as published by the Free Software Foundation; either
  * version 2.1 of the License, or (at your option) any later version.
  *
  * This library is distributed in the hope that it will be useful,
  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  * Lesser General Public License for more details.
  *
  * You should have received a copy of the GNU Lesser General Public
  * License along with this library; if not, write to the Free Software
  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  */

 #include <stdio.h>
 #include <windows.h>
 #include <advpub.h>
 #include <assert.h>
 #include "wine/test.h"

 #define TEST_STRING1 "C:\\Program Files\\Application Name"
 #define TEST_STRING2 "%49001%\\Application Name"

 static HRESULT (WINAPI *pGetVersionFromFile)(LPSTR,LPDWORD,LPDWORD,BOOL);
 static HRESULT (WINAPI *pDelNode)(LPCSTR,DWORD);
 static HRESULT (WINAPI *pTranslateInfString)(LPSTR,LPSTR,LPSTR,LPSTR,LPSTR,DWORD,LPDWORD,LPVOID);

 static void version_test(void)
 {
     HRESULT hr;
     DWORD major, minor;

     major = minor = 0;
     hr = pGetVersionFromFile("kernel32.dll", &major, &minor, FALSE);
     ok (hr == S_OK, "GetVersionFromFileEx(kernel32.dll) failed, returned "
         "0x%08lx\n", hr);

     trace("kernel32.dll Language ID: 0x%08lx, Codepage ID: 0x%08lx\n",
            major, minor);

     major = minor = 0;
     hr = pGetVersionFromFile("kernel32.dll", &major, &minor, TRUE);
     ok (hr == S_OK, "GetVersionFromFileEx(kernel32.dll) failed, returned "
         "0x%08lx\n", hr);

     trace("kernel32.dll version: %d.%d.%d.%d\n", HIWORD(major), LOWORD(major),
           HIWORD(minor), LOWORD(minor));
 }

 static void delnode_test(void)
 {
     HRESULT hr;
     HANDLE hn;
     CHAR currDir[MAX_PATH];
     int currDirLen;

     /* Native DelNode apparently does not support relative paths, so we use
        absolute paths for testing */
     currDirLen = GetCurrentDirectoryA(sizeof(currDir) / sizeof(CHAR), currDir);
     assert(currDirLen > 0 && currDirLen < sizeof(currDir) / sizeof(CHAR));

     if(currDir[currDirLen - 1] == '\\')
         currDir[--currDirLen] = 0;

     /* Simple tests; these should fail. */
     hr = pDelNode(NULL, 0);
     ok (hr == E_FAIL, "DelNode called with NULL pathname should return E_FAIL\n");
     hr = pDelNode("", 0);
     ok (hr == E_FAIL, "DelNode called with empty pathname should return E_FAIL\n");

     /* Test deletion of a file. */
     hn = CreateFile("DelNodeTestFile1", GENERIC_WRITE, 0, NULL,
         CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
     assert(hn != INVALID_HANDLE_VALUE);
     CloseHandle(hn);
     hr = pDelNode(lstrcat(currDir, "\\DelNodeTestFile1"), 0);
     ok (hr == S_OK, "DelNode failed deleting a single file\n");
     currDir[currDirLen] = '\0';

     /* Test deletion of an empty directory. */
     CreateDirectoryA("DelNodeTestDir", NULL);
     hr = pDelNode(lstrcat(currDir, "\\DelNodeTestDir"), 0);
     ok (hr == S_OK, "DelNode failed deleting an empty directory\n");
     currDir[currDirLen] = '\0';

     /* Test deletion of a directory containing one file. */
     CreateDirectoryA("DelNodeTestDir", NULL);
     hn = CreateFile("DelNodeTestDir\\DelNodeTestFile1", GENERIC_WRITE, 0, NULL,
         CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
     assert(hn != INVALID_HANDLE_VALUE);
     CloseHandle(hn);
     hr = pDelNode(lstrcat(currDir, "\\DelNodeTestDir"), 0);
     ok (hr == S_OK, "DelNode failed deleting a directory containing one file\n");
     currDir[currDirLen] = '\0';

     /* Test deletion of a directory containing multiple files. */
     CreateDirectoryA("DelNodeTestDir", NULL);
     hn = CreateFile("DelNodeTestDir\\DelNodeTestFile1", GENERIC_WRITE, 0, NULL,
         CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
     assert(hn != INVALID_HANDLE_VALUE);
     CloseHandle(hn);
     hn = CreateFile("DelNodeTestDir\\DelNodeTestFile2", GENERIC_WRITE, 0, NULL,
         CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
     assert(hn != INVALID_HANDLE_VALUE);
     CloseHandle(hn);
     hn = CreateFile("DelNodeTestDir\\DelNodeTestFile3", GENERIC_WRITE, 0, NULL,
         CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
     assert(hn != INVALID_HANDLE_VALUE);
     CloseHandle(hn);
     hr = pDelNode(lstrcat(currDir, "\\DelNodeTestDir"), 0);
     ok (hr == S_OK, "DelNode failed deleting a directory containing multiple files\n");
     currDir[currDirLen] = '\0';
 }

 static void append_str(char **str, const char *data)
 {
     sprintf(*str, data);
     *str += strlen(*str);
 }

 static void create_inf_file()
 {
     char data[1024];
     char *ptr = data;
     DWORD dwNumberOfBytesWritten;
     HANDLE hf = CreateFile("c:\\test.inf", GENERIC_WRITE, 0, NULL,
                            CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);

     append_str(&ptr, "[Version]\n");
     append_str(&ptr, "Signature=\"$Chicago$\"\n");
     append_str(&ptr, "[CustInstDestSection]\n");
     append_str(&ptr, "49001=ProgramFilesDir\n");
     append_str(&ptr, "[ProgramFilesDir]\n");
     append_str(&ptr, "HKLM,\"Software\\Microsoft\\Windows\\CurrentVersion\",");
     append_str(&ptr, "\"ProgramFilesDir\",,\"%%24%%\\%%LProgramF%%\"\n");
     append_str(&ptr, "[section]\n");
     append_str(&ptr, "NotACustomDestination=Version\n");
     append_str(&ptr, "CustomDestination=CustInstDestSection\n");
     append_str(&ptr, "[Options.NTx86]\n");
     append_str(&ptr, "49001=ProgramFilesDir\n");
     append_str(&ptr, "InstallDir=%%49001%%\\%%DefaultAppPath%%\n");
     append_str(&ptr, "CustomHDestination=CustInstDestSection\n");
     append_str(&ptr, "[Strings]\n");
     append_str(&ptr, "DefaultAppPath=\"Application Name\"\n");
     append_str(&ptr, "LProgramF=\"Program Files\"\n");

     WriteFile(hf, data, ptr - data, &dwNumberOfBytesWritten, NULL);
     CloseHandle(hf);
 }

 static void translateinfstring_test()
 {
     HRESULT hr;
     char buffer[MAX_PATH];
     DWORD dwSize;

     create_inf_file();

     /* pass in a couple invalid parameters */
     hr = pTranslateInfString(NULL, NULL, NULL, NULL, buffer, MAX_PATH, &dwSize, NULL);
     ok(hr == E_INVALIDARG, "Expected E_INVALIDARG, got 0x%08x\n", (UINT)hr);

     /* try to open an inf file that doesn't exist */
     hr = pTranslateInfString("c:\\a.inf", "Options.NTx86", "Options.NTx86",
                              "InstallDir", buffer, MAX_PATH, &dwSize, NULL);
     ok(hr == HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND) || hr == E_INVALIDARG,
        "Expected 0x80070002 or E_INVALIDARG, got 0x%08x\n", (UINT)hr);

     /* try a nonexistent section */
     buffer[0] = 0;
     hr = pTranslateInfString("c:\\test.inf", "idontexist", "Options.NTx86",
                              "InstallDir", buffer, MAX_PATH, &dwSize, NULL);
     ok(hr == S_OK, "Expected S_OK, got 0x%08x\n", (UINT)hr);
     ok(!strcmp(buffer, TEST_STRING2), "Expected %s, got %s\n", TEST_STRING2, buffer);
     ok(dwSize == 25, "Expected size 25, got %ld\n", dwSize);

     buffer[0] = 0;
     /* try other nonexistent section */
     hr = pTranslateInfString("c:\\test.inf", "Options.NTx86", "idontexist",
                              "InstallDir", buffer, MAX_PATH, &dwSize, NULL);
     ok(hr == SPAPI_E_LINE_NOT_FOUND || hr == E_INVALIDARG,
        "Expected SPAPI_E_LINE_NOT_FOUND or E_INVALIDARG, got 0x%08x\n", (UINT)hr);

     buffer[0] = 0;
     /* try nonexistent key */
     hr = pTranslateInfString("c:\\test.inf", "Options.NTx86", "Options.NTx86",
                              "notvalid", buffer, MAX_PATH, &dwSize, NULL);
     ok(hr == SPAPI_E_LINE_NOT_FOUND || hr == E_INVALIDARG,
        "Expected SPAPI_E_LINE_NOT_FOUND or E_INVALIDARG, got 0x%08x\n", (UINT)hr);

     buffer[0] = 0;
     /* test the behavior of pszInstallSection */
     hr = pTranslateInfString("c:\\test.inf", "section", "Options.NTx86",
                              "InstallDir", buffer, MAX_PATH, &dwSize, NULL);
     ok(hr == ERROR_SUCCESS || hr == E_FAIL,
        "Expected ERROR_SUCCESS or E_FAIL, got 0x%08x\n", (UINT)hr);

     if(hr == ERROR_SUCCESS)
     todo_wine
     {
         ok(!strcmp(buffer, TEST_STRING1), "Expected %s, got %s\n", TEST_STRING1, buffer);
         ok(dwSize == 34, "Expected size 34, got %ld\n", dwSize);
     }

     buffer[0] = 0;
     /* try without a pszInstallSection */
     hr = pTranslateInfString("c:\\test.inf", NULL, "Options.NTx86",
                              "InstallDir", buffer, MAX_PATH, &dwSize, NULL);
     ok(hr == S_OK, "Expected S_OK, got 0x%08x\n", (UINT)hr);
     ok(!strcmp(buffer, TEST_STRING2), "Expected %s, got %s\n", TEST_STRING2, buffer);
     ok(dwSize == 25, "Expected size 25, got %ld\n", dwSize);

     DeleteFile("c:\\a.inf");
     DeleteFile("c:\\test.inf");
 }

 START_TEST(advpack)
 {
     HMODULE hdll;

     hdll = LoadLibraryA("advpack.dll");
     if (!hdll)
         return;

     pGetVersionFromFile = (void*)GetProcAddress(hdll, "GetVersionFromFile");
     pDelNode = (void*)GetProcAddress(hdll, "DelNode");
     pTranslateInfString = (void*)GetProcAddress(hdll, "TranslateInfString");
     if (!pGetVersionFromFile || !pDelNode || !pTranslateInfString)
         return;

     version_test();
     delnode_test();
     translateinfstring_test();

     FreeLibrary(hdll);
 }
	/*
	* Unit tests for advpack.dll
	*
	* Copyright (C) 2005 Robert Reif
	* Copyright (C) 2005 Sami Aario
	*
	* This library is free software; you can redistribute it and/or
	* modify it under the terms of the GNU Lesser General Public
	* License as published by the Free Software Foundation; either
	* version 2.1 of the License, or (at your option) any later version.
	*
	* This library is distributed in the hope that it will be useful,
	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
	* Lesser General Public License for more details.
	*
	* You should have received a copy of the GNU Lesser General Public
	* License along with this library; if not, write to the Free Software
	* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
	*/

	#include <stdio.h>
	#include <windows.h>
	#include <advpub.h>
	#include <assert.h>
	#include "wine/test.h"

	#define TEST_STRING1 "C:\\Program Files\\Application Name"
	#define TEST_STRING2 "%49001%\\Application Name"

	static HRESULT (WINAPI *pGetVersionFromFile)(LPSTR,LPDWORD,LPDWORD,BOOL);
	static HRESULT (WINAPI *pDelNode)(LPCSTR,DWORD);
	static HRESULT (WINAPI *pTranslateInfString)(LPSTR,LPSTR,LPSTR,LPSTR,LPSTR,DWORD,LPDWORD,LPVOID);

	static void version_test(void)
	{
	HRESULT hr;
	DWORD major, minor;

	major = minor = 0;
	hr = pGetVersionFromFile("kernel32.dll", &major, &minor, FALSE);
	ok (hr == S_OK, "GetVersionFromFileEx(kernel32.dll) failed, returned "
	"0x%08lx\n", hr);

	trace("kernel32.dll Language ID: 0x%08lx, Codepage ID: 0x%08lx\n",
	major, minor);

	major = minor = 0;
	hr = pGetVersionFromFile("kernel32.dll", &major, &minor, TRUE);
	ok (hr == S_OK, "GetVersionFromFileEx(kernel32.dll) failed, returned "
	"0x%08lx\n", hr);

	trace("kernel32.dll version: %d.%d.%d.%d\n", HIWORD(major), LOWORD(major),
	HIWORD(minor), LOWORD(minor));
	}

	static void delnode_test(void)
	{
	HRESULT hr;
	HANDLE hn;
	CHAR currDir[MAX_PATH];
	int currDirLen;

	/* Native DelNode apparently does not support relative paths, so we use
	absolute paths for testing */
	currDirLen = GetCurrentDirectoryA(sizeof(currDir) / sizeof(CHAR), currDir);
	assert(currDirLen > 0 && currDirLen < sizeof(currDir) / sizeof(CHAR));

	if(currDir[currDirLen - 1] == '\\')
	currDir[--currDirLen] = 0;

	/* Simple tests; these should fail. */
	hr = pDelNode(NULL, 0);
	ok (hr == E_FAIL, "DelNode called with NULL pathname should return E_FAIL\n");
	hr = pDelNode("", 0);
	ok (hr == E_FAIL, "DelNode called with empty pathname should return E_FAIL\n");

	/* Test deletion of a file. */
	hn = CreateFile("DelNodeTestFile1", GENERIC_WRITE, 0, NULL,
	CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
	assert(hn != INVALID_HANDLE_VALUE);
	CloseHandle(hn);
	hr = pDelNode(lstrcat(currDir, "\\DelNodeTestFile1"), 0);
	ok (hr == S_OK, "DelNode failed deleting a single file\n");
	currDir[currDirLen] = '\0';

	/* Test deletion of an empty directory. */
	CreateDirectoryA("DelNodeTestDir", NULL);
	hr = pDelNode(lstrcat(currDir, "\\DelNodeTestDir"), 0);
	ok (hr == S_OK, "DelNode failed deleting an empty directory\n");
	currDir[currDirLen] = '\0';

	/* Test deletion of a directory containing one file. */
	CreateDirectoryA("DelNodeTestDir", NULL);
	hn = CreateFile("DelNodeTestDir\\DelNodeTestFile1", GENERIC_WRITE, 0, NULL,
	CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
	assert(hn != INVALID_HANDLE_VALUE);
	CloseHandle(hn);
	hr = pDelNode(lstrcat(currDir, "\\DelNodeTestDir"), 0);
	ok (hr == S_OK, "DelNode failed deleting a directory containing one file\n");
	currDir[currDirLen] = '\0';

	/* Test deletion of a directory containing multiple files. */
	CreateDirectoryA("DelNodeTestDir", NULL);
	hn = CreateFile("DelNodeTestDir\\DelNodeTestFile1", GENERIC_WRITE, 0, NULL,
	CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
	assert(hn != INVALID_HANDLE_VALUE);
	CloseHandle(hn);
	hn = CreateFile("DelNodeTestDir\\DelNodeTestFile2", GENERIC_WRITE, 0, NULL,
	CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
	assert(hn != INVALID_HANDLE_VALUE);
	CloseHandle(hn);
	hn = CreateFile("DelNodeTestDir\\DelNodeTestFile3", GENERIC_WRITE, 0, NULL,
	CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
	assert(hn != INVALID_HANDLE_VALUE);
	CloseHandle(hn);
	hr = pDelNode(lstrcat(currDir, "\\DelNodeTestDir"), 0);
	ok (hr == S_OK, "DelNode failed deleting a directory containing multiple files\n");
	currDir[currDirLen] = '\0';
	}

	static void append_str(char *str, const char data)
	{
	sprintf(*str, data);
	str += strlen(str);
	}

	static void create_inf_file()
	{
	char data[1024];
	char *ptr = data;
	DWORD dwNumberOfBytesWritten;
	HANDLE hf = CreateFile("c:\\test.inf", GENERIC_WRITE, 0, NULL,
	CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);

	append_str(&ptr, "[Version]\n");
	append_str(&ptr, "Signature=\"$Chicago$\"\n");
	append_str(&ptr, "[CustInstDestSection]\n");
	append_str(&ptr, "49001=ProgramFilesDir\n");
	append_str(&ptr, "[ProgramFilesDir]\n");
	append_str(&ptr, "HKLM,\"Software\\Microsoft\\Windows\\CurrentVersion\",");
	append_str(&ptr, "\"ProgramFilesDir\",,\"%%24%%\\%%LProgramF%%\"\n");
	append_str(&ptr, "[section]\n");
	append_str(&ptr, "NotACustomDestination=Version\n");
	append_str(&ptr, "CustomDestination=CustInstDestSection\n");
	append_str(&ptr, "[Options.NTx86]\n");
	append_str(&ptr, "49001=ProgramFilesDir\n");
	append_str(&ptr, "InstallDir=%%49001%%\\%%DefaultAppPath%%\n");
	append_str(&ptr, "CustomHDestination=CustInstDestSection\n");
	append_str(&ptr, "[Strings]\n");
	append_str(&ptr, "DefaultAppPath=\"Application Name\"\n");
	append_str(&ptr, "LProgramF=\"Program Files\"\n");

	WriteFile(hf, data, ptr - data, &dwNumberOfBytesWritten, NULL);
	CloseHandle(hf);
	}

	static void translateinfstring_test()
	{
	HRESULT hr;
	char buffer[MAX_PATH];
	DWORD dwSize;

	create_inf_file();

	/* pass in a couple invalid parameters */
	hr = pTranslateInfString(NULL, NULL, NULL, NULL, buffer, MAX_PATH, &dwSize, NULL);
	ok(hr == E_INVALIDARG, "Expected E_INVALIDARG, got 0x%08x\n", (UINT)hr);

	/* try to open an inf file that doesn't exist */
	hr = pTranslateInfString("c:\\a.inf", "Options.NTx86", "Options.NTx86",
	"InstallDir", buffer, MAX_PATH, &dwSize, NULL);
	ok(hr == HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND) \|\| hr == E_INVALIDARG,
	"Expected 0x80070002 or E_INVALIDARG, got 0x%08x\n", (UINT)hr);

	/* try a nonexistent section */
	buffer[0] = 0;
	hr = pTranslateInfString("c:\\test.inf", "idontexist", "Options.NTx86",
	"InstallDir", buffer, MAX_PATH, &dwSize, NULL);
	ok(hr == S_OK, "Expected S_OK, got 0x%08x\n", (UINT)hr);
	ok(!strcmp(buffer, TEST_STRING2), "Expected %s, got %s\n", TEST_STRING2, buffer);
	ok(dwSize == 25, "Expected size 25, got %ld\n", dwSize);

	buffer[0] = 0;
	/* try other nonexistent section */
	hr = pTranslateInfString("c:\\test.inf", "Options.NTx86", "idontexist",
	"InstallDir", buffer, MAX_PATH, &dwSize, NULL);
	ok(hr == SPAPI_E_LINE_NOT_FOUND \|\| hr == E_INVALIDARG,
	"Expected SPAPI_E_LINE_NOT_FOUND or E_INVALIDARG, got 0x%08x\n", (UINT)hr);

	buffer[0] = 0;
	/* try nonexistent key */
	hr = pTranslateInfString("c:\\test.inf", "Options.NTx86", "Options.NTx86",
	"notvalid", buffer, MAX_PATH, &dwSize, NULL);
	ok(hr == SPAPI_E_LINE_NOT_FOUND \|\| hr == E_INVALIDARG,
	"Expected SPAPI_E_LINE_NOT_FOUND or E_INVALIDARG, got 0x%08x\n", (UINT)hr);

	buffer[0] = 0;
	/* test the behavior of pszInstallSection */
	hr = pTranslateInfString("c:\\test.inf", "section", "Options.NTx86",
	"InstallDir", buffer, MAX_PATH, &dwSize, NULL);
	ok(hr == ERROR_SUCCESS \|\| hr == E_FAIL,
	"Expected ERROR_SUCCESS or E_FAIL, got 0x%08x\n", (UINT)hr);

	if(hr == ERROR_SUCCESS)
	todo_wine
	{
	ok(!strcmp(buffer, TEST_STRING1), "Expected %s, got %s\n", TEST_STRING1, buffer);
	ok(dwSize == 34, "Expected size 34, got %ld\n", dwSize);
	}

	buffer[0] = 0;
	/* try without a pszInstallSection */
	hr = pTranslateInfString("c:\\test.inf", NULL, "Options.NTx86",
	"InstallDir", buffer, MAX_PATH, &dwSize, NULL);
	ok(hr == S_OK, "Expected S_OK, got 0x%08x\n", (UINT)hr);
	ok(!strcmp(buffer, TEST_STRING2), "Expected %s, got %s\n", TEST_STRING2, buffer);
	ok(dwSize == 25, "Expected size 25, got %ld\n", dwSize);

	DeleteFile("c:\\a.inf");
	DeleteFile("c:\\test.inf");
	}

	START_TEST(advpack)
	{
	HMODULE hdll;

	hdll = LoadLibraryA("advpack.dll");
	if (!hdll)
	return;

	pGetVersionFromFile = (void*)GetProcAddress(hdll, "GetVersionFromFile");
	pDelNode = (void*)GetProcAddress(hdll, "DelNode");
	pTranslateInfString = (void*)GetProcAddress(hdll, "TranslateInfString");
	if (!pGetVersionFromFile \|\| !pDelNode \|\| !pTranslateInfString)
	return;

	version_test();
	delnode_test();
	translateinfstring_test();

	FreeLibrary(hdll);
	}