Skip to content

Commit aa9a848

Browse files
committed
Take the logger from test_utils into fuzz::test_utils
1 parent 9a72207 commit aa9a848

File tree

4 files changed

+19
-8
lines changed

4 files changed

+19
-8
lines changed

fuzz/fuzz_targets/chanmon_deser_target.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ impl Writer for VecWriter {
3030
#[inline]
3131
pub fn do_test(data: &[u8]) {
3232
reset_rng_state();
33-
let logger = Arc::new(test_logger::TestLogger{});
33+
let logger = Arc::new(test_logger::TestLogger::new("".to_owned()));
3434
if let Ok((latest_block_hash, monitor)) = <(Sha256dHash, channelmonitor::ChannelMonitor)>::read(&mut Cursor::new(data), logger.clone()) {
3535
let mut w = VecWriter(Vec::new());
3636
monitor.write_for_disk(&mut w).unwrap();

fuzz/fuzz_targets/full_stack_target.rs

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -554,7 +554,7 @@ pub fn do_test(data: &[u8], logger: &Arc<Logger>) {
554554
#[cfg(feature = "afl")]
555555
fn main() {
556556
fuzz!(|data| {
557-
let logger: Arc<Logger> = Arc::new(test_logger::TestLogger{});
557+
let logger: Arc<Logger> = Arc::new(test_logger::TestLogger::new("".to_owned()));
558558
do_test(data, &logger);
559559
});
560560
}
@@ -565,7 +565,7 @@ fn main() {
565565
fn main() {
566566
loop {
567567
fuzz!(|data| {
568-
let logger: Arc<Logger> = Arc::new(test_logger::TestLogger{});
568+
let logger: Arc<Logger> = Arc::new(test_logger::TestLogger::new("".to_owned()));
569569
do_test(data, &logger);
570570
});
571571
}
@@ -575,7 +575,7 @@ fn main() {
575575
#[macro_use] extern crate libfuzzer_sys;
576576
#[cfg(feature = "libfuzzer_fuzz")]
577577
fuzz_target!(|data: &[u8]| {
578-
let logger: Arc<Logger> = Arc::new(test_logger::TestLogger{});
578+
let logger: Arc<Logger> = Arc::new(test_logger::TestLogger::new("".to_owned()));
579579
do_test(data, &logger);
580580
});
581581

@@ -589,7 +589,7 @@ mod tests {
589589

590590
#[test]
591591
fn duplicate_crash() {
592-
let logger: Arc<Logger> = Arc::new(test_logger::TestLogger{});
592+
let logger: Arc<Logger> = Arc::new(test_logger::TestLogger::new("".to_owned()));
593593
super::do_test(&::hex::decode("00").unwrap(), &logger);
594594
}
595595

fuzz/fuzz_targets/router_target.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -154,7 +154,7 @@ pub fn do_test(data: &[u8]) {
154154
}
155155
}
156156

157-
let logger: Arc<Logger> = Arc::new(test_logger::TestLogger{});
157+
let logger: Arc<Logger> = Arc::new(test_logger::TestLogger::new("".to_owned()));
158158
let chain_monitor = Arc::new(DummyChainWatcher {
159159
input: Arc::clone(&input),
160160
});

fuzz/fuzz_targets/utils/test_logger.rs

Lines changed: 13 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,22 @@
11
use lightning::util::logger::{Logger, Record};
2+
pub struct TestLogger {
3+
#[cfg(test)]
4+
id: String,
5+
}
26

3-
pub struct TestLogger {}
7+
impl TestLogger {
8+
pub fn new(_id: String) -> TestLogger {
9+
TestLogger {
10+
#[cfg(test)]
11+
id: _id
12+
}
13+
}
14+
}
415

516
impl Logger for TestLogger {
617
fn log(&self, record: &Record) {
718
#[cfg(test)]
8-
println!("{:<5} [{} : {}, {}] {}", record.level.to_string(), record.module_path, record.file, record.line, record.args);
19+
println!("{:<5} {} [{} : {}, {}] {}", record.level.to_string(), self.id, record.module_path, record.file, record.line, record.args);
920
#[cfg(not(test))]
1021
let _ = format!("{}", record.args);
1122
}

0 commit comments

Comments
 (0)