Something went wrong. Try again.
Reactos
Something went wrong. Try again.
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742/* * SetupAPI device class-related functions * * Copyright 2000 Andreas Mohr for CodeWeavers * 2005-2006 Hervé Poussineau (hpoussin@reactos.org) * * 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., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA */
#include "setupapi_private.h"
#include <wingdi.h>#include <shellapi.h>#include <strsafe.h>
/* Unicode constants */static const WCHAR BackSlash[] = {'\\',0};static const WCHAR ClassGUID[] = {'C','l','a','s','s','G','U','I','D',0};static const WCHAR ClassInstall32[] = {'C','l','a','s','s','I','n','s','t','a','l','l','3','2',0};static const WCHAR DotServices[] = {'.','S','e','r','v','i','c','e','s',0};static const WCHAR InterfaceInstall32[] = {'I','n','t','e','r','f','a','c','e','I','n','s','t','a','l','l','3','2',0};static const WCHAR SetupapiDll[] = {'s','e','t','u','p','a','p','i','.','d','l','l',0};
typedef BOOL(WINAPI* PROPERTY_PAGE_PROVIDER) ( IN PSP_PROPSHEETPAGE_REQUEST PropPageRequest, IN LPFNADDPROPSHEETPAGE fAddFunc, IN LPARAM lParam);typedef BOOL(*UPDATE_CLASS_PARAM_HANDLER) ( IN HDEVINFO DeviceInfoSet, IN PSP_DEVINFO_DATA DeviceInfoData OPTIONAL, IN PSP_CLASSINSTALL_HEADER ClassInstallParams OPTIONAL, IN DWORD ClassInstallParamsSize);
static BOOLSETUP_PropertyChangeHandler( IN HDEVINFO DeviceInfoSet, IN PSP_DEVINFO_DATA DeviceInfoData OPTIONAL, IN PSP_CLASSINSTALL_HEADER ClassInstallParams OPTIONAL, IN DWORD ClassInstallParamsSize);
static BOOLSETUP_PropertyAddPropertyAdvancedHandler( IN HDEVINFO DeviceInfoSet, IN PSP_DEVINFO_DATA DeviceInfoData OPTIONAL, IN PSP_CLASSINSTALL_HEADER ClassInstallParams OPTIONAL, IN DWORD ClassInstallParamsSize);
typedef struct _INSTALL_PARAMS_DATA{ DI_FUNCTION Function; UPDATE_CLASS_PARAM_HANDLER UpdateHandler; ULONG ParamsSize; LONG FieldOffset;} INSTALL_PARAMS_DATA;
#define ADD_PARAM_HANDLER(Function, UpdateHandler, ParamsType, ParamsField) \ { Function, UpdateHandler, sizeof(ParamsType), FIELD_OFFSET(struct ClassInstallParams, ParamsField) },
static const INSTALL_PARAMS_DATA InstallParamsData[] = { ADD_PARAM_HANDLER(DIF_PROPERTYCHANGE, SETUP_PropertyChangeHandler, SP_PROPCHANGE_PARAMS, PropChangeParams) ADD_PARAM_HANDLER(DIF_ADDPROPERTYPAGE_ADVANCED, SETUP_PropertyAddPropertyAdvancedHandler, SP_ADDPROPERTYPAGE_DATA, AddPropertyPageData)};#undef ADD_PARAM_HANDLER
#define UNKNOWN_ICON_INDEX 18
/*********************************************************************** * SetupDiDestroyClassImageList(SETUPAPI.@) */BOOL WINAPISetupDiDestroyClassImageList( IN PSP_CLASSIMAGELIST_DATA ClassImageListData){ struct ClassImageList *list; BOOL ret = FALSE;
TRACE("%p\n", ClassImageListData);
if (!ClassImageListData) SetLastError(ERROR_INVALID_PARAMETER); else if (ClassImageListData->cbSize != sizeof(SP_CLASSIMAGELIST_DATA)) SetLastError(ERROR_INVALID_USER_BUFFER); else if ((list = (struct ClassImageList *)ClassImageListData->Reserved) == NULL) SetLastError(ERROR_INVALID_USER_BUFFER); else if (list->magic != SETUP_CLASS_IMAGE_LIST_MAGIC) SetLastError(ERROR_INVALID_USER_BUFFER); else { /* If Reserved wasn't NULL, then this is valid too */ if (ClassImageListData->ImageList) { ImageList_Destroy(ClassImageListData->ImageList); ClassImageListData->ImageList = NULL; }
MyFree(list); ClassImageListData->Reserved = 0;
ret = TRUE; }
TRACE("Returning %d\n", ret); return ret;}
LONGSETUP_CreateDevicesList( IN OUT struct DeviceInfoSet *list, IN PCWSTR MachineName OPTIONAL, IN CONST GUID *Class OPTIONAL, IN PCWSTR Enumerator OPTIONAL){ PWCHAR Buffer = NULL; DWORD BufferLength = 4096; PCWSTR InstancePath; struct DeviceInfo *deviceInfo; WCHAR ClassGuidBuffer[MAX_GUID_STRING_LEN]; DWORD ClassGuidBufferSize; GUID ClassGuid; DEVINST dnDevInst; CONFIGRET cr;
Buffer = HeapAlloc(GetProcessHeap(), 0, BufferLength); if (!Buffer) return ERROR_NOT_ENOUGH_MEMORY;
do { cr = CM_Get_Device_ID_List_ExW(Enumerator, Buffer, BufferLength / sizeof(WCHAR), Enumerator ? CM_GETIDLIST_FILTER_ENUMERATOR : CM_GETIDLIST_FILTER_NONE, list->hMachine); if (cr == CR_BUFFER_SMALL) { if (Buffer) HeapFree(GetProcessHeap(), 0, Buffer); BufferLength *= 2; Buffer = HeapAlloc(GetProcessHeap(), 0, BufferLength); if (!Buffer) return ERROR_NOT_ENOUGH_MEMORY; } else if (cr != CR_SUCCESS) { TRACE("CM_Get_Device_ID_List_ExW() failed with status 0x%x\n", cr); if (Buffer) HeapFree(GetProcessHeap(), 0, Buffer); return (cr == CR_REGISTRY_ERROR) ? ERROR_INVALID_DATA : GetErrorCodeFromCrCode(cr); } } while (cr != CR_SUCCESS);
for (InstancePath = Buffer; *InstancePath != UNICODE_NULL; InstancePath += wcslen(InstancePath) + 1) { /* Check that device really exists */ TRACE("Checking %S\n", InstancePath); cr = CM_Locate_DevNode_Ex(&dnDevInst, (DEVINSTID_W)InstancePath, CM_LOCATE_DEVNODE_NORMAL, list->hMachine); if (cr != CR_SUCCESS) { ERR("CM_Locate_DevNode_ExW('%S') failed with status 0x%x\n", InstancePath, cr); continue; }
/* Retrieve GUID of this device */ ClassGuidBufferSize = sizeof(ClassGuidBuffer); cr = CM_Get_DevNode_Registry_Property_ExW(dnDevInst, CM_DRP_CLASSGUID, NULL, ClassGuidBuffer, &ClassGuidBufferSize, 0, list->hMachine); if (cr == CR_SUCCESS) { ClassGuidBuffer[MAX_GUID_STRING_LEN - 2] = '\0'; /* Replace the } by a NULL character */ if (UuidFromStringW(&ClassGuidBuffer[1], &ClassGuid) != RPC_S_OK) { /* Bad GUID, skip the entry */ ERR("Invalid ClassGUID '%S' for device %S\n", ClassGuidBuffer, InstancePath); continue; } } else { TRACE("Using default class GUID_NULL for device %S\n", InstancePath); memcpy(&ClassGuid, &GUID_NULL, sizeof(GUID)); }
if (Class && !IsEqualIID(&ClassGuid, Class)) { TRACE("Skipping %S due to wrong class GUID\n", InstancePath); continue; }
/* Good! Create a device info element */ if (!CreateDeviceInfo(list, InstancePath, &ClassGuid, &deviceInfo)) { ERR("Failed to create info for %S\n", InstancePath); HeapFree(GetProcessHeap(), 0, Buffer); return GetLastError(); }
TRACE("Adding device %s to list\n", debugstr_w(InstancePath)); InsertTailList(&list->ListHead, &deviceInfo->ListEntry); }
HeapFree(GetProcessHeap(), 0, Buffer); return ERROR_SUCCESS;}
static BOOLSETUP_GetIconIndex( IN HKEY hClassKey, OUT PINT ImageIndex){ LPWSTR Buffer = NULL; DWORD dwRegType, dwLength; LONG rc; BOOL ret = FALSE;
/* Read icon registry key */ rc = RegQueryValueExW(hClassKey, REGSTR_VAL_INSICON, NULL, &dwRegType, NULL, &dwLength); if (rc != ERROR_SUCCESS) { SetLastError(rc); goto cleanup; } else if (dwRegType != REG_SZ) { SetLastError(ERROR_INVALID_INDEX); goto cleanup; } Buffer = MyMalloc(dwLength + sizeof(WCHAR)); if (!Buffer) { SetLastError(ERROR_NOT_ENOUGH_MEMORY); goto cleanup; } rc = RegQueryValueExW(hClassKey, REGSTR_VAL_INSICON, NULL, NULL, (LPBYTE)Buffer, &dwLength); if (rc != ERROR_SUCCESS) { SetLastError(rc); goto cleanup; } /* make sure the returned buffer is NULL-terminated */ Buffer[dwLength / sizeof(WCHAR)] = 0;
/* Transform icon value to a INT */ *ImageIndex = atoiW(Buffer); ret = TRUE;
cleanup: MyFree(Buffer); return ret;}
/*********************************************************************** * SetupDiGetClassImageIndex (SETUPAPI.@) */BOOL WINAPISetupDiGetClassImageIndex( IN PSP_CLASSIMAGELIST_DATA ClassImageListData, IN CONST GUID *ClassGuid, OUT PINT ImageIndex){ struct ClassImageList *list; BOOL ret = FALSE;
TRACE("%p %s %p\n", ClassImageListData, debugstr_guid(ClassGuid), ImageIndex);
if (!ClassImageListData || !ClassGuid || !ImageIndex) SetLastError(ERROR_INVALID_PARAMETER); else if (ClassImageListData->cbSize != sizeof(SP_CLASSIMAGELIST_DATA)) SetLastError(ERROR_INVALID_USER_BUFFER); else if ((list = (struct ClassImageList *)ClassImageListData->Reserved) == NULL) SetLastError(ERROR_INVALID_USER_BUFFER); else if (list->magic != SETUP_CLASS_IMAGE_LIST_MAGIC) SetLastError(ERROR_INVALID_USER_BUFFER); else { DWORD i;
for (i = 0; i < list->NumberOfGuids; i++) { if (IsEqualIID(ClassGuid, &list->Guids[i])) break; }
if (i == list->NumberOfGuids || list->IconIndexes[i] < 0) SetLastError(ERROR_FILE_NOT_FOUND); else { *ImageIndex = list->IconIndexes[i]; ret = TRUE; } }
TRACE("Returning %d\n", ret); return ret;}
/*********************************************************************** * SetupDiGetClassImageList(SETUPAPI.@) */BOOL WINAPISetupDiGetClassImageList( OUT PSP_CLASSIMAGELIST_DATA ClassImageListData){ return SetupDiGetClassImageListExW(ClassImageListData, NULL, NULL);}
/*********************************************************************** * SetupDiGetClassImageListExA(SETUPAPI.@) */BOOL WINAPISetupDiGetClassImageListExA( OUT PSP_CLASSIMAGELIST_DATA ClassImageListData, IN PCSTR MachineName OPTIONAL, IN PVOID Reserved){ PWSTR MachineNameW = NULL; BOOL ret;
if (MachineName) { MachineNameW = pSetupMultiByteToUnicode(MachineName, CP_ACP); if (MachineNameW == NULL) return FALSE; }
ret = SetupDiGetClassImageListExW(ClassImageListData, MachineNameW, Reserved);
MyFree(MachineNameW);
return ret;}
static BOOL WINAPISETUP_GetClassIconInfo(IN CONST GUID *ClassGuid, OUT PINT OutIndex, OUT LPWSTR *OutDllName){ LPWSTR Buffer = NULL; INT iconIndex = -UNKNOWN_ICON_INDEX; HKEY hKey = INVALID_HANDLE_VALUE; BOOL ret = FALSE;
if (ClassGuid) { hKey = SetupDiOpenClassRegKey(ClassGuid, KEY_QUERY_VALUE); if (hKey != INVALID_HANDLE_VALUE) { SETUP_GetIconIndex(hKey, &iconIndex); } }
if (iconIndex > 0) { /* Look up icon in dll specified by Installer32 or EnumPropPages32 key */ PWCHAR Comma; LONG rc; DWORD dwRegType, dwLength; rc = RegQueryValueExW(hKey, REGSTR_VAL_INSTALLER_32, NULL, &dwRegType, NULL, &dwLength); if (rc == ERROR_SUCCESS && dwRegType == REG_SZ) { Buffer = MyMalloc(dwLength + sizeof(WCHAR)); if (Buffer == NULL) { SetLastError(ERROR_NOT_ENOUGH_MEMORY); goto cleanup; } rc = RegQueryValueExW(hKey, REGSTR_VAL_INSTALLER_32, NULL, NULL, (LPBYTE)Buffer, &dwLength); if (rc != ERROR_SUCCESS) { SetLastError(rc); goto cleanup; } /* make sure the returned buffer is NULL-terminated */ Buffer[dwLength / sizeof(WCHAR)] = 0; } else if (ERROR_SUCCESS == (rc = RegQueryValueExW(hKey, REGSTR_VAL_ENUMPROPPAGES_32, NULL, &dwRegType, NULL, &dwLength)) && dwRegType == REG_SZ) { Buffer = MyMalloc(dwLength + sizeof(WCHAR)); if (Buffer == NULL) { SetLastError(ERROR_NOT_ENOUGH_MEMORY); goto cleanup; } rc = RegQueryValueExW(hKey, REGSTR_VAL_ENUMPROPPAGES_32, NULL, NULL, (LPBYTE)Buffer, &dwLength); if (rc != ERROR_SUCCESS) { SetLastError(rc); goto cleanup; } /* make sure the returned buffer is NULL-terminated */ Buffer[dwLength / sizeof(WCHAR)] = 0; } else { /* Unable to find where to load the icon */ SetLastError(ERROR_FILE_NOT_FOUND); goto cleanup; } Comma = strchrW(Buffer, ','); if (!Comma) { SetLastError(ERROR_GEN_FAILURE); goto cleanup; } *Comma = '\0'; *OutDllName = Buffer; } else { /* Look up icon in setupapi.dll */ iconIndex = -iconIndex; *OutDllName = NULL; }
*OutIndex = iconIndex; ret = TRUE;
TRACE("Icon index %d, dll name %s\n", iconIndex, debugstr_w(*OutDllName ? *OutDllName : SetupapiDll));
cleanup:
if (hKey != INVALID_HANDLE_VALUE) RegCloseKey(hKey);
if (Buffer && !ret) MyFree(Buffer);
return ret;}
/*********************************************************************** * SetupDiGetClassImageListExW(SETUPAPI.@) */BOOL WINAPISetupDiGetClassImageListExW( OUT PSP_CLASSIMAGELIST_DATA ClassImageListData, IN PCWSTR MachineName OPTIONAL, IN PVOID Reserved){ BOOL ret = FALSE;
TRACE("%p %p %p\n", ClassImageListData, debugstr_w(MachineName), Reserved);
if (!ClassImageListData) SetLastError(ERROR_INVALID_PARAMETER); else if (ClassImageListData->cbSize != sizeof(SP_CLASSIMAGELIST_DATA)) SetLastError(ERROR_INVALID_USER_BUFFER); else if (Reserved) SetLastError(ERROR_INVALID_PARAMETER); else { struct ClassImageList *list = NULL; HDC hDC; DWORD RequiredSize; DWORD ilMask, bkColor; HICON hIcon; DWORD size; INT i, bpp; UINT idx;
/* Get list of all class GUIDs in given computer */ ret = SetupDiBuildClassInfoListExW( 0, NULL, 0, &RequiredSize, MachineName, NULL); if (!ret && GetLastError() != ERROR_INSUFFICIENT_BUFFER) goto cleanup;
size = sizeof(struct ClassImageList) + (sizeof(GUID) + sizeof(INT)) * RequiredSize; list = HeapAlloc(GetProcessHeap(), 0, size); if (!list) { SetLastError(ERROR_NOT_ENOUGH_MEMORY); goto cleanup; } list->magic = SETUP_CLASS_IMAGE_LIST_MAGIC; list->NumberOfGuids = RequiredSize; list->Guids = (GUID*)(list + 1); list->IconIndexes = (INT*)((ULONG_PTR)(list + 1) + sizeof(GUID) * RequiredSize);
ret = SetupDiBuildClassInfoListExW( 0, list->Guids, list->NumberOfGuids, &RequiredSize, MachineName, NULL); if (!ret) goto cleanup; else if (RequiredSize != list->NumberOfGuids) { /* Hm. Class list changed since last call. Ignore * this case as it should be very rare */ SetLastError(ERROR_GEN_FAILURE); ret = FALSE; goto cleanup; }
/* Prepare a HIMAGELIST */ InitCommonControls();
hDC = GetDC(NULL); if (!hDC) goto cleanup;
bpp = GetDeviceCaps(hDC, BITSPIXEL); ReleaseDC(NULL, hDC);
if (bpp <= 4) ilMask = ILC_COLOR4; else if (bpp <= 8) ilMask = ILC_COLOR8; else if (bpp <= 16) ilMask = ILC_COLOR16; else if (bpp <= 24) ilMask = ILC_COLOR24; else if (bpp <= 32) ilMask = ILC_COLOR32; else ilMask = ILC_COLOR;
ilMask |= ILC_MASK;
ClassImageListData->ImageList = ImageList_Create(16, 16, ilMask, 100, 10); if (!ClassImageListData->ImageList) goto cleanup;
ClassImageListData->Reserved = (ULONG_PTR)list;
/* For some reason, Windows sets the list background to COLOR_WINDOW */ bkColor = GetSysColor(COLOR_WINDOW); ImageList_SetBkColor(ClassImageListData->ImageList, bkColor);
/* Now, we "simply" need to load icons associated with all class guids, * and put their index in the image list in the IconIndexes array */ for (i = 0; i < list->NumberOfGuids; i++) { INT miniIconIndex; LPWSTR DllName = NULL;
if (SETUP_GetClassIconInfo(&list->Guids[i], &miniIconIndex, &DllName)) { if (DllName && ExtractIconExW(DllName, -miniIconIndex, NULL, &hIcon, 1) == 1) { list->IconIndexes[i] = ImageList_AddIcon(ClassImageListData->ImageList, hIcon); } else if(!DllName) { hIcon = LoadImage(hInstance, MAKEINTRESOURCE(miniIconIndex), IMAGE_ICON, 16, 16, LR_DEFAULTCOLOR); list->IconIndexes[i] = ImageList_AddIcon(ClassImageListData->ImageList, hIcon); }
if(hIcon) DestroyIcon(hIcon); else list->IconIndexes[i] = -1;
if(DllName) MyFree(DllName); } else { list->IconIndexes[i] = -1; /* Special value to indicate that the icon is unavailable */ } }
/* Finally, add the overlay icons to the image list */ for (i = 0; i <= 2; i++) { hIcon = LoadImage(hInstance, MAKEINTRESOURCE(500 + i), IMAGE_ICON, 16, 16, LR_DEFAULTCOLOR); if (hIcon) { idx = ImageList_AddIcon(ClassImageListData->ImageList, hIcon); if (idx != -1) ImageList_SetOverlayImage(ClassImageListData->ImageList, idx, i + 1); DestroyIcon(hIcon); } }
ret = TRUE;
cleanup: if (!ret) { if (ClassImageListData->Reserved) SetupDiDestroyClassImageList(ClassImageListData); else if (list) MyFree(list); } }
TRACE("Returning %d\n", ret); return ret;}
/*********************************************************************** * SetupDiGetClassInstallParamsA(SETUPAPI.@) */BOOL WINAPISetupDiGetClassInstallParamsA( IN HDEVINFO DeviceInfoSet, IN PSP_DEVINFO_DATA DeviceInfoData OPTIONAL, OUT PSP_CLASSINSTALL_HEADER ClassInstallParams OPTIONAL, IN DWORD ClassInstallParamsSize, OUT PDWORD RequiredSize OPTIONAL){ FIXME("SetupDiGetClassInstallParamsA(%p %p %p %lu %p) Stub\n", DeviceInfoSet, DeviceInfoData, ClassInstallParams, ClassInstallParamsSize, RequiredSize); return FALSE;}
/*********************************************************************** * SetupDiGetClassInstallParamsW(SETUPAPI.@) */BOOL WINAPISetupDiGetClassInstallParamsW( IN HDEVINFO DeviceInfoSet, IN PSP_DEVINFO_DATA DeviceInfoData OPTIONAL, OUT PSP_CLASSINSTALL_HEADER ClassInstallParams OPTIONAL, IN DWORD ClassInstallParamsSize, OUT PDWORD RequiredSize OPTIONAL){ FIXME("SetupDiGetClassInstallParamsW(%p %p %p %lu %p) Stub\n", DeviceInfoSet, DeviceInfoData, ClassInstallParams, ClassInstallParamsSize, RequiredSize); return FALSE;}
/*********************************************************************** * SetupDiLoadClassIcon(SETUPAPI.@) */BOOL WINAPISetupDiLoadClassIcon( IN CONST GUID *ClassGuid, OUT HICON *LargeIcon OPTIONAL, OUT PINT MiniIconIndex OPTIONAL){ INT iconIndex = 0; LPWSTR DllName = NULL; BOOL ret = FALSE; HICON hIcon = NULL;
if (LargeIcon) { if(!SETUP_GetClassIconInfo(ClassGuid, &iconIndex, &DllName)) goto cleanup;
if (!DllName || ExtractIconExW(DllName, -iconIndex, &hIcon, NULL, 1) != 1 || hIcon == NULL) { /* load the default unknown device icon if ExtractIcon failed */ if(DllName) iconIndex = UNKNOWN_ICON_INDEX;
hIcon = LoadImage(hInstance, MAKEINTRESOURCE(iconIndex), IMAGE_ICON, 32, 32, LR_DEFAULTCOLOR);
if(!hIcon) goto cleanup; }
*LargeIcon = hIcon; }
if (MiniIconIndex) *MiniIconIndex = iconIndex;
ret = TRUE;
cleanup:
if(DllName) MyFree(DllName);
TRACE("Returning %d\n", ret); return ret;}
/*********************************************************************** * SetupDiInstallClassExW (SETUPAPI.@) */HKEYSETUP_CreateClassKey(HINF hInf);BOOL WINAPISetupDiInstallClassExW( IN HWND hwndParent OPTIONAL, IN PCWSTR InfFileName OPTIONAL, IN DWORD Flags, IN HSPFILEQ FileQueue OPTIONAL, IN CONST GUID *InterfaceClassGuid OPTIONAL, IN PVOID Reserved1, IN PVOID Reserved2){ BOOL ret = FALSE;
TRACE("%p %s 0x%lx %p %s %p %p\n", hwndParent, debugstr_w(InfFileName), Flags, FileQueue, debugstr_guid(InterfaceClassGuid), Reserved1, Reserved2);
if (!InfFileName) { FIXME("Case not implemented: InfFileName NULL\n"); SetLastError(ERROR_CALL_NOT_IMPLEMENTED); } else if (Flags & ~(DI_NOVCP | DI_NOBROWSE | DI_FORCECOPY | DI_QUIETINSTALL)) { TRACE("Unknown flags: 0x%08lx\n", Flags & ~(DI_NOVCP | DI_NOBROWSE | DI_FORCECOPY | DI_QUIETINSTALL)); SetLastError(ERROR_INVALID_FLAGS); } else if ((Flags & DI_NOVCP) && FileQueue == NULL) SetLastError(ERROR_INVALID_PARAMETER); else if (Reserved1 != NULL) SetLastError(ERROR_INVALID_PARAMETER); else if (Reserved2 != NULL) SetLastError(ERROR_INVALID_PARAMETER); else { HDEVINFO hDeviceInfo = INVALID_HANDLE_VALUE; SP_DEVINSTALL_PARAMS_W InstallParams; WCHAR SectionName[MAX_PATH]; HINF hInf = INVALID_HANDLE_VALUE; HKEY hRootKey = INVALID_HANDLE_VALUE; PVOID callback_context = NULL;
hDeviceInfo = SetupDiCreateDeviceInfoList(NULL, NULL); if (hDeviceInfo == INVALID_HANDLE_VALUE) goto cleanup;
InstallParams.cbSize = sizeof(SP_DEVINSTALL_PARAMS); if (!SetupDiGetDeviceInstallParamsW(hDeviceInfo, NULL, &InstallParams)) goto cleanup;
InstallParams.Flags &= ~(DI_NOVCP | DI_NOBROWSE | DI_QUIETINSTALL); InstallParams.Flags |= Flags & (DI_NOVCP | DI_NOBROWSE | DI_QUIETINSTALL); if (Flags & DI_NOVCP) InstallParams.FileQueue = FileQueue; if (!SetupDiSetDeviceInstallParamsW(hDeviceInfo, NULL, &InstallParams)) goto cleanup;
/* Open the .inf file */ hInf = SetupOpenInfFileW( InfFileName, NULL, INF_STYLE_WIN4, NULL); if (hInf == INVALID_HANDLE_VALUE) goto cleanup;
/* Try to append a layout file */ SetupOpenAppendInfFileW(NULL, hInf, NULL);
if (InterfaceClassGuid) { /* Retrieve the actual section name */ ret = SetupDiGetActualSectionToInstallW( hInf, InterfaceInstall32, SectionName, MAX_PATH, NULL, NULL); if (!ret) goto cleanup;
/* Open registry key related to this interface */ /* FIXME: What happens if the key doesn't exist? */ hRootKey = SetupDiOpenClassRegKeyExW(InterfaceClassGuid, KEY_ENUMERATE_SUB_KEYS, DIOCR_INTERFACE, NULL, NULL); if (hRootKey == INVALID_HANDLE_VALUE) goto cleanup;
/* SetupDiCreateDeviceInterface??? */ FIXME("Installing an interface is not implemented\n"); SetLastError(ERROR_CALL_NOT_IMPLEMENTED); } else { /* Create or open the class registry key 'HKLM\CurrentControlSet\Class\{GUID}' */ hRootKey = SETUP_CreateClassKey(hInf); if (hRootKey == INVALID_HANDLE_VALUE) goto cleanup;
/* Retrieve the actual section name */ ret = SetupDiGetActualSectionToInstallW( hInf, ClassInstall32, SectionName, MAX_PATH - strlenW(DotServices), NULL, NULL); if (!ret) goto cleanup;
callback_context = SetupInitDefaultQueueCallback(hwndParent); if (!callback_context) goto cleanup;
ret = SetupInstallFromInfSectionW( hwndParent, hInf, SectionName, SPINST_REGISTRY | SPINST_FILES | SPINST_BITREG | SPINST_INIFILES | SPINST_INI2REG, hRootKey, NULL, /* FIXME: SourceRootPath */ !(Flags & DI_NOVCP) && (Flags & DI_FORCECOPY) ? SP_COPY_FORCE_IN_USE : 0, /* CopyFlags */ SetupDefaultQueueCallbackW, callback_context, hDeviceInfo, NULL); if (!ret) goto cleanup;
/* OPTIONAL: Install .Services section */ lstrcatW(SectionName, DotServices); SetupInstallServicesFromInfSectionExW( hInf, SectionName, 0, hDeviceInfo, NULL, NULL, NULL); ret = TRUE; }
cleanup: if (hDeviceInfo != INVALID_HANDLE_VALUE) SetupDiDestroyDeviceInfoList(hDeviceInfo); if (hInf != INVALID_HANDLE_VALUE) SetupCloseInfFile(hInf); if (hRootKey != INVALID_HANDLE_VALUE) RegCloseKey(hRootKey); SetupTermDefaultQueueCallback(callback_context); }
TRACE("Returning %d\n", ret); return ret;}
/*********************************************************************** * Helper functions for SetupDiSetClassInstallParamsW */static BOOLSETUP_PropertyChangeHandler( IN HDEVINFO DeviceInfoSet, IN PSP_DEVINFO_DATA DeviceInfoData, IN PSP_CLASSINSTALL_HEADER ClassInstallParams OPTIONAL, IN DWORD ClassInstallParamsSize){ PSP_PROPCHANGE_PARAMS PropChangeParams = (PSP_PROPCHANGE_PARAMS)ClassInstallParams; BOOL ret = FALSE;
if (!DeviceInfoData) SetLastError(ERROR_INVALID_PARAMETER); else if (ClassInstallParamsSize != sizeof(SP_PROPCHANGE_PARAMS)) SetLastError(ERROR_INVALID_PARAMETER); else if (PropChangeParams && PropChangeParams->StateChange != DICS_ENABLE && PropChangeParams->StateChange != DICS_DISABLE && PropChangeParams->StateChange != DICS_PROPCHANGE && PropChangeParams->StateChange != DICS_START && PropChangeParams->StateChange != DICS_STOP) SetLastError(ERROR_INVALID_FLAGS); else if (PropChangeParams && PropChangeParams->Scope != DICS_FLAG_GLOBAL && PropChangeParams->Scope != DICS_FLAG_CONFIGSPECIFIC) SetLastError(ERROR_INVALID_FLAGS); else if (PropChangeParams && (PropChangeParams->StateChange == DICS_START || PropChangeParams->StateChange == DICS_STOP) && PropChangeParams->Scope != DICS_FLAG_CONFIGSPECIFIC) SetLastError(ERROR_INVALID_USER_BUFFER); else { PSP_PROPCHANGE_PARAMS *CurrentPropChangeParams; struct DeviceInfo *deviceInfo = (struct DeviceInfo *)DeviceInfoData->Reserved; CurrentPropChangeParams = &deviceInfo->ClassInstallParams.PropChangeParams;
if (*CurrentPropChangeParams) { MyFree(*CurrentPropChangeParams); *CurrentPropChangeParams = NULL; } if (PropChangeParams) { *CurrentPropChangeParams = MyMalloc(ClassInstallParamsSize); if (!*CurrentPropChangeParams) { SetLastError(ERROR_NOT_ENOUGH_MEMORY); goto done; } memcpy(*CurrentPropChangeParams, PropChangeParams, ClassInstallParamsSize); } ret = TRUE; }
done: return ret;}
static BOOLSETUP_PropertyAddPropertyAdvancedHandler( IN HDEVINFO DeviceInfoSet, IN PSP_DEVINFO_DATA DeviceInfoData OPTIONAL, IN PSP_CLASSINSTALL_HEADER ClassInstallParams OPTIONAL, IN DWORD ClassInstallParamsSize){ PSP_ADDPROPERTYPAGE_DATA AddPropertyPageData = (PSP_ADDPROPERTYPAGE_DATA)ClassInstallParams; BOOL ret = FALSE;
if (ClassInstallParamsSize != sizeof(SP_PROPCHANGE_PARAMS)) SetLastError(ERROR_INVALID_PARAMETER); else if (AddPropertyPageData && AddPropertyPageData->Flags != 0) SetLastError(ERROR_INVALID_FLAGS); else if (AddPropertyPageData && AddPropertyPageData->NumDynamicPages >= MAX_INSTALLWIZARD_DYNAPAGES) SetLastError(ERROR_INVALID_USER_BUFFER); else { PSP_ADDPROPERTYPAGE_DATA *CurrentAddPropertyPageData; if (!DeviceInfoData) { struct DeviceInfoSet *list = (struct DeviceInfoSet *)DeviceInfoSet; CurrentAddPropertyPageData = &list->ClassInstallParams.AddPropertyPageData; } else { struct DeviceInfo *deviceInfo = (struct DeviceInfo *)DeviceInfoData->Reserved; CurrentAddPropertyPageData = &deviceInfo->ClassInstallParams.AddPropertyPageData; } if (*CurrentAddPropertyPageData) { MyFree(*CurrentAddPropertyPageData); *CurrentAddPropertyPageData = NULL; } if (AddPropertyPageData) { *CurrentAddPropertyPageData = MyMalloc(ClassInstallParamsSize); if (!*CurrentAddPropertyPageData) { SetLastError(ERROR_NOT_ENOUGH_MEMORY); goto done; } memcpy(*CurrentAddPropertyPageData, AddPropertyPageData, ClassInstallParamsSize); } ret = TRUE; }
done: return ret;}
/*********************************************************************** * SetupDiSetClassInstallParamsW (SETUPAPI.@) */BOOL WINAPISetupDiSetClassInstallParamsW( IN HDEVINFO DeviceInfoSet, IN PSP_DEVINFO_DATA DeviceInfoData OPTIONAL, IN PSP_CLASSINSTALL_HEADER ClassInstallParams OPTIONAL, IN DWORD ClassInstallParamsSize){ struct DeviceInfoSet *list; BOOL ret = FALSE;
TRACE("%p %p %p %lu\n", DeviceInfoSet, DeviceInfoData, ClassInstallParams, ClassInstallParamsSize);
if (!DeviceInfoSet) SetLastError(ERROR_INVALID_PARAMETER); else if (DeviceInfoSet == (HDEVINFO)INVALID_HANDLE_VALUE) SetLastError(ERROR_INVALID_HANDLE); else if ((list = (struct DeviceInfoSet *)DeviceInfoSet)->magic != SETUP_DEVICE_INFO_SET_MAGIC) SetLastError(ERROR_INVALID_HANDLE); else if (DeviceInfoData && DeviceInfoData->cbSize != sizeof(SP_DEVINFO_DATA)) SetLastError(ERROR_INVALID_USER_BUFFER); else if (ClassInstallParams && ClassInstallParams->cbSize != sizeof(SP_CLASSINSTALL_HEADER)) SetLastError(ERROR_INVALID_USER_BUFFER); else if (ClassInstallParams && ClassInstallParamsSize < sizeof(SP_CLASSINSTALL_HEADER)) SetLastError(ERROR_INVALID_PARAMETER); else if (!ClassInstallParams && ClassInstallParamsSize != 0) SetLastError(ERROR_INVALID_PARAMETER); else { SP_DEVINSTALL_PARAMS_W InstallParams; BOOL Result;
InstallParams.cbSize = sizeof(SP_DEVINSTALL_PARAMS_W); Result = SetupDiGetDeviceInstallParamsW(DeviceInfoSet, DeviceInfoData, &InstallParams); if (!Result) goto done;
if (ClassInstallParams) { DWORD i; /* Check parameters in ClassInstallParams */ for (i = 0; i < sizeof(InstallParamsData) / sizeof(InstallParamsData[0]); i++) { if (InstallParamsData[i].Function == ClassInstallParams->InstallFunction) { ret = InstallParamsData[i].UpdateHandler( DeviceInfoSet, DeviceInfoData, ClassInstallParams, ClassInstallParamsSize); if (ret) { InstallParams.Flags |= DI_CLASSINSTALLPARAMS; ret = SetupDiSetDeviceInstallParamsW(DeviceInfoSet, DeviceInfoData, &InstallParams); } goto done; } } ERR("InstallFunction %u has no associated update handler\n", ClassInstallParams->InstallFunction); SetLastError(ERROR_CALL_NOT_IMPLEMENTED); goto done; } else { InstallParams.Flags &= ~DI_CLASSINSTALLPARAMS; ret = SetupDiSetDeviceInstallParamsW(DeviceInfoSet, DeviceInfoData, &InstallParams); } }
done: TRACE("Returning %d\n", ret); return ret;}
/*********************************************************************** * SetupDiGetClassDevPropertySheetsA(SETUPAPI.@) */BOOL WINAPISetupDiGetClassDevPropertySheetsA( IN HDEVINFO DeviceInfoSet, IN PSP_DEVINFO_DATA DeviceInfoData OPTIONAL, IN LPPROPSHEETHEADERA PropertySheetHeader, IN DWORD PropertySheetHeaderPageListSize, OUT PDWORD RequiredSize OPTIONAL, IN DWORD PropertySheetType){ PROPSHEETHEADERW psh; BOOL ret = FALSE;
TRACE("%p %p %p 0%lx %p 0x%lx\n", DeviceInfoSet, DeviceInfoData, PropertySheetHeader, PropertySheetHeaderPageListSize, RequiredSize, PropertySheetType);
if(PropertySheetHeader) { psh.dwFlags = PropertySheetHeader->dwFlags; psh.phpage = PropertySheetHeader->phpage; psh.nPages = PropertySheetHeader->nPages; }
ret = SetupDiGetClassDevPropertySheetsW(DeviceInfoSet, DeviceInfoData, PropertySheetHeader ? &psh : NULL, PropertySheetHeaderPageListSize, RequiredSize, PropertySheetType); if (ret) { PropertySheetHeader->nPages = psh.nPages; }
TRACE("Returning %d\n", ret); return ret;}
struct ClassDevPropertySheetsData{ LPPROPSHEETHEADERW PropertySheetHeader; DWORD PropertySheetHeaderPageListSize; DWORD NumberOfPages; BOOL DontCancel;};
static BOOL WINAPISETUP_GetClassDevPropertySheetsCallback( IN HPROPSHEETPAGE hPropSheetPage, IN OUT LPARAM lParam){ struct ClassDevPropertySheetsData *PropPageData;
PropPageData = (struct ClassDevPropertySheetsData *)lParam;
PropPageData->NumberOfPages++;
if (PropPageData->PropertySheetHeader->nPages < PropPageData->PropertySheetHeaderPageListSize) { PropPageData->PropertySheetHeader->phpage[PropPageData->PropertySheetHeader->nPages] = hPropSheetPage; PropPageData->PropertySheetHeader->nPages++; return TRUE; }
return PropPageData->DontCancel;}
static DWORDSETUP_GetValueString( IN HKEY hKey, IN LPWSTR lpValueName, OUT LPWSTR *lpString){ LPWSTR lpBuffer; DWORD dwLength = 0; DWORD dwRegType; DWORD rc;
*lpString = NULL;
RegQueryValueExW(hKey, lpValueName, NULL, &dwRegType, NULL, &dwLength);
if (dwLength == 0 || dwRegType != REG_SZ) return ERROR_FILE_NOT_FOUND;
lpBuffer = HeapAlloc(GetProcessHeap(), 0, dwLength + sizeof(WCHAR)); if (lpBuffer == NULL) return ERROR_NOT_ENOUGH_MEMORY;
rc = RegQueryValueExW(hKey, lpValueName, NULL, NULL, (LPBYTE)lpBuffer, &dwLength); if (rc != ERROR_SUCCESS) { HeapFree(GetProcessHeap(), 0, lpBuffer); return rc; }
lpBuffer[dwLength / sizeof(WCHAR)] = UNICODE_NULL;
*lpString = lpBuffer;
return ERROR_SUCCESS;}
staticBOOLSETUP_CallInstaller( IN DI_FUNCTION InstallFunction, IN HDEVINFO DeviceInfoSet, IN PSP_DEVINFO_DATA DeviceInfoData OPTIONAL, IN PSP_ADDPROPERTYPAGE_DATA PageData){ PSP_CLASSINSTALL_HEADER pClassInstallParams = NULL; DWORD dwSize = 0; DWORD dwError; BOOL ret = TRUE;
/* Get the size of the old class install parameters */ if (!SetupDiGetClassInstallParams(DeviceInfoSet, DeviceInfoData, NULL, 0, &dwSize)) { dwError = GetLastError(); if (dwError != ERROR_INSUFFICIENT_BUFFER) { ERR("SetupDiGetClassInstallParams failed (Error %lu)\n", dwError); return FALSE; } }
/* Allocate a buffer for the old class install parameters */ pClassInstallParams = HeapAlloc(GetProcessHeap(), 0, dwSize); if (pClassInstallParams == NULL) { ERR("Failed to allocate the parameters buffer!\n"); return FALSE; }
/* Save the old class install parameters */ if (!SetupDiGetClassInstallParams(DeviceInfoSet, DeviceInfoData, pClassInstallParams, dwSize, &dwSize)) { ERR("SetupDiGetClassInstallParams failed (Error %lu)\n", GetLastError()); ret = FALSE; goto done; }
/* Set the new class install parameters */ if (!SetupDiSetClassInstallParams(DeviceInfoSet, DeviceInfoData, &PageData->ClassInstallHeader, sizeof(SP_ADDPROPERTYPAGE_DATA))) { ERR("SetupDiSetClassInstallParams failed (Error %lu)\n", dwError); ret = FALSE; goto done; }
/* Call the installer */ ret = SetupDiCallClassInstaller(InstallFunction, DeviceInfoSet, DeviceInfoData); if (ret == FALSE) { ERR("SetupDiCallClassInstaller failed\n"); goto done; }
/* Read the new class installer parameters */ if (!SetupDiGetClassInstallParams(DeviceInfoSet, DeviceInfoData, &PageData->ClassInstallHeader, sizeof(SP_ADDPROPERTYPAGE_DATA), NULL)) { ERR("SetupDiGetClassInstallParams failed (Error %lu)\n", GetLastError()); ret = FALSE; goto done; }
done: /* Restore and free the old class install parameters */ if (pClassInstallParams != NULL) { SetupDiSetClassInstallParams(DeviceInfoSet, DeviceInfoData, pClassInstallParams, dwSize);
HeapFree(GetProcessHeap(), 0, pClassInstallParams); }
return ret;}
/*********************************************************************** * SetupDiGetClassDevPropertySheetsW(SETUPAPI.@) */BOOL WINAPISetupDiGetClassDevPropertySheetsW( IN HDEVINFO DeviceInfoSet, IN PSP_DEVINFO_DATA DeviceInfoData OPTIONAL, IN OUT LPPROPSHEETHEADERW PropertySheetHeader, IN DWORD PropertySheetHeaderPageListSize, OUT PDWORD RequiredSize OPTIONAL, IN DWORD PropertySheetType){ struct DeviceInfoSet *devInfoSet = NULL; struct DeviceInfo *devInfo = NULL; BOOL ret = FALSE;
TRACE("%p %p %p 0%lx %p 0x%lx\n", DeviceInfoSet, DeviceInfoData, PropertySheetHeader, PropertySheetHeaderPageListSize, RequiredSize, PropertySheetType);
if (!DeviceInfoSet) SetLastError(ERROR_INVALID_HANDLE); else if (((struct DeviceInfoSet *)DeviceInfoSet)->magic != SETUP_DEVICE_INFO_SET_MAGIC) SetLastError(ERROR_INVALID_HANDLE); else if ((devInfoSet = (struct DeviceInfoSet *)DeviceInfoSet)->magic != SETUP_DEVICE_INFO_SET_MAGIC) SetLastError(ERROR_INVALID_HANDLE); else if (!PropertySheetHeader) SetLastError(ERROR_INVALID_PARAMETER); else if (PropertySheetHeader->dwFlags & PSH_PROPSHEETPAGE) SetLastError(ERROR_INVALID_FLAGS); else if (DeviceInfoData && DeviceInfoData->cbSize != sizeof(SP_DEVINFO_DATA)) SetLastError(ERROR_INVALID_USER_BUFFER); else if (!DeviceInfoData && IsEqualIID(&devInfoSet->ClassGuid, &GUID_NULL)) SetLastError(ERROR_INVALID_PARAMETER); else if (PropertySheetType != DIGCDP_FLAG_ADVANCED && PropertySheetType != DIGCDP_FLAG_BASIC && PropertySheetType != DIGCDP_FLAG_REMOTE_ADVANCED && PropertySheetType != DIGCDP_FLAG_REMOTE_BASIC) SetLastError(ERROR_INVALID_PARAMETER); else { HKEY hKey = INVALID_HANDLE_VALUE; SP_PROPSHEETPAGE_REQUEST Request; LPWSTR PropPageProvider = NULL; HMODULE hModule = NULL; PROPERTY_PAGE_PROVIDER pPropPageProvider = NULL; struct ClassDevPropertySheetsData PropPageData; SP_ADDPROPERTYPAGE_DATA InstallerPropPageData; DWORD InitialNumberOfPages, i; DWORD rc;
if (DeviceInfoData) devInfo = (struct DeviceInfo *)DeviceInfoData->Reserved;
/* Get the class property page provider */ if (devInfoSet->hmodClassPropPageProvider == NULL) { hKey = SetupDiOpenClassRegKeyExW(devInfo ? &devInfo->ClassGuid : &devInfoSet->ClassGuid, KEY_QUERY_VALUE, DIOCR_INSTALLER, NULL/*devInfoSet->MachineName + 2*/, NULL); if (hKey != INVALID_HANDLE_VALUE) { rc = SETUP_GetValueString(hKey, REGSTR_VAL_ENUMPROPPAGES_32, &PropPageProvider); if (rc == ERROR_SUCCESS) { rc = GetFunctionPointer(PropPageProvider, &hModule, (PVOID*)&pPropPageProvider); if (rc != ERROR_SUCCESS) { SetLastError(ERROR_INVALID_PROPPAGE_PROVIDER); goto cleanup; }
devInfoSet->hmodClassPropPageProvider = hModule; devInfoSet->pClassPropPageProvider = pPropPageProvider;
HeapFree(GetProcessHeap(), 0, PropPageProvider); PropPageProvider = NULL; }
RegCloseKey(hKey); hKey = INVALID_HANDLE_VALUE; } }
/* Get the device property page provider */ if (devInfo != NULL && devInfo->hmodDevicePropPageProvider == NULL) { hKey = SETUPDI_OpenDrvKey(devInfoSet->HKLM, devInfo, KEY_QUERY_VALUE); if (hKey != INVALID_HANDLE_VALUE) { rc = SETUP_GetValueString(hKey, REGSTR_VAL_ENUMPROPPAGES_32, &PropPageProvider); if (rc == ERROR_SUCCESS) { rc = GetFunctionPointer(PropPageProvider, &hModule, (PVOID*)&pPropPageProvider); if (rc != ERROR_SUCCESS) { SetLastError(ERROR_INVALID_PROPPAGE_PROVIDER); goto cleanup; }
devInfo->hmodDevicePropPageProvider = hModule; devInfo->pDevicePropPageProvider = pPropPageProvider;
HeapFree(GetProcessHeap(), 0, PropPageProvider); PropPageProvider = NULL; }
RegCloseKey(hKey); hKey = INVALID_HANDLE_VALUE; } }
InitialNumberOfPages = PropertySheetHeader->nPages;
Request.cbSize = sizeof(SP_PROPSHEETPAGE_REQUEST); Request.PageRequested = SPPSR_ENUM_ADV_DEVICE_PROPERTIES; Request.DeviceInfoSet = DeviceInfoSet; Request.DeviceInfoData = DeviceInfoData;
PropPageData.PropertySheetHeader = PropertySheetHeader; PropPageData.PropertySheetHeaderPageListSize = PropertySheetHeaderPageListSize; PropPageData.NumberOfPages = 0; PropPageData.DontCancel = (RequiredSize != NULL) ? TRUE : FALSE;
/* Call the class property page provider */ if (devInfoSet->pClassPropPageProvider != NULL) ((PROPERTY_PAGE_PROVIDER)devInfoSet->pClassPropPageProvider)(&Request, SETUP_GetClassDevPropertySheetsCallback, (LPARAM)&PropPageData);
/* Call the device property page provider */ if (devInfo != NULL && devInfo->pDevicePropPageProvider != NULL) ((PROPERTY_PAGE_PROVIDER)devInfo->pDevicePropPageProvider)(&Request, SETUP_GetClassDevPropertySheetsCallback, (LPARAM)&PropPageData);
/* Call the class installer and add the returned pages */ ZeroMemory(&InstallerPropPageData, sizeof(SP_ADDPROPERTYPAGE_DATA)); InstallerPropPageData.ClassInstallHeader.cbSize = sizeof(SP_CLASSINSTALL_HEADER); InstallerPropPageData.ClassInstallHeader.InstallFunction = DIF_ADDPROPERTYPAGE_ADVANCED; InstallerPropPageData.hwndWizardDlg = PropertySheetHeader->hwndParent;
if (SETUP_CallInstaller(DIF_ADDPROPERTYPAGE_ADVANCED, DeviceInfoSet, DeviceInfoData, &InstallerPropPageData)) { for (i = 0; i < InstallerPropPageData.NumDynamicPages; i++) { if (PropPageData.PropertySheetHeader->nPages < PropertySheetHeaderPageListSize) { PropPageData.PropertySheetHeader->phpage[PropPageData.PropertySheetHeader->nPages] = InstallerPropPageData.DynamicPages[i]; PropPageData.PropertySheetHeader->nPages++; } }
PropPageData.NumberOfPages += InstallerPropPageData.NumDynamicPages; }
if (RequiredSize) *RequiredSize = PropPageData.NumberOfPages;
if (InitialNumberOfPages + PropPageData.NumberOfPages <= PropertySheetHeaderPageListSize) { ret = TRUE; } else { SetLastError(ERROR_INSUFFICIENT_BUFFER); }
cleanup: if (hKey != INVALID_HANDLE_VALUE) RegCloseKey(hKey);
if (PropPageProvider != NULL) HeapFree(GetProcessHeap(), 0, PropPageProvider); }
TRACE("Returning %d\n", ret); return ret;}
/*********************************************************************** * SetupDiGetClassRegistryPropertyA(SETUPAPI.@) */BOOL WINAPISetupDiGetClassRegistryPropertyA( IN CONST GUID *ClassGuid, IN DWORD Property, OUT PDWORD PropertyRegDataType OPTIONAL, OUT PBYTE PropertyBuffer, IN DWORD PropertyBufferSize, OUT PDWORD RequiredSize OPTIONAL, IN PCSTR MachineName OPTIONAL, IN PVOID Reserved){ HMACHINE hMachine = NULL; DWORD PropLength = 0; DWORD Error = ERROR_SUCCESS; CONFIGRET cr;
TRACE("%s %lu %p %p %lu %p %s %p\n", debugstr_guid(ClassGuid), Property, PropertyRegDataType, PropertyBuffer, PropertyBufferSize, RequiredSize, MachineName, Reserved);
if (Reserved != NULL) { SetLastError(ERROR_INVALID_PARAMETER); return FALSE; }
if (MachineName) { cr = CM_Connect_MachineA(MachineName, &hMachine); if (cr != CR_SUCCESS) goto done; }
if (Property >= SPCRP_MAXIMUM_PROPERTY) { cr = CR_INVALID_PROPERTY; goto done; }
PropLength = PropertyBufferSize; cr = CM_Get_Class_Registry_PropertyA((LPGUID)ClassGuid, Property + (CM_DRP_DEVICEDESC - SPDRP_DEVICEDESC), PropertyRegDataType, PropertyBuffer, &PropLength, 0, hMachine); if ((cr == CR_SUCCESS) || (cr == CR_BUFFER_SMALL)) { if (RequiredSize) *RequiredSize = PropLength; }
done: if (cr != CR_SUCCESS) { switch (cr) { case CR_INVALID_DEVINST : Error = ERROR_NO_SUCH_DEVINST; break;
case CR_INVALID_PROPERTY : Error = ERROR_INVALID_REG_PROPERTY; break;
case CR_BUFFER_SMALL : Error = ERROR_INSUFFICIENT_BUFFER; break;
default : Error = GetErrorCodeFromCrCode(cr); } }
if (hMachine != NULL) CM_Disconnect_Machine(hMachine);
SetLastError(Error); return (cr == CR_SUCCESS);}
/*********************************************************************** * SetupDiGetClassRegistryPropertyW(SETUPAPI.@) */BOOL WINAPISetupDiGetClassRegistryPropertyW( IN CONST GUID *ClassGuid, IN DWORD Property, OUT PDWORD PropertyRegDataType OPTIONAL, OUT PBYTE PropertyBuffer, IN DWORD PropertyBufferSize, OUT PDWORD RequiredSize OPTIONAL, IN PCWSTR MachineName OPTIONAL, IN PVOID Reserved){ HMACHINE hMachine = NULL; DWORD PropLength = 0; DWORD Error = ERROR_SUCCESS; CONFIGRET cr;
TRACE("%s %lu %p %p %lu %p %s %p\n", debugstr_guid(ClassGuid), Property, PropertyRegDataType, PropertyBuffer, PropertyBufferSize, RequiredSize, debugstr_w(MachineName), Reserved);
if (Reserved != NULL) { SetLastError(ERROR_INVALID_PARAMETER); return FALSE; }
if (MachineName) { cr = CM_Connect_MachineW(MachineName, &hMachine); if (cr != CR_SUCCESS) goto done; }
if (Property >= SPCRP_MAXIMUM_PROPERTY) { cr = CR_INVALID_PROPERTY; goto done; }
PropLength = PropertyBufferSize; cr = CM_Get_Class_Registry_PropertyW((LPGUID)ClassGuid, Property + (CM_DRP_DEVICEDESC - SPDRP_DEVICEDESC), PropertyRegDataType, PropertyBuffer, &PropLength, 0, hMachine); if ((cr == CR_SUCCESS) || (cr == CR_BUFFER_SMALL)) { if (RequiredSize) *RequiredSize = PropLength; }
done: if (cr != CR_SUCCESS) { switch (cr) { case CR_INVALID_DEVINST : Error = ERROR_NO_SUCH_DEVINST; break;
case CR_INVALID_PROPERTY : Error = ERROR_INVALID_REG_PROPERTY; break;
case CR_BUFFER_SMALL : Error = ERROR_INSUFFICIENT_BUFFER; break;
default : Error = GetErrorCodeFromCrCode(cr); } }
if (hMachine != NULL) CM_Disconnect_Machine(hMachine);
SetLastError(Error); return (cr == CR_SUCCESS);}
/*********************************************************************** * SetupDiSetClassRegistryPropertyA(SETUPAPI.@) */BOOL WINAPISetupDiSetClassRegistryPropertyA( IN CONST GUID *ClassGuid, IN DWORD Property, IN CONST BYTE *PropertyBuffer OPTIONAL, IN DWORD PropertyBufferSize, IN PCSTR MachineName OPTIONAL, IN PVOID Reserved){ HMACHINE hMachine = NULL; DWORD Error = ERROR_SUCCESS; CONFIGRET cr;
TRACE("%s %lu %p %lu %s %p\n", debugstr_guid(ClassGuid), Property, PropertyBuffer, PropertyBufferSize, MachineName, Reserved);
if (Reserved != NULL) { SetLastError(ERROR_INVALID_PARAMETER); return FALSE; }
if (MachineName) { cr = CM_Connect_MachineA(MachineName, &hMachine); if (cr != CR_SUCCESS) goto done; }
if (Property >= SPCRP_MAXIMUM_PROPERTY) { cr = CR_INVALID_PROPERTY; goto done; }
cr = CM_Set_Class_Registry_PropertyA((LPGUID)ClassGuid, Property + (CM_DRP_DEVICEDESC - SPDRP_DEVICEDESC), PropertyBuffer, PropertyBufferSize, 0, hMachine);
done: if (cr != CR_SUCCESS) { switch (cr) { case CR_INVALID_DEVINST: Error = ERROR_NO_SUCH_DEVINST; break;
case CR_INVALID_PROPERTY: Error = ERROR_INVALID_REG_PROPERTY; break;
case CR_BUFFER_SMALL: Error = ERROR_INSUFFICIENT_BUFFER; break;
default : Error = GetErrorCodeFromCrCode(cr); } }
if (hMachine != NULL) CM_Disconnect_Machine(hMachine);
SetLastError(Error); return (cr == CR_SUCCESS);}
/*********************************************************************** * SetupDiSetClassRegistryPropertyW(SETUPAPI.@) */BOOL WINAPISetupDiSetClassRegistryPropertyW( IN CONST GUID *ClassGuid, IN DWORD Property, IN CONST BYTE *PropertyBuffer OPTIONAL, IN DWORD PropertyBufferSize, IN PCWSTR MachineName OPTIONAL, IN PVOID Reserved){ HMACHINE hMachine = NULL; DWORD Error = ERROR_SUCCESS; CONFIGRET cr;
TRACE("%s %lu %p %lu %s %p\n", debugstr_guid(ClassGuid), Property, PropertyBuffer, PropertyBufferSize, debugstr_w(MachineName), Reserved);
if (Reserved != NULL) { SetLastError(ERROR_INVALID_PARAMETER); return FALSE; }
if (MachineName) { cr = CM_Connect_MachineW(MachineName, &hMachine); if (cr != CR_SUCCESS) goto done; }
if (Property >= SPCRP_MAXIMUM_PROPERTY) { cr = CR_INVALID_PROPERTY; goto done; }
cr = CM_Set_Class_Registry_PropertyW((LPGUID)ClassGuid, Property + (CM_DRP_DEVICEDESC - SPDRP_DEVICEDESC), PropertyBuffer, PropertyBufferSize, 0, hMachine);
done: if (cr != CR_SUCCESS) { switch (cr) { case CR_INVALID_DEVINST: Error = ERROR_NO_SUCH_DEVINST; break;
case CR_INVALID_PROPERTY: Error = ERROR_INVALID_REG_PROPERTY; break;
case CR_BUFFER_SMALL: Error = ERROR_INSUFFICIENT_BUFFER; break;
default : Error = GetErrorCodeFromCrCode(cr); } }
if (hMachine != NULL) CM_Disconnect_Machine(hMachine);
SetLastError(Error); return (cr == CR_SUCCESS);}