61 lines
1.7 KiB
C
61 lines
1.7 KiB
C
|
#if !defined(_FUSION_GSGENCTX_H_INCLUDED_)
|
||
|
|
#define _FUSION_GSGENCTX_H_INCLUDED_
|
||
|
|
|
||
|
|
#pragma once
|
||
|
|
|
||
|
|
#include "fusionheap.h"
|
||
|
|
|
||
|
|
class CGSGenCtx
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
static BOOL Create(
|
||
|
|
PGUID_SECTION_GENERATION_CONTEXT *GSGenContext,
|
||
|
|
ULONG DataFormatVersion,
|
||
|
|
GUID_SECTION_GENERATION_CONTEXT_CALLBACK_FUNCTION CallbackFunction,
|
||
|
|
PVOID CallbackContext
|
||
|
|
);
|
||
|
|
|
||
|
|
BOOL Add(const GUID &rGuid, PVOID DataContext, ULONG AssemblyRosterIndex, DWORD DuplicateErrorCode);
|
||
|
|
BOOL Find(const GUID &rGuid, PVOID *DataContext);
|
||
|
|
BOOL GetSectionSize(PSIZE_T SectionSize);
|
||
|
|
BOOL GetSectionData(SIZE_T BufferSize, PVOID Buffer, SIZE_T *BytesWritten);
|
||
|
|
PVOID GetCallbackContext() { return m_CallbackContext; }
|
||
|
|
|
||
|
|
VOID DeleteYourself() { FUSION_DELETE_SINGLETON(this); }
|
||
|
|
|
||
|
|
~CGSGenCtx();
|
||
|
|
protected:
|
||
|
|
CGSGenCtx();
|
||
|
|
|
||
|
|
GUID_SECTION_GENERATION_CONTEXT_CALLBACK_FUNCTION m_CallbackFunction;
|
||
|
|
PVOID m_CallbackContext;
|
||
|
|
|
||
|
|
class Entry
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
Entry() : m_DataContext(NULL), m_Next(NULL) { }
|
||
|
|
~Entry() { }
|
||
|
|
|
||
|
|
BOOL Initialize(const GUID &Guid, PVOID DataContext, ULONG AssemblyRosterIndex);
|
||
|
|
BOOL GetEntryDataSize(CGSGenCtx *pSSGenCtx, SIZE_T &rSize);
|
||
|
|
BOOL GetEntryData(CGSGenCtx *pSSGenCtx, SIZE_T BufferSize, PVOID Buffer, SIZE_T *BytesWritten);
|
||
|
|
|
||
|
|
GUID m_Guid;
|
||
|
|
PVOID m_DataContext;
|
||
|
|
Entry *m_Next;
|
||
|
|
ULONG m_AssemblyRosterIndex;
|
||
|
|
};
|
||
|
|
|
||
|
|
friend Entry;
|
||
|
|
|
||
|
|
static int __cdecl SortGuidSectionEntries(const void *elem1, const void *elem2);
|
||
|
|
|
||
|
|
ULONG m_EntryCount;
|
||
|
|
ULONG m_HashTableSize;
|
||
|
|
Entry *m_FirstEntry;
|
||
|
|
Entry *m_LastEntry;
|
||
|
|
ULONG m_DataFormatVersion;
|
||
|
|
};
|
||
|
|
|
||
|
|
#endif
|