Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2025-02-22 08:25:03 +00:00
Code Issues Projects Releases Packages Wiki Activity

[HOTPLUG] Display class icons for removable devices

Browse source
This commit is contained in:
Eric Kohl 2020-03-31 22:43:42 +02:00
parent d37799ea3e
commit 45e79efd06
1 changed files with 75 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

82
dll/cpl/hotplug/hotplug.c
View file

@ -11,6 +11,12 @@
#define NDEBUG
#include <debug.h>
typedef struct _HOTPLUG_DATA
{
SP_CLASSIMAGELIST_DATA ImageListData;
} HOTPLUG_DATA, *PHOTPLUG_DATA;
// globals
HINSTANCE hApplet = 0;
@ -23,7 +29,9 @@ APPLET Applets[NUM_APPLETS] =
static
VOID
EnumHotpluggedDevices(HWND hwndDeviceTree)
EnumHotpluggedDevices(
HWND hwndDeviceTree,
PHOTPLUG_DATA pHotplugData)
{
WCHAR szDisplayName[40];
SP_DEVINFO_DATA did = { 0 };
@ -32,6 +40,10 @@ EnumHotpluggedDevices(HWND hwndDeviceTree)
DWORD dwCapabilities, dwSize;
ULONG ulStatus, ulProblem;
TVINSERTSTRUCTW tvItem;
// HTREEITEM hTreeItem;
WCHAR szGuidString[MAX_GUID_STRING_LEN];
GUID ClassGuid;
INT nClassImage;
CONFIGRET cr;
TreeView_DeleteAllItems(hwndDeviceTree);
@ -84,20 +96,42 @@ EnumHotpluggedDevices(HWND hwndDeviceTree)
if (cr != CR_SUCCESS)
wcscpy(szDisplayName, L"Unknown Device");
/* Get the class GUID */
dwSize = sizeof(szGuidString);
cr = CM_Get_DevNode_Registry_Property(did.DevInst,
CM_DRP_CLASSGUID,
NULL,
szGuidString,
&dwSize,
0);
if (cr == CR_SUCCESS)
{
pSetupGuidFromString(szGuidString, &ClassGuid);
}
else
{
// memcpy(&ClassGuid, GUID_DEVCLASS_UNKNOWN, sizeof(GUID));
}
/* Get the image for the class this device is in */
SetupDiGetClassImageIndex(&pHotplugData->ImageListData,
&ClassGuid,
&nClassImage);
/* Add it to the device tree */
ZeroMemory(&tvItem, sizeof(tvItem));
tvItem.hParent = TVI_ROOT;
tvItem.hInsertAfter = TVI_FIRST;
tvItem.item.mask = TVIF_STATE | TVIF_TEXT /*| TVIF_PARAM | TVIF_IMAGE | TVIF_SELECTEDIMAGE */;
tvItem.item.mask = TVIF_STATE | TVIF_TEXT /*| TVIF_PARAM*/ | TVIF_IMAGE | TVIF_SELECTEDIMAGE;
tvItem.item.state = TVIS_EXPANDED;
tvItem.item.stateMask = TVIS_EXPANDED;
tvItem.item.pszText = szDisplayName;
// tvItem.item.iImage = IMAGE_SOUND_SECTION;
// tvItem.item.iSelectedImage = IMAGE_SOUND_SECTION;
tvItem.item.iImage = nClassImage;
tvItem.item.iSelectedImage = nClassImage;
tvItem.item.lParam = (LPARAM)NULL;
/*hTreeItem = */ TreeView_InsertItem(hwndDeviceTree, &tvItem);
/*hTreeItem =*/ TreeView_InsertItem(hwndDeviceTree, &tvItem);
@ -116,12 +150,31 @@ SafeRemovalDlgProc(
WPARAM wParam,
LPARAM lParam)
{
PHOTPLUG_DATA pHotplugData;
UNREFERENCED_PARAMETER(lParam);
pHotplugData = (PHOTPLUG_DATA)GetWindowLongPtr(hwndDlg, DWLP_USER);
switch (uMsg)
{
case WM_INITDIALOG:
EnumHotpluggedDevices(GetDlgItem(hwndDlg, IDC_SAFE_REMOVE_DEVICE_TREE));
pHotplugData = HeapAlloc(GetProcessHeap(), 0, sizeof(HOTPLUG_DATA));
if (pHotplugData != NULL)
{
SetWindowLongPtr(hwndDlg, DWLP_USER, (LONG_PTR)pHotplugData);
pHotplugData->ImageListData.cbSize = sizeof(pHotplugData->ImageListData);
SetupDiGetClassImageList(&pHotplugData->ImageListData);
TreeView_SetImageList(GetDlgItem(hwndDlg, IDC_SAFE_REMOVE_DEVICE_TREE),
pHotplugData->ImageListData.ImageList,
TVSIL_NORMAL);
EnumHotpluggedDevices(GetDlgItem(hwndDlg, IDC_SAFE_REMOVE_DEVICE_TREE),
pHotplugData);
}
return TRUE;
case WM_COMMAND:
@ -148,7 +201,12 @@ SafeRemovalDlgProc(
if (wParam == 1)
{
KillTimer(hwndDlg, 1);
EnumHotpluggedDevices(GetDlgItem(hwndDlg, IDC_SAFE_REMOVE_DEVICE_TREE));
if (pHotplugData != NULL)
{
EnumHotpluggedDevices(GetDlgItem(hwndDlg, IDC_SAFE_REMOVE_DEVICE_TREE),
pHotplugData);
}
}
break;
@ -156,6 +214,16 @@ SafeRemovalDlgProc(
KillTimer(hwndDlg, 1);
EndDialog(hwndDlg, TRUE);
break;
case WM_DESTROY:
if (pHotplugData != NULL)
{
SetupDiDestroyClassImageList(&pHotplugData->ImageListData);
HeapFree(GetProcessHeap(), 0, pHotplugData);
SetWindowLongPtr(hwndDlg, DWLP_USER, (LONG_PTR)NULL);
}
break;
}
return FALSE;