Skip to content

Commit 9823983

Browse files
committed
Take the logger from test_utils into fuzz::test_utils
1 parent cff5e8b commit 9823983

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
@@ -533,7 +533,7 @@ pub fn do_test(data: &[u8], logger: &Arc<Logger>) {
533533
#[cfg(feature = "afl")]
534534
fn main() {
535535
fuzz!(|data| {
536-
let logger: Arc<Logger> = Arc::new(test_logger::TestLogger{});
536+
let logger: Arc<Logger> = Arc::new(test_logger::TestLogger::new("".to_owned()));
537537
do_test(data, &logger);
538538
});
539539
}
@@ -544,7 +544,7 @@ fn main() {
544544
fn main() {
545545
loop {
546546
fuzz!(|data| {
547-
let logger: Arc<Logger> = Arc::new(test_logger::TestLogger{});
547+
let logger: Arc<Logger> = Arc::new(test_logger::TestLogger::new("".to_owned()));
548548
do_test(data, &logger);
549549
});
550550
}
@@ -554,7 +554,7 @@ fn main() {
554554
#[macro_use] extern crate libfuzzer_sys;
555555
#[cfg(feature = "libfuzzer_fuzz")]
556556
fuzz_target!(|data: &[u8]| {
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

@@ -568,7 +568,7 @@ mod tests {
568568

569569
#[test]
570570
fn duplicate_crash() {
571-
let logger: Arc<Logger> = Arc::new(test_logger::TestLogger{});
571+
let logger: Arc<Logger> = Arc::new(test_logger::TestLogger::new("".to_owned()));
572572
super::do_test(&::hex::decode("00").unwrap(), &logger);
573573
}
574574

fuzz/fuzz_targets/router_target.rs

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

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

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)