149 lines
7.7 KiB
C
149 lines
7.7 KiB
C
|
//***************************************************************************
|
||
|
|
//
|
||
|
|
// (c) 1999-2001 by Microsoft Corp. All Rights Reserved.
|
||
|
|
//
|
||
|
|
// sqlexec.h
|
||
|
|
//
|
||
|
|
// cvadai 6-May-1999 created.
|
||
|
|
//
|
||
|
|
//***************************************************************************
|
||
|
|
|
||
|
|
#ifndef _SQLEXEC_H_
|
||
|
|
#define _SQLEXEC_H_
|
||
|
|
|
||
|
|
// This file defines a class that maintains a cache of
|
||
|
|
// SQL connections. Each instance of this class only
|
||
|
|
// contains one set of logon credentials.
|
||
|
|
// Each controller by default is only entitled to
|
||
|
|
// a maximum of 20 connections.
|
||
|
|
// ===================================================
|
||
|
|
|
||
|
|
#define SQL_STRING_LIMIT 3970
|
||
|
|
#define REPDRVR_NAME_LIMIT REPDRVR_MAX_LONG_STRING_SIZE
|
||
|
|
#define REPDRVR_PATH_LIMIT 450
|
||
|
|
|
||
|
|
#include <std.h>
|
||
|
|
#include <comutil.h>
|
||
|
|
#include <oledb.h>
|
||
|
|
#include <oledberr.h>
|
||
|
|
#include <seqstream.h>
|
||
|
|
#include <objcache.h>
|
||
|
|
#include <sqlcache.h>
|
||
|
|
#include <wbemint.h>
|
||
|
|
#include <repcache.h>
|
||
|
|
#include <coresvc.h>
|
||
|
|
|
||
|
|
//*******************************************************
|
||
|
|
//
|
||
|
|
// CSQLExecute
|
||
|
|
//
|
||
|
|
//*******************************************************
|
||
|
|
|
||
|
|
// This needs to deal with blob data,
|
||
|
|
// and support batching.
|
||
|
|
|
||
|
|
class _declspec( dllexport ) CSQLExecute
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
|
||
|
|
static HRESULT ExecuteQuery(IDBCreateCommand *pCmd, LPCWSTR lpSQL, IRowset **ppIRowset = NULL, DWORD *dwNumRows = NULL, ...);
|
||
|
|
static HRESULT ExecuteQuery(IDBInitialize *pDBInit, LPCWSTR lpSQL, IRowset **ppIRowset = NULL, DWORD *dwNumRows = NULL);
|
||
|
|
|
||
|
|
static HRESULT ExecuteQueryAsync(IDBInitialize *pDBInit, LPCWSTR lpSQL, IDBAsynchStatus **ppIAsync, DWORD *dwNumRows = NULL);
|
||
|
|
|
||
|
|
static HRESULT IsDataReady(IDBAsynchStatus *pIAsync);
|
||
|
|
static HRESULT CancelQuery(IDBAsynchStatus *pIAsync);
|
||
|
|
|
||
|
|
static HRESULT GetNumColumns(IRowset *pIRowset, IMalloc *pMalloc, int &iNumCols);
|
||
|
|
static HRESULT GetDataType(IRowset *pIRowset, int iPos, IMalloc *pMalloc, DWORD &dwType,
|
||
|
|
DWORD &dwSize, DWORD &dwPrec, DWORD &dwScale, LPWSTR *lpColumnName = NULL);
|
||
|
|
static HRESULT GetColumnValue(IRowset *pIRowset, int iPos, IMalloc *pMalloc, HROW **pRow, VARIANT &vValue, LPWSTR * lpColumnName=NULL);
|
||
|
|
|
||
|
|
static HRESULT ReadImageValue (IRowset *pRowset, int iPos, HROW **pRow, BYTE **pBuffer, DWORD &dwLen);
|
||
|
|
static HRESULT WriteImageValue(IDBCreateCommand *pCmd, LPCWSTR lpSQL, int iPos, BYTE *pValue, DWORD dwLen);
|
||
|
|
|
||
|
|
static HRESULT GetWMIError(IUnknown *pErrorObj);
|
||
|
|
static HRESULT GetWMIError(long ErrorID);
|
||
|
|
static void SetDBNumeric (DB_NUMERIC &Id, SQL_ID ObjId);
|
||
|
|
static void ClearBindingInfo(DBBINDING *binding);
|
||
|
|
static void SetBindingInfo(DBBINDING *binding, ULONG iOrdinal, ULONG uSize, DBPARAMIO io, ULONG maxlen, DBTYPE type, BYTE bPrecision);
|
||
|
|
static void SetParamBindInfo (DBPARAMBINDINFO &BindInfo, LPWSTR pszType, LPWSTR lpName, ULONG uSize, DWORD dwFlags, BYTE bPrecision);
|
||
|
|
|
||
|
|
static void GetInt64 (DB_NUMERIC *pId, wchar_t **ppBuffer);
|
||
|
|
static __int64 GetInt64 (DB_NUMERIC *pId);
|
||
|
|
|
||
|
|
static void SetVariant(DWORD dwType, VARIANT *pValue, BYTE *pData, DWORD dwTargetType);
|
||
|
|
|
||
|
|
protected:
|
||
|
|
|
||
|
|
CSQLExecute(){};
|
||
|
|
~CSQLExecute(){};
|
||
|
|
};
|
||
|
|
|
||
|
|
|
||
|
|
//***************************************************************************
|
||
|
|
//
|
||
|
|
// CSQLExecProcedure - class for executing procedures as efficiently as
|
||
|
|
// possible.
|
||
|
|
//
|
||
|
|
//***************************************************************************
|
||
|
|
|
||
|
|
class _declspec (dllexport) CSQLExecProcedure
|
||
|
|
{
|
||
|
|
typedef std::vector <SQL_ID> SQLIDs;
|
||
|
|
|
||
|
|
public:
|
||
|
|
static HRESULT GetObjectIdByPath (CSQLConnection *pConn, LPCWSTR lpPath, SQL_ID &dObjectId, SQL_ID &dClassId,
|
||
|
|
SQL_ID *dScopeId, BOOL *bDeleted=NULL); // sp_GetInstanceID
|
||
|
|
|
||
|
|
static HRESULT GetHierarchy(CSQLConnection *pConn, SQL_ID dClassId); // sp_GetChildClassList, sp_GetParentList
|
||
|
|
static HRESULT GetNextUnkeyedPath(CSQLConnection *pConn, SQL_ID dClassId, _bstr_t &sNewPath); // sp_GetNextUnkeyedPath
|
||
|
|
static HRESULT GetNextKeyhole(CSQLConnection *pConn, DWORD iPropertyId, SQL_ID &dNewId); // sp_GetNextKeyhole
|
||
|
|
|
||
|
|
static HRESULT DeleteProperty(CSQLConnection *pConn, DWORD iPropertyId); // sp_DeleteClassData
|
||
|
|
static HRESULT DeleteInstanceData (CSQLConnection *pConn, SQL_ID dObjectId, DWORD iPropertyId, DWORD iPos = -1);
|
||
|
|
static HRESULT CheckKeyMigration(CSQLConnection *pConn, LPWSTR lpObjectKey, LPWSTR lpClassName, SQL_ID dClassId, SQL_ID dScopeID,
|
||
|
|
SQL_ID *pIDs, DWORD iNumIDs);
|
||
|
|
static HRESULT NeedsToCheckKeyMigration(BOOL &bCheck);
|
||
|
|
static HRESULT Execute (CSQLConnection *pConn, LPCWSTR lpProcName, CWStringArray &arrValues,
|
||
|
|
IRowset **ppIRowset);
|
||
|
|
static HRESULT EnumerateSubScopes (CSQLConnection *pConn, SQL_ID dScopeId);
|
||
|
|
static HRESULT InsertScopeMap (CSQLConnection *pConn, SQL_ID dScopeId, LPCWSTR lpScopePath, SQL_ID );
|
||
|
|
|
||
|
|
static HRESULT GetClassInfo (CSQLConnection *pConn, SQL_ID dClassId, SQL_ID &dSuperClassId, BYTE **pBuffer, DWORD &dwBuffLen);
|
||
|
|
static HRESULT HasInstances(CSQLConnection *pConn, SQL_ID dClassId, SQL_ID *pDerivedIds, DWORD iNumDerived, BOOL &bInstancesExist);
|
||
|
|
static HRESULT InsertClass (CSQLConnection *pConn, LPCWSTR lpClassName, LPCWSTR lpObjectKey, LPCWSTR lpObjectPath, SQL_ID dScopeID,
|
||
|
|
SQL_ID dParentClassId, SQL_ID dDynasty, DWORD iState, BYTE *pClassBuff, DWORD dwClassBuffLen, DWORD iClassFlags, DWORD iInsertFlags, SQL_ID &dNewId);
|
||
|
|
static HRESULT UpdateClassBlob (CSQLConnection *pConn, SQL_ID dClassId, _IWmiObject *pObj);
|
||
|
|
static HRESULT InsertClassData (CSQLConnection *pConn, IWbemClassObject *pObj, CSchemaCache *pCache,SQL_ID dScopeId, SQL_ID dClassId, LPCWSTR lpPropName,
|
||
|
|
DWORD CIMType, DWORD StorageType,LPCWSTR lpValue, SQL_ID dRefClassId, DWORD iPropID, DWORD iFlags,
|
||
|
|
DWORD iFlavor, BOOL iSkipValid, DWORD &iNewPropId, SQL_ID dOrigClassId=0, BOOL *bIsKey = NULL);
|
||
|
|
static HRESULT InsertBlobData (CSQLConnection *pConn, SQL_ID dClassId, SQL_ID dObjectId, DWORD iPropertyId, BYTE *pImage, DWORD iPos, DWORD dwNumBytes);
|
||
|
|
static HRESULT InsertPropertyBatch (CSQLConnection *pConn, LPCWSTR lpObjectKey, LPCWSTR lpPath, LPCWSTR lpClassName,
|
||
|
|
SQL_ID dClassId, SQL_ID dScopeId, DWORD iFlags, InsertPropValues *pVals, DWORD iNumVals, SQL_ID &dNewObjectId);
|
||
|
|
static HRESULT InsertBatch (CSQLConnection *pConn, SQL_ID dObjectId, SQL_ID dScopeId, SQL_ID dClassId,
|
||
|
|
InsertQfrValues *pVals, DWORD iNumVals=10);
|
||
|
|
static HRESULT ObjectExists (CSQLConnection *pConn,SQL_ID dId, BOOL &bExists, SQL_ID *dClassId, SQL_ID *dScopeId, BOOL bDeletedOK = FALSE);
|
||
|
|
static HRESULT RenameSubscopes (CSQLConnection *pConn, LPWSTR lpOldPath, LPWSTR lpOldKey, LPWSTR lpNewPath, LPWSTR lpNewKey);
|
||
|
|
|
||
|
|
static HRESULT GetSecurityDescriptor(CSQLConnection *pConn, SQL_ID dObjectId,
|
||
|
|
PNTSECURITY_DESCRIPTOR * ppSD, DWORD &dwBuffLen,
|
||
|
|
DWORD dwFlags);
|
||
|
|
static HRESULT EnumerateSecuredChildren(CSQLConnection *pConn, CSchemaCache *pCache, SQL_ID dObjectId, SQL_ID dClassId, SQL_ID dScopeId,
|
||
|
|
SQLIDs &ObjIds, SQLIDs &ClassIds, SQLIDs &ScopeIds);
|
||
|
|
// Special functions for handling ESS
|
||
|
|
|
||
|
|
static HRESULT InsertUncommittedEvent (CSQLConnection *pConn,LPCWSTR lpGUID, LPWSTR lpNamespace, LPWSTR lpClassName, IWbemClassObject *pOldObj,
|
||
|
|
IWbemClassObject *pNewObj, CSchemaCache *pCache);
|
||
|
|
static HRESULT DeleteUncommittedEvents (CSQLConnection *pConn,LPCWSTR lpGUID, CSchemaCache *pCache, CObjectCache *pObjCache);
|
||
|
|
static HRESULT CommitEvents(CSQLConnection *pConn, _IWmiCoreServices *pESS, LPCWSTR lpNamespace,
|
||
|
|
LPCWSTR lpGUID, CSchemaCache *pCache, CObjectCache *pObjCache);
|
||
|
|
|
||
|
|
private:
|
||
|
|
CSQLExecProcedure() {};
|
||
|
|
~CSQLExecProcedure() {};
|
||
|
|
};
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
#endif // _SQLEXEC_H_
|