hotplug.c File Reference

#include "hotplug.h"
#include <initguid.h>
#include <devguid.h>
#include <debug.h>

Include dependency graph for hotplug.c:

Go to the source code of this file.

Classes
struct	_HOTPLUG_DATA

Macros
#define	NDEBUG

Typedefs
typedef struct _HOTPLUG_DATA	HOTPLUG_DATA
typedef struct _HOTPLUG_DATA *	PHOTPLUG_DATA

Functions
static DWORD	GetHotPlugFlags (VOID)
static DWORD	SetHotPlugFlags (_In_ DWORD dwFlags)
static HTREEITEM	InsertDeviceTreeItem (_In_ HWND hwndDeviceTree, _In_ HTREEITEM hParent, _In_ DWORD DevInst, _In_ PHOTPLUG_DATA pHotplugData)
static VOID	RecursiveInsertSubDevices (_In_ HWND hwndDeviceTree, _In_ HTREEITEM hParentItem, _In_ DWORD ParentDevInst, _In_ PHOTPLUG_DATA pHotplugData)
static VOID	EnumHotpluggedDevices (HWND hwndDeviceTree, PHOTPLUG_DATA pHotplugData)
static VOID	UpdateButtons (HWND hwndDlg)
static VOID	ShowContextMenu (HWND hwndDlg, HWND hwndTreeView, PHOTPLUG_DATA pHotplugData)
INT_PTR CALLBACK	SafeRemovalDlgProc (HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
LONG APIENTRY	InitApplet (HWND hwnd, UINT uMsg, LPARAM wParam, LPARAM lParam)
LONG CALLBACK	CPlApplet (HWND hwndCPl, UINT uMsg, LPARAM lParam1, LPARAM lParam2)
INT WINAPI	DllMain (HINSTANCE hinstDLL, DWORD dwReason, LPVOID lpvReserved)

Variables
HINSTANCE	hApplet = 0
APPLET	Applets [NUM_APPLETS]

Macro Definition Documentation

NDEBUG

#define NDEBUG

Definition at line 14 of file hotplug.c.

Typedef Documentation

HOTPLUG_DATA

typedef struct _HOTPLUG_DATA HOTPLUG_DATA

PHOTPLUG_DATA

typedef struct _HOTPLUG_DATA * PHOTPLUG_DATA

Function Documentation

CPlApplet()

LONG CALLBACK CPlApplet	(	HWND	hwndCPl,
		UINT	uMsg,
		LPARAM	lParam1,
		LPARAM	lParam2
	)

Definition at line 534 of file hotplug.c.

{
    UINT i = (UINT)lParam1;
 
    switch(uMsg)
    {
        case CPL_INIT:
            return TRUE;
 
        case CPL_GETCOUNT:
            return NUM_APPLETS;
 
        case CPL_INQUIRE:
            if (i < NUM_APPLETS)
            {
                CPLINFO *CPlInfo = (CPLINFO*)lParam2;
                CPlInfo->lData = 0;
                CPlInfo->idIcon = Applets[i].idIcon;
                CPlInfo->idName = Applets[i].idName;
                CPlInfo->idInfo = Applets[i].idDescription;
            }
            else
            {
                return TRUE;
            }
            break;
 
        case CPL_DBLCLK:
            if (i < NUM_APPLETS)
                Applets[i].AppletProc(hwndCPl, uMsg, lParam1, lParam2);
            else
                return TRUE;
            break;
 
        case CPL_STARTWPARMSW:
            if (i < NUM_APPLETS)
                return Applets[i].AppletProc(hwndCPl, uMsg, lParam1, lParam2);
            break;
    }
    return FALSE;
}

DllMain()

INT WINAPI DllMain	(	HINSTANCE	hinstDLL,
		DWORD	dwReason,
		LPVOID	lpvReserved
	)

Definition at line 583 of file hotplug.c.

{
    UNREFERENCED_PARAMETER(lpvReserved);
 
    switch (dwReason)
    {
        case DLL_PROCESS_ATTACH:
        case DLL_THREAD_ATTACH:
            hApplet = hinstDLL;
            break;
    }
    return TRUE;
}

EnumHotpluggedDevices()

static VOID EnumHotpluggedDevices ( HWND  hwndDeviceTree, PHOTPLUG_DATA  pHotplugData  )

static

Definition at line 231 of file hotplug.c.

{
    SP_DEVINFO_DATA did = { 0 };
    HDEVINFO hdev;
    int idev;
    DWORD dwCapabilities, dwSize;
    ULONG ulStatus, ulProblem;
    HTREEITEM hTreeItem;
    CONFIGRET cr;
 
    DPRINT1("EnumHotpluggedDevices()\n");
 
    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)
        {
            hTreeItem = InsertDeviceTreeItem(hwndDeviceTree,
                                             TVI_ROOT,
                                             did.DevInst,
                                             pHotplugData);
 
            if ((hTreeItem != NULL) && (pHotplugData->dwFlags & HOTPLUG_DISPLAY_DEVICE_COMPONENTS))
            {
                RecursiveInsertSubDevices(hwndDeviceTree,
                                          hTreeItem,
                                          did.DevInst,
                                          pHotplugData);
            }
        }
    }
 
    SetupDiDestroyDeviceInfoList(hdev);
}

Referenced by SafeRemovalDlgProc().

GetHotPlugFlags()

static DWORD GetHotPlugFlags ( VOID  )

static

Definition at line 39 of file hotplug.c.

{
    HKEY hKey = NULL;
    DWORD dwFlags = 0;
    DWORD dwSize, dwError;
 
    dwError = RegOpenKeyExW(HKEY_CURRENT_USER,
                            REGSTR_PATH_SYSTRAY,
                            0,
                            KEY_READ,
                            &hKey);
    if (dwError != ERROR_SUCCESS)
        goto done;
 
    dwSize = sizeof(dwFlags);
    dwError = RegQueryValueExW(hKey,
                               L"HotPlugFlags",
                               NULL,
                               NULL,
                               (LPBYTE)&dwFlags,
                               &dwSize);
    if (dwError != ERROR_SUCCESS)
        dwFlags = 0;
 
done:
    if (hKey != NULL)
        RegCloseKey(hKey);
 
    return dwFlags;
}

Referenced by SafeRemovalDlgProc().

InitApplet()

LONG APIENTRY InitApplet	(	HWND	hwnd,
		UINT	uMsg,
		LPARAM	wParam,
		LPARAM	lParam
	)

Definition at line 514 of file hotplug.c.

{
    DPRINT("InitApplet()\n");
 
    DialogBox(hApplet,
              MAKEINTRESOURCE(IDD_SAFE_REMOVE_HARDWARE_DIALOG),
              hwnd,
              SafeRemovalDlgProc);
 
    // TODO
    return TRUE;
}

InsertDeviceTreeItem()

static HTREEITEM InsertDeviceTreeItem ( _In_ HWND  hwndDeviceTree, _In_ HTREEITEM  hParent, _In_ DWORD  DevInst, _In_ PHOTPLUG_DATA  pHotplugData  )

static

Definition at line 108 of file hotplug.c.

{
    WCHAR szDisplayName[40];
    WCHAR szGuidString[MAX_GUID_STRING_LEN];
    TVINSERTSTRUCTW tvItem;
    GUID ClassGuid;
    INT nClassImage;
    DWORD dwSize;
    CONFIGRET cr;
 
    /* Get the device description */
    dwSize = sizeof(szDisplayName);
    cr = CM_Get_DevNode_Registry_Property(DevInst,
                                          CM_DRP_DEVICEDESC,
                                          NULL,
                                          szDisplayName,
                                          &dwSize,
                                          0);
    if (cr != CR_SUCCESS)
        wcscpy(szDisplayName, L"Unknown Device");
 
    /* Get the class GUID */
    dwSize = sizeof(szGuidString);
    cr = CM_Get_DevNode_Registry_Property(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 = hParent;
    tvItem.hInsertAfter = TVI_LAST;
 
    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 = nClassImage;
    tvItem.item.iSelectedImage = nClassImage;
    tvItem.item.lParam = (LPARAM)NULL;
 
    return TreeView_InsertItem(hwndDeviceTree, &tvItem);
}

Referenced by EnumHotpluggedDevices(), and RecursiveInsertSubDevices().

RecursiveInsertSubDevices()

static VOID RecursiveInsertSubDevices ( _In_ HWND  hwndDeviceTree, _In_ HTREEITEM  hParentItem, _In_ DWORD  ParentDevInst, _In_ PHOTPLUG_DATA  pHotplugData  )

static

Definition at line 174 of file hotplug.c.

{
    HTREEITEM hTreeItem;
    DEVINST ChildDevInst;
    CONFIGRET cr;
 
    DPRINT("RecursiveInsertSubDevices()\n");
 
    cr = CM_Get_Child(&ChildDevInst, ParentDevInst, 0);
    if (cr != CR_SUCCESS)
    {
        DPRINT("No child! %lu\n", cr);
        return;
    }
 
    hTreeItem = InsertDeviceTreeItem(hwndDeviceTree,
                                     hParentItem,
                                     ChildDevInst,
                                     pHotplugData);
    if (hTreeItem != NULL)
    {
        RecursiveInsertSubDevices(hwndDeviceTree,
                                  hTreeItem,
                                  ChildDevInst,
                                  pHotplugData);
    }
 
    for (;;)
    {
        cr = CM_Get_Sibling(&ChildDevInst, ChildDevInst, 0);
        if (cr != CR_SUCCESS)
        {
            DPRINT("No sibling! %lu\n", cr);
            return;
        }
 
        hTreeItem = InsertDeviceTreeItem(hwndDeviceTree,
                                         hParentItem,
                                         ChildDevInst,
                                         pHotplugData);
        if (hTreeItem != NULL)
        {
            RecursiveInsertSubDevices(hwndDeviceTree,
                                      hTreeItem,
                                      ChildDevInst,
                                      pHotplugData);
        }
    }
}

Referenced by EnumHotpluggedDevices(), and RecursiveInsertSubDevices().

SafeRemovalDlgProc()

INT_PTR CALLBACK SafeRemovalDlgProc	(	HWND	hwndDlg,
		UINT	uMsg,
		WPARAM	wParam,
		LPARAM	lParam
	)

Definition at line 349 of file hotplug.c.

{
    PHOTPLUG_DATA pHotplugData;
 
    pHotplugData = (PHOTPLUG_DATA)GetWindowLongPtr(hwndDlg, DWLP_USER);
 
    switch (uMsg)
    {
        case WM_INITDIALOG:
            pHotplugData = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(HOTPLUG_DATA));
            if (pHotplugData != NULL)
            {
                WCHAR szWindowTitle[MAX_PATH];
 
                SetWindowLongPtr(hwndDlg, DWLP_USER, (LONG_PTR)pHotplugData);
 
                if (LoadStringW(hApplet,
                                IDS_CPLNAME,
                                szWindowTitle,
                                ARRAYSIZE(szWindowTitle)))
                {
                    SetWindowTextW(hwndDlg, szWindowTitle);
                }
 
                pHotplugData->hIcon = (HICON)LoadImageW(hApplet,
                                                        MAKEINTRESOURCEW(IDI_HOTPLUG),
                                                        IMAGE_ICON,
                                                        GetSystemMetrics(SM_CXICON),
                                                        GetSystemMetrics(SM_CYICON),
                                                        LR_DEFAULTCOLOR);
                pHotplugData->hIconSm = (HICON)LoadImageW(hApplet,
                                                          MAKEINTRESOURCEW(IDI_HOTPLUG),
                                                          IMAGE_ICON,
                                                          GetSystemMetrics(SM_CXSMICON),
                                                          GetSystemMetrics(SM_CYSMICON),
                                                          LR_DEFAULTCOLOR);
                SendMessageW(hwndDlg, WM_SETICON, ICON_BIG, (LPARAM)pHotplugData->hIcon);
                SendMessageW(hwndDlg, WM_SETICON, ICON_SMALL, (LPARAM)pHotplugData->hIconSm);
 
                pHotplugData->ImageListData.cbSize = sizeof(pHotplugData->ImageListData);
                SetupDiGetClassImageList(&pHotplugData->ImageListData);
 
                pHotplugData->hPopupMenu = LoadMenu(hApplet, MAKEINTRESOURCE(IDM_POPUP_DEVICE_TREE));
 
                pHotplugData->dwFlags = GetHotPlugFlags();
 
                if (pHotplugData->dwFlags & HOTPLUG_DISPLAY_DEVICE_COMPONENTS)
                    Button_SetCheck(GetDlgItem(hwndDlg, IDC_SAFE_REMOVE_DISPLAY_COMPONENTS), BST_CHECKED);
 
                TreeView_SetImageList(GetDlgItem(hwndDlg, IDC_SAFE_REMOVE_DEVICE_TREE),
                                      pHotplugData->ImageListData.ImageList,
                                      TVSIL_NORMAL);
 
                EnumHotpluggedDevices(GetDlgItem(hwndDlg, IDC_SAFE_REMOVE_DEVICE_TREE),
                                      pHotplugData);
                UpdateButtons(hwndDlg);
            }
            return TRUE;
 
        case WM_COMMAND:
            switch (LOWORD(wParam))
            {
                case IDCLOSE:
                    KillTimer(hwndDlg, 1);
                    EndDialog(hwndDlg, TRUE);
                    break;
 
                case IDC_SAFE_REMOVE_DISPLAY_COMPONENTS:
                    if (HIWORD(wParam) == BN_CLICKED)
                    {
                        if (pHotplugData != NULL)
                        {
                            if (Button_GetCheck(GetDlgItem(hwndDlg, IDC_SAFE_REMOVE_DISPLAY_COMPONENTS)) == BST_CHECKED)
                                pHotplugData->dwFlags |= HOTPLUG_DISPLAY_DEVICE_COMPONENTS;
                            else
                                pHotplugData->dwFlags &= ~HOTPLUG_DISPLAY_DEVICE_COMPONENTS;
 
                            SetHotPlugFlags(pHotplugData->dwFlags);
 
                            EnumHotpluggedDevices(GetDlgItem(hwndDlg, IDC_SAFE_REMOVE_DEVICE_TREE),
                                                  pHotplugData);
                        }
                    }
                    break;
            }
            break;
 
        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);
 
                if (pHotplugData != NULL)
                {
                    EnumHotpluggedDevices(GetDlgItem(hwndDlg, IDC_SAFE_REMOVE_DEVICE_TREE),
                                          pHotplugData);
                    UpdateButtons(hwndDlg);
                }
            }
            break;
 
        case WM_NOTIFY:
            if (((LPNMHDR)lParam)->idFrom == IDC_SAFE_REMOVE_DEVICE_TREE)
            {
                if (((LPNMHDR)lParam)->code == NM_RCLICK)
                {
                    if (pHotplugData != NULL)
                    {
                        ShowContextMenu(hwndDlg,
                                        ((LPNMHDR)lParam)->hwndFrom,
                                        pHotplugData);
                        return TRUE;
                    }
                }
            }
            break;
 
        case WM_CLOSE:
            KillTimer(hwndDlg, 1);
            EndDialog(hwndDlg, TRUE);
            break;
 
        case WM_DESTROY:
            if (pHotplugData != NULL)
            {
                if (pHotplugData->hPopupMenu != NULL)
                    DestroyMenu(pHotplugData->hPopupMenu);
 
                SetupDiDestroyClassImageList(&pHotplugData->ImageListData);
 
                if (pHotplugData->hIconSm)
                {
                    DestroyIcon(pHotplugData->hIconSm);
                }
 
                if (pHotplugData->hIcon)
                {
                    DestroyIcon(pHotplugData->hIcon);
                }
 
                HeapFree(GetProcessHeap(), 0, pHotplugData);
                SetWindowLongPtr(hwndDlg, DWLP_USER, (LONG_PTR)NULL);
            }
            break;
    }
 
    return FALSE;
}

Referenced by InitApplet().

SetHotPlugFlags()

static DWORD SetHotPlugFlags ( _In_ DWORD  dwFlags )

static

Definition at line 73 of file hotplug.c.

{
    HKEY hKey = NULL;
    DWORD dwError;
 
    dwError = RegCreateKeyExW(HKEY_CURRENT_USER,
                              REGSTR_PATH_SYSTRAY,
                              0,
                              NULL,
                              REG_OPTION_NON_VOLATILE,
                              KEY_WRITE,
                              NULL,
                              &hKey,
                              NULL);
    if (dwError != ERROR_SUCCESS)
        goto done;
 
    dwError = RegSetValueExW(hKey,
                             L"HotPlugFlags",
                             0,
                             REG_DWORD,
                             (LPBYTE)&dwFlags,
                             sizeof(dwFlags));
 
done:
    if (hKey != NULL)
        RegCloseKey(hKey);
 
    return dwError;
}

Referenced by SafeRemovalDlgProc().

ShowContextMenu()

static VOID ShowContextMenu ( HWND  hwndDlg, HWND  hwndTreeView, PHOTPLUG_DATA  pHotplugData  )

static

Definition at line 319 of file hotplug.c.

{
    HTREEITEM hTreeItem;
    RECT rc;
    POINT pt;
 
    hTreeItem = TreeView_GetSelection(hwndTreeView);
    if (hTreeItem == NULL)
        return;
 
    TreeView_GetItemRect(hwndTreeView, hTreeItem, &rc, TRUE);
 
    pt.x = (rc.left + rc.right) / 2;
    pt.y = (rc.top + rc.bottom) / 2;
    ClientToScreen(hwndTreeView, &pt);
    TrackPopupMenu(GetSubMenu(pHotplugData->hPopupMenu, 0),
                   TPM_LEFTALIGN | TPM_TOPALIGN,
                   pt.x,
                   pt.y,
                   0,
                   hwndDlg,
                   NULL);
}

Referenced by OpenDialogProc(), SafeRemovalDlgProc(), SaveDialogProc(), and VfdPageDlgProc().

UpdateButtons()

static VOID UpdateButtons ( HWND  hwndDlg )

static

Definition at line 305 of file hotplug.c.

{
    BOOL bEnabled;
 
    bEnabled = (TreeView_GetCount(GetDlgItem(hwndDlg, IDC_SAFE_REMOVE_DEVICE_TREE)) != 0);
 
    EnableWindow(GetDlgItem(hwndDlg, IDC_SAFE_REMOVE_PROPERTIES), bEnabled);
    EnableWindow(GetDlgItem(hwndDlg, IDC_SAFE_REMOVE_STOP), bEnabled);
}

Referenced by SafeRemovalDlgProc().

Variable Documentation

Applets

APPLET Applets[NUM_APPLETS]

Initial value:

=
{
    {IDI_HOTPLUG, IDS_CPLNAME, IDS_CPLDESCRIPTION, InitApplet}
}

Definition at line 32 of file hotplug.c.

Referenced by CPlApplet().

hApplet

HINSTANCE hApplet = 0

Definition at line 29 of file hotplug.c.

Referenced by DllMain(), InitApplet(), and SafeRemovalDlgProc().