dlls/setupapi/tests/query.c - wine - Git at Google

 /*
  * Unit tests for setupapi.dll query functions
  *
  * Copyright (C) 2006 James Hawkins
  *
  * 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 <setupapi.h>
 #include "wine/test.h"

 /* function pointers */
 static HMODULE hSetupAPI;
 static void (WINAPI *pSetupCloseInfFile)(HINF);
 static BOOL (WINAPI *pSetupGetInfInformationA)(LPCVOID,DWORD,PSP_INF_INFORMATION,DWORD,PDWORD);
 static HINF (WINAPI *pSetupOpenInfFileA)(PCSTR,PCSTR,DWORD,PUINT);
 static BOOL (WINAPI *pSetupQueryInfFileInformationA)(PSP_INF_INFORMATION,UINT,PSTR,DWORD,PDWORD);

 CHAR CURR_DIR[MAX_PATH];
 CHAR WIN_DIR[MAX_PATH];

 static void init_function_pointers(void)
 {
     hSetupAPI = LoadLibraryA("setupapi.dll");

     if (hSetupAPI)
     {
         pSetupCloseInfFile = (void *)GetProcAddress(hSetupAPI, "SetupCloseInfFile");
         pSetupGetInfInformationA = (void *)GetProcAddress(hSetupAPI, "SetupGetInfInformationA");
         pSetupOpenInfFileA = (void *)GetProcAddress(hSetupAPI, "SetupOpenInfFileA");
         pSetupQueryInfFileInformationA = (void *)GetProcAddress(hSetupAPI, "SetupQueryInfFileInformationA");
     }
 }

 static void get_directories(void)
 {
     int len;

     GetCurrentDirectoryA(MAX_PATH, CURR_DIR);
     len = lstrlenA(CURR_DIR);

     if(len && (CURR_DIR[len-1] == '\\'))
         CURR_DIR[len-1] = 0;

     GetWindowsDirectoryA(WIN_DIR, MAX_PATH);
     len = lstrlenA(WIN_DIR);

     if (len && (WIN_DIR[len-1] == '\\'))
         WIN_DIR[len-1] = 0;
 }

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

 static void create_inf_file(LPSTR filename)
 {
     char data[1024];
     char *ptr = data;
     DWORD dwNumberOfBytesWritten;
     HANDLE hf = CreateFile(filename, GENERIC_WRITE, 0, NULL,
                            CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);

     append_str(&ptr, "[Version]\n");
     append_str(&ptr, "Signature=\"$Chicago$\"\n");
     append_str(&ptr, "AdvancedINF=2.5\n");

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

 static BOOL check_info_filename(PSP_INF_INFORMATION info, LPSTR test)
 {
     LPSTR filename;
     DWORD size;
     BOOL ret = FALSE;

     if (!pSetupQueryInfFileInformationA(info, 0, NULL, 0, &size))
         return FALSE;

     filename = HeapAlloc(GetProcessHeap(), 0, size);
     if (!filename)
         return FALSE;

     pSetupQueryInfFileInformationA(info, 0, filename, size, &size);

     if (!lstrcmpiA(test, filename))
         ret = TRUE;

     HeapFree(GetProcessHeap(), 0, filename);
     return ret;
 }

 static PSP_INF_INFORMATION alloc_inf_info(LPSTR filename, DWORD search, PDWORD size)
 {
     PSP_INF_INFORMATION info;
     BOOL ret;

     ret = pSetupGetInfInformationA(filename, search, NULL, 0, size);
     if (!ret)
         return NULL;

     info = HeapAlloc(GetProcessHeap(), 0, *size);
     return info;
 }

 static void test_SetupGetInfInformation(void)
 {
     PSP_INF_INFORMATION info;
     CHAR inf_filename[MAX_PATH];
     CHAR inf_one[MAX_PATH], inf_two[MAX_PATH];
     DWORD size;
     HINF hinf;
     BOOL ret;

     lstrcpyA(inf_filename, CURR_DIR);
     lstrcatA(inf_filename, "\\");
     lstrcatA(inf_filename, "test.inf");

     /* try an invalid inf handle */
     size = 0xdeadbeef;
     SetLastError(0xbeefcafe);
     ret = pSetupGetInfInformationA(NULL, INFINFO_INF_SPEC_IS_HINF, NULL, 0, &size);
     ok(ret == FALSE, "Expected SetupGetInfInformation to fail\n");
     ok(GetLastError() == ERROR_INVALID_HANDLE,
        "Expected ERROR_INVALID_HANDLE, got %ld\n", GetLastError());
     ok(size == 0xdeadbeef, "Expected size to remain unchanged\n");

     /* try an invalid inf filename */
     size = 0xdeadbeef;
     SetLastError(0xbeefcafe);
     ret = pSetupGetInfInformationA(NULL, INFINFO_INF_NAME_IS_ABSOLUTE, NULL, 0, &size);
     ok(ret == FALSE, "Expected SetupGetInfInformation to fail\n");
     ok(GetLastError() == ERROR_INVALID_PARAMETER,
        "Expected ERROR_INVALID_PARAMETER, got %ld\n", GetLastError());
     ok(size == 0xdeadbeef, "Expected size to remain unchanged\n");

     create_inf_file(inf_filename);

     /* try an invalid search flag */
     size = 0xdeadbeef;
     SetLastError(0xbeefcafe);
     ret = pSetupGetInfInformationA(inf_filename, -1, NULL, 0, &size);
     ok(ret == FALSE, "Expected SetupGetInfInformation to fail\n");
     ok(GetLastError() == ERROR_INVALID_PARAMETER,
        "Expected ERROR_INVALID_PARAMETER, got %ld\n", GetLastError());
     ok(size == 0xdeadbeef, "Expected size to remain unchanged\n");

     /* try a nonexistent inf file */
     size = 0xdeadbeef;
     SetLastError(0xbeefcafe);
     ret = pSetupGetInfInformationA("idontexist", INFINFO_INF_NAME_IS_ABSOLUTE, NULL, 0, &size);
     ok(ret == FALSE, "Expected SetupGetInfInformation to fail\n");
     ok(GetLastError() == ERROR_FILE_NOT_FOUND,
        "Expected ERROR_FILE_NOT_FOUND, got %ld\n", GetLastError());
     ok(size == 0xdeadbeef, "Expected size to remain unchanged\n");

     /* successfully open the inf file */
     size = 0xdeadbeef;
     ret = pSetupGetInfInformationA(inf_filename, INFINFO_INF_NAME_IS_ABSOLUTE, NULL, 0, &size);
     ok(ret == TRUE, "Expected SetupGetInfInformation to succeed\n");
     ok(size != 0xdeadbeef, "Expected a valid size on return\n");

     /* set ReturnBuffer to NULL and ReturnBufferSize to non-zero */
     SetLastError(0xbeefcafe);
     ret = pSetupGetInfInformationA(inf_filename, INFINFO_INF_NAME_IS_ABSOLUTE, NULL, size, &size);
     ok(ret == FALSE, "Expected SetupGetInfInformation to fail\n");
     ok(GetLastError() == ERROR_INVALID_PARAMETER,
        "Expected ERROR_INVALID_PARAMETER, got %ld\n", GetLastError());

     info = HeapAlloc(GetProcessHeap(), 0, size);

     /* try valid ReturnBuffer but too small size */
     SetLastError(0xbeefcafe);
     ret = pSetupGetInfInformationA(inf_filename, INFINFO_INF_NAME_IS_ABSOLUTE, info, size - 1, &size);
     ok(ret == FALSE, "Expected SetupGetInfInformation to fail\n");
     ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER,
        "Expected ERROR_INSUFFICIENT_BUFFER, got %ld\n", GetLastError());

     /* successfully get the inf information */
     ret = pSetupGetInfInformationA(inf_filename, INFINFO_INF_NAME_IS_ABSOLUTE, info, size, &size);
     ok(ret == TRUE, "Expected SetupGetInfInformation to succeed\n");
     ok(check_info_filename(info, inf_filename), "Expected returned filename to be equal\n");

     HeapFree(GetProcessHeap(), 0, info);

     /* try the INFINFO_INF_SPEC_IS_HINF search flag */
     hinf = pSetupOpenInfFileA(inf_filename, NULL, INF_STYLE_WIN4, NULL);
     info = alloc_inf_info(hinf, INFINFO_INF_SPEC_IS_HINF, &size);
     ret = pSetupGetInfInformationA(hinf, INFINFO_INF_SPEC_IS_HINF, info, size, &size);
     ok(ret == TRUE, "Expected SetupGetInfInformation to succeed\n");
     ok(check_info_filename(info, inf_filename), "Expected returned filename to be equal\n");
     pSetupCloseInfFile(hinf);

     lstrcpyA(inf_one, WIN_DIR);
     lstrcatA(inf_one, "\\inf\\");
     lstrcatA(inf_one, "test.inf");
     create_inf_file(inf_one);

     lstrcpyA(inf_two, WIN_DIR);
     lstrcatA(inf_two, "\\system32\\");
     lstrcatA(inf_two, "test.inf");
     create_inf_file(inf_two);

     HeapFree(GetProcessHeap(), 0, info);
     info = alloc_inf_info("test.inf", INFINFO_DEFAULT_SEARCH, &size);

     /* test the INFINFO_DEFAULT_SEARCH search flag */
     ret = pSetupGetInfInformationA("test.inf", INFINFO_DEFAULT_SEARCH, info, size, &size);
     ok(ret == TRUE, "Expected SetupGetInfInformation to succeed\n");
     ok(check_info_filename(info, inf_one), "Expected returned filename to be equal\n");

     HeapFree(GetProcessHeap(), 0, info);
     info = alloc_inf_info("test.inf", INFINFO_REVERSE_DEFAULT_SEARCH, &size);

     /* test the INFINFO_REVERSE_DEFAULT_SEARCH search flag */
     ret = pSetupGetInfInformationA("test.inf", INFINFO_REVERSE_DEFAULT_SEARCH, info, size, &size);
     ok(ret == TRUE, "Expected SetupGetInfInformation to succeed\n");
     ok(check_info_filename(info, inf_two), "Expected returned filename to be equal\n");

     DeleteFileA(inf_filename);
     DeleteFileA(inf_one);
     DeleteFileA(inf_two);
 }

 START_TEST(query)
 {
     init_function_pointers();
     get_directories();

     test_SetupGetInfInformation();
 }
	/*
	* Unit tests for setupapi.dll query functions
	*
	* Copyright (C) 2006 James Hawkins
	*
	* 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 <setupapi.h>
	#include "wine/test.h"

	/* function pointers */
	static HMODULE hSetupAPI;
	static void (WINAPI *pSetupCloseInfFile)(HINF);
	static BOOL (WINAPI *pSetupGetInfInformationA)(LPCVOID,DWORD,PSP_INF_INFORMATION,DWORD,PDWORD);
	static HINF (WINAPI *pSetupOpenInfFileA)(PCSTR,PCSTR,DWORD,PUINT);
	static BOOL (WINAPI *pSetupQueryInfFileInformationA)(PSP_INF_INFORMATION,UINT,PSTR,DWORD,PDWORD);

	CHAR CURR_DIR[MAX_PATH];
	CHAR WIN_DIR[MAX_PATH];

	static void init_function_pointers(void)
	{
	hSetupAPI = LoadLibraryA("setupapi.dll");

	if (hSetupAPI)
	{
	pSetupCloseInfFile = (void *)GetProcAddress(hSetupAPI, "SetupCloseInfFile");
	pSetupGetInfInformationA = (void *)GetProcAddress(hSetupAPI, "SetupGetInfInformationA");
	pSetupOpenInfFileA = (void *)GetProcAddress(hSetupAPI, "SetupOpenInfFileA");
	pSetupQueryInfFileInformationA = (void *)GetProcAddress(hSetupAPI, "SetupQueryInfFileInformationA");
	}
	}

	static void get_directories(void)
	{
	int len;

	GetCurrentDirectoryA(MAX_PATH, CURR_DIR);
	len = lstrlenA(CURR_DIR);

	if(len && (CURR_DIR[len-1] == '\\'))
	CURR_DIR[len-1] = 0;

	GetWindowsDirectoryA(WIN_DIR, MAX_PATH);
	len = lstrlenA(WIN_DIR);

	if (len && (WIN_DIR[len-1] == '\\'))
	WIN_DIR[len-1] = 0;
	}

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

	static void create_inf_file(LPSTR filename)
	{
	char data[1024];
	char *ptr = data;
	DWORD dwNumberOfBytesWritten;
	HANDLE hf = CreateFile(filename, GENERIC_WRITE, 0, NULL,
	CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);

	append_str(&ptr, "[Version]\n");
	append_str(&ptr, "Signature=\"$Chicago$\"\n");
	append_str(&ptr, "AdvancedINF=2.5\n");

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

	static BOOL check_info_filename(PSP_INF_INFORMATION info, LPSTR test)
	{
	LPSTR filename;
	DWORD size;
	BOOL ret = FALSE;

	if (!pSetupQueryInfFileInformationA(info, 0, NULL, 0, &size))
	return FALSE;

	filename = HeapAlloc(GetProcessHeap(), 0, size);
	if (!filename)
	return FALSE;

	pSetupQueryInfFileInformationA(info, 0, filename, size, &size);

	if (!lstrcmpiA(test, filename))
	ret = TRUE;

	HeapFree(GetProcessHeap(), 0, filename);
	return ret;
	}

	static PSP_INF_INFORMATION alloc_inf_info(LPSTR filename, DWORD search, PDWORD size)
	{
	PSP_INF_INFORMATION info;
	BOOL ret;

	ret = pSetupGetInfInformationA(filename, search, NULL, 0, size);
	if (!ret)
	return NULL;

	info = HeapAlloc(GetProcessHeap(), 0, *size);
	return info;
	}

	static void test_SetupGetInfInformation(void)
	{
	PSP_INF_INFORMATION info;
	CHAR inf_filename[MAX_PATH];
	CHAR inf_one[MAX_PATH], inf_two[MAX_PATH];
	DWORD size;
	HINF hinf;
	BOOL ret;

	lstrcpyA(inf_filename, CURR_DIR);
	lstrcatA(inf_filename, "\\");
	lstrcatA(inf_filename, "test.inf");

	/* try an invalid inf handle */
	size = 0xdeadbeef;
	SetLastError(0xbeefcafe);
	ret = pSetupGetInfInformationA(NULL, INFINFO_INF_SPEC_IS_HINF, NULL, 0, &size);
	ok(ret == FALSE, "Expected SetupGetInfInformation to fail\n");
	ok(GetLastError() == ERROR_INVALID_HANDLE,
	"Expected ERROR_INVALID_HANDLE, got %ld\n", GetLastError());
	ok(size == 0xdeadbeef, "Expected size to remain unchanged\n");

	/* try an invalid inf filename */
	size = 0xdeadbeef;
	SetLastError(0xbeefcafe);
	ret = pSetupGetInfInformationA(NULL, INFINFO_INF_NAME_IS_ABSOLUTE, NULL, 0, &size);
	ok(ret == FALSE, "Expected SetupGetInfInformation to fail\n");
	ok(GetLastError() == ERROR_INVALID_PARAMETER,
	"Expected ERROR_INVALID_PARAMETER, got %ld\n", GetLastError());
	ok(size == 0xdeadbeef, "Expected size to remain unchanged\n");

	create_inf_file(inf_filename);

	/* try an invalid search flag */
	size = 0xdeadbeef;
	SetLastError(0xbeefcafe);
	ret = pSetupGetInfInformationA(inf_filename, -1, NULL, 0, &size);
	ok(ret == FALSE, "Expected SetupGetInfInformation to fail\n");
	ok(GetLastError() == ERROR_INVALID_PARAMETER,
	"Expected ERROR_INVALID_PARAMETER, got %ld\n", GetLastError());
	ok(size == 0xdeadbeef, "Expected size to remain unchanged\n");

	/* try a nonexistent inf file */
	size = 0xdeadbeef;
	SetLastError(0xbeefcafe);
	ret = pSetupGetInfInformationA("idontexist", INFINFO_INF_NAME_IS_ABSOLUTE, NULL, 0, &size);
	ok(ret == FALSE, "Expected SetupGetInfInformation to fail\n");
	ok(GetLastError() == ERROR_FILE_NOT_FOUND,
	"Expected ERROR_FILE_NOT_FOUND, got %ld\n", GetLastError());
	ok(size == 0xdeadbeef, "Expected size to remain unchanged\n");

	/* successfully open the inf file */
	size = 0xdeadbeef;
	ret = pSetupGetInfInformationA(inf_filename, INFINFO_INF_NAME_IS_ABSOLUTE, NULL, 0, &size);
	ok(ret == TRUE, "Expected SetupGetInfInformation to succeed\n");
	ok(size != 0xdeadbeef, "Expected a valid size on return\n");

	/* set ReturnBuffer to NULL and ReturnBufferSize to non-zero */
	SetLastError(0xbeefcafe);
	ret = pSetupGetInfInformationA(inf_filename, INFINFO_INF_NAME_IS_ABSOLUTE, NULL, size, &size);
	ok(ret == FALSE, "Expected SetupGetInfInformation to fail\n");
	ok(GetLastError() == ERROR_INVALID_PARAMETER,
	"Expected ERROR_INVALID_PARAMETER, got %ld\n", GetLastError());

	info = HeapAlloc(GetProcessHeap(), 0, size);

	/* try valid ReturnBuffer but too small size */
	SetLastError(0xbeefcafe);
	ret = pSetupGetInfInformationA(inf_filename, INFINFO_INF_NAME_IS_ABSOLUTE, info, size - 1, &size);
	ok(ret == FALSE, "Expected SetupGetInfInformation to fail\n");
	ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER,
	"Expected ERROR_INSUFFICIENT_BUFFER, got %ld\n", GetLastError());

	/* successfully get the inf information */
	ret = pSetupGetInfInformationA(inf_filename, INFINFO_INF_NAME_IS_ABSOLUTE, info, size, &size);
	ok(ret == TRUE, "Expected SetupGetInfInformation to succeed\n");
	ok(check_info_filename(info, inf_filename), "Expected returned filename to be equal\n");

	HeapFree(GetProcessHeap(), 0, info);

	/* try the INFINFO_INF_SPEC_IS_HINF search flag */
	hinf = pSetupOpenInfFileA(inf_filename, NULL, INF_STYLE_WIN4, NULL);
	info = alloc_inf_info(hinf, INFINFO_INF_SPEC_IS_HINF, &size);
	ret = pSetupGetInfInformationA(hinf, INFINFO_INF_SPEC_IS_HINF, info, size, &size);
	ok(ret == TRUE, "Expected SetupGetInfInformation to succeed\n");
	ok(check_info_filename(info, inf_filename), "Expected returned filename to be equal\n");
	pSetupCloseInfFile(hinf);

	lstrcpyA(inf_one, WIN_DIR);
	lstrcatA(inf_one, "\\inf\\");
	lstrcatA(inf_one, "test.inf");
	create_inf_file(inf_one);

	lstrcpyA(inf_two, WIN_DIR);
	lstrcatA(inf_two, "\\system32\\");
	lstrcatA(inf_two, "test.inf");
	create_inf_file(inf_two);

	HeapFree(GetProcessHeap(), 0, info);
	info = alloc_inf_info("test.inf", INFINFO_DEFAULT_SEARCH, &size);

	/* test the INFINFO_DEFAULT_SEARCH search flag */
	ret = pSetupGetInfInformationA("test.inf", INFINFO_DEFAULT_SEARCH, info, size, &size);
	ok(ret == TRUE, "Expected SetupGetInfInformation to succeed\n");
	ok(check_info_filename(info, inf_one), "Expected returned filename to be equal\n");

	HeapFree(GetProcessHeap(), 0, info);
	info = alloc_inf_info("test.inf", INFINFO_REVERSE_DEFAULT_SEARCH, &size);

	/* test the INFINFO_REVERSE_DEFAULT_SEARCH search flag */
	ret = pSetupGetInfInformationA("test.inf", INFINFO_REVERSE_DEFAULT_SEARCH, info, size, &size);
	ok(ret == TRUE, "Expected SetupGetInfInformation to succeed\n");
	ok(check_info_filename(info, inf_two), "Expected returned filename to be equal\n");

	DeleteFileA(inf_filename);
	DeleteFileA(inf_one);
	DeleteFileA(inf_two);
	}

	START_TEST(query)
	{
	init_function_pointers();
	get_directories();

	test_SetupGetInfInformation();
	}