Add HOME to environment of interactive unix shells

wyzmgmdg · Dec 24, 2021 · 5dbb75d · 5dbb75d
1 parent 51d087a
commit 5dbb75d
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/ssh_session_unix.go b/ssh_session_unix.go
@@ -24,6 +24,7 @@ import (
 	"io"
 	"log"
 	"os/exec"
+	"os/user"
 
 	"github.com/creack/pty"
 	"github.com/gliderlabs/ssh"
@@ -36,6 +37,9 @@ func createPty(s ssh.Session, shell string) {
 	)
 
 	cmd.Env = append(cmd.Env, fmt.Sprintf("TERM=%s", ptyReq.Term))
+	if currentUser, err := user.Current(); err == nil {
+		cmd.Env = append(cmd.Env, fmt.Sprintf("HOME=%s", currentUser.HomeDir))
+	}
 	f, err := pty.Start(cmd)
 	if err != nil {
 		log.Fatalln("Could not start shell:", err)