fix(server): fixed possible recursive includes in preprocessing

server/src/cache.rs

@@ -16,8 +16,10 @@ impl Sites {
         std::env::set_current_dir(path)?;
         let mut home_file = std::fs::File::open("home.html")?;
 
+        let mut visited_inodes = Vec::new();
+
         Ok(Self {
-            home: parser::generate_site(&mut home_file)?
+            home: parser::generate_site(&mut home_file, &mut visited_inodes)?
         })
     }
 

server/src/parser.rs

@@ -1,9 +1,17 @@
-use std::io::{Read, read_to_string};
+use std::{io::Read, os::unix::fs::MetadataExt};
 use anyhow::{Error, Result};
-use log::{error, info};
+use log::info;
 
-pub fn generate_site(base: &mut std::fs::File) -> Result<String> {
+pub fn generate_site(base: &mut std::fs::File, visited_inodes: &mut Vec<u64>) -> Result<String> {
-    let include_regex = regex::Regex::new("\\s+@@@include ((?:[^\\/]*\\/)*)(.*)@@@")?;
+    let inode = base.metadata()?.ino();
+
+    if visited_inodes.contains(&inode) {
+        return Err(Error::msg("Recursive file inclusion detected!"));
+    } else {
+        visited_inodes.push(inode);
+    }
+
+    let include_regex = regex::Regex::new("@@@include ((?:[^\\/]*\\/)*)(.*)@@@")?;
 
     let mut content = String::new();
     base.read_to_string(&mut content)?;
@@ -17,11 +25,13 @@ pub fn generate_site(base: &mut std::fs::File) -> Result<String> {
 
             let mut file = std::fs::File::open(parts.get(1).expect("This should never happen due to the regex check."))?;
 
-            lines.push(generate_site(&mut file)?);
+            lines.push(generate_site(&mut file, visited_inodes)?);
         } else {
             lines.push(line.to_owned());
         }
     }
 
+    let _ = visited_inodes.pop();
+
     Ok(lines.join("\n"))
 }