eject.c

Go to the documentation of this file.

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