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.
 
 
 
 
 
 

267 lines
12 KiB

/********************************************************************
Copyright (c) 1999 Microsoft Corporation
Module Name:
PCH_ResourceMemRange.CPP
Abstract:
WBEM provider class implementation for PCH_ResourceMemRange class
Revision History:
Ghim-Sim Chua (gschua) 04/27/99
- Created
********************************************************************/
#include "pchealth.h"
#include "PCH_ResourceMemRange.h"
// #include "confgmgr.h"
/////////////////////////////////////////////////////////////////////////////
// tracing stuff
#ifdef THIS_FILE
#undef THIS_FILE
#endif
static char __szTraceSourceFile[] = __FILE__;
#define THIS_FILE __szTraceSourceFile
#define TRACE_ID DCID_RESOURCEMEMRANGE
CPCH_ResourceMemRange MyPCH_ResourceMemRangeSet (PROVIDER_NAME_PCH_RESOURCEMEMRANGE, PCH_NAMESPACE) ;
// Property names
//===============
const static WCHAR* pBase = L"Base" ;
const static WCHAR* pCategory = L"Category" ;
const static WCHAR* pTimeStamp = L"TimeStamp" ;
const static WCHAR* pChange = L"Change" ;
const static WCHAR* pEnd = L"End" ;
const static WCHAR* pMax = L"Max" ;
const static WCHAR* pMin = L"Min" ;
const static WCHAR* pName = L"Name" ;
/*****************************************************************************
*
* FUNCTION : CPCH_ResourceMemRange::EnumerateInstances
*
* DESCRIPTION : Returns all the instances of this class.
*
* INPUTS : A pointer to the MethodContext for communication with WinMgmt.
* A long that contains the flags described in
* IWbemServices::CreateInstanceEnumAsync. Note that the following
* flags are handled by (and filtered out by) WinMgmt:
* WBEM_FLAG_DEEP
* WBEM_FLAG_SHALLOW
* WBEM_FLAG_RETURN_IMMEDIATELY
* WBEM_FLAG_FORWARD_ONLY
* WBEM_FLAG_BIDIRECTIONAL
*
* RETURNS : WBEM_S_NO_ERROR if successful
*
* COMMENTS : TO DO: All instances on the machine should be returned here.
* If there are no instances, return WBEM_S_NO_ERROR.
* It is not an error to have no instances.
*
*****************************************************************************/
HRESULT CPCH_ResourceMemRange::EnumerateInstances(
MethodContext* pMethodContext,
long lFlags
)
{
TraceFunctEnter("CPCH_ResourceIRQ::EnumerateInstances");
HRESULT hRes = WBEM_S_NO_ERROR;
REFPTRCOLLECTION_POSITION posList;
// Instances
CComPtr<IEnumWbemClassObject> pDeviceMemAddressEnumInst;
CComPtr<IEnumWbemClassObject> pWin32AllocatedResourceEnumInst;
// Objects
IWbemClassObjectPtr pWin32AllocatedResourceObj;
IWbemClassObjectPtr pDeviceMemAddressObj;
// Variants
CComVariant varAntecedent;
CComVariant varDependent;
CComVariant varStartingAddress;
CComVariant varPNPEntity;
// Query Strings
CComBSTR bstrWin32AllocatedResourceQuery = L"Select Antecedent, Dependent FROM win32_Allocatedresource";
CComBSTR bstrDeviceMemAddressQuery = L"Select StartingAddress, EndingAddress FROM Win32_DeviceMemoryAddress WHERE StartingAddress = ";
CComBSTR bstrDeviceMemAddressQueryString;
// Return Values;
ULONG ulWin32AllocatedResourceRetVal = 0;
ULONG ulDeviceMemAddressRetVal = 0;
// Integers
int i;
int nStAddren;
int nIter;
// Pattern Strings
LPCSTR strDeviceMemAddressPattern = "Win32_DeviceMemoryAddress.StartingAddress=";
LPCSTR strPNPEntityPattern = "Win32_PnPEntity.DeviceID=";
// Chars
LPSTR strSource;
LPSTR pDest;
BOOL fValidInt;
CComBSTR bstrPropertyAntecedent=L"Antecedent";
CComBSTR bstrPropertyDependent=L"Dependent";
// Enumerate the instances of Win32_PNPAllocatedResource
hRes = ExecWQLQuery(&pWin32AllocatedResourceEnumInst, bstrWin32AllocatedResourceQuery);
if (FAILED(hRes))
{
// Cannot get any properties.
goto END;
}
// Query Succeeded
while(WBEM_S_NO_ERROR == pWin32AllocatedResourceEnumInst->Next(WBEM_INFINITE, 1, &pWin32AllocatedResourceObj, &ulWin32AllocatedResourceRetVal))
{
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
// Starting Address //
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
// Get the Antecedent Value
hRes = pWin32AllocatedResourceObj->Get(bstrPropertyAntecedent, 0, &varAntecedent, NULL, NULL);
if (FAILED(hRes))
{
// Could not get the antecedent
ErrorTrace(TRACE_ID, "GetVariant on Win32_AllocatedResource:Antecedent Field failed.");
}
else
{
// Got the Antecedent. Search its value to see if it is a DeviceMemAddress.
// varAntecedent set to antecedent. Copy this to bstrResult
varAntecedent.ChangeType(VT_BSTR, NULL);
{
USES_CONVERSION;
strSource = OLE2A(varAntecedent.bstrVal);
}
// Check if it is DeviceMemoryAddress by comparing with the known pattern of DeviceMemoryAddress.
pDest = strstr(strSource,strDeviceMemAddressPattern);
if(pDest != NULL)
{
// This is DeviceMemoryAddress Resource instance
// Can get the Device Memory Starting Address
// Advance the pointer to the end of the pattern so the pointer is
// positioned at Starting Address
pDest += lstrlen(strDeviceMemAddressPattern);
// Formulate the Query String
bstrDeviceMemAddressQueryString = bstrDeviceMemAddressQuery;
bstrDeviceMemAddressQueryString.Append(pDest);
// At this point the WQL Query can be used to get the win32_DeviceMemoryAddress Instance.
// Added the following line because you need to clear the CComPtr before you query the second time.
pDeviceMemAddressEnumInst = NULL;
hRes = ExecWQLQuery(&pDeviceMemAddressEnumInst, bstrDeviceMemAddressQueryString);
if (SUCCEEDED(hRes))
{
// Query Succeeded. Get the Instance Object
if(WBEM_S_NO_ERROR == pDeviceMemAddressEnumInst->Next(WBEM_INFINITE, 1, &pDeviceMemAddressObj, &ulDeviceMemAddressRetVal))
{
// Create a new instance of PCH_ResourceMemRange Class based on the passed-in MethodContext
CInstancePtr pPCHResourceMemRangeInstance(CreateNewInstance(pMethodContext), false);
// Created a New Instance of PCH_ResourceMemRange Successfully.
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
// StartingAddress //
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
CopyProperty(pDeviceMemAddressObj, L"StartingAddress", pPCHResourceMemRangeInstance, pBase);
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
// EndingAddress //
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
CopyProperty(pDeviceMemAddressObj, L"EndingAddress", pPCHResourceMemRangeInstance, pEnd);
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
// Name //
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
// CopyProperty(pWin32AllocatedResourceObj, L"Dependent", pPCHResourceMemRangeInstance, pName);
// Get the Dependent Value
hRes = pWin32AllocatedResourceObj->Get(bstrPropertyDependent, 0, &varDependent, NULL, NULL);
if (FAILED(hRes))
{
// Could not get the Dependent
ErrorTrace(TRACE_ID, "GetVariant on Win32_AllocatedResource:Dependent Field failed.");
}
else
{
// Got the Dependent. Search its value to point to PNPEntity.
// varDependent set to Dependent. Copy this to bstrResult
varDependent.ChangeType(VT_BSTR, NULL);
{
USES_CONVERSION;
strSource = OLE2A(varDependent.bstrVal);
}
// Search for Win32_PNPEntity Pattern.
pDest = strstr(strSource,strPNPEntityPattern);
if(pDest)
{
// Advance the pointer to point to the PNPEntity Name
pDest += lstrlen(strPNPEntityPattern);
// Copy the PNPEntity Name.....
varPNPEntity = pDest;
// Set the Name
hRes = pPCHResourceMemRangeInstance->SetVariant(pName, varPNPEntity);
if (FAILED(hRes))
{
ErrorTrace(TRACE_ID, "SetVariant on win32_AllocatedResource.IRQ Number Failed!");
// Proceed Anyway
}
}
}
// All the properties in pPCHResourceMemRange are set
hRes = pPCHResourceMemRangeInstance->Commit();
if (FAILED(hRes))
{
// Cannot commit the Instance
ErrorTrace(TRACE_ID, "Commit on Instance failed.");
} // end of if FAILED(hRes)
}
} // end of Query Succeeded.
} // end of if pDest != NULL
} // end of else got the antecedent
} //end of Allocated Resource Instances.
END:
TraceFunctLeave();
return hRes ;
// pInstance->SetVariant(pBase, <Property Value>);
// pInstance->SetVariant(pCategory, <Property Value>);
// pInstance->SetVariant(pTimeStamp, <Property Value>);
// pInstance->SetVariant(pChange, <Property Value>);
// pInstance->SetVariant(pEnd, <Property Value>);
// pInstance->SetVariant(pMax, <Property Value>);
// pInstance->SetVariant(pMin, <Property Value>);
// pInstance->SetVariant(pName, <Property Value>);
}