Вопрос # 902

Вопрос задал: SMaks (статус: 1-ый класс)
Вопрос отправлен: 7 сентября 2007, 23:54
Состояние вопроса: открыт, ответов: 1.

Ответ отправил: Матвеев Игорь Владимирович (статус: Студент)
Время отправки: 8 сентября 2007, 02:13
Оценка за ответ: 3

Матвеев Игорь Владимирович (статус: Студент), 10 сентября 2007, 00:21 [#1]:

asm // Метка начала функции
   jmp @@skeep
     db 53h, 43h, 0BBh, 00h, 62h, 5Bh, 79h, 80h
   @@skeep:
end;

SMaks (статус: 1-ый класс), 10 сентября 2007, 14:29 [#2]:

Матвеев Игорь Владимирович (статус: Студент), 11 сентября 2007, 14:16 [#3]:

// Функция которую будем патчить
procedure TestFunc();
var
  vTime: Integer;
begin
 asm    // Метка начала кода, патчинг нужно будет начинать сразу за меткой
   jmp @@skeep;
   db 53h, 43h, 0BBh, 00h, 62h, 5Bh, 79h, 80h
   @@skeep:
 end;
 
 // Разные бесполезные вычисления (время и условие, чтобы компилятор не пытался
 // вырезать этот код как неиспользуемый)
 vTime := GetTickCount();
 if vTime = 0 then Beep;
 // Вывод
 Beep;
 ShowMessage(\'Hello, this is a original TestFunc() code\');
 
 asm    // Метка конца кода, патчинг нужно будет кончаться сразу перед меткой
   jmp @@skeep;
   db 52h, 42h, 0BAh, 0FFh, 61h, 5Ah, 78h, 7Fh
   @@skeep:
 end;
end;
 
// Эта функция ищет кусок кода между метками
function SearchBlock(const StartPtr: Pointer; MaxSize: Integer; var ResultPtr: Pointer; var Size: Integer): Boolean;
const
  markOne : array[0..7] of Byte = ($80, $79, $5B, $62, $00, $BB, $43, $53);
  markTwo : array[0..7] of Byte = ($7F, $78, $5A, $61, $FF, $BA, $42, $52);
var
  i, j : Integer;
  Ptr1,
  Ptr2 : Pointer;
begin
 // При поиске нельзя сравнивать метку с константной последовательностью байт,
 // поскольку мы можем наткнуться при поиске не на метку, а на саму эту константу
 // (в данном случае это почти невозможно, поскольку мы знаем откуда начинать поиск,
 // но вообще в реальных проектах это нужно учитывать). Поэтому бедем искать
 // последний байт, а потом сравнивать задом-наперед.
 
 // Ищем стартовую метку
 for i := Integer(StartPtr) to Integer(StartPtr) + MaxSize do
   if Byte(Pointer(i)^) = markOne[0] then                      // Ищем последний байт
     begin
       for j := 0 to 7 do                                      // Проверяем байты сзади
         if Byte(Pointer(i-j)^) <> markOne[j] then             // с меткой
           begin
             Ptr1 := 0;                // Различие
             break;
           end else if j = 7 then Ptr1 := Pointer(i);
       if Integer(Ptr1) <> 0 then Break;       // Метка найдена
  end;
 ResultPtr := Pointer(Integer(Ptr1)+1);
 // Ищем завершающую метку
 for i := Integer(ResultPtr) to Integer(StartPtr) + MaxSize do
   if Byte(Pointer(i)^) = markTwo[0] then                      // Ищем последний байт
     begin
       for j := 0 to 7 do                                      // Проверяем байты сзади
         if Byte(Pointer(i-j)^) <> markTwo[j] then             // с меткой
           begin
             Ptr2 := 0;                // Различие
             break;
           end else if j = 7 then Ptr2 := Pointer(i);
       if Integer(Ptr2) <> 0 then Break;       // Метка найдена
  end;
 Size := Integer(Ptr2)-7-2 - Integer(ResultPtr); // Отнимаем 7 байт метки и 2
                                                 // байта jmp перехода
 if (ResultPtr <> nil) and (Size <> 0) then
   Result := True else Result := False;
end;
 
// Тестовый вызов тестовой функции!
procedure TForm1.Button1Click(Sender: TObject);
begin
 TestFunc()
end;
 
// Патчинг
procedure TForm1.Button2Click(Sender: TObject);
var
  p : Pointer;
  s : Integer;
  k : Cardinal;
  i : Integer;
begin
 // ШАГ №1 - Найти интересующий нас кусок кода
 if not SearchBlock(@TestFunc, 4024, p, s) then
   begin
     MessageDlg(\'Ошибка: не найдены поисковые метки\', mtError, [mbOk], -1);
     Exit;
   end;
 // ШАГ №2 - Сделаем память доступной для записи
 if not VirtualProtect(p, s, PAGE_READWRITE, k) then
   begin
     MessageDlg(\'Ошибка: не получилось установить для памяти атрибут PAGE_READWRITE\', mtError, [mbOk], -1);
     Exit;
   end;
 // ШАГ №3 - Непосредстенно патчинг
 try
   for i := 0 to s-1 do
     Byte( Pointer(Integer(p)+i)^ ) := $90;   // Заполняем все nop\'ами 
 except
   on EAccessViolation do MessageDlg(\'Ошибка: невозможно записать в память\', mtError, [mbOk], -1);
 end;
end;