客户端配置，端口白名单等

server/proxy/udp.go

@@ -0,0 +1,64 @@
+package proxy
+
+import (
+	"github.com/cnlh/nps/bridge"
+	"github.com/cnlh/nps/lib/common"
+	"github.com/cnlh/nps/lib/conn"
+	"github.com/cnlh/nps/lib/file"
+	"github.com/cnlh/nps/lib/pool"
+	"net"
+	"strings"
+)
+
+type UdpModeServer struct {
+	server
+	listener *net.UDPConn
+	udpMap   map[string]*conn.Conn
+}
+
+func NewUdpModeServer(bridge *bridge.Bridge, task *file.Tunnel) *UdpModeServer {
+	s := new(UdpModeServer)
+	s.bridge = bridge
+	s.udpMap = make(map[string]*conn.Conn)
+	s.task = task
+	return s
+}
+
+//开始
+func (s *UdpModeServer) Start() error {
+	var err error
+	s.listener, err = net.ListenUDP("udp", &net.UDPAddr{net.ParseIP("0.0.0.0"), s.task.Port, ""})
+	if err != nil {
+		return err
+	}
+	buf := pool.BufPoolUdp.Get().([]byte)
+	for {
+		n, addr, err := s.listener.ReadFromUDP(buf)
+		if err != nil {
+			if strings.Contains(err.Error(), "use of closed network connection") {
+				break
+			}
+			continue
+		}
+		go s.process(addr, buf[:n])
+	}
+	return nil
+}
+
+func (s *UdpModeServer) process(addr *net.UDPAddr, data []byte) {
+	link := conn.NewLink(s.task.Client.GetId(), common.CONN_UDP, s.task.Target, s.task.Client.Cnf.CompressEncode, s.task.Client.Cnf.CompressDecode, s.task.Client.Cnf.Crypt, nil, s.task.Flow, s.listener, s.task.Client.Rate, addr)
+	if err := s.checkFlow(); err != nil {
+		return
+	}
+	if tunnel, err := s.bridge.SendLinkInfo(s.task.Client.Id, link); err != nil {
+		return
+	} else {
+		s.task.Flow.Add(len(data), 0)
+		tunnel.SendMsg(data, link)
+		pool.PutBufPoolUdp(data)
+	}
+}
+
+func (s *UdpModeServer) Close() error {
+	return s.listener.Close()
+}