Leaked source code of windows server 2003
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.
 
 
 
 
 
 

607 lines
15 KiB

//+--------------------------------------------------------------------------
//
// Microsoft Windows
// Copyright (C) Microsoft Corporation, 1996 - 1999
//
// File: db.h
//
// Contents: Cert Server Database interface implementation
//
//---------------------------------------------------------------------------
#include "resource.h" // main symbols
typedef struct _DBSEEKDATA
{
DWORD SeekFlags; // CST_*
JET_GRBIT grbitSeekRange; // JetSeek flags if CST_SEEKINDEXRANGE
// this is where to seek to retrieve
// end-of-range key
JET_GRBIT grbitInitial; // JetMove or JetSeek flags: set initial cursor
// Where to set the cursor initially
JET_GRBIT grbitRange; // JetSetIndexRange flags if CST_SEEKINDEXRANGE
// other flags to be ingested while setting
// range (bitRange UpperLimit, Inclusive)
} DBSEEKDATA;
// _Create() CreateFlags:
#define CF_DATABASE 0x00000001
#define CF_MISSINGTABLES 0x00000002
#define CF_MISSINGCOLUMNS 0x00000004
#define CF_MISSINGINDEXES 0x00000008
#ifdef DBG_CERTSRV
#define CSASSERTTHREAD(pcs) \
{ \
DWORD dwThreadId = GetCurrentThreadId(); \
if ((pcs)->dwThreadId != dwThreadId) \
{ \
DBGPRINT((DBG_SS_CERTDB, "Session tid=%d, Current tid=%d\n", (pcs)->dwThreadId, dwThreadId)); \
} \
CSASSERT((pcs)->dwThreadId == dwThreadId); \
}
#endif
HRESULT
InitGlobalWriterState(VOID);
HRESULT
UnInitGlobalWriterState(VOID);
VOID
DBFreeParms();
HRESULT
DBInitParms(
IN DWORD cSession,
IN DWORD DBFlags,
OPTIONAL IN WCHAR const *pwszEventSource,
OPTIONAL IN WCHAR const *pwszLogDir,
OPTIONAL IN WCHAR const *pwszSystemDir,
OPTIONAL IN WCHAR const *pwszTempDir,
OUT JET_INSTANCE *pInstance);
#if DBG_CERTSRV
VOID
dbgcat(
IN OUT WCHAR *pwszBuf,
IN WCHAR const *pwszAdd);
WCHAR const *
wszSeekgrbit(
JET_GRBIT grbit);
WCHAR const *
wszMovecrow(
IN LONG cRow);
WCHAR const *
wszSetIndexRangegrbit(
JET_GRBIT grbit);
WCHAR const *
wszMakeKeygrbit(
JET_GRBIT grbit);
WCHAR const *
wszCSFFlags(
IN LONG Flags);
WCHAR const *
wszCSTFlags(
IN LONG Flags);
WCHAR const *
wszSeekOperator(
IN LONG SeekOperator);
WCHAR const *
wszSortOperator(
IN LONG SortOrder);
VOID
dbDumpValue(
IN DWORD dwSubSystemId,
OPTIONAL IN DBTABLE const *pdt,
IN BYTE const *pbValue,
IN DWORD cbValue);
VOID
dbDumpColumn(
IN DWORD dwSubSystemId,
IN DBTABLE const *pdt,
IN BYTE const *pbValue,
IN DWORD cbValue);
#endif // DBG_CERTSRV
class CCertDB:
public ICertDB,
//public ISupportErrorInfoImpl<&IID_ICertDB>,
public CComObjectRoot,
public CComCoClass<CCertDB, &CLSID_CCertDB>
{
public:
CCertDB();
~CCertDB();
BEGIN_COM_MAP(CCertDB)
COM_INTERFACE_ENTRY(ICertDB)
//COM_INTERFACE_ENTRY(ISupportErrorInfo)
END_COM_MAP()
DECLARE_NOT_AGGREGATABLE(CCertDB)
// Remove the comment from the line above if you don't want your object to
// support aggregation. The default is to support it
DECLARE_REGISTRY(
CCertDB,
wszCLASS_CERTDB TEXT(".1"),
wszCLASS_CERTDB,
IDS_CERTDB_DESC,
THREADFLAGS_BOTH)
// ICertDB
public:
STDMETHOD(Open)(
/* [in] */ DWORD Flags,
/* [in] */ DWORD cSession,
/* [in] */ WCHAR const *pwszEventSource,
/* [in] */ WCHAR const *pwszDBFile,
/* [in] */ WCHAR const *pwszLogDir,
/* [in] */ WCHAR const *pwszSystemDir,
/* [in] */ WCHAR const *pwszTempDir);
STDMETHOD(ShutDown)(
/* [in] */ DWORD dwFlags);
STDMETHOD(OpenRow)(
/* [in] */ DWORD dwFlags,
/* [in] */ DWORD RowId,
/* [in] */ WCHAR const *pwszSerialNumberOrCertHash, // OPTIONAL
/* [out] */ ICertDBRow **pprow);
STDMETHOD(OpenView)(
/* [in] */ DWORD ccvr,
/* [in] */ CERTVIEWRESTRICTION const *acvr,
/* [in] */ DWORD ccolOut,
/* [in] */ DWORD const *acolOut,
/* [in] */ DWORD const dwFlags,
/* [out] */ IEnumCERTDBRESULTROW **ppenum);
STDMETHOD(EnumCertDBColumn)(
/* [in] */ DWORD dwTable,
/* [out] */ IEnumCERTDBCOLUMN **ppenum);
STDMETHOD(OpenBackup)(
/* [in] */ LONG grbitJet,
/* [out] */ ICertDBBackup **ppBackup);
STDMETHOD(GetDefaultColumnSet)(
/* [in] */ DWORD iColumnSetDefault,
/* [in] */ DWORD cColumnIds,
/* [out] */ DWORD *pcColumnIds,
/* [out, ref] */ DWORD *pColumnIds);
// CCertDB
HRESULT BeginTransaction(
IN CERTSESSION *pcs,
IN BOOL fPrepareUpdate);
HRESULT CommitTransaction(
IN CERTSESSION *pcs,
IN BOOL fCommit,
IN BOOL fSoftCommit);
HRESULT ReleaseSession(
IN CERTSESSION *pcs);
HRESULT BackupBegin(
IN LONG grbitJet);
HRESULT BackupGetDBFileList(
IN OUT DWORD *pcwcList,
OUT WCHAR *pwszzList); // OPTIONAL
HRESULT BackupGetLogFileList(
IN OUT DWORD *pcwcList,
OUT WCHAR *pwszzList); // OPTIONAL
HRESULT BackupOpenFile(
IN WCHAR const *pwszFile,
OUT JET_HANDLE *phFileDB,
OPTIONAL OUT ULARGE_INTEGER *pliSize);
HRESULT BackupReadFile(
IN JET_HANDLE hFileDB,
OUT BYTE *pb,
IN DWORD cb,
OUT DWORD *pcb);
HRESULT BackupCloseFile(
IN JET_HANDLE hFileDB);
HRESULT BackupTruncateLog();
HRESULT BackupEnd();
HRESULT SetProperty(
IN CERTSESSION *pcs,
IN DBTABLE const *pdt,
IN DWORD cbProp,
IN BYTE const *pbProp); // OPTIONAL
HRESULT GetProperty(
IN CERTSESSION *pcs,
IN DBTABLE const *pdt,
OPTIONAL IN ICertDBComputedColumn *pIComputedColumn,
IN OUT DWORD *pcbProp,
OUT BYTE *pbProp); // OPTIONAL
HRESULT SetAttribute(
IN CERTSESSION *pcs,
IN WCHAR const *pwszAttributeName,
IN DWORD cbValue,
IN BYTE const *pbValue); // OPTIONAL
HRESULT GetAttribute(
IN CERTSESSION *pcs,
IN WCHAR const *pwszAttributeName,
IN OUT DWORD *pcbValue,
OUT BYTE *pbValue); // OPTIONAL
HRESULT SetExtension(
IN CERTSESSION *pcs,
IN WCHAR const *pwszExtensionName,
IN DWORD dwExtFlags,
IN DWORD cbValue,
IN BYTE const *pbValue); // OPTIONAL
HRESULT GetExtension(
IN CERTSESSION *pcs,
IN WCHAR const *pwszExtensionName,
OUT DWORD *pdwExtFlags,
IN OUT DWORD *pcbValue,
OUT BYTE *pbValue); // OPTIONAL
HRESULT CopyRequestNames(
IN CERTSESSION *pcs);
HRESULT GetColumnType(
IN LONG ColumnIndex,
OUT DWORD *pType);
HRESULT EnumCertDBColumnNext(
IN DWORD dwTable,
IN ULONG ielt,
IN ULONG celt,
OUT CERTDBCOLUMN *rgelt,
OUT ULONG *pielt,
OUT ULONG *pceltFetched);
HRESULT EnumCertDBResultRowNext(
IN CERTSESSION *pcs,
IN DWORD ccvr,
IN CERTVIEWRESTRICTION const *acvr,
IN DWORD ccolOut,
IN DWORD const *acolOut,
IN LONG cskip,
OPTIONAL IN ICertDBComputedColumn *pIComputedColumn,
IN ULONG celt,
OUT CERTDBRESULTROW *rgelt,
OUT ULONG *pceltFetched,
OUT LONG *pcskipped);
HRESULT ReleaseResultRow(
IN ULONG celt,
IN OUT CERTDBRESULTROW *rgelt);
HRESULT EnumerateSetup(
IN CERTSESSION *pcs,
IN OUT DWORD *pFlags,
OUT JET_TABLEID *ptableid);
HRESULT EnumerateNext(
IN CERTSESSION *pcs,
IN OUT DWORD *pFlags,
IN JET_TABLEID tableid,
IN LONG cskip,
IN ULONG celt,
OUT CERTDBNAME *rgelt,
OUT ULONG *pceltFetched);
HRESULT EnumerateClose(
IN CERTSESSION *pcs,
IN JET_TABLEID tableid);
HRESULT OpenTables(
IN CERTSESSION *pcs,
OPTIONAL IN CERTVIEWRESTRICTION const *pcvr);
HRESULT CloseTables(
IN CERTSESSION *pcs);
HRESULT Delete(
IN CERTSESSION *pcs);
HRESULT CloseTable(
IN CERTSESSION *pcs,
IN JET_TABLEID tableid);
HRESULT MapPropId(
IN WCHAR const *pwszPropName,
IN DWORD dwFlags,
OUT DBTABLE *pdtOut);
HRESULT TestShutDownState();
#if DBG_CERTSRV
VOID DumpRestriction(
IN DWORD dwSubSystemId,
IN LONG i,
IN CERTVIEWRESTRICTION const *pcvr);
#endif // DBG_CERTSRV
private:
HRESULT _AllocateSession(
OUT CERTSESSION **ppcs);
HRESULT _BackupGetFileList(
IN BOOL fDBFiles,
IN OUT DWORD *pcwcList,
OPTIONAL OUT WCHAR *pwszzList);
HRESULT _CreateTable(
IN DWORD CreateFlags, // CF_*
IN CERTSESSION *pcs,
IN DBCREATETABLE const *pct);
HRESULT _Create(
IN DWORD CreateFlags, // CF_*
IN CHAR const *pszDatBaseName);
HRESULT _CreateIndex(
IN CERTSESSION *pcs,
IN JET_TABLEID tableid,
IN CHAR const *pszIndexName,
IN CHAR const *pchKey,
IN DWORD cbKey,
IN DWORD flags);
HRESULT _AddColumn(
IN CERTSESSION *pcs,
IN JET_TABLEID tableid,
IN DBTABLE const *pdt);
HRESULT _BuildColumnIds(
IN CERTSESSION *pcs,
IN CHAR const *pszTableName,
IN DBTABLE *pdt);
HRESULT _ConvertOldColumnData(
IN CERTSESSION *pcs,
IN CHAR const *pszTableName,
IN DBAUXDATA const *pdbaux,
IN DBTABLE *pdt);
HRESULT _ConvertColumnData(
IN CERTSESSION *pcs,
IN JET_TABLEID tableid,
IN DWORD RowId,
IN DBTABLE const *pdt,
IN DBAUXDATA const *pdbaux,
IN OUT BYTE **ppbBuf,
IN OUT DWORD *pcbBuf);
HRESULT _AddKeyLengthColumn(
IN CERTSESSION *pcs,
IN JET_TABLEID tableid,
IN DWORD RowId,
IN DBTABLE const *pdtPublicKey,
IN DBTABLE const *pdtPublicKeyAlgorithm,
IN DBTABLE const *pdtPublicKeyParameters,
IN DBTABLE const *pdtPublicKeyLength,
IN DBAUXDATA const *pdbaux,
IN OUT BYTE **ppbBuf,
IN OUT DWORD *pcbBuf);
HRESULT _AddCallerName(
IN CERTSESSION *pcs,
IN JET_TABLEID tableid,
IN DWORD RowId,
IN DBTABLE const *pdtCallerName,
IN DBTABLE const *pdtRequesterName,
IN DBAUXDATA const *pdbaux,
IN OUT BYTE **ppbBuf,
IN OUT DWORD *pcbBuf);
HRESULT _SetHashColumnIfEmpty(
IN CERTSESSION *pcs,
IN JET_TABLEID tableid,
IN DWORD RowId,
IN DBTABLE const *pdtHash,
IN DBAUXDATA const *pdbaux,
IN BYTE const *pbHash,
IN DWORD cbHash);
HRESULT _AddCertColumns(
IN CERTSESSION *pcs,
IN JET_TABLEID tableid,
IN DWORD RowId,
IN DBTABLE const *pdtCertHash,
IN DBTABLE const *pdtSKI,
IN DBTABLE const *pdtCert,
IN DBAUXDATA const *pdbaux,
IN OUT BYTE **ppbBuf,
IN OUT DWORD *pcbBuf);
HRESULT _SetColumn(
IN JET_SESID SesId,
IN JET_TABLEID tableid,
IN DBTABLE const *pdt,
IN JET_COLUMNID columnid,
IN DWORD cbProp,
IN BYTE const *pbProp); // OPTIONAL
HRESULT _OpenTableRow(
IN CERTSESSION *pcs,
IN DBAUXDATA const *pdbaux,
OPTIONAL IN CERTVIEWRESTRICTION const *pcvr,
OUT CERTSESSIONTABLE *pTable,
OUT DWORD *pdwRowIdMismatch);
HRESULT _UpdateTable(
IN CERTSESSION *pcs,
IN JET_TABLEID tableid);
HRESULT _OpenTable(
IN CERTSESSION *pcs,
IN DBAUXDATA const *pdbaux,
IN CERTVIEWRESTRICTION const *pcvr,
IN OUT CERTSESSIONTABLE *pTable);
HRESULT _SetIndirect(
IN CERTSESSION *pcs,
IN OUT CERTSESSIONTABLE *pTable,
IN WCHAR const *pwszName,
IN DWORD const *pdwExtFlags, // OPTIONAL
IN DWORD cbValue,
IN BYTE const *pbValue); // OPTIONAL
HRESULT _GetIndirect(
IN CERTSESSION *pcs,
IN OUT CERTSESSIONTABLE *pTable,
IN WCHAR const *pwszName,
OUT DWORD *pdwExtFlags, // OPTIONAL
IN OUT DWORD *pcbValue,
OUT BYTE *pbValue); // OPTIONAL
DBTABLE const *_MapTable(
IN WCHAR const *pwszPropName,
IN DBTABLE const *pdt);
HRESULT _MapPropIdIndex(
IN DWORD ColumnIndex,
OUT DBTABLE const **ppdt,
OPTIONAL OUT DWORD *pType);
HRESULT _MapTableToIndex(
IN DBTABLE const *pdt,
OUT DWORD *pColumnIndex);
HRESULT _RetrieveColumnBuffer(
IN CERTSESSION *pcs,
IN JET_TABLEID tableid,
IN DBTABLE const *pdt,
IN JET_COLUMNID columnid,
IN OUT DWORD *pcbProp,
IN OUT BYTE **ppbBuf,
IN OUT DWORD *pcbBuf);
HRESULT _RetrieveColumn(
IN CERTSESSION *pcs,
IN JET_TABLEID tableid,
IN DBTABLE const *pdt,
IN JET_COLUMNID columnid,
OPTIONAL IN ICertDBComputedColumn *pIComputedColumn,
IN OUT DWORD *pcbData,
OUT BYTE *pbData);
HRESULT _CompareColumnValue(
IN CERTSESSION *pcs,
IN CERTVIEWRESTRICTION const *pcvr,
OPTIONAL IN ICertDBComputedColumn *pIComputedColumn);
HRESULT _EnumerateMove(
IN CERTSESSION *pcs,
IN OUT DWORD *pFlags,
IN DBAUXDATA const *pdbaux,
IN JET_TABLEID tableid,
IN LONG cskip);
HRESULT _MakeSeekKey(
IN CERTSESSION *pcs,
IN JET_TABLEID tableid,
IN DBTABLE const *pdt,
IN BYTE const *pbValue,
IN DWORD cbValue);
HRESULT _SeekTable(
IN CERTSESSION *pcs,
IN JET_TABLEID tableid,
IN CERTVIEWRESTRICTION const *pcvr,
IN DBTABLE const *pdt,
OPTIONAL IN ICertDBComputedColumn *pIComputedColumn,
IN DWORD dwPosition,
OUT DWORD *pTableFlags
DBGPARM(IN DBAUXDATA const *pdbaux));
HRESULT _MoveTable(
IN CERTSESSION *pcs,
IN DWORD ccvr,
IN CERTVIEWRESTRICTION const *acvr,
OPTIONAL IN ICertDBComputedColumn *pIComputedColumn,
IN LONG cskip,
OUT LONG *pcskipped);
HRESULT _GetResultRow(
IN CERTSESSION *pcs,
IN DWORD ccvr,
IN CERTVIEWRESTRICTION const *acvr,
IN LONG cskip,
IN DWORD ccolOut,
IN DWORD const *acolOut,
OPTIONAL IN ICertDBComputedColumn *pIComputedColumn,
OUT CERTDBRESULTROW *pelt,
OUT LONG *pcskipped);
HRESULT _JetSeekFromRestriction(
IN CERTVIEWRESTRICTION const *pcvr,
IN DWORD dwPosition,
OUT DBSEEKDATA *pSeekData);
HRESULT _DupString(
OPTIONAL IN WCHAR const *pwszPrefix,
IN WCHAR const *pwszIn,
OUT WCHAR **ppwszOut);
HRESULT _Rollback(
IN CERTSESSION *pcs);
#if DBG_CERTSRV
HRESULT _DumpRowId(
IN CHAR const *psz,
IN CERTSESSION *pcs,
IN JET_TABLEID tableid);
HRESULT _DumpColumn(
IN CHAR const *psz,
IN CERTSESSION *pcs,
IN JET_TABLEID tableid,
IN DBTABLE const *pdt,
OPTIONAL IN ICertDBComputedColumn *pIComputedColumn);
#endif // DBG_CERTSRV
BOOL m_fDBOpen;
BOOL m_fDBReadOnly;
BOOL m_fDBRestart;
BOOL m_fFoundOldColumns;
BOOL m_fAddedNewColumns;
JET_INSTANCE m_Instance;
CERTSESSION *m_aSession;
DWORD m_cSession;
DWORD m_cbPage;
CRITICAL_SECTION m_critsecSession;
CRITICAL_SECTION m_critsecAutoIncTables;
BOOL m_cCritSec;
BOOL m_fPendingShutDown;
};