Как изменить реализацию (обход) объявленной извне функции - PullRequest
Новая
       14

Как изменить реализацию (обход) объявленной извне функции

12 голосов
/ 02 августа 2011

У меня есть сторонняя функция 

function DataCompare(const S1, S2: string; APartial: Boolean): Boolean;
begin
   ...
end;

Используется в другом стороннем устройстве.

Я хочу заменить тело функции во время выполнения другой новой реализацией.

Возможно ли это? Я предполагаю, что будет необходимость некоторого взлома (аля VirtualMemoryUnprotect). Не ассемблерное решение очень приветствуется.

Ответы [ 2 ]

36 голосов
/ 02 августа 2011

Да, вы можете сделать это, используя функции ReadProcessMemory и WriteProcessMemory для исправления кода текущего процесса. По сути, вы получаете адрес процедуры или функции для исправления, а затем вставляете инструкцию Jump в адрес новой процедуры.

Проверьте этот код 

Uses
  uThirdParty; //this is the unit where the original DataCompare function is declarated

type
  //strctures to hold the address and instructions to patch
  TJumpOfs = Integer;
  PPointer = ^Pointer;

  PXRedirCode = ^TXRedirCode;
  TXRedirCode = packed record
    Jump: Byte;
    Offset: TJumpOfs;
  end;

  PAbsoluteIndirectJmp = ^TAbsoluteIndirectJmp;
  TAbsoluteIndirectJmp = packed record
    OpCode: Word;
    Addr: PPointer;
  end;

var
 DataCompareBackup: TXRedirCode; //Store the original address of the function to patch


//this is the implementation of the new function
function DataCompareHack(const S1, S2: string; APartial: Boolean): Boolean;
begin
  //here write your own code
end;

//get the address of a procedure or method of a function 
function GetActualAddr(Proc: Pointer): Pointer;
begin
  if Proc <> nil then
  begin
    if (Win32Platform = VER_PLATFORM_WIN32_NT) and (PAbsoluteIndirectJmp(Proc).OpCode = $25FF) then
      Result := PAbsoluteIndirectJmp(Proc).Addr^
    else
      Result := Proc;
  end
  else
    Result := nil;
end;

//patch the original function or procedure
procedure HookProc(Proc, Dest: Pointer; var BackupCode: TXRedirCode);
var
  n: {$IFDEF VER230}NativeUInt{$ELSE}DWORD{$ENDIF};
  Code: TXRedirCode;
begin
  Proc := GetActualAddr(Proc);
  Assert(Proc <> nil);
  //store the address of the original procedure to patch
  if ReadProcessMemory(GetCurrentProcess, Proc, @BackupCode, SizeOf(BackupCode), n) then
  begin
    Code.Jump := $E9;
    Code.Offset := PAnsiChar(Dest) - PAnsiChar(Proc) - SizeOf(Code);
    //replace the target procedure address  with the new one.
    WriteProcessMemory(GetCurrentProcess, Proc, @Code, SizeOf(Code), n);
  end;
end;
//restore the original address of the hooked function or procedure
procedure UnhookProc(Proc: Pointer; var BackupCode: TXRedirCode);
var
  n: {$IFDEF VER230}NativeUInt{$ELSE}Cardinal{$ENDIF};
begin
  if (BackupCode.Jump <> 0) and (Proc <> nil) then
  begin
    Proc := GetActualAddr(Proc);
    Assert(Proc <> nil);
    WriteProcessMemory(GetCurrentProcess, Proc, @BackupCode, SizeOf(BackupCode), n);
    BackupCode.Jump := 0;
  end;
end;

//Patch the original procedure or function
procedure HookDataCompare;
begin
  //look how is passed the address of the original procedure (including the unit name)
  HookProc(@uThirdParty.DataCompare, @DataCompareHack, DataCompareBackup);
end;

//restore the address of the original procedure or function
procedure UnHookDataCompare;
begin
  UnhookProc(@uThirdParty.DataCompare, DataCompareBackup);
end;


initialization
 HookDataCompare;
finalization
 UnHookDataCompare;
end.

Теперь каждый раз, когда вы запускаете свое приложение и вызываете функцию DataCompare, будет выполняться инструкция перехода (на новый адрес), в результате чего будет вызываться функция DataCompareHack.

2 голосов
/ 02 августа 2011

Я думаю, что JCL имеет несколько утилит для такого рода вещей ... Я сам не использовал его, но быстро осмотрел и следующие вещи выглядят многообещающе: 

jclSysUtils.WriteProtectedMemory()
jclPeImage.TJclPeMapImgHooks.ReplaceImport()

Я думаю, jclHookExcept.JclHookExceptions() демонстрирует, как их использовать.

...