proxy

type ANSIFilter struct {
	// contains filtered or unexported fields
}

type CloseConnectionRequest struct {
	SessionID   string `json:"session_id"`
	CloseReason string `json:"close_reason,omitempty"`
}

type Connection struct {
	// 会话ID
	SessionID string `json:"session_id"`
	// 协议类型
	Protocol Protocol `json:"protocol"`
	// 目标信息
	ResourceID string `json:"resource_id"`
	TargetHost string `json:"target_host"`
	TargetPort int    `json:"target_port"`
	// 用户信息
	Username string `json:"username"`
	UserID   string `json:"user_id"`
	ClientIP string `json:"client_ip"`
	// 状态信息
	Status     ConnectionStatus `json:"status"`
	CreateTime time.Time        `json:"create_time"`
	CloseTime  *time.Time       `json:"close_time,omitempty"`
	// 数据统计
	BytesSent     int64 `json:"bytes_sent"`
	BytesReceived int64 `json:"bytes_received"`
	// 扩展信息
	Metadata map[string]string `json:"metadata,omitempty"`
}

type ConnectionHandler interface {
	// HandleConnection 处理单个客户端连接
	// 返回错误时，TCPServer 会关闭连接
	HandleConnection(ctx context.Context, conn net.Conn) error
}

type ConnectionSet struct {
	Total int64         `json:"total"`
	Items []*Connection `json:"items"`
}

type ConnectionStatus string

type CreateConnectionRequest struct {
	// 协议类型
	Protocol Protocol `json:"protocol"`
	// 目标资源ID
	ResourceID string `json:"resource_id"`
	// 目标主机
	TargetHost string `json:"target_host"`
	// 目标端口
	TargetPort int `json:"target_port"`
	// 认证信息
	Username string `json:"username"`
	Password string `json:"password,omitempty"`
	// 用户信息（审计用）
	UserID   string `json:"user_id"`
	ClientIP string `json:"client_ip"`
	// 扩展信息
	Metadata map[string]string `json:"metadata,omitempty"`
}

type DangerousPattern struct {
	// 操作类型（e.g., "DROP", "DELETE", "TRUNCATE"）
	Operation string
	// 风险等级
	RiskLevel RiskLevel
	// 是否应该拦截
	ShouldBlock bool
	// 拦截原因
	BlockReason string
}

type Direction string

type GetConnectionRequest struct {
	SessionID string `json:"session_id"`
}

type ListConnectionsRequest struct {
	Page     int64            `json:"page"`
	PageSize int64            `json:"page_size"`
	Protocol Protocol         `json:"protocol,omitempty"`
	UserID   string           `json:"user_id,omitempty"`
	Status   ConnectionStatus `json:"status,omitempty"`
}

type ParseResult struct {
	// 解析出的事件列表
	Events []*audit.OperationEvent
	// 是否应该拦截
	ShouldBlock bool
	// 拦截原因
	BlockReason string
}

type Parser interface {
	// Parse 解析数据流，返回审计事件
	// data: 原始数据
	// direction: 数据流方向
	// sessionID: 会话ID
	Parse(ctx context.Context, data []byte, direction Direction, sessionID string) (*ParseResult, error)

	// DetectRisk 检测风险等级
	DetectRisk(event *audit.OperationEvent) RiskLevel

	// ShouldBlock 判断是否应该拦截
	ShouldBlock(event *audit.OperationEvent) bool
}

type Protocol string

type Recorder interface {
	// Start 开始录制
	Start(ctx context.Context, sessionID string, metadata map[string]interface{}) error

	// Write 写入数据
	// timestamp: 相对于会话开始的时间偏移（秒）
	// ioType: 输入输出类型（'i' input, 'o' output）
	// data: 数据内容
	Write(timestamp float64, ioType rune, data []byte) error

	// Stop 结束录制，返回录像文件路径
	Stop(ctx context.Context) (recordingPath string, err error)

	// Status 获取录制状态
	Status() RecordingStatus

	// Protocol 返回录制器支持的协议
	Protocol() audit.Protocol
}

type RecorderFactory interface {
	// GetRecorder 根据协议获取录制器
	GetRecorder(protocol audit.Protocol) (Recorder, error)
}

type RecordingStatus string

type RiskLevel string

type Service interface {
	// 创建代理连接
	CreateConnection(context.Context, *CreateConnectionRequest) (*Connection, error)

	// 获取连接信息
	GetConnection(context.Context, *GetConnectionRequest) (*Connection, error)

	// 列出所有连接
	ListConnections(context.Context, *ListConnectionsRequest) (*ConnectionSet, error)

	// 关闭连接
	CloseConnection(context.Context, *CloseConnectionRequest) error
}

type TCPServer struct {
	// contains filtered or unexported fields
}

type TCPServerMetrics struct {
	TotalConnections   atomic.Int64 // 总连接数
	ActiveConnections  atomic.Int64 // 活跃连接数
	FailedConnections  atomic.Int64 // 失败连接数
	TotalBytesSent     atomic.Int64 // 总发送字节数
	TotalBytesReceived atomic.Int64 // 总接收字节数
	LastStartTime      time.Time    // 最后启动时间
}

type TCPSession struct {
	SessionID     string
	ClientConn    net.Conn
	CreateTime    time.Time
	BytesSent     atomic.Int64
	BytesReceived atomic.Int64
	// contains filtered or unexported fields
}

type WebSocketConn interface {
	io.ReadWriteCloser
	// SetReadDeadline 设置读超时
	SetReadDeadline(t time.Time) error
	// SetWriteDeadline 设置写超时
	SetWriteDeadline(t time.Time) error
}