mirror of https://github.com/pwndbg/pwndbg.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
98 lines
4.1 KiB
Python
98 lines
4.1 KiB
Python
import tests
|
|
from pwndbg.commands.nearpc import emulate
|
|
from pwndbg.commands.nearpc import nearpc
|
|
from pwndbg.commands.nearpc import pdisass
|
|
from pwndbg.commands.windbg import u
|
|
|
|
EMULATE_DISASM_BINARY = tests.binaries.get("emulate_disasm.out")
|
|
EMULATE_DISASM_LOOP_BINARY = tests.binaries.get("emulate_disasm_loop.out")
|
|
|
|
|
|
def test_emulate_disasm(start_binary):
|
|
"""
|
|
Tests emulate command and its caching behavior
|
|
"""
|
|
start_binary(EMULATE_DISASM_BINARY)
|
|
|
|
disasm_with_emu_0x400080 = [
|
|
" ► 0x400080 <_start> jmp label <label>",
|
|
" ↓",
|
|
" 0x400083 <label> nop ",
|
|
" 0x400084 add byte ptr [rax], al",
|
|
" 0x400086 add byte ptr [rax], al",
|
|
" 0x400088 add byte ptr [rax], al",
|
|
" 0x40008a add byte ptr [rax], al",
|
|
" 0x40008c add byte ptr [rax], al",
|
|
" 0x40008e add byte ptr [rax], al",
|
|
" 0x400090 add byte ptr [rax], al",
|
|
" 0x400092 add byte ptr [rax], al",
|
|
" 0x400094 add byte ptr [rax], al",
|
|
]
|
|
|
|
disasm_without_emu_0x400080 = [
|
|
" ► 0x400080 <_start> jmp label <label>",
|
|
" ",
|
|
" 0x400082 <_start+2> nop ",
|
|
" 0x400083 <label> nop ",
|
|
" 0x400084 add byte ptr [rax], al",
|
|
" 0x400086 add byte ptr [rax], al",
|
|
" 0x400088 add byte ptr [rax], al",
|
|
" 0x40008a add byte ptr [rax], al",
|
|
" 0x40008c add byte ptr [rax], al",
|
|
" 0x40008e add byte ptr [rax], al",
|
|
" 0x400090 add byte ptr [rax], al",
|
|
" 0x400092 add byte ptr [rax], al",
|
|
]
|
|
|
|
compare_output_emu(disasm_with_emu_0x400080)
|
|
compare_output_without_emu(disasm_without_emu_0x400080)
|
|
|
|
|
|
def test_emulate_disasm_loop(start_binary):
|
|
start_binary(EMULATE_DISASM_LOOP_BINARY)
|
|
|
|
disasm_with_emu_0x400080 = [
|
|
" ► 0x400080 <_start> movabs rsi, string <0x400094>",
|
|
" 0x40008a <_start+10> mov rdi, rsp",
|
|
" 0x40008d <_start+13> mov ecx, 3",
|
|
" 0x400092 <_start+18> rep movsb byte ptr [rdi], byte ptr [rsi]",
|
|
" ↓",
|
|
" 0x400092 <_start+18> rep movsb byte ptr [rdi], byte ptr [rsi]",
|
|
" ↓",
|
|
" 0x400092 <_start+18> rep movsb byte ptr [rdi], byte ptr [rsi]",
|
|
" ↓",
|
|
" 0x400092 <_start+18> rep movsb byte ptr [rdi], byte ptr [rsi]",
|
|
" 0x400094 <string> xor dword ptr [rdx], esi",
|
|
" 0x400096 <string+2> xor esi, dword ptr [rsi]",
|
|
" 0x40009d add byte ptr [rax], al",
|
|
" 0x40009f add byte ptr [rax], al",
|
|
]
|
|
|
|
disasm_without_emu_0x400080 = [
|
|
" ► 0x400080 <_start> movabs rsi, string <0x400094>",
|
|
" 0x40008a <_start+10> mov rdi, rsp",
|
|
" 0x40008d <_start+13> mov ecx, 3",
|
|
" 0x400092 <_start+18> rep movsb byte ptr [rdi], byte ptr [rsi]",
|
|
" 0x400094 <string> xor dword ptr [rdx], esi",
|
|
" 0x400096 <string+2> xor esi, dword ptr [rsi]",
|
|
" 0x40009d add byte ptr [rax], al",
|
|
" 0x40009f add byte ptr [rax], al",
|
|
" 0x4000a1 add byte ptr [rax], al",
|
|
" 0x4000a3 add byte ptr [rax], al",
|
|
" 0x4000a5 add byte ptr [rax], al",
|
|
]
|
|
|
|
compare_output_emu(disasm_with_emu_0x400080)
|
|
compare_output_without_emu(disasm_without_emu_0x400080)
|
|
|
|
|
|
def compare_output_emu(emu_0x400080):
|
|
assert emulate(to_string=True) == emu_0x400080
|
|
|
|
|
|
def compare_output_without_emu(emu_0x400080):
|
|
assert nearpc(to_string=True) == emu_0x400080
|
|
assert emulate(to_string=True, emulate=False) == emu_0x400080
|
|
assert pdisass(to_string=True) == emu_0x400080
|
|
assert u(to_string=True) == emu_0x400080
|