windows-nt/Source/XPSP1/NT/shell/osshell/snapins/devmgr/snapin/cnode.h

559 lines
13 KiB
C
Raw Normal View History

2020-09-26 03:20:57 -05:00
#ifndef __CNODE_H__
#define __CNODE_H__
/*++
Copyright (C) 1997-1999 Microsoft Corporation
Module Name:
cnode.h
Abstract:
header file for cnode.cpp
Author:
William Hsieh (williamh) created
Revision History:
--*/
class CResultItem;
class CResultComputer;
class CResultClass;
class CResultDevice;
class CDriverList;
class CDriver;
class CHwProfile;
class CMachine;
class CItemIdentifier;
class CDeviceIdentifier;
class CClassIdentifier;
class CComputerIdentifier;
class CResourceIdentifier;
class CResourceTypeIdentifier;
#define ALL_LOG_CONF BOOT_LOG_CONF + ALLOC_LOG_CONF + FORCED_LOG_CONF
inline
COOKIE_TYPE CookieType(RESOURCEID ResType)
{
if (ResType_Mem == ResType)
{
return COOKIE_TYPE_RESULTITEM_RESOURCE_MEMORY;
}
else if (ResType_IO == ResType)
{
return COOKIE_TYPE_RESULTITEM_RESOURCE_IO;
}
else if (ResType_DMA == ResType)
{
return COOKIE_TYPE_RESULTITEM_RESOURCE_DMA;
}
else if (ResType_IRQ == ResType)
{
return COOKIE_TYPE_RESULTITEM_RESOURCE_IRQ;
}
else
{
ASSERT(FALSE);
return COOKIE_TYPE_UNKNOWN;
}
}
///
/// class to represent a result pane item
///
class CResultItem
{
public:
CResultItem() : m_pMachine(NULL)
{}
virtual ~CResultItem() {};
LPCTSTR GetDisplayName() const
{
return (LPCTSTR)m_strDisplayName;
}
int GetImageIndex()
{
return m_iImage;
}
virtual CItemIdentifier* CreateIdentifier() = 0;
CMachine* m_pMachine;
protected:
int m_iImage;
String m_strDisplayName;
};
class CClass : public CResultItem
{
public:
CClass(CMachine* pMachine, LPGUID pGuid);
virtual ~CClass();
BOOL GetFirstDevice(CDevice** ppDevice, PVOID& pContext);
BOOL GetNextDevice(CDevice** ppDevice, PVOID& Context);
operator GUID&()
{
return m_Guid;
}
operator LPGUID()
{
return &m_Guid;
}
BOOL operator ==(const CClass& OtherClass)
{
return IsEqualGUID(m_Guid, (GUID&)OtherClass);
}
CDevInfoList* GetDevInfoList(HWND hwndParent = NULL);
HICON LoadIcon();
void AddDevice(CDevice* pDevice);
int GetNumberOfDevices(BOOL Hidden = FALSE)
{
return Hidden ? m_TotalDevices :
m_TotalDevices - m_TotalHiddenDevices;
}
BOOL NoDisplay()
{
return m_NoDisplay;
}
virtual CItemIdentifier* CreateIdentifier();
void PropertyChanged();
CPropSheetData m_psd;
private:
CClass(const CClass& ClassSrc);
CClass& operator=(const CClass& ClassSrc);
GUID m_Guid;
CList<CDevice*, CDevice*> m_listDevice;
BOOL m_NoDisplay;
POSITION m_pos;
CDevInfoList* m_pDevInfoList;
int m_TotalDevices;
int m_TotalHiddenDevices;
};
class CDevice : public CResultItem
{
public:
CDevice() : m_pClass(NULL), m_pSibling(NULL), m_pChild(NULL),
m_pParent(NULL), m_bLaunchTroubleShooter(FALSE)
{}
CDevice(CMachine* pMachine, CClass* pClass, PSP_DEVINFO_DATA pDevData);
operator SP_DEVINFO_DATA&()
{
return m_DevData;
}
operator PSP_DEVINFO_DATA()
{
return &m_DevData;
}
BOOL operator ==(LPCTSTR DeviceID)
{
return (0 == m_strDeviceID.CompareNoCase(DeviceID));
}
BOOL operator ==(CDevice& OtherDevice);
CDevice* GetChild()
{
return m_pChild;
}
CDevice* GetParent()
{
return m_pParent;
}
CDevice* GetSibling()
{
return m_pSibling;
}
void SetChild(CDevice* pDevice)
{
m_pChild = pDevice;
}
void SetParent(CDevice* pDevice)
{
m_pParent = pDevice;
}
void SetSibling(CDevice* pDevice)
{
m_pSibling = pDevice;
}
CClass* GetClass()
{
return m_pClass;
}
virtual DEVNODE GetDevNode()
{
return m_DevData.DevInst;
}
LPCTSTR GetDeviceID() const
{
return (m_strDeviceID.IsEmpty()) ? NULL : (LPCTSTR)m_strDeviceID;
}
void SetClass(CClass* pClass)
{
m_pClass = pClass;
}
BOOL IsHidden();
BOOL IsPhantom();
BOOL NoShowInDM();
BOOL IsPCMCIA();
BOOL IsPCIDevice();
BOOL IsPnpDevice();
BOOL IsBiosDevice();
BOOL GetMFG(TCHAR* pBuffer, DWORD Size, DWORD* pRequiredSize)
{
return m_pMachine->DiGetDeviceMFGString(&m_DevData, pBuffer, Size, pRequiredSize);
}
virtual CItemIdentifier* CreateIdentifier();
CDevice* FindMFParent();
HICON LoadClassIcon();
BOOL GetStatus(DWORD* pStatus, DWORD* pProblem);
BOOL GetCapabilities(DWORD* pCapabilities);
BOOL GetPowerCapabilities(DWORD* pCapabilities);
BOOL IsRAW();
BOOL IsUninstallable();
BOOL IsDisableable();
BOOL IsDisabled();
BOOL IsStateDisabled();
BOOL IsStarted();
BOOL IsMFChild();
BOOL IsSpecialMFChild();
BOOL HasProblem();
BOOL NeedsRestart();
BOOL GetConfigFlags(DWORD* pFlags);
BOOL GetConfigSpecificConfigFlags(DWORD* pCSStatus);
BOOL GetKnownLogConf(LOG_CONF* plc, DWORD* plcType);
BOOL HasResources();
void GetMFGString(String& strMFG);
void GetProviderString(String& strMFG);
void GetDriverDateString(String& strMFG);
void GetDriverVersionString(String& strMFG);
LPCTSTR GetClassDisplayName();
void ClassGuid(GUID& ClassGuid)
{
ASSERT(m_pClass);
ClassGuid = *m_pClass;
}
BOOL NoChangeUsage();
CDriver* CreateDriver();
BOOL HasDrivers();
DWORD EnableDisableDevice(HWND hDlg, BOOL Enabling);
void PropertyChanged();
CPropSheetData m_psd;
BOOL m_bLaunchTroubleShooter;
private:
CDevice(const CDevice& DeviceSrc);
CDevice& operator=(const CDevice& DeviceSrc);
CDevice* m_pParent;
CDevice* m_pSibling;
CDevice* m_pChild;
String m_strDeviceID;
SP_DEVINFO_DATA m_DevData;
CClass* m_pClass;
};
class CComputer : public CDevice
{
public:
CComputer(CMachine* pMachine, DEVNODE dnRoot);
virtual DEVNODE GetDevNode()
{
return m_dnRoot;
}
virtual CItemIdentifier* CreateIdentifier();
private:
DEVNODE m_dnRoot;
CComputer(const CComputer& ComputerSrc);
CComputer& operator=(const CComputer& ComputerSrc);
};
class CResource : public CResultItem
{
public:
CResource(CDevice* pDevice, RESOURCEID ResType, DWORDLONG dlBase, DWORDLONG dlLen,
BOOL Forced, BOOL Free = FALSE);
BOOL IsForced()
{
return m_Forced;
}
void GetValue(DWORDLONG* pdlBase, DWORDLONG* pdlLen) const
{
ASSERT(pdlBase && pdlLen);
*pdlBase = m_dlBase;
*pdlLen = m_dlLen;
}
virtual CItemIdentifier* CreateIdentifier();
void GetRangeString(String& strRange);
LPCTSTR GetViewName()
{
return (LPCTSTR)m_strViewName;
}
CDevice* GetDevice()
{
return m_pDevice;
}
CResource* GetChild()
{
return m_pChild;
}
CResource* GetSibling()
{
return m_pSibling;
}
CResource* GetParent()
{
return m_pParent;
}
void SetChild(CResource* pRes)
{
m_pChild = pRes;
}
void SetParent(CResource* pRes)
{
m_pParent = pRes;
}
void SetSibling(CResource* pRes)
{
m_pSibling = pRes;
}
BOOL operator <=(const CResource& resSrc);
BOOL SameRange(const CResource& resSrc);
BOOL EnclosedBy(const CResource& resSrc);
RESOURCEID ResType()
{
return m_ResType;
}
CDevice* m_pDevice; // BUGBUG - this should not be public
private:
CResource(const CResource& resSrc);
CResource& operator=(const CResource& resSrc);
RESOURCEID m_ResType;
DWORDLONG m_dlBase;
DWORDLONG m_dlEnd;
DWORDLONG m_dlLen;
BOOL m_Forced;
CResource* m_pChild;
CResource* m_pSibling;
CResource* m_pParent;
// COOKIE_TYPE m_CookieType;
BOOL m_Allocated;
String m_strViewName;
};
#define pIRQResData(pData) ((IRQ_RESOURCE*)pData)
#define pDMAResData(pData) ((DMA_RESOURCE*)pData)
#define pMemResData(pData) ((MEM_RESOURCE*)pData)
#define pIOResData(pData) ((IO_RESOURCE*)pData)
class CResourceType : public CResultItem
{
public:
CResourceType(CMachine* pMachine, RESOURCEID ResType);
virtual CItemIdentifier* CreateIdentifier();
CResource* GetChild()
{
return m_pChild;
}
CResourceType* GetSibling()
{
return m_pSibling;
}
CComputer* GetParent()
{
return m_pParent;
}
void SetChild(CResource* pRes)
{
m_pChild = pRes;
}
void SetParent(CComputer* pComp)
{
m_pParent = pComp;
}
void SetSibling(CResourceType* pResT)
{
m_pSibling = pResT;
}
RESOURCEID GetResType()
{
return m_ResType;
}
private:
CResourceType(const CResourceType& resSrc);
CResourceType& operator=(const CResourceType& resSrc);
RESOURCEID m_ResType;
CResource* m_pChild;
CResourceType* m_pSibling;
CComputer* m_pParent;
};
class CResourceList
{
public:
CResourceList(CDevice* pDevice, RESOURCEID ResType,
ULONG LogConfType = ALLOC_LOG_CONF, ULONG AltLogConfType = BOOT_LOG_CONF);
CResourceList(CMachine* pMachine, RESOURCEID ResType,
ULONG LogConfType = ALLOC_LOG_CONF, ULONG AltLogConfType = BOOT_LOG_CONF);
~CResourceList();
int GetCount()
{
return m_listRes.GetCount();
}
BOOL GetFirst(CResource** ppRes, PVOID& pContext);
BOOL GetNext(CResource** ppRes, PVOID& Context);
BOOL CreateResourceTree(CResource** ppRoot);
static BOOL ExtractResourceValue(RESOURCEID ResType, PVOID pData,
DWORDLONG* pdlBase, DWORDLONG* pdlLen);
private:
BOOL InsertResourceToTree(CResource* pRes, CResource* pResRoot, BOOL ForcedInsert);
CResourceList(const CResourceList& Src);
CResourceList& operator=(const CResourceList& Src);
void CreateSubtreeResourceList(CDevice* pDeviceStart, RESOURCEID ResType,
ULONG LogConfType, ULONG AltLogConfType);
CList<CResource*, CResource*> m_listRes;
void InsertResourceToList(CResource* pRes);
#define pIRQResData(pData) ((IRQ_RESOURCE*)pData)
#define pDMAResData(pData) ((DMA_RESOURCE*)pData)
#define pMemResData(pData) ((MEM_RESOURCE*)pData)
#define pIOResData(pData) ((IO_RESOURCE*)pData)
};
class CItemIdentifier
{
public:
CItemIdentifier()
{}
virtual ~CItemIdentifier()
{}
virtual BOOL operator==(CCookie& Cookie) = 0;
};
class CClassIdentifier : public CItemIdentifier
{
public:
CClassIdentifier(CClass& Class)
{
m_Guid = (GUID&)Class;
}
virtual BOOL operator==(CCookie& Cookie)
{
return COOKIE_TYPE_RESULTITEM_CLASS == Cookie.GetType() &&
IsEqualGUID(m_Guid, *((CClass*)Cookie.GetResultItem()));
}
private:
GUID m_Guid;
};
class CDeviceIdentifier : public CItemIdentifier
{
public:
CDeviceIdentifier(CDevice& Device)
{
ASSERT(Device.GetDeviceID());
m_strDeviceId = Device.GetDeviceID();
}
virtual BOOL operator==(CCookie& Cookie)
{
return COOKIE_TYPE_RESULTITEM_DEVICE == Cookie.GetType() &&
!lstrcmpi(m_strDeviceId, ((CDevice*)Cookie.GetResultItem())->GetDeviceID());
}
private:
String m_strDeviceId;
};
class CResourceIdentifier : public CItemIdentifier
{
public:
CResourceIdentifier(CResource& Res)
{
m_CookieType = CookieType(Res.ResType());
m_strDisplayName = Res.GetDisplayName();
}
virtual BOOL operator==(CCookie& Cookie)
{
return m_CookieType == Cookie.GetType() &&
!lstrcmpi((LPCTSTR)m_strDisplayName, Cookie.GetResultItem()->GetDisplayName());
}
private:
COOKIE_TYPE m_CookieType;
String m_strDisplayName;
};
class CResourceTypeIdentifier : public CItemIdentifier
{
public:
CResourceTypeIdentifier(CResourceType& ResType)
{
m_ResType = ResType.GetResType();
}
virtual BOOL operator==(CCookie& Cookie)
{
return COOKIE_TYPE_RESULTITEM_RESTYPE == Cookie.GetType() &&
m_ResType == ((CResourceType*)Cookie.GetResultItem())->GetResType();
}
private:
RESOURCEID m_ResType;
};
class CComputerIdentifier : public CItemIdentifier
{
public:
CComputerIdentifier(CComputer& Computer)
{
ASSERT(Computer.GetDisplayName());
m_strName = Computer.GetDisplayName();
}
virtual BOOL operator==(CCookie& Cookie)
{
return COOKIE_TYPE_RESULTITEM_COMPUTER == Cookie.GetType() &&
!lstrcmpi((LPCTSTR)m_strName, Cookie.GetResultItem()->GetDisplayName());
}
private:
String m_strName;
};
#endif // __CNODE_H__