X-Git-Url: http://git.meshlink.io/?a=blobdiff_plain;f=src%2Fpmtu.c;h=36152cffec2af6d8d23a61f32d5a8fa7ffa202c9;hb=689fd74cf976264f54d9c61f09ef5338fa5c5f23;hp=d91919eed40f9e95a653f00d7cad55389cd293a7;hpb=0669c133ef0150e9772711588651c1c4d8e45267;p=meshlink

diff --git a/src/pmtu.c b/src/pmtu.c
index d91919ee..36152cff 100644
--- a/src/pmtu.c
+++ b/src/pmtu.c
@@ -91,6 +91,7 @@ static void udp_probe_timeout_handler(event_loop_t *loop, void *data) {
 	logger(mesh, MESHLINK_INFO, "Too much time has elapsed since last UDP ping response from %s, stopping UDP communication", n->name);
 	n->status.udp_confirmed = false;
 	n->mtuprobes = 0;
+	n->udpprobes = 0;
 	n->minmtu = 0;
 	n->maxmtu = MTU;
 
@@ -161,6 +162,8 @@ void udp_probe_h(meshlink_handle_t *mesh, node_t *n, vpn_packet_t *packet, uint1
 		n->status.udp_confirmed = true;
 	}
 
+	n->udpprobes = 0;
+
 	// Reset the UDP ping timer.
 
 	timeout_del(&mesh->loop, &n->udp_ping_timeout);
@@ -219,15 +222,25 @@ static void send_udp_probe_packet(meshlink_handle_t *mesh, node_t *n, int len) {
 static void try_udp(meshlink_handle_t *mesh, node_t *n) {
 	/* Probe request */
 
+	if(n->udpprobes < -3) {
+		/* We lost three UDP probes, UDP status is no longer unconfirmed */
+		udp_probe_timeout_handler(&mesh->loop, n);
+	}
+
 	struct timespec elapsed;
+
 	timespec_sub(&mesh->loop.now, &n->last_udp_probe_sent, &elapsed);
 
-	int interval = n->status.udp_confirmed ? 10 : 2;
+	int interval = (n->status.udp_confirmed && n->udpprobes >= 0) ? 10 : 2;
 
 	if(elapsed.tv_sec >= interval) {
 		n->last_udp_probe_sent = mesh->loop.now;
 		send_udp_probe_packet(mesh, n, MIN_PROBE_SIZE);
 
+		if(n->status.udp_confirmed) {
+			n->udpprobes--;
+		}
+
 		if(!n->status.udp_confirmed && n->prevedge) {
 			n->status.broadcast = true;
 			send_udp_probe_packet(mesh, n, MIN_PROBE_SIZE);