Go-SSH简易服务端如何优雅关闭连接？

问题核心 #

直接在exec请求回复后调用channel.Close()会导致Git客户端返回Exit status -1甚至本地出现"broken pipe"错误，本质是违背了SSH通道的交互流程和Git的协议要求。

channel.Close()的工作机制 #

SSH通道是双向字节流通道，调用Close()会向远端发送SSH_MSG_CHANNEL_CLOSE消息，告知对方不再发送数据；远端收到后会回复同样的消息，随后通道彻底关闭。如果此时远端仍在尝试向通道写入数据（比如Git客户端还在传输仓库数据），就会触发"broken pipe"错误。另外，Git客户端期望收到明确的命令退出状态，而非突然的通道关闭，这就是Exit status -1的原因。

正确的处理流程 #

针对Git的exec类请求（如git-receive-pack），必须遵循完整的交互流程：

1. 解析exec请求的payload，获取Git命令的具体参数
2. 启动对应命令，绑定通道与命令的标准输入/输出/错误流
3. 等待命令执行完成，且所有数据传输完毕
4. 向客户端发送exit-status请求，告知命令的退出码
5. 最后调用channel.Close()关闭通道

修改后的示例代码 #

func(in <-chan *ssh.Request) {
    for req := range in {
        switch req.Type {
        case "exec":
            // 解析exec请求携带的命令参数
            var cmdPayload struct {
                Command string
            }
            if err := ssh.Unmarshal(req.Payload, &cmdPayload); err != nil {
                req.Reply(false, nil)
                continue
            }

            channel.Stderr().Write([]byte("Request received!\n"))
            req.Reply(true, nil)

            // 启动Git命令（以git-receive-pack为例）
            cmd := exec.Command("git", "receive-pack", cmdPayload.Command)
            
            // 绑定命令与通道的标准流
            stdin, err := cmd.StdinPipe()
            if err != nil {
                channel.Stderr().Write([]byte(err.Error() + "\n"))
                break
            }
            stdout, err := cmd.StdoutPipe()
            if err != nil {
                channel.Stderr().Write([]byte(err.Error() + "\n"))
                break
            }
            stderr, err := cmd.StderrPipe()
            if err != nil {
                channel.Stderr().Write([]byte(err.Error() + "\n"))
                break
            }

            if err := cmd.Start(); err != nil {
                channel.Stderr().Write([]byte(err.Error() + "\n"))
                break
            }

            // 异步处理数据传输，确保双向数据流完成
            var streamWg sync.WaitGroup
            streamWg.Add(3)
            
            go func() {
                io.Copy(stdin, channel)
                stdin.Close() // 告知命令输入已结束
                streamWg.Done()
            }()
            go func() {
                io.Copy(channel, stdout)
                streamWg.Done()
            }()
            go func() {
                io.Copy(channel.Stderr(), stderr)
                streamWg.Done()
            }()

            // 等待所有数据传输完成，以及命令执行结束
            streamWg.Wait()
            cmd.Wait()

            // 向客户端发送标准退出状态，这是Git客户端期望的收尾步骤
            exitCode := uint32(cmd.ProcessState.ExitCode())
            statusPayload := ssh.Marshal(&struct{ Status uint32 }{exitCode})
            channel.SendRequest("exit-status", false, statusPayload)

            // 最后安全关闭通道
            channel.Close()
        default:
            req.Reply(false, nil)
        }
    }
    wg.Done()
}(requests)

关键注意点 #

• 必须解析exec请求的payload：Git客户端会把具体的仓库路径等参数放在这里，直接忽略会导致命令执行失败
• 不能提前关闭通道：必须等命令执行完、所有数据传输完毕，再发送退出状态后关闭，否则会中断Git的同步流程
• 绑定流的完整性：要同时处理输入、输出、错误流，确保Git的所有数据能正常双向传输

内容的提问来源于stack exchange，提问作者samvv