|
|
/********************************************************************
Copyright (c) 1999 Microsoft Corporation
Module Name: PCH_PrinterDriver.CPP
Abstract: WBEM provider class implementation for PCH_PrinterDriver class
Revision History:
Ghim-Sim Chua (gschua) 04/27/99 - Created
Kalyani Narlanka (kalyanin) 05/11/99 - Added Code to get all the properties for this class
********************************************************************/
#include "pchealth.h"
#include "PCH_PrinterDriver.h"
/////////////////////////////////////////////////////////////////////////////
// tracing stuff
#ifdef THIS_FILE
#undef THIS_FILE
#endif
static char __szTraceSourceFile[] = __FILE__;#define THIS_FILE __szTraceSourceFile
#define TRACE_ID DCID_PRINTERDRIVER
CPCH_PrinterDriver MyPCH_PrinterDriverSet (PROVIDER_NAME_PCH_PRINTERDRIVER, PCH_NAMESPACE) ;
// Property names
//===============
const static WCHAR* pTimeStamp = L"TimeStamp" ;const static WCHAR* pChange = L"Change" ;const static WCHAR* pDate = L"Date" ;const static WCHAR* pFilename = L"Filename" ;const static WCHAR* pManufacturer = L"Manufacturer" ;const static WCHAR* pName = L"Name" ;const static WCHAR* pSize = L"Size" ;const static WCHAR* pVersion = L"Version" ;const static WCHAR* pPath = L"Path" ;
/*****************************************************************************
** FUNCTION : CPCH_PrinterDriver::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_PrinterDriver::EnumerateInstances( MethodContext* pMethodContext, long lFlags ){
TraceFunctEnter("CPCH_PrinterDriver::EnumerateInstances");
// Begin Declarations
HRESULT hRes = WBEM_S_NO_ERROR;
// Query String
CComBSTR bstrPrinterQuery = L"Select DeviceID, PortName FROM win32_printer";
// Instances
CComPtr<IEnumWbemClassObject> pPrinterEnumInst; CInstance *pPCHPrinterDriverInstance;
// SystemTime
SYSTEMTIME stUTCTime;
// Objects
IWbemClassObjectPtr pPrinterObj; IWbemClassObjectPtr pFileObj;
// Unsigned Longs....
ULONG ulPrinterRetVal = 0; ULONG uiReturn = 0;
// File Status structure
struct _stat filestat;
// Strings
CComBSTR bstrPrinterDriverWithPath; CComBSTR bstrPrinterDriver; CComBSTR bstrProperty; CComBSTR bstrDeviceID = L"DeviceID";
LPCWSTR lpctstrPortName = L"PortName"; LPCWSTR lpctstrFileSize = L"FileSize"; LPCWSTR lpctstrLastModified = L"LastModified"; LPCWSTR lpctstrManufacturer = L"Manufacturer"; LPCWSTR lpctstrVersion = L"Version"; LPCTSTR lpctstrComma = _T(","); LPCTSTR lpctstrDrvExtension = _T(".drv"); LPCTSTR lpctstrDevices = _T("Devices"); LPCWSTR lpctstrDeviceID = L"DeviceID";
TCHAR tchDeviceID[MAX_PATH];
TCHAR tchBuffer[MAX_PATH]; TCHAR *ptchToken;
CComVariant varValue; CComVariant varSnapshot = "Snapshot"; BOOL fDriverFound;
BOOL fCommit = FALSE;
// Get the date and time to update the TimeStamp Field
GetSystemTime(&stUTCTime);
// Execute the query to get DeviceID, PORTName FROM Win32_Printer
// Class.
// pPrinterEnumInst contains a pointer to the list of instances returned.
tchDeviceID[0] = 0;
hRes = ExecWQLQuery(&pPrinterEnumInst, bstrPrinterQuery); if (FAILED(hRes)) { // Cannot get any properties.
goto END; } // Query Succeeded!
// Enumerate the instances from pPrinterEnumInstance
// Get the next instance into pPrinterObj object.
while(WBEM_S_NO_ERROR == pPrinterEnumInst->Next(WBEM_INFINITE, 1, &pPrinterObj, &ulPrinterRetVal)) { // Create a new instance of PCH_PrinterDriver Class based on the passed-in MethodContext
CInstancePtr pPCHPrinterDriverInstance(CreateNewInstance(pMethodContext),false);
// Created a New Instance of PCH_PrinterDriver Successfully.
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
// TIME STAMP //
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
hRes = pPCHPrinterDriverInstance->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 = pPCHPrinterDriverInstance->SetVariant(pChange, varSnapshot); if (FAILED(hRes)) { // Could not Set the Change Property
// Continue anyway
ErrorTrace(TRACE_ID, "SetCHString on Change Field failed."); }
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
// NAME //
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
hRes = CopyProperty(pPrinterObj, lpctstrDeviceID, pPCHPrinterDriverInstance, pName); if(SUCCEEDED(hRes)) { fCommit = TRUE; }
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
// PATH //
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
CopyProperty(pPrinterObj, lpctstrPortName, pPCHPrinterDriverInstance, pPath); /////////////////////////////////////////////////////////////////////////////////////////////////////////////
// FILENAME //
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
// Get the File Name i.e.driver from the INI file.
// Use the DeviceID of win32_printer class to get the PCH_Printer.FileName
// Get the device ID and use it to pick up the driver from win.ini File
hRes = pPrinterObj->Get(bstrDeviceID, 0, &varValue, NULL, NULL); if(SUCCEEDED(hRes)) { // Got the DeviceID
// Now call GetProfileString to get the Driver
USES_CONVERSION; _tcscpy(tchDeviceID,W2T(varValue.bstrVal)); if (GetProfileString(lpctstrDevices, tchDeviceID, "\0", tchBuffer, MAX_PATH) > 1) { // tchBuffer contains a string of two tokens, first the driver, second the PathName
// Get the driver
ptchToken = _tcstok(tchBuffer,lpctstrComma); if(ptchToken != NULL) { // Got the Driver Name
bstrPrinterDriver = ptchToken; varValue = ptchToken; // Use this to set the FileName
hRes = pPCHPrinterDriverInstance->SetVariant(pFilename, varValue); if (FAILED(hRes)) { // Could not Set the FileName Property
// Continue anyway
ErrorTrace(TRACE_ID, "Set Variant on Change Field failed."); }
// Now get the properties of the File
// Concatenate ".drv" to get the driver's actual Name
bstrPrinterDriver.Append(lpctstrDrvExtension);
// Get the Complete Path of the File
fDriverFound = getCompletePath(bstrPrinterDriver, bstrPrinterDriverWithPath); if (fDriverFound) { // Got the complete Path Call GetCIMDataFile Function to get
// properties of this file.
if (SUCCEEDED(GetCIMDataFile(bstrPrinterDriverWithPath, &pFileObj))) {
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
// VERSION //
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
CopyProperty(pFileObj, lpctstrVersion, pPCHPrinterDriverInstance, pVersion);
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
// FILESIZE //
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
CopyProperty(pFileObj, lpctstrFileSize, pPCHPrinterDriverInstance, pSize);
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
// DATE //
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
CopyProperty(pFileObj, lpctstrLastModified, pPCHPrinterDriverInstance, pDate);
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
// MANUFACTURER //
/////////////////////////////////////////////////////////////////////////////////////////////////////////////
CopyProperty(pFileObj, lpctstrManufacturer, pPCHPrinterDriverInstance, pManufacturer);
} //end of SUCCEEDED...
} // end of if fDriverFound
} // end of if (ptchToken != NULL)
} // end of GetProfileString...
}// end of got the DeviceID
// All the properties are set. Commit the instance
hRes = pPCHPrinterDriverInstance->Commit(); if(FAILED(hRes)) { // Could not Commit the instance
ErrorTrace(TRACE_ID, "Could not commit the instance"); }
} // end of While
END: TraceFunctLeave(); return hRes;
}
|
