Fix panic on incomplete closing frontmatter fence

Closes GH-121. Reviewed-by: Titus Wormer <[email protected]>
wooorm · Aug 9, 2024 · 2c83b0d · 2c83b0d
1 parent d0788c9
commit 2c83b0d
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 7 deletions.
diff --git a/src/construct/document.rs b/src/construct/document.rs
@@ -565,14 +565,16 @@ fn resolve(tokenizer: &mut Tokenizer) {
                 inject_index = child_index + 1;
             }
 
-            if let Some(mut exits) = tokenizer.tokenize_state.document_exits[line].take() {
-                let mut exit_index = 0;
-                while exit_index < exits.len() {
-                    exits[exit_index].point = point.clone();
-                    exit_index += 1;
+            if line < tokenizer.tokenize_state.document_exits.len() {
+                if let Some(mut exits) = tokenizer.tokenize_state.document_exits[line].take() {
+                    let mut exit_index = 0;
+                    while exit_index < exits.len() {
+                        exits[exit_index].point = point.clone();
+                        exit_index += 1;
+                    }
+
+                    child.map.add(inject_index, 0, exits);
                 }
-
-                child.map.add(inject_index, 0, exits);
             }
 
             line += 1;

diff --git a/tests/frontmatter.rs b/tests/frontmatter.rs
@@ -61,6 +61,12 @@ fn frontmatter() -> Result<(), message::Message> {
         "should not support 2 markers in a closing fence"
     );
 
+    assert_eq!(
+        to_html_with_options("---\n--\n", &frontmatter)?,
+        "<hr />\n<p>--</p>\n",
+        "should not panic if newline after 2 marker closing fence"
+    );
+
     assert_eq!(
         to_html_with_options("---\n----", &frontmatter)?,
         "<hr />\n<hr />",