2015-10-30 10:56:46 -04:00
|
|
|
package dokodemo
|
|
|
|
|
|
|
|
import (
|
|
|
|
"io"
|
|
|
|
"net"
|
|
|
|
"sync"
|
|
|
|
|
|
|
|
"github.com/v2ray/v2ray-core/app"
|
|
|
|
"github.com/v2ray/v2ray-core/common/alloc"
|
|
|
|
"github.com/v2ray/v2ray-core/common/log"
|
|
|
|
v2net "github.com/v2ray/v2ray-core/common/net"
|
|
|
|
"github.com/v2ray/v2ray-core/common/retry"
|
|
|
|
)
|
|
|
|
|
|
|
|
type DokodemoDoor struct {
|
2015-12-06 12:21:15 -05:00
|
|
|
config Config
|
2015-12-05 16:55:45 -05:00
|
|
|
accepting bool
|
|
|
|
address v2net.Address
|
2015-12-11 06:01:20 -05:00
|
|
|
space app.Space
|
2015-10-30 10:56:46 -04:00
|
|
|
}
|
|
|
|
|
2015-12-11 06:01:20 -05:00
|
|
|
func NewDokodemoDoor(space app.Space, config Config) *DokodemoDoor {
|
2015-12-06 12:21:15 -05:00
|
|
|
return &DokodemoDoor{
|
|
|
|
config: config,
|
|
|
|
space: space,
|
|
|
|
address: config.Address(),
|
2015-10-30 10:56:46 -04:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-12-02 15:44:01 -05:00
|
|
|
func (this *DokodemoDoor) Listen(port v2net.Port) error {
|
2015-11-10 18:08:43 -05:00
|
|
|
this.accepting = true
|
|
|
|
|
2015-12-06 12:21:15 -05:00
|
|
|
if this.config.Network().HasNetwork(v2net.TCPNetwork) {
|
2015-10-30 10:56:46 -04:00
|
|
|
err := this.ListenTCP(port)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
}
|
2015-12-06 12:21:15 -05:00
|
|
|
if this.config.Network().HasNetwork(v2net.UDPNetwork) {
|
2015-11-10 18:08:43 -05:00
|
|
|
err := this.ListenUDP(port)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
}
|
2015-10-30 10:56:46 -04:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2015-12-02 15:44:01 -05:00
|
|
|
func (this *DokodemoDoor) ListenUDP(port v2net.Port) error {
|
2015-11-10 18:08:43 -05:00
|
|
|
udpConn, err := net.ListenUDP("udp", &net.UDPAddr{
|
|
|
|
IP: []byte{0, 0, 0, 0},
|
|
|
|
Port: int(port),
|
|
|
|
Zone: "",
|
|
|
|
})
|
|
|
|
if err != nil {
|
|
|
|
log.Error("Dokodemo failed to listen on port %d: %v", port, err)
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
go this.handleUDPPackets(udpConn)
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (this *DokodemoDoor) handleUDPPackets(udpConn *net.UDPConn) {
|
|
|
|
defer udpConn.Close()
|
|
|
|
for this.accepting {
|
|
|
|
buffer := alloc.NewBuffer()
|
|
|
|
nBytes, addr, err := udpConn.ReadFromUDP(buffer.Value)
|
|
|
|
buffer.Slice(0, nBytes)
|
|
|
|
if err != nil {
|
|
|
|
buffer.Release()
|
|
|
|
log.Error("Dokodemo failed to read from UDP: %v", err)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
packet := v2net.NewPacket(v2net.NewUDPDestination(this.address), buffer, false)
|
2015-12-05 16:55:45 -05:00
|
|
|
ray := this.space.PacketDispatcher().DispatchToOutbound(packet)
|
2015-11-10 18:08:43 -05:00
|
|
|
close(ray.InboundInput())
|
|
|
|
|
|
|
|
for payload := range ray.InboundOutput() {
|
|
|
|
udpConn.WriteToUDP(payload.Value, addr)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-12-02 15:44:01 -05:00
|
|
|
func (this *DokodemoDoor) ListenTCP(port v2net.Port) error {
|
2015-10-30 10:56:46 -04:00
|
|
|
tcpListener, err := net.ListenTCP("tcp", &net.TCPAddr{
|
|
|
|
IP: []byte{0, 0, 0, 0},
|
|
|
|
Port: int(port),
|
|
|
|
Zone: "",
|
|
|
|
})
|
|
|
|
if err != nil {
|
|
|
|
log.Error("Dokodemo failed to listen on port %d: %v", port, err)
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
go this.AcceptTCPConnections(tcpListener)
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (this *DokodemoDoor) AcceptTCPConnections(tcpListener *net.TCPListener) {
|
|
|
|
for this.accepting {
|
|
|
|
retry.Timed(100, 100).On(func() error {
|
|
|
|
connection, err := tcpListener.AcceptTCP()
|
|
|
|
if err != nil {
|
|
|
|
log.Error("Dokodemo failed to accept new connections: %v", err)
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
go this.HandleTCPConnection(connection)
|
|
|
|
return nil
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func (this *DokodemoDoor) HandleTCPConnection(conn *net.TCPConn) {
|
|
|
|
defer conn.Close()
|
|
|
|
|
|
|
|
packet := v2net.NewPacket(v2net.NewTCPDestination(this.address), nil, true)
|
2015-12-05 16:55:45 -05:00
|
|
|
ray := this.space.PacketDispatcher().DispatchToOutbound(packet)
|
2015-10-30 10:56:46 -04:00
|
|
|
|
|
|
|
var inputFinish, outputFinish sync.Mutex
|
|
|
|
inputFinish.Lock()
|
|
|
|
outputFinish.Lock()
|
|
|
|
|
2015-12-06 12:21:15 -05:00
|
|
|
reader := v2net.NewTimeOutReader(this.config.Timeout(), conn)
|
2015-10-30 10:56:46 -04:00
|
|
|
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) {
|
|
|
|
v2net.ReaderToChan(input, reader)
|
|
|
|
finish.Unlock()
|
|
|
|
close(input)
|
|
|
|
}
|
|
|
|
|
|
|
|
func dumpOutput(writer io.Writer, output <-chan *alloc.Buffer, finish *sync.Mutex) {
|
|
|
|
v2net.ChanToWriter(writer, output)
|
|
|
|
finish.Unlock()
|
|
|
|
}
|