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.
 
 
 
 
 
 

410 lines
19 KiB

//=============================================================================
// Contains the refresh function for a simple query function (a simple query
// is one which involves a single WMI class and little post processing).
//=============================================================================
#include "stdafx.h"
#include "category.h"
#include "dataset.h"
#include "wmiabstraction.h"
#include "resourcemap.h"
//-----------------------------------------------------------------------------
// These functions implement features found in the new versions of MFC (new
// than what we're currently building with).
//-----------------------------------------------------------------------------
void StringReplace(CString & str, LPCTSTR szLookFor, LPCTSTR szReplaceWith)
{
CString strWorking(str);
CString strReturn;
CString strLookFor(szLookFor);
CString strReplaceWith(szReplaceWith);
int iLookFor = strLookFor.GetLength();
int iNext;
while (!strWorking.IsEmpty())
{
iNext = strWorking.Find(strLookFor);
if (iNext == -1)
{
strReturn += strWorking;
strWorking.Empty();
}
else
{
strReturn += strWorking.Left(iNext);
strReturn += strReplaceWith;
strWorking = strWorking.Right(strWorking.GetLength() - (iNext + iLookFor));
}
}
str = strReturn;
}
//-----------------------------------------------------------------------------
// The CSimpleQuery class encapsulates the simple query data. An array of these
// is created containing an element for each category to use this refresh
// function.
//-----------------------------------------------------------------------------
class CSimpleQuery
{
public:
CSimpleQuery(DWORD dwIndex, LPCTSTR szClass, LPCTSTR szProperties, UINT uiColumns, BOOL fShowResources = FALSE, BOOL fBlankDividers = FALSE) :
m_dwIndex(dwIndex),
m_strClass(szClass),
m_strProperties(szProperties),
m_uiColumns(uiColumns),
m_fBlankDividers(fBlankDividers),
m_fShowResources(fShowResources) {};
~CSimpleQuery() {};
public:
DWORD m_dwIndex; // the index of this particular query
CString m_strClass; // name of WMI class to enumerate
CString m_strProperties; // properties to enumerate (in order displayed, comma delimited)
BOOL m_fShowResources; // show resources used (must be two column "item|value" format)
UINT m_uiColumns; // resource ID for the string containing the data to go in columns
BOOL m_fBlankDividers; // insert a blank line between each instance
};
// TBD - need to mark some items as advanced
CSimpleQuery aSimpleQueries[] =
{
CSimpleQuery(QUERY_CDROM, _T("Win32_CDRomDrive"), _T("Drive, Description, MediaLoaded, MediaType, Name, Manufacturer, Status, TransferRate, MSIAdvancedSCSITargetId, MSIAdvancedPNPDeviceID"), IDS_CDROMCOLUMNS, TRUE, TRUE),
CSimpleQuery(QUERY_SERVICES, _T("Win32_Service"), _T("DisplayName, Name, State, StartMode, ServiceType, PathName, ErrorControl, StartName, TagId"), IDS_SERVICES1, FALSE),
CSimpleQuery(QUERY_PROGRAMGROUP, _T("Win32_ProgramGroup"), _T("GroupName, Name, UserName"), IDS_PROGRAMGROUP1, FALSE),
CSimpleQuery(QUERY_STARTUP, _T("Win32_StartupCommand"), _T("Caption, Command, User, Location"), IDS_STARTUP1, FALSE),
CSimpleQuery(QUERY_KEYBOARD, _T("Win32_Keyboard"), _T("Description, Name, Layout, MSIAdvancedPNPDeviceID, NumberOfFunctionKeys"), IDS_KEYBOARD1, TRUE, TRUE),
CSimpleQuery(QUERY_POINTDEV, _T("Win32_PointingDevice"), _T("HardwareType, NumberOfButtons, Status, MSIAdvancedPNPDeviceID, MSIAdvancedPowerManagementSupported, MSIAdvancedDoubleSpeedThreshold, MSIAdvancedHandedness"), IDS_POINTDEV1, TRUE, TRUE),
CSimpleQuery(QUERY_MODEM, _T("Win32_POTSModem"), _T("Caption, Description, DeviceID, DeviceType, AttachedTo, AnswerMode, MSIAdvancedPNPDeviceID, MSIAdvancedProviderName, MSIAdvancedModemInfPath, MSIAdvancedModemInfSection, MSIAdvancedBlindOff, MSIAdvancedBlindOn, CompressionOff, CompressionOn, ErrorControlForced, ErrorControlOff, ErrorControlOn, MSIAdvancedFlowControlHard, MSIAdvancedFlowControlOff, MSIAdvancedFlowControlSoft, MSIAdvancedDCB, MSIAdvancedDefault, MSIAdvancedInactivityTimeout, MSIAdvancedModulationBell, MSIAdvancedModulationCCITT, MSIAdvancedPrefix, MSIAdvancedPulse, MSIAdvancedReset, MSIAdvancedResponsesKeyName, SpeakerModeDial, SpeakerModeOff, SpeakerModeOn, SpeakerModeSetup, SpeakerVolumeHigh, SpeakerVolumeLow, SpeakerVolumeMed, MSIAdvancedStringFormat, MSIAdvancedTerminator, MSIAdvancedTone"), IDS_MODEM1, TRUE, TRUE),
CSimpleQuery(QUERY_NETPROT, _T("Win32_NetworkProtocol"), _T("Name, ConnectionlessService, GuaranteesDelivery, GuaranteesSequencing, MSIAdvancedMaximumAddressSize, MSIAdvancedMaximumMessageSize, MSIAdvancedMessageOriented, MSIAdvancedMinimumAddressSize, MSIAdvancedPseudoStreamOriented, MSIAdvancedSupportsBroadcasting, MSIAdvancedSupportsConnectData, MSIAdvancedSupportsDisconnectData, MSIAdvancedSupportsEncryption, MSIAdvancedSupportsExpeditedData, MSIAdvancedSupportsGracefulClosing, MSIAdvancedSupportsGuaranteedBandwidth, MSIAdvancedSupportsMulticasting"), IDS_NETPROT1, FALSE, TRUE),
CSimpleQuery(QUERY_ENVVAR, _T("Win32_Environment"), _T("Name, VariableValue, UserName"), IDS_ENVVAR1, FALSE),
CSimpleQuery(QUERY_SOUNDDEV, _T("Win32_SoundDevice"), _T("Caption, Manufacturer, Status, MSIAdvancedPNPDeviceID"), IDS_SOUNDDEV1, TRUE, TRUE),
CSimpleQuery(QUERY_DISPLAY, _T("Win32_VideoController"), _T("Name, MSIAdvancedPNPDeviceID, VideoProcessor, AdapterCompatibility, MSIAdvancedDescription, MSIAdvancedAdapterRAM, MSIAdvancedInstalledDisplayDrivers, DriverVersion, MSIAdvancedInfFilename, MSIAdvancedInfSection, MSIAdvancedNumberOfColorPlanes, MSIAdvancedCurrentNumberOfColors, CurrentHorizontalResolution, CurrentVerticalResolution, CurrentRefreshRate, CurrentBitsPerPixel"), IDS_DISPLAY1, TRUE, TRUE),
CSimpleQuery(QUERY_INFRARED, _T("Win32_InfraredDevice"), _T("Caption"), IDS_INFRARED1, TRUE, TRUE),
CSimpleQuery(QUERY_PARALLEL, _T("Win32_ParallelPort"), _T("Name, MSIAdvancedPNPDeviceID"), IDS_PARALLEL1, TRUE, TRUE),
CSimpleQuery(QUERY_PRINTER, _T("Win32_Printer"), _T("Name, DriverName, PortName, ServerName"), IDS_PRINTER1, FALSE),
CSimpleQuery(QUERY_NETCONNECTION, _T("Win32_NetworkConnection"), _T("LocalName, RemoteName, ResourceType, ConnectionType, UserName"), IDS_NETCONNECTION1, FALSE),
CSimpleQuery(QUERY_DRIVER, _T("Win32_SystemDriver"), _T("Name, Description, PathName, ServiceType, Started, StartMode, State, Status, ErrorControl, AcceptPause, AcceptStop"), IDS_DRIVER1, FALSE),
CSimpleQuery(QUERY_SIGNEDDRIVER, _T("Win32_PnPSignedDriver"), _T("DeviceName, IsSigned, DeviceClass, DriverVersion, DriverDate, Manufacturer, InfName, DriverName, DeviceID"), IDS_SIGNEDDRIVER1, FALSE),
CSimpleQuery(QUERY_IDE, _T("Win32_IDEController"), _T("Caption, Manufacturer, Status, PNPDeviceID"), IDS_IDE1, TRUE, TRUE),
CSimpleQuery(QUERY_SCSI, _T("Win32_SCSIController"), _T("Caption, Manufacturer, Status, MSIAdvancedPNPDeviceID"), IDS_SCSI1, TRUE, TRUE),
CSimpleQuery(QUERY_PRINTJOBS, _T("Win32_PrintJob"), _T("Document, Size, Owner, Notify, Status, TimeSubmitted, StartTime, UntilTime, ElapsedTime, PagesPrinted, JobId, Priority, Parameters, DriverName, PrintProcessor, HostPrintQueue, DataType, Name"), IDS_PRINTJOBS1, FALSE),
CSimpleQuery(0, _T(""), _T(""), 0)
};
//-----------------------------------------------------------------------------
// The refresh function which processes the simple query. The correct query
// is found in the query array. For each instance of the class, the column
// data will be inserted into the array of column values, and the format
// flags will be replaced with the actual data.
//-----------------------------------------------------------------------------
BOOL ProcessColumnString(CMSIValue * pValue, CWMIObject * pObject, CString * pstrProperties);
BOOL GetResourcesFromPNPID(CWMIHelper * pWMI, CResourceMap * pResourceMap, LPCTSTR szPNPID, CPtrList * aColValues);
HRESULT NetAdapter(CWMIHelper * pWMI, volatile BOOL * pfCancel, CPtrList * aColValues, int iColCount, CResourceMap * pResourceMap);
HRESULT SerialPort(CWMIHelper * pWMI, volatile BOOL * pfCancel, CPtrList * aColValues, int iColCount, CResourceMap * pResourceMap);
HRESULT SimpleQuery(CWMIHelper * pWMI, DWORD dwIndex, volatile BOOL * pfCancel, CPtrList * aColValues, int iColCount, void ** ppCache)
{
HRESULT hr = S_OK;
// Find the correct query.
CSimpleQuery * pQuery = aSimpleQueries;
while (pQuery->m_dwIndex && pQuery->m_dwIndex != dwIndex)
pQuery++;
if (pQuery->m_dwIndex != dwIndex && dwIndex != QUERY_NETADAPTER && dwIndex != QUERY_SERIALPORT)
{
ASSERT(0 && "bad dwIndex to SimpleQuery()");
return E_FAIL;
}
// Make sure we have a resource map (or can create one).
if (ppCache)
{
if (pWMI && *ppCache == NULL)
{
*ppCache = (void *) new CResourceMap;
if (*ppCache)
{
hr = ((CResourceMap *) *ppCache)->Initialize(pWMI);
if (FAILED(hr))
{
delete ((CResourceMap *) *ppCache);
*ppCache = (void *) NULL;
}
}
}
else if (pWMI == NULL && *ppCache)
{
delete ((CResourceMap *) *ppCache);
return S_OK;
}
}
CResourceMap * pResourceMap = (CResourceMap *) *ppCache;
// This is a nice way to cache to resource map for multiple functions, but it's
// a monumental pain when we remote to a different machine:
//
// CResourceMap * pResourceMap = gResourceMap.GetResourceMap(pWMI);
// if (pResourceMap == NULL)
// return hr;
// Check to see if this is one of the special case queries.
if (dwIndex == QUERY_NETADAPTER)
return NetAdapter(pWMI, pfCancel, aColValues, iColCount, pResourceMap);
if (dwIndex == QUERY_SERIALPORT)
return SerialPort(pWMI, pfCancel, aColValues, iColCount, pResourceMap);
// Enumerate the requested class.
CWMIObjectCollection * pCollection = NULL;
hr = pWMI->Enumerate(pQuery->m_strClass, &pCollection, pQuery->m_strProperties);
if (SUCCEEDED(hr))
{
CWMIObject * pObject = NULL;
while (S_OK == pCollection->GetNext(&pObject))
{
CString strProperties = pQuery->m_strProperties;
// If this is the second or later iteration, then we should add a blank
// set of entries to the columns.
if (pQuery->m_fBlankDividers)
pWMI->AppendBlankLine(aColValues, iColCount);
pWMI->AddObjectToOutput(aColValues, iColCount, pObject, strProperties, pQuery->m_uiColumns);
// If so marked, we should see what resources (drivers, IRQs, etc.) are
// associated with this object and add them to the list.
if (pQuery->m_fShowResources)
{
ASSERT(iColCount == 2);
CString strPNPID = pObject->GetString(_T("PNPDeviceID"));
if (!strPNPID.IsEmpty())
GetResourcesFromPNPID(pWMI, pResourceMap, strPNPID, aColValues);
}
}
delete pObject;
delete pCollection;
}
return hr;
}
//-----------------------------------------------------------------------------
// Given a resource map and a PNP ID string, this function will add all the
// resources used by that PNP device (IRQs, drivers, etc.) to a two column
// array of string lists.
//-----------------------------------------------------------------------------
BOOL GetResourcesFromPNPID(CWMIHelper * pWMI, CResourceMap * pResourceMap, LPCTSTR szPNPID, CPtrList * aColValues)
{
CString strPath, strResourcePath;
CString strPNPDeviceID(szPNPID);
StringReplace(strPNPDeviceID, _T("\\"), _T("\\\\"));
strPath.Format(_T("Win32_PnPEntity.DeviceID=\"%s\""), strPNPDeviceID);
CStringList * pResourceList = pResourceMap->Lookup(strPath);
if (pResourceList)
{
::AfxSetResourceHandle(_Module.GetResourceInstance());
for (POSITION pos = pResourceList->GetHeadPosition(); pos != NULL;)
{
strResourcePath = pResourceList->GetNext(pos);
CWMIObject * pResourceObject;
if (SUCCEEDED(pWMI->GetObject(strResourcePath, &pResourceObject)))
{
CString strClass;
if (SUCCEEDED(pResourceObject->GetValueString(_T("__CLASS"), &strClass)))
{
CString strItem;
if (strClass == _T("Win32_IRQResource"))
strItem.LoadString(IDS_IRQCHANNEL);
else if (strClass == _T("Win32_PortResource"))
strItem.LoadString(IDS_IOPORT);
else if (strClass == _T("Win32_DMAChannel"))
strItem.LoadString(IDS_DMACHANNEL);
else if (strClass == _T("Win32_DeviceMemoryAddress"))
strItem.LoadString(IDS_MEMORYADDRESS);
else if (strClass == _T("CIM_DataFile"))
strItem.LoadString(IDS_DRIVER);
if (!strItem.IsEmpty())
{
CString strValue;
if (SUCCEEDED(pResourceObject->GetValueString(_T("Caption"), &strValue)))
{
if (strClass == _T("CIM_DataFile"))
{
CString strVersion, strDate, strSize;
pResourceObject->GetInterpretedValue(_T("FileSize"), _T("%z"), _T('z'), &strSize, NULL);
pResourceObject->GetValueString(_T("Version"), &strVersion);
pResourceObject->GetInterpretedValue(_T("CreationDate"), _T("%t"), _T('t'), &strDate, NULL);
if (!strVersion.IsEmpty() || !strDate.IsEmpty() || !strSize.IsEmpty())
strValue += CString(_T(" (")) + strVersion + CString(_T(", ")) + strSize + CString(_T(", ")) + strDate + CString(_T(")"));
}
pWMI->AppendCell(aColValues[0], strItem, 0);
pWMI->AppendCell(aColValues[1], strValue, 0);
}
}
}
delete pResourceObject;
} // if we could get the object
} // for enumerating through the list
} // if there is a list
return TRUE;
}
//-----------------------------------------------------------------------------
// The specific query for the network adapter class.
//-----------------------------------------------------------------------------
HRESULT NetAdapter(CWMIHelper * pWMI, volatile BOOL * pfCancel, CPtrList * aColValues, int iColCount, CResourceMap * pResourceMap)
{
HRESULT hr = S_OK;
LPCTSTR szNetworkAdapterProperties = _T("Caption, AdapterType, MSIAdvancedProductName, MSIAdvancedInstalled, MSIAdvancedPNPDeviceID, MSIAdvancedTimeOfLastReset, MSIAdvancedIndex");
LPCTSTR szNetworkAdapterConfigProperties = _T("ServiceName, IPAddress, IPSubnet, DefaultIPGateway, DHCPEnabled, MSIAdvancedDHCPServer, MSIAdvancedDHCPLeaseExpires, MSIAdvancedDHCPLeaseObtained, MACAddress, Index");
CPtrList listNetAdapterConfig;
CWMIObjectCollection * pCollection = NULL;
hr = pWMI->Enumerate(_T("Win32_NetworkAdapter"), &pCollection, szNetworkAdapterProperties);
if (SUCCEEDED(hr))
{
CWMIObject * pObject = NULL;
while (S_OK == pCollection->GetNext(&pObject))
{
pWMI->AppendBlankLine(aColValues, iColCount);
pWMI->AddObjectToOutput(aColValues, iColCount, pObject, szNetworkAdapterProperties, IDS_NETWORKADAPTER1);
DWORD dwIndex;
if (SUCCEEDED(pObject->GetValueDWORD(_T("Index"), &dwIndex)))
{
if (listNetAdapterConfig.IsEmpty())
{
// Enumerate the class and cache the objects.
CWMIObjectCollection * pConfigCollection = NULL;
if (SUCCEEDED(pWMI->Enumerate(_T("Win32_NetworkAdapterConfiguration"), &pConfigCollection, szNetworkAdapterConfigProperties)))
{
CWMIObject * pConfigObject = NULL;
while (S_OK == pConfigCollection->GetNext(&pConfigObject))
{
DWORD dwConfigIndex;
if (SUCCEEDED(pConfigObject->GetValueDWORD(_T("Index"), &dwConfigIndex)))
if (dwConfigIndex == dwIndex)
pWMI->AddObjectToOutput(aColValues, iColCount, pConfigObject, szNetworkAdapterConfigProperties, IDS_NETWORKADAPTER2);
listNetAdapterConfig.AddTail((void *)pConfigObject);
pConfigObject = NULL;
}
delete pConfigObject;
delete pConfigCollection;
}
}
else
{
// Look through the list of cached objects.
for (POSITION pos = listNetAdapterConfig.GetHeadPosition(); pos != NULL;)
{
CWMIObject * pConfigObject = (CWMIObject *)listNetAdapterConfig.GetNext(pos);
DWORD dwConfigIndex;
if (pConfigObject && SUCCEEDED(pConfigObject->GetValueDWORD(_T("Index"), &dwConfigIndex)))
if (dwConfigIndex == dwIndex)
pWMI->AddObjectToOutput(aColValues, iColCount, pConfigObject, szNetworkAdapterConfigProperties, IDS_NETWORKADAPTER2);
}
}
}
CString strPNPID = pObject->GetString(_T("PNPDeviceID"));
if (!strPNPID.IsEmpty())
GetResourcesFromPNPID(pWMI, pResourceMap, strPNPID, aColValues);
}
delete pObject;
delete pCollection;
while (!listNetAdapterConfig.IsEmpty())
{
CWMIObject * pDeleteObject = (CWMIObject *)listNetAdapterConfig.RemoveHead();
if (pDeleteObject)
delete pDeleteObject;
}
}
return hr;
}
//-----------------------------------------------------------------------------
// The specific query for the serial port class.
//
// TBD - cache the config values to save re-enumerating.
//-----------------------------------------------------------------------------
HRESULT SerialPort(CWMIHelper * pWMI, volatile BOOL * pfCancel, CPtrList * aColValues, int iColCount, CResourceMap * pResourceMap)
{
HRESULT hr = S_OK;
LPCTSTR szSerialPortProperties = _T("Name, Status, MSIAdvancedPNPDeviceID, MSIAdvancedMaximumInputBufferSize, MSIAdvancedMaximumOutputBufferSize, MSIAdvancedSettableBaudRate, MSIAdvancedSettableDataBits, MSIAdvancedSettableFlowControl, MSIAdvancedSettableParity, MSIAdvancedSettableParityCheck, MSIAdvancedSettableStopBits, MSIAdvancedSettableRLSD, MSIAdvancedSupportsRLSD, MSIAdvancedSupports16BitMode, MSIAdvancedSupportsSpecialCharacters, MSIAdvancedDeviceID");
LPCTSTR szSerialPortConfigProperties = _T("BaudRate, BitsPerByte, StopBits, Parity, IsBusy, MSIAdvancedAbortReadWriteOnError, MSIAdvancedBinaryModeEnabled, MSIAdvancedContinueXMitOnXOff, MSIAdvancedCTSOutflowControl, MSIAdvancedDiscardNULLBytes, MSIAdvancedDSROutflowControl, MSIAdvancedDSRSensitivity, MSIAdvancedDTRFlowControlType, MSIAdvancedEOFCharacter, MSIAdvancedErrorReplaceCharacter, MSIAdvancedErrorReplacementEnabled, MSIAdvancedEventCharacter, MSIAdvancedParityCheckEnabled, MSIAdvancedRTSFlowControlType, MSIAdvancedXOffCharacter, MSIAdvancedXOffXMitThreshold, MSIAdvancedXOnCharacter, MSIAdvancedXOnXMitThreshold, MSIAdvancedXOnXOffInFlowControl, MSIAdvancedXOnXOffOutFlowControl, Name");
CWMIObjectCollection * pCollection = NULL;
hr = pWMI->Enumerate(_T("Win32_SerialPort"), &pCollection, szSerialPortProperties);
if (SUCCEEDED(hr))
{
CWMIObject * pObject = NULL;
while (S_OK == pCollection->GetNext(&pObject))
{
pWMI->AppendBlankLine(aColValues, iColCount);
pWMI->AddObjectToOutput(aColValues, iColCount, pObject, szSerialPortProperties, IDS_SERIALPORT1);
CString strDeviceID;
if (SUCCEEDED(pObject->GetValueString(_T("DeviceID"), &strDeviceID)))
{
CWMIObjectCollection * pConfigCollection = NULL;
if (SUCCEEDED(pWMI->Enumerate(_T("Win32_SerialPortConfiguration"), &pConfigCollection, szSerialPortConfigProperties)))
{
CWMIObject * pConfigObject = NULL;
while (S_OK == pConfigCollection->GetNext(&pConfigObject))
{
CString strConfigDeviceID;
if (SUCCEEDED(pConfigObject->GetValueString(_T("Name"), &strConfigDeviceID)))
if (strDeviceID.CompareNoCase(strConfigDeviceID) == 0)
pWMI->AddObjectToOutput(aColValues, iColCount, pConfigObject, szSerialPortConfigProperties, IDS_SERIALPORT2);
}
delete pConfigObject;
delete pConfigCollection;
}
}
CString strPNPID = pObject->GetString(_T("PNPDeviceID"));
if (!strPNPID.IsEmpty())
GetResourcesFromPNPID(pWMI, pResourceMap, strPNPID, aColValues);
}
delete pObject;
delete pCollection;
}
return hr;
}