Преглед изворни кода

Use buf.PacketReader when UDPOverride is available

RPRX пре 4 година
родитељ
комит
e1a5392beb
1 измењених фајлова са 1 додато и 9 уклоњено
  1. 1 9
      proxy/freedom/freedom.go

+ 1 - 9
proxy/freedom/freedom.go

@@ -195,11 +195,10 @@ func NewPacketReader(conn net.Conn, UDPOverride net.Destination) buf.Reader {
 	if statConn != nil {
 		counter = statConn.ReadCounter
 	}
-	if c, ok := iConn.(*internet.PacketConnWrapper); ok {
+	if c, ok := iConn.(*internet.PacketConnWrapper); ok && UDPOverride.Address == nil && UDPOverride.Port == 0 {
 		return &PacketReader{
 			PacketConnWrapper: c,
 			Counter:           counter,
-			UDPOverride:       UDPOverride,
 		}
 	}
 	return &buf.PacketReader{Reader: conn}
@@ -208,7 +207,6 @@ func NewPacketReader(conn net.Conn, UDPOverride net.Destination) buf.Reader {
 type PacketReader struct {
 	*internet.PacketConnWrapper
 	stats.Counter
-	UDPOverride net.Destination
 }
 
 func (r *PacketReader) ReadMultiBuffer() (buf.MultiBuffer, error) {
@@ -225,12 +223,6 @@ func (r *PacketReader) ReadMultiBuffer() (buf.MultiBuffer, error) {
 		Port:    net.Port(d.(*net.UDPAddr).Port),
 		Network: net.Network_UDP,
 	}
-	if r.UDPOverride.Address != nil {
-		b.UDP.Address = r.UDPOverride.Address
-	}
-	if r.UDPOverride.Port != 0 {
-		b.UDP.Port = r.UDPOverride.Port
-	}
 	if r.Counter != nil {
 		r.Counter.Add(int64(n))
 	}