starainrt
f20eb653ae
- 拆分原有单体 ssh.go，按职责重组为 types、utils、transport、login、keepalive、session、exec、pool、shell、terminal、forward、hostkey、state 等模块，并补充平台相关实现 - 重做登录与连接运行时，补齐基于 context 的建连、jump/proxy 链路、可配置认证顺序，以及 Unix/Windows 下的 ssh-agent 支持 - 新增正式非交互执行模型 ExecRequest/ExecResult，支持流式输出、溢出统计、超时控制，以及 posix/powershell/cmd/raw 多方言执行 - 保留旧 shell 风格兼容接口，同时让路径/用户探测等 helper 具备跨 shell fallback，避免 Windows 目标继续硬依赖 POSIX 命令 - 新增 TerminalSession 作为原始交互终端基座，提供 IO attach、resize、signal/control、退出状态与关闭原因管理 - 重构端口转发语义，默认复用当前 SSH 连接，并显式提供 detached 的本地/动态转发模式承载隔离场景 - 梳理 keepalive 与取消语义，区分仅取消本次操作和关闭整条连接，并统一连接状态与传输关闭路径 - 围绕新的 session/连接生命周期重做执行池与运行时支撑 - 大幅增强 SFTP 传输链路，补齐更安全的原子替换、校验、进度回调、重试隔离、可复用 client 生命周期与失败语义 - 新增取消语义、keepalive、SFTP、forward、terminal input 等关键回归测试，提升核心链路稳定性
122 lines
2.3 KiB
Go
122 lines
2.3 KiB
Go
package starssh
|
|
|
|
import (
|
|
"errors"
|
|
"io"
|
|
"net"
|
|
"strings"
|
|
|
|
"golang.org/x/crypto/ssh"
|
|
)
|
|
|
|
func (s *StarSSH) Close() error {
|
|
return s.closeTransport(true)
|
|
}
|
|
|
|
func (s *StarSSH) NewSession() (*ssh.Session, error) {
|
|
return s.NewPTYSession(nil)
|
|
}
|
|
|
|
func (s *StarSSH) NewExecSession() (*ssh.Session, error) {
|
|
client, err := s.requireSSHClient()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return NewExecSession(client)
|
|
}
|
|
|
|
func (s *StarSSH) NewPTYSession(config *TerminalConfig) (*ssh.Session, error) {
|
|
client, err := s.requireSSHClient()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return NewPTYSession(client, config)
|
|
}
|
|
|
|
func NewTransferSession(client *ssh.Client) (*ssh.Session, error) {
|
|
return NewExecSession(client)
|
|
}
|
|
|
|
func NewExecSession(client *ssh.Client) (*ssh.Session, error) {
|
|
if client == nil {
|
|
return nil, errors.New("ssh client is nil")
|
|
}
|
|
return client.NewSession()
|
|
}
|
|
|
|
func NewSession(client *ssh.Client) (*ssh.Session, error) {
|
|
return NewPTYSession(client, nil)
|
|
}
|
|
|
|
func NewPTYSession(client *ssh.Client, config *TerminalConfig) (*ssh.Session, error) {
|
|
if client == nil {
|
|
return nil, errors.New("ssh client is nil")
|
|
}
|
|
|
|
session, err := client.NewSession()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
cfg := normalizeTerminalConfig(config)
|
|
if err := session.RequestPty(cfg.Term, cfg.Rows, cfg.Columns, cfg.Modes); err != nil {
|
|
_ = session.Close()
|
|
return nil, err
|
|
}
|
|
|
|
return session, nil
|
|
}
|
|
|
|
func normalizeTerminalConfig(config *TerminalConfig) TerminalConfig {
|
|
cfg := TerminalConfig{
|
|
Term: defaultPTYTerm,
|
|
Rows: defaultPTYRows,
|
|
Columns: defaultPTYColumns,
|
|
Modes: ssh.TerminalModes{
|
|
ssh.ECHO: 1,
|
|
ssh.TTY_OP_ISPEED: 14400,
|
|
ssh.TTY_OP_OSPEED: 14400,
|
|
},
|
|
}
|
|
if config == nil {
|
|
return cfg
|
|
}
|
|
|
|
if strings.TrimSpace(config.Term) != "" {
|
|
cfg.Term = config.Term
|
|
}
|
|
if config.Rows > 0 {
|
|
cfg.Rows = config.Rows
|
|
}
|
|
if config.Columns > 0 {
|
|
cfg.Columns = config.Columns
|
|
}
|
|
if len(config.Modes) > 0 {
|
|
cfg.Modes = config.Modes
|
|
}
|
|
return cfg
|
|
}
|
|
|
|
func normalizeAlreadyClosedError(err error) error {
|
|
if err == nil {
|
|
return nil
|
|
}
|
|
if errors.Is(err, net.ErrClosed) {
|
|
return nil
|
|
}
|
|
if errors.Is(err, io.EOF) {
|
|
return nil
|
|
}
|
|
if strings.Contains(err.Error(), "use of closed network connection") {
|
|
return nil
|
|
}
|
|
return err
|
|
}
|
|
|
|
func closeUpstream(upstream *StarSSH) error {
|
|
if upstream == nil {
|
|
return nil
|
|
}
|
|
return upstream.Close()
|
|
}
|