diff options
-rw-r--r-- | differential-tests/tests/capstone-differential.rs | 2 | ||||
-rw-r--r-- | tests/test.rs | 93 |
2 files changed, 48 insertions, 47 deletions
diff --git a/differential-tests/tests/capstone-differential.rs b/differential-tests/tests/capstone-differential.rs index 3e76321..ea74f50 100644 --- a/differential-tests/tests/capstone-differential.rs +++ b/differential-tests/tests/capstone-differential.rs @@ -608,7 +608,7 @@ fn capstone_differential() { */ } - const NR_THREADS: u64 = 64; + const NR_THREADS: u64 = 512; let range_size = (u32::MAX as u64 + 1) / NR_THREADS; diff --git a/tests/test.rs b/tests/test.rs index 5007464..bae941d 100644 --- a/tests/test.rs +++ b/tests/test.rs @@ -8,59 +8,30 @@ mod armv8; use yaxpeax_arch::{Arch, Decoder, U8Reader}; use std::fmt::Write; -#[test] -#[ignore] -fn test_armv7_does_not_panic() { - let armv7 = <yaxpeax_arm::armv7::ARMv7 as Arch>::Decoder::default(); +fn test_range<A: Arch>(decoder: &A::Decoder, start: u64, end: u64) +where + for <'a> U8Reader<'a>: yaxpeax_arch::Reader<<A as Arch>::Address, <A as Arch>::Word>, + <A as Arch>::Instruction: std::fmt::Display +{ - for i in 0..=u32::MAX { - let bytes = i.to_le_bytes(); - let res = armv7.decode(&mut U8Reader::new(&bytes)); - if let Ok(instr) = res { - let s = instr.to_string(); - drop(s); - } - } -} -#[test] -#[ignore] -fn test_armv7_thumb_does_not_panic() { - let mut armv7_t = <yaxpeax_arm::armv7::ARMv7 as Arch>::Decoder::default(); - armv7_t.set_thumb_mode(true); + let mut instr = A::Instruction::default(); + let mut s = String::new(); - for i in 0..=u32::MAX { + for i in start..=end { + if i & 0x01_ff_ff_ff == 0 { + eprintln!("case {:08x}", i); + } + let i = i as u32; let bytes = i.to_le_bytes(); - let res = armv7_t.decode(&mut U8Reader::new(&bytes)); - if let Ok(instr) = res { - let s = instr.to_string(); - drop(s); + let res = decoder.decode_into(&mut instr, &mut U8Reader::new(&bytes)); + if let Ok(()) = res { + s.clear(); + write!(s, "{}", instr).unwrap(); } } } -#[test] -#[ignore] -fn test_armv8_does_not_panic() { - fn test_range(start: u64, end: u64) { - let armv8 = <yaxpeax_arm::armv8::a64::ARMv8 as Arch>::Decoder::default(); - - let mut instr = <yaxpeax_arm::armv8::a64::ARMv8 as yaxpeax_arch::Arch>::Instruction::default(); - let mut s = String::new(); - - for i in start..=end { - if i & 0x01_ff_ff_ff == 0 { - eprintln!("case {:08x}", i); - } - let i = i as u32; - let bytes = i.to_le_bytes(); - let res = armv8.decode_into(&mut instr, &mut U8Reader::new(&bytes)); - if let Ok(()) = res { - s.clear(); - write!(s, "{}", instr).unwrap(); - } - } - } - +fn par_test_u32(test_range: fn(u64, u64)) { const NR_THREADS: u64 = 512; const RANGE_SIZE: u64 = (u32::MAX as u64 + 1) / NR_THREADS; @@ -76,3 +47,33 @@ fn test_armv8_does_not_panic() { handle.join().unwrap(); } } + +#[test] +#[ignore] +fn test_armv7_does_not_panic() { + par_test_u32(|start, end| { + let armv7 = <yaxpeax_arm::armv7::ARMv7 as Arch>::Decoder::default(); + + test_range::<yaxpeax_arm::armv7::ARMv7>(&armv7, start, end); + }); +} +#[test] +#[ignore] +fn test_armv7_thumb_does_not_panic() { + par_test_u32(|start, end| { + let mut armv7_t = <yaxpeax_arm::armv7::ARMv7 as Arch>::Decoder::default(); + armv7_t.set_thumb_mode(true); + + test_range::<yaxpeax_arm::armv7::ARMv7>(&armv7_t, start, end); + }); +} + +#[test] +#[ignore] +fn test_armv8_does_not_panic() { + par_test_u32(|start, end| { + let armv8 = <yaxpeax_arm::armv8::a64::ARMv8 as Arch>::Decoder::default(); + + test_range::<yaxpeax_arm::armv8::a64::ARMv8>(&armv8, start, end); + }); +} |