fixes #370 #368

ehang-io · Jan 15, 2020 · f72e356 · f72e356
1 parent 7260565
commit f72e356
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/server/proxy/udp.go b/server/proxy/udp.go
@@ -55,9 +55,10 @@ func (s *UdpModeServer) process(addr *net.UDPAddr, data []byte) {
 	}
 	defer s.task.Client.AddConn()
 	link := conn.NewLink(common.CONN_UDP, s.task.Target.TargetStr, s.task.Client.Cnf.Crypt, s.task.Client.Cnf.Compress, addr.String(), s.task.Target.LocalProxy)
-	if target, err := s.bridge.SendLinkInfo(s.task.Client.Id, link, s.task); err != nil {
+	if clientConn, err := s.bridge.SendLinkInfo(s.task.Client.Id, link, s.task); err != nil {
 		return
 	} else {
+		target := conn.GetConn(clientConn, s.task.Client.Cnf.Crypt, s.task.Client.Cnf.Compress, nil, true)
 		s.task.Flow.Add(int64(len(data)), 0)
 		buf := common.BufPoolUdp.Get().([]byte)
 		defer common.BufPoolUdp.Put(buf)