diff --git a/src/hook.c b/src/hook.c index 8faa796..743f255 100644 --- a/src/hook.c +++ b/src/hook.c @@ -104,7 +104,10 @@ void Hook_Create(char *moduleName, char *functionName, PROC newFunction, PROC *f #endif if (HookingMethod == 1) + { Hook_PatchIAT(GetModuleHandle(NULL), moduleName, functionName, newFunction); + Hook_PatchIAT(GetModuleHandle("storm.dll"), moduleName, functionName, newFunction); + } } void Hook_Revert(char *moduleName, char *functionName, PROC newFunction, PROC *function) @@ -126,6 +129,12 @@ void Hook_Revert(char *moduleName, char *functionName, PROC newFunction, PROC *f moduleName, functionName, GetProcAddress(GetModuleHandle(moduleName), functionName)); + + Hook_PatchIAT( + GetModuleHandle("storm.dll"), + moduleName, + functionName, + GetProcAddress(GetModuleHandle(moduleName), functionName)); } }