mirror of
https://github.com/reactos/reactos.git
synced 2024-09-06 19:02:32 +00:00
138 lines
3.5 KiB
C
138 lines
3.5 KiB
C
![]() |
/*
|
||
|
* PROJECT: Safely Remove Hardware Applet
|
||
|
* LICENSE: GPL-2.0-or-later (https://spdx.org/licenses/GPL-2.0-or-later)
|
||
|
* PURPOSE: Device removal support
|
||
|
* COPYRIGHT: Copyright 2023 Thamatip Chitpong <thamatip.chitpong@reactos.org>
|
||
|
*/
|
||
|
|
||
|
#include "hotplug.h"
|
||
|
|
||
|
#include <strsafe.h>
|
||
|
|
||
|
DEVINST
|
||
|
GetDeviceInstForRemoval(
|
||
|
_In_ PHOTPLUG_DATA pHotplugData)
|
||
|
{
|
||
|
HTREEITEM hItem, hParentItem;
|
||
|
TVITEMW tvItem;
|
||
|
|
||
|
hItem = TreeView_GetSelection(pHotplugData->hwndDeviceTree);
|
||
|
if (!hItem)
|
||
|
return 0;
|
||
|
|
||
|
/* Find root item */
|
||
|
hParentItem = TreeView_GetParent(pHotplugData->hwndDeviceTree, hItem);
|
||
|
while (hParentItem)
|
||
|
{
|
||
|
hItem = hParentItem;
|
||
|
hParentItem = TreeView_GetParent(pHotplugData->hwndDeviceTree, hItem);
|
||
|
}
|
||
|
|
||
|
ZeroMemory(&tvItem, sizeof(tvItem));
|
||
|
tvItem.mask = TVIF_PARAM;
|
||
|
tvItem.hItem = hItem;
|
||
|
|
||
|
TreeView_GetItem(pHotplugData->hwndDeviceTree, &tvItem);
|
||
|
|
||
|
return tvItem.lParam;
|
||
|
}
|
||
|
|
||
|
static
|
||
|
VOID
|
||
|
FillConfirmDeviceList(
|
||
|
_In_ HWND hwndCfmDeviceList,
|
||
|
_In_ PHOTPLUG_DATA pHotplugData)
|
||
|
{
|
||
|
LVCOLUMNW lvColumn;
|
||
|
|
||
|
ZeroMemory(&lvColumn, sizeof(lvColumn));
|
||
|
lvColumn.mask = LVCF_FMT;
|
||
|
lvColumn.fmt = LVCFMT_LEFT | LVCFMT_IMAGE;
|
||
|
ListView_InsertColumn(hwndCfmDeviceList, 0, &lvColumn);
|
||
|
|
||
|
CfmListEnumDevices(hwndCfmDeviceList, pHotplugData);
|
||
|
|
||
|
ListView_SetColumnWidth(hwndCfmDeviceList, 0, LVSCW_AUTOSIZE_USEHEADER);
|
||
|
}
|
||
|
|
||
|
static
|
||
|
VOID
|
||
|
SafeRemoveDevice(
|
||
|
_In_ DEVINST DevInst,
|
||
|
_In_opt_ HWND hwndDlg)
|
||
|
{
|
||
|
PNP_VETO_TYPE VetoType = PNP_VetoTypeUnknown;
|
||
|
CONFIGRET cr;
|
||
|
|
||
|
cr = CM_Request_Device_EjectW(DevInst, &VetoType, NULL, 0, 0);
|
||
|
if (cr != CR_SUCCESS && VetoType == PNP_VetoTypeUnknown)
|
||
|
{
|
||
|
LPCWSTR pszFormat = L"";
|
||
|
WCHAR szError[64];
|
||
|
|
||
|
/* NOTES: IDS_EJECT_ERROR_FORMAT resource has to be explicitly NULL-terminated
|
||
|
* so we can use the string directly without having to make a copy of it. */
|
||
|
LoadStringW(hApplet, IDS_EJECT_ERROR_FORMAT, (LPWSTR)&pszFormat, 0);
|
||
|
StringCbPrintfW(szError, sizeof(szError), pszFormat, cr);
|
||
|
|
||
|
MessageBoxW(hwndDlg, szError, NULL, MB_ICONEXCLAMATION | MB_OK);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
INT_PTR
|
||
|
CALLBACK
|
||
|
ConfirmRemovalDlgProc(
|
||
|
_In_ HWND hwndDlg,
|
||
|
_In_ UINT uMsg,
|
||
|
_In_ WPARAM wParam,
|
||
|
_In_ LPARAM lParam)
|
||
|
{
|
||
|
PHOTPLUG_DATA pHotplugData;
|
||
|
|
||
|
pHotplugData = (PHOTPLUG_DATA)GetWindowLongPtrW(hwndDlg, DWLP_USER);
|
||
|
|
||
|
switch (uMsg)
|
||
|
{
|
||
|
case WM_INITDIALOG:
|
||
|
{
|
||
|
HWND hwndDevList;
|
||
|
|
||
|
pHotplugData = (PHOTPLUG_DATA)lParam;
|
||
|
SetWindowLongPtrW(hwndDlg, DWLP_USER, (LONG_PTR)pHotplugData);
|
||
|
|
||
|
hwndDevList = GetDlgItem(hwndDlg, IDC_CONFIRM_STOP_DEVICE_LIST);
|
||
|
|
||
|
ListView_SetImageList(hwndDevList,
|
||
|
pHotplugData->ImageListData.ImageList,
|
||
|
LVSIL_SMALL);
|
||
|
|
||
|
FillConfirmDeviceList(hwndDevList, pHotplugData);
|
||
|
|
||
|
return TRUE;
|
||
|
}
|
||
|
|
||
|
case WM_COMMAND:
|
||
|
{
|
||
|
switch (LOWORD(wParam))
|
||
|
{
|
||
|
case IDOK:
|
||
|
SafeRemoveDevice(GetDeviceInstForRemoval(pHotplugData), hwndDlg);
|
||
|
EndDialog(hwndDlg, LOWORD(wParam));
|
||
|
break;
|
||
|
|
||
|
case IDCANCEL:
|
||
|
EndDialog(hwndDlg, LOWORD(wParam));
|
||
|
break;
|
||
|
}
|
||
|
|
||
|
break;
|
||
|
}
|
||
|
|
||
|
case WM_DESTROY:
|
||
|
SetWindowLongPtrW(hwndDlg, DWLP_USER, (LONG_PTR)NULL);
|
||
|
break;
|
||
|
}
|
||
|
|
||
|
return FALSE;
|
||
|
}
|