187 lines
5 KiB
C++
187 lines
5 KiB
C++
|
//+---------------------------------------------------------------------------
|
||
|
|
//
|
||
|
|
// Microsoft Windows
|
||
|
|
// Copyright (C) Microsoft Corporation, 1992 - 1997.
|
||
|
|
//
|
||
|
|
// File: session.hxx
|
||
|
|
//
|
||
|
|
// Contents: TSession interfaces.
|
||
|
|
//
|
||
|
|
// History: 3-30-97 MohamedN Created
|
||
|
|
//
|
||
|
|
//----------------------------------------------------------------------------
|
||
|
|
|
||
|
|
#pragma once
|
||
|
|
|
||
|
|
#include "datasrc.hxx"
|
||
|
|
#include <impiunk.hxx>
|
||
|
|
#include <mparser.h>
|
||
|
|
|
||
|
|
|
||
|
|
//+---------------------------------------------------------------------------
|
||
|
|
//
|
||
|
|
// Class: CDBSession
|
||
|
|
//
|
||
|
|
// Purpose: exposes ole-db session interfaces.
|
||
|
|
//
|
||
|
|
// History: 3-30-97 MohamedN Created
|
||
|
|
//
|
||
|
|
// Notes:
|
||
|
|
//
|
||
|
|
//----------------------------------------------------------------------------
|
||
|
|
|
||
|
|
|
||
|
|
class CDBSession : public IOpenRowset,
|
||
|
|
public IGetDataSource,
|
||
|
|
public ISessionProperties,
|
||
|
|
public IDBCreateCommand
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
|
||
|
|
CDBSession( CDataSrc & dataSrc,
|
||
|
|
IUnknown * pUnkOuter,
|
||
|
|
IUnknown ** ppUnkInner,
|
||
|
|
IParserSession* pIPSession,
|
||
|
|
HANDLE hToken );
|
||
|
|
|
||
|
|
~CDBSession(void);
|
||
|
|
|
||
|
|
|
||
|
|
//
|
||
|
|
// IUnknown methods
|
||
|
|
//
|
||
|
|
STDMETHOD(QueryInterface) ( THIS_ REFIID riid, LPVOID *ppiuk )
|
||
|
|
{
|
||
|
|
// Delegate to outer unknown if aggregated
|
||
|
|
return _pUnkOuter->QueryInterface( riid, ppiuk );
|
||
|
|
}
|
||
|
|
|
||
|
|
STDMETHOD(RealQueryInterface) (THIS_ REFIID riid, LPVOID *ppiuk );
|
||
|
|
|
||
|
|
STDMETHOD_(ULONG, AddRef) (THIS) { return _pUnkOuter->AddRef(); }
|
||
|
|
|
||
|
|
STDMETHOD_(ULONG, Release) (THIS) { return _pUnkOuter->Release(); }
|
||
|
|
|
||
|
|
//
|
||
|
|
// IGetDataSource methods
|
||
|
|
//
|
||
|
|
|
||
|
|
STDMETHOD (GetDataSource) (THIS_ REFIID riid, IUnknown **ppDataSource );
|
||
|
|
|
||
|
|
|
||
|
|
//
|
||
|
|
// ISessionProperties methods
|
||
|
|
//
|
||
|
|
|
||
|
|
STDMETHOD (GetProperties) (THIS_
|
||
|
|
ULONG cPropertySets,
|
||
|
|
const DBPROPIDSET rgPropertySets[],
|
||
|
|
ULONG* pcProperties,
|
||
|
|
DBPROPSET** prgProperties);
|
||
|
|
|
||
|
|
STDMETHOD (SetProperties) (THIS_
|
||
|
|
ULONG cProperties,
|
||
|
|
DBPROPSET rgProperties[]);
|
||
|
|
|
||
|
|
//
|
||
|
|
// ICreateCommand methods
|
||
|
|
//
|
||
|
|
|
||
|
|
STDMETHOD (CreateCommand) (THIS_ IUnknown*, REFIID, IUnknown **);
|
||
|
|
|
||
|
|
//
|
||
|
|
// IOpenRowset methods
|
||
|
|
//
|
||
|
|
|
||
|
|
STDMETHODIMP OpenRowset (IUnknown * pUnkOuter,
|
||
|
|
DBID * pTableID,
|
||
|
|
DBID * pIndexID,
|
||
|
|
REFIID riid,
|
||
|
|
ULONG cPropertySets,
|
||
|
|
DBPROPSET rgPropertySets[],
|
||
|
|
IUnknown ** ppRowset);
|
||
|
|
//
|
||
|
|
// Access Functions
|
||
|
|
//
|
||
|
|
IParserSession * GetParserSession() { return _xIPSession.GetPointer(); }
|
||
|
|
CDataSrc * GetDataSrcPtr() { return &_dataSrc; }
|
||
|
|
IUnknown * GetOuterUnk() { return _pUnkOuter; }
|
||
|
|
|
||
|
|
HANDLE GetLogonToken() { return _xSessionToken.Get(); }
|
||
|
|
|
||
|
|
private:
|
||
|
|
|
||
|
|
LPUNKNOWN _pUnkOuter;
|
||
|
|
CImpIUnknown<CDBSession> _impIUnknown;
|
||
|
|
|
||
|
|
CDataSrc & _dataSrc;
|
||
|
|
|
||
|
|
CMutexSem _mtxSess;
|
||
|
|
|
||
|
|
CMSessProps _UtlProps;
|
||
|
|
|
||
|
|
|
||
|
|
//
|
||
|
|
// SQL Text Parser support
|
||
|
|
//
|
||
|
|
XInterface<IParserSession> _xIPSession;
|
||
|
|
|
||
|
|
//
|
||
|
|
// IOpenRowset helper
|
||
|
|
//
|
||
|
|
static SCODE SetOpenRowsetProperties(
|
||
|
|
ICommandText* pICmdText,
|
||
|
|
ULONG cPropertySets,
|
||
|
|
DBPROPSET rgPropertySets[]);
|
||
|
|
|
||
|
|
static SCODE MarkOpenRowsetProperties(
|
||
|
|
ICommandText* pICmdText,
|
||
|
|
ULONG cPropertySets,
|
||
|
|
DBPROPSET rgPropertySets[]
|
||
|
|
);
|
||
|
|
|
||
|
|
static void MarkPropInError(
|
||
|
|
ULONG cPropertySets,
|
||
|
|
DBPROPSET * rgPropertySets,
|
||
|
|
GUID * pguidPropSet,
|
||
|
|
DBPROP * pProp );
|
||
|
|
//
|
||
|
|
// ISupportErrorInfo
|
||
|
|
//
|
||
|
|
CCIOleDBError _ErrorInfo;
|
||
|
|
SHandle _xSessionToken;
|
||
|
|
};
|
||
|
|
|
||
|
|
//+---------------------------------------------------------------------------
|
||
|
|
//
|
||
|
|
// Class: CImpersonateSessionUser
|
||
|
|
//
|
||
|
|
// Purpose:
|
||
|
|
//
|
||
|
|
// History: 01-24-99 danleg Created
|
||
|
|
//
|
||
|
|
// Notes:
|
||
|
|
//
|
||
|
|
//----------------------------------------------------------------------------
|
||
|
|
|
||
|
|
class CImpersonateSessionUser
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
|
||
|
|
CImpersonateSessionUser( HANDLE hToken = INVALID_HANDLE_VALUE );
|
||
|
|
~CImpersonateSessionUser();
|
||
|
|
|
||
|
|
void Revert();
|
||
|
|
|
||
|
|
private:
|
||
|
|
|
||
|
|
void CachePrevToken();
|
||
|
|
void Impersonate();
|
||
|
|
HANDLE DupToken( HANDLE hToken );
|
||
|
|
|
||
|
|
BOOL _fImpersonated;
|
||
|
|
|
||
|
|
SHandle _xPrevToken;
|
||
|
|
SHandle _xSessionToken;
|
||
|
|
};
|