mirror of
https://github.com/Gouryella/drip.git
synced 2026-02-27 14:50:52 +00:00
0c19c3300c
- Introduce pooled tunnel sessions (TunnelID/DataConnect) on client/server - Proxy HTTP/HTTPS via raw HTTP over yamux streams; pipe TCP streams directly - Move UI/stats into internal/shared; refactor CLI tunnel helpers; drop msgpack/hpack legacy
65 lines
2.8 KiB
Go
65 lines
2.8 KiB
Go
package protocol
|
|
|
|
import json "github.com/goccy/go-json"
|
|
|
|
// PoolCapabilities advertises client connection pool capabilities
|
|
type PoolCapabilities struct {
|
|
MaxDataConns int `json:"max_data_conns"` // Maximum data connections client supports
|
|
Version int `json:"version"` // Protocol version for pool features
|
|
}
|
|
|
|
// RegisterRequest is sent by client to register a tunnel
|
|
type RegisterRequest struct {
|
|
Token string `json:"token"` // Authentication token
|
|
CustomSubdomain string `json:"custom_subdomain"` // Optional custom subdomain
|
|
TunnelType TunnelType `json:"tunnel_type"` // http, tcp, udp
|
|
LocalPort int `json:"local_port"` // Local port to forward to
|
|
|
|
// Connection pool fields (optional, for multi-connection support)
|
|
ConnectionType string `json:"connection_type,omitempty"` // "primary" or empty for legacy
|
|
TunnelID string `json:"tunnel_id,omitempty"` // For data connections to join
|
|
PoolCapabilities *PoolCapabilities `json:"pool_capabilities,omitempty"` // Client pool capabilities
|
|
}
|
|
|
|
// RegisterResponse is sent by server after successful registration
|
|
type RegisterResponse struct {
|
|
Subdomain string `json:"subdomain"` // Assigned subdomain
|
|
Port int `json:"port,omitempty"` // Assigned TCP port (for TCP tunnels)
|
|
URL string `json:"url"` // Full tunnel URL
|
|
Message string `json:"message"` // Success message
|
|
|
|
// Connection pool fields (optional, for multi-connection support)
|
|
TunnelID string `json:"tunnel_id,omitempty"` // Unique tunnel identifier
|
|
SupportsDataConn bool `json:"supports_data_conn,omitempty"` // Server supports multi-connection
|
|
RecommendedConns int `json:"recommended_conns,omitempty"` // Suggested data connection count
|
|
}
|
|
|
|
// DataConnectRequest is sent by data connections to join a tunnel
|
|
type DataConnectRequest struct {
|
|
TunnelID string `json:"tunnel_id"` // Tunnel to join
|
|
Token string `json:"token"` // Same auth token as primary
|
|
ConnectionID string `json:"connection_id"` // Unique connection identifier
|
|
}
|
|
|
|
// DataConnectResponse acknowledges data connection
|
|
type DataConnectResponse struct {
|
|
Accepted bool `json:"accepted"` // Whether connection was accepted
|
|
ConnectionID string `json:"connection_id"` // Echoed connection ID
|
|
Message string `json:"message,omitempty"` // Optional message
|
|
}
|
|
|
|
// ErrorMessage represents an error
|
|
type ErrorMessage struct {
|
|
Code string `json:"code"` // Error code
|
|
Message string `json:"message"` // Error message
|
|
}
|
|
|
|
// Marshal helpers for control plane messages (JSON encoding)
|
|
func MarshalJSON(v interface{}) ([]byte, error) {
|
|
return json.Marshal(v)
|
|
}
|
|
|
|
func UnmarshalJSON(data []byte, v interface{}) error {
|
|
return json.Unmarshal(data, v)
|
|
}
|