Source code of Windows XP (NT5)
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.
 
 
 
 
 
 

481 lines
16 KiB

/******************************************************************
PrintSys.CPP -- WMI provider class implementation
Generated by Microsoft WMI Code Generation Engine
TO DO: - See individual function headers
- When linking, make sure you link to framedyd.lib &
msvcrtd.lib (debug) or framedyn.lib & msvcrt.lib (retail).
Description:
******************************************************************/
#include "pchealth.h"
#include "PrintSys.h"
/////////////////////////////////////////////////////////////////////////////
// tracing stuff
#ifdef THIS_FILE
#undef THIS_FILE
#endif
static char __szTraceSourceFile[] = __FILE__;
#define THIS_FILE __szTraceSourceFile
#define TRACE_ID DCID_PRINTERDRIVER
/////////////////////////////////////////////////////////////////////////////
// initialization
CPrintSys MyPrintSysSet (PROVIDER_NAME_PRINTSYS, PCH_NAMESPACE) ;
/////////////////////////////////////////////////////////////////////////////
// Property names
const static WCHAR *c_wszGenDrv = L"GenDrv";
const static WCHAR *c_wszName = L"Name";
const static WCHAR *c_wszPath = L"Path";
const static WCHAR *c_wszUniDrv = L"UniDrv";
const static WCHAR *c_wszUsePrintMgrSpooling = L"UsePrintMgrSpooling";
//////////////////////////////////////////////////////////////////////////////
// construction / destruction
// ***************************************************************************
CPrintSys::CPrintSys (LPCWSTR lpwszName, LPCWSTR lpwszNameSpace ) :
Provider(lpwszName, lpwszNameSpace)
{
m_pParamOut = NULL;
m_pCurrent = NULL;
m_pParamIn = NULL;
m_lFlags = 0;
}
// ***************************************************************************
CPrintSys::~CPrintSys()
{
}
//////////////////////////////////////////////////////////////////////////////
// exposed methods
// *****************************************************************************
HRESULT CPrintSys::EnumerateInstances(MethodContext *pMethodContext, long lFlags)
{
TraceFunctEnter("CPrintSys::EnumerateInstances");
HRESULT hr = WBEM_S_NO_ERROR;
// Create a new instance of PCH_Printer Class based on the passed-in MethodContext
CInstancePtr pPrintSysInst(CreateNewInstance(pMethodContext), false);
hr = this->GetObject(pPrintSysInst, 0);
TraceFunctLeave();
return hr;
}
// *****************************************************************************
HRESULT CPrintSys::ExecMethod (const CInstance& Instance,
const BSTR bstrMethodName,
CInstance *pInParams, CInstance *pOutParams,
long lFlags)
{
return WBEM_E_PROVIDER_NOT_CAPABLE;
}
// *****************************************************************************
HRESULT CPrintSys::GetObject(CInstance* pInstance, long lFlags)
{
TraceFunctEnter("CPrintSys::GetObject");
HRESULT hr = WBEM_S_NO_ERROR;
// Objects
IWbemClassObjectPtr pFileObj = NULL;
// Variants
CComVariant varValue;
CComVariant varNotAvail = L"Not Available";
// Strings
CComBSTR bstrDriverWithPath;
CComBSTR bstrDetails;
CComBSTR bstrVersion = L"Version";
CComBSTR bstrFileSize = L"FileSize";
CComBSTR bstrModifiedDate = L"LastModified";
LPCTSTR lpctstrUniDriver = _T("unidrv.dll");
LPCTSTR lpctstrGenDriver = _T("gendrv.dll");
LPCTSTR lpctstrWindows = _T("Windows");
LPCTSTR lpctstrDevice = _T("Device");
LPCTSTR lpctstrNoUniDrv = _T("(unidrv.dll) = NotInstalled");
LPCTSTR lpctstrNoGenDrv = _T("(gendrv.dll) = NotInstalled");
LPCTSTR lpctstrPrintersHive = _T("System\\CurrentControlSet\\Control\\Print\\Printers");
LPCTSTR lpctstrYes = _T("yes");
LPCTSTR lpctstrAttributes = _T("Attributes");
LPCTSTR lpctstrSpooler = _T("Spooler");
TCHAR tchBuffer[MAX_PATH + 1];
TCHAR tchPrinterKeyName[MAX_PATH + 1];
TCHAR *ptchToken;
// Booleans
BOOL fDriverFound = FALSE;
BOOL fAttribFound = FALSE;
// DWORDs
DWORD dwSize;
DWORD dwIndex;
DWORD dwType;
// Return Values;
ULONG ulPrinterAttribs;
LONG lRegRetVal;
struct tm tm;
WBEMTime wbemtime;
HKEY hkeyPrinter = NULL;
HKEY hkeyPrinters = NULL;
FILETIME ft;
// *** Set the properties associated with the default printer
// In "win.ini" file under "Windows" section "Device" represents the default printer
if(GetProfileString(lpctstrWindows, lpctstrDevice, "\0", tchBuffer, MAX_PATH) > 1)
{
// The Above GetProfileString returns "printerName", "PrinterDriver"
// and "PrinterPath" seperated by commas. Ignore "PrinterDriver"
// and use the other two to set the properties.
ptchToken = _tcstok(tchBuffer, _T(","));
if(ptchToken != NULL)
{
// ** name (token 1)
varValue = ptchToken;
if (pInstance->SetVariant(c_wszName, varValue) == FALSE)
ErrorTrace(TRACE_ID, "SetVariant on Name failed.");
// ** path (token 3)
ptchToken = _tcstok(NULL, _T(","));
if(ptchToken != NULL)
{
// gotta skip the 2nd token cuz it's the printer dirver & we
// don't give a rat's patoshki about it at this moment...
ptchToken = _tcstok(NULL, _T(","));
if(ptchToken != NULL)
{
varValue = ptchToken;
if (pInstance->SetVariant(c_wszPath, varValue) == FALSE)
ErrorTrace(TRACE_ID, "Set Variant on Path failed.");
}
}
}
}
// couldn't fetch the properties of the default printer, so shove in some
// default values...
else
{
// set Name to "Not Available"
if (pInstance->SetVariant(c_wszName, varNotAvail) == FALSE)
ErrorTrace(TRACE_ID, "Se Variant on Name failed.");
// set Path to "Not Available"
if (pInstance->SetVariant(c_wszPath, varValue) == FALSE)
ErrorTrace(TRACE_ID, "Set Variant on Path failed.");
}
// *** Set the properties associated with using print manager spooling
// First try to get the Spooling information from the registry which is
// available if there are any installed printers
// HKLM\system\CCS\Control\Print\Printers
lRegRetVal = RegOpenKeyEx(HKEY_LOCAL_MACHINE, lpctstrPrintersHive, 0, KEY_READ, &hkeyPrinters);
if(lRegRetVal == ERROR_SUCCESS)
{
// Enumerate the keys under this hive.
ZeroMemory(&ft, sizeof(ft));
dwIndex = 0;
dwSize = MAX_PATH;
lRegRetVal = RegEnumKeyEx(hkeyPrinters, dwIndex, tchPrinterKeyName, &dwSize, NULL, NULL, NULL, &ft);
if(lRegRetVal == ERROR_SUCCESS)
{
// There is atleast one printer installed.
lRegRetVal = RegOpenKeyEx(hkeyPrinters, tchPrinterKeyName, 0, KEY_READ, &hkeyPrinter);
if(lRegRetVal == ERROR_SUCCESS)
{
// Opened the first printer key
// Query for , regname "Attributes"
dwSize = sizeof(DWORD);
lRegRetVal = RegQueryValueEx(hkeyPrinter, lpctstrAttributes, NULL, &dwType, (LPBYTE)&ulPrinterAttribs, &dwSize);
if(lRegRetVal == ERROR_SUCCESS)
{
// if the PRINTER_ATTRIBUTE_DIRECT bit in ulPrinterAttribs
// is set, then we have spooling...
if((ulPrinterAttribs & PRINTER_ATTRIBUTE_DIRECT) != 0)
varValue = VARIANT_FALSE;
else
varValue = VARIANT_TRUE;
if (ulPrinterAttribs > 0)
fAttribFound = TRUE;
}
}
}
}
if (hkeyPrinter != NULL)
{
RegCloseKey(hkeyPrinter);
hkeyPrinter = NULL;
}
if (hkeyPrinters != NULL)
{
RegCloseKey(hkeyPrinters);
hkeyPrinters = NULL;
}
if(fAttribFound == FALSE)
{
// If not get the "spooler" key value from the win.ini file. If the entry is not present default to "yes".
if(GetProfileString(lpctstrWindows, lpctstrSpooler, _T("yes"), tchBuffer, MAX_PATH) > 1)
{
// if it's yes, then we have spooling...
if(_tcsicmp(tchBuffer, lpctstrYes) == 0)
varValue = VARIANT_TRUE;
else
varValue = VARIANT_FALSE;
}
}
// Set the Spooling Property.
if (pInstance->SetVariant(c_wszUsePrintMgrSpooling, varValue) == FALSE)
ErrorTrace(TRACE_ID, "SetVariant on usePrintManagerSpooling failed.");
// *** Set the properties associated with using print manager spooling
// Get the complete path for unidrv.dll
fDriverFound = getCompletePath(lpctstrUniDriver, bstrDriverWithPath);
if(fDriverFound)
{
// Need to use GetCIMDataFile to get the unidriver properties
if (SUCCEEDED(GetCIMDataFile(bstrDriverWithPath, &pFileObj)))
{
bstrDetails.Empty();
// Get the Version & append it to the value string...
varValue.Clear();
hr = pFileObj->Get(bstrVersion, 0, &varValue, NULL, NULL);
if(SUCCEEDED(hr))
{
if(varValue.vt == VT_BSTR)
{
bstrDetails.Append(varValue.bstrVal);
bstrDetails.Append(_T(" "));
}
}
// Get the FileSize & append it to the value string...
varValue.Clear();
hr = pFileObj->Get(bstrFileSize, 0, &varValue, NULL, NULL);
if(SUCCEEDED(hr))
{
if(varValue.vt == VT_BSTR)
{
bstrDetails.Append(varValue.bstrVal);
bstrDetails.Append(_T(" "));
}
}
// Get the date & time & append them to the value string...
varValue.Clear();
hr = pFileObj->Get(bstrModifiedDate, 0, &varValue, NULL, NULL);
if(SUCCEEDED(hr))
{
if(varValue.vt == VT_BSTR)
{
WCHAR *pwsz;
// there is a slight problem when WMI returns to us a
// time that has a '*' in it. The WBEMTime class plain
// refuses to deal with it. So change '*'s to '0's...
for (pwsz = varValue.bstrVal; *pwsz != L'\0'; pwsz++)
{
if (*pwsz == L'*')
*pwsz = L'0';
}
wbemtime = varValue.bstrVal;
if(wbemtime.GetStructtm(&tm))
{
varValue = asctime(&tm);
bstrDetails.Append(varValue.bstrVal);
}
}
}
// set the value
varValue.vt = VT_BSTR;
varValue.bstrVal = bstrDetails.Detach();
}
// Since I don't seem to have unindrv.dll installed, I can't verify
// this, but going by the other WMI providers, GetObject doesn't seem
// to like it when u release objects fetched by it. So, I am not
// going to release it.
if (pFileObj != NULL)
{
// pFileObj->Release();
pFileObj = NULL;
}
}
// the unidriver dll isn't present. Use a default value
else
{
varValue.Clear();
varValue = lpctstrNoUniDrv;
}
// set the property
if (pInstance->SetVariant(c_wszUniDrv, varValue) == FALSE)
ErrorTrace(TRACE_ID, "SetVariant on UniDriver failed.");
// *** Set the properties associated with using print manager spooling
// Get the complete path for gendrv.dll
bstrDriverWithPath.Empty();
fDriverFound = getCompletePath(lpctstrGenDriver, bstrDriverWithPath);
if(fDriverFound)
{
bstrDetails.Empty();
// Need to use GetCIMDataFile to get the gen driver properties
if(SUCCEEDED(GetCIMDataFile(bstrDriverWithPath, &pFileObj)))
{
// Get the Version & append it to the value string...
varValue.Clear();
hr = pFileObj->Get(bstrVersion, 0, &varValue, NULL, NULL);
if(SUCCEEDED(hr))
{
if(varValue.vt == VT_BSTR)
{
bstrDetails.Append(varValue.bstrVal);
bstrDetails.Append(_T(" "));
}
}
// Get the FileSize & append it to the value string...
varValue.Clear();
hr = pFileObj->Get(bstrFileSize, 0, &varValue, NULL, NULL);
if(SUCCEEDED(hr))
{
if(varValue.vt == VT_BSTR)
{
bstrDetails.Append(varValue.bstrVal);
bstrDetails.Append(_T(" "));
}
}
// Get the date & time & append them to the value string...
varValue.Clear();
hr = pFileObj->Get(bstrModifiedDate, 0, &varValue, NULL, NULL);
if(SUCCEEDED(hr))
{
if(varValue.vt == VT_BSTR)
{
WCHAR *pwsz;
// there is a slight problem when WMI returns to us a
// time that has a '*' in it. The WBEMTime class plain
// refuses to deal with it. So change '*'s to '0's...
for (pwsz = varValue.bstrVal; *pwsz != L'\0'; pwsz++)
{
if (*pwsz == L'*')
*pwsz = L'0';
}
wbemtime = varValue.bstrVal;
if(wbemtime.GetStructtm(&tm))
{
varValue = asctime(&tm);
bstrDetails.Append(varValue.bstrVal);
}
}
}
// set the value
varValue.vt = VT_BSTR;
varValue.bstrVal = bstrDetails.Detach();
}
// Since I don't seem to have gendrv.dll installed, I can't verify
// this, but going by the other WMI providers, GetObject doesn't seem
// to like it when u release objects fetched by it. So, I am not
// going to release it.
if (pFileObj != NULL)
{
// pFileObj->Release();
pFileObj = NULL;
}
}
// the gen driver dll is not present...
else
{
varValue.Clear();
varValue = lpctstrNoGenDrv;
}
if (pInstance->SetVariant(c_wszGenDrv, varValue) == FALSE)
ErrorTrace(TRACE_ID, "SetVariant on GenDrv failed.");
// WOOHOO!! We can commit now
hr = pInstance->Commit();
if(FAILED(hr))
ErrorTrace(TRACE_ID, "Error on commiting!");
TraceFunctLeave();
return hr;
}
// *****************************************************************************
HRESULT CPrintSys::ExecQuery(MethodContext *pMethodContext,
CFrameworkQuery& Query, long lFlags)
{
return WBEM_E_PROVIDER_NOT_CAPABLE;
}
// *****************************************************************************
HRESULT CPrintSys::PutInstance(const CInstance& Instance, long lFlags)
{
return WBEM_E_PROVIDER_NOT_CAPABLE;
}
// *****************************************************************************
HRESULT CPrintSys::DeleteInstance(const CInstance& Instance, long lFlags)
{
return WBEM_E_PROVIDER_NOT_CAPABLE;
}