diff options
Diffstat (limited to 'test/test.rs')
| -rw-r--r-- | test/test.rs | 401 |
1 files changed, 2 insertions, 399 deletions
diff --git a/test/test.rs b/test/test.rs index 177e748..7dd54ea 100644 --- a/test/test.rs +++ b/test/test.rs @@ -5,402 +5,5 @@ extern crate test; extern crate yaxpeax_arch; extern crate yaxpeax_arm; -use yaxpeax_arch::{Decodable, LengthedInstruction}; -use yaxpeax_arm::armv7::{ARMv7, Instruction, ConditionCode, Operands, Opcode, ShiftSpec}; - -fn test_decode(data: [u8; 4], expected: Instruction) { - let mut instr = Instruction::blank(); - instr.decode_into(data.to_vec()); - assert!( - instr == expected, - "decode error for {:02x}{:02x}{:02x}{:02x}:\n decoded: {:?}\n expected: {:?}\n", - data[0], data[1], data[2], data[3], - instr, expected - ); -} - -fn test_display(data: [u8; 4], expected: &'static str) { - let mut instr = Instruction::blank(); - instr.decode_into(data.to_vec()); - let text = format!("{}", instr); - assert!( - text == expected, - "display error for {:02x}{:02x}{:02x}{:02x}:\n decoded: {:?}\n displayed: {}\n expected: {}\n", - data[0], data[1], data[2], data[3], - instr, - text, expected - ); -} - -#[test] -fn test_decode_str_ldr() { - test_decode( - [0x24, 0xc0, 0x9f, 0xe5], - Instruction { - condition: ConditionCode::AL, - opcode: Opcode::LDR(true, true, false), - operands: Operands::RegImm(12, 0x24), - s: false - } - ); - test_decode( - [0x10, 0x00, 0x9f, 0xe5], - Instruction { - condition: ConditionCode::AL, - opcode: Opcode::LDR(true, true, false), - operands: Operands::RegImm(0, 0x10), - s: false - } - ); - test_decode( - [0x04, 0x20, 0x2d, 0xe5], - Instruction { - condition: ConditionCode::AL, - opcode: Opcode::STR(false, true, true), - operands: Operands::TwoRegImm(13, 2, 4), - s: false - } - ); - test_decode( - [0x04, 0x00, 0x2d, 0xe5], - Instruction { - condition: ConditionCode::AL, - opcode: Opcode::STR(false, true, true), - operands: Operands::TwoRegImm(13, 0, 4), - s: false - } - ); - test_decode( - [0x14, 0x30, 0x9f, 0xe5], - Instruction { - condition: ConditionCode::AL, - opcode: Opcode::LDR(true, true, false), - operands: Operands::RegImm(3, 0x14), - s: false - } - ); - test_decode( - [0x14, 0x20, 0x9f, 0xe5], - Instruction { - condition: ConditionCode::AL, - opcode: Opcode::LDR(true, true, false), - operands: Operands::RegImm(2, 0x14), - s: false - } - ); -} - -#[test] -fn test_decode_misc() { - test_display( - [0x02, 0x00, 0xa0, 0xe3], - "mov r0, 0x2" - ); - test_display( - [0xe8, 0x10, 0x9f, 0xe5], - "ldr r1, [pc, #0x3a0]" - ); -} - -#[test] -fn test_decode_pop() { - test_decode( - [0x04, 0x10, 0x9d, 0xe4], - Instruction { - condition: ConditionCode::AL, - opcode: Opcode::LDR(true, false, false), - operands: Operands::TwoRegImm(13, 1, 4), - s: false - } - ); - test_display( - [0x04, 0x10, 0x9d, 0xe4], - "pop {r1}" - ); - test_decode( - [0xf0, 0x40, 0x2d, 0xe9], - Instruction { - condition: ConditionCode::AL, - opcode: Opcode::STM(false, true, true, false), - operands: Operands::RegRegList(13, 16624), - s: false - } - ); - test_display( - [0xf0, 0x40, 0x2d, 0xe9], - "push {r4, r5, r6, r7, lr}" - ); - test_decode( - [0xf0, 0x80, 0xbd, 0x18], - Instruction { - condition: ConditionCode::NE, - opcode: Opcode::LDM(true, false, true, false), - operands: Operands::RegRegList(13, 33008), - s: false - } - ); - test_display( - [0xf0, 0x80, 0xbd, 0x18], - "popne {r4, r5, r6, r7, pc}" - ); -} - -#[test] -fn test_decode_mov() { - test_decode( - [0x0d, 0x20, 0xa0, 0xe1], - Instruction { - condition: ConditionCode::AL, - opcode: Opcode::MOV, - operands: Operands::TwoOperand(2, 13), - s: false - } - ); - test_decode( - [0x00, 0xb0, 0xa0, 0xe3], - Instruction { - condition: ConditionCode::AL, - opcode: Opcode::MOV, - operands: Operands::RegImm(11, 0), - s: false - } - ); -} - -#[test] -fn test_decode_arithmetic() { - test_decode( - [0x03, 0x30, 0x8f, 0xe0], - Instruction { - condition: ConditionCode::AL, - opcode: Opcode::ADD, - operands: Operands::ThreeOperand(3, 15, 3), - s: false - } - ); - test_decode( - [0x03, 0x30, 0x66, 0xe0], - Instruction { - condition: ConditionCode::AL, - opcode: Opcode::RSB, - operands: Operands::ThreeOperand(3, 6, 3), - s: false - } - ); - test_decode( - [0x43, 0x31, 0xa0, 0xe1], - Instruction { - condition: ConditionCode::AL, - opcode: Opcode::MOV, - operands: Operands::ThreeOperandWithShift(3, 0, 3, ShiftSpec::Immediate(10)), - s: false - } - ); - test_decode( - [0x01, 0x50, 0x43, 0xe2], - Instruction { - condition: ConditionCode::AL, - opcode: Opcode::SUB, - operands: Operands::RegImm(3, 20481), - s: false - } - ); -} - -#[test] -fn test_decode_mul() { - test_decode( - [0x9c, 0x7d, 0x0b, 0x00], - Instruction { - condition: ConditionCode::EQ, - opcode: Opcode::MUL, - operands: Operands::MulThreeRegs(11, 12, 13), - s: false - } - ); - test_decode( - [0x90, 0x79, 0x09, 0x00], - Instruction { - condition: ConditionCode::EQ, - opcode: Opcode::MUL, - operands: Operands::MulThreeRegs(9, 0, 9), - s: false - } - ); - test_decode( - [0x94, 0x79, 0x09, 0x00], - Instruction { - condition: ConditionCode::EQ, - opcode: Opcode::MUL, - operands: Operands::MulThreeRegs(9, 4, 9), - s: false - } - ); -} - -static instruction_bytes: [u8; 4 * 60] = [ - 0x24, 0xc0, 0x9f, 0xe5, - 0x00, 0xb0, 0xa0, 0xe3, - 0x04, 0x10, 0x9d, 0xe4, - 0x0d, 0x20, 0xa0, 0xe1, - 0x04, 0x20, 0x2d, 0xe5, - 0x04, 0x00, 0x2d, 0xe5, - 0x10, 0x00, 0x9f, 0xe5, - 0x10, 0x30, 0x9f, 0xe5, - 0x04, 0xc0, 0x2d, 0xe5, - 0x4b, 0xfe, 0xff, 0xeb, - 0xd5, 0xfd, 0xff, 0xeb, - 0x90, 0x79, 0x09, 0x00, - 0x64, 0xd0, 0x01, 0x00, - 0x94, 0x79, 0x09, 0x00, - 0x14, 0x30, 0x9f, 0xe5, - 0x14, 0x20, 0x9f, 0xe5, - 0x03, 0x30, 0x8f, 0xe0, - 0x02, 0x10, 0x93, 0xe7, - 0x00, 0x00, 0x51, 0xe3, - 0x0e, 0xf0, 0xa0, 0x01, - 0x01, 0xfe, 0xff, 0xea, - 0x58, 0x75, 0x09, 0x00, - 0xec, 0x02, 0x00, 0x00, - 0xf0, 0x40, 0x2d, 0xe9, - 0x54, 0x70, 0x9f, 0xe5, - 0x00, 0x30, 0xd7, 0xe5, - 0x00, 0x00, 0x53, 0xe3, - 0xf0, 0x80, 0xbd, 0x18, - 0x48, 0x60, 0x9f, 0xe5, - 0x48, 0x30, 0x9f, 0xe5, - 0x48, 0x40, 0x9f, 0xe5, - 0x03, 0x30, 0x66, 0xe0, - 0x43, 0x31, 0xa0, 0xe1, - 0x00, 0x20, 0x94, 0xe5, - 0x01, 0x50, 0x43, 0xe2, - 0x05, 0x00, 0x52, 0xe1, - 0x06, 0x00, 0x00, 0x2a, - 0x01, 0x30, 0x82, 0xe2, - 0x00, 0x30, 0x84, 0xe5, - 0x0f, 0xe0, 0xa0, 0xe1, - 0x03, 0xf1, 0x96, 0xe7, - 0x00, 0x20, 0x94, 0xe5, - 0x05, 0x00, 0x52, 0xe1, - 0xf8, 0xff, 0xff, 0x3a, - 0x01, 0x30, 0xa0, 0xe3, - 0x00, 0x30, 0xc7, 0xe5, - 0xf0, 0x80, 0xbd, 0xe8, - 0x9c, 0x7d, 0x0b, 0x00, - 0xa0, 0x33, 0x0b, 0x00, - 0xa4, 0x33, 0x0b, 0x00, - 0xa0, 0x7d, 0x0b, 0x00, - 0x04, 0xe0, 0x2d, 0xe5, - 0x04, 0xf0, 0x9d, 0xe4, - 0x24, 0x00, 0x9f, 0xe5, - 0x00, 0x30, 0x90, 0xe5, - 0x00, 0x00, 0x53, 0xe3, - 0x04, 0xe0, 0x2d, 0xe5, - 0x04, 0xf0, 0x9d, 0x04, - 0x14, 0x30, 0x9f, 0xe5, - 0x00, 0x00, 0x53, 0xe3 - ]; - - -#[test] -fn test_decode_span() { - let mut i = 0u32; - while i < instruction_bytes.len() as u32 { - let mut instr = Instruction::blank(); - instr.decode_into(instruction_bytes[(i as usize)..].iter().map(|x| *x)); - println!( - "Decoded {:02x}{:02x}{:02x}{:02x}: {}", //{:?}\n {}", - instruction_bytes[i as usize], - instruction_bytes[i as usize + 1], - instruction_bytes[i as usize + 2], - instruction_bytes[i as usize + 3], -// instr, - instr); - i += instr.len(); - } - panic!("done"); -} -/* - * from debian 5.0.10 bash 3.2-4_arm - * 0x0001bee4 24c09fe5 ldr ip, sym.__libc_csu_fini - * 0x0001bee8 00b0a0e3 mov fp, 0 - * 0x0001beec 04109de4 pop {r1} - * 0x0001bef0 0d20a0e1 mov r2, sp - * 0x0001bef4 04202de5 str r2, [sp, -4]! - * 0x0001bef8 04002de5 str r0, [sp, -4]! - * 0x0001befc 10009fe5 ldr r0, sym.main - * 0x0001bf00 10309fe5 ldr r3, sym.__libc_csu_init - * 0x0001bf04 04c02de5 str ip, [sp, -4]! - * 0x0001bf08 4bfeffeb bl sym.imp.__libc_start_main - * 0x0001bf0c d5fdffeb bl sym.imp.abort - * 0x0001bf10 90790900 muleq sb, r0, sb - * 0x0001bf14 64d00100 andeq sp, r1, r4, rrx - * 0x0001bf18 94790900 muleq sb, r4, sb - * 0x0001bf1c 14309fe5 ldr r3, [0x0001bf38] - * 0x0001bf20 14209fe5 ldr r2, [0x0001bf3c] - * 0x0001bf24 03308fe0 add r3, pc, r3 - * 0x0001bf28 021093e7 ldr r1, [r3, r2] - * 0x0001bf2c 000051e3 cmp r1, 0 - * 0x0001bf30 0ef0a001 moveq pc, lr - * 0x0001bf34 01feffea b loc.imp.__gmon_start__ - * 0x0001bf38 58750900 andeq r7, sb, r8, asr r5 - * 0x0001bf3c ec020000 andeq r0, r0, ip, ror 5 - * 0x0001bf40 f0402de9 push {r4, r5, r6, r7, lr} - * 0x0001bf44 54709fe5 ldr r7, [0x0001bfa0] - * 0x0001bf48 0030d7e5 ldrb r3, [r7] - * 0x0001bf4c 000053e3 cmp r3, 0 - * 0x0001bf50 f080bd18 popne {r4, r5, r6, r7, pc} - * 0x0001bf54 48609fe5 ldr r6, [0x0001bfa4] - * 0x0001bf58 48309fe5 ldr r3, [0x0001bfa8] - * 0x0001bf5c 48409fe5 ldr r4, [0x0001bfac] - * 0x0001bf60 033066e0 rsb r3, r6, r3 - * 0x0001bf64 4331a0e1 asr r3, r3, 2 - * 0x0001bf68 002094e5 ldr r2, [r4] - * 0x0001bf6c 015043e2 sub r5, r3, 1 - * 0x0001bf70 050052e1 cmp r2, r5 - * 0x0001bf74 0600002a bhs 0x1bf94 - * 0x0001bf78 013082e2 add r3, r2, 1 - * 0x0001bf7c 003084e5 str r3, [r4] - * 0x0001bf80 0fe0a0e1 mov lr, pc - * 0x0001bf84 03f196e7 ldr pc, [r6, r3, lsl 2] - * 0x0001bf88 002094e5 ldr r2, [r4] - * 0x0001bf8c 050052e1 cmp r2, r5 - * 0x0001bf90 f8ffff3a blo 0x1bf78 - * 0x0001bf94 0130a0e3 mov r3, 1 - * 0x0001bf98 0030c7e5 strb r3, [r7] - * 0x0001bf9c f080bde8 pop {r4, r5, r6, r7, pc} - * 0x0001bfa0 9c7d0b00 muleq fp, ip, sp - * 0x0001bfa4 a0330b00 andeq r3, fp, r0, lsr 7 - * 0x0001bfa8 a4330b00 andeq r3, fp, r4, lsr 7 - * 0x0001bfac a07d0b00 andeq r7, fp, r0, lsr 27 - * 0x0001bfb0 04e02de5 str lr, [sp, -4]! - * 0x0001bfb4 04f09de4 pop {pc} - * 0x0001bfb8 24009fe5 ldr r0, [0x0001bfe4] - * 0x0001bfbc 003090e5 ldr r3, [r0] - * 0x0001bfc0 000053e3 cmp r3, 0 - * 0x0001bfc4 04e02de5 str lr, [sp, -4]! - * 0x0001bfc8 04f09d04 popeq {pc} - * 0x0001bfcc 14309fe5 ldr r3, [0x0001bfe8] - * 0x0001bfd0 000053e3 cmp r3, 0 - */ - -use test::Bencher; -#[bench] -pub fn bench_60000_instrs(b: &mut Bencher) { - b.iter(|| { - for i in (0..1000) { - let mut iter = instruction_bytes.iter().map(|x| *x); - let mut result = Instruction::blank(); - loop { - match result.decode_into(&mut iter) { - Some(result) => { - test::black_box(&result); - }, - None => { - break; - } - } - } - } - }); -} +// mod armv7; +mod armv8; |