Something went wrong. Try again.
Reactos
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333/* * SetupAPI device class-related functions tests * * Copyright 2006 Hervé Poussineau * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU General Public Licence as * published by the Free Software Foundation; either version 2 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 * General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this library; if not, write to the Free Software Foundation, * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA */
#include <assert.h>#include <stdarg.h>#include <stdio.h>
#include "windef.h"#include "winbase.h"#include "wingdi.h"#include "winuser.h"#include "winreg.h"#include "cfgmgr32.h"#include "setupapi.h"
#include "wine/test.h"
#define ok_lasterr(err) \ ok( GetLastError() == (err), \ "Expected error %lx, got %lx\n", (DWORD)(err), GetLastError() )
static GUID test_class_guid = { 0x4d36e967, 0xe325, 0x11ce, { 0xbf, 0xc1, 0x08, 0x00, 0x2b, 0xe1, 0x03, 0x18 } };static char test_class_name[MAX_CLASS_NAME_LEN] = "DiskDrive";
static void test_SetupDiBuildClassInfoList(void){ LPGUID guid_list = NULL; DWORD required_size, size;
SetLastError( 0xdeadbeef ); ok( !SetupDiBuildClassInfoList( 0, NULL, 0, NULL ), "Fail expected" ); ok_lasterr( ERROR_INVALID_PARAMETER );
SetLastError( 0xdeadbeef ); ok( !SetupDiBuildClassInfoList( 0, NULL, 0, &required_size ), "Fail expected\n" ); ok_lasterr( ERROR_INSUFFICIENT_BUFFER );
guid_list = HeapAlloc( GetProcessHeap(), 0, ( required_size + 1 ) * sizeof( GUID ) ); if ( !guid_list ) return;
SetLastError( 0xdeadbeef ); ok( SetupDiBuildClassInfoList( 0, guid_list, required_size, &size ), "Error reported %lx\n", GetLastError() ); ok( size == required_size, "Expected size %lu, got %lu\n", required_size, size ); SetLastError( 0xdeadbeef ); ok( SetupDiBuildClassInfoList( 0, guid_list, required_size + 1, &size ), "Error reported %lx\n", GetLastError() ); ok( size == required_size, "Expected size %lu, got %lu\n", required_size, size );
if ( size > 0 ) { /* That's better to use the first class found, as we know for sure that it exists */ memcpy(&test_class_guid, &guid_list[0], sizeof( GUID ) ); ok( SetupDiClassNameFromGuidA( &test_class_guid, test_class_name, sizeof( test_class_name ), NULL ), "Error reported %lx\n", GetLastError() ); } HeapFree( GetProcessHeap(), 0, guid_list );}
static void test_SetupDiClassGuidsFromNameA(void){ LPGUID guid_list = NULL; DWORD required_size, size;
SetLastError( 0xdeadbeef ); ok( !SetupDiClassGuidsFromNameA( NULL, NULL, 0, NULL ), "Fail expected\n" ); ok_lasterr( ERROR_INVALID_PARAMETER );
SetLastError( 0xdeadbeef ); ok( !SetupDiClassGuidsFromNameA( NULL, NULL, 0, &required_size ), "Fail expected\n" ); ok_lasterr( ERROR_INVALID_PARAMETER );
SetLastError( 0xdeadbeef ); ok( SetupDiClassGuidsFromNameA( "", NULL, 0, &required_size ), "Error reported %lx\n", GetLastError() ); ok( required_size == 0, "Expected 0, got %lu\n", required_size );
SetLastError( 0xdeadbeef ); ok( !SetupDiClassGuidsFromNameA( test_class_name, NULL, 0, &required_size ), "Fail expected\n" ); ok_lasterr( ERROR_INSUFFICIENT_BUFFER ); ok( required_size > 0, "Expected > 0, got %lu\n", required_size );
guid_list = HeapAlloc( GetProcessHeap(), 0, ( required_size + 1 ) * sizeof( GUID ) ); if ( !guid_list ) return;
SetLastError( 0xdeadbeef ); ok( SetupDiClassGuidsFromNameA( test_class_name, guid_list, required_size, &size ), "Error reported %lx\n", GetLastError() ); ok( size == required_size, "Expected size %lu, got %lu\n", required_size, size ); ok( IsEqualIID( &guid_list[0], &test_class_guid ), "Expected %s, got %s for class %s\n", debugstr_guid( &test_class_guid ), debugstr_guid( &guid_list[0] ), test_class_name ); SetLastError( 0xdeadbeef ); ok( SetupDiClassGuidsFromNameA( test_class_name, guid_list, required_size + 1, &size ), "Error reported %lx\n", GetLastError() ); ok( size == required_size, "Expected size %lu, got %lu\n", required_size, size ); ok( IsEqualIID( &guid_list[0], &test_class_guid ), "Expected %s, got %s for class %s\n", debugstr_guid( &test_class_guid ), debugstr_guid( &guid_list[0] ), test_class_name );
HeapFree( GetProcessHeap(), 0, guid_list );}
static void test_SetupDiClassNameFromGuidA(void){ CHAR* class_name = NULL; DWORD required_size, size;
SetLastError( 0xdeadbeef ); ok( !SetupDiClassNameFromGuidA( NULL, NULL, 0, NULL ), "Fail expected\n" ); ok_lasterr( ERROR_INVALID_CLASS );
SetLastError( 0xdeadbeef ); ok( !SetupDiClassNameFromGuidA( NULL, NULL, 0, &required_size ), "Fail expected\n" ); ok_lasterr( ERROR_INVALID_CLASS );
SetLastError( 0xdeadbeef ); ok( !SetupDiClassNameFromGuidA( &test_class_guid, NULL, 0, &required_size ), "Fail expected\n" ); ok_lasterr( ERROR_INSUFFICIENT_BUFFER ); ok( required_size > 0, "Expected > 0, got %lu\n", required_size ); ok( required_size < MAX_CLASS_NAME_LEN, "Expected < %u, got %lu for GUID %s\n", MAX_CLASS_NAME_LEN, required_size, debugstr_guid( &test_class_guid ) );
class_name = HeapAlloc( GetProcessHeap(), 0, required_size ); if ( !class_name ) return;
SetLastError( 0xdeadbeef ); ok( SetupDiClassNameFromGuidA( &test_class_guid, class_name, required_size, &size ), "Error reported %lx\n", GetLastError() ); ok( size == required_size, "Expected size %lu, got %lu\n", required_size, size ); ok( !strcmp( class_name, test_class_name ), "Expected %s, got %s\n", test_class_name, class_name ); SetLastError( 0xdeadbeef ); ok( SetupDiClassNameFromGuidA( &test_class_guid, class_name, required_size + 1, &size ), "Error reported %lx\n", GetLastError() ); ok( size == required_size, "Expected size %lu, got %lu\n", required_size, size ); ok( !strcmp( class_name, test_class_name ), "Expected %s, got %s\n", test_class_name, class_name );
HeapFree( GetProcessHeap(), 0, class_name );}
static void test_SetupDiGetClassDescriptionA(void){ CHAR* class_desc = NULL; DWORD required_size, size;
SetLastError( 0xdeadbeef ); ok( !SetupDiGetClassDescriptionA( NULL, NULL, 0, NULL ), "Fail expected\n" ); ok_lasterr( ERROR_INVALID_PARAMETER );
SetLastError( 0xdeadbeef ); ok( !SetupDiGetClassDescriptionA( NULL, NULL, 0, &required_size ), "Fail expected\n" ); ok_lasterr( ERROR_INVALID_PARAMETER );
SetLastError( 0xdeadbeef ); ok( !SetupDiGetClassDescriptionA( &test_class_guid, NULL, 0, &required_size ), "Fail expected\n" ); ok_lasterr( ERROR_INSUFFICIENT_BUFFER ); ok( required_size > 0, "Expected > 0, got %lu\n", required_size ); ok( required_size < LINE_LEN, "Expected < %u, got %lu\n", LINE_LEN, required_size );
class_desc = HeapAlloc( GetProcessHeap(), 0, required_size ); if ( !class_desc ) return;
SetLastError( 0xdeadbeef ); ok( SetupDiGetClassDescriptionA( &test_class_guid, class_desc, required_size, &size ), "Error reported %lx\n", GetLastError() ); ok( size == required_size, "Expected size %lu, got %lu\n", required_size, size ); SetLastError( 0xdeadbeef ); ok( SetupDiGetClassDescriptionA( &test_class_guid, class_desc, required_size + 1, &size ), "Error reported %lx\n", GetLastError() ); ok( size == required_size, "Expected size %lu, got %lu\n", required_size, size );
HeapFree( GetProcessHeap(), 0, class_desc );}
static void test_SetupDiGetClassDevsA(void){ HDEVINFO device_info;
SetLastError( 0xdeadbeef ); device_info = SetupDiGetClassDevs( NULL, NULL, NULL, 0 ); ok( device_info == INVALID_HANDLE_VALUE, "Fail expected\n" ); ok_lasterr( ERROR_INVALID_PARAMETER );
SetLastError( 0xdeadbeef ); device_info = SetupDiGetClassDevs( NULL, NULL, NULL, DIGCF_ALLCLASSES ); ok( device_info != INVALID_HANDLE_VALUE, "Error reported %lx\n", GetLastError() ); SetLastError( 0xdeadbeef ); ok( SetupDiDestroyDeviceInfoList( device_info ), "Error reported %lx\n", GetLastError() );
SetLastError( 0xdeadbeef ); device_info = SetupDiGetClassDevs( NULL, NULL, NULL, DIGCF_DEVICEINTERFACE ); ok( device_info == INVALID_HANDLE_VALUE, "Fail expected\n" ); ok_lasterr( ERROR_INVALID_PARAMETER );
SetLastError( 0xdeadbeef ); device_info = SetupDiGetClassDevs( &test_class_guid, NULL, NULL, 0 ); ok( device_info != INVALID_HANDLE_VALUE, "Error reported %lx\n", GetLastError() ); SetLastError( 0xdeadbeef ); ok( SetupDiDestroyDeviceInfoList( device_info ), "Error reported %lx\n", GetLastError() );
SetLastError( 0xdeadbeef ); device_info = SetupDiGetClassDevs( NULL, "(invalid enumerator)", NULL, DIGCF_ALLCLASSES ); ok( device_info == INVALID_HANDLE_VALUE, "Fail expected\n" ); ok_lasterr( ERROR_INVALID_DATA );
SetLastError( 0xdeadbeef ); device_info = SetupDiGetClassDevs( NULL, "Root", NULL, DIGCF_ALLCLASSES ); ok( device_info != INVALID_HANDLE_VALUE, "Error reported %lx\n", GetLastError() ); SetLastError( 0xdeadbeef ); ok( SetupDiDestroyDeviceInfoList( device_info ), "Error reported %lx\n", GetLastError() );}
static void test_SetupDiGetClassDevsExW(void){ const GUID not_existing_guid = { 0xdeadbeef, 0xdead, 0xbeef, { 0x88, 0x99, 0xaa, 0xbb, 0xcc, 0xdd, 0xee, 0xff } }; HDEVINFO dev_info; BOOL ret; SP_DEVINFO_DATA info;
dev_info = SetupDiGetClassDevsExW(¬_existing_guid, NULL, NULL, DIGCF_PRESENT | DIGCF_DEVICEINTERFACE, NULL, NULL, NULL); ok(dev_info != INVALID_HANDLE_VALUE, "Expected success\n");
ZeroMemory(&info, sizeof(info)); info.cbSize = sizeof(info); ret = SetupDiEnumDeviceInfo(dev_info, 0, &info); ok (!ret, "Expected failure.\n"); ok_lasterr(ERROR_NO_MORE_ITEMS);
SetupDiDestroyDeviceInfoList(dev_info);}
static void test_SetupDiOpenClassRegKeyExA(void){ HKEY hkey; LONG err;
SetLastError( 0xdeadbeef ); hkey = SetupDiOpenClassRegKeyExA( NULL, 0, 0, NULL, NULL ); ok( hkey == INVALID_HANDLE_VALUE, "Fail expected\n" ); ok_lasterr( ERROR_INVALID_FLAGS );
SetLastError( 0xdeadbeef ); hkey = SetupDiOpenClassRegKeyExA( NULL, 0, DIOCR_INSTALLER | DIOCR_INTERFACE, NULL, NULL ); ok( hkey == INVALID_HANDLE_VALUE, "Fail expected\n" ); ok_lasterr( ERROR_INVALID_FLAGS );
SetLastError( 0xdeadbeef ); hkey = SetupDiOpenClassRegKeyExA( NULL, 0, DIOCR_INSTALLER, NULL, NULL ); ok( hkey == INVALID_HANDLE_VALUE, "Fail expected\n" ); ok_lasterr( ERROR_INVALID_CLASS );
SetLastError( 0xdeadbeef ); hkey = SetupDiOpenClassRegKeyExA( NULL, 0, DIOCR_INTERFACE, NULL, NULL ); ok( hkey == INVALID_HANDLE_VALUE, "Fail expected\n" ); ok_lasterr( ERROR_INVALID_CLASS );
SetLastError( 0xdeadbeef ); hkey = SetupDiOpenClassRegKeyExA( NULL, KEY_QUERY_VALUE, DIOCR_INSTALLER, NULL, NULL ); ok( hkey != INVALID_HANDLE_VALUE, "Got error %lx\n", GetLastError() ); err = RegCloseKey( hkey ); ok( err == ERROR_SUCCESS, "Got error %lx\n", err );
SetLastError( 0xdeadbeef ); hkey = SetupDiOpenClassRegKeyExA( NULL, KEY_QUERY_VALUE, DIOCR_INTERFACE, NULL, NULL ); ok( hkey != INVALID_HANDLE_VALUE, "Got error %lx\n", GetLastError() ); err = RegCloseKey( hkey ); ok( err == ERROR_SUCCESS, "Got error %lx\n", err );
SetLastError( 0xdeadbeef ); hkey = SetupDiOpenClassRegKeyExA( &test_class_guid, KEY_QUERY_VALUE, DIOCR_INSTALLER, NULL, NULL ); ok( hkey != INVALID_HANDLE_VALUE, "Got error %lx\n", GetLastError() ); err = RegCloseKey( hkey ); ok( err == ERROR_SUCCESS, "Got error %lx\n", err );
err = RegOpenKeyExA(HKEY_LOCAL_MACHINE, "System\\CurrentControlSet\\Control\\Class", 0, KEY_SET_VALUE, &hkey); ok( err == ERROR_SUCCESS, "Got error %lx\n", err );}
START_TEST(devclass){ test_SetupDiBuildClassInfoList(); test_SetupDiClassGuidsFromNameA(); test_SetupDiClassNameFromGuidA(); test_SetupDiGetClassDescriptionA(); test_SetupDiGetClassDevsA(); test_SetupDiOpenClassRegKeyExA(); test_SetupDiGetClassDevsExW();}