2013-02-17 15:51:07 +00:00
|
|
|
/*
|
|
|
|
* PROJECT: Safely Remove Hardware Applet
|
|
|
|
* LICENSE: GPL - See COPYING in the top level directory
|
|
|
|
* FILE: dll/cpl/hotplug/hotplug.c
|
|
|
|
* PURPOSE: applet initialization
|
2013-09-15 13:06:05 +00:00
|
|
|
* PROGRAMMERS: Johannes Anderwald (johannes.anderwald@reactos.org)
|
2013-02-17 15:51:07 +00:00
|
|
|
*/
|
|
|
|
|
|
|
|
#include "hotplug.h"
|
|
|
|
|
2020-03-29 22:37:45 +00:00
|
|
|
#define NDEBUG
|
|
|
|
#include <debug.h>
|
|
|
|
|
2020-03-31 20:43:42 +00:00
|
|
|
typedef struct _HOTPLUG_DATA
|
|
|
|
{
|
|
|
|
SP_CLASSIMAGELIST_DATA ImageListData;
|
|
|
|
} HOTPLUG_DATA, *PHOTPLUG_DATA;
|
|
|
|
|
|
|
|
|
2013-02-17 15:51:07 +00:00
|
|
|
// globals
|
|
|
|
HINSTANCE hApplet = 0;
|
|
|
|
|
|
|
|
/* Applets */
|
|
|
|
APPLET Applets[NUM_APPLETS] =
|
|
|
|
{
|
2020-03-29 22:37:45 +00:00
|
|
|
{IDI_HOTPLUG, IDS_CPLNAME, IDS_CPLDESCRIPTION, InitApplet}
|
2013-02-17 15:51:07 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
|
2020-03-30 21:11:44 +00:00
|
|
|
static
|
|
|
|
VOID
|
2020-03-31 20:43:42 +00:00
|
|
|
EnumHotpluggedDevices(
|
|
|
|
HWND hwndDeviceTree,
|
|
|
|
PHOTPLUG_DATA pHotplugData)
|
2020-03-30 21:11:44 +00:00
|
|
|
{
|
|
|
|
WCHAR szDisplayName[40];
|
|
|
|
SP_DEVINFO_DATA did = { 0 };
|
|
|
|
HDEVINFO hdev;
|
|
|
|
int idev;
|
|
|
|
DWORD dwCapabilities, dwSize;
|
|
|
|
ULONG ulStatus, ulProblem;
|
|
|
|
TVINSERTSTRUCTW tvItem;
|
2020-03-31 20:43:42 +00:00
|
|
|
// HTREEITEM hTreeItem;
|
|
|
|
WCHAR szGuidString[MAX_GUID_STRING_LEN];
|
|
|
|
GUID ClassGuid;
|
|
|
|
INT nClassImage;
|
2020-03-30 21:11:44 +00:00
|
|
|
CONFIGRET cr;
|
|
|
|
|
|
|
|
TreeView_DeleteAllItems(hwndDeviceTree);
|
|
|
|
|
|
|
|
hdev = SetupDiGetClassDevs(NULL, NULL, 0, DIGCF_ALLCLASSES | DIGCF_PRESENT);
|
|
|
|
if (hdev == INVALID_HANDLE_VALUE)
|
|
|
|
return;
|
|
|
|
|
|
|
|
did.cbSize = sizeof(did);
|
|
|
|
|
|
|
|
/* Enumerate all the attached devices */
|
|
|
|
for (idev = 0; SetupDiEnumDeviceInfo(hdev, idev, &did); idev++)
|
|
|
|
{
|
|
|
|
ulStatus = 0;
|
|
|
|
ulProblem = 0;
|
|
|
|
|
|
|
|
cr = CM_Get_DevNode_Status(&ulStatus,
|
|
|
|
&ulProblem,
|
|
|
|
did.DevInst,
|
|
|
|
0);
|
|
|
|
if (cr != CR_SUCCESS)
|
|
|
|
continue;
|
|
|
|
|
|
|
|
dwCapabilities = 0,
|
|
|
|
dwSize = sizeof(dwCapabilities);
|
|
|
|
cr = CM_Get_DevNode_Registry_Property(did.DevInst,
|
|
|
|
CM_DRP_CAPABILITIES,
|
|
|
|
NULL,
|
|
|
|
&dwCapabilities,
|
|
|
|
&dwSize,
|
|
|
|
0);
|
|
|
|
if (cr != CR_SUCCESS)
|
|
|
|
continue;
|
|
|
|
|
|
|
|
/* Add devices that require safe removal to the device tree */
|
|
|
|
if ( (dwCapabilities & CM_DEVCAP_REMOVABLE) &&
|
|
|
|
!(dwCapabilities & CM_DEVCAP_DOCKDEVICE) &&
|
|
|
|
!(dwCapabilities & CM_DEVCAP_SURPRISEREMOVALOK) &&
|
|
|
|
((dwCapabilities & CM_DEVCAP_EJECTSUPPORTED) || (ulStatus & DN_DISABLEABLE)) &&
|
|
|
|
ulProblem == 0)
|
|
|
|
{
|
|
|
|
/* Get the device description */
|
|
|
|
dwSize = sizeof(szDisplayName);
|
|
|
|
cr = CM_Get_DevNode_Registry_Property(did.DevInst,
|
|
|
|
CM_DRP_DEVICEDESC,
|
|
|
|
NULL,
|
|
|
|
szDisplayName,
|
|
|
|
&dwSize,
|
|
|
|
0);
|
|
|
|
if (cr != CR_SUCCESS)
|
|
|
|
wcscpy(szDisplayName, L"Unknown Device");
|
|
|
|
|
2020-03-31 20:43:42 +00:00
|
|
|
/* 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);
|
|
|
|
|
2020-03-30 21:11:44 +00:00
|
|
|
/* Add it to the device tree */
|
|
|
|
ZeroMemory(&tvItem, sizeof(tvItem));
|
|
|
|
tvItem.hParent = TVI_ROOT;
|
|
|
|
tvItem.hInsertAfter = TVI_FIRST;
|
|
|
|
|
2020-03-31 20:43:42 +00:00
|
|
|
tvItem.item.mask = TVIF_STATE | TVIF_TEXT /*| TVIF_PARAM*/ | TVIF_IMAGE | TVIF_SELECTEDIMAGE;
|
2020-03-30 21:11:44 +00:00
|
|
|
tvItem.item.state = TVIS_EXPANDED;
|
|
|
|
tvItem.item.stateMask = TVIS_EXPANDED;
|
|
|
|
tvItem.item.pszText = szDisplayName;
|
2020-03-31 20:43:42 +00:00
|
|
|
tvItem.item.iImage = nClassImage;
|
|
|
|
tvItem.item.iSelectedImage = nClassImage;
|
2020-03-30 21:11:44 +00:00
|
|
|
tvItem.item.lParam = (LPARAM)NULL;
|
|
|
|
|
2020-03-31 20:43:42 +00:00
|
|
|
/*hTreeItem =*/ TreeView_InsertItem(hwndDeviceTree, &tvItem);
|
2020-03-30 21:11:44 +00:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
SetupDiDestroyDeviceInfoList(hdev);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2020-03-29 22:37:45 +00:00
|
|
|
INT_PTR
|
|
|
|
CALLBACK
|
|
|
|
SafeRemovalDlgProc(
|
|
|
|
HWND hwndDlg,
|
|
|
|
UINT uMsg,
|
|
|
|
WPARAM wParam,
|
|
|
|
LPARAM lParam)
|
|
|
|
{
|
2020-03-31 20:43:42 +00:00
|
|
|
PHOTPLUG_DATA pHotplugData;
|
|
|
|
|
2020-03-29 22:37:45 +00:00
|
|
|
UNREFERENCED_PARAMETER(lParam);
|
|
|
|
|
2020-03-31 20:43:42 +00:00
|
|
|
|
|
|
|
pHotplugData = (PHOTPLUG_DATA)GetWindowLongPtr(hwndDlg, DWLP_USER);
|
|
|
|
|
2020-03-29 22:37:45 +00:00
|
|
|
switch (uMsg)
|
|
|
|
{
|
|
|
|
case WM_INITDIALOG:
|
2020-03-31 20:43:42 +00:00
|
|
|
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);
|
|
|
|
}
|
2020-03-29 22:37:45 +00:00
|
|
|
return TRUE;
|
|
|
|
|
|
|
|
case WM_COMMAND:
|
|
|
|
switch (LOWORD(wParam))
|
|
|
|
{
|
|
|
|
case IDCLOSE:
|
2020-03-30 21:11:44 +00:00
|
|
|
KillTimer(hwndDlg, 1);
|
2020-03-29 22:37:45 +00:00
|
|
|
EndDialog(hwndDlg, TRUE);
|
|
|
|
break;
|
|
|
|
|
|
|
|
}
|
|
|
|
break;
|
2020-03-30 21:11:44 +00:00
|
|
|
|
|
|
|
case WM_DEVICECHANGE:
|
|
|
|
switch (wParam)
|
|
|
|
{
|
|
|
|
case DBT_DEVNODES_CHANGED:
|
|
|
|
SetTimer(hwndDlg, 1, 500, NULL);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
|
|
|
|
case WM_TIMER:
|
|
|
|
if (wParam == 1)
|
|
|
|
{
|
|
|
|
KillTimer(hwndDlg, 1);
|
2020-03-31 20:43:42 +00:00
|
|
|
|
|
|
|
if (pHotplugData != NULL)
|
|
|
|
{
|
|
|
|
EnumHotpluggedDevices(GetDlgItem(hwndDlg, IDC_SAFE_REMOVE_DEVICE_TREE),
|
|
|
|
pHotplugData);
|
|
|
|
}
|
2020-03-30 21:11:44 +00:00
|
|
|
}
|
|
|
|
break;
|
|
|
|
|
|
|
|
case WM_CLOSE:
|
|
|
|
KillTimer(hwndDlg, 1);
|
|
|
|
EndDialog(hwndDlg, TRUE);
|
|
|
|
break;
|
2020-03-31 20:43:42 +00:00
|
|
|
|
|
|
|
case WM_DESTROY:
|
|
|
|
if (pHotplugData != NULL)
|
|
|
|
{
|
|
|
|
SetupDiDestroyClassImageList(&pHotplugData->ImageListData);
|
|
|
|
|
|
|
|
HeapFree(GetProcessHeap(), 0, pHotplugData);
|
|
|
|
SetWindowLongPtr(hwndDlg, DWLP_USER, (LONG_PTR)NULL);
|
|
|
|
}
|
|
|
|
break;
|
2020-03-29 22:37:45 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
return FALSE;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2013-02-17 15:51:07 +00:00
|
|
|
LONG
|
|
|
|
APIENTRY
|
|
|
|
InitApplet(
|
2020-03-29 22:37:45 +00:00
|
|
|
HWND hwnd,
|
2013-02-17 15:51:07 +00:00
|
|
|
UINT uMsg,
|
|
|
|
LPARAM wParam,
|
|
|
|
LPARAM lParam)
|
|
|
|
{
|
2020-03-29 22:37:45 +00:00
|
|
|
DPRINT1("InitApplet()\n");
|
|
|
|
|
|
|
|
DialogBox(hApplet,
|
|
|
|
MAKEINTRESOURCE(IDD_SAFE_REMOVE_HARDWARE_DIALOG),
|
|
|
|
hwnd,
|
|
|
|
SafeRemovalDlgProc);
|
|
|
|
|
2013-02-17 15:51:07 +00:00
|
|
|
// TODO
|
2020-03-29 22:37:45 +00:00
|
|
|
return TRUE;
|
2013-02-17 15:51:07 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
LONG
|
|
|
|
CALLBACK
|
|
|
|
CPlApplet(
|
|
|
|
HWND hwndCPl,
|
|
|
|
UINT uMsg,
|
|
|
|
LPARAM lParam1,
|
|
|
|
LPARAM lParam2)
|
|
|
|
{
|
2020-03-29 22:37:45 +00:00
|
|
|
UINT i = (UINT)lParam1;
|
|
|
|
|
2013-02-17 15:51:07 +00:00
|
|
|
switch(uMsg)
|
|
|
|
{
|
|
|
|
case CPL_INIT:
|
|
|
|
return TRUE;
|
2020-03-29 22:37:45 +00:00
|
|
|
|
2013-02-17 15:51:07 +00:00
|
|
|
case CPL_GETCOUNT:
|
|
|
|
return NUM_APPLETS;
|
2020-03-29 22:37:45 +00:00
|
|
|
|
2013-02-17 15:51:07 +00:00
|
|
|
case CPL_INQUIRE:
|
2020-03-29 22:37:45 +00:00
|
|
|
{
|
|
|
|
CPLINFO *CPlInfo = (CPLINFO*)lParam2;
|
|
|
|
CPlInfo->lData = 0;
|
|
|
|
CPlInfo->idIcon = Applets[i].idIcon;
|
|
|
|
CPlInfo->idName = Applets[i].idName;
|
|
|
|
CPlInfo->idInfo = Applets[i].idDescription;
|
|
|
|
}
|
2013-02-17 15:51:07 +00:00
|
|
|
break;
|
2020-03-29 22:37:45 +00:00
|
|
|
|
2013-02-17 15:51:07 +00:00
|
|
|
case CPL_DBLCLK:
|
2020-03-29 22:37:45 +00:00
|
|
|
Applets[i].AppletProc(hwndCPl, uMsg, lParam1, lParam2);
|
2013-02-17 15:51:07 +00:00
|
|
|
break;
|
2020-03-29 22:37:45 +00:00
|
|
|
|
|
|
|
case CPL_STARTWPARMSW:
|
|
|
|
return Applets[i].AppletProc(hwndCPl, uMsg, lParam1, lParam2);
|
2013-02-17 15:51:07 +00:00
|
|
|
}
|
|
|
|
return FALSE;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
INT
|
|
|
|
WINAPI
|
|
|
|
DllMain(
|
|
|
|
HINSTANCE hinstDLL,
|
|
|
|
DWORD dwReason,
|
|
|
|
LPVOID lpvReserved)
|
|
|
|
{
|
|
|
|
UNREFERENCED_PARAMETER(lpvReserved);
|
|
|
|
|
2020-03-29 22:37:45 +00:00
|
|
|
switch (dwReason)
|
2013-02-17 15:51:07 +00:00
|
|
|
{
|
2020-03-29 22:37:45 +00:00
|
|
|
case DLL_PROCESS_ATTACH:
|
|
|
|
case DLL_THREAD_ATTACH:
|
|
|
|
hApplet = hinstDLL;
|
|
|
|
break;
|
2013-02-17 15:51:07 +00:00
|
|
|
}
|
|
|
|
return TRUE;
|
|
|
|
}
|