From 0b81bd91e45a232012ea567f08ea4db5674c3119 Mon Sep 17 00:00:00 2001
From: Page Fault <p4gefau1t@gmail.com>
Date: Sat, 13 Jun 2020 16:19:47 +0000
Subject: [PATCH] fix tproxy udp

---
 tunnel/tproxy/server.go | 16 ++++++++++++++--
 1 file changed, 14 insertions(+), 2 deletions(-)

diff --git a/tunnel/tproxy/server.go b/tunnel/tproxy/server.go
index 3e9e8b2..d8f87dc 100644
--- a/tunnel/tproxy/server.go
+++ b/tunnel/tproxy/server.go
@@ -99,16 +99,28 @@ func (s *Server) packetDispatchLoop() {
 		}
 		s.mapping[src.String()] = conn
 		s.mappingLock.Unlock()
-
 		conn.Input <- buf[:n]
 		s.packetChan <- conn
 
 		go func(conn *dokodemo.PacketConn) {
 			defer conn.Close()
+			back, err := tproxy.DialUDP(
+				"udp",
+				&net.UDPAddr{
+					IP:   conn.M.IP,
+					Port: conn.M.Port,
+				},
+				conn.Source.(*net.UDPAddr),
+			)
+			if err != nil {
+				log.Error(common.NewError("failed to dial tproxy udp").Base(err))
+				return
+			}
+			defer back.Close()
 			for {
 				select {
 				case payload := <-conn.Output:
-					_, err := s.udpListener.WriteTo(payload, conn.Source)
+					_, err := back.Write(payload)
 					if err != nil {
 						log.Error(common.NewError("tproxy udp write error").Base(err))
 						return