mirror of https://github.com/tongzx/nt5src
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.
637 lines
17 KiB
637 lines
17 KiB
//-----------------------------------------------------------------------------
|
|
//
|
|
// Microsoft Windows
|
|
// Copyright (C) Microsoft Corporation, 1992 - 1995.
|
|
//
|
|
// File: CRSInfo.cxx
|
|
//
|
|
// Contents: IRowsetInfo and IGetRow methods
|
|
//
|
|
// Functions:
|
|
//
|
|
// Notes:
|
|
//
|
|
//
|
|
// History: 08/30/96 | RenatoB | Created
|
|
//----------------------------------------------------------------------------
|
|
|
|
|
|
//-----------------------------------------------------------------------------
|
|
// @class CRowsetInfo | embedding of Rowset,
|
|
// to give our IrowsetInfo interface
|
|
//
|
|
//
|
|
//-----------------------------------------------------------------------------
|
|
// Includes
|
|
#include "oleds.hxx"
|
|
#include "atl.h"
|
|
#include "row.hxx"
|
|
|
|
//+---------------------------------------------------------------------------
|
|
//
|
|
// Function: CRowsetInfo::QueryInterface
|
|
//
|
|
// Synopsis: @mfunc QueryInterface.
|
|
//
|
|
// Update: This function should never be called as RowsetInfo is now contained
|
|
// in CRowset.
|
|
//
|
|
//-----------------------------------------------------------------------.
|
|
HRESULT
|
|
CRowsetInfo::QueryInterface(
|
|
REFIID riid,
|
|
LPVOID * ppv
|
|
)
|
|
{
|
|
ADsAssert(FALSE);
|
|
|
|
RRETURN( E_INVALIDARG );
|
|
}
|
|
|
|
//+---------------------------------------------------------------------------
|
|
//
|
|
// Function: CRowsetInfo::CRowsetInfo
|
|
//
|
|
// Synopsis: @mfunc Ctor
|
|
//
|
|
// Arguments:
|
|
//
|
|
//
|
|
// Returns: @rdesc NONE
|
|
//
|
|
// Modifies:
|
|
//
|
|
// History: 08/30/96 RenatoB Created
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
CRowsetInfo::CRowsetInfo(
|
|
IUnknown * pUnkOuter, // controlling unknown)
|
|
IUnknown * pParentObject, // RowProvider
|
|
CSessionObject * pCSession, // Session that created rowset
|
|
CCommandObject * pCCommand, // Command object that created rowset
|
|
CRowProvider * pRowProvider // Row provider pointer
|
|
)
|
|
{
|
|
//
|
|
// Asserts
|
|
//
|
|
ADsAssert(pRowProvider);
|
|
|
|
_pUnkOuter = (pUnkOuter == NULL) ? (IRowsetInfo FAR *)this : pUnkOuter;
|
|
_pRowset = NULL;
|
|
_pParentObject = pParentObject;
|
|
_pCSession = pCSession;
|
|
_pCCommand = pCCommand;
|
|
|
|
// AddRef twice - once for command/session object, once for parent object
|
|
if( NULL == _pCCommand )
|
|
{
|
|
_pCSession->AddRef();
|
|
_pCSession->AddRef();
|
|
}
|
|
else if( NULL == _pCSession )
|
|
{
|
|
_pCCommand->AddRef();
|
|
_pCCommand->AddRef();
|
|
}
|
|
else // shouldn't get here
|
|
ADsAssert(FALSE);
|
|
|
|
_pRowProvider = pRowProvider;
|
|
_pRowProvider->AddRef();
|
|
|
|
_pMalloc = NULL;
|
|
|
|
if( _pCCommand != NULL )
|
|
_pCCommand->IncrementOpenRowsets();
|
|
|
|
//this section is for IRowsetInfo and IUnknown methods.
|
|
|
|
InitializeCriticalSection(&_csRowsetInfo);
|
|
}
|
|
|
|
|
|
//+---------------------------------------------------------------------------
|
|
//
|
|
// Function: CRowsetInfo::~CRowsetInfo
|
|
//
|
|
// Synopsis: @mfunc Dtor
|
|
//
|
|
// Arguments:
|
|
//
|
|
//
|
|
// Returns: @rdesc NONE
|
|
//
|
|
// Modifies:
|
|
//
|
|
// History: 08/30/96 RenatoB Created
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
CRowsetInfo::~CRowsetInfo()
|
|
{
|
|
if( _pMalloc != NULL )
|
|
_pMalloc->Release();
|
|
|
|
if( _pCCommand != NULL ) {
|
|
_pCCommand->DecrementOpenRowsets();
|
|
_pCCommand->Release();
|
|
}
|
|
|
|
if( _pCSession != NULL ) {
|
|
_pCSession->Release();
|
|
}
|
|
|
|
if( _pParentObject != NULL ) {
|
|
_pParentObject->Release();
|
|
}
|
|
|
|
if( _pRowProvider != NULL ) {
|
|
_pRowProvider->Release();
|
|
}
|
|
|
|
DeleteCriticalSection(&_csRowsetInfo);
|
|
}
|
|
|
|
//+---------------------------------------------------------------------------
|
|
//
|
|
// Function: CRowsetInfo::FInit
|
|
//
|
|
// Synopsis: @mfunc Initializer
|
|
//
|
|
// Synopsis: @mfunc Initializer
|
|
//
|
|
// Arguments:
|
|
//
|
|
//
|
|
// Returns: @rdesc NONE
|
|
// @flag S_OK | Interface is supported
|
|
// @flag E_OUTOFMEMORY | Not enough memory
|
|
// @flag E_INVALIDARG | One or more arguments are invalid.
|
|
//
|
|
// Modifies:
|
|
//
|
|
// History: 08/30/96 RenatoB Created
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
HRESULT
|
|
CRowsetInfo::FInit(
|
|
IUnknown * pRowset) // rowset interface
|
|
{
|
|
HRESULT hr;
|
|
hr = S_OK;
|
|
|
|
if( FAILED(CoGetMalloc(MEMCTX_TASK, &_pMalloc)) )
|
|
RRETURN( E_OUTOFMEMORY );
|
|
|
|
// Should not AddRef here because RowsetInfo is contained in CRowset
|
|
_pRowset = pRowset;
|
|
|
|
RRETURN( hr );
|
|
}
|
|
|
|
//+---------------------------------------------------------------------------
|
|
//
|
|
// Function: CRowsetInfo::GetReferencedRowset
|
|
//
|
|
// Synopsis: @mfunc Returns an interface pointer to the rowset to which a bookmark applies
|
|
//
|
|
// Called by: Client
|
|
// Called when: Any time
|
|
//
|
|
// Arguments:
|
|
// iOrdinal [in] Bookmark column for which to get related rowset. Must be 0 in this impl.
|
|
// riid [in] IID of the interface pointer to return in *ppReferencedRowset.
|
|
// ppReferencedRowset[out] pointer to Rowset object referenced by Bookmark
|
|
//
|
|
//
|
|
// Returns: @rdesc NONE
|
|
// S_OK | Interface is supported
|
|
// E_INVALIDARG | ppReferencedRowset was a NULL pointer
|
|
// E_FAIL | provider specific error
|
|
// E_NOINTERFACE | Rowset does not support interface
|
|
// DB_E_NOTAREFENCEDCOLUMN | iOrdinal was not 0
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
HRESULT
|
|
CRowsetInfo::GetReferencedRowset(
|
|
DBORDINAL iOrdinal,
|
|
REFIID riid,
|
|
IUnknown ** ppReferencedRowset
|
|
)
|
|
{
|
|
CAutoBlock cab (&_csRowsetInfo);
|
|
|
|
//
|
|
// Asserts
|
|
//
|
|
ADsAssert(_pRowProvider);
|
|
ADsAssert(_pRowset);
|
|
|
|
if( ppReferencedRowset == NULL )
|
|
RRETURN( E_INVALIDARG );
|
|
|
|
*ppReferencedRowset = NULL;
|
|
|
|
if( iOrdinal >= _pRowProvider->GetColumnCount() )
|
|
RRETURN( DB_E_BADORDINAL );
|
|
|
|
if( iOrdinal != 0 )
|
|
RRETURN( DB_E_NOTAREFERENCECOLUMN );
|
|
|
|
RRETURN(_pRowset->QueryInterface(
|
|
riid,
|
|
(void**)ppReferencedRowset)
|
|
);
|
|
};
|
|
|
|
//+---------------------------------------------------------------------------
|
|
//
|
|
// Function: CRowsetInfo::GetSpecificaton
|
|
//
|
|
// Synopsis: @mfunc Returns an interface pointer to the command that created the rowset
|
|
//
|
|
// Called by: Client
|
|
// Called when: Any time
|
|
//
|
|
// Arguments:
|
|
// riid [in] IID of the interface pointer to return in *ppSpecification.
|
|
// ppSpecification [out] pointer to command
|
|
//
|
|
//
|
|
// Returns: @rdesc NONE
|
|
// S_OK | Interface is supported
|
|
// E_INVALIDARG | ppSpecification was a NULL pointer
|
|
// E_FAIL | provider specific error
|
|
// E_NOINTERFACE | Command does not support interface
|
|
// S_FALSE | Rowset does not have command that created it
|
|
//----------------------------------------------------------------------------
|
|
HRESULT
|
|
CRowsetInfo::GetSpecification(
|
|
REFIID riid,
|
|
IUnknown ** ppSpecification
|
|
)
|
|
{
|
|
CAutoBlock cab(&_csRowsetInfo);
|
|
|
|
if( ppSpecification == NULL )
|
|
RRETURN( E_INVALIDARG );
|
|
|
|
*ppSpecification = NULL;
|
|
|
|
if( _pParentObject == NULL )
|
|
RRETURN( S_FALSE );
|
|
|
|
RRETURN( _pParentObject->QueryInterface(riid, (void**)ppSpecification) );
|
|
};
|
|
|
|
//+---------------------------------------------------------------------------
|
|
//
|
|
// Function: CRowsetInfo::GetProperties
|
|
//
|
|
// Synopsis: @mfunc GetProperties
|
|
// Called by: Client.
|
|
// Called when: Any time.
|
|
//
|
|
// Arguments:
|
|
// cPropertyIDSets[in] The number of DBPROPIDSET structures in rgProperttySets
|
|
// rgPropertyIDSets[in] Array of cPropertyIDSets of DBPROIDSET structs.
|
|
// pcPropertySets[out] number of DBPROPSET returned in *prgPropertySets
|
|
// prgPropertySets[out] pointer to array of DBPROPSET structures, having the
|
|
// value and status of requested properties
|
|
//
|
|
// Returns: @rdesc HRESULT
|
|
// S_OK Success
|
|
// DB_S_ERRORSOCCURRED Values were not supported for some properties
|
|
// E_FAIL A provider-specific error occurred
|
|
// E_INVALIDARG cPropertyIDSets > 0 and rgPropertyIdSEts = NULL
|
|
// pcPropertySets or prgPropertySets was NULL pointer
|
|
// In al element of rgPropertyIDSets, cPropertyIDs was not zero
|
|
// and rgPropertyIDs was a Null pointer
|
|
// E_OUTOFMEMORY Provider could not allocate memory
|
|
// DB_E_ERRORSOCCURRED Values were not returned for any properties
|
|
//-----------------------------------------------------------------------.
|
|
HRESULT
|
|
CRowsetInfo::GetProperties(
|
|
const ULONG cPropertyIDSets,
|
|
const DBPROPIDSET rgPropertyIDSets[],
|
|
ULONG * pcPropertySets,
|
|
DBPROPSET ** prgPropertySets
|
|
)
|
|
{
|
|
// should never get here because this call is handled by CRowset
|
|
ADsAssert(FALSE);
|
|
|
|
RRETURN( E_FAIL );
|
|
};
|
|
|
|
//+---------------------------------------------------------------------------
|
|
//
|
|
// Function: CRowsetInfo::GetRowFromHROW
|
|
//
|
|
// Synopsis: @mfunc Returns an interface pointer to a row in the rowset.
|
|
//
|
|
// Called by: Client
|
|
// Called when: Any time
|
|
//
|
|
// Arguments:
|
|
// IUnknown *pUnkOuter Outer unknown.
|
|
// HROW hRow Handle to row.
|
|
// REFIID riid, Interface requested.
|
|
// [out]IUnknown **ppUnk returned interface
|
|
//
|
|
//
|
|
// Returns: HRESULT
|
|
//----------------------------------------------------------------------------
|
|
HRESULT
|
|
CRowsetInfo::GetRowFromHROW(
|
|
IUnknown *pUnkOuter,
|
|
HROW hRow,
|
|
REFIID riid,
|
|
IUnknown * *ppUnk,
|
|
BOOL fIsTearOff,
|
|
BOOL fAllAttrs
|
|
)
|
|
{
|
|
#if (!defined(BUILD_FOR_NT40))
|
|
HRESULT hr = S_OK;
|
|
CAutoBlock cab(&_csRowsetInfo);
|
|
auto_tm<WCHAR> lpwszURL;
|
|
auto_rel<IGetDataSource> pGetDataSource;
|
|
auto_rel<ICommand> pCommand;
|
|
auto_rel<IRowset> pRowset;
|
|
auto_rel<IRowset> pRealRowset;
|
|
CComObject<CRow> *pRow = NULL;
|
|
auto_rel<IRow> pRowDelete;
|
|
CCredentials creds;
|
|
CCredentials *pCreds;
|
|
PWSTR pwszUserID = NULL, pwszPassword = NULL;
|
|
|
|
//Initialize return arguments
|
|
if (ppUnk)
|
|
*ppUnk = NULL;
|
|
|
|
//We currently don't support aggregation.
|
|
if (pUnkOuter != NULL)
|
|
RRETURN(DB_E_NOAGGREGATION);
|
|
|
|
if (ppUnk == NULL)
|
|
RRETURN(E_INVALIDARG);
|
|
|
|
if ( _pParentObject == NULL)
|
|
RRETURN(S_FALSE);
|
|
|
|
TRYBLOCK
|
|
PWSTR pURL;
|
|
ADsAssert(_pRowset != NULL);
|
|
hr = _pRowset->QueryInterface(__uuidof(IRowset), (void **)&pRealRowset);
|
|
BAIL_ON_FAILURE(hr);
|
|
hr = _pRowProvider->GetURLFromHROW( hRow,&pURL, pRealRowset);
|
|
BAIL_ON_FAILURE(hr);
|
|
|
|
//assign returned ptr to auto_tm class for auto deletion.
|
|
lpwszURL = pURL;
|
|
|
|
*ppUnk = NULL;
|
|
|
|
//
|
|
// Get the session and QI for any mandatory interface -
|
|
// we query for IGetDataSource.
|
|
//
|
|
hr = _pParentObject->QueryInterface(
|
|
IID_IGetDataSource,
|
|
(void**)&pGetDataSource);
|
|
|
|
if (FAILED(hr))
|
|
{
|
|
hr = _pParentObject->QueryInterface(
|
|
IID_ICommand,
|
|
(void**)&pCommand);
|
|
BAIL_ON_FAILURE(hr);
|
|
|
|
hr = pCommand->GetDBSession (
|
|
IID_IGetDataSource,
|
|
(LPUNKNOWN *)&pGetDataSource);
|
|
BAIL_ON_FAILURE(hr);
|
|
}
|
|
|
|
//Get user id and password properties for authentication
|
|
//If the RowProvider has cached special credentials, we use them,
|
|
//Otherwise we use the DBPROP_AUTH_USERID and DBPROP_AUTH_PASSWORD
|
|
//properties in the originating DataSource object.
|
|
if ((pCreds = _pRowProvider->GetCredentials()) == NULL)
|
|
{
|
|
hr = GetCredentials(pGetDataSource, creds);
|
|
BAIL_ON_FAILURE(hr);
|
|
pCreds = &creds;
|
|
}
|
|
|
|
hr = pCreds->GetUserName(&pwszUserID);
|
|
BAIL_ON_FAILURE(hr);
|
|
hr = pCreds->GetPassword(&pwszPassword);
|
|
BAIL_ON_FAILURE(hr);
|
|
|
|
//Instantiate a row object.
|
|
hr = CComObject<CRow>::CreateInstance(&pRow);
|
|
if (FAILED(hr))
|
|
BAIL_ON_FAILURE(hr = E_OUTOFMEMORY);
|
|
|
|
//To make sure we delete the row object in case
|
|
//we encounter errors after this point.
|
|
pRow->AddRef();
|
|
pRowDelete = pRow;
|
|
|
|
//Initialize the row object.
|
|
hr = pRow->Initialize( lpwszURL,
|
|
(IUnknown *)pGetDataSource,
|
|
(IUnknown *)_pRowset,
|
|
hRow,
|
|
pwszUserID,
|
|
pwszPassword,
|
|
DBBINDURLFLAG_READ, //we are currently a read-only provider.
|
|
fIsTearOff, // see comment at beginning of row.cxx
|
|
!fAllAttrs, // see comment at beginning of row.cxx
|
|
_pRowProvider
|
|
);
|
|
|
|
if (FAILED(hr))
|
|
{
|
|
if (INVALID_CREDENTIALS_ERROR(hr))
|
|
{
|
|
BAIL_ON_FAILURE(hr = DB_SEC_E_PERMISSIONDENIED);
|
|
}
|
|
else if (hr == E_NOINTERFACE)
|
|
{
|
|
BAIL_ON_FAILURE(hr);
|
|
}
|
|
else
|
|
{
|
|
BAIL_ON_FAILURE(hr = DB_E_NOTFOUND);
|
|
}
|
|
}
|
|
|
|
//Get the requested interface on the Row.
|
|
hr = pRow->QueryInterface(riid, (void**)ppUnk);
|
|
if (FAILED(hr))
|
|
BAIL_ON_FAILURE(hr = E_NOINTERFACE);
|
|
|
|
CATCHBLOCKBAIL(hr)
|
|
|
|
if (pwszUserID)
|
|
FreeADsStr(pwszUserID);
|
|
if (pwszPassword)
|
|
FreeADsStr(pwszPassword);
|
|
|
|
RRETURN(S_OK);
|
|
|
|
error:
|
|
if (pwszUserID)
|
|
FreeADsStr(pwszUserID);
|
|
if (pwszPassword)
|
|
FreeADsStr(pwszPassword);
|
|
|
|
RRETURN(hr);
|
|
|
|
#else
|
|
RRETURN(E_FAIL);
|
|
#endif
|
|
}
|
|
|
|
|
|
|
|
//+---------------------------------------------------------------------------
|
|
//
|
|
// Function: CRowsetInfo::GetURLFromHROW
|
|
//
|
|
// Synopsis: @mfunc Returns an interface pointer to a row in the rowset.
|
|
//
|
|
// Called by: Client
|
|
// Called when: Any time
|
|
//
|
|
// Arguments:
|
|
// HROW hRow Handle to row.
|
|
// LPOLESTR ppwszURL, URL name for the row.
|
|
//
|
|
//
|
|
// Returns: HRESULT
|
|
//----------------------------------------------------------------------------
|
|
HRESULT
|
|
CRowsetInfo::GetURLFromHROW(
|
|
HROW hRow,
|
|
LPOLESTR *ppwszURL
|
|
)
|
|
{
|
|
#if (!defined(BUILD_FOR_NT40))
|
|
CAutoBlock cab(&_csRowsetInfo);
|
|
LPWSTR lpwszURL = NULL;
|
|
auto_rel<IRowset> pRealRowset;
|
|
HRESULT hr = S_OK;
|
|
|
|
if (ppwszURL == NULL)
|
|
RRETURN(E_INVALIDARG);
|
|
|
|
ADsAssert(_pRowset != NULL);
|
|
hr = _pRowset->QueryInterface(__uuidof(IRowset), (void **)&pRealRowset);
|
|
BAIL_ON_FAILURE(hr);
|
|
|
|
// Call row provider to get the URL.
|
|
hr = _pRowProvider->GetURLFromHROW( hRow,ppwszURL, pRealRowset);
|
|
BAIL_ON_FAILURE(hr);
|
|
|
|
error:
|
|
|
|
RRETURN(hr);
|
|
#else
|
|
RRETURN(E_FAIL);
|
|
#endif
|
|
};
|
|
|
|
|
|
|
|
//+---------------------------------------------------------------------------
|
|
//
|
|
// Function: CRowsetInfo::GetCredentials
|
|
//
|
|
// Synopsis: @mfunc Gets credentials stored in INIT properties.
|
|
//
|
|
// Arguments:
|
|
// pSession [in] IGetDataSource interface ptr on Session.
|
|
// refCredentials [in] CCredentials containing user id and password.
|
|
//
|
|
// Returns: HRESULT
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
HRESULT CRowsetInfo::GetCredentials(
|
|
IGetDataSource *pSession,
|
|
CCredentials &refCredentials
|
|
)
|
|
{
|
|
#if (!defined(BUILD_FOR_NT40))
|
|
HRESULT hr = S_OK;
|
|
auto_rel<IDBProperties> pDBProp;
|
|
ULONG i, j, cPropertySets = 0;
|
|
|
|
Assert(pSession);
|
|
|
|
hr = pSession->GetDataSource(
|
|
__uuidof(IDBProperties),
|
|
(IUnknown **)&pDBProp);
|
|
|
|
BAIL_ON_FAILURE(hr);
|
|
|
|
DBPROPID propids[2];
|
|
propids[0] = DBPROP_AUTH_USERID;
|
|
propids[1] = DBPROP_AUTH_PASSWORD;
|
|
|
|
DBPROPIDSET rgPropertyIDSets[1];
|
|
rgPropertyIDSets[0].rgPropertyIDs = propids;
|
|
rgPropertyIDSets[0].cPropertyIDs = 2;
|
|
rgPropertyIDSets[0].guidPropertySet = DBPROPSET_DBINIT;
|
|
|
|
DBPROPSET *prgPropertySets;
|
|
|
|
hr = pDBProp->GetProperties(
|
|
1,
|
|
rgPropertyIDSets,
|
|
&cPropertySets,
|
|
&prgPropertySets);
|
|
|
|
if (hr == DB_E_ERRORSOCCURRED)
|
|
BAIL_ON_FAILURE(hr);
|
|
|
|
for (i = 0; i < cPropertySets; i++)
|
|
{
|
|
for (j = 0; j < prgPropertySets[i].cProperties; j++)
|
|
{
|
|
DBPROP *pProp = &prgPropertySets[i].rgProperties[j];
|
|
Assert(pProp);
|
|
if (pProp->dwStatus == S_OK &&
|
|
pProp->dwPropertyID == DBPROP_AUTH_USERID)
|
|
refCredentials.SetUserName(V_BSTR(&pProp->vValue));
|
|
else if (pProp->dwStatus == S_OK &&
|
|
pProp->dwPropertyID == DBPROP_AUTH_PASSWORD)
|
|
refCredentials.SetPassword(V_BSTR(&pProp->vValue));
|
|
else
|
|
continue;
|
|
}
|
|
}
|
|
|
|
error:
|
|
for (i = 0; i < cPropertySets; i++)
|
|
{
|
|
for (j = 0; j < prgPropertySets[i].cProperties; j++)
|
|
{
|
|
DBPROP *pProp = &(prgPropertySets[i].rgProperties[j]);
|
|
Assert(pProp);
|
|
FreeDBID(&pProp->colid);
|
|
VariantClear(&pProp->vValue);
|
|
}
|
|
|
|
CoTaskMemFree(prgPropertySets[i].rgProperties);
|
|
}
|
|
CoTaskMemFree(prgPropertySets);
|
|
|
|
RRETURN ( hr );
|
|
#else
|
|
RRETURN(E_FAIL);
|
|
#endif
|
|
}
|