Відстеження з середини DLL її приєднання та від'єднання до процесу

Тему про відстеження з середини DLL її приєднання до процесу та від'єднання можна знайти в сотнях статей. Наприклад, детально це розписано в блозі Gunsmoker'а. Але в більшості видно одну й ту ж саму помилку. А саме:

library MyDLL; 

procedure DllMain(Reason: Integer); 
begin 
  { This is called when Reason = DETACH } 
end; 

begin 
  { This executes when a process ATTACHes. } 
  DllProc := @DllMain; 
end. 

Реалізувавши таким чином бібліотеку можна бути впевненим, що основний код бібліотеки буде викликаний 4 рази на подіях:

  DLL_PROCESS_ATTACH = 1;              // Executable attaches to DLL
  DLL_THREAD_ATTACH = 2;               // Thread in exe calls DLL
  DLL_THREAD_DETACH = 3;               // Thread leaves DLL
  DLL_PROCESS_DETACH = 0;              // Exe detaches from DLL

Відповідно, присвоєння DllProc := @DllMain також буде викликано 4 рази, при цьому вказівник на DllMain може бути втрачено. Тому варто його зберегти, а при виході з бібліотеки відновити:

library MyDLL; 

uses ...;

var
  OldDllProc : TDLLProc;

procedure DllMain(Reason: Integer);
begin
  case Reason of
    DLL_PROCESS_ATTACH:
      begin
        OldDllProc := @DllProc;
        DllProc    := @MyProc;
      end;
    DLL_PROCESS_DETACH:
      begin
        DllProc := OldDllProc;
      end;
  end;
end;

begin
  if (DllProc = nil) then
    DllMain(DLL_PROCESS_ATTACH);
end.

Таким чином можна уникнути втрат пам'яті та виконання зайвого або навіть шкідливого в даній ситуації коду.