From 65991e53f32028fc0e75491e2d541d7f72d555a0 Mon Sep 17 00:00:00 2001 From: Hatter Jiang Date: Sun, 18 Jul 2021 23:21:13 +0800 Subject: [PATCH] chore: home --- main.go | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/main.go b/main.go index 627f856..6436410 100644 --- a/main.go +++ b/main.go @@ -46,13 +46,13 @@ func main() { } ssh.Handle(func(s ssh.Session) { - // authorizedKey := gossh.MarshalAuthorizedKey(s.PublicKey()) io.WriteString(s, WELCOME) cmd := exec.Command("/bin/bash") ptyReq, winCh, isPty := s.Pty() if isPty { cmd.Env = append(cmd.Env, fmt.Sprintf("TERM=%s", ptyReq.Term)) + cmd.Env = append(cmd.Env, fmt.Sprintf("HOME=/root")) f, err := pty.Start(cmd) if err != nil { panic(err) @@ -67,6 +67,7 @@ func main() { }() io.Copy(s, f) // stdout cmd.Wait() + s.Exit(0) } else { io.WriteString(s, "No PTY requested.\n") s.Exit(1)