From 0fff2a6aa0832b1cabf381e0c970f0fd47223224 Mon Sep 17 00:00:00 2001 From: iximeow Date: Mon, 22 Mar 2021 23:08:54 -0700 Subject: port long-mode decoder updates to protected-mode --- test/protected_mode/display.rs | 131 +++++++++++++++++++++++++++++++++++++++++ 1 file changed, 131 insertions(+) create mode 100644 test/protected_mode/display.rs (limited to 'test/protected_mode/display.rs') diff --git a/test/protected_mode/display.rs b/test/protected_mode/display.rs new file mode 100644 index 0000000..8426a0a --- /dev/null +++ b/test/protected_mode/display.rs @@ -0,0 +1,131 @@ +use std::fmt::Write; + +use yaxpeax_arch::{AddressBase, Decoder, LengthedInstruction}; +use yaxpeax_x86::protected_mode::{DisplayStyle, InstDecoder}; + +fn test_display(data: &[u8], expected: &'static str) { + test_display_under(&InstDecoder::default(), data, expected); +} + +fn test_display_under(decoder: &InstDecoder, data: &[u8], expected: &'static str) { + let mut hex = String::new(); + for b in data { + write!(hex, "{:02x}", b).unwrap(); + } + match decoder.decode(data.into_iter().map(|x| *x)) { + Ok(instr) => { + let text = format!("{}", instr.display_with(DisplayStyle::C)); + assert!( + text == expected, + "display error for {}:\n decoded: {:?} under decoder {}\n displayed: {}\n expected: {}\n", + hex, + instr, + decoder, + text, + expected + ); + // while we're at it, test that the instruction is as long, and no longer, than its + // input + assert_eq!((0u32.wrapping_offset(instr.len()).to_linear()) as usize, data.len(), "instruction length is incorrect, wanted instruction {}", expected); + }, + Err(e) => { + assert!(false, "decode error ({}) for {} under decoder {}:\n expected: {}\n", e, hex, decoder, expected); + } + } +} + +// decided i do not like at&t syntax much at all. not going to write a formatter for it. some test +// cases will live on in case someone else feels like adding one, or i get mad enough to do it. +#[ignore] +#[test] +fn test_instructions_atnt() { + // just modrm + test_display(&[0x33, 0x08], "xor (%eax), %ecx"); + test_display(&[0x33, 0x20], "xor (%eax), %esp"); + test_display(&[0x33, 0x05, 0x78, 0x56, 0x34, 0x12], "xor (0x12345678), %eax"); + test_display(&[0x33, 0x41, 0x23], "xor 0x23(%ecx), %eax"); + test_display(&[0x33, 0x81, 0x23, 0x01, 0x65, 0x43], "xor %0x43650123, %eax"); + test_display(&[0x33, 0xc1], "xor %ecx, %eax"); + + // sib + test_display(&[0x33, 0x04, 0x25, 0x11, 0x22, 0x33, 0x44], "xor (0x44332211), %eax"); + test_display(&[0x41, 0x33, 0x04, 0x25, 0x11, 0x22, 0x33, 0x44], "xor (0x44332211), %eax"); + + test_display(&[0x33, 0x44, 0x65, 0x11], "xor 0x11(%r13), %eax"); + + test_display(&[0x33, 0x34, 0x25, 0x20, 0x30, 0x40, 0x50], "xor 0x50403020, %esi"); + + test_display(&[0x0f, 0xe7, 0x03], "movntq %mm0, (%ebx)"); + + test_display(&[0x0f, 0x7f, 0x0f], "movq %mm1, (%edi)"); + test_display(&[0x0f, 0xc4, 0xc0, 0x14], "pinsrw $0x14, %eax, %mm0"); + + test_display(&[0x0f, 0xd1, 0x00], "psrlw (%eax), %mm0"); + test_display(&[0x0f, 0xe5, 0x3d, 0xaa, 0xbb, 0xcc, 0x77], "pmulhw 0x77ccbbaa, %mm7"); +} + +#[test] +fn test_instructions_c() { + // just modrm + test_display(&[0x33, 0x08], "ecx ^= [eax]"); + test_display(&[0x33, 0x20], "esp ^= [eax]"); + test_display(&[0x33, 0x05, 0x78, 0x56, 0x34, 0x12], "eax ^= [0x12345678]"); + test_display(&[0x33, 0x41, 0x23], "eax ^= [ecx + 0x23]"); + test_display(&[0x33, 0x81, 0x23, 0x01, 0x65, 0x43], "eax ^= [ecx + 0x43650123]"); + test_display(&[0x33, 0xc1], "eax ^= ecx"); + + // sib + test_display(&[0x33, 0x04, 0x25, 0x11, 0x22, 0x33, 0x44], "eax ^= [0x44332211]"); + + test_display(&[0x33, 0x44, 0x65, 0x11], "eax ^= [ebp + 0x11]"); + + test_display(&[0x33, 0x34, 0x25, 0x20, 0x30, 0x40, 0x50], "esi ^= [0x50403020]"); + + test_display(&[0x0f, 0xe7, 0x03], "[ebx] = movntq(mm0)"); + + test_display(&[0x0f, 0x7f, 0x0f], "[edi] = movq(mm1)"); + test_display(&[0x0f, 0xc4, 0xc0, 0x14], "mm0 = pinsrw(mm0, eax, 0x14)"); + + test_display(&[0x0f, 0xd1, 0x00], "mm0 = psrlw(mm0, [eax])"); + test_display(&[0x0f, 0xe5, 0x3d, 0xaa, 0xbb, 0xcc, 0x77], "mm7 = pmulhw(mm7, [0x77ccbbaa])"); + + test_display(&[0xf3, 0xa5], "rep dword { es:[edi++] = ds:[esi++] }"); + test_display(&[0xf3, 0x66, 0xa5], "rep word { es:[edi++] = ds:[esi++] }"); + test_display(&[0xf3, 0xa4], "rep byte { es:[edi++] = ds:[esi++] }"); + + test_display(&[0xf6, 0xc2, 0x18], "eflags = flags(dl & 0x18)"); + test_display(&[0xf6, 0xc2, 0x18], "eflags = flags(dl & 0x18)"); + test_display(&[0x84, 0xc0], "eflags = flags(al & al)"); + test_display(&[0x85, 0xc0], "eflags = flags(eax & eax)"); + test_display(&[0x3a, 0xc0], "eflags = flags(al - al)"); + test_display(&[0x3b, 0xc0], "eflags = flags(eax - eax)"); + + test_display(&[0x0f, 0xbc, 0xd3], "edx = lsb(ebx) (x86 bsf)"); + test_display(&[0xf3, 0x0f, 0xbc, 0xd3], "edx = lsb(ebx)"); + // test_display(&[0x41, 0x0f, 0xbc, 0xd3], "edx = lsb(ebx) (x86 bsf"); // for non-bm1 + test_display(&[0x0f, 0xbd, 0xd3], "edx = msb(ebx)"); + // test_display(&[0x41, 0x0f, 0xbc, 0xd3], "edx = lsb(ebx) (x86 bsr"); // for non-bm1 + test_display(&[0xd2, 0xc0], "al = al rol cl"); + test_display(&[0xd2, 0xc8], "al = al ror cl"); + test_display(&[0xd2, 0xd0], "al = al rcl cl"); + test_display(&[0xd2, 0xd8], "al = al rcr cl"); + test_display(&[0xd2, 0xe0], "al = al << cl"); + test_display(&[0xd2, 0xe8], "al = al >> cl"); + test_display(&[0xd2, 0xf0], "al = al <<< cl"); + test_display(&[0xd2, 0xf8], "al = al >>> cl"); + + test_display(&[0xc4, 0xc3, 0x7b, 0xf0, 0x01, 0x05], "eax = [ecx] ror 0x5 (x86 rorx)"); + test_display(&[0xc4, 0xc2, 0xe3, 0xf7, 0x01], "eax = [ecx] >> ebx (x86 shrx)"); + test_display(&[0xc4, 0xc2, 0xe1, 0xf7, 0x01], "eax = [ecx] << ebx (x86 shlx)"); + + test_display(&[0xd2, 0xe0], "al = al << cl"); + + test_display(&[0x66, 0x0f, 0xac, 0xcf, 0x11], "di = shrd(di, cx, 0x11)"); + test_display(&[0x0f, 0xa5, 0xc9], "ecx = shld(ecx, ecx, cl)"); + + test_display(&[0x66, 0x0f, 0x38, 0xf6, 0x01], "eax += [ecx] + eflags.cf"); + + test_display(&[0xfe, 0x00], "byte [eax]++"); + test_display(&[0x66, 0xff, 0x08], "word [eax]--"); + test_display(&[0xff, 0x00], "dword [eax]++"); +} -- cgit v1.1