Leaked source code of windows server 2003
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

768 lines
22 KiB

//+-------------------------------------------------------------------------
//
// Microsoft Windows
//
// Copyright (C) Microsoft Corporation
//
// File: init.cpp
//
//--------------------------------------------------------------------------
#include "hotplug.h"
#define HOTPLUG_CLASS_NAME TEXT("HotPlugClass")
VOID
HotPlugDeviceTree(
HWND hwndParent,
BOOLEAN HotPlugTree
)
{
CONFIGRET ConfigRet;
DEVICETREE DeviceTree;
ZeroMemory(&DeviceTree, sizeof(DeviceTree));
DeviceTree.HotPlugTree = HotPlugTree;
InitializeListHead(&DeviceTree.ChildSiblingList);
DialogBoxParam(hHotPlug,
MAKEINTRESOURCE(DLG_DEVTREE),
hwndParent,
DevTreeDlgProc,
(LPARAM)&DeviceTree
);
return;
}
DWORD
WINAPI
HotPlugRemovalVetoedW(
HWND hwnd,
HINSTANCE hInst,
LPWSTR szCmd,
int nShow
)
{
UNREFERENCED_PARAMETER(hwnd);
UNREFERENCED_PARAMETER(hInst);
UNREFERENCED_PARAMETER(nShow);
return HandleVetoedOperation(szCmd, VETOED_REMOVAL);
}
DWORD
WINAPI
HotPlugEjectVetoedW(
HWND hwnd,
HINSTANCE hInst,
LPWSTR szCmd,
int nShow
)
{
UNREFERENCED_PARAMETER(hwnd);
UNREFERENCED_PARAMETER(hInst);
UNREFERENCED_PARAMETER(nShow);
return HandleVetoedOperation(szCmd, VETOED_EJECT);
}
DWORD
WINAPI
HotPlugStandbyVetoedW(
HWND hwnd,
HINSTANCE hInst,
LPWSTR szCmd,
int nShow
)
{
UNREFERENCED_PARAMETER(hwnd);
UNREFERENCED_PARAMETER(hInst);
UNREFERENCED_PARAMETER(nShow);
return HandleVetoedOperation(szCmd, VETOED_STANDBY);
}
DWORD
WINAPI
HotPlugHibernateVetoedW(
HWND hwnd,
HINSTANCE hInst,
LPWSTR szCmd,
int nShow
)
{
UNREFERENCED_PARAMETER(hwnd);
UNREFERENCED_PARAMETER(hInst);
UNREFERENCED_PARAMETER(nShow);
return HandleVetoedOperation(szCmd, VETOED_HIBERNATE);
}
DWORD
WINAPI
HotPlugWarmEjectVetoedW(
HWND hwnd,
HINSTANCE hInst,
LPWSTR szCmd,
int nShow
)
{
UNREFERENCED_PARAMETER(hwnd);
UNREFERENCED_PARAMETER(hInst);
UNREFERENCED_PARAMETER(nShow);
return HandleVetoedOperation(szCmd, VETOED_WARM_EJECT);
}
DWORD
WINAPI
HandleVetoedOperation(
LPWSTR szCmd,
VETOED_OPERATION VetoedOperation
)
{
HANDLE hPipeRead;
HANDLE hEvent;
PNP_VETO_TYPE vetoType;
DWORD bytesRead;
VETO_DEVICE_COLLECTION removalVetoCollection;
//
// Open the specified name pipe and event.
//
if (!OpenPipeAndEventHandles(szCmd,
&hPipeRead,
&hEvent)) {
return 1;
}
ASSERT((hEvent != NULL) && (hEvent != INVALID_HANDLE_VALUE));
ASSERT((hPipeRead != NULL) && (hPipeRead != INVALID_HANDLE_VALUE));
//
// The first DWORD is the VetoType
//
if (!ReadFile(hPipeRead,
(LPVOID)&vetoType,
sizeof(PNP_VETO_TYPE),
&bytesRead,
NULL)) {
CloseHandle(hPipeRead);
SetEvent(hEvent);
CloseHandle(hEvent);
return 1;
}
//
// Now drain all the removal strings. Note that some of them will be
// device instance paths (definitely the first)
//
DeviceCollectionBuildFromPipe(
hPipeRead,
CT_VETOED_REMOVAL_NOTIFICATION,
(PDEVICE_COLLECTION) &removalVetoCollection
);
//
// We are finished reading from the pipe, so close the handle and tell
// umpnpmgr that it can continue.
//
CloseHandle(hPipeRead);
SetEvent(hEvent);
CloseHandle(hEvent);
//
// There should always be one device as that is the device who's removal
// was vetoed.
//
ASSERT(removalVetoCollection.dc.NumDevices);
//
// Invent the VetoedOperation "VETOED_UNDOCK" from an eject containing
// another dock.
//
if (removalVetoCollection.dc.DockInList) {
if (VetoedOperation == VETOED_EJECT) {
VetoedOperation = VETOED_UNDOCK;
} else if (VetoedOperation == VETOED_WARM_EJECT) {
VetoedOperation = VETOED_WARM_UNDOCK;
}
}
removalVetoCollection.VetoType = vetoType;
removalVetoCollection.VetoedOperation = VetoedOperation;
VetoedRemovalUI(&removalVetoCollection);
DeviceCollectionDestroy(
(PDEVICE_COLLECTION) &removalVetoCollection
);
return 1;
}
DWORD
WINAPI
HotPlugSafeRemovalNotificationW(
HWND hwnd,
HINSTANCE hInst,
LPWSTR szCmd,
int nShow
)
{
HANDLE hPipeRead, hEvent;
DEVICE_COLLECTION safeRemovalCollection;
MSG Msg;
WNDCLASS wndClass;
HWND hSafeRemovalWnd;
HANDLE hHotplugIconEvent;
UNREFERENCED_PARAMETER(hwnd);
UNREFERENCED_PARAMETER(hInst);
UNREFERENCED_PARAMETER(nShow);
//
// Open the specified name pipe and event.
//
if (!OpenPipeAndEventHandles(szCmd,
&hPipeRead,
&hEvent)) {
return 1;
}
ASSERT((hEvent != NULL) && (hEvent != INVALID_HANDLE_VALUE));
ASSERT((hPipeRead != NULL) && (hPipeRead != INVALID_HANDLE_VALUE));
//
// Read out the device ID list from the Pipe
//
DeviceCollectionBuildFromPipe(
hPipeRead,
CT_SAFE_REMOVAL_NOTIFICATION,
&safeRemovalCollection
);
//
// On success or error, we are finished reading from the pipe, so close the
// handle and tell umpnpmgr it can continue.
//
CloseHandle(hPipeRead);
SetEvent(hEvent);
CloseHandle(hEvent);
//
// If we have any devices then bring up the safe removal dialog
//
if (safeRemovalCollection.NumDevices) {
if (!GetClassInfo(hHotPlug, HOTPLUG_CLASS_NAME, &wndClass)) {
ZeroMemory(&wndClass, sizeof(wndClass));
wndClass.lpfnWndProc = (safeRemovalCollection.DockInList)
? DockSafeRemovalBalloonProc
: SafeRemovalBalloonProc;
wndClass.hInstance = hHotPlug;
wndClass.lpszClassName = HOTPLUG_CLASS_NAME;
if (!RegisterClass(&wndClass)) {
goto clean0;
}
}
//
// In order to prevent multiple similar icons on the tray, we will
// create a named event that will be used to serialize the UI.
//
// Note that if we can't create the event for some reason then we will just
// display the UI. This might cause multiple icons, but it is better
// than not displaying any UI at all.
//
hHotplugIconEvent = CreateEvent(NULL,
FALSE,
TRUE,
safeRemovalCollection.DockInList
? TEXT("Local\\Dock_TaskBarIcon_Event")
: TEXT("Local\\HotPlug_TaskBarIcon_Event")
);
if (hHotplugIconEvent) {
WaitForSingleObject(hHotplugIconEvent, INFINITE);
}
if (!safeRemovalCollection.DockInList) {
//
// First disable the hotplug service so that the icon will go away from
// the taskbar. We do this just in case there are any other hotplug devices
// in the machine since we don't want multiple hotplug icons
// showing up in the taskbar.
//
// NOTE: We don't need to do this for the safe to undock case since
// the docking icon is different.
//
SysTray_EnableService(STSERVICE_HOTPLUG, FALSE);
}
hSafeRemovalWnd = CreateWindowEx(WS_EX_TOOLWINDOW,
HOTPLUG_CLASS_NAME,
TEXT(""),
WS_DLGFRAME | WS_BORDER | WS_DISABLED,
CW_USEDEFAULT,
CW_USEDEFAULT,
0,
0,
NULL,
NULL,
hHotPlug,
(LPVOID)&safeRemovalCollection
);
if (hSafeRemovalWnd != NULL) {
while (IsWindow(hSafeRemovalWnd)) {
if (GetMessage(&Msg, NULL, 0, 0)) {
TranslateMessage(&Msg);
DispatchMessage(&Msg);
}
}
}
//
// Set the Event so the next surprise removal process can go to work
// and then close the event handle.
//
if (hHotplugIconEvent) {
SetEvent(hHotplugIconEvent);
CloseHandle(hHotplugIconEvent);
}
if (!safeRemovalCollection.DockInList) {
//
// Re-enable the hotplug service so that the icon can show back up in
// the taskbar if we have any hotplug devices.
//
SysTray_EnableService(STSERVICE_HOTPLUG, TRUE);
}
}
clean0:
DeviceCollectionDestroy(&safeRemovalCollection);
return 1;
}
DWORD
WINAPI
HotPlugDriverBlockedW(
HWND hwnd,
HINSTANCE hInst,
LPWSTR szCmd,
int nShow
)
{
HANDLE hPipeRead, hEvent;
DEVICE_COLLECTION blockedDriverCollection;
HANDLE hDriverBlockIconEvent = NULL;
HANDLE hDriverBlockEvent = NULL;
TCHAR szEventName[MAX_PATH];
UNREFERENCED_PARAMETER(hwnd);
UNREFERENCED_PARAMETER(hInst);
UNREFERENCED_PARAMETER(nShow);
//
// Open the specified name pipe and event.
//
if (OpenPipeAndEventHandles(szCmd,
&hPipeRead,
&hEvent) == FALSE) {
return 1;
}
ASSERT((hEvent != NULL) && (hEvent != INVALID_HANDLE_VALUE));
ASSERT((hPipeRead != NULL) && (hPipeRead != INVALID_HANDLE_VALUE));
//
// Read out the list of blocked driver GUIDs from the Pipe. Note that for
// the CT_BLOCKED_DRIVER_NOTIFICATION collection type, we use only the
// DeviceInstanceId field of each collection entry (which is OK because
// MAX_GUID_STRING_LEN << MAX_DEVICE_ID_LEN). All other fields are skipped.
//
DeviceCollectionBuildFromPipe(
hPipeRead,
CT_BLOCKED_DRIVER_NOTIFICATION,
&blockedDriverCollection
);
//
// On success or error, we are finished reading from the pipe, so close the
// handle and tell umpnpmgr it can continue.
//
CloseHandle(hPipeRead);
SetEvent(hEvent);
CloseHandle(hEvent);
//
// Since the balloons can hang around for numerous seconds, or longer if
// there is no user input to the system, we need to make sure that we only
// have one driver block event queued for any given type of driver block.
// This way if there is an attempt to load a driver many times in a row, we
// won't queue up numerous driver blocked ballons for the same driver.
// We will do this by creating a local event that includes the GUID of the
// blocked driver or no GUID if we are showing the generic balloon.
//
if (SUCCEEDED(StringCchPrintf(szEventName,
SIZECHARS(szEventName),
TEXT("Local\\DRIVERBLOCK-%s"),
(blockedDriverCollection.NumDevices == 1)
? DeviceCollectionGetDeviceInstancePath(&blockedDriverCollection, 0)
: TEXT("ALL")
))) {
hDriverBlockEvent = CreateEvent(NULL,
FALSE,
TRUE,
szEventName
);
if (hDriverBlockEvent) {
if (WaitForSingleObject(hDriverBlockEvent, 0) != WAIT_OBJECT_0) {
//
// This means that this driver block balloon is either being
// displayed, or in the queue to be displayed, so we can just
// exit this process.
//
goto clean0;
}
}
}
//
// In order to prevent multipe driver blocked icons and ballons showing up
// on the taskbar together and stepping on each other, we will create a
// named event that will be used to serialize the driver blocked icons and
// balloon UI.
//
// Note that if we can't create the event for some reason then we will just
// display the UI. This might cause multiple driver blocked icons, but it
// is better than not displaying any UI at all.
//
// Also note that we can coexist with normal hotplug icon. As such we have
// a different event name and a different icon.
//
hDriverBlockIconEvent = CreateEvent(NULL,
FALSE,
TRUE,
TEXT("Local\\HotPlug_DriverBlockedIcon_Event")
);
if (hDriverBlockIconEvent) {
for (;;) {
DWORD waitStatus;
waitStatus = MsgWaitForMultipleObjects(1,
&hDriverBlockIconEvent,
FALSE,
INFINITE,
QS_ALLINPUT);
if (waitStatus == WAIT_OBJECT_0) {
//
// The current driver block icon went away so it is our turn.
//
break;
} else if (waitStatus == (WAIT_OBJECT_0 + 1)) {
//
// Message in the queue.
//
MSG msg;
while (PeekMessage(&msg, NULL, 0, 0, PM_REMOVE)) {
if (msg.message == WM_CLOSE) {
goto clean0;
}
TranslateMessage(&msg);
DispatchMessage(&msg);
}
} else {
//
// This shouldn't happen.
//
goto clean0;
}
}
}
//
// Show the balloon.
//
DisplayDriverBlockBalloon(&blockedDriverCollection);
//
// Since the balloon has now gone away, set the event so if we get another
// block on the same driver it will display another balloon.
//
if (hDriverBlockEvent) {
SetEvent(hDriverBlockEvent);
}
clean0:
//
// Set the Event so the next blocked driver process can go to work and then
// close the event handle.
//
if (hDriverBlockIconEvent) {
SetEvent(hDriverBlockIconEvent);
CloseHandle(hDriverBlockIconEvent);
}
if (hDriverBlockEvent) {
CloseHandle(hDriverBlockEvent);
}
//
// Destroy the collection.
//
DeviceCollectionDestroy(&blockedDriverCollection);
return 1;
}
DWORD
WINAPI
HotPlugChildWithInvalidIdW(
HWND hwnd,
HINSTANCE hInst,
LPWSTR szCmd,
int nShow
)
{
HANDLE hPipeRead, hEvent;
DEVICE_COLLECTION childWithInvalidIdCollection;
HANDLE hChildWithInvalidIdIconEvent = NULL;
HANDLE hChildWithInvalidIdEvent = NULL;
TCHAR szEventName[MAX_PATH];
UNREFERENCED_PARAMETER(hwnd);
UNREFERENCED_PARAMETER(hInst);
UNREFERENCED_PARAMETER(nShow);
//
// Open the specified name pipe and event.
//
if (!OpenPipeAndEventHandles(szCmd,
&hPipeRead,
&hEvent)) {
return 1;
}
ASSERT((hEvent != NULL) && (hEvent != INVALID_HANDLE_VALUE));
ASSERT((hPipeRead != NULL) && (hPipeRead != INVALID_HANDLE_VALUE));
//
// Read out the device instance Id of the parent that has a child device
// with an invalid Id. Note, if we are passed multiple device instance
// Ids, we will only display UI for the first one in the list.
//
DeviceCollectionBuildFromPipe(
hPipeRead,
CT_CHILD_WITH_INVALID_ID_NOTIFICATION,
&childWithInvalidIdCollection
);
//
// On success or error, we are finished reading from the pipe, so close the
// handle and tell umpnpmgr it can continue.
//
CloseHandle(hPipeRead);
SetEvent(hEvent);
CloseHandle(hEvent);
//
// Since the balloons can hang around for numerous seconds, or longer if
// there is no user input to the system, we need to make sure that we only
// have one invalid child event queued for any given parent with an invalid
// child.
//
StringCchPrintf(szEventName,
SIZECHARS(szEventName),
TEXT("Local\\CHILDWITHINVALIDID-%s"),
DeviceCollectionGetDeviceInstancePath(&childWithInvalidIdCollection, 0)
);
hChildWithInvalidIdEvent = CreateEvent(NULL,
FALSE,
TRUE,
szEventName
);
if (hChildWithInvalidIdEvent) {
if (WaitForSingleObject(hChildWithInvalidIdEvent, 0) != WAIT_OBJECT_0) {
//
// This means that this invalid child balloon is either being
// displayed, or in the queue to be displayed, so we can just
// exit this process.
//
goto clean0;
}
}
//
// In order to prevent multipe invalid child icons and ballons showing up
// on the taskbar together and stepping on each other, we will create a
// named event that will be used to serialize the invalid child icons and
// balloon UI.
//
// Note that if we can't create the event for some reason then we will just
// display the UI. This might cause multiple invalid child icons, but it
// is better than not displaying any UI at all.
//
// Also note that we can coexist with normal hotplug icon. As such we have
// a different event name and a different icon.
//
hChildWithInvalidIdIconEvent = CreateEvent(NULL,
FALSE,
TRUE,
TEXT("Local\\HotPlug_ChildWithInvalidId_Event")
);
if (hChildWithInvalidIdIconEvent) {
for (;;) {
DWORD waitStatus;
waitStatus = MsgWaitForMultipleObjects(1,
&hChildWithInvalidIdIconEvent,
FALSE,
INFINITE,
QS_ALLINPUT);
if (waitStatus == WAIT_OBJECT_0) {
//
// The current invalid child icon went away so it is our turn.
//
break;
} else if (waitStatus == (WAIT_OBJECT_0 + 1)) {
//
// Message in the queue.
//
MSG msg;
while (PeekMessage(&msg, NULL, 0, 0, PM_REMOVE)) {
if (msg.message == WM_CLOSE) {
goto clean0;
}
TranslateMessage(&msg);
DispatchMessage(&msg);
}
} else {
//
// This shouldn't happen.
//
goto clean0;
}
}
}
//
// Show the balloon.
//
DisplayChildWithInvalidIdBalloon(&childWithInvalidIdCollection);
//
// Since the balloon has now gone away, set the event so if we get another
// invalid child device it will display another balloon.
//
if (hChildWithInvalidIdEvent) {
SetEvent(hChildWithInvalidIdEvent);
}
clean0:
//
// Set the Event so the next invalid child process can go to work and then
// close the event handle.
//
if (hChildWithInvalidIdIconEvent) {
SetEvent(hChildWithInvalidIdIconEvent);
CloseHandle(hChildWithInvalidIdIconEvent);
}
if (hChildWithInvalidIdEvent) {
CloseHandle(hChildWithInvalidIdEvent);
}
//
// Destroy the collection.
//
DeviceCollectionDestroy(&childWithInvalidIdCollection);
return 1;
}
LONG
CPlApplet(
HWND hWnd,
WORD uMsg,
DWORD_PTR lParam1,
LRESULT lParam2
)
{
LPNEWCPLINFO lpCPlInfo;
LPCPLINFO lpOldCPlInfo;
UNREFERENCED_PARAMETER(lParam1);
switch (uMsg) {
case CPL_INIT:
return TRUE;
case CPL_GETCOUNT:
return 1;
case CPL_INQUIRE:
lpOldCPlInfo = (LPCPLINFO)(LPARAM)lParam2;
lpOldCPlInfo->lData = 0L;
lpOldCPlInfo->idIcon = IDI_HOTPLUGICON;
lpOldCPlInfo->idName = IDS_HOTPLUGNAME;
lpOldCPlInfo->idInfo = IDS_HOTPLUGINFO;
return TRUE;
case CPL_NEWINQUIRE:
lpCPlInfo = (LPNEWCPLINFO)(LPARAM)lParam2;
lpCPlInfo->hIcon = LoadIcon(hHotPlug, MAKEINTRESOURCE(IDI_HOTPLUGICON));
LoadString(hHotPlug, IDS_HOTPLUGNAME, lpCPlInfo->szName, SIZECHARS(lpCPlInfo->szName));
LoadString(hHotPlug, IDS_HOTPLUGINFO, lpCPlInfo->szInfo, SIZECHARS(lpCPlInfo->szInfo));
lpCPlInfo->dwHelpContext = IDH_HOTPLUGAPPLET;
lpCPlInfo->dwSize = sizeof(NEWCPLINFO);
lpCPlInfo->lData = 0;
lpCPlInfo->szHelpFile[0] = '\0';
return TRUE;
case CPL_DBLCLK:
HotPlugDeviceTree(hWnd, TRUE);
break;
default:
break;
}
return 0L;
}