Code:

unit EditorVMTandDMTTables;

interface

// функция служит для выяснения существования VMT у класса

// возвращает True, если класс имеет VMT и False - если нет

function IsVMTExist(Cls: TClass): Boolean;

// процедура служит для замены адреса метода в VMT класса со смещением

// Offset(должно быть кратно 4) новым адресом, хранящимся в NewMet

// примечание: перед вызовом этой процедуры проверяйте существование

// VMT у класса функцией IsVMTExist

procedure VirtMethodReplace(Cls: TClass; Offset: LongWord; NewMet: Pointer); overload;

// процедура служит для замены адреса метода, хранящегося в OldMet,

// в VMT класса новым адресом, хранящимся в NewMet

// примечание: перед вызовом этой процедуры проверяйте существование

// VMT у класса функцией IsVMTExist

procedure VirtMethodReplace(Cls: TClass; OldMet, NewMet: Pointer); overload;

// функция служит для замены адреса динамического метода класса с индексом,

// хранящимся в Index, новым адресом, хранящимся в NewMet

// возвращает True, если метод с данным индексом найден и False - если нет

function DynMethodReplace(Cls: TClass; Index: Word; NewMet: Pointer): Boolean; overload;

// функция служит для замены адреса динамического метода класса, хранящегося

// в OldMet, новым адресом, хранящимся в NewMet

// возвращает True, если метод с данным адресом найден и False - если нет

function DynMethodReplace(Cls: TClass; OldMet, NewMet: Pointer): Boolean; overload;

implementation

// функция служит для получения указателя на байт, следующий за адресом

// последнего метода в VMT класса

// возвращает nil в случае, если у класса нет VMT

// функция является "внутренней" в модуле

// (используется другими подпрограммами и не объявлена в секции interface)

// , поэтому используйте её только если

// Вы полностью уверены в своих действиях(она изменяет "рабочие" регистры

// ECX и EDX)

function GetVMTEnd(Cls: TClass): Pointer;

asm

// Вход: Cls --> EAX

// Выход: Result --> EAX

PUSH EBX

MOV ECX, 8

MOV EBX, -1

MOV EDX, vmtSelfPtr

@@cycle:

ADD EDX, 4

CMP [EAX + EDX], EAX

JE @@vmt_not_found

JB @@continue

CMP [EAX + EDX], EBX

JAE @@continue

MOV EBX, [EAX + EDX]

@@continue:

DEC ECX

JNZ @@cycle

MOV EAX, EBX

JMP @@exit

@@vmt_not_found:

XOR EAX, EAX

@@exit:

POP EBX

end;

function IsVMTExist(Cls: TClass): Boolean;

asm

// Вход: Cls --> EAX

// Выход: Result --> AL

CALL GetVMTEnd

TEST EAX, EAX

JZ @@vmt_not_found

MOV AL, 1

@@vmt_not_found:

end;

procedure VirtMethodReplace(Cls: TClass; Offset: LongWord; NewMet: Pointer); overload;

asm

// Вход: Cls --> EAX, Offset --> EDX, NewMet --> ECX

MOV [EAX + EDX], ECX

end;

procedure VirtMethodReplace(Cls: TClass; OldMet, NewMet: Pointer); overload;

asm

// Вход: Cls --> EAX, OldMet --> EDX, NewMet --> ECX

PUSH EDI

MOV EDI, EAX

PUSH ECX

PUSH EDX

PUSH EAX

CALL GetVMTEnd

POP EDX

SUB EAX, EDX

SHR EAX, 2

POP EDX

POP ECX

PUSH ECX

MOV ECX, EAX

MOV EAX, EDX

POP EDX

REPNE SCASD

JNE @@OldMet_not_found

MOV [EDI - 4], EDX

@@OldMet_not_found:

POP EDI

end;

function DynMethodReplace(Cls: TClass; Index: Word; NewMet: Pointer): Boolean; overload;

asm

// Вход: Cls --> EAX, Index --> DX, NewMet --> ECX

// Выход: Result --> AL

PUSH EDI

PUSH ESI

MOV ESI, ECX

XOR EAX, EDX

XOR EDX, EAX

XOR EAX, EDX

JMP @@start

@@cycle:

MOV EDX, [EDX]

@@start:

MOV EDI, [EDX].vmtDynamicTable

TEST EDI, EDI

JZ @@get_parent_dmt

MOVZX ECX, WORD PTR [EDI]

PUSH ECX

ADD EDI, 2

REPNE SCASW

JE @@Index_found

POP ECX

@@get_parent_dmt:

MOV EDX, [EDX].vmtParent

TEST EDX, EDX

JNZ @@cycle

JMP @@Index_not_found

@@Index_found:

POP EAX

SHL EAX, 1

SUB EAX, ECX

MOV [EDI + EAX * 2 - 4], ESI

MOV AL, 1

JMP @@exit

@@Index_not_found:

XOR AL, AL

@@exit:

POP ESI

POP EDI

end;

function DynMethodReplace(Cls: TClass; OldMet, NewMet: Pointer): Boolean; overload;

asm

// Вход: Cls --> EAX, OldMet --> EDX, NewMet --> ECX

// Выход: Result --> AL

PUSH EDI

PUSH ESI

MOV ESI, ECX

XOR EAX, EDX

XOR EDX, EAX

XOR EAX, EDX

JMP @@start

@@cycle:

MOV EDX, [EDX]

@@start:

MOV EDI, [EDX].vmtDynamicTable

TEST EDI, EDI

JZ @@get_parent_dmt

MOVZX ECX, WORD PTR [EDI]

LEA EDI, EDI + 2 * ECX + 2

REPNE SCASD

JE @@OldMet_found

@@get_parent_dmt:

MOV EDX, [EDX].vmtParent

TEST EDX, EDX

JNZ @@cycle

JMP @@OldMet_not_found

@@OldMet_found:

MOV [EDI - 4], ESI

MOV AL, 1

JMP @@exit

@@OldMet_not_found:

XOR AL, AL

@@exit:

POP ESI

POP EDI

end;

end.