Wish SSH 代码审查

快速参考

审查清单

• - [ ] 主机密钥从文件加载或安全生成
• [ ] 中间件顺序正确（日志记录优先，认证尽早）
• [ ] 使用会话上下文管理每个连接的状态
• [ ] 优雅关闭处理活动会话
• [ ] 为终端应用处理PTY请求
• [ ] 连接限制防止资源耗尽
• [ ] 超时中间件防止连接挂起
• [ ] BubbleTea中间件配置正确

关键模式

服务器设置

go
// 正确 - 完整的服务器设置
s, err := wish.NewServer(
wish.WithAddress(fmt.Sprintf(%s:%d, host, port)),
wish.WithHostKeyPath(.ssh/id_ed25519),
wish.WithMiddleware(
logging.Middleware(), // 优先：记录所有连接
activeterm.Middleware(), // 处理终端尺寸
bubbletea.Middleware(teaHandler),
),
)
if err != nil {
return fmt.Errorf(创建服务器: %w, err)
}

优雅关闭

go
// 错误 - 突然关闭
log.Fatal(s.ListenAndServe())

// 正确 - 优雅关闭
done := make(chan os.Signal, 1)
signal.Notify(done, os.Interrupt, syscall.SIGTERM)

go func() {
if err := s.ListenAndServe(); err != nil && !errors.Is(err, ssh.ErrServerClosed) {
log.Error(服务器错误, error, err)
}
}()

<-done
ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
defer cancel()
if err := s.Shutdown(ctx); err != nil {
log.Error(关闭错误, error, err)
}

BubbleTea处理器

go
func teaHandler(s ssh.Session) (tea.Model, []tea.ProgramOption) {
pty, , := s.Pty()

model := NewModel(pty.Window.Width, pty.Window.Height)

return model, []tea.ProgramOption{
tea.WithAltScreen(),
tea.WithMouseCellMotion(),
}
}

何时加载参考文档

• - 审查服务器初始化 → server.md
• 审查认证、会话状态 → sessions.md

审查问题

1. 1. 主机密钥是否安全处理？
2. 中间件顺序是否正确？
3. 是否实现了优雅关闭？
4. PTY窗口尺寸是否传递给TUI？
5. 是否配置了连接超时？