CDeviceView Class Reference

#include <DeviceView.h>

Collaboration diagram for CDeviceView:

Public Member Functions
	CDeviceView (HWND hMainWnd)
	~CDeviceView (void)
bool	Initialize ()
bool	Uninitialize ()
LRESULT	OnSize (_In_ int x, _In_ int y, _In_ int cx, _In_ int cy)
LRESULT	OnDoubleClick (_In_ LPNMHDR NmHdr)
LRESULT	OnRightClick (_In_ LPNMHDR NmHdr)
LRESULT	OnContextMenu (_In_ LPARAM lParam)
LRESULT	OnAction (UINT Action)
VOID	Refresh (_In_ ViewType Type, _In_ bool ScanForChanges, _In_ bool UpdateView)
VOID	DisplayPropertySheet ()
VOID	SetFocus ()
VOID	SetHiddenDevices (_In_ bool ShowHidden)
ViewType	GetCurrentView ()
bool	CreateActionMenu (_In_ HMENU OwnerMenu, _In_ bool MainMenu)
CNode *	GetSelectedNode ()
bool	SelDeviceIsStarted ()
bool	SelDeviceIsInstalled ()

Private Member Functions
bool	AddRootDevice ()
bool	RefreshDeviceList ()
bool	ListDevicesByConnection ()
bool	ListDevicesByType ()
bool	GetNextClass (_In_ ULONG ClassIndex, _Out_ LPGUID ClassGuid, _Out_ HDEVINFO *hDevInfo)
bool	RecurseChildDevices (_In_ DEVINST ParentDevice, _In_ HTREEITEM hParentTreeItem)
bool	EnableSelectedDevice (_In_ bool Enable, _Out_ bool &NeedsReboot)
bool	UpdateSelectedDevice (_Out_ bool &NeedsReboot)
bool	UninstallSelectedDevice ()
bool	RunAddHardwareWizard ()
bool	GetChildDevice (_In_ DEVINST ParentDevInst, _Out_ PDEVINST DevInst)
bool	GetSiblingDevice (_In_ DEVINST PrevDevice, _Out_ PDEVINST DevInst)
HTREEITEM	InsertIntoTreeView (_In_opt_ HTREEITEM hParent, _In_ CNode *Node)
void	BuildActionMenuForNode (_In_ HMENU OwnerMenu, _In_ CNode *Node, _In_ bool MainMenu)
HTREEITEM	RecurseFindDevice (_In_ HTREEITEM hParentItem, _In_ CNode *Node)
void	SelectNode (_In_ CNode *Node)
void	EmptyDeviceView ()
CNode *	GetNode (_In_ LPTV_ITEMW TvItem)
CClassNode *	GetClassNode (_In_ LPGUID ClassGuid)
CDeviceNode *	GetDeviceNode (_In_ DEVINST Device)
void	EmptyLists ()

Static Private Member Functions
static unsigned int __stdcall	RefreshThread (void *Param)

Private Attributes
HWND	m_hMainWnd
HWND	m_hTreeView
HWND	m_hPropertyDialog
HMENU	m_hMenu
ViewType	m_ViewType
HTREEITEM	m_hTreeRoot
bool	m_ShowHidden
CRootNode *	m_RootNode
CAtlList< CClassNode * >	m_ClassNodeList
CAtlList< CDeviceNode * >	m_DeviceNodeList
SP_CLASSIMAGELIST_DATA	m_ImageListData

Detailed Description

Definition at line 15 of file DeviceView.h.

Constructor & Destructor Documentation

CDeviceView()

CDeviceView::CDeviceView

(

HWND

hMainWnd

)

Definition at line 35 of file DeviceView.cpp.

      :
    m_hMainWnd(hMainWnd),
    m_hTreeView(NULL),
    m_hPropertyDialog(NULL),
    m_hMenu(NULL),
    m_ViewType(DevicesByType),
    m_ShowHidden(false),
    m_RootNode(NULL)
{
    ZeroMemory(&m_ImageListData, sizeof(SP_CLASSIMAGELIST_DATA));
}

~CDeviceView()

CDeviceView::~CDeviceView

(

void

)

Definition at line 49 of file DeviceView.cpp.

{
}

Member Function Documentation

AddRootDevice()

bool CDeviceView::AddRootDevice ( )

private

Definition at line 353 of file DeviceView.cpp.

{
    m_hTreeRoot = InsertIntoTreeView(NULL, m_RootNode);
    return (m_hTreeRoot != NULL);
}

BuildActionMenuForNode()

void CDeviceView::BuildActionMenuForNode ( _In_ HMENU  OwnerMenu, _In_ CNode *  Node, _In_ bool  MainMenu  )

private

Definition at line 842 of file DeviceView.cpp.

{
    // Create a separator structure
    MENUITEMINFOW MenuSeparator = { 0 };
    MenuSeparator.cbSize = sizeof(MENUITEMINFOW);
    MenuSeparator.fType = MFT_SEPARATOR;
 
    // Setup the
    MENUITEMINFOW MenuItemInfo = { 0 };
    MenuItemInfo.cbSize = sizeof(MENUITEMINFOW);
    MenuItemInfo.fMask = MIIM_ID | MIIM_STRING | MIIM_DATA | MIIM_SUBMENU;
    MenuItemInfo.fType = MFT_STRING;
 
    CAtlStringW String;
    int i = 0;
 
    // Device nodes have extra data
    if (Node->GetNodeType() == DeviceNode)
    {
        CDeviceNode *DeviceNode = dynamic_cast<CDeviceNode *>(Node);
 
        if (DeviceNode->CanUpdate())
        {
            String.LoadStringW(g_hThisInstance, IDS_MENU_UPDATE);
            MenuItemInfo.wID = IDM_UPDATE_DRV;
            MenuItemInfo.dwTypeData = String.GetBuffer();
            InsertMenuItemW(OwnerMenu, i, TRUE, &MenuItemInfo);
            i++;
        }
 
        if (DeviceNode->IsDisabled())
        {
            String.LoadStringW(g_hThisInstance, IDS_MENU_ENABLE);
            MenuItemInfo.wID = IDM_ENABLE_DRV;
            MenuItemInfo.dwTypeData = String.GetBuffer();
            InsertMenuItemW(OwnerMenu, i, TRUE, &MenuItemInfo);
            i++;
        }
 
        if (DeviceNode->CanDisable() && !DeviceNode->IsDisabled())
        {
            String.LoadStringW(g_hThisInstance, IDS_MENU_DISABLE);
            MenuItemInfo.wID = IDM_DISABLE_DRV;
            MenuItemInfo.dwTypeData = String.GetBuffer();
            InsertMenuItemW(OwnerMenu, i, TRUE, &MenuItemInfo);
            i++;
        }
 
        if (DeviceNode->CanUninstall())
        {
            String.LoadStringW(g_hThisInstance, IDS_MENU_UNINSTALL);
            MenuItemInfo.wID = IDM_UNINSTALL_DRV;
            MenuItemInfo.dwTypeData = String.GetBuffer();
            InsertMenuItemW(OwnerMenu, i, TRUE, &MenuItemInfo);
            i++;
        }
 
        InsertMenuItemW(OwnerMenu, i, TRUE, &MenuSeparator);
        i++;
    }
 
    // All nodes have the scan option
    String.LoadStringW(g_hThisInstance, IDS_MENU_SCAN);
    MenuItemInfo.wID = IDM_SCAN_HARDWARE;
    MenuItemInfo.dwTypeData = String.GetBuffer();
    InsertMenuItemW(OwnerMenu, i, TRUE, &MenuItemInfo);
    i++;
 
    if ((Node->GetNodeType() == RootNode) || (MainMenu == true))
    {
        String.LoadStringW(g_hThisInstance, IDS_MENU_ADD);
        MenuItemInfo.wID = IDM_ADD_HARDWARE;
        MenuItemInfo.dwTypeData = String.GetBuffer();
        InsertMenuItemW(OwnerMenu, i, TRUE, &MenuItemInfo);
        i++;
    }
 
    if (Node->HasProperties())
    {
        InsertMenuItemW(OwnerMenu, i, TRUE, &MenuSeparator);
        i++;
 
        String.LoadStringW(g_hThisInstance, IDS_MENU_PROPERTIES);
        MenuItemInfo.wID = IDM_PROPERTIES;
        MenuItemInfo.dwTypeData = String.GetBuffer();
        InsertMenuItemW(OwnerMenu, i, TRUE, &MenuItemInfo);
        i++;
 
        SetMenuDefaultItem(OwnerMenu, IDC_PROPERTIES, FALSE);
    }
}

Referenced by CreateActionMenu(), and OnContextMenu().

CreateActionMenu()

bool CDeviceView::CreateActionMenu	(	_In_ HMENU	OwnerMenu,
		_In_ bool	MainMenu
	)

Definition at line 325 of file DeviceView.cpp.

{
    CNode *Node = GetSelectedNode();
    if (Node)
    {
        BuildActionMenuForNode(OwnerMenu, Node, MainMenu);
        return true;
    }
 
    return false;
}

Referenced by CDeviceManager::OnCommand(), and CDeviceManager::OnNotify().

DisplayPropertySheet()

void CDeviceView::DisplayPropertySheet

(

)

Definition at line 305 of file DeviceView.cpp.

{
    CNode *Node = GetSelectedNode();
    if (Node && Node->HasProperties())
    {
        DevicePropertiesExW(m_hTreeView,
                            NULL,
                            Node->GetDeviceId(),
                            1,//DPF_EXTENDED,
                            FALSE);
    }
}

Referenced by OnAction(), OnDoubleClick(), and CDeviceManager::OnNotify().

EmptyDeviceView()

void CDeviceView::EmptyDeviceView ( )

private

Definition at line 1048 of file DeviceView.cpp.

{
    (VOID)TreeView_DeleteAllItems(m_hTreeView);
}

Referenced by Uninitialize().

EmptyLists()

void CDeviceView::EmptyLists ( )

private

Definition at line 1123 of file DeviceView.cpp.

{
    CNode *Node;
 
    while (!m_ClassNodeList.IsEmpty())
    {
        Node = m_ClassNodeList.RemoveTail();
        delete Node;
    }
 
    while (!m_DeviceNodeList.IsEmpty())
    {
        Node = m_DeviceNodeList.RemoveTail();
        delete Node;
    }
}

Referenced by RefreshDeviceList().

EnableSelectedDevice()

bool CDeviceView::EnableSelectedDevice ( _In_ bool  Enable, _Out_ bool &  NeedsReboot  )

private

Definition at line 686 of file DeviceView.cpp.

{
    CDeviceNode *Node = dynamic_cast<CDeviceNode *>(GetSelectedNode());
    if (Node == nullptr)
        return false;
 
    if (Enable == false)
    {
        CAtlStringW str;
        if (str.LoadStringW(g_hThisInstance, IDS_CONFIRM_DISABLE))
        {
            if (MessageBoxW(m_hMainWnd,
                            str,
                            Node->GetDisplayName(),
                            MB_YESNO | MB_ICONWARNING | MB_DEFBUTTON2) != IDYES)
            {
                return false;
            }
        }
    }
 
    return Node->EnableDevice(Enable, NeedsReboot);
}

Referenced by OnAction().

GetChildDevice()

bool CDeviceView::GetChildDevice ( _In_ DEVINST  ParentDevInst, _Out_ PDEVINST  DevInst  )

private

Definition at line 780 of file DeviceView.cpp.

{
    CONFIGRET cr;
    cr = CM_Get_Child(DevInst,
                      ParentDevInst,
                      0);
    return (cr == CR_SUCCESS);
}

Referenced by RecurseChildDevices().

GetClassNode()

CClassNode * CDeviceView::GetClassNode ( _In_ LPGUID  ClassGuid )

private

Definition at line 1055 of file DeviceView.cpp.

{
    POSITION Pos;
    CClassNode *Node = nullptr;
 
    Pos = m_ClassNodeList.GetHeadPosition();
    if (Pos == NULL)
        return nullptr;
 
    do
    {
        Node = m_ClassNodeList.GetNext(Pos);
        if (IsEqualGUID(*Node->GetClassGuid(), *ClassGuid))
        {
            ATLASSERT(Node->GetNodeType() == ClassNode);
            break;
        }
 
        Node = nullptr;
 
    } while (Pos != NULL);
 
    return Node;
}

Referenced by ListDevicesByType().

GetCurrentView()

ViewType CDeviceView::GetCurrentView ( )

inline

Definition at line 78 of file DeviceView.h.

{ return m_ViewType; }

Referenced by OnAction(), CDeviceManager::OnCommand(), and CDeviceManager::OnCreate().

GetDeviceNode()

CDeviceNode * CDeviceView::GetDeviceNode ( _In_ DEVINST  Device )

private

Definition at line 1083 of file DeviceView.cpp.

{
    POSITION Pos;
    CDeviceNode *Node = nullptr;
 
    Pos = m_DeviceNodeList.GetHeadPosition();
    if (Pos == NULL)
        return nullptr;
 
    do
    {
        Node = m_DeviceNodeList.GetNext(Pos);
        if (Node->GetDeviceInst() == Device)
        {
            ATLASSERT(Node->GetNodeType() == DeviceNode);
            break;
        }
 
        Node = nullptr;
 
    } while (Pos != NULL);
 
    return Node;
}

Referenced by ListDevicesByType(), and RecurseChildDevices().

GetNextClass()

bool CDeviceView::GetNextClass ( _In_ ULONG  ClassIndex, _Out_ LPGUID  ClassGuid, _Out_ HDEVINFO *  hDevInfo  )

private

Definition at line 360 of file DeviceView.cpp.

{
    CONFIGRET cr;
 
    // Get the next class in the list
    cr = CM_Enumerate_Classes(ClassIndex,
                              ClassGuid,
                              0);
    if (cr != CR_SUCCESS)
        return false;
 
    // We only want the devices for this class
    *hDevInfo = SetupDiGetClassDevsW(ClassGuid,
                                     NULL,
                                     NULL,
                                     DIGCF_PRESENT);
 
    return (hDevInfo != INVALID_HANDLE_VALUE);
}

Referenced by ListDevicesByType(), and RefreshDeviceList().

GetNode()

CNode * CDeviceView::GetNode ( _In_ LPTV_ITEMW  TvItem )

private

Definition at line 1110 of file DeviceView.cpp.

{
    TvItem->mask = TVIF_PARAM;
    if (TreeView_GetItem(m_hTreeView, TvItem))
    {
        return (CNode *)TvItem->lParam;
    }
    return nullptr;
}

Referenced by GetSelectedNode().

GetSelectedNode()

CNode * CDeviceView::GetSelectedNode

(

)

Definition at line 341 of file DeviceView.cpp.

{
    TV_ITEM TvItem;
    TvItem.hItem = TreeView_GetSelection(m_hTreeView);
    return GetNode(&TvItem);
}

Referenced by CreateActionMenu(), DisplayPropertySheet(), EnableSelectedDevice(), OnContextMenu(), UninstallSelectedDevice(), UpdateSelectedDevice(), and CDeviceManager::UpdateToolbar().

GetSiblingDevice()

bool CDeviceView::GetSiblingDevice ( _In_ DEVINST  PrevDevice, _Out_ PDEVINST  DevInst  )

private

Definition at line 793 of file DeviceView.cpp.

{
    CONFIGRET cr;
    cr = CM_Get_Sibling(DevInst,
                        PrevDevice,
                        0);
    return (cr == CR_SUCCESS);
}

Referenced by RecurseChildDevices().

Initialize()

bool CDeviceView::Initialize

(

)

Definition at line 54 of file DeviceView.cpp.

{
    // Get the device image list
    m_ImageListData.cbSize = sizeof(SP_CLASSIMAGELIST_DATA);
    BOOL bSuccess = SetupDiGetClassImageList(&m_ImageListData);
    if (bSuccess == FALSE)
        return false;
 
    // Create the main treeview
    m_hTreeView = CreateWindowExW(WS_EX_CLIENTEDGE,
                                  WC_TREEVIEW,
                                  NULL,
                                  WS_CHILD | WS_VISIBLE | WS_BORDER | TVS_HASLINES |
                                  TVS_HASBUTTONS | TVS_SHOWSELALWAYS | TVS_LINESATROOT,
                                  0, 0, 0, 0,
                                  m_hMainWnd,
                                  (HMENU)IDC_TREEVIEW,
                                  g_hThisInstance,
                                  NULL);
    if (m_hTreeView)
    {
        // Set the image list against the treeview
        (void)TreeView_SetImageList(m_hTreeView,
                                    m_ImageListData.ImageList,
                                    TVSIL_NORMAL);
 
        // Give the treeview arrows instead of +/- boxes (on Win7)
        SetWindowTheme(m_hTreeView, L"explorer", NULL);
 
        // Create the root node
        m_RootNode = new CRootNode(&m_ImageListData);
        m_RootNode->SetupNode();
    }
 
 
 
    return !!(m_hTreeView);
}

Referenced by CDeviceManager::OnCreate().

InsertIntoTreeView()

HTREEITEM CDeviceView::InsertIntoTreeView ( _In_opt_ HTREEITEM  hParent, _In_ CNode *  Node  )

private

Definition at line 806 of file DeviceView.cpp.

{
    LPWSTR lpLabel;
    lpLabel = Node->GetDisplayName();
 
    TV_ITEMW tvi;
    TV_INSERTSTRUCT tvins;
    ZeroMemory(&tvi, sizeof(tvi));
    ZeroMemory(&tvins, sizeof(tvins));
 
    tvi.mask = TVIF_TEXT | TVIF_PARAM | TVIF_IMAGE | TVIF_SELECTEDIMAGE;
    tvi.pszText = lpLabel;
    tvi.cchTextMax = wcslen(lpLabel);
    tvi.lParam = (LPARAM)Node;
    tvi.iImage = Node->GetClassImage();
    tvi.iSelectedImage = Node->GetClassImage();
 
    // try to cast it to a device node. This will only succeed if it's the correct type
    CDeviceNode *DeviceNode = dynamic_cast<CDeviceNode *>(Node);
    if (DeviceNode && DeviceNode->GetOverlayImage())
    {
        tvi.mask |= TVIF_STATE;
        tvi.stateMask = TVIS_OVERLAYMASK;
        tvi.state = INDEXTOOVERLAYMASK(DeviceNode->GetOverlayImage());
    }
 
    tvins.item = tvi;
    tvins.hParent = hParent;
 
    return TreeView_InsertItem(m_hTreeView, &tvins);
}

Referenced by AddRootDevice(), ListDevicesByType(), and RecurseChildDevices().

ListDevicesByConnection()

bool CDeviceView::ListDevicesByConnection ( )

private

Definition at line 576 of file DeviceView.cpp.

{
    // Walk the device tree and add all the devices
    (void)RecurseChildDevices(m_RootNode->GetDeviceInst(), m_hTreeRoot);
 
    // Expand the root item
    (void)TreeView_Expand(m_hTreeView,
                          m_hTreeRoot,
                          TVE_EXPAND);
 
    return true;
}

ListDevicesByType()

bool CDeviceView::ListDevicesByType ( )

private

Definition at line 444 of file DeviceView.cpp.

{
    CClassNode *ClassNode;
    CDeviceNode *DeviceNode;
    HDEVINFO hDevInfo;
    HTREEITEM hTreeItem = NULL;
    GUID ClassGuid;
    INT ClassIndex;
    BOOL bClassSuccess, bSuccess;
 
    ClassIndex = 0;
    do
    {
        // Loop through all the device classes
        bClassSuccess = GetNextClass(ClassIndex, &ClassGuid, &hDevInfo);
        if (bClassSuccess)
        {
            bool AddedParent = false;
            INT DeviceIndex = 0;
            bool MoreItems = false;
 
            // Get the cached class node
            ClassNode = GetClassNode(&ClassGuid);
            if (ClassNode == nullptr)
            {
                ClassIndex++;
                continue;
            }
 
            // Check if this is a hidden class
            if (IsEqualGUID(ClassGuid, GUID_DEVCLASS_LEGACYDRIVER) ||
                IsEqualGUID(ClassGuid, GUID_DEVCLASS_VOLUME))
            {
                // Ignore this device if we aren't displaying hidden devices
                if (m_ShowHidden == FALSE)
                {
                    ClassIndex++;
                    continue;
                }
            }
 
            do
            {
                // Get a handle to all the devices in this class
                SP_DEVINFO_DATA DeviceInfoData;
                ZeroMemory(&DeviceInfoData, sizeof(SP_DEVINFO_DATA));
                DeviceInfoData.cbSize = sizeof(SP_DEVINFO_DATA);
                bSuccess = SetupDiEnumDeviceInfo(hDevInfo,
                                                 DeviceIndex,
                                                 &DeviceInfoData);
                if (bSuccess == FALSE && GetLastError() == ERROR_NO_MORE_ITEMS)
                    MoreItems = false;
 
                if (bSuccess)
                {
                    MoreItems = true;
 
                    // Get the cached device node
                    DeviceNode = GetDeviceNode(DeviceInfoData.DevInst);
                    if (DeviceNode == nullptr)
                    {
                        DeviceIndex++;
                        continue;
                    }
 
                    // Check if this is a hidden device
                    if (DeviceNode->IsHidden())
                    {
                        // Ignore this device if we aren't displaying hidden devices
                        if (m_ShowHidden == FALSE)
                        {
                            DeviceIndex++;
                            continue;
                        }
                    }
 
                    // We have a device, we need to add the parent if it hasn't yet been added
                    if (AddedParent == false)
                    {
                        // Insert the new class under the root item
                        hTreeItem = InsertIntoTreeView(m_hTreeRoot,
                                                       ClassNode);
                        AddedParent = true;
                    }
 
                    // Add the device under the class item node
                    (void)InsertIntoTreeView(hTreeItem, DeviceNode);
 
                    // Expand the class if it has a problem device
                    if (DeviceNode->HasProblem())
                    {
                        (void)TreeView_Expand(m_hTreeView,
                                              hTreeItem,
                                              TVE_EXPAND);
                    }
                }
 
                DeviceIndex++;
 
            } while (MoreItems);
 
            // If this class has devices, sort them alphabetically
            if (AddedParent == true)
            {
                (void)TreeView_SortChildren(m_hTreeView,
                                            hTreeItem,
                                            0);
            }
        }
 
        ClassIndex++;
 
    } while (bClassSuccess);
 
    // Sort the classes alphabetically
    (void)TreeView_SortChildren(m_hTreeView,
                                m_hTreeRoot,
                                0);
 
    // Expand the root item
    (void)TreeView_Expand(m_hTreeView,
                          m_hTreeRoot,
                          TVE_EXPAND);
 
    // Pre-select the root item
    (VOID)TreeView_SelectItem(m_hTreeView,
                              m_hTreeRoot);
 
    return 0;
}

OnAction()

LRESULT CDeviceView::OnAction

(

UINT

Action

)

Definition at line 243 of file DeviceView.cpp.

{
    switch (Action)
    {
        case IDM_PROPERTIES:
        {
            DisplayPropertySheet();
            break;
        }
 
        case IDM_SCAN_HARDWARE:
        {
            Refresh(GetCurrentView(),
                    true,
                    true);
            break;
        }
 
        case IDM_ENABLE_DRV:
        {
            bool NeedsReboot;
            if (EnableSelectedDevice(true, NeedsReboot) &&
                NeedsReboot)
            {
                MessageBox(m_hMainWnd, L"Rebooting", L"Enable", MB_OK);
            }
            break;
        }
 
        case IDM_DISABLE_DRV:
        {
            bool NeedsReboot;
            EnableSelectedDevice(false, NeedsReboot);
            break;
        }
 
        case IDM_UPDATE_DRV:
        {
            bool NeedsReboot;
            UpdateSelectedDevice(NeedsReboot);
            break;
        }
 
        case IDM_UNINSTALL_DRV:
        {
            UninstallSelectedDevice();
            break;
        }
 
        case IDM_ADD_HARDWARE:
        {
            RunAddHardwareWizard();
            break;
        }
    }
 
    return 0;
}

Referenced by CDeviceManager::OnCommand().

OnContextMenu()

LRESULT CDeviceView::OnContextMenu

(

_In_ LPARAM

lParam

)

Definition at line 169 of file DeviceView.cpp.

{
    HTREEITEM hSelected = TreeView_GetSelection(m_hTreeView);
 
    RECT rc;
    if (TreeView_GetItemRect(m_hTreeView,
                             hSelected,
                             &rc,
                             TRUE))
    {
        POINT pt;
        if (GetCursorPos(&pt) &&
            ScreenToClient(m_hTreeView, &pt) &&
            PtInRect(&rc, pt))
        {
            CNode *Node = GetSelectedNode();
            if (Node)
            {
                // Create the context menu
                HMENU hContextMenu = CreatePopupMenu();
 
                // Add the actions for this node
                BuildActionMenuForNode(hContextMenu, Node, false);
 
                INT xPos = GET_X_LPARAM(lParam);
                INT yPos = GET_Y_LPARAM(lParam);
 
                // Display the menu
                TrackPopupMenuEx(hContextMenu,
                                 TPM_RIGHTBUTTON,
                                 xPos,
                                 yPos,
                                 m_hMainWnd,
                                 NULL);
 
                DestroyMenu(hContextMenu);
            }
        }
    }
 
    return 0;
}

Referenced by CDeviceManager::OnContext().

OnDoubleClick()

LRESULT CDeviceView::OnDoubleClick

(

_In_ LPNMHDR

NmHdr

)

Definition at line 128 of file DeviceView.cpp.

{
    TVHITTESTINFO hitInfo;
    HTREEITEM hItem;
 
    GetCursorPos(&hitInfo.pt);
    ScreenToClient(m_hTreeView, &hitInfo.pt);
 
    // Check if we are trying to double-click an item
    hItem = TreeView_HitTest(m_hTreeView, &hitInfo);
    if (hItem != NULL && (hitInfo.flags & (TVHT_ONITEM | TVHT_ONITEMICON)))
    {
        DisplayPropertySheet();
    }
 
    return 0;
}

Referenced by CDeviceManager::OnNotify().

OnRightClick()

LRESULT CDeviceView::OnRightClick

(

_In_ LPNMHDR

NmHdr

)

Definition at line 149 of file DeviceView.cpp.

{
    TVHITTESTINFO hitInfo;
    HTREEITEM hItem;
 
    GetCursorPos(&hitInfo.pt);
    ScreenToClient(m_hTreeView, &hitInfo.pt);
 
    hItem = TreeView_HitTest(m_hTreeView, &hitInfo);
    if (hItem != NULL && (hitInfo.flags & (TVHT_ONITEM | TVHT_ONITEMICON)))
    {
        TreeView_SelectItem(m_hTreeView, hItem);
    }
 
    return 0;
}

Referenced by CDeviceManager::OnNotify().

OnSize()

LRESULT CDeviceView::OnSize	(	_In_ int	x,
		_In_ int	y,
		_In_ int	cx,
		_In_ int	cy
	)

Definition at line 108 of file DeviceView.cpp.

{
    // Resize the treeview
    SetWindowPos(m_hTreeView,
                 NULL,
                 x,
                 y,
                 cx,
                 cy,
                 SWP_NOZORDER);
 
    return 0;
}

Referenced by CDeviceManager::OnSize().

RecurseChildDevices()

bool CDeviceView::RecurseChildDevices ( _In_ DEVINST  ParentDevice, _In_ HTREEITEM  hParentTreeItem  )

private

Definition at line 590 of file DeviceView.cpp.

{
    HTREEITEM hDevItem = NULL;
    DEVINST Device;
    bool HasProblem = false;
    bool bSuccess;
 
    // Check if the parent has any child devices
    if (GetChildDevice(ParentDevice, &Device) == FALSE)
        return true;
 
    // Get the cached device node
    CDeviceNode *DeviceNode;
    DeviceNode = dynamic_cast<CDeviceNode *>(GetDeviceNode(Device));
    if (DeviceNode == nullptr)
    {
        return false;
    }
 
    // Don't show hidden devices if not requested
    if ((m_ShowHidden == TRUE) || (!(DeviceNode->IsHidden())))
    {
        // Add this device to the tree under its parent
        hDevItem = InsertIntoTreeView(hParentTreeItem,
                                      DeviceNode);
        if (hDevItem)
        {
            // Check if this child has any children itself
            if (!RecurseChildDevices(Device, hDevItem))
                HasProblem = true;
        }
 
        if (DeviceNode->HasProblem())
        {
            HasProblem = true;
        }
    }
 
 
    // Check for siblings
    for (;;)
    {
        // Check if the parent device has anything at the same level
        bSuccess = GetSiblingDevice(Device, &Device);
        if (bSuccess == FALSE)
            break;
 
        DeviceNode = dynamic_cast<CDeviceNode *>(GetDeviceNode(Device));
        if (DeviceNode == nullptr)
        {
            continue;
        }
 
        // Don't show hidden devices if not requested
        if ((m_ShowHidden == TRUE) || (!(DeviceNode->IsHidden())))
        {
            if (DeviceNode->HasProblem())
            {
                HasProblem = true;
            }
 
            // Add this device to the tree under its parent
            hDevItem = InsertIntoTreeView(hParentTreeItem,
                                          DeviceNode);
            if (hDevItem)
            {
                // Check if this child has any children itself
                if (!RecurseChildDevices(Device, hDevItem))
                    HasProblem = true;
            }
        }
    }
 
    (void)TreeView_SortChildren(m_hTreeView,
                                hParentTreeItem,
                                0);
 
    // Expand the class if it has a problem device
    if (HasProblem == true)
    {
        (void)TreeView_Expand(m_hTreeView,
                              hParentTreeItem,
                              TVE_EXPAND);
    }
 
    // If there was a problem, expand the ancestors
    if (HasProblem)
        return false;
 
    return true;
}

Referenced by ListDevicesByConnection(), and RecurseChildDevices().

RecurseFindDevice()

HTREEITEM CDeviceView::RecurseFindDevice ( _In_ HTREEITEM  hParentItem, _In_ CNode *  Node  )

private

Definition at line 939 of file DeviceView.cpp.

{
    HTREEITEM FoundItem;
    HTREEITEM hItem;
    TVITEMW tvItem;
    CNode *FoundNode;
 
    // Check if this node has any children
    hItem = TreeView_GetChild(m_hTreeView, hParentItem);
    if (hItem == NULL)
        return NULL;
 
    // The lParam contains the node pointer data
    tvItem.hItem = hItem;
    tvItem.mask = TVIF_PARAM;
    if (TreeView_GetItem(m_hTreeView, &tvItem) &&
        tvItem.lParam != NULL)
    {
        // check for a matching node
        FoundNode = reinterpret_cast<CNode *>(tvItem.lParam);
        if ((FoundNode->GetNodeType() == Node->GetNodeType()) &&
            (IsEqualGUID(*FoundNode->GetClassGuid(), *Node->GetClassGuid())))
        {
            // check if this is a class node, or a device with matching ID's
            if ((FoundNode->GetNodeType() == ClassNode) ||
                (wcscmp(FoundNode->GetDeviceId(), Node->GetDeviceId()) == 0))
            {
                return hItem;
            }
        }
    }
 
    // This node may have its own children
    FoundItem = RecurseFindDevice(hItem, Node);
    if (FoundItem)
        return FoundItem;
 
    // Loop all the siblings
    for (;;)
    {
        // Get the next item at this level
        hItem = TreeView_GetNextSibling(m_hTreeView, hItem);
        if (hItem == NULL)
            break;
 
        // The lParam contains the node pointer data
        tvItem.hItem = hItem;
        tvItem.mask = TVIF_PARAM;
        if (TreeView_GetItem(m_hTreeView, &tvItem))
        {
            // check for a matching class
            FoundNode = reinterpret_cast<CNode *>(tvItem.lParam);
            if ((FoundNode->GetNodeType() == Node->GetNodeType()) &&
                (IsEqualGUID(*FoundNode->GetClassGuid(), *Node->GetClassGuid())))
            {
                // check if this is a class node, or a device with matching ID's
                if ((FoundNode->GetNodeType() == ClassNode) ||
                    (wcscmp(FoundNode->GetDeviceId(), Node->GetDeviceId()) == 0))
                {
                    return hItem;
                }
            }
        }
 
        // This node may have its own children
        FoundItem = RecurseFindDevice(hItem, Node);
        if (FoundItem)
            return FoundItem;
    }
 
    return hItem;
}

Referenced by RecurseFindDevice(), and SelectNode().

Refresh()

void CDeviceView::Refresh	(	_In_ ViewType	Type,
		_In_ bool	ScanForChanges,
		_In_ bool	UpdateView
	)

Definition at line 216 of file DeviceView.cpp.

{
    // Enum devices on a separate thread to keep the gui responsive
 
    m_ViewType = Type;
 
    RefreshThreadData *ThreadData;
    ThreadData = new RefreshThreadData;
    ThreadData->This = this;
    ThreadData->ScanForChanges = ScanForChanges;
    ThreadData->UpdateView = UpdateView;
 
    HANDLE hThread;
    hThread = (HANDLE)_beginthreadex(NULL,
                                     0,
                                     RefreshThread,
                                     ThreadData,
                                     0,
                                     NULL);
    if (hThread) CloseHandle(hThread);
}

Referenced by CDeviceManager::RefreshView().

RefreshDeviceList()

bool CDeviceView::RefreshDeviceList ( )

private

Definition at line 1141 of file DeviceView.cpp.

{
    GUID ClassGuid;
    CClassNode *ClassNode;
    CDeviceNode *DeviceNode;
    HDEVINFO hDevInfo;
    SP_DEVINFO_DATA DeviceInfoData;
    DWORD i;
    BOOL Success;
 
    ULONG ClassIndex = 0;
 
    EmptyLists();
 
    if (m_RootNode) delete m_RootNode;
    m_RootNode = new CRootNode(&m_ImageListData);
    m_RootNode->SetupNode();
 
    // Loop through all the classes
    do
    {
        Success = GetNextClass(ClassIndex, &ClassGuid, &hDevInfo);
        if (Success)
        {
            // Create a new class node and add it to the list
            ClassNode = new CClassNode(&ClassGuid, &m_ImageListData);
            if (ClassNode->SetupNode())
            {
                m_ClassNodeList.AddTail(ClassNode);
            }
 
            SetupDiDestroyDeviceInfoList(hDevInfo);
        }
        ClassIndex++;
    } while (Success);
 
    // Get all the devices on the local machine
    hDevInfo = SetupDiGetClassDevsW(NULL,
                                    0,
                                    0,
                                    DIGCF_PRESENT | DIGCF_ALLCLASSES);
    if (hDevInfo == INVALID_HANDLE_VALUE)
    {
        return false;
    }
 
    // loop though all the devices
    DeviceInfoData.cbSize = sizeof(SP_DEVINFO_DATA);
    for (i = 0;; i++)
    {
        // Get the devinst for this device
        Success = SetupDiEnumDeviceInfo(hDevInfo, i, &DeviceInfoData);
        if (Success == FALSE)
            break;
 
        // create a new device node and add it to the list
        DeviceNode = new CDeviceNode(DeviceInfoData.DevInst, &m_ImageListData);
        if (DeviceNode->SetupNode())
        {
            m_DeviceNodeList.AddTail(DeviceNode);
        }
        else
        {
            ATLASSERT(FALSE);
        }
    }
 
    SetupDiDestroyDeviceInfoList(hDevInfo);
 
    return TRUE;
}

RefreshThread()

unsigned int __stdcall CDeviceView::RefreshThread ( void *  Param )

staticprivate

Definition at line 384 of file DeviceView.cpp.

{
    RefreshThreadData *ThreadData = (RefreshThreadData *)Param;
    CDeviceView *This = ThreadData->This;
 
    // Get a copy of the currently selected node
    CNode *LastSelectedNode = This->GetSelectedNode();
    if (LastSelectedNode == nullptr || (LastSelectedNode->GetNodeType() == RootNode))
    {
        LastSelectedNode = new CRootNode(*This->m_RootNode);
    }
    else if (LastSelectedNode->GetNodeType() == ClassNode)
    {
        LastSelectedNode = new CClassNode(*dynamic_cast<CClassNode *>(LastSelectedNode));
    }
    else if (LastSelectedNode->GetNodeType() == DeviceNode)
    {
        LastSelectedNode = new CDeviceNode(*dynamic_cast<CDeviceNode *>(LastSelectedNode));
    }
 
    // Empty the treeview
    This->EmptyDeviceView();
 
    // Re-add the root node to the tree
    if (This->AddRootDevice() == false)
        return 0;
 
    // Refresh the devices only if requested
    if (ThreadData->ScanForChanges)
    {
        This->RefreshDeviceList();
    }
 
    // display the type of view the user wants
    switch (This->m_ViewType)
    {
        case DevicesByType:
            (void)This->ListDevicesByType();
            break;
 
        case DevicesByConnection:
            (VOID)This->ListDevicesByConnection();
            break;
 
        case ResourcesByType:
            break;
 
        case ResourcesByConnection:
            break;
    }
 
    This->SelectNode(LastSelectedNode);
 
    delete ThreadData;
 
    return 0;
}

Referenced by Refresh().

RunAddHardwareWizard()

bool CDeviceView::RunAddHardwareWizard ( )

private

Definition at line 756 of file DeviceView.cpp.

{
    PADDHARDWAREWIZARD pAddHardwareWizard;
    HMODULE hModule;
 
    hModule = LoadLibraryW(L"hdwwiz.cpl");
    if (hModule == NULL)
        return false;
 
    pAddHardwareWizard = (PADDHARDWAREWIZARD)GetProcAddress(hModule,
                                                            "AddHardwareWizard");
    if (pAddHardwareWizard == NULL)
    {
        FreeLibrary(hModule);
        return false;
    }
 
    pAddHardwareWizard(m_hMainWnd, NULL);
 
    FreeLibrary(hModule);
    return true;
}

Referenced by OnAction().

SelDeviceIsInstalled()

bool CDeviceView::SelDeviceIsInstalled

(

)

SelDeviceIsStarted()

bool CDeviceView::SelDeviceIsStarted

(

)

SelectNode()

void CDeviceView::SelectNode ( _In_ CNode *  Node )

private

Definition at line 1016 of file DeviceView.cpp.

{
    HTREEITEM hRoot, hItem;
 
    // Check if there are any items in the tree
    hRoot = TreeView_GetRoot(m_hTreeView);
    if (hRoot == NULL)
        return;
 
    // If we don't want to set select a node, just select root
    if (Node == nullptr || Node->GetNodeType() == RootNode)
    {
        TreeView_SelectItem(m_hTreeView, hRoot);
        return;
    }
 
    // Scan the tree looking for the node we want
    hItem = RecurseFindDevice(hRoot, Node);
    if (hItem)
    {
        TreeView_SelectItem(m_hTreeView, hItem);
    }
    else
    {
        TreeView_SelectItem(m_hTreeView, hRoot);
    }
}

SetFocus()

void CDeviceView::SetFocus

(

)

Definition at line 319 of file DeviceView.cpp.

{
    ::SetFocus(m_hTreeView);
}

Referenced by CDeviceManager::OnActivate(), CDeviceManager::OnCommand(), and SetFocus().

SetHiddenDevices()

VOID CDeviceView::SetHiddenDevices ( _In_ bool  ShowHidden )

inline

Definition at line 73 of file DeviceView.h.

    {
        m_ShowHidden = ShowHidden;
    }

Referenced by CDeviceManager::OnCommand().

Uninitialize()

bool CDeviceView::Uninitialize

(

)

Definition at line 94 of file DeviceView.cpp.

{
    EmptyDeviceView();
 
    if (m_ImageListData.ImageList != NULL)
    {
        SetupDiDestroyClassImageList(&m_ImageListData);
        ZeroMemory(&m_ImageListData, sizeof(SP_CLASSIMAGELIST_DATA));
    }
 
    return true;
}

Referenced by CDeviceManager::OnDestroy().

UninstallSelectedDevice()

bool CDeviceView::UninstallSelectedDevice ( )

private

Definition at line 733 of file DeviceView.cpp.

{
    CDeviceNode *Node = dynamic_cast<CDeviceNode *>(GetSelectedNode());
    if (Node == nullptr)
        return false;
 
    CAtlStringW str;
    if (str.LoadStringW(g_hThisInstance, IDS_CONFIRM_UNINSTALL))
    {
        if (MessageBoxW(m_hMainWnd,
                        str,
                        Node->GetDisplayName(),
                        MB_YESNO | MB_ICONWARNING | MB_DEFBUTTON2) != IDYES)
        {
            return false;
        }
    }
 
    return Node->UninstallDevice();
}

Referenced by OnAction().

UpdateSelectedDevice()

bool CDeviceView::UpdateSelectedDevice ( _Out_ bool &  NeedsReboot )

private

Definition at line 714 of file DeviceView.cpp.

{
    CDeviceNode *Node = dynamic_cast<CDeviceNode *>(GetSelectedNode());
    if (Node == nullptr)
        return false;
 
    DWORD dwReboot;
    if (InstallDevInst(m_hMainWnd, Node->GetDeviceId(), TRUE, &dwReboot))
    {
        NeedsReboot = false;
        return true;
    }
 
    return false;
}

Referenced by OnAction().

Member Data Documentation

m_ClassNodeList

CAtlList<CClassNode *> CDeviceView::m_ClassNodeList

private

Definition at line 27 of file DeviceView.h.

Referenced by EmptyLists(), GetClassNode(), and RefreshDeviceList().

m_DeviceNodeList

CAtlList<CDeviceNode *> CDeviceView::m_DeviceNodeList

private

Definition at line 28 of file DeviceView.h.

Referenced by EmptyLists(), GetDeviceNode(), and RefreshDeviceList().

m_hMainWnd

HWND CDeviceView::m_hMainWnd

private

Definition at line 18 of file DeviceView.h.

Referenced by EnableSelectedDevice(), Initialize(), OnAction(), OnContextMenu(), RunAddHardwareWizard(), UninstallSelectedDevice(), and UpdateSelectedDevice().

m_hMenu

HMENU CDeviceView::m_hMenu

private

Definition at line 21 of file DeviceView.h.

m_hPropertyDialog

HWND CDeviceView::m_hPropertyDialog

private

Definition at line 20 of file DeviceView.h.

m_hTreeRoot

HTREEITEM CDeviceView::m_hTreeRoot

private

Definition at line 23 of file DeviceView.h.

Referenced by AddRootDevice(), ListDevicesByConnection(), and ListDevicesByType().

m_hTreeView

HWND CDeviceView::m_hTreeView

private

Definition at line 19 of file DeviceView.h.

Referenced by DisplayPropertySheet(), EmptyDeviceView(), GetNode(), GetSelectedNode(), Initialize(), InsertIntoTreeView(), ListDevicesByConnection(), ListDevicesByType(), OnContextMenu(), OnDoubleClick(), OnRightClick(), OnSize(), RecurseChildDevices(), RecurseFindDevice(), SelectNode(), and SetFocus().

m_ImageListData

SP_CLASSIMAGELIST_DATA CDeviceView::m_ImageListData

private

Definition at line 29 of file DeviceView.h.

Referenced by CDeviceView(), Initialize(), RefreshDeviceList(), and Uninitialize().

m_RootNode

CRootNode* CDeviceView::m_RootNode

private

Definition at line 26 of file DeviceView.h.

Referenced by AddRootDevice(), Initialize(), ListDevicesByConnection(), and RefreshDeviceList().

m_ShowHidden

bool CDeviceView::m_ShowHidden

private

Definition at line 24 of file DeviceView.h.

Referenced by ListDevicesByType(), RecurseChildDevices(), and SetHiddenDevices().

m_ViewType

ViewType CDeviceView::m_ViewType

private

Definition at line 22 of file DeviceView.h.

Referenced by GetCurrentView(), and Refresh().

---

The documentation for this class was generated from the following files:

• dll/win32/devmgr/devmgmt/DeviceView.h
• dll/win32/devmgr/devmgmt/DeviceView.cpp