1:
2:
3:
4:
5:
6:
7:
8:
9:
10:
11:
12:
13:
14:
15:
16:
17:
18:
19:
20:
21:
22:
23:
24:
25:
26:
27:
28:
29:
30:
31:
32:
33:
34:
35:
36:
37:
38:
39:
40:
41:
42:
43:
44:
45:
46:
47:
48:
49:
50:
51:
52:
53:
54:
55:
56:
57:
58:
59:
60:
61:
62:
63:
64:
65:
66:
67:
68:
69:
70:
71:
72:
73:
74:
75:
76:
77:
78:
79:
80:
81:
82:
83:
84:
85:
86:
87:
88:
89:
90:
91:
92:
93:
94:
95:
96:
97:
98:
99:
100:
101:
102:
103:
104:
105:
106:
107:
108:
109:
110:
111:
112:
113:
114:
115:
116:
117:
118:
119:
120:
121:
122:
123:
124:
125:
126:
127:
128:
129:
130:
131:
132:
133:
134:
135:
136:
137:
138:
139:
140:
141:
142:
143:
144:
145:
146:
147:
148:
149:
|
library MatHook;
uses
Windows,
Messages;
type
THookRec = record
hMatHook: HHOOK;
hMatWnd: HWND;
oldProc: Integer;
end;
var
map: DWord;
buf: ^THookRec;
// new window proc - runs in context of target process
function MatWndProc(Handle: hWnd; Msg: uInt; wp: wParam; lp: lParam): LongInt; stdcall;
begin
try
case Msg of
WM_CREATE:
begin
MessageBox(0, GetCommandLine, 'Command Line parameter(s)', MB_OK);
end;
// user definied message to stop subclassing
// (RegisterWindowMessage would be a better choice instead of WM_USER message!)
WM_USER + 1:
begin
// delete custom menu entries (quick'n'dirty)
SetWindowLong(buf^.hMatWnd, GWL_WNDPROC, buf^.oldProc);
end;
end;
Result := CallWindowProc(Pointer(buf^.oldProc), Handle, Msg, wp, lp);
except
Result := 0;
end;
end;
// hook proc - waits for target window to be created
function MatHookProc(nCode: Integer; wp: wParam; lp: lParam): LongInt; stdcall;
var
hTemp: hWnd;
szClass: array[0..255] of Char;
begin
try
if (nCode >= HC_ACTION) then
begin
Case nCode of
HCBT_CREATEWND:
begin
hTemp := HWND(wp);
FillChar(szClass, 256, 0);
GetClassName(hTemp, szClass, 256);
if (szClass = 'Notepad') then
begin
buf^.hMatWnd := htemp;
buf^.oldProc := GetWindowLong(buf^.hMatWnd, GWL_WNDPROC);
SetWindowLong(buf^.hMatWnd, GWL_WNDPROC, Integer(@MatWndProc));
end;
end;
HCBT_DESTROYWND:
begin
hTemp := HWND(wp);
FillChar(szClass, 256, 0);
GetClassName(hTemp, szClass, 256);
if (szClass = 'Notepad') then
begin
SetWindowLong(buf^.hMatWnd, GWL_WNDPROC, buf^.OldProc);
end;
end;
end;
end;
Result := CallNextHookEx(buf^.hMatHook, nCode, wp, lp);
except
Result := 0;
end;
end;
// sets up hook
function SetHook: Boolean; stdcall; export;
begin
try
Result := false;
if (not assigned(buf)) then
begin
map := CreateFileMapping(DWord(-1), nil, PAGE_READWRITE, 0, SizeOf(THookRec), 'HookRecMemBlock');
buf := MapViewOfFile(map, FILE_MAP_ALL_ACCESS, 0, 0, 0);
buf^.hMatHook := SetWindowsHookEx(WH_CBT, @MatHookProc, hInstance, 0);
Result := true;
end;
except
Result := false;
end;
end;
// removes hook
function RemoveHook: Boolean; stdcall; export;
begin
Result := false;
if (assigned(buf)) then
begin
// tell our new wnd proc to stop subclassing
// (has to be done in context of target process)
SendMessage(buf^.hMatWnd, wm_User + 1, 1, 0);
if (buf^.hMatHook <> 0) then UnhookWindowsHookEx(buf^.hMatHook);
buf^.hMatHook := 0;
UnmapViewOfFile(buf);
buf := nil;
Result := true;
end;
end;
// DLL entry point
procedure DllEntry(dwReason: DWord);
begin
Case dwReason of
DLL_PROCESS_ATTACH:
begin
if (not assigned(buf)) then
begin
map := OpenFileMapping(FILE_MAP_ALL_ACCESS, false, 'HookRecMemBlock');
buf := MapViewOfFile(map, FILE_MAP_ALL_ACCESS, 0, 0, 0);
CloseHandle(map);
map := 0;
end;
end;
DLL_PROCESS_DETACH:
begin
UnmapViewOfFile(buf);
buf := nil;
end;
end;
end;
exports
SetHook,
RemoveHook;
// main
begin
DisableThreadLibraryCalls(hInstance);
DllProc := @DLLEntry;
DllEntry(DLL_PROCESS_ATTACH);
end.
|
