Prechádzať zdrojové kódy

refactor udp worker

Darien Raymond 7 rokov pred
rodič
commit
59958fed68
1 zmenil súbory, kde vykonal 4 pridanie a 2 odobranie
  1. 4 2
      app/proxyman/inbound/worker.go

+ 4 - 2
app/proxyman/inbound/worker.go

@@ -210,8 +210,10 @@ func (w *udpWorker) getConnection(id connID) (*udpConn, bool) {
 
 func (w *udpWorker) callback(b *buf.Buffer, source net.Destination, originalDest net.Destination) {
 	id := connID{
-		src:  source,
-		dest: originalDest,
+		src: source,
+	}
+	if originalDest.IsValid() {
+		id.dest = originalDest
 	}
 	conn, existing := w.getConnection(id)
 	select {