From af3710abebe63758ad1f01bec4a9229734ce642a Mon Sep 17 00:00:00 2001
From: SashaXser <24498484+SashaXser@users.noreply.github.com>
Date: Sun, 10 Sep 2023 14:50:10 +0400
Subject: [PATCH] Update goodbyedpi.c

---
 src/goodbyedpi.c | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/goodbyedpi.c b/src/goodbyedpi.c
index d6cbb5b..9d1b2d9 100644
--- a/src/goodbyedpi.c
+++ b/src/goodbyedpi.c
@@ -184,12 +184,17 @@ static void add_filter_str(int proto, int port) {
     size_t new_filter_size = strlen(filter_string) + (proto == IPPROTO_UDP ? strlen(udp) : strlen(tcp)) + 16;
     char *new_filter = malloc(new_filter_size);
 
-    sprintf(new_filter, proto == IPPROTO_UDP ? udp : tcp, port, port);
+    if (proto == IPPROTO_UDP) {
+        snprintf(new_filter, new_filter_size, udp, port, port);
+    } else {
+        snprintf(new_filter, new_filter_size, tcp, port, port);
+    }
 
     free(filter_string);
     filter_string = new_filter;
 }
 
+
 static void add_ip_id_str(int id) {
     const char *ipid = " or ip.Id == %d";
     char *addfilter = malloc(strlen(ipid) + 16);