reactos/dll/cpl/hotplug/eject.c

138 lines
3.5 KiB
C
Raw Normal View History

/*
* 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;
}