mirror of
https://github.com/e1732a364fed/v2ray_simple.git
synced 2026-04-23 00:57:07 +08:00
e1732a364fed
100 lines
2.1 KiB
Go
100 lines
2.1 KiB
Go
package proxy
|
|
|
|
import (
|
|
"io"
|
|
"net"
|
|
"net/url"
|
|
|
|
"github.com/e1732a364fed/v2ray_simple/netLayer"
|
|
"github.com/e1732a364fed/v2ray_simple/utils"
|
|
)
|
|
|
|
const (
|
|
DirectName = "direct"
|
|
DirectURL = DirectName + "://"
|
|
)
|
|
|
|
//implements ClientCreator for direct
|
|
type DirectCreator struct{}
|
|
|
|
func (DirectCreator) URLToDialConf(url *url.URL, format int) (*DialConf, error) {
|
|
d := &DialConf{}
|
|
|
|
return d, nil
|
|
}
|
|
|
|
func (DirectCreator) NewClient(dc *DialConf) (Client, error) {
|
|
d := &DirectClient{}
|
|
|
|
if dc.SendThrough != "" {
|
|
st, err := netLayer.StrToNetAddr(netLayer.MixNetworkName, dc.SendThrough)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
d.localTCPAddr = st.(*netLayer.TCP_or_UDPAddr).TCPAddr
|
|
d.localUDPAddr = st.(*netLayer.TCP_or_UDPAddr).UDPAddr
|
|
|
|
}
|
|
|
|
return d, nil
|
|
}
|
|
|
|
type DirectClient struct {
|
|
Base
|
|
|
|
localTCPAddr *net.TCPAddr
|
|
localUDPAddr *net.UDPAddr
|
|
}
|
|
|
|
func (*DirectClient) Name() string { return DirectName }
|
|
|
|
func (*DirectClient) MultiTransportLayer() bool {
|
|
return true
|
|
}
|
|
|
|
//若 underlay 为nil，则会对target进行拨号, 否则返回underlay本身
|
|
func (d *DirectClient) Handshake(underlay net.Conn, firstPayload []byte, target netLayer.Addr) (result io.ReadWriteCloser, err error) {
|
|
|
|
if underlay == nil {
|
|
|
|
if d.Sockopt != nil {
|
|
result, err = target.DialWithOpt(d.Sockopt, d.localTCPAddr)
|
|
} else {
|
|
result, err = target.Dial(nil, d.localTCPAddr)
|
|
}
|
|
|
|
} else {
|
|
result = underlay
|
|
|
|
}
|
|
if err != nil {
|
|
return
|
|
}
|
|
if len(firstPayload) > 0 {
|
|
_, err = result.Write(firstPayload)
|
|
utils.PutBytes(firstPayload)
|
|
|
|
}
|
|
|
|
return
|
|
|
|
}
|
|
|
|
//direct的Client的 EstablishUDPChannel 直接 监听一个udp端口，无视传入的net.Conn.
|
|
// 这是因为要考虑到fullcone.
|
|
func (d *DirectClient) EstablishUDPChannel(_ net.Conn, firstPayload []byte, target netLayer.Addr) (netLayer.MsgConn, error) {
|
|
|
|
if len(firstPayload) == 0 {
|
|
|
|
return netLayer.NewUDPMsgConn(d.localUDPAddr, d.IsFullcone, false, d.Sockopt)
|
|
|
|
} else {
|
|
mc, err := netLayer.NewUDPMsgConn(d.localUDPAddr, d.IsFullcone, false, d.Sockopt)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return mc, mc.WriteMsgTo(firstPayload, target)
|
|
|
|
}
|
|
}
|