LyScript 實現Hook改寫MessageBox

LyScript 可實現自定義彙編指令的替換功能，用戶可以自行編寫一段彙編指令，將程序中特定的通用函數進行功能改寫與轉向操作，此功能原理是簡單的Hook操作。

首先我們先來實現一個Hook模板，在代碼中實現中轉機制，如下代碼以MessageBoxA函數爲案例實現修改彙編參數傳遞。

from LyScript32 import MyDebug

# 傳入彙編列表,寫出到內存
def assemble(dbg, address=0, asm_list=[]):
    asm_len_count = 0
    for index in range(0,len(asm_list)):
        # 寫出到內存
        dbg.assemble_at(address, asm_list[index])
        # print("地址: {} --> 長度計數器: {} --> 寫出: {}".format(hex(address + asm_len_count), asm_len_count,asm_list[index]))
        # 得到asm長度
        asm_len_count = dbg.assemble_code_size(asm_list[index])
        # 地址每次遞增
        address = address + asm_len_count

if __name__ == "__main__":
    dbg = MyDebug()
    connect_flag = dbg.connect()
    print("連接狀態: {}".format(connect_flag))

    # 找到MessageBoxA
    messagebox_address = dbg.get_module_from_function("user32.dll","MessageBoxA")
    print("MessageBoxA內存地址 = {}".format(hex(messagebox_address)))

    # 分配空間
    HookMem = dbg.create_alloc(1024)
    print("自定義內存空間: {}".format(hex(HookMem)))

    # 寫出FindWindowA內存地址,跳轉地址
    asm = [
        f"push {hex(HookMem)}",
        "ret"
    ]

    # 將列表中的彙編指令寫出到內存
    assemble(dbg,messagebox_address,asm)

    dbg.close()

上方代碼中可以看到，首先獲取到MessageBoxA函數內存地址，然後我們通過dbg.create_alloc(1024)分配一段空間，並利用assemble()函數寫出一個跳轉指令。

此段代碼執行後，MessageBoxA處的指令將被替換，跳轉到我們自己分配的內存中去。

接着我們就來實現功能改寫，將彈窗中的消息替換成我們自己的版權信息，此處先給出代碼。

from LyScript32 import MyDebug

# 傳入彙編列表,寫出到內存
def assemble(dbg, address=0, asm_list=[]):
    asm_len_count = 0
    for index in range(0,len(asm_list)):
        # 寫出到內存
        dbg.assemble_at(address, asm_list[index])
        # print("地址: {} --> 長度計數器: {} --> 寫出: {}".format(hex(address + asm_len_count), asm_len_count,asm_list[index]))
        # 得到asm長度
        asm_len_count = dbg.assemble_code_size(asm_list[index])
        # 地址每次遞增
        address = address + asm_len_count

if __name__ == "__main__":
    dbg = MyDebug()
    connect_flag = dbg.connect()
    print("連接狀態: {}".format(connect_flag))

    # 找到MessageBoxA
    messagebox_address = dbg.get_module_from_function("user32.dll","MessageBoxA")
    print("MessageBoxA內存地址 = {}".format(hex(messagebox_address)))

    # 分配空間
    HookMem = dbg.create_alloc(1024)
    print("自定義內存空間: {}".format(hex(HookMem)))

    # 寫出FindWindowA內存地址,跳轉地址
    asm = [
        f"push {hex(HookMem)}",
        "ret"
    ]

    # 將列表中的彙編指令寫出到內存
    assemble(dbg,messagebox_address,asm)

    # 定義兩個變量,存放字符串
    MsgBoxAddr = dbg.create_alloc(512)
    MsgTextAddr = dbg.create_alloc(512)

    # 填充字符串內容
    # lyshark 標題
    txt = [0x6c, 0x79, 0x73, 0x68, 0x61, 0x72, 0x6b]
    # 內容 lyshark.com
    box = [0x6C, 0x79, 0x73, 0x68, 0x61, 0x72, 0x6B, 0x2E, 0x63, 0x6F, 0x6D]

    for txt_count in range(0,len(txt)):
        dbg.write_memory_byte(MsgBoxAddr + txt_count, txt[txt_count])

    for box_count in range(0,len(box)):
        dbg.write_memory_byte(MsgTextAddr + box_count, box[box_count])

    print("標題地址: {} 內容: {}".format(hex(MsgBoxAddr),hex(MsgTextAddr)))

    # 此處是MessageBox替換後的片段
    PatchCode =\
    [
        "mov edi, edi",
        "push ebp",
        "mov ebp,esp",
        "push -1",
        "push 0",
        "push dword ptr ss:[ebp+0x14]",
        f"push {hex(MsgBoxAddr)}",
        f"push {hex(MsgTextAddr)}",
        "push dword ptr ss:[ebp+0x8]",
        "call 0x76030E20",
        "pop ebp",
        "ret 0x10"
    ]

    # 寫出到自定義內存
    assemble(dbg, HookMem, PatchCode)

    print("地址已被替換,可以運行了.")
    dbg.set_debug("Run")
    dbg.set_debug("Run")

    dbg.close()

首先程序運行後，會經過assemble(dbg,messagebox_address,asm)彙編寫出的位置，此處是一個跳轉，直接跳轉到我們自己申請的內存空間內。

當EIP走到此處後，跳轉到我們自己構建的彈窗位置，此處的代碼如下。

當彈窗運行後，讀入的內存地址有兩處MsgBoxAddr是消息MsgTextAddr是文本，這兩處位置都被python中的push {hex()}替換掉了，當運行彈窗後，就是執行我們自己的函數。