fix(cli): confine agent runtime paths

This commit is contained in:
ZacharyZhang-NY committed 2026-07-19 16:26:00 -04:00
1 parent c9d1075585
commit 7bf84a35e0
10 files changed
+399 -51

No files matched your search

+80
View File
@@ -0,0 +1,80 @@
package main
import (
"os"
"path/filepath"
"strings"
"testing"
)
func setMainTestHome(t *testing.T) string {
t.Helper()
home := t.TempDir()
t.Setenv("HOME", home)
t.Setenv("USERPROFILE", home)
return home
}
func TestGuardScriptPathRejectsUnsafeAgentID(t *testing.T) {
setMainTestHome(t)
_, err := guardScriptPathForAgent("../escape")
if err == nil || !strings.Contains(err.Error(), "invalid agent ID") {
t.Fatalf("guardScriptPathForAgent() error = %v", err)
}
}
func TestFindAgentIDByNameReturnsValidatedDirectoryIdentity(t *testing.T) {
home := setMainTestHome(t)
runtimeDirectory := filepath.Join(home, ".elydora", "agent-1")
if err := os.MkdirAll(runtimeDirectory, 0700); err != nil {
t.Fatalf("MkdirAll() error = %v", err)
}
config := []byte(`{"agent_name":"opencode","agent_id":"agent-1"}`)
if err := os.WriteFile(filepath.Join(runtimeDirectory, "config.json"), config, 0600); err != nil {
t.Fatalf("WriteFile() error = %v", err)
}
agentID, err := findAgentIDByName("opencode")
if err != nil {
t.Fatalf("findAgentIDByName() error = %v", err)
}
if agentID != "agent-1" {
t.Fatalf("agent ID = %q, want agent-1", agentID)
}
}
func TestFindAgentIDByNameRejectsConfigDirectoryMismatch(t *testing.T) {
home := setMainTestHome(t)
runtimeDirectory := filepath.Join(home, ".elydora", "stored-directory")
if err := os.MkdirAll(runtimeDirectory, 0700); err != nil {
t.Fatalf("MkdirAll() error = %v", err)
}
config := []byte(`{"agent_name":"opencode","agent_id":"different-agent"}`)
if err := os.WriteFile(filepath.Join(runtimeDirectory, "config.json"), config, 0600); err != nil {
t.Fatalf("WriteFile() error = %v", err)
}
_, err := findAgentIDByName("opencode")
if err == nil || !strings.Contains(err.Error(), "crosses its runtime directory") {
t.Fatalf("findAgentIDByName() error = %v", err)
}
}
func TestFindAgentIDByNameRejectsAmbiguousRuntimes(t *testing.T) {
home := setMainTestHome(t)
for _, agentID := range []string{"agent-1", "agent-2"} {
runtimeDirectory := filepath.Join(home, ".elydora", agentID)
if err := os.MkdirAll(runtimeDirectory, 0700); err != nil {
t.Fatalf("MkdirAll() error = %v", err)
}
config := []byte(`{"agent_name":"opencode","agent_id":"` + agentID + `"}`)
if err := os.WriteFile(filepath.Join(runtimeDirectory, "config.json"), config, 0600); err != nil {
t.Fatalf("WriteFile() error = %v", err)
}
}
_, err := findAgentIDByName("opencode")
if err == nil || !strings.Contains(err.Error(), "multiple") {
t.Fatalf("findAgentIDByName() error = %v", err)
}
}