From 90956826a5e267e8c9d220561e8b2506efe97d02 Mon Sep 17 00:00:00 2001
From: Jesse Luehrs <doy@tozt.net>
Date: Thu, 18 Nov 2021 23:13:57 +0000
Subject: add afl fuzzer

---
 fuzz/src/main.rs | 66 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 66 insertions(+)
 create mode 100644 fuzz/src/main.rs

(limited to 'fuzz/src')

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"
+            );
+        }
+    });
+}
-- 
cgit v1.2.3-54-g00ecf