|
|
// ScriptSupport.cpp: implementation for our scripting support class CScriptSupport
//
// Copyright (c)1997-2001 Microsoft Corporation
//
//////////////////////////////////////////////////////////////////////
#include "precomp.h"
#include "ScriptSupport.h"
/*
Routine Description:
Name:
CScriptSupport::CScriptSupport
Functionality:
Constructor
Virtual: No
Arguments:
None
Return Value:
None
Notes:
*/
CScriptSupport::CScriptSupport (){ //
// this is not part of WMI provider, so, we need to do the security
//
HRESULT hr = ::CoInitializeSecurity( NULL, -1, NULL, NULL, RPC_C_AUTHN_LEVEL_CONNECT, RPC_C_IMP_LEVEL_IDENTIFY, NULL, EOAC_NONE, 0 );
}
/*
Routine Description:
Name:
CScriptSupport::~CScriptSupport
Functionality:
Destructor
Virtual: Yes.
Arguments:
None
Return Value:
None
Notes:
*/CScriptSupport::~CScriptSupport (){}
/*
Routine Description:
Name:
CScriptSupport::InterfaceSupportsErrorInfo
Functionality:
Queries if we support IErrorInfo
Virtual: Yes (part of ISupportErrorInfo)
Arguments:
riid - Interface ID (guid).
Return Value:
S_OK
Notes: $undone:shawnwu, This is just testing for now. Don't check in the code.
*/ STDMETHODIMP CScriptSupport::InterfaceSupportsErrorInfo ( REFIID riid ){ //return (riid == IID_INetSecProvMgr) ? NOERROR : ResultFromScode(S_FALSE);
return S_FALSE;}
/*
Routine Description:
Name:
CScriptSupport::get_RandomPortLower
Functionality:
Get the random port's lower bound.
Virtual: Yes (part of INetSecProvMgr)
Arguments:
plLower - Receivs the lower bound of random port range.
Return Value:
S_OK
Notes: $undone:shawnwu, This is just testing for now. Don't check in the code.
*/ STDMETHODIMP CScriptSupport::get_RandomPortLower ( OUT long * plLower ){ *plLower = 65000;
return S_OK;}
/*
Routine Description:
Name:
CScriptSupport::gett_RandomPortUpper
Functionality:
Get the random port's upper bound.
Virtual: Yes (part of INetSecProvMgr)
Arguments:
plUpper - Receivs the upper bound of random port range.
Return Value:
S_OK
Notes: $undone:shawnwu, This is just testing for now. Don't check in the code.
*/ STDMETHODIMP CScriptSupport::get_RandomPortUpper ( OUT long * plUpper ){ *plUpper = 65012;
return S_OK;}
/*
Routine Description:
Name:
CScriptSupport::GetNamespace
Functionality:
Private helper for finding the provider's service interface given its namespace string.
Virtual: No.
Arguments:
bstrNamespace - Namespace string.
ppNS - Receives the namespace.
Return Value:
Success: S_OK.
Failure: other error codes.
Notes: $undone:shawnwu, This is just testing for now. Don't check in the code.
*/
HRESULTCScriptSupport::GetNamespace ( IN BSTR bstrNamespace, OUT IWbemServices ** ppNS ){ if (ppNS == NULL) { return E_INVALIDARG; }
*ppNS = NULL;
CComPtr<IWbemLocator> srpLocator; HRESULT hr = ::CoCreateInstance(CLSID_WbemLocator, 0, CLSCTX_INPROC_SERVER, IID_IWbemLocator, (LPVOID *) &srpLocator );
if (SUCCEEDED(hr) && srpLocator) { //
// Ask the locator to find the SCE provider.
//
CComPtr<IWbemServices> srpNamespace;
hr = srpLocator->ConnectServer(bstrNamespace, NULL, NULL, NULL, 0, NULL, NULL, &srpNamespace);
if (SUCCEEDED(hr) && srpNamespace) { //
// Set the proxy so that impersonation of the client occurs.
//
hr = ::CoSetProxyBlanket( srpNamespace, RPC_C_AUTHN_WINNT, RPC_C_AUTHZ_NONE, NULL, RPC_C_AUTHN_LEVEL_CALL, RPC_C_IMP_LEVEL_IMPERSONATE, NULL, EOAC_NONE ); if (SUCCEEDED(hr)) { *ppNS = srpNamespace.Detach(); hr = S_OK; } } }
return hr;
}
/*
Routine Description:
Name:
CScriptSupport::ExecuteQuery
Functionality:
Execute the given query.
Virtual: Yes (part of INetSecProvMgr)
Arguments:
bstrNaemspace - The provider namespace.
bstrQuery - The query to be executed.
plSucceeded - Receives the execution result. It is = 1 if succeeded and 0 otherwise.
Return Value:
S_OK
Notes: $undone:shawnwu, This is just testing for now. Don't check in the code.
*/ STDMETHODIMP CScriptSupport::ExecuteQuery ( IN BSTR bstrNamespace, IN BSTR bstrQuery, IN BSTR bstrDelimiter, IN BSTR bstrPropName, OUT BSTR * pbstrResult ){ *pbstrResult = 0;
CComPtr<IWbemServices> srpNamespace; HRESULT hr = GetNamespace(bstrNamespace, &srpNamespace);
if (SUCCEEDED(hr)) { CComPtr<IEnumWbemClassObject> srpEnum;
//
// querying the objects
//
hr = srpNamespace->ExecQuery(L"WQL", bstrQuery, WBEM_FLAG_RETURN_IMMEDIATELY | WBEM_FLAG_FORWARD_ONLY, NULL, &srpEnum );
if (SUCCEEDED(hr)) { //
// find out how many we have
//
ULONG nEnum = 0;
CComVariant var;
//
// push the property values to this vector for later packing
//
std::vector<BSTR> vecPropValues;
//
// total length of the result string, we need a 0 terminator, that is it is initialized to 1
//
long lTotLen = 1;
//
// Length of the delimiter
//
long lDelLen = wcslen(bstrDelimiter);
while (SUCCEEDED(hr)) { CComPtr<IWbemClassObject> srpObj; hr = srpEnum->Next(WBEM_INFINITE, 1, &srpObj, &nEnum); if (SUCCEEDED(hr) && srpObj != NULL) { hr = srpObj->Get(bstrPropName, 0, &var, NULL, NULL); if (SUCCEEDED(hr) && var.vt != VT_EMPTY && var.vt != VT_NULL) { CComVariant varResult; if (SUCCEEDED(::VariantChangeType(&varResult, &var, 0, VT_BSTR))) { vecPropValues.push_back(varResult.bstrVal); lTotLen += wcslen(varResult.bstrVal) + lDelLen; varResult.vt = VT_EMPTY; } } } else { break; } } *pbstrResult = ::SysAllocStringLen(NULL, lTotLen);
if (*pbstrResult != NULL) {
//
// running head of copying
//
LPWSTR pDest = *pbstrResult; pDest[0] = L'\0';
for (int i = 0; i < vecPropValues.size(); i++) { wcscpy(pDest, vecPropValues[i]); pDest += wcslen(vecPropValues[i]); wcscpy(pDest, bstrDelimiter); pDest += lDelLen; } } } }
return S_OK;}
/*
Routine Description:
Name:
CScriptSupport::GetProperty
Functionality:
Get the property value of the given object's given property.
Virtual: Yes (part of INetSecProvMgr)
Arguments:
bstrNaemspace - The provider namespace.
bstrObjectPath - The path for the object.
bstrPropName - The name of the property.
pvarValue - Receives the value in string format.
Return Value:
S_OK
Notes: $undone:shawnwu, This is just testing for now. Don't check in the code.
*/ STDMETHODIMP CScriptSupport::GetProperty ( IN BSTR bstrNamespace, IN BSTR bstrObjectPath, IN BSTR bstrPropName, OUT VARIANT * pvarValue ){ ::VariantInit(pvarValue);
CComPtr<IWbemServices> srpNamespace; HRESULT hr = GetNamespace(bstrNamespace, &srpNamespace);
if (SUCCEEDED(hr)) { CComPtr<IWbemClassObject> srpObj;
hr = srpNamespace->GetObject( bstrObjectPath, 0, NULL, &srpObj, NULL ); if (SUCCEEDED(hr)) { hr = srpObj->Get(bstrPropName, 0, pvarValue, NULL, NULL); } }
return hr;}
|
