/********************************************************************
Copyright (c) 1999 Microsoft Corporation
Module Name:
PCH_BIOS.CPP
Abstract:
WBEM provider class implementation for PCH_BIOS class
Revision History:
Ghim-Sim Chua (gschua) 05/05/99
- Created
Kalyani Narlanka (kalyanin) 05/12/99
- Added Code to get all the properties of this class
Kalyani Narlanka (kalyanin) 05/18/99
********************************************************************/
#include "pchealth.h"
#include "PCH_BIOS.h"
/////////////////////////////////////////////////////////////////////////////
// tracing stuff
#ifdef THIS_FILE
#undef THIS_FILE
#endif
static char __szTraceSourceFile[] = __FILE__;
#define THIS_FILE __szTraceSourceFile
#define TRACE_ID DCID_BIOS
CPCH_BIOS MyPCH_BIOSSet (PROVIDER_NAME_PCH_BIOS, PCH_NAMESPACE) ;
// Property names
//===============
const static WCHAR* pBIOSDate = L"BIOSDate" ;
const static WCHAR* pBIOSName = L"BIOSName" ;
const static WCHAR* pBIOSVersion = L"BIOSVersion" ;
const static WCHAR* pCPU = L"CPU" ;
const static WCHAR* pINFName = L"INFName" ;
const static WCHAR* pMachineType = L"MachineType" ;
const static WCHAR* pDriver = L"Driver" ;
const static WCHAR* pDriverDate = L"DriverDate" ;
const static WCHAR* pChange = L"Change";
const static WCHAR* pTimeStamp = L"TimeStamp";
/*****************************************************************************
*
* FUNCTION : CPCH_BIOS::EnumerateInstances
*
* DESCRIPTION : Returns the instance 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
*
* SYSNOPSIS : There is only instance of this class at any time. This function gives this
* instance.
* If there are no instances, returns WBEM_S_NO_ERROR.
* It is not an error to have no instances.
*
*****************************************************************************/
HRESULT CPCH_BIOS::EnumerateInstances ( MethodContext* pMethodContext, long lFlags )
{
// Begin Declarations
//
TraceFunctEnter("CPCH_BIOS::EnumerateInstances");
HRESULT hRes = WBEM_S_NO_ERROR;
HRESULT hRes1;
HRESULT hRes2;
// Query String
CComBSTR bstrBIOSQuery = L"Select Name, ReleaseDate, Version FROM win32_BIOS";
CComBSTR bstrProcessorQuery = L"Select DeviceId, Name FROM win32_processor";
CComBSTR bstrComputerSystemQuery = L"Select Name, Description FROM win32_computerSystem";
CComBSTR bstrDriver;
// Registry Hive where BIOS Info is stored
LPCTSTR lpctstrSystemHive = _T("System\\CurrentControlSet\\Services\\Class\\System");
// Registry Names of interest
LPCTSTR lpctstrDriverDesc = _T("DriverDesc");
LPCTSTR lpctstrINFName = _T("INFPath");
LPCTSTR lpctstrDriverDate = _T("DriverDate");
LPCTSTR lpctstrSystem = _T("System\\");
// Property Names
LPCWSTR lpctstrReleaseDate = L"ReleaseDate";
LPCWSTR lpctstrName = L"Name";
LPCWSTR lpctstrVersion = L"Version";
LPCWSTR lpctstrDescription = L"Description";
LPCTSTR lpctstrSystemBoard = _T("System Board");
// Strings
TCHAR tchSubSystemKeyName[MAX_PATH];
TCHAR tchDriverDescValue[MAX_PATH];
TCHAR tchDriverDateValue[MAX_PATH];
TCHAR tchINFNameValue[MAX_PATH];
// Instances
CComPtr<IEnumWbemClassObject> pBIOSEnumInst;
CComPtr<IEnumWbemClassObject> pProcessorEnumInst;
CComPtr<IEnumWbemClassObject> pComputerSystemEnumInst;
// Instances
// CInstancePtr pPCHBIOSInstance;
// Objects
IWbemClassObjectPtr pBIOSObj; // BUGBUG : WMI asserts if we use CComPtr
IWbemClassObjectPtr pProcessorObj; // BUGBUG : WMI asserts if we use CComPtr
IWbemClassObjectPtr pComputerSystemObj; // BUGBUG : WMI asserts if we use CComPtr
// Variants
CComVariant varDriver;
CComVariant varDriverDate;
CComVariant varINFName;
CComVariant varSnapshot = "SnapShot";
// Unsigned Longs....
ULONG ulBIOSRetVal = 0;
ULONG ulProcessorRetVal = 0;
ULONG ulComputerSystemRetVal = 0;
LONG lRegRetVal;
// SystemTime
SYSTEMTIME stUTCTime;
// Registry Keys
HKEY hkeySystem;
HKEY hkeySubSystem;
// DWORDs
DWORD dwIndex = 0;
DWORD dwSize = MAX_PATH;
DWORD dwType;
// Boolean
BOOL fContinueEnum = FALSE;
BOOL fCommit = FALSE;
// FileTime
PFILETIME pFileTime = NULL;
// End Declarations
// Create a new instance of PCH_BIOS Class based on the passed-in MethodContext
CInstancePtr pPCHBIOSInstance(CreateNewInstance(pMethodContext), false);
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
// TIME STAMP //
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
// Get the date and time to update the TimeStamp Field
GetSystemTime(&stUTCTime);
hRes = pPCHBIOSInstance->SetDateTime(pTimeStamp, WBEMTime(stUTCTime));
if (FAILED(hRes))
{
// Could not Set the Time Stamp
// Continue anyway
ErrorTrace(TRACE_ID, "SetDateTime on Timestamp Field failed.");
}
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
// CHANGE //
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
hRes = pPCHBIOSInstance->SetVariant(pChange, varSnapshot);
if (FAILED(hRes))
{
// Could not Set the Change Property
// Continue anyway
ErrorTrace(TRACE_ID, "Set Variant on Change Field failed.");
}
// Execute the query to get Name, ReleaseDate, Version FROM Win32_BIOS
// Class.
// pBIOSEnumInst contains a pointer to the instance returned.
hRes = ExecWQLQuery(&pBIOSEnumInst, bstrBIOSQuery );
if (SUCCEEDED(hRes))
{
// Query Succeeded!
// Get the instance Object.
if((pBIOSEnumInst->Next(WBEM_INFINITE, 1, &pBIOSObj, &ulBIOSRetVal)) == WBEM_S_NO_ERROR)
{
// Get Name, Date and Version
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
// BIOSDATE //
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
CopyProperty(pBIOSObj, lpctstrReleaseDate, pPCHBIOSInstance, pBIOSDate);
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
// BIOSNAME //
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
hRes = CopyProperty(pBIOSObj, lpctstrName, pPCHBIOSInstance, pBIOSName);
if(SUCCEEDED(hRes))
{
fCommit = TRUE;
}
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
// BIOSVERSION //
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
CopyProperty(pBIOSObj, lpctstrVersion, pPCHBIOSInstance, pBIOSVersion);
}
}
// Done with Win32_BIOS Class
// Now query Win32_Processor Class to get "CPU" property
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
// CPU //
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
hRes = ExecWQLQuery(&pProcessorEnumInst, bstrProcessorQuery);
if (SUCCEEDED(hRes))
{
// Query Succeeded!
// Get the instance Object.
if((pProcessorEnumInst->Next(WBEM_INFINITE, 1, &pProcessorObj, &ulProcessorRetVal)) == WBEM_S_NO_ERROR)
{
// Get Name
CopyProperty(pProcessorObj, lpctstrName, pPCHBIOSInstance, pCPU);
}
}
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
// MACHINETYPE //
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
hRes = ExecWQLQuery(&pComputerSystemEnumInst, bstrComputerSystemQuery);
if (SUCCEEDED(hRes))
{
// Query Succeeded!
// Get the instance Object.
if((pComputerSystemEnumInst->Next(WBEM_INFINITE, 1, &pComputerSystemObj, &ulComputerSystemRetVal)) == WBEM_S_NO_ERROR)
{
// Get "Description"
CopyProperty(pComputerSystemObj, lpctstrDescription, pPCHBIOSInstance, pMachineType);
}
}
// Get the remaining properties i.e. INFName, Driver and DriverDate from the Registry
// This is present in one of the keys under the HIVE "HKLM\System\CCS\Services\Class\System"
// Enumerate keys under this hive until the regname "DeviceDesc" equals "System Board"
// Once you hit "DeviceDesc" = "System Board" get the INFpath, Driver
// DriverDate from there.
lRegRetVal = RegOpenKeyEx(HKEY_LOCAL_MACHINE, lpctstrSystemHive, 0, KEY_READ, &hkeySystem);
if(lRegRetVal == ERROR_SUCCESS)
{
// Opened the Registry key.
// Enumerate the keys under this hive. One of the keys has
// DeviceDesc = "system Board".
lRegRetVal = RegEnumKeyEx(hkeySystem, dwIndex, tchSubSystemKeyName, &dwSize, NULL, NULL, NULL, pFileTime);
if(lRegRetVal == ERROR_SUCCESS)
{
fContinueEnum = TRUE;
}
while(fContinueEnum)
{
// Open the SubKey.
lRegRetVal = RegOpenKeyEx(hkeySystem, tchSubSystemKeyName, 0, KEY_READ, &hkeySubSystem);
if(lRegRetVal == ERROR_SUCCESS)
{
// Opened the SubKey
// Query for , regname "DriverDesc "
dwSize = MAX_PATH;
lRegRetVal = RegQueryValueEx(hkeySubSystem, lpctstrDriverDesc , NULL, &dwType, (LPBYTE)tchDriverDescValue, &dwSize);
if(lRegRetVal == ERROR_SUCCESS)
{
// Compare if the value is equal to "System Board"
if(_tcsicmp(tchDriverDescValue, lpctstrSystemBoard) == 0)
{
// The following statements could
try
{
// Found the Right DriverDesc
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
// DRIVER //
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
// Driver = system+lptstrSubSystemKeyName
bstrDriver = lpctstrSystem;
bstrDriver.Append(tchSubSystemKeyName);
varDriver = bstrDriver.Copy();
hRes2 = pPCHBIOSInstance->SetVariant(pDriver, varDriver);
if(FAILED(hRes2))
{
// Could not Set the DRIVER Property
// Continue anyway
ErrorTrace(TRACE_ID, "Set variant on Driver Failed.");
}
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
// DRIVERDATE //
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
// Query for DriverDate
dwSize = MAX_PATH;
lRegRetVal = RegQueryValueEx(hkeySubSystem, lpctstrDriverDate, NULL, &dwType, (LPBYTE)tchDriverDescValue, &dwSize);
if(lRegRetVal == ERROR_SUCCESS)
{
// Set the DriverDate
varDriverDate = tchDriverDescValue;
hRes2 = pPCHBIOSInstance->SetVariant(pDriverDate, varDriverDate);
if(FAILED(hRes2))
{
// Could not Set the DRIVERDATE Property
// Continue anyway
ErrorTrace(TRACE_ID, "Set variant on DriverDate Failed.");
}
}
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
// INFNAME //
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
// Query for INFName
dwSize = MAX_PATH;
lRegRetVal = RegQueryValueEx(hkeySubSystem, lpctstrINFName, NULL, &dwType, (LPBYTE)tchINFNameValue, &dwSize);
if(lRegRetVal == ERROR_SUCCESS)
{
// Set the INFName
varINFName = tchINFNameValue;
hRes2 = pPCHBIOSInstance->SetVariant(pINFName, varINFName);
if(FAILED(hRes2))
{
// Could not Set the INFNAME Property
// Continue anyway
ErrorTrace(TRACE_ID, "Set variant on INFNAME Property Failed.");
}
}
// Need not enumerate the rest of the keys
fContinueEnum = FALSE;
}
catch(...)
{
lRegRetVal = RegCloseKey(hkeySubSystem);
lRegRetVal = RegCloseKey(hkeySystem);
throw;
}
} // end of strcmp
} // end of Succeeded hRes2
// Close the Opened Regkey
lRegRetVal = RegCloseKey(hkeySubSystem);
if(lRegRetVal != ERROR_SUCCESS)
{
// Could not close the reg Key
ErrorTrace(TRACE_ID, "RegClose Sub Key Failed.");
}
}
// Check to see if further enumeration is required.
// continue to enumerate.
if(fContinueEnum)
{
dwSize = MAX_PATH;
dwIndex++;
lRegRetVal = RegEnumKeyEx(hkeySystem, dwIndex, tchSubSystemKeyName, &dwSize, NULL, NULL, NULL, pFileTime);
if(lRegRetVal != ERROR_SUCCESS)
{
fContinueEnum = FALSE;
}
}
} // end of while
lRegRetVal = RegCloseKey(hkeySystem);
if(lRegRetVal != ERROR_SUCCESS)
{
// Could not close the reg Key
ErrorTrace(TRACE_ID, "RegClose Key Failed.");
}
}
// Got all the properties for PCH_BIOS Class
if(fCommit)
{
hRes = pPCHBIOSInstance->Commit();
if(FAILED(hRes))
{
// Could not Commit the instance
ErrorTrace(TRACE_ID, "Commit on PCHBiosInstance Failed");
}
}
TraceFunctLeave();
return hRes ;
}