Chore: reset udp timeout after sending each packet (#1260)

chlins · Feb 26, 2021 · f536861 · f536861
1 parent f63c9eb
commit f536861
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/tunnel/connection.go b/tunnel/connection.go
@@ -104,8 +104,13 @@ func handleUDPToRemote(packet C.UDPPacket, pc C.PacketConn, metadata *C.Metadata
 		return errors.New("udp addr invalid")
 	}
 
-	_, err := pc.WriteTo(packet.Data(), addr)
-	return err
+	if _, err := pc.WriteTo(packet.Data(), addr); err != nil {
+		return err
+	}
+	// reset timeout
+	pc.SetReadDeadline(time.Now().Add(udpTimeout))
+
+	return nil
 }
 
 func handleUDPToLocal(packet C.UDPPacket, pc net.PacketConn, key string, fAddr net.Addr) {