Don't ignore context in DialUDP

SagerNet · Aug 28, 2021 · 03170d9 · 03170d9
1 parent 5f767d1
commit 03170d9
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/transport/internet/udp/dispatcher.go b/transport/internet/udp/dispatcher.go
@@ -121,13 +121,15 @@ func handleInput(ctx context.Context, conn *connEntry, dest net.Destination, cal
 }
 
 type dispatcherConn struct {
+	ctx        context.Context
 	dispatcher *Dispatcher
 	cache      chan *udp.Packet
 	done       *done.Instance
 }
 
 func DialDispatcher(ctx context.Context, dispatcher routing.Dispatcher) (net.PacketConn, error) {
 	c := &dispatcherConn{
+		ctx:   ctx,
 		cache: make(chan *udp.Packet, 16),
 		done:  done.New(),
 	}
@@ -168,8 +170,7 @@ func (c *dispatcherConn) WriteTo(p []byte, addr net.Addr) (int, error) {
 	n := copy(raw, p)
 	buffer.Resize(0, int32(n))
 
-	ctx := context.Background()
-	c.dispatcher.Dispatch(ctx, net.DestinationFromAddr(addr), buffer)
+	c.dispatcher.Dispatch(c.ctx, net.DestinationFromAddr(addr), buffer)
 	return n, nil
 }