1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
|
// this was something of a misfeature for these formatters..
#[test]
#[allow(deprecated)]
fn formatters_are_not_feature_gated() {
use yaxpeax_arch::display::{
u8_hex, u16_hex, u32_hex, u64_hex,
signed_i8_hex, signed_i16_hex, signed_i32_hex, signed_i64_hex
};
let _ = u8_hex(10);
let _ = u16_hex(10);
let _ = u32_hex(10);
let _ = u64_hex(10);
let _ = signed_i8_hex(10);
let _ = signed_i16_hex(10);
let _ = signed_i32_hex(10);
let _ = signed_i64_hex(10);
}
#[cfg(feature="alloc")]
#[test]
fn display_sink_write_hex_helpers() {
use yaxpeax_arch::display::{DisplaySink};
// for u8/i8/u16/i16 we can exhaustively test. we'll leave the rest for fuzzers.
let mut buf = String::new();
for i in 0..=u8::MAX {
buf.clear();
buf.write_u8(i).expect("write succeeds");
assert_eq!(buf, format!("{:x}", i));
buf.clear();
buf.write_prefixed_u8(i).expect("write succeeds");
assert_eq!(buf, format!("0x{:x}", i));
let expected = if (i as i8) < 0 {
format!("-0x{:x}", (i as i8).unsigned_abs())
} else {
format!("0x{:x}", i)
};
buf.clear();
buf.write_prefixed_i8(i as i8).expect("write succeeds");
assert_eq!(buf, expected);
}
for i in 0..=u16::MAX {
buf.clear();
buf.write_u16(i).expect("write succeeds");
assert_eq!(buf, format!("{:x}", i));
buf.clear();
buf.write_prefixed_u16(i).expect("write succeeds");
assert_eq!(buf, format!("0x{:x}", i));
let expected = if (i as i16) < 0 {
format!("-0x{:x}", (i as i16).unsigned_abs())
} else {
format!("0x{:x}", i)
};
buf.clear();
buf.write_prefixed_i16(i as i16).expect("write succeeds");
assert_eq!(buf, expected);
}
}
#[cfg(feature="alloc")]
#[test]
fn sinks_are_equivalent() {
use yaxpeax_arch::display::{DisplaySink, FmtSink};
use yaxpeax_arch::testkit::DisplaySinkWriteComparator;
let mut bare = String::new();
let mut through_sink = String::new();
for i in 0..u16::MAX {
bare.clear();
through_sink.clear();
let mut out = FmtSink::new(&mut through_sink);
let mut comparator = DisplaySinkWriteComparator::new(
&mut out,
|sink| { sink.inner_ref().as_str() },
&mut bare,
|sink| { sink.as_str() },
);
comparator.write_u16(i).expect("write succeeds");
comparator.write_prefixed_u16(i).expect("write succeeds");
comparator.write_prefixed_i16(i as i16).expect("write succeeds");
}
}
#[cfg(all(feature="alloc", feature="color-new"))]
#[test]
fn ansi_sink_works() {
use yaxpeax_arch::color_new::ansi::AnsiDisplaySink;
use yaxpeax_arch::display::DisplaySink;
let mut buf = String::new();
let mut ansi_sink = AnsiDisplaySink::new(&mut buf, yaxpeax_arch::color_new::DefaultColors);
ansi_sink.span_start_immediate();
ansi_sink.write_prefixed_u8(0x80).expect("write succeeds");
ansi_sink.span_end_immediate();
ansi_sink.write_fixed_size("(").expect("write succeeds");
ansi_sink.span_start_register();
ansi_sink.write_fixed_size("rbp").expect("write succeeds");
ansi_sink.span_end_register();
ansi_sink.write_fixed_size(")").expect("write succeeds");
drop(ansi_sink);
assert_eq!(buf, "\x1b[37m0x80\x1b[39m(\x1b[38;5;6mrbp\x1b[39m)");
}
|
