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.
75 lines
1.4 KiB
75 lines
1.4 KiB
#pragma once
|
|
#include "modlist.h"
|
|
|
|
class CModuleListSet;
|
|
|
|
enum MTE_FLAGS
|
|
{
|
|
MTE_DEFAULT = 0x00000000,
|
|
MTE_DELAY_LOADED = 0x00000001,
|
|
};
|
|
|
|
class CModuleTreeEntry
|
|
{
|
|
public:
|
|
CModule* m_pModule;
|
|
CModule* m_pImportModule;
|
|
|
|
DWORD m_dwFlags; // MTE_FLAGS
|
|
};
|
|
|
|
|
|
// Flags for HrGetModuleBindings.
|
|
//
|
|
enum GMB_FLAGS
|
|
{
|
|
GMBF_DEFAULT = 0x00000000,
|
|
GMBF_ADD_TO_MLSET = 0x00000001,
|
|
};
|
|
|
|
class CModuleTree : public vector<CModuleTreeEntry>
|
|
{
|
|
public:
|
|
CModuleList Modules;
|
|
|
|
public:
|
|
~CModuleTree ();
|
|
|
|
HRESULT
|
|
HrAddEntry (
|
|
IN CModule* pMod,
|
|
IN CModule* pImport,
|
|
IN DWORD dwFlags);
|
|
|
|
HRESULT
|
|
HrGetModuleBindings (
|
|
IN const CModule* pMod,
|
|
IN DWORD dwFlags /* GMB_FLAGS */,
|
|
OUT CModuleListSet* pSet) const;
|
|
|
|
CModuleTreeEntry*
|
|
PFindFirstEntryWithModule (
|
|
IN const CModule* pMod) const;
|
|
|
|
CModuleTreeEntry*
|
|
PFindFirstEntryAfterModuleGroup (
|
|
IN const CModule* pMod) const;
|
|
|
|
CModuleTreeEntry*
|
|
PBinarySearchEntry (
|
|
IN const CModule* pMod,
|
|
IN const CModule* pImport,
|
|
OUT CModuleTreeEntry** pInsertPosition OPTIONAL) const;
|
|
|
|
private:
|
|
CModuleTreeEntry*
|
|
PBinarySearchEntryByModule (
|
|
IN const CModule* pMod,
|
|
OUT CModuleTreeEntry** pInsertPosition OPTIONAL) const;
|
|
|
|
#if DBG
|
|
VOID DbgVerifySorted ();
|
|
#else
|
|
VOID DbgVerifySorted () {}
|
|
#endif
|
|
};
|