Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add afl fuzz tests #272

Merged
merged 3 commits into from
Jul 6, 2021
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -29,6 +29,8 @@ exclude = [
# The fuzz crate can't be compiled or tested without the 'cargo fuzz' command,
# so exclude it from normal builds.
"fuzz",
# Same counts for the afl fuzz targets
"afl",
]

[lib]
Expand Down
27 changes: 27 additions & 0 deletions FUZZING.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,27 @@
# Fuzzing

Prost ships a few fuzz tests, using both libfuzzer and aflfuzz.


## afl

To run the afl fuzz tests, first install cargo-afl:

cargo install -f afl

Then build a fuzz target and run afl on it:

cd afl/<target>/
cargo afl build --bin fuzz-target
cargo afl fuzz -i in -o out target/debug/fuzz-target

To reproduce a crash:

cd afl/<target>/
cargo build --bin reproduce
cargo run --bin reproduce -- out/crashes/<crashfile>


## libfuzzer

TODO
2 changes: 2 additions & 0 deletions afl/.gitignore
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
out/
core.*
18 changes: 18 additions & 0 deletions afl/proto3/Cargo.toml
Original file line number Diff line number Diff line change
@@ -0,0 +1,18 @@
[package]
name = "fuzz-target-proto3"
version = "0.1.0"
authors = ["Prost developers"]
edition = "2018"

[[bin]]
name = "fuzz-target"
path = "src/main.rs"

[[bin]]
name = "reproduce"
path = "src/reproduce.rs"

[dependencies]
afl = "0.4"
protobuf = { path = "../../protobuf/" }
tests = { path = "../../tests/" }
30 changes: 30 additions & 0 deletions afl/proto3/README.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,30 @@
# proto3 fuzz tests

## Test corpus

The test message `testmessage` was created like this:

```rust
use prost::Message;
use protobuf::test_messages::proto3::TestAllTypesProto3;

fn main() {
let msg = TestAllTypesProto3 {
optional_int32: 42,
optional_fixed64: 9983748923,
optional_bool: true,
recursive_message: Some(
Box::new(TestAllTypesProto3 {
repeated_int32: vec![1, 2, 99, 50, -5],
..Default::default()
})
),
repeated_sfixed32: vec![1, -1, 1, -1],
repeated_float: vec![-1.0, 10.10, 1.337, std::f32::NAN],
..Default::default()
};
let mut buf = vec![];
msg.encode(&mut buf).unwrap();
std::fs::write("proto3-default.bin", buf).unwrap();
}
```
Empty file added afl/proto3/in/empty
Empty file.
Binary file added afl/proto3/in/testmessage
Binary file not shown.
10 changes: 10 additions & 0 deletions afl/proto3/src/main.rs
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
use afl::fuzz;

use protobuf::test_messages::proto3::TestAllTypesProto3;
use tests::roundtrip;

fn main() {
fuzz!(|data: &[u8]| {
let _ = roundtrip::<TestAllTypesProto3>(data).unwrap_error();
});
}
13 changes: 13 additions & 0 deletions afl/proto3/src/reproduce.rs
Original file line number Diff line number Diff line change
@@ -0,0 +1,13 @@
use protobuf::test_messages::proto3::TestAllTypesProto3;
use tests::roundtrip;

fn main() {
let args: Vec<String> = std::env::args().collect();
if args.len() != 2 {
println!("Usage: {} <path-to-crash>", args[0]);
std::process::exit(1);
}

let data = std::fs::read(&args[1]).expect(&format!("Could not open file {}", args[1]));
let _ = roundtrip::<TestAllTypesProto3>(&data).unwrap_error();
}