feat(config): restrict flows to Paris and Dublin strategies (#1007)

fujiapple852 · Feb 12, 2024 · 3ea259b · 3ea259b
1 parent 559c97b
commit 3ea259b
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/src/config.rs b/src/config.rs
@@ -334,6 +334,16 @@ impl TrippyConfig {
         Self::build_config(args, cfg_file, platform)
     }
 
+    /// The maximum number of flows allowed.
+    ///
+    /// This is restricted to 1 for the classic strategy.
+    pub fn max_flows(&self) -> usize {
+        match self.multipath_strategy {
+            MultipathStrategy::Classic => 1,
+            _ => self.tui_max_flows,
+        }
+    }
+
     #[allow(clippy::too_many_lines)]
     fn build_config(args: Args, cfg_file: ConfigFile, platform: &Platform) -> anyhow::Result<Self> {
         let &Platform {

diff --git a/src/main.rs b/src/main.rs
@@ -218,7 +218,7 @@ fn start_tracer(
         tracer_config,
         channel_config,
         cfg.tui_max_samples,
-        cfg.tui_max_flows,
+        cfg.max_flows(),
     );
     let trace_data = backend.trace();
     thread::Builder::new()
@@ -344,7 +344,7 @@ fn make_tui_config(args: &TrippyConfig) -> TuiConfig {
         args.tui_geoip_mode,
         args.tui_max_addrs,
         args.tui_max_samples,
-        args.tui_max_flows,
+        args.max_flows(),
         args.tui_theme,
         &args.tui_bindings,
         &args.tui_custom_columns,