使用 go 启动交互式 ssh 会话

Question

我有这种“某种”工作，但缺少的是：

1. 控制键不起作用。“ctrl-c”会杀死整个 go 脚本，“ctrl-r”不会按预期工作，等等。
2. 向上和向下箭头打印^[[A^[[D^[[C

下面的代码是我从这个包中获得的： https: //github.com/nanobox-io/golang-ssh我单独使用该包进行了测试，它工作得很好。我看不到该包正在做什么，但我的脚本中没有。我还包含了用于创建到主机的隧道的代码（如果相关的话）。

这是示例代码。请忽略恐慌的使用，它使我在处理新代码时更容易调试它。我不希望将此代码在当前状态下用于生产环境。

func Start(hostName string) {
    var (
        termWidth, termHeight = 80, 24
    )

    jumpAddr := "..."
    hostAddr := "..."

    jumpConfig := ssh.ClientConfig{
        // ...
    }

    hostConfig := ssh.ClientConfig{
        // ...
    }

    jumpConn, err := ssh.Dial("tcp", jumpAddr+":22", &jumpConfig)
    ifpanic(err)

    hostConn, err := jumpConn.Dial("tcp", hostAddr+":22")
    ifpanic(err)

    hostSSHConn, chans, reqs, err := ssh.NewClientConn(hostConn, hostName+":22", &hostConfig)
    ifpanic(err)

    hostClient := ssh.NewClient(hostSSHConn, chans, reqs)
    session, err := hostClient.NewSession()
    ifpanic(err)

    session.Stdout = os.Stdout
    session.Stderr = os.Stderr
    session.Stdin = os.Stdin

    modes := ssh.TerminalModes{
        ssh.ECHO: 1,
    }

    fd := os.Stdin.Fd()

    oldState, err := term.MakeRaw(fd)
    ifpanic(err)

    defer ifpanic(term.RestoreTerminal(fd, oldState))

    winsize, err := term.GetWinsize(fd)
    if err == nil {
        termWidth = int(winsize.Width)
        termHeight = int(winsize.Height)
    }

    if err := session.RequestPty("xterm", termHeight, termWidth, modes); err != nil {
        panic(err)
    }

    if err := session.Shell(); err != nil {
        panic(err)
    }

    go monWinCh(session, os.Stdout.Fd())

    if err := session.Wait(); err != nil {
        panic(err)
    }
}

func ifpanic(err error) {
    if err != nil {
        panic(err)
    }
}

func monWinCh(session *ssh.Session, fd uintptr) {
    sigs := make(chan os.Signal, 1)

    signal.Notify(sigs, syscall.SIGWINCH)
    defer signal.Stop(sigs)

    for range sigs {
        if _, err := session.SendRequest("window-change", false, termSize(fd)); err != nil {
            panic(err)
        }
    }
}

func termSize(fd uintptr) []byte {
    size := make([]byte, 16)

    winsize, err := term.GetWinsize(fd)
    if err != nil {
        binary.BigEndian.PutUint32(size, uint32(80))
        binary.BigEndian.PutUint32(size[4:], uint32(24))
        return size
    }

    binary.BigEndian.PutUint32(size, uint32(winsize.Width))
    binary.BigEndian.PutUint32(size[4:], uint32(winsize.Height))

    return size
}

Answer 1

您可以使用通道捕获操作系统信号，然后将它们重定向到 SSH 会话：

import (
    "os"
    "os/signal"
    "syscall"
)
//...

sigChannel := make(chan os.Signal)
signal.Notify(sigChannel, syscall.SIGINT, syscall.SIGTERM)
go func() {
    for {
        sig, ok := <-sigChannel
        if !ok {
            break
        }

        switch sig {
        case syscall.SIGINT:
            session.Signal(ssh.SIGINT)
        case syscall.SIGTERM:
            session.Signal(ssh.SIGTERM)
        }
    }
}()

注意：不要忘记正确关闭信号通道，以便你的 goroutine 可以正常退出。

此代码不会捕获所有信号，但您可以在此处找到可用信号的列表，并选择您想要处理的信号。