mirror of
https://github.com/reactos/reactos.git
synced 2024-08-29 14:48:37 +00:00
367 lines
12 KiB
C++
367 lines
12 KiB
C++
/*
|
|
* PROJECT: ReactOS system libraries
|
|
* LICENSE: GPL - See COPYING in the top level directory
|
|
* FILE: dll/shellext/stobject/hotplug.cpp
|
|
* PURPOSE: Removable devices notification icon handler
|
|
* PROGRAMMERS: Shriraj Sawant a.k.a SR13 <sr.official@hotmail.com>
|
|
*/
|
|
|
|
#include "precomp.h"
|
|
|
|
#include <atlsimpcoll.h>
|
|
#include <dbt.h>
|
|
#include <cfgmgr32.h>
|
|
#include <shlwapi.h>
|
|
|
|
#define DISPLAY_NAME_LEN 40
|
|
|
|
CSimpleArray<DEVINST> g_devList;
|
|
CString g_strHotplugTooltip;
|
|
static HICON g_hIconHotplug = NULL;
|
|
static WCHAR g_strMenuSel[DISPLAY_NAME_LEN];
|
|
static BOOL g_IsRemoving = FALSE;
|
|
|
|
/*++
|
|
* @name EnumHotpluggedDevices
|
|
*
|
|
* Enumerates the connected safely removable devices.
|
|
*
|
|
* @param devList
|
|
* List of device instances, representing the currently attached devices.
|
|
*
|
|
* @return The error code.
|
|
*
|
|
*--*/
|
|
HRESULT EnumHotpluggedDevices(CSimpleArray<DEVINST> &devList)
|
|
{
|
|
devList.RemoveAll(); // Clear current devList
|
|
HDEVINFO hdev = SetupDiGetClassDevs(NULL, NULL, 0, DIGCF_ALLCLASSES | DIGCF_PRESENT);
|
|
if (INVALID_HANDLE_VALUE == hdev)
|
|
return E_HANDLE;
|
|
SP_DEVINFO_DATA did = { 0 };
|
|
did.cbSize = sizeof(did);
|
|
|
|
// Enumerate all the attached devices.
|
|
for (int idev = 0; SetupDiEnumDeviceInfo(hdev, idev, &did); idev++)
|
|
{
|
|
DWORD dwCapabilities = 0, dwSize = sizeof(dwCapabilities);
|
|
ULONG ulStatus = 0, ulProblem = 0;
|
|
CONFIGRET cr = CM_Get_DevNode_Status(&ulStatus, &ulProblem, did.DevInst, 0);
|
|
if (cr != CR_SUCCESS)
|
|
continue;
|
|
|
|
cr = CM_Get_DevNode_Registry_Property(did.DevInst, CM_DRP_CAPABILITIES, NULL, &dwCapabilities, &dwSize, 0);
|
|
if (cr != CR_SUCCESS)
|
|
continue;
|
|
|
|
// Filter and make list of only the appropriate safely removable devices.
|
|
if ( (dwCapabilities & CM_DEVCAP_REMOVABLE) &&
|
|
!(dwCapabilities & CM_DEVCAP_DOCKDEVICE) &&
|
|
!(dwCapabilities & CM_DEVCAP_SURPRISEREMOVALOK) &&
|
|
((dwCapabilities & CM_DEVCAP_EJECTSUPPORTED) || (ulStatus & DN_DISABLEABLE)) &&
|
|
!ulProblem)
|
|
{
|
|
devList.Add(did.DevInst);
|
|
}
|
|
}
|
|
SetupDiDestroyDeviceInfoList(hdev);
|
|
|
|
if (NO_ERROR != GetLastError() && ERROR_NO_MORE_ITEMS != GetLastError())
|
|
{
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
/*++
|
|
* @name NotifyBalloon
|
|
*
|
|
* Pops the balloon notification of the given notification icon.
|
|
*
|
|
* @param pSysTray
|
|
* Provides interface for acquiring CSysTray information as required.
|
|
* @param szTitle
|
|
* Title for the balloon notification.
|
|
* @param szInfo
|
|
* Main content for the balloon notification.
|
|
* @param uId
|
|
* Represents the particular notification icon.
|
|
*
|
|
* @return The error code.
|
|
*
|
|
*--*/
|
|
HRESULT NotifyBalloon(CSysTray* pSysTray, LPCWSTR szTitle = NULL, LPCWSTR szInfo = NULL, UINT uId = ID_ICON_HOTPLUG)
|
|
{
|
|
NOTIFYICONDATA nim = { 0 };
|
|
|
|
nim.cbSize = sizeof(nim);
|
|
nim.uID = uId;
|
|
nim.hWnd = pSysTray->GetHWnd();
|
|
|
|
nim.uFlags = NIF_INFO;
|
|
nim.uTimeout = 10;
|
|
nim.dwInfoFlags = NIIF_INFO;
|
|
|
|
StringCchCopy(nim.szInfoTitle, _countof(nim.szInfoTitle), szTitle);
|
|
StringCchCopy(nim.szInfo, _countof(nim.szInfo), szInfo);
|
|
BOOL ret = Shell_NotifyIcon(NIM_MODIFY, &nim);
|
|
|
|
Sleep(10000); /* As per windows, the balloon notification remains visible for atleast 10 sec.
|
|
This timer maintains the same condition.
|
|
Also it is required so that the icon doesn't hide instantly after last device is removed,
|
|
as that will prevent popping of notification.
|
|
*/
|
|
StringCchCopy(nim.szInfoTitle, _countof(nim.szInfoTitle), L"");
|
|
StringCchCopy(nim.szInfo, _countof(nim.szInfo), L"");
|
|
ret = Shell_NotifyIcon(NIM_MODIFY, &nim);
|
|
g_IsRemoving = FALSE; /* This flag is used to prevent instant icon hiding after last device is removed.
|
|
The above timer maintains the required state for the same.
|
|
*/
|
|
return ret ? S_OK : E_FAIL;
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE Hotplug_Init(_In_ CSysTray * pSysTray)
|
|
{
|
|
TRACE("Hotplug_Init\n");
|
|
|
|
g_hIconHotplug = LoadIcon(g_hInstance, MAKEINTRESOURCE(IDI_HOTPLUG_OK));
|
|
g_strHotplugTooltip.LoadStringW(IDS_HOTPLUG_REMOVE_1);
|
|
|
|
EnumHotpluggedDevices(g_devList);
|
|
|
|
if (g_devList.GetSize() > 0)
|
|
return pSysTray->NotifyIcon(NIM_ADD, ID_ICON_HOTPLUG, g_hIconHotplug, g_strHotplugTooltip);
|
|
else
|
|
return pSysTray->NotifyIcon(NIM_ADD, ID_ICON_HOTPLUG, g_hIconHotplug, g_strHotplugTooltip, NIS_HIDDEN);
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE Hotplug_Update(_In_ CSysTray * pSysTray)
|
|
{
|
|
TRACE("Hotplug_Update\n");
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT STDMETHODCALLTYPE Hotplug_Shutdown(_In_ CSysTray * pSysTray)
|
|
{
|
|
TRACE("Hotplug_Shutdown\n");
|
|
|
|
DestroyIcon(g_hIconHotplug);
|
|
g_hIconHotplug = NULL;
|
|
|
|
return pSysTray->NotifyIcon(NIM_DELETE, ID_ICON_HOTPLUG, NULL, NULL);
|
|
}
|
|
|
|
static void _RunHotplug(CSysTray * pSysTray)
|
|
{
|
|
ShellExecuteW(pSysTray->GetHWnd(),
|
|
L"open",
|
|
L"rundll32.exe",
|
|
L"shell32.dll,Control_RunDLL hotplug.dll",
|
|
NULL,
|
|
SW_SHOWNORMAL);
|
|
}
|
|
|
|
static void _ShowContextMenu(CSysTray * pSysTray)
|
|
{
|
|
HMENU hPopup = CreatePopupMenu();
|
|
ULONG ulLength = DISPLAY_NAME_LEN * sizeof(WCHAR);
|
|
|
|
for (INT index = 0; index < g_devList.GetSize(); index++)
|
|
{
|
|
WCHAR dispName[DISPLAY_NAME_LEN];
|
|
CString menuName;
|
|
CONFIGRET cr = CM_Get_DevNode_Registry_Property(g_devList[index], CM_DRP_DEVICEDESC, NULL, dispName, &ulLength, 0);
|
|
if (cr != CR_SUCCESS)
|
|
StrCpyW(dispName, L"Unknown Device");
|
|
|
|
menuName.Format(IDS_HOTPLUG_REMOVE_3, dispName);
|
|
AppendMenuW(hPopup, MF_STRING, index+1, menuName);
|
|
}
|
|
|
|
SetForegroundWindow(pSysTray->GetHWnd());
|
|
DWORD flags = TPM_RETURNCMD | TPM_NONOTIFY | TPM_RIGHTALIGN | TPM_BOTTOMALIGN;
|
|
POINT pt;
|
|
GetCursorPos(&pt);
|
|
|
|
DWORD id = TrackPopupMenuEx(hPopup, flags,
|
|
pt.x, pt.y,
|
|
pSysTray->GetHWnd(), NULL);
|
|
|
|
if (id > 0)
|
|
{
|
|
id--; // since array indices starts from zero.
|
|
CONFIGRET cr = CM_Get_DevNode_Registry_Property(g_devList[id], CM_DRP_DEVICEDESC, NULL, g_strMenuSel, &ulLength, 0);
|
|
if (cr != CR_SUCCESS)
|
|
StrCpyW(g_strMenuSel, L"Unknown Device");
|
|
|
|
cr = CM_Request_Device_Eject_Ex(g_devList[id], 0, 0, 0, 0, 0);
|
|
if (cr != CR_SUCCESS)
|
|
{
|
|
WCHAR strInfo[128];
|
|
swprintf(strInfo, L"Problem Ejecting %wS", g_strMenuSel);
|
|
MessageBox(0, L"The device cannot be stopped right now! Try stopping it again later!", strInfo, MB_OK | MB_ICONEXCLAMATION);
|
|
}
|
|
else
|
|
{
|
|
//MessageBox(0, L"Device ejected successfully!! You can safely remove the device now!", L"Safely Remove Hardware", MB_OK | MB_ICONINFORMATION);
|
|
g_IsRemoving = TRUE;
|
|
g_devList.RemoveAt(id); /* thing is.. even after removing id at this point, the devnode_change occurs after some seconds of sucessful removal
|
|
and since pendrive is still plugged in it gets enumerated, if problem number is not filtered.
|
|
*/
|
|
}
|
|
}
|
|
|
|
DestroyMenu(hPopup);
|
|
}
|
|
|
|
static void _ShowContextMenuR(CSysTray * pSysTray)
|
|
{
|
|
CString strMenu((LPWSTR)IDS_HOTPLUG_REMOVE_2);
|
|
HMENU hPopup = CreatePopupMenu();
|
|
AppendMenuW(hPopup, MF_STRING, IDS_HOTPLUG_REMOVE_2, strMenu);
|
|
SetMenuDefaultItem(hPopup, IDS_HOTPLUG_REMOVE_2, FALSE);
|
|
|
|
SetForegroundWindow(pSysTray->GetHWnd());
|
|
DWORD flags = TPM_RETURNCMD | TPM_NONOTIFY | TPM_RIGHTALIGN | TPM_BOTTOMALIGN;
|
|
POINT pt;
|
|
GetCursorPos(&pt);
|
|
|
|
DWORD id = TrackPopupMenuEx(hPopup, flags,
|
|
pt.x, pt.y,
|
|
pSysTray->GetHWnd(), NULL);
|
|
|
|
if (id == IDS_HOTPLUG_REMOVE_2)
|
|
{
|
|
_RunHotplug(pSysTray);
|
|
}
|
|
|
|
DestroyMenu(hPopup);
|
|
}
|
|
|
|
|
|
VOID
|
|
HotplugDeviceTimer(
|
|
_In_ CSysTray *pSysTray)
|
|
{
|
|
TRACE("HotplugDeviceTimer()\n");
|
|
|
|
EnumHotpluggedDevices(g_devList);
|
|
|
|
if (g_devList.GetSize() > 0)
|
|
pSysTray->NotifyIcon(NIM_MODIFY, ID_ICON_HOTPLUG, g_hIconHotplug, g_strHotplugTooltip);
|
|
else
|
|
pSysTray->NotifyIcon(NIM_MODIFY, ID_ICON_HOTPLUG, g_hIconHotplug, g_strHotplugTooltip, NIS_HIDDEN);
|
|
}
|
|
|
|
|
|
HRESULT STDMETHODCALLTYPE Hotplug_Message(_In_ CSysTray * pSysTray, UINT uMsg, WPARAM wParam, LPARAM lParam, LRESULT &lResult)
|
|
{
|
|
TRACE("Hotplug_Message uMsg=%d, wParam=%x, lParam=%x\n", uMsg, wParam, lParam);
|
|
|
|
switch (uMsg)
|
|
{
|
|
case WM_USER + 220:
|
|
TRACE("Hotplug_Message: WM_USER+220\n");
|
|
if (wParam == HOTPLUG_SERVICE_FLAG)
|
|
{
|
|
if (lParam)
|
|
{
|
|
pSysTray->EnableService(HOTPLUG_SERVICE_FLAG, TRUE);
|
|
return Hotplug_Init(pSysTray);
|
|
}
|
|
else
|
|
{
|
|
pSysTray->EnableService(HOTPLUG_SERVICE_FLAG, FALSE);
|
|
return Hotplug_Shutdown(pSysTray);
|
|
}
|
|
}
|
|
return S_FALSE;
|
|
|
|
case WM_USER + 221:
|
|
TRACE("Hotplug_Message: WM_USER+221\n");
|
|
if (wParam == HOTPLUG_SERVICE_FLAG)
|
|
{
|
|
lResult = (LRESULT)pSysTray->IsServiceEnabled(HOTPLUG_SERVICE_FLAG);
|
|
return S_OK;
|
|
}
|
|
return S_FALSE;
|
|
|
|
case WM_TIMER:
|
|
if (wParam == HOTPLUG_TIMER_ID)
|
|
{
|
|
KillTimer(pSysTray->GetHWnd(), HOTPLUG_TIMER_ID);
|
|
_ShowContextMenu(pSysTray);
|
|
}
|
|
else if (wParam == HOTPLUG_DEVICE_TIMER_ID)
|
|
{
|
|
KillTimer(pSysTray->GetHWnd(), HOTPLUG_DEVICE_TIMER_ID);
|
|
HotplugDeviceTimer(pSysTray);
|
|
}
|
|
break;
|
|
|
|
case ID_ICON_HOTPLUG:
|
|
switch (lParam)
|
|
{
|
|
case WM_LBUTTONDOWN:
|
|
SetTimer(pSysTray->GetHWnd(), HOTPLUG_TIMER_ID, GetDoubleClickTime(), NULL);
|
|
break;
|
|
|
|
case WM_LBUTTONUP:
|
|
break;
|
|
|
|
case WM_LBUTTONDBLCLK:
|
|
KillTimer(pSysTray->GetHWnd(), HOTPLUG_TIMER_ID);
|
|
_RunHotplug(pSysTray);
|
|
break;
|
|
|
|
case WM_RBUTTONDOWN:
|
|
break;
|
|
|
|
case WM_RBUTTONUP:
|
|
_ShowContextMenuR(pSysTray);
|
|
break;
|
|
|
|
case WM_RBUTTONDBLCLK:
|
|
break;
|
|
|
|
case WM_MOUSEMOVE:
|
|
break;
|
|
}
|
|
return S_OK;
|
|
|
|
case WM_DEVICECHANGE:
|
|
switch (wParam)
|
|
{
|
|
case DBT_DEVNODES_CHANGED:
|
|
TRACE("WM_DEVICECHANGE : DBT_DEVNODES_CHANGED\n");
|
|
SetTimer(pSysTray->GetHWnd(), HOTPLUG_DEVICE_TIMER_ID, 100, NULL);
|
|
lResult = true;
|
|
break;
|
|
|
|
case DBT_DEVICEARRIVAL:
|
|
break;
|
|
case DBT_DEVICEQUERYREMOVE:
|
|
break;
|
|
case DBT_DEVICEQUERYREMOVEFAILED:
|
|
break;
|
|
case DBT_DEVICEREMOVECOMPLETE:
|
|
WCHAR strInfo[128];
|
|
swprintf(strInfo, L"The %wS can now be safely removed from the system.", g_strMenuSel);
|
|
NotifyBalloon(pSysTray, L"Safe to Remove Hardware", strInfo);
|
|
|
|
lResult = true;
|
|
break;
|
|
case DBT_DEVICEREMOVEPENDING:
|
|
break;
|
|
}
|
|
return S_OK;
|
|
|
|
default:
|
|
TRACE("Hotplug_Message received for unknown ID %d, ignoring.\n");
|
|
return S_FALSE;
|
|
}
|
|
|
|
return S_FALSE;
|
|
}
|