add afl fuzzer

3 files changed, 80 insertions, 0 deletions

diff --git a/fuzz/.gitignore b/fuzz/.gitignore
new file mode 100644
index 0000000..0a25f36
--- /dev/null
+++ b/fuzz/.gitignore
@@ -0,0 +1,2 @@
+/target
+/out
diff --git a/fuzz/Cargo.toml b/fuzz/Cargo.toml
new file mode 100644
index 0000000..8f6621d
--- /dev/null
+++ b/fuzz/Cargo.toml
@@ -0,0 +1,12 @@
+[package]
+name = "fuzz"
+version = "0.1.0"
+edition = "2021"
+
+# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
+
+[dependencies]
+afl = "*"
+vt100 = { path = ".." }
+serde = { version = "1.0.130", features = ["derive"] }
+serde_json = "1.0.69"
diff --git a/fuzz/src/main.rs b/fuzz/src/main.rs
new file mode 100644
index 0000000..ee90e3c
--- /dev/null
+++ b/fuzz/src/main.rs
@@ -0,0 +1,66 @@
+#[path = "../../tests/helpers/mod.rs"]
+mod helpers;
+
+fn main() {
+    afl::fuzz!(|data: &[u8]| {
+        let mut vt_base = vt100::Parser::default();
+        let mut vt_diff = vt100::Parser::default();
+        let mut prev_screen = vt_base.screen().clone();
+        let empty_screen = vt100::Parser::default().screen().clone();
+        for byte in data {
+            vt_base.process(&[*byte]);
+
+            let mut vt_full = vt100::Parser::default();
+            vt_full.process(&vt_base.screen().state_formatted());
+            vt_full.process(&vt_base.screen().bells_diff(&empty_screen));
+            assert!(
+                helpers::compare_screens(vt_base.screen(), vt_full.screen()),
+                "full"
+            );
+
+            let mut vt_diff_empty = vt100::Parser::default();
+            vt_diff_empty
+                .process(&vt_base.screen().state_diff(&empty_screen));
+            vt_diff_empty
+                .process(&vt_base.screen().bells_diff(&empty_screen));
+            assert!(
+                helpers::compare_screens(
+                    vt_base.screen(),
+                    vt_diff_empty.screen()
+                ),
+                "diff-empty"
+            );
+
+            vt_diff.process(&vt_base.screen().state_diff(&prev_screen));
+            vt_diff.process(&vt_base.screen().bells_diff(&empty_screen));
+            prev_screen = vt_base.screen().clone();
+            assert!(
+                helpers::compare_screens(vt_base.screen(), vt_diff.screen()),
+                "diff"
+            );
+
+            let mut vt_rows = vt100::Parser::default();
+            let mut wrapped = false;
+            for (idx, row) in
+                vt_base.screen().rows_formatted(0, 80).enumerate()
+            {
+                vt_rows.process(b"\x1b[m");
+                if !wrapped {
+                    vt_rows.process(format!("\x1b[{}H", idx + 1).as_bytes());
+                }
+                vt_rows.process(&row);
+                wrapped =
+                    vt_base.screen().row_wrapped(idx.try_into().unwrap());
+            }
+            vt_rows.process(&vt_base.screen().cursor_state_formatted());
+            vt_rows.process(&vt_base.screen().attributes_formatted());
+            vt_rows.process(&vt_base.screen().input_mode_formatted());
+            vt_rows.process(&vt_base.screen().title_formatted());
+            vt_rows.process(&vt_base.screen().bells_diff(&empty_screen));
+            assert!(
+                helpers::compare_screens(vt_base.screen(), vt_rows.screen()),
+                "rows"
+            );
+        }
+    });
+}