f

Author: nekohasekai

tcpip/udpnat/service.go

@@ -71,19 +71,19 @@ func (s *Service) HandlePacket(source netip.AddrPort, destination netip.AddrPort
 	}
 }
 
-func (s *Service) HandlePacketControl(source netip.AddrPort, destination netip.AddrPort, packet *buffer.PacketBuffer, controlMessage []byte, constructor func(natConn *NATConn) (context.Context, N.PacketWriter)) {
+func (s *Service) HandlePacketControl(source netip.AddrPort, packet *buffer.PacketBuffer, controlMessage []byte, constructor func(natConn *NATConn) (context.Context, N.PacketWriter)) {
 	conn, loaded := s.table.LoadOrStore(source, createNatConn)
 	if !loaded {
 		conn.ctx, conn.writer = constructor(conn)
 		conn.ctx, conn.cancel = common.ContextWithCancelCause(conn.ctx)
 		*conn = NATConn{
 			source:      source,
-			destination: destination,
+			destination: packet.Destination,
 			recvChan:    make(chan *buffer.PacketBuffer, s.capacity),
 		}
 	} else if common.Done(conn.ctx) {
 		s.table.Delete(source)
-		s.HandlePacketControl(source, destination, packet, controlMessage, constructor)
+		s.HandlePacketControl(source, packet, controlMessage, constructor)
 		return
 	}
 	if udpcontrol.SocketControlMessageBufferSize > 0 {