Added regex assertion for station_id

filecoin-station · May 2, 2024 · 6620505 · 6620505
1 parent 12fe9b8
commit 6620505
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/daemon/Cargo.toml b/daemon/Cargo.toml
@@ -18,6 +18,7 @@ atomicwrites = "0.4.3"
 clap = { version = "4.5.4", features = ["derive", "env"] }
 env_logger.workspace = true
 log.workspace = true
+regex = "1.10.4"
 serde.workspace = true
 serde_json = "1.0.116"
 tokio = { workspace = true }

diff --git a/daemon/main.rs b/daemon/main.rs
@@ -10,6 +10,7 @@ use std::time::Duration;
 
 use args::CliArgs;
 use clap::Parser;
+use regex::Regex;
 
 use zinnia_runtime::anyhow::{anyhow, Context, Error, Result};
 use zinnia_runtime::{
@@ -87,6 +88,10 @@ async fn run(config: CliArgs) -> Result<RunOutput> {
         rng_seed: None,
     };
 
+    if !Regex::new(r"^[0-9a-fA-F]{88}$").unwrap().is_match(&runtime_config.station_id) {
+        return Err(anyhow!("Invalid station_id format"));
+    }
+
     // TODO: handle module exit and restart it
     // https://github.com/filecoin-station/zinnia/issues/146
     log::info!("Starting module {main_module}");