Validate controller diagnostics, reboot protection, and terminal authentication

1 files changed, 6 insertions, 2 deletions

diff --git a/src/linux/mod.rs b/src/linux/mod.rs
index fd9c471..c6c10bd 100644
--- a/src/linux/mod.rs
+++ b/src/linux/mod.rs
@@ -104,8 +104,12 @@ impl Runtime for LinuxRuntime {
         }
         self.apply(connection, generation).await
     }
-    async fn probe(&mut self, connection: &Trojan, duration: u64) -> Result<xray::Probe> {
-        xray::Probe::start(connection, duration).await
+    async fn probe(
+        &mut self,
+        connection: &Trojan,
+        duration: u64,
+    ) -> Result<Box<dyn crate::runtime::ProbeSession>> {
+        Ok(Box::new(xray::Probe::start(connection, duration).await?))
     }
     async fn disable(&mut self) -> Result<()> {
         xray::systemctl(&["stop", "roentgen-xray.service"]).await?;