diff options
Diffstat (limited to 'test')
-rw-r--r-- | test/lib_test.rs | 6 | ||||
-rw-r--r-- | test/test.rs | 111 |
2 files changed, 117 insertions, 0 deletions
diff --git a/test/lib_test.rs b/test/lib_test.rs new file mode 100644 index 0000000..946e67b --- /dev/null +++ b/test/lib_test.rs @@ -0,0 +1,6 @@ +#[test] +fn test_disasm() { + let mut instr = Instruction::invalid(); + arch::x86_64::instr::decode_one(&[0x33, 0xc0], &mut instr); + assert_eq!(1, 1); +} diff --git a/test/test.rs b/test/test.rs new file mode 100644 index 0000000..c4133e6 --- /dev/null +++ b/test/test.rs @@ -0,0 +1,111 @@ +extern crate x86_rs; + +use x86_rs::{Instruction, Opcode, decode_one}; + +fn decode(bytes: &[u8]) -> Option<Instruction> { + let mut instr = Instruction::invalid(); + match decode_one(bytes, &mut instr) { + Some(()) => Some(instr), + None => None + } +} + +#[test] +fn test_mov() { + assert_eq!(&format!("{}", decode( + &[0x48, 0xc7, 0x04, 0x24, 0x00, 0x00, 0x00, 0x00] + ).unwrap()), "mov [rsp], 0x0"); + assert_eq!(&format!("{}", decode( + &[0x48, 0x89, 0x44, 0x24, 0x08] + ).unwrap()), "mov [rsp + 0x8], rax"); + assert_eq!(&format!("{}", decode( + &[0x48, 0x89, 0x43, 0x18] + ).unwrap()), "mov [rbx + 0x18], rax"); + assert_eq!(&format!("{}", decode( + &[0x48, 0xc7, 0x43, 0x10, 0x00, 0x00, 0x00] + ).unwrap()), "mov [rbx + 0x10], 0x0"); + assert_eq!(&format!("{}", decode( + &[0x49, 0x89, 0x4e, 0x08] + ).unwrap()), "mov [r14 + 0x8], rcx"); + assert_eq!(&format!("{}", decode( + &[0x48, 0x8b, 0x32] + ).unwrap()), "mov rsi, [rdx]"); + assert_eq!(&format!("{}", decode( + &[0x49, 0x89, 0x46, 0x10] + ).unwrap()), "mov [r14 + 0x10], rax"); + assert_eq!(&format!("{}", decode( + &[0x4d, 0x0f, 0x43, 0xec, 0x49] + ).unwrap()), "cmovae r13, r12"); + assert_eq!(&format!("{}", decode( + &[0x0f, 0xb6, 0x06] + ).unwrap()), "movzx eax, byte [rsi]"); +} + +#[test] +fn test_control_flow() { + assert_eq!(&format!("{}", decode( + &[0x73, 0x31] + ).unwrap()), "jnb 0x31"); + assert_eq!(&format!("{}", decode( + &[0x72, 0x5a] + ).unwrap()), "jb 0x5a"); + assert_eq!(&format!("{}", decode( + &[0x0f, 0x86, 0x8b, 0x01, 0x00, 0x00] + ).unwrap()), "jna 0x18b"); + assert_eq!(&format!("{}", decode( + &[0x74, 0x47] + ).unwrap()), "jz 0x47"); + assert_eq!(&format!("{}", decode( + &[0xff, 0x15, 0x7e, 0x72, 0x24, 0x00] + ).unwrap()), "call [rip + 0x24727e]"); + assert_eq!(&format!("{}", decode( + &[0xc3] + ).unwrap()), "ret"); +} + +#[test] +fn test_test_cmp() { + assert_eq!(&format!("{}", decode( + &[0x48, 0x3d, 0x01, 0xf0, 0xff, 0xff] + ).unwrap()), "cmp rax, 0xfffffffffffff001"); + assert_eq!(&format!("{}", decode( + &[0x3d, 0x01, 0xf0, 0xff, 0xff] + ).unwrap()), "cmp eax, 0xfffff001"); + assert_eq!(&format!("{}", decode( + &[0x48, 0x83, 0xf8, 0xff] + ).unwrap()), "cmp rax, 0xffffffffffffffff"); + assert_eq!(&format!("{}", decode( + &[0x48, 0x39, 0xc6] + ).unwrap()), "cmp rsi, rax"); +} + +#[test] +fn test_misc() { + assert_eq!(&format!("{}", decode( + &[0x33, 0xc0] + ).unwrap()), "xor eax, eax"); + assert_eq!(&format!("{}", decode( + &[0x48, 0x8d, 0x53, 0x08] + ).unwrap()), "lea rdx, [rbx + 0x8]"); + assert_eq!(&format!("{}", decode( + &[0x31, 0xc9] + ).unwrap()), "xor ecx, ecx"); + assert_eq!(&format!("{}", decode( + &[0x48, 0x29, 0xc8] + ).unwrap()), "sub rax, rcx"); + assert_eq!(&format!("{}", decode( + &[0x48, 0x03, 0x0b] + ).unwrap()), "add rcx, [rbx]"); + assert_eq!(&format!("{}", decode( + &[0x5b] + ).unwrap()), "pop rbx"); + assert_eq!(&format!("{}", decode( + &[0x41, 0x5e] + ).unwrap()), "pop r14"); + assert_eq!(&format!("{}", decode( + &[0x48, 0x8d, 0x0c, 0x12] + ).unwrap()), "lea rcx, [rdx + rdx]"); + assert_eq!(&format!("{}", decode( + &[0xf6, 0xc2, 0x18] + ).unwrap()), "test dl, 0x18"); +} |