From 90e2e053d25feabd2dcf384dfc262a8bcdbe3ab8 Mon Sep 17 00:00:00 2001 From: Cody Hiar Date: Sat, 5 Mar 2022 12:53:15 -0700 Subject: Initial Commit --- termshare/internal/daemon/daemon.go | 56 +++++++++++++++++++++++++++++++++++++ 1 file changed, 56 insertions(+) create mode 100644 termshare/internal/daemon/daemon.go (limited to 'termshare/internal/daemon') diff --git a/termshare/internal/daemon/daemon.go b/termshare/internal/daemon/daemon.go new file mode 100644 index 0000000..465cdff --- /dev/null +++ b/termshare/internal/daemon/daemon.go @@ -0,0 +1,56 @@ +package daemon + +import ( + "fmt" + "bufio" + "io" + "log" + "os" + "os/exec" + "syscall" + "strconv" + "unsafe" + + "github.com/gliderlabs/ssh" + "github.com/creack/pty" +) + +func setWinsize(f *os.File, w, h int) { + syscall.Syscall(syscall.SYS_IOCTL, f.Fd(), uintptr(syscall.TIOCSWINSZ), + uintptr(unsafe.Pointer(&struct{ h, w, x, y uint16 }{uint16(h), uint16(w), 0, 0}))) +} + +func RunDaemon(port string) { + ssh.Handle(func(s ssh.Session) { + cmd := exec.Command("bash") + ptyReq, winCh, isPty := s.Pty() + if isPty { + cmd.Env = append(cmd.Env, fmt.Sprintf("TERM=%s", ptyReq.Term)) + f, err := pty.Start(cmd) + if err != nil { + panic(err) + } + go func() { + for win := range winCh { + setWinsize(f, win.Width, win.Height) + } + }() + go func() { + io.Copy(f, s) // stdin + }() + io.Copy(s, f) // stdout + cmd.Wait() + } else { + io.WriteString(s, "No PTY requested.\n") + reader := bufio.NewReader(s) + for { + input, _, _ := reader.ReadRune() + io.WriteString(os.Stdout, strconv.QuoteRune(input)) + } + s.Exit(1) + } + }) + + log.Println("starting ssh server on port " + port + " ...") + log.Fatal(ssh.ListenAndServe(":" + port, nil)) +} -- cgit v1.2.3