mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-09-03 02:04:34 -04:00
60 lines
1.2 KiB
Go
60 lines
1.2 KiB
Go
|
package hub
|
||
|
|
||
|
import (
|
||
|
"net"
|
||
|
|
||
|
"github.com/v2ray/v2ray-core/common/alloc"
|
||
|
v2net "github.com/v2ray/v2ray-core/common/net"
|
||
|
)
|
||
|
|
||
|
type UDPPayloadHandler func(*alloc.Buffer, v2net.Destination)
|
||
|
|
||
|
type UDPHub struct {
|
||
|
conn *net.UDPConn
|
||
|
callback UDPPayloadHandler
|
||
|
accepting bool
|
||
|
}
|
||
|
|
||
|
func ListenUDP(port v2net.Port, callback UDPPayloadHandler) (*UDPHub, error) {
|
||
|
udpConn, err := net.ListenUDP("udp", &net.UDPAddr{
|
||
|
IP: []byte{0, 0, 0, 0},
|
||
|
Port: int(port),
|
||
|
})
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
hub := &UDPHub{
|
||
|
conn: udpConn,
|
||
|
callback: callback,
|
||
|
}
|
||
|
go hub.start()
|
||
|
return hub, nil
|
||
|
}
|
||
|
|
||
|
func (this *UDPHub) Close() {
|
||
|
this.accepting = false
|
||
|
this.conn.Close()
|
||
|
}
|
||
|
|
||
|
func (this *UDPHub) WriteTo(payload []byte, dest v2net.Destination) (int, error) {
|
||
|
return this.conn.WriteToUDP(payload, &net.UDPAddr{
|
||
|
IP: dest.Address().IP(),
|
||
|
Port: int(dest.Port()),
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func (this *UDPHub) start() {
|
||
|
this.accepting = true
|
||
|
for this.accepting {
|
||
|
buffer := alloc.NewBuffer()
|
||
|
nBytes, addr, err := this.conn.ReadFromUDP(buffer.Value)
|
||
|
if err != nil {
|
||
|
buffer.Release()
|
||
|
continue
|
||
|
}
|
||
|
buffer.Slice(0, nBytes)
|
||
|
dest := v2net.UDPDestination(v2net.IPAddress(addr.IP), v2net.Port(addr.Port))
|
||
|
go this.callback(buffer, dest)
|
||
|
}
|
||
|
}
|