Skip to content

Commit ec692f9

Browse files
committed
add fuzz target for BlindedMessagePath
Adds fuzz target for BlindedMessagePath that implements `Readable` to check for correct deserialisation
1 parent 808d1dc commit ec692f9

File tree

6 files changed

+153
-0
lines changed

6 files changed

+153
-0
lines changed

fuzz/src/bin/gen_target.sh

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -82,3 +82,5 @@ GEN_TEST msg_stfu msg_targets::
8282
GEN_TEST msg_splice_init msg_targets::
8383
GEN_TEST msg_splice_ack msg_targets::
8484
GEN_TEST msg_splice_locked msg_targets::
85+
86+
GEN_TEST msg_blinded_message_path msg_targets::
Lines changed: 120 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,120 @@
1+
// This file is Copyright its original authors, visible in version control
2+
// history.
3+
//
4+
// This file is licensed under the Apache License, Version 2.0 <LICENSE-APACHE
5+
// or http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
6+
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your option.
7+
// You may not use this file except in accordance with one or both of these
8+
// licenses.
9+
10+
// This file is auto-generated by gen_target.sh based on target_template.txt
11+
// To modify it, modify target_template.txt and run gen_target.sh instead.
12+
13+
#![cfg_attr(feature = "libfuzzer_fuzz", no_main)]
14+
#![cfg_attr(rustfmt, rustfmt_skip)]
15+
16+
#[cfg(not(fuzzing))]
17+
compile_error!("Fuzz targets need cfg=fuzzing");
18+
19+
#[cfg(not(hashes_fuzz))]
20+
compile_error!("Fuzz targets need cfg=hashes_fuzz");
21+
22+
#[cfg(not(secp256k1_fuzz))]
23+
compile_error!("Fuzz targets need cfg=secp256k1_fuzz");
24+
25+
extern crate lightning_fuzz;
26+
use lightning_fuzz::msg_targets::msg_blinded_message_path::*;
27+
28+
#[cfg(feature = "afl")]
29+
#[macro_use] extern crate afl;
30+
#[cfg(feature = "afl")]
31+
fn main() {
32+
fuzz!(|data| {
33+
msg_blinded_message_path_run(data.as_ptr(), data.len());
34+
});
35+
}
36+
37+
#[cfg(feature = "honggfuzz")]
38+
#[macro_use] extern crate honggfuzz;
39+
#[cfg(feature = "honggfuzz")]
40+
fn main() {
41+
loop {
42+
fuzz!(|data| {
43+
msg_blinded_message_path_run(data.as_ptr(), data.len());
44+
});
45+
}
46+
}
47+
48+
#[cfg(feature = "libfuzzer_fuzz")]
49+
#[macro_use] extern crate libfuzzer_sys;
50+
#[cfg(feature = "libfuzzer_fuzz")]
51+
fuzz_target!(|data: &[u8]| {
52+
msg_blinded_message_path_run(data.as_ptr(), data.len());
53+
});
54+
55+
#[cfg(feature = "stdin_fuzz")]
56+
fn main() {
57+
use std::io::Read;
58+
59+
let mut data = Vec::with_capacity(8192);
60+
std::io::stdin().read_to_end(&mut data).unwrap();
61+
msg_blinded_message_path_run(data.as_ptr(), data.len());
62+
}
63+
64+
#[test]
65+
fn run_test_cases() {
66+
use std::fs;
67+
use std::io::Read;
68+
use lightning_fuzz::utils::test_logger::StringBuffer;
69+
70+
use std::sync::{atomic, Arc};
71+
{
72+
let data: Vec<u8> = vec![0];
73+
msg_blinded_message_path_run(data.as_ptr(), data.len());
74+
}
75+
let mut threads = Vec::new();
76+
let threads_running = Arc::new(atomic::AtomicUsize::new(0));
77+
if let Ok(tests) = fs::read_dir("test_cases/msg_blinded_message_path") {
78+
for test in tests {
79+
let mut data: Vec<u8> = Vec::new();
80+
let path = test.unwrap().path();
81+
fs::File::open(&path).unwrap().read_to_end(&mut data).unwrap();
82+
threads_running.fetch_add(1, atomic::Ordering::AcqRel);
83+
84+
let thread_count_ref = Arc::clone(&threads_running);
85+
let main_thread_ref = std::thread::current();
86+
threads.push((path.file_name().unwrap().to_str().unwrap().to_string(),
87+
std::thread::spawn(move || {
88+
let string_logger = StringBuffer::new();
89+
90+
let panic_logger = string_logger.clone();
91+
let res = if ::std::panic::catch_unwind(move || {
92+
msg_blinded_message_path_test(&data, panic_logger);
93+
}).is_err() {
94+
Some(string_logger.into_string())
95+
} else { None };
96+
thread_count_ref.fetch_sub(1, atomic::Ordering::AcqRel);
97+
main_thread_ref.unpark();
98+
res
99+
})
100+
));
101+
while threads_running.load(atomic::Ordering::Acquire) > 32 {
102+
std::thread::park();
103+
}
104+
}
105+
}
106+
let mut failed_outputs = Vec::new();
107+
for (test, thread) in threads.drain(..) {
108+
if let Some(output) = thread.join().unwrap() {
109+
println!("\nOutput of {}:\n{}\n", test, output);
110+
failed_outputs.push(test);
111+
}
112+
}
113+
if !failed_outputs.is_empty() {
114+
println!("Test cases which failed: ");
115+
for case in failed_outputs {
116+
println!("{}", case);
117+
}
118+
panic!();
119+
}
120+
}

fuzz/src/msg_targets/gen_target.sh

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -72,3 +72,5 @@ GEN_TEST lightning::ln::msgs::Stfu test_msg_simple ""
7272
GEN_TEST lightning::ln::msgs::SpliceInit test_msg_simple ""
7373
GEN_TEST lightning::ln::msgs::SpliceAck test_msg_simple ""
7474
GEN_TEST lightning::ln::msgs::SpliceLocked test_msg_simple ""
75+
76+
GEN_TEST lightning::blinded_path::message::BlindedMessagePath test_msg_simple ""

fuzz/src/msg_targets/mod.rs

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -48,3 +48,4 @@ pub mod msg_stfu;
4848
pub mod msg_splice_init;
4949
pub mod msg_splice_ack;
5050
pub mod msg_splice_locked;
51+
pub mod msg_blinded_message_path;
Lines changed: 27 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,27 @@
1+
// This file is Copyright its original authors, visible in version control
2+
// history.
3+
//
4+
// This file is licensed under the Apache License, Version 2.0 <LICENSE-APACHE
5+
// or http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
6+
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your option.
7+
// You may not use this file except in accordance with one or both of these
8+
// licenses.
9+
10+
// This file is auto-generated by gen_target.sh based on msg_target_template.txt
11+
// To modify it, modify msg_target_template.txt and run gen_target.sh instead.
12+
13+
#![cfg_attr(rustfmt, rustfmt_skip)]
14+
15+
use crate::msg_targets::utils::VecWriter;
16+
use crate::utils::test_logger;
17+
18+
#[inline]
19+
pub fn msg_blinded_message_path_test<Out: test_logger::Output>(data: &[u8], _out: Out) {
20+
test_msg_simple!(lightning::blinded_path::message::BlindedMessagePath, data);
21+
}
22+
23+
#[no_mangle]
24+
pub extern "C" fn msg_blinded_message_path_run(data: *const u8, datalen: usize) {
25+
let data = unsafe { std::slice::from_raw_parts(data, datalen) };
26+
test_msg_simple!(lightning::blinded_path::message::BlindedMessagePath, data);
27+
}

fuzz/targets.h

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -67,3 +67,4 @@ void msg_stfu_run(const unsigned char* data, size_t data_len);
6767
void msg_splice_init_run(const unsigned char* data, size_t data_len);
6868
void msg_splice_ack_run(const unsigned char* data, size_t data_len);
6969
void msg_splice_locked_run(const unsigned char* data, size_t data_len);
70+
void msg_blinded_message_path_run(const unsigned char* data, size_t data_len);

0 commit comments

Comments
 (0)