mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-09-06 03:54:22 -04:00
151 lines
3.7 KiB
Go
151 lines
3.7 KiB
Go
package dokodemo
|
|
|
|
import (
|
|
"io"
|
|
"sync"
|
|
|
|
"github.com/v2ray/v2ray-core/app/dispatcher"
|
|
"github.com/v2ray/v2ray-core/common/alloc"
|
|
v2io "github.com/v2ray/v2ray-core/common/io"
|
|
"github.com/v2ray/v2ray-core/common/log"
|
|
v2net "github.com/v2ray/v2ray-core/common/net"
|
|
"github.com/v2ray/v2ray-core/proxy"
|
|
"github.com/v2ray/v2ray-core/transport/hub"
|
|
)
|
|
|
|
type DokodemoDoor struct {
|
|
tcpMutex sync.RWMutex
|
|
udpMutex sync.RWMutex
|
|
config *Config
|
|
accepting bool
|
|
address v2net.Address
|
|
port v2net.Port
|
|
packetDispatcher dispatcher.PacketDispatcher
|
|
tcpListener *hub.TCPHub
|
|
udpHub *hub.UDPHub
|
|
udpServer *hub.UDPServer
|
|
listeningPort v2net.Port
|
|
}
|
|
|
|
func NewDokodemoDoor(config *Config, packetDispatcher dispatcher.PacketDispatcher) *DokodemoDoor {
|
|
return &DokodemoDoor{
|
|
config: config,
|
|
packetDispatcher: packetDispatcher,
|
|
address: config.Address,
|
|
port: config.Port,
|
|
}
|
|
}
|
|
|
|
func (this *DokodemoDoor) Port() v2net.Port {
|
|
return this.listeningPort
|
|
}
|
|
|
|
func (this *DokodemoDoor) Close() {
|
|
this.accepting = false
|
|
if this.tcpListener != nil {
|
|
this.tcpMutex.Lock()
|
|
this.tcpListener.Close()
|
|
this.tcpListener = nil
|
|
this.tcpMutex.Unlock()
|
|
}
|
|
if this.udpHub != nil {
|
|
this.udpMutex.Lock()
|
|
this.udpHub.Close()
|
|
this.udpHub = nil
|
|
this.udpMutex.Unlock()
|
|
}
|
|
}
|
|
|
|
func (this *DokodemoDoor) Listen(port v2net.Port) error {
|
|
if this.accepting {
|
|
if this.listeningPort == port {
|
|
return nil
|
|
} else {
|
|
return proxy.ErrorAlreadyListening
|
|
}
|
|
}
|
|
this.listeningPort = port
|
|
this.accepting = true
|
|
|
|
if this.config.Network.HasNetwork(v2net.TCPNetwork) {
|
|
err := this.ListenTCP(port)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
if this.config.Network.HasNetwork(v2net.UDPNetwork) {
|
|
err := this.ListenUDP(port)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (this *DokodemoDoor) ListenUDP(port v2net.Port) error {
|
|
this.udpServer = hub.NewUDPServer(this.packetDispatcher)
|
|
udpHub, err := hub.ListenUDP(port, this.handleUDPPackets)
|
|
if err != nil {
|
|
log.Error("Dokodemo failed to listen on port ", port, ": ", err)
|
|
return err
|
|
}
|
|
this.udpMutex.Lock()
|
|
this.udpHub = udpHub
|
|
this.udpMutex.Unlock()
|
|
return nil
|
|
}
|
|
|
|
func (this *DokodemoDoor) handleUDPPackets(payload *alloc.Buffer, dest v2net.Destination) {
|
|
packet := v2net.NewPacket(v2net.UDPDestination(this.address, this.port), payload, false)
|
|
this.udpServer.Dispatch(dest, packet, func(packet v2net.Packet) {
|
|
defer packet.Chunk().Release()
|
|
this.udpMutex.RLock()
|
|
if !this.accepting {
|
|
this.udpMutex.RUnlock()
|
|
return
|
|
}
|
|
this.udpHub.WriteTo(packet.Chunk().Value, packet.Destination())
|
|
this.udpMutex.RUnlock()
|
|
})
|
|
}
|
|
|
|
func (this *DokodemoDoor) ListenTCP(port v2net.Port) error {
|
|
tcpListener, err := hub.ListenTCP(port, this.HandleTCPConnection)
|
|
if err != nil {
|
|
log.Error("Dokodemo: Failed to listen on port ", port, ": ", err)
|
|
return err
|
|
}
|
|
this.tcpMutex.Lock()
|
|
this.tcpListener = tcpListener
|
|
this.tcpMutex.Unlock()
|
|
return nil
|
|
}
|
|
|
|
func (this *DokodemoDoor) HandleTCPConnection(conn *hub.TCPConn) {
|
|
defer conn.Close()
|
|
|
|
packet := v2net.NewPacket(v2net.TCPDestination(this.address, this.port), nil, true)
|
|
ray := this.packetDispatcher.DispatchToOutbound(packet)
|
|
|
|
var inputFinish, outputFinish sync.Mutex
|
|
inputFinish.Lock()
|
|
outputFinish.Lock()
|
|
|
|
reader := v2net.NewTimeOutReader(this.config.Timeout, conn)
|
|
go dumpInput(reader, ray.InboundInput(), &inputFinish)
|
|
go dumpOutput(conn, ray.InboundOutput(), &outputFinish)
|
|
|
|
outputFinish.Lock()
|
|
}
|
|
|
|
func dumpInput(reader io.Reader, input chan<- *alloc.Buffer, finish *sync.Mutex) {
|
|
v2io.RawReaderToChan(input, reader)
|
|
finish.Unlock()
|
|
close(input)
|
|
}
|
|
|
|
func dumpOutput(writer io.Writer, output <-chan *alloc.Buffer, finish *sync.Mutex) {
|
|
v2io.ChanToRawWriter(writer, output)
|
|
finish.Unlock()
|
|
}
|