]> git.bitcoin.ninja Git - rust-lightning/commitdiff
Take the logger from test_utils into fuzz::test_utils
authorMatt Corallo <git@bluematt.me>
Tue, 8 Jan 2019 20:06:43 +0000 (15:06 -0500)
committerMatt Corallo <git@bluematt.me>
Thu, 24 Jan 2019 18:16:59 +0000 (13:16 -0500)
fuzz/fuzz_targets/chanmon_deser_target.rs
fuzz/fuzz_targets/full_stack_target.rs
fuzz/fuzz_targets/router_target.rs
fuzz/fuzz_targets/utils/test_logger.rs

index 9ddf52c662ec7712acccfb0613100176eb9108ce..fb8f0bf6f120f49d6225168b18f4a9aaf55b340e 100644 (file)
@@ -30,7 +30,7 @@ impl Writer for VecWriter {
 #[inline]
 pub fn do_test(data: &[u8]) {
        reset_rng_state();
-       let logger = Arc::new(test_logger::TestLogger{});
+       let logger = Arc::new(test_logger::TestLogger::new("".to_owned()));
        if let Ok((latest_block_hash, monitor)) = <(Sha256dHash, channelmonitor::ChannelMonitor)>::read(&mut Cursor::new(data), logger.clone()) {
                let mut w = VecWriter(Vec::new());
                monitor.write_for_disk(&mut w).unwrap();
index 9930a44d2558cf54922fa8c1878bcf6a5965effb..a4697fd939df6741fe4241636c5611e4ecee36c6 100644 (file)
@@ -554,7 +554,7 @@ pub fn do_test(data: &[u8], logger: &Arc<Logger>) {
 #[cfg(feature = "afl")]
 fn main() {
        fuzz!(|data| {
-               let logger: Arc<Logger> = Arc::new(test_logger::TestLogger{});
+               let logger: Arc<Logger> = Arc::new(test_logger::TestLogger::new("".to_owned()));
                do_test(data, &logger);
        });
 }
@@ -565,7 +565,7 @@ fn main() {
 fn main() {
        loop {
                fuzz!(|data| {
-                       let logger: Arc<Logger> = Arc::new(test_logger::TestLogger{});
+                       let logger: Arc<Logger> = Arc::new(test_logger::TestLogger::new("".to_owned()));
                        do_test(data, &logger);
                });
        }
@@ -575,7 +575,7 @@ fn main() {
 #[macro_use] extern crate libfuzzer_sys;
 #[cfg(feature = "libfuzzer_fuzz")]
 fuzz_target!(|data: &[u8]| {
-       let logger: Arc<Logger> = Arc::new(test_logger::TestLogger{});
+       let logger: Arc<Logger> = Arc::new(test_logger::TestLogger::new("".to_owned()));
        do_test(data, &logger);
 });
 
@@ -589,7 +589,7 @@ mod tests {
 
        #[test]
        fn duplicate_crash() {
-               let logger: Arc<Logger> = Arc::new(test_logger::TestLogger{});
+               let logger: Arc<Logger> = Arc::new(test_logger::TestLogger::new("".to_owned()));
                super::do_test(&::hex::decode("00").unwrap(), &logger);
        }
 
index 8938deefe00890dcb4ec34b5fa6fabf49cbc0b0c..3a40d39855e63b9ce0bf627f0b65c1afe2ad0081 100644 (file)
@@ -154,7 +154,7 @@ pub fn do_test(data: &[u8]) {
                }
        }
 
-       let logger: Arc<Logger> = Arc::new(test_logger::TestLogger{});
+       let logger: Arc<Logger> = Arc::new(test_logger::TestLogger::new("".to_owned()));
        let chain_monitor = Arc::new(DummyChainWatcher {
                input: Arc::clone(&input),
        });
index f828d0638f17726690c2c9b83807866c634287bd..097d001dbb25f80e0049a6c1526c379365a0f49f 100644 (file)
@@ -1,11 +1,22 @@
 use lightning::util::logger::{Logger, Record};
+pub struct TestLogger {
+       #[cfg(test)]
+       id: String,
+}
 
-pub struct TestLogger {}
+impl TestLogger {
+       pub fn new(_id: String) -> TestLogger {
+               TestLogger {
+                       #[cfg(test)]
+                       id: _id
+               }
+       }
+}
 
 impl Logger for TestLogger {
        fn log(&self, record: &Record) {
                #[cfg(test)]
-               println!("{:<5} [{} : {}, {}] {}", record.level.to_string(), record.module_path, record.file, record.line, record.args);
+               println!("{:<5} {} [{} : {}, {}] {}", record.level.to_string(), self.id, record.module_path, record.file, record.line, record.args);
                #[cfg(not(test))]
                let _ = format!("{}", record.args);
        }