diff --git a/cmd/olliesrv/server.go b/cmd/olliesrv/server.go index 56ca5e0..cf5f9ea 100644 --- a/cmd/olliesrv/server.go +++ b/cmd/olliesrv/server.go @@ -585,9 +585,9 @@ func (s *Server) pathType(path string) string { if sess.Core == nil { return "" } - // Validate agent Name - agentName := sess.Core.Agent().Name() - if agentName == "" || parts[2] != agentName { + // Validate target agent by name or ID. + ag := sess.Core.FindAgent(parts[2]) + if ag == nil { return "" } // /session/{id}/agent/{aid} — directory @@ -600,7 +600,7 @@ func (s *Server) pathType(path string) string { } // /session/{id}/agent/{aid}/proc/{pid} — verify exists if len(parts) == 5 && parts[3] == "proc" { - for _, p := range sess.Core.Agent().ListDetached() { + for _, p := range ag.ListDetached() { if fmt.Sprintf("%d", p.PID) == parts[4] { return "file" }