From 6d7f3976f33b0d4fbbece60d6b7d549b13107674 Mon Sep 17 00:00:00 2001
From: Luca Deri <deri@ntop.org>
Date: Mon, 9 Dec 2019 18:02:48 +0100
Subject: Code cleanup

---
 src/lib/protocols/stun.c | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

(limited to 'src/lib/protocols/stun.c')

diff --git a/src/lib/protocols/stun.c b/src/lib/protocols/stun.c
index e95965f6b..d5114204f 100644
--- a/src/lib/protocols/stun.c
+++ b/src/lib/protocols/stun.c
@@ -350,9 +350,13 @@ static ndpi_int_stun_t ndpi_int_check_stun(struct ndpi_detection_module_struct *
 
             memset(flow->host_server_name, 0, sizeof(flow->host_server_name));
 
-            for(j=0; j<i; j++)
-              flow->host_server_name[j] = payload[k++];
-
+            for(j=0; j<i; j++) {
+	      if((k+i) < payload_length)
+		flow->host_server_name[j] = payload[k++];
+	      else
+		break;
+	    }
+	    
 #ifdef DEBUG_STUN
             printf("==> [%s]\n", flow->host_server_name);
 #endif
-- 
cgit v1.2.3