Session file security updates (#3071)

2025-06-25 12:02:48 -07:00
parent 8a32128461
commit 85f284b4cf
10 changed files with 446 additions and 178 deletions
@@ -372,9 +372,17 @@ impl Agent {
            })?;

        // Get the session file path
-        let session_path = crate::session::storage::get_path(
+        let session_path = match crate::session::storage::get_path(
            crate::session::storage::Identifier::Name(session_id.to_string()),
-        );
+        ) {
+            Ok(path) => path,
+            Err(e) => {
+                return Err(ToolError::ExecutionError(format!(
+                    "Invalid session ID '{}': {}",
+                    session_id, e
+                )));
+            }
+        };

        // Check if session file exists
        if !session_path.exists() {