主页 / coding / 问题 / 79422755
Accepted
xrfang
Asked: 2025-02-08 14:58:00 +0800 CST

如何在Go中实现ssh会话日志记录？

  • 772

我正在编写一个简单的堡垒服务，它将记录 ssh 会话的输入/输出。ssh 连接部分如下所示：

func connect(user, host string, logger sessionLogger) error {
    cmd := exec.Command("ssh", user+"@"+host)
    t, err := pty.Start(cmd)
    if err != nil {
        return err
    }
    defer t.Close()
    go func() {
        buf := make([]byte, 1024)
        for {
            n, err := t.Read(buf)
            if err != nil {
                break
            }
            os.Stdout.Write(buf[:n])
            logger(buf[:n])
        }
    }()
    go func() { io.Copy(t, os.Stdin) }()
    return cmd.Wait()
}

问题是用户输入的所有命令都会回显到 PTY 两次，如下所示：

在此处输入图片描述

我曾尝试使用 cmd 的 stdin/stdout，包括根本不使用 pty，但遇到了各种奇怪的问题，例如没有命令提示符的输出等。其中大多数与命令的 stdout 的性质有关（它必须是 pty）。

这是目前为止最好的结果了。不过我需要消除过多的命令回音。

linux

1 个回答

  1. Best Answer

    您需要使用原始模式来防止重复输入

        // import "golang.org/x/term"
    ...
    defer t.Close()

    oldState, err := term.MakeRaw(int(os.Stdin.Fd()))
    if err != nil {
        return err
    }
    defer term.Restore(int(os.Stdin.Fd()), oldState)
    ...
    • 1

相关问题