fix(agent): set home fallback when user home does not exist

testitok · Nov 9, 2023 · 5b7d9cb · 5b7d9cb
1 parent f3030a1
commit 5b7d9cb
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/pkg/agent/server/command/command_native.go b/pkg/agent/server/command/command_native.go
@@ -11,6 +11,7 @@ import (
 	"syscall"
 
 	"github.com/shellhub-io/shellhub/pkg/agent/pkg/osauth"
+	log "github.com/sirupsen/logrus"
 )
 
 func NewCmd(u *osauth.User, shell, term, host string, command ...string) *exec.Cmd {
@@ -34,7 +35,14 @@ func NewCmd(u *osauth.User, shell, term, host string, command ...string) *exec.C
 		"SHELL=" + shell,
 		"SHELLHUB_HOST=" + host,
 	}
-	cmd.Dir = u.HomeDir
+
+	if _, err := os.Stat(u.HomeDir); err != nil {
+		log.WithError(err).WithField("dir", u.HomeDir).Warn("setting user's home directory to /")
+
+		cmd.Dir = "/"
+	} else {
+		cmd.Dir = u.HomeDir
+	}
 
 	if os.Geteuid() == 0 {
 		cmd.SysProcAttr = &syscall.SysProcAttr{}