356 lines
12 KiB
C++
356 lines
12 KiB
C++
|
|
||
|
|
//***************************************************************************
|
||
|
|
//
|
||
|
|
// (c) 1999-2001 by Microsoft Corp. All Rights Reserved.
|
||
|
|
//
|
||
|
|
// sqlit.cpp
|
||
|
|
//
|
||
|
|
// cvadai 19-Mar-99 Created as prototype for Quasar.
|
||
|
|
//
|
||
|
|
//***************************************************************************
|
||
|
|
|
||
|
|
#define _SQLIT_CPP_
|
||
|
|
#pragma warning( disable : 4786 ) // identifier was truncated to 'number' characters in the
|
||
|
|
#pragma warning( disable : 4251 ) // needs to have dll-interface to be used by clients of class
|
||
|
|
|
||
|
|
#define DBINITCONSTANTS // Initialize OLE constants...
|
||
|
|
#define INITGUID // ...once in each app.
|
||
|
|
#define _WIN32_DCOM
|
||
|
|
#include "precomp.h"
|
||
|
|
|
||
|
|
#include <std.h>
|
||
|
|
#include <sqlutils.h>
|
||
|
|
#include <repdrvr.h>
|
||
|
|
#include <sqlit.h>
|
||
|
|
#include <wbemint.h>
|
||
|
|
|
||
|
|
//***************************************************************************
|
||
|
|
//
|
||
|
|
// CWmiDbIterator::CWmiDbIterator
|
||
|
|
//
|
||
|
|
//***************************************************************************
|
||
|
|
|
||
|
|
CWmiDbIterator::CWmiDbIterator()
|
||
|
|
{
|
||
|
|
m_pStatus = NULL;
|
||
|
|
m_pRowset = NULL;
|
||
|
|
m_pSession = NULL;
|
||
|
|
m_pIMalloc = NULL;
|
||
|
|
m_pConn = NULL;
|
||
|
|
m_uRefCount = 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
//***************************************************************************
|
||
|
|
//
|
||
|
|
// CWmiDbIterator::~CWmiDbIterator
|
||
|
|
//
|
||
|
|
//***************************************************************************
|
||
|
|
CWmiDbIterator::~CWmiDbIterator()
|
||
|
|
{
|
||
|
|
Cancel(0);
|
||
|
|
if (m_pSession)
|
||
|
|
m_pSession->Release();
|
||
|
|
if (m_pIMalloc)
|
||
|
|
m_pIMalloc->Release();
|
||
|
|
}
|
||
|
|
|
||
|
|
//***************************************************************************
|
||
|
|
//
|
||
|
|
// CWmiDbIterator::QueryInterface
|
||
|
|
//
|
||
|
|
//***************************************************************************
|
||
|
|
|
||
|
|
HRESULT STDMETHODCALLTYPE CWmiDbIterator::QueryInterface
|
||
|
|
(REFIID riid,
|
||
|
|
void __RPC_FAR *__RPC_FAR *ppvObject)
|
||
|
|
{
|
||
|
|
*ppvObject = 0;
|
||
|
|
|
||
|
|
if (IID_IUnknown==riid || IID_IWmiDbIterator==riid )
|
||
|
|
{
|
||
|
|
*ppvObject = (IWmiDbIterator *)this;
|
||
|
|
AddRef();
|
||
|
|
return S_OK;
|
||
|
|
}
|
||
|
|
|
||
|
|
return E_NOINTERFACE;
|
||
|
|
}
|
||
|
|
|
||
|
|
//***************************************************************************
|
||
|
|
//
|
||
|
|
// CWmiDbIterator::AddRef
|
||
|
|
//
|
||
|
|
//***************************************************************************
|
||
|
|
|
||
|
|
ULONG STDMETHODCALLTYPE CWmiDbIterator::AddRef()
|
||
|
|
{
|
||
|
|
InterlockedIncrement((LONG *) &m_uRefCount);
|
||
|
|
return m_uRefCount;
|
||
|
|
}
|
||
|
|
|
||
|
|
//***************************************************************************
|
||
|
|
//
|
||
|
|
// CWmiDbIterator::Release
|
||
|
|
//
|
||
|
|
//***************************************************************************
|
||
|
|
|
||
|
|
ULONG STDMETHODCALLTYPE CWmiDbIterator::Release()
|
||
|
|
{
|
||
|
|
ULONG uNewCount = InterlockedDecrement((LONG *) &m_uRefCount);
|
||
|
|
if (0 != uNewCount)
|
||
|
|
return uNewCount;
|
||
|
|
delete this;
|
||
|
|
return WBEM_S_NO_ERROR;
|
||
|
|
}
|
||
|
|
//***************************************************************************
|
||
|
|
//
|
||
|
|
// CWmiDbIterator::Cancel
|
||
|
|
//
|
||
|
|
//***************************************************************************
|
||
|
|
|
||
|
|
HRESULT STDMETHODCALLTYPE CWmiDbIterator::Cancel(
|
||
|
|
/* [in] */ DWORD dwFlags)
|
||
|
|
{
|
||
|
|
HRESULT hr = WBEM_S_NO_ERROR;
|
||
|
|
|
||
|
|
hr = CSQLExecute::CancelQuery(m_pStatus);
|
||
|
|
|
||
|
|
if (m_pStatus)
|
||
|
|
m_pStatus->Release();
|
||
|
|
m_pStatus = NULL;
|
||
|
|
|
||
|
|
if (m_pConn)
|
||
|
|
{
|
||
|
|
((CWmiDbController *)m_pSession->m_pController)->ConnCache.ReleaseConnection(m_pConn, hr);
|
||
|
|
m_pConn = NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (m_pRowset)
|
||
|
|
m_pRowset->Release();
|
||
|
|
m_pRowset = NULL;
|
||
|
|
|
||
|
|
return hr;
|
||
|
|
}
|
||
|
|
|
||
|
|
//***************************************************************************
|
||
|
|
//
|
||
|
|
// CWmiDbIterator::NextBatch
|
||
|
|
//
|
||
|
|
//***************************************************************************
|
||
|
|
|
||
|
|
HRESULT STDMETHODCALLTYPE CWmiDbIterator::NextBatch(
|
||
|
|
/* [in] */ DWORD dwNumRequested,
|
||
|
|
/* [in] */ DWORD dwTimeOutSeconds,
|
||
|
|
/* [in] */ DWORD dwFlags,
|
||
|
|
/* [in] */ DWORD dwRequestedHandleType,
|
||
|
|
/* [in] */ REFIID riid,
|
||
|
|
/* [out] */ DWORD __RPC_FAR *pdwNumReturned,
|
||
|
|
/* [iid_is][length_is][size_is][out] */ LPVOID __RPC_FAR *ppObjects)
|
||
|
|
{
|
||
|
|
|
||
|
|
HRESULT hr = WBEM_S_NO_ERROR, hrRet = WBEM_S_NO_ERROR;
|
||
|
|
bool bImmediate = !(dwRequestedHandleType & WMIDB_HANDLE_TYPE_SUBSCOPED);
|
||
|
|
|
||
|
|
if (!m_pSession || !(m_pSession->m_pController) ||
|
||
|
|
((CWmiDbController *)m_pSession->m_pController)->m_dwCurrentStatus == WBEM_E_SHUTTING_DOWN)
|
||
|
|
return WBEM_E_SHUTTING_DOWN;
|
||
|
|
|
||
|
|
if (!dwNumRequested || !ppObjects)
|
||
|
|
return WBEM_E_INVALID_PARAMETER;
|
||
|
|
|
||
|
|
if (dwRequestedHandleType == WMIDB_HANDLE_TYPE_INVALID &&
|
||
|
|
riid == IID_IWmiDbHandle)
|
||
|
|
return WBEM_E_INVALID_PARAMETER;
|
||
|
|
|
||
|
|
// FIXME: We need to create a readahead cache.
|
||
|
|
|
||
|
|
if (dwFlags & WMIDB_FLAG_LOOKAHEAD ||
|
||
|
|
(riid != IID_IWmiDbHandle &&
|
||
|
|
riid != IID_IWbemClassObject &&
|
||
|
|
riid != IID__IWmiObject))
|
||
|
|
/// UuidCompare(pIIDRequestedInterface, &IID_IWmiDbHandle, NULL) ||
|
||
|
|
// UuidCompare(pIIDRequestedInterface, &IID_IWbemClassObject, NULL))
|
||
|
|
return WBEM_E_NOT_SUPPORTED;
|
||
|
|
|
||
|
|
if (pdwNumReturned)
|
||
|
|
*pdwNumReturned = 0;
|
||
|
|
|
||
|
|
if (!m_pStatus && !m_pRowset)
|
||
|
|
return WBEM_S_NO_MORE_DATA;
|
||
|
|
|
||
|
|
// For each ObjectId, do we instantiate a new handle,
|
||
|
|
// and increment a background ref count on the object itself?
|
||
|
|
// How do we keep track of the handles that are in use??
|
||
|
|
|
||
|
|
try
|
||
|
|
{
|
||
|
|
|
||
|
|
HROW *pRow = NULL;
|
||
|
|
VARIANT vTemp;
|
||
|
|
VariantInit(&vTemp);
|
||
|
|
int iNumRetrieved = 0;
|
||
|
|
IRowset *pIRowset = NULL;
|
||
|
|
|
||
|
|
if (m_pStatus)
|
||
|
|
{
|
||
|
|
hr = CSQLExecute::IsDataReady(m_pStatus);
|
||
|
|
|
||
|
|
// TO DO: Wait if we are pending. Fail for now.
|
||
|
|
|
||
|
|
if (SUCCEEDED(hr))
|
||
|
|
{
|
||
|
|
hr = m_pStatus->QueryInterface(IID_IRowset, (void **)&pIRowset);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else
|
||
|
|
pIRowset = m_pRowset;
|
||
|
|
|
||
|
|
if (SUCCEEDED(hr) && pIRowset)
|
||
|
|
{
|
||
|
|
// TO DO: Take the timeout value into consideration!!!
|
||
|
|
|
||
|
|
hr = CSQLExecute::GetColumnValue(pIRowset, 1, m_pIMalloc, &pRow, vTemp);
|
||
|
|
while (SUCCEEDED(hr) && hr != WBEM_S_NO_MORE_DATA && iNumRetrieved < dwNumRequested)
|
||
|
|
{
|
||
|
|
|
||
|
|
if (!m_pSession || !(m_pSession->m_pController) ||
|
||
|
|
((CWmiDbController *)m_pSession->m_pController)->m_dwCurrentStatus == WBEM_E_SHUTTING_DOWN)
|
||
|
|
{
|
||
|
|
hrRet = WBEM_E_SHUTTING_DOWN;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
// At this point, we need to check the cache
|
||
|
|
// to make sure we don't already have one of these,
|
||
|
|
// and that its not locked.
|
||
|
|
|
||
|
|
SQL_ID dID = 0;
|
||
|
|
|
||
|
|
if (vTemp.vt == VT_BSTR)
|
||
|
|
dID = _wtoi64(vTemp.bstrVal);
|
||
|
|
else if (vTemp.vt == VT_I4)
|
||
|
|
dID = vTemp.lVal;
|
||
|
|
|
||
|
|
SQL_ID dClassID = 0, dScopeID = 0;
|
||
|
|
DWORD dwLockType = 0;
|
||
|
|
|
||
|
|
VariantClear(&vTemp);
|
||
|
|
hr = CSQLExecute::GetColumnValue(pIRowset, 2, m_pIMalloc, &pRow, vTemp);
|
||
|
|
if (vTemp.vt == VT_BSTR)
|
||
|
|
dClassID = _wtoi64(vTemp.bstrVal);
|
||
|
|
else if (vTemp.vt == VT_I4)
|
||
|
|
dClassID = vTemp.lVal;
|
||
|
|
|
||
|
|
VariantClear(&vTemp);
|
||
|
|
hr = CSQLExecute::GetColumnValue(pIRowset, 3, m_pIMalloc, &pRow, vTemp);
|
||
|
|
|
||
|
|
if (vTemp.vt == VT_BSTR)
|
||
|
|
dScopeID = _wtoi64(vTemp.bstrVal);
|
||
|
|
else if (vTemp.vt == VT_I4)
|
||
|
|
dScopeID = vTemp.lVal;
|
||
|
|
|
||
|
|
VariantClear(&vTemp);
|
||
|
|
|
||
|
|
//hr = ((CWmiDbSession *)m_pSession)->VerifyObjectSecurity(NULL, dID, dClassID, dScopeID, 0, WBEM_ENABLE);
|
||
|
|
if (SUCCEEDED(hr))
|
||
|
|
{
|
||
|
|
CWmiDbHandle *pTemp = new CWmiDbHandle;
|
||
|
|
if (pTemp)
|
||
|
|
{
|
||
|
|
DWORD dwVersion = 0;
|
||
|
|
// Obtain a lock for this object
|
||
|
|
// =============================
|
||
|
|
pTemp->m_pSession = m_pSession;
|
||
|
|
|
||
|
|
hr = ((CWmiDbController *)m_pSession->m_pController)->LockCache.AddLock(bImmediate, dID, dwRequestedHandleType, pTemp,
|
||
|
|
dScopeID, dClassID, &((CWmiDbController *)m_pSession->m_pController)->SchemaCache, false,
|
||
|
|
0, 0, &dwVersion);
|
||
|
|
|
||
|
|
if (FAILED(hr))
|
||
|
|
{
|
||
|
|
delete pTemp;
|
||
|
|
// If they failed to get a handle, what do we do?
|
||
|
|
// Ignore it and continue, I guess.
|
||
|
|
hrRet = WBEM_S_PARTIAL_RESULTS;
|
||
|
|
ppObjects[iNumRetrieved] = NULL;
|
||
|
|
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
((CWmiDbController *)m_pSession->m_pController)->AddHandle();
|
||
|
|
pTemp->AddRef();
|
||
|
|
pTemp->m_dwVersion = dwVersion;
|
||
|
|
pTemp->m_dwHandleType = dwRequestedHandleType;
|
||
|
|
pTemp->m_dClassId = dClassID;
|
||
|
|
pTemp->m_dObjectId = dID;
|
||
|
|
if (dwFlags & WBEM_FLAG_USE_SECURITY_DESCRIPTOR)
|
||
|
|
pTemp->m_bSecDesc = TRUE;
|
||
|
|
|
||
|
|
if (pTemp->m_dClassId == MAPPEDNSCLASSID)
|
||
|
|
pTemp->m_bDefault = FALSE;
|
||
|
|
|
||
|
|
pTemp->m_dScopeId = dScopeID;
|
||
|
|
|
||
|
|
if (riid == IID_IWmiDbHandle)
|
||
|
|
{
|
||
|
|
ppObjects[iNumRetrieved] = pTemp;
|
||
|
|
iNumRetrieved++;
|
||
|
|
}
|
||
|
|
else if (riid == IID_IWbemClassObject ||
|
||
|
|
riid == IID__IWmiObject)
|
||
|
|
{
|
||
|
|
IWbemClassObject *pTemp2 = NULL;
|
||
|
|
hr = pTemp->QueryInterface(IID_IWbemClassObject, (void **)&pTemp2);
|
||
|
|
ppObjects[iNumRetrieved] = pTemp2;
|
||
|
|
if (FAILED(hr))
|
||
|
|
hrRet = WBEM_S_PARTIAL_RESULTS;
|
||
|
|
else
|
||
|
|
iNumRetrieved++;
|
||
|
|
pTemp->Release();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
// *pQueryResult = NULL; // What do we do here? Cancel, I assume.
|
||
|
|
hrRet = WBEM_E_OUT_OF_MEMORY;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
else
|
||
|
|
hrRet = WBEM_S_PARTIAL_RESULTS;
|
||
|
|
|
||
|
|
if (m_pSession && ((CWmiDbSession *)m_pSession)->m_pController)
|
||
|
|
((CWmiDbController *)m_pSession->m_pController)->IncrementHitCount(false);
|
||
|
|
|
||
|
|
VariantClear(&vTemp);
|
||
|
|
|
||
|
|
hr = pIRowset->ReleaseRows(1, pRow, NULL, NULL, NULL);
|
||
|
|
delete pRow;
|
||
|
|
pRow = NULL;
|
||
|
|
if (iNumRetrieved == dwNumRequested)
|
||
|
|
break;
|
||
|
|
hr = CSQLExecute::GetColumnValue(pIRowset, 1, m_pIMalloc, &pRow, vTemp);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if (pdwNumReturned)
|
||
|
|
*pdwNumReturned = iNumRetrieved;
|
||
|
|
|
||
|
|
// Null out m_pStatus if there are no more results!!!
|
||
|
|
if (hr == WBEM_S_NO_MORE_DATA)
|
||
|
|
{
|
||
|
|
hrRet = WBEM_S_NO_MORE_DATA;
|
||
|
|
Cancel(0);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
catch (...)
|
||
|
|
{
|
||
|
|
hrRet = WBEM_E_CRITICAL_ERROR;
|
||
|
|
}
|
||
|
|
|
||
|
|
return hrRet;
|
||
|
|
}
|
||
|
|
|