246 lines
9.4 KiB
C++
246 lines
9.4 KiB
C++
|
|
||
|
|
// Row.h : Declaration of the CRow
|
||
|
|
|
||
|
|
#if (!defined(BUILD_FOR_NT40))
|
||
|
|
#ifndef __ROW_H_
|
||
|
|
#define __ROW_H_
|
||
|
|
|
||
|
|
#define DBGUID_ROWURL {0x0C733AB6L,0x2A1C,0x11CE,{0xAD,0xE5,0x00,0xAA,0x00,0x44,0x77,0x3D}}
|
||
|
|
#define NT_SEC_DESC_ATTR L"ntsecuritydescriptor"
|
||
|
|
|
||
|
|
extern const OLEDBDECLSPEC DBID DBROWCOL_ROWURL = {DBGUID_ROWURL, DBKIND_GUID_PROPID, (LPOLESTR)0};
|
||
|
|
|
||
|
|
/////////////////////////////////////////////////////////////////////////////
|
||
|
|
// CRow
|
||
|
|
//
|
||
|
|
class ATL_NO_VTABLE CRow :
|
||
|
|
INHERIT_TRACKING,
|
||
|
|
public CComObjectRootEx<CComMultiThreadModel>,
|
||
|
|
public CComCoClass<CRow, &CLSID_Row>,
|
||
|
|
public ISupportErrorInfo,
|
||
|
|
public IRow,
|
||
|
|
public IColumnsInfo2,
|
||
|
|
public IConvertType,
|
||
|
|
public IGetSession
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
CRow()
|
||
|
|
{
|
||
|
|
m_pUnkMarshaler = NULL;
|
||
|
|
|
||
|
|
m_pSession = NULL;
|
||
|
|
m_pSourceRowset = NULL;
|
||
|
|
m_hRow = DB_NULL_HROW;
|
||
|
|
m_cMaxColumns = -1;
|
||
|
|
m_pAttrInfo = NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
DECLARE_GET_CONTROLLING_UNKNOWN()
|
||
|
|
|
||
|
|
BEGIN_COM_MAP(CRow)
|
||
|
|
COM_INTERFACE_ENTRY(IRow)
|
||
|
|
COM_INTERFACE_ENTRY(IColumnsInfo2)
|
||
|
|
COM_INTERFACE_ENTRY(IColumnsInfo)
|
||
|
|
COM_INTERFACE_ENTRY(IGetSession)
|
||
|
|
COM_INTERFACE_ENTRY(IConvertType)
|
||
|
|
COM_INTERFACE_ENTRY_AGGREGATE(IID_IMarshal, m_pUnkMarshaler.p)
|
||
|
|
END_COM_MAP()
|
||
|
|
|
||
|
|
HRESULT FinalConstruct()
|
||
|
|
{
|
||
|
|
m_cSourceRowsetColumns = 0;
|
||
|
|
m_pSourceRowsetColumnInfo = NULL;
|
||
|
|
m_pSourceRowsetStringsBuffer = NULL;
|
||
|
|
m_pRowProvider = NULL;
|
||
|
|
|
||
|
|
RRETURN( CoCreateFreeThreadedMarshaler(
|
||
|
|
GetControllingUnknown(), &m_pUnkMarshaler.p));
|
||
|
|
}
|
||
|
|
|
||
|
|
void FinalRelease()
|
||
|
|
{
|
||
|
|
//Release Free Threaded Marshaler.
|
||
|
|
m_pUnkMarshaler.Release();
|
||
|
|
|
||
|
|
//If this is a row in a rowset, free rowset
|
||
|
|
//column info and release the row handle.
|
||
|
|
if (m_pSourceRowset)
|
||
|
|
{
|
||
|
|
if (m_cSourceRowsetColumns)
|
||
|
|
{
|
||
|
|
CoTaskMemFree(m_pSourceRowsetColumnInfo);
|
||
|
|
CoTaskMemFree(m_pSourceRowsetStringsBuffer);
|
||
|
|
}
|
||
|
|
auto_rel<IRowset> pRowset;
|
||
|
|
HRESULT hr = m_pSourceRowset->QueryInterface(__uuidof(IRowset),
|
||
|
|
(void **)&pRowset);
|
||
|
|
if (FAILED(hr) || !pRowset)
|
||
|
|
return;
|
||
|
|
pRowset->ReleaseRows(1, &m_hRow, NULL, NULL, NULL);
|
||
|
|
m_pRowProvider = NULL;
|
||
|
|
}
|
||
|
|
if(m_pAttrInfo)
|
||
|
|
FreeADsMem(m_pAttrInfo);
|
||
|
|
}
|
||
|
|
|
||
|
|
CComPtr<IUnknown> m_pUnkMarshaler;
|
||
|
|
|
||
|
|
// ISupportsErrorInfo
|
||
|
|
STDMETHOD(InterfaceSupportsErrorInfo)(REFIID riid);
|
||
|
|
|
||
|
|
// IRow
|
||
|
|
STDMETHOD(GetColumns)(
|
||
|
|
/* [in] */ DBORDINAL cColumns,
|
||
|
|
/* [size_is][out][in] */ DBCOLUMNACCESS rgColumns[ ]);
|
||
|
|
|
||
|
|
|
||
|
|
STDMETHOD(GetSourceRowset)(
|
||
|
|
/* [in] */ REFIID riid,
|
||
|
|
/* [iid_is][out] */ IUnknown __RPC_FAR *__RPC_FAR *ppRowset,
|
||
|
|
/* [out] */ HROW *phRow);
|
||
|
|
|
||
|
|
STDMETHOD(Open)(
|
||
|
|
/* [unique][in] */ IUnknown *pUnkOuter,
|
||
|
|
/* [in] */ DBID *pColumnID,
|
||
|
|
/* [in] */ REFGUID rguidColumnType,
|
||
|
|
/* [in] */ DWORD dwBindFlags,
|
||
|
|
/* [in] */ REFIID riid,
|
||
|
|
/* [iid_is][out] */ IUnknown **ppUnk);
|
||
|
|
|
||
|
|
//IColumnsInfo2 : IColumnsInfo
|
||
|
|
STDMETHOD(GetColumnInfo)(
|
||
|
|
/* [out][in] */ DBORDINAL *pcColumns,
|
||
|
|
/* [size_is][size_is][out] */ DBCOLUMNINFO **prgInfo,
|
||
|
|
/* [out] */ OLECHAR **ppStringsBuffer);
|
||
|
|
|
||
|
|
STDMETHOD(MapColumnIDs)(
|
||
|
|
/* [in] */ DBORDINAL cColumnIDs,
|
||
|
|
/* [size_is][in] */ const DBID rgColumnIDs[ ],
|
||
|
|
/* [size_is][out] */ DBORDINAL rgColumns[ ]);
|
||
|
|
|
||
|
|
STDMETHOD(GetRestrictedColumnInfo)(
|
||
|
|
/* [in] */ DBORDINAL cColumnIDMasks,
|
||
|
|
/* [in] */ const DBID rgColumnIDMasks[ ],
|
||
|
|
/* [in] */ DWORD dwFlags,
|
||
|
|
/* [out][in] */ DBORDINAL *pcColumns,
|
||
|
|
/* [size_is][size_is][out] */ DBID **prgColumnIDs,
|
||
|
|
/* [size_is][size_is][out] */ DBCOLUMNINFO **prgColumnInfo,
|
||
|
|
/* [out] */ OLECHAR **ppStringsBuffer);
|
||
|
|
|
||
|
|
//IConvertType
|
||
|
|
STDMETHOD(CanConvert)(
|
||
|
|
/* [in] */ DBTYPE wFromType,
|
||
|
|
/* [in] */ DBTYPE wToType,
|
||
|
|
/* [in] */ DBCONVERTFLAGS dwConvertFlags);
|
||
|
|
|
||
|
|
//IGetSession
|
||
|
|
STDMETHOD(GetSession)(
|
||
|
|
REFIID riid,
|
||
|
|
IUnknown **ppunkSession);
|
||
|
|
|
||
|
|
//Internal methods
|
||
|
|
|
||
|
|
//public
|
||
|
|
public:
|
||
|
|
//Initializes a standalone CRow object.
|
||
|
|
STDMETHOD(Initialize)( PWSTR pwszURL,
|
||
|
|
IUnknown *pUnkSession,
|
||
|
|
IAuthenticate* pAuthenticate,
|
||
|
|
DWORD dwBindFlags,
|
||
|
|
BOOL fIsTearOff,
|
||
|
|
BOOL fGetColInfoFromRowset,
|
||
|
|
CCredentials *pSessCreds,
|
||
|
|
bool fBind = TRUE
|
||
|
|
);
|
||
|
|
|
||
|
|
//Initializes a CRow object that represents a Row in a Rowset.
|
||
|
|
STDMETHOD(Initialize) ( PWSTR pwszURL,
|
||
|
|
IUnknown* pUnkSession,
|
||
|
|
IUnknown* pUnkSourceRowset,
|
||
|
|
HROW hRow,
|
||
|
|
PWSTR pwszUserName,
|
||
|
|
PWSTR pwszPassword,
|
||
|
|
DWORD dwBindFlags,
|
||
|
|
BOOL fIsTearOff,
|
||
|
|
BOOL fGetColInfoFromRowset,
|
||
|
|
CRowProvider *pRowProvider
|
||
|
|
);
|
||
|
|
|
||
|
|
public:
|
||
|
|
|
||
|
|
private:
|
||
|
|
//Internal methods
|
||
|
|
|
||
|
|
//Function to check if a given column name
|
||
|
|
//matches one of the source rowset's columns.
|
||
|
|
//If a match is found, it returns the index
|
||
|
|
//into the source rowset column array.
|
||
|
|
int IsSourceRowsetColumn(PWCHAR pwszColumnName);
|
||
|
|
|
||
|
|
//Functions to check if a given column name matches
|
||
|
|
//Column ID Mask criteria.
|
||
|
|
bool fMatchesMaskCriteria(PWCHAR pwszColumnName,
|
||
|
|
ULONG cColumnIDMasks,
|
||
|
|
const DBID rgColumnIDMasks[ ]);
|
||
|
|
|
||
|
|
bool fMatchesMaskCriteria(DBID columnid,
|
||
|
|
ULONG cColumnIDMasks,
|
||
|
|
const DBID rgColumnIDMasks[ ]);
|
||
|
|
|
||
|
|
//Function to check if a property is multi-valued.
|
||
|
|
HRESULT GetSchemaAttributes(
|
||
|
|
PWCHAR pwszColumnName,
|
||
|
|
VARIANT_BOOL *pfMultiValued,
|
||
|
|
long *plMaxRange
|
||
|
|
);
|
||
|
|
|
||
|
|
//Function to get size and DBTYPE of a property
|
||
|
|
HRESULT GetTypeAndSize(
|
||
|
|
ADSTYPE dwADsType,
|
||
|
|
CComBSTR& bstrPropName,
|
||
|
|
DBTYPE* pdbType,
|
||
|
|
ULONG* pulSise
|
||
|
|
);
|
||
|
|
|
||
|
|
//Function to get the root of schema tree eg. "LDAP://Schema"
|
||
|
|
//This function gets this from m_pADsobj using ADSI functions
|
||
|
|
//and stores the string in m_bstrSchemaRoot.
|
||
|
|
HRESULT GetSchemaRoot();
|
||
|
|
|
||
|
|
//Function to get columns from source rowset if this is a tear-off row.
|
||
|
|
HRESULT GetSourceRowsetColumns(
|
||
|
|
ULONG cColumns,
|
||
|
|
DBCOLUMNACCESS rgColumns[ ],
|
||
|
|
ULONG *pcErrors
|
||
|
|
);
|
||
|
|
|
||
|
|
//Function to get the status from HRESULT
|
||
|
|
DBSTATUS StatusFromHRESULT(HRESULT hr);
|
||
|
|
|
||
|
|
HRESULT GetSecurityDescriptor(DBCOLUMNACCESS *pColumn,
|
||
|
|
BOOL fMultiValued);
|
||
|
|
|
||
|
|
int IgnorecbMaxLen(DBTYPE wType);
|
||
|
|
|
||
|
|
auto_rel<IADs> m_pADsObj;
|
||
|
|
CCredentials m_objCredentials;
|
||
|
|
CComBSTR m_objURL;
|
||
|
|
CComBSTR m_bstrSchemaRoot;
|
||
|
|
auto_cs m_autocs;
|
||
|
|
auto_rel<IUnknown> m_pSession;
|
||
|
|
auto_rel<IUnknown> m_pSourceRowset;
|
||
|
|
HROW m_hRow;
|
||
|
|
DWORD m_dwBindFlags;
|
||
|
|
DBORDINAL m_cSourceRowsetColumns;
|
||
|
|
DBCOLUMNINFO *m_pSourceRowsetColumnInfo;
|
||
|
|
OLECHAR *m_pSourceRowsetStringsBuffer;
|
||
|
|
CRowProvider *m_pRowProvider;
|
||
|
|
BOOL m_fIsTearOff;
|
||
|
|
BOOL m_fGetColInfoFromRowset;
|
||
|
|
PADS_ATTR_INFO m_pAttrInfo;
|
||
|
|
LONG m_cMaxColumns;
|
||
|
|
};
|
||
|
|
|
||
|
|
#endif //__ROW_H_
|
||
|
|
#endif
|