mirror of
https://github.com/e1732a364fed/v2ray_simple.git
synced 2025-12-24 13:27:56 +08:00
158 lines
4.6 KiB
Go
158 lines
4.6 KiB
Go
package proxy
|
||
|
||
import (
|
||
"io"
|
||
"net"
|
||
"strings"
|
||
"sync"
|
||
"time"
|
||
|
||
"github.com/e1732a364fed/v2ray_simple/netLayer"
|
||
"github.com/e1732a364fed/v2ray_simple/utils"
|
||
"github.com/xtaci/smux"
|
||
)
|
||
|
||
// 规定，如果 proxy的server的handshake如果返回的是具有内层mux的连接，该连接要实现 MuxMarker 接口.
|
||
type MuxMarker interface {
|
||
io.ReadWriteCloser
|
||
IsMux() bool
|
||
}
|
||
|
||
// 实现 utils.MuxMarker, utils.User
|
||
type UserReadWrapper struct {
|
||
utils.User
|
||
netLayer.ReadWrapper
|
||
Mux bool
|
||
}
|
||
|
||
func (w *UserReadWrapper) IsMux() bool { return w.Mux }
|
||
|
||
// some client may 建立tcp连接后首先由客户端读服务端的数据？虽较少见但确实存在.
|
||
// Anyway firstpayload might not be read, and we should try to reduce this delay.
|
||
// 也有可能是有人用 nc 来测试，也会遇到这种读不到 firstpayload 的情况
|
||
const FirstPayloadTimeout = time.Millisecond * 100
|
||
|
||
// Client is used to dial a server.
|
||
// Because Server is "target agnostic", Client's Handshake requires a target addr as param.
|
||
//
|
||
// A Client has all the data of all layers in its VSI model.
|
||
// Once a Client is fully defined, the flow of the data is fully defined.
|
||
type Client interface {
|
||
BaseInterface
|
||
|
||
//Perform handshake when request is TCP。firstPayload 用于如 vless/trojan 这种 没有握手包的协议，可为空。
|
||
Handshake(underlay net.Conn, firstPayload []byte, target netLayer.Addr) (wrappedConn io.ReadWriteCloser, err error)
|
||
|
||
//Establish a channel and constantly request data for each UDP addr through this channel. firstpayload and target can be empty theoretically, depending on the implementation.
|
||
EstablishUDPChannel(underlay net.Conn, firstPayload []byte, target netLayer.Addr) (netLayer.MsgConn, error)
|
||
|
||
//If udp is send through multiple connection or not
|
||
IsUDP_MultiChannel() bool
|
||
|
||
//get/listen a useable inner mux
|
||
GetClientInnerMuxSession(wrc io.ReadWriteCloser) *smux.Session
|
||
InnerMuxEstablished() bool
|
||
CloseInnerMuxSession()
|
||
|
||
//用于在拨号时选用一个特定的ip拨号。
|
||
LocalTCPAddr() *net.TCPAddr
|
||
LocalUDPAddr() *net.UDPAddr
|
||
|
||
GetCreator() ClientCreator
|
||
|
||
sync.Locker //用于锁定 innerMux
|
||
}
|
||
|
||
type UserClient interface {
|
||
Client
|
||
GetUser() utils.User
|
||
}
|
||
|
||
// Server is used for listening clients.
|
||
// Because Server is "target agnostic"，Handshake should return the target addr that the Client requested.
|
||
//
|
||
// A Server has all the data of all layers in its VSI model.
|
||
// Once a Server is fully defined, the flow of the data is fully defined.
|
||
type Server interface {
|
||
BaseInterface
|
||
|
||
//net.Conn is for TCP request, netLayer.MsgConn is for UDP request.
|
||
// 约定，如果error返回的是 utils.ErrHandled， 则调用代码停止进一步处理。
|
||
Handshake(underlay net.Conn) (net.Conn, netLayer.MsgConn, netLayer.Addr, error)
|
||
|
||
//get/listen a useable inner mux
|
||
GetServerInnerMuxSession(wlc io.ReadWriteCloser) *smux.Session
|
||
|
||
//tproxy,tun 和 shadowsocks(udp) 都用到了 SelfListen
|
||
//
|
||
//is表示开启自监听; 此时若 tcp=1, 表示监听tcp, 若tcp=0, 表示自己不监听tcp, 但需要vs进行监听; 若tcp<0, 则表示自己不监听, 也不要vs监听; udp同理; 开启SelfListen同时表明 Server实现了 ListenerServer
|
||
SelfListen() (is bool, tcp, udp int)
|
||
}
|
||
|
||
type ListenerServer interface {
|
||
Server
|
||
|
||
//非阻塞
|
||
StartListen(func(netLayer.TCPRequestInfo), func(netLayer.UDPRequestInfo)) io.Closer
|
||
}
|
||
|
||
type UserServer interface {
|
||
Server
|
||
utils.UserContainer
|
||
}
|
||
|
||
// FullName can fully represent the VSI model for a proxy.
|
||
// We think tcp/udp/kcp/raw_socket is FirstName，protocol of the proxy is LastName, and the rest is MiddleName。
|
||
//
|
||
// An Example of a full name: tcp+tls+ws+vless.
|
||
func GetFullName(pc BaseInterface) string {
|
||
|
||
return getFullNameBuilder(pc, pc.Name()).String()
|
||
}
|
||
|
||
// return GetFullName(pc) + "://" + pc.AddrStr() (+ #tag)
|
||
func GetVSI_url(pc BaseInterface, targetNetwork string) string {
|
||
n := pc.Name()
|
||
|
||
sb := getFullNameBuilder(pc, n)
|
||
sb.WriteString("://")
|
||
if n == DirectName {
|
||
if targetNetwork == "tcp" {
|
||
if lta := pc.GetBase().LTA; lta != nil {
|
||
sb.WriteString(lta.String())
|
||
}
|
||
} else if targetNetwork == "udp" {
|
||
if lua := pc.GetBase().LUA; lua != nil {
|
||
sb.WriteString(lua.String())
|
||
}
|
||
}
|
||
|
||
} else {
|
||
sb.WriteString(pc.AddrStr())
|
||
}
|
||
|
||
if t := pc.GetTag(); t != "" {
|
||
sb.WriteByte('#')
|
||
sb.WriteString(t)
|
||
}
|
||
|
||
return sb.String()
|
||
}
|
||
|
||
func getFullNameBuilder(pc BaseInterface, n string) *strings.Builder {
|
||
|
||
var sb strings.Builder
|
||
sb.WriteString(pc.Network())
|
||
sb.WriteString(pc.MiddleName())
|
||
sb.WriteString(n)
|
||
|
||
if i, innerProxyName := pc.HasInnerMux(); i == 2 {
|
||
sb.WriteString("+smux+")
|
||
sb.WriteString(innerProxyName)
|
||
|
||
}
|
||
|
||
return &sb
|
||
|
||
}
|