mirror of https://github.com/v2ray/v2ray-core
53 lines
1.2 KiB
Go
53 lines
1.2 KiB
Go
package udp
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"v2ray.com/core/common/net"
|
|
)
|
|
|
|
type Server struct {
|
|
Port net.Port
|
|
MsgProcessor func(msg []byte) []byte
|
|
accepting bool
|
|
conn *net.UDPConn
|
|
}
|
|
|
|
func (server *Server) Start() (net.Destination, error) {
|
|
conn, err := net.ListenUDP("udp", &net.UDPAddr{
|
|
IP: []byte{127, 0, 0, 1},
|
|
Port: int(server.Port),
|
|
Zone: "",
|
|
})
|
|
if err != nil {
|
|
return net.Destination{}, err
|
|
}
|
|
server.Port = net.Port(conn.LocalAddr().(*net.UDPAddr).Port)
|
|
server.conn = conn
|
|
go server.handleConnection(conn)
|
|
localAddr := conn.LocalAddr().(*net.UDPAddr)
|
|
return net.UDPDestination(net.IPAddress(localAddr.IP), net.Port(localAddr.Port)), nil
|
|
}
|
|
|
|
func (server *Server) handleConnection(conn *net.UDPConn) {
|
|
server.accepting = true
|
|
for server.accepting {
|
|
buffer := make([]byte, 2*1024)
|
|
nBytes, addr, err := conn.ReadFromUDP(buffer)
|
|
if err != nil {
|
|
fmt.Printf("Failed to read from UDP: %v\n", err)
|
|
continue
|
|
}
|
|
|
|
response := server.MsgProcessor(buffer[:nBytes])
|
|
if _, err := conn.WriteToUDP(response, addr); err != nil {
|
|
fmt.Println("Failed to write to UDP: ", err.Error())
|
|
}
|
|
}
|
|
}
|
|
|
|
func (server *Server) Close() {
|
|
server.accepting = false
|
|
server.conn.Close()
|
|
}
|