Source code of Windows XP (NT5)
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

//-----------------------------------------------------------------------------
//
// 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
}