Extended Inline API Hooking

Share your advanced PureBasic knowledge/code with the community.
User avatar
Rescator
Addict
Addict
Posts: 1769
Joined: Sat Feb 19, 2005 5:05 pm
Location: Norway

Re: Extended Inline API Hooking

Post by Rescator »

I know that, but read the topic title and look at the source in the first post again.

It doesn't really hook the API (OS) function now does it? It just changes the function table pointer (!) in the "current" process,
that's what I reacted too originally when I saw the post.

It's not like I could guess it was Part B of a larger code now could I? ;)

(I just felt like pointing that confusion out as too often people just dump code into Tips'n'Tricks and it's not always easy to know what it's intended for or if it's a complete example or is supposed to be used with other code to really work, I liked how simple it was and felt a need to point out the few issues I found with it.)
PyroStrex
User
User
Posts: 61
Joined: Mon Mar 22, 2010 3:08 pm

Re: Extended Inline API Hooking

Post by PyroStrex »

@Rescator
Hmm, taking all into one post, so basically you want me to do what? Make also an example to demonstrate the injection and ejection process? You should list it down :P. I was also trying to improve my code :D. Umm, about the Unicode function, can you name one function?
Thorium
Addict
Addict
Posts: 1271
Joined: Sat Aug 15, 2009 6:59 pm

Re: Extended Inline API Hooking

Post by Thorium »

PyroStrex wrote: Edit: Oh, and can you also tell me what other function that cannot be hooked? ASM view would be nice :)
For example DbgBreakPoint.
It's only a 2 byte procedure.

Code: Select all

int 3
ret
PyroStrex
User
User
Posts: 61
Joined: Mon Mar 22, 2010 3:08 pm

Re: Extended Inline API Hooking

Post by PyroStrex »

Thorium wrote:
PyroStrex wrote: Edit: Oh, and can you also tell me what other function that cannot be hooked? ASM view would be nice :)
For example DbgBreakPoint.
It's only a 2 byte procedure.

Code: Select all

int 3
ret
Owh, i can do IAT Hooking for this. If fails, EAT Hooking will try to support. But dunno if it will still fail though. I will go and take a look.
PyroStrex
User
User
Posts: 61
Joined: Mon Mar 22, 2010 3:08 pm

Re: Extended Inline API Hooking

Post by PyroStrex »

Added The UnHook Function and the example also updated.
PureGuy
Enthusiast
Enthusiast
Posts: 102
Joined: Mon Aug 30, 2010 11:51 am

Re: Extended Inline API Hooking

Post by PureGuy »

I know this a bit old topic, but a very useful one.
My ASM knowledge is very poor.
Does anyone now how to get this function working for x64?
Opcode
Enthusiast
Enthusiast
Posts: 137
Joined: Thu Jul 18, 2013 4:58 am

Re: Extended Inline API Hooking

Post by Opcode »

Updated 5.7X as a module. Removed OpenProcess so it will only work for the current process that it's running in (you can re-add it if needed).

Detour.pbi

Code: Select all

DeclareModule CHook
  
  Import "Kernel32.lib"
    GetProcAddress_(hModule.l, lpProcName.p-ascii) As "_GetProcAddress@8"
  EndImport
  
  Declare.b InlineHook(LibraryName.s, LibraryFunction.s, NewFunction.l, OldFunction.l)
  Declare.b InlineUnhook(LibraryName.s, LibraryFunction.s, NewFunction.l, OldFunction.l)
  
EndDeclareModule

Module CHook
  
  ; --------------------------------------------------
  ;   Install Hook
  ; --------------------------------------------------
  
  Procedure.b InlineHook(LibraryName.s, LibraryFunction.s, NewFunction.l, OldFunction.l)
    
    Protected.l OldAddr, NewAddr, RetAddr, CodeAddress, OldProtection
    
    OldAddr = GetProcAddress_(GetModuleHandle_(LibraryName), LibraryFunction)
    NewAddr = NewFunction
    RetAddr = OldAddr + 5
    
    CodeAddress = VirtualAllocEx_(GetCurrentProcess_(), #Null, 10, #MEM_COMMIT, #PAGE_EXECUTE_READWRITE)
    
    If CodeAddress = #Null
      ProcedureReturn #False
    EndIf
    
    VirtualProtect_(OldAddr, 5, #PAGE_EXECUTE_READWRITE, @OldProtection)
    
    ; Trampoline
    CopyMemory(OldAddr, CodeAddress, 5)
    PokeB(CodeAddress + 5, $E9)
    PokeL(CodeAddress + 6, RetAddr - (CodeAddress + 10))
    
    ; JMP To Detour Function
    PokeB(OldAddr + 0, $E9)
    PokeL(OldAddr + 1, ((NewAddr - OldAddr) - 5))
    
    VirtualProtect_(OldAddr, 5, OldProtection, @OldProtection)
    
    ; Update Prototype Pointer To Point To Trampoline
    PokeL(OldFunction, CodeAddress)
    
    ProcedureReturn #True
    
  EndProcedure
  
  ; --------------------------------------------------
  ;   Remove Hook
  ; --------------------------------------------------
  
  Procedure.b InlineUnhook(LibraryName.s, LibraryFunction.s, NewFunction.l, OldFunction.l)
    
    Protected.l OldAddr, CodeAddress, OldProtection
    
    OldAddr = GetProcAddress_(GetModuleHandle_(LibraryName), LibraryFunction)
    CodeAddress = PeekL(OldFunction)
    
    If CodeAddress = #Null
      ProcedureReturn #False
    EndIf
    
    ; Restore Bytes Of Original Function
    VirtualProtect_(OldAddr, 5, #PAGE_EXECUTE_READWRITE, @OldProtection)
    CopyMemory(CodeAddress, OldAddr, 5)
    VirtualProtect_(OldAddr, 5, OldProtection, @OldProtection)
    
    ; Free Trampoline
    FillMemory(CodeAddress, 10, 0)
    VirtualFreeEx_(GetCurrentProcess_(), CodeAddress, 0, #MEM_RELEASE)
    
    ; Update Prototype Pointer To Point To Hook Function
    PokeL(OldFunction, NewFunction)
    
    ProcedureReturn #True
    
  EndProcedure
  
EndModule
Example.pb

Code: Select all

XIncludeFile "Detour.pbi"

Prototype.l RealMessageBoxW(hWnd.l, lpText.s, lpCaption.s, uType.l)

Global o_MessageBoxW.RealMessageBoxW

Procedure.l MyMessageBoxW(hWnd.l, lpText.s, lpCaption.s, uType.l)
  ProcedureReturn o_MessageBoxW(hWnd, "MessageBoxW Is Now Hooked!", lpCaption, uType)
EndProcedure

CHook::InlineHook("User32.dll", "MessageBoxW", @MyMessageBoxW(), @o_MessageBoxW)
MessageRequester("Demo", "This Is Test A!")

CHook::InlineUnhook("User32.dll", "MessageBoxW", @MyMessageBoxW(), @o_MessageBoxW)
MessageRequester("Demo", "This Is Test B!")
Last edited by Opcode on Thu Oct 24, 2019 2:14 am, edited 3 times in total.
Mistrel
Addict
Addict
Posts: 3415
Joined: Sat Jun 30, 2007 8:04 pm

Re: Extended Inline API Hooking

Post by Mistrel »

I'm not well versed in these kind of things. The example works but it fails for detouring GetTickCount() with an invalid memory access:

Code: Select all

Prototype.l o_GetTickCount()

Global o_GetTickCount.o_GetTickCount

Procedure.l MyGetTickCount()
  ProcedureReturn o_GetTickCount()
EndProcedure

CHook::InlineHook("Kernel32.dll","GetTickCount",@MyGetTickCount(),@o_GetTickCount)

MyGetTickCount()
This function doesn't seem to work with the solution from the hot patching thread either:

viewtopic.php?f=12&t=47406
Opcode
Enthusiast
Enthusiast
Posts: 137
Joined: Thu Jul 18, 2013 4:58 am

Re: Extended Inline API Hooking

Post by Opcode »

Mistrel wrote:I'm not well versed in these kind of things. The example works but it fails for detouring GetTickCount() with an invalid memory access:

Code: Select all

Prototype.l o_GetTickCount()

Global o_GetTickCount.o_GetTickCount

Procedure.l MyGetTickCount()
  ProcedureReturn o_GetTickCount()
EndProcedure

CHook::InlineHook("Kernel32.dll","GetTickCount",@MyGetTickCount(),@o_GetTickCount)

MyGetTickCount()
This function doesn't seem to work with the solution from the hot patching thread either:

viewtopic.php?f=12&t=47406
This should work, however retrieving the real tick count via o_GetTickCount() causes a crash. My guess is it gets patched by jmp unconditionally (e.g. hook has to be removed to call it).

Code: Select all

XIncludeFile "Detour.pbi"

Prototype.i RealGetTickCount()

Global o_GetTickCount.RealGetTickCount

Procedure.i MyGetTickCount()
  ProcedureReturn 1337
EndProcedure

Debug GetTickCount_() ; Will display real tick count.

CHook::InlineHook("Kernel32.dll", "GetTickCount", @MyGetTickCount(), @o_GetTickCount)

Debug GetTickCount_() ; Will display 1337 from detour.
Mistrel
Addict
Addict
Posts: 3415
Joined: Sat Jun 30, 2007 8:04 pm

Re: Extended Inline API Hooking

Post by Mistrel »

Is there a way to fix this to also support calling the old function? I want to adjust the result and need to call the original function for as a reference.
User avatar
Mijikai
Addict
Addict
Posts: 1360
Joined: Sun Sep 11, 2016 2:17 pm

Re: Extended Inline API Hooking

Post by Mijikai »

Mistrel wrote:Is there a way to fix this to also support calling the old function? I want to adjust the result and need to call the original function for as a reference.
Either unhook the function, call it and rehook (bad) or you just jump over the opcodes
of the hook by adjusting the function address.

Or be really lazy and call/use the Nt/Rtl/syscall counterpart.

Or hook the IAT to easily call the original function.
Mistrel
Addict
Addict
Posts: 3415
Joined: Sat Jun 30, 2007 8:04 pm

Re: Extended Inline API Hooking

Post by Mistrel »

I don't understand how to do any of that. Except maybe call the Nt counterpart.
Opcode
Enthusiast
Enthusiast
Posts: 137
Joined: Thu Jul 18, 2013 4:58 am

Re: Extended Inline API Hooking

Post by Opcode »

Mistrel wrote:I don't understand how to do any of that. Except maybe call the Nt counterpart.
That's one possibility.

Code: Select all

XIncludeFile "Detour.pbi"

Prototype.l RealGetTickCount()

Global o_GetTickCount.RealGetTickCount

Procedure.l MyGetTickCount()
  
  Protected.l Result = NtGetTickCount_()
  ProcedureReturn Result / 2
  
EndProcedure

Debug GetTickCount_() ; Returns real tick count.

CHook::InlineHook("Kernel32.dll", "GetTickCount", @MyGetTickCount(), @o_GetTickCount)

Debug GetTickCount_() ; Returns real tick count / 2.
Post Reply