Add gettext fuzz target

Author: kdarkhan

.github/workflows/test.yml

@@ -61,6 +61,12 @@ jobs:
           cargo fuzz run normalize -- -only_ascii=1 -max_total_time=30
           cargo fuzz cmin normalize
 
+      - name: Run gettext fuzzer and minimize corpus
+        run: |
+          cd i18n-helpers
+          cargo fuzz run gettext -- -only_ascii=1 -max_total_time=30
+          cargo fuzz cmin normalize
+
   clippy:
     name: Clippy
     runs-on: ubuntu-latest

i18n-helpers/fuzz/Cargo.lock

@@ -8,13 +8,13 @@ publish = false
 cargo-fuzz = true
 
 [dependencies]
+arbitrary = { version = "1.3.1", features = ["derive"] }
 libfuzzer-sys = "0.4.0"
+mdbook = { version = "0.4.25", default-features = false }
+mdbook-i18n-helpers = { path = ".." }
 polib = "0.2.0"
 pretty_assertions = "1.3.0"
 
-[dependencies.mdbook-i18n-helpers]
-path = ".."
-
 # Prevent this from interfering with workspaces
 [workspace]
 members = ["."]
@@ -33,3 +33,9 @@ name = "normalize"
 path = "fuzz_targets/normalize.rs"
 test = false
 doc = false
+
+[[bin]]
+name = "gettext"
+path = "fuzz_targets/gettext.rs"
+test = false
+doc = false

i18n-helpers/fuzz/Cargo.toml

@@ -0,0 +1,12 @@
+#![no_main]
+
+use libfuzzer_sys::fuzz_target;
+use mdbook_i18n_helpers::gettext::translate_book;
+use mdbook_i18n_helpers_fuzz::{create_book, create_catalog, BookItem};
+
+fuzz_target!(|inputs: (Vec<(&str, &str)>, Vec<BookItem>)| {
+    let (translations, book_items) = inputs;
+    let catalog = create_catalog(translations);
+    let mut book = create_book(book_items);
+    translate_book(&catalog, &mut book)
+});

i18n-helpers/fuzz/fuzz_targets/gettext.rs

@@ -2,24 +2,9 @@
 
 use libfuzzer_sys::fuzz_target;
 use mdbook_i18n_helpers::normalize::normalize;
-use polib::catalog::Catalog;
-use polib::message::Message;
-use polib::metadata::CatalogMetadata;
+use mdbook_i18n_helpers_fuzz::create_catalog;
 
 fuzz_target!(|translations: Vec<(&str, &str)>| {
     let catalog = create_catalog(translations);
     let _ = normalize(catalog);
 });
-
-fn create_catalog(translations: Vec<(&str, &str)>) -> Catalog {
-    let mut catalog = Catalog::new(CatalogMetadata::new());
-    for (idx, (msgid, msgstr)) in translations.iter().enumerate() {
-        let message = Message::build_singular()
-            .with_source(format!("foo.md:{idx}"))
-            .with_msgid(String::from(*msgid))
-            .with_msgstr(String::from(*msgstr))
-            .done();
-        catalog.append_or_update(message);
-    }
-    catalog
-}

i18n-helpers/fuzz/fuzz_targets/normalize.rs

@@ -0,0 +1,52 @@
+use arbitrary::Arbitrary;
+use mdbook::book::{Book, Chapter};
+use polib::catalog::Catalog;
+use polib::message::Message;
+use polib::metadata::CatalogMetadata;
+use std::path::PathBuf;
+
+/// Generate a random Catalog for fuzzing.
+pub fn create_catalog(translations: Vec<(&str, &str)>) -> Catalog {
+    let mut catalog = Catalog::new(CatalogMetadata::new());
+    for (idx, (msgid, msgstr)) in translations.iter().enumerate() {
+        let message = Message::build_singular()
+            .with_source(format!("foo.md:{idx}"))
+            .with_msgid(String::from(*msgid))
+            .with_msgstr(String::from(*msgstr))
+            .done();
+        catalog.append_or_update(message);
+    }
+    catalog
+}
+
+/// Generate a random Book for fuzzing.
+pub fn create_book(book_items: Vec<BookItem>) -> Book {
+    let mut book = Book::new();
+    for item in book_items.into_iter() {
+        book.push_item(item);
+    }
+    book
+}
+
+/// Wrapper enum for generating arbitrary `BookItem`s.
+#[derive(Arbitrary, Debug)]
+pub enum BookItem {
+    Chapter { name: String, content: String },
+    Separator,
+    PartTitle(String),
+}
+
+impl From<BookItem> for mdbook::book::BookItem {
+    fn from(other: BookItem) -> mdbook::book::BookItem {
+        match other {
+            BookItem::Chapter { name, content } => mdbook::book::BookItem::Chapter(Chapter::new(
+                &name,
+                content,
+                PathBuf::new(),
+                Vec::new(),
+            )),
+            BookItem::Separator => mdbook::book::BookItem::Separator,
+            BookItem::PartTitle(title) => mdbook::book::BookItem::PartTitle(title),
+        }
+    }
+}