mirror of
https://github.com/bolucat/Archive.git
synced 2025-10-11 11:00:29 +08:00
58 lines
1.4 KiB
Go
58 lines
1.4 KiB
Go
package transporter
|
|
|
|
import (
|
|
"net"
|
|
|
|
"github.com/Ehco1996/ehco/internal/cmgr"
|
|
"github.com/Ehco1996/ehco/internal/constant"
|
|
"github.com/Ehco1996/ehco/internal/relay/conf"
|
|
"github.com/Ehco1996/ehco/pkg/lb"
|
|
)
|
|
|
|
type TCPHandShakeF func(remote *lb.Node) (net.Conn, error)
|
|
|
|
type RelayClient interface {
|
|
TCPHandShake(remote *lb.Node) (net.Conn, error)
|
|
RelayTCPConn(c net.Conn, handshakeF TCPHandShakeF) error
|
|
}
|
|
|
|
func newRelayClient(base *baseTransporter) (RelayClient, error) {
|
|
switch base.cfg.TransportType {
|
|
case constant.RelayTypeRaw:
|
|
return newRawClient(base)
|
|
case constant.RelayTypeWS:
|
|
return newWsClient(base)
|
|
case constant.RelayTypeWSS:
|
|
return newWssClient(base)
|
|
case constant.RelayTypeMWSS:
|
|
return newMwssClient(base)
|
|
case constant.RelayTypeMTCP:
|
|
return newMtcpClient(base)
|
|
default:
|
|
panic("unsupported transport type" + base.cfg.TransportType)
|
|
}
|
|
}
|
|
|
|
type RelayServer interface {
|
|
ListenAndServe() error
|
|
Close() error
|
|
}
|
|
|
|
func NewRelayServer(cfg *conf.Config, cmgr cmgr.Cmgr) (RelayServer, error) {
|
|
base := NewBaseTransporter(cfg, cmgr)
|
|
switch cfg.ListenType {
|
|
case constant.RelayTypeRaw:
|
|
return newRawServer(base)
|
|
case constant.RelayTypeWS:
|
|
return newWsServer(base)
|
|
case constant.RelayTypeWSS:
|
|
return newWssServer(base)
|
|
case constant.RelayTypeMWSS:
|
|
return newMwssServer(base)
|
|
case constant.RelayTypeMTCP:
|
|
return newMtcpServer(base)
|
|
default:
|
|
panic("unsupported transport type" + cfg.ListenType)
|
|
}
|
|
}
|