From e4605c81a347f83abd2d5c15f54495a1df6cf0e0 Mon Sep 17 00:00:00 2001
From: Christian Deacon <christianmdeacon@gmail.com>
Date: Fri, 21 Mar 2025 18:20:52 -0400
Subject: [PATCH] Add util function to parse port ranges.

---
 src/loader/utils/helpers.c | 64 ++++++++++++++++++++++++++++++++++++++
 src/loader/utils/helpers.h | 13 +++++++-
 2 files changed, 76 insertions(+), 1 deletion(-)

diff --git a/src/loader/utils/helpers.c b/src/loader/utils/helpers.c
index e039060..d587b03 100644
--- a/src/loader/utils/helpers.c
+++ b/src/loader/utils/helpers.c
@@ -121,4 +121,68 @@ u64 get_boot_nano_time()
     sysinfo(&sys);
 
     return sys.uptime * 1e9;
+}
+
+/**
+ * Parses a port range string and returns the minimum and maximum port.
+ * 
+ * @param range_str The port range string.
+ * 
+ * @return The port range as port_range_t type. Fields will be set to 0 on failure.
+ */
+port_range_t parse_port_range(const char* range_str)
+{
+    port_range_t ret = {0};
+
+    if (!range_str)
+    {
+        return ret;
+    }
+
+    // Copy range string.
+    char range_str_copy[24];
+    strncpy(range_str_copy, range_str, sizeof(range_str_copy) - 1);
+    range_str_copy[sizeof(range_str_copy) - 1] = '\0';
+
+    // First scan for port ranges with ":".
+    char* start = range_str_copy;
+    char* end = strchr(range_str_copy, '-');
+
+    if (!end)
+    {
+        end = strchr(range_str_copy, ':');
+    }
+
+    if (end)
+    {
+        *end = '\0';
+        end++;
+    }
+
+    char *end_ptr = NULL;
+
+    ret.min = strtol(start, &end_ptr, 10);
+
+    if (end_ptr == start || (*end_ptr != '\0' && !isspace((unsigned char)*end_ptr)))
+    {
+        return ret;
+    }
+
+    if (end)
+    {
+        ret.max = strtol(end, &end_ptr, 10);
+
+        if (end_ptr == end || (*end_ptr != '\0' && !isspace((unsigned char)*end_ptr)))
+        {
+            return ret;
+        }
+    }
+    else
+    {
+        ret.max = ret.min;
+    }
+
+    ret.success = 1;
+
+    return ret;
 }
\ No newline at end of file
diff --git a/src/loader/utils/helpers.h b/src/loader/utils/helpers.h
index b5e3ddc..31c9389 100644
--- a/src/loader/utils/helpers.h
+++ b/src/loader/utils/helpers.h
@@ -7,6 +7,9 @@
 #include <stdio.h>
 #include <stdlib.h>
 
+#include <string.h>
+#include <ctype.h>
+
 #include <sys/sysinfo.h>
 
 struct ip_range
@@ -15,6 +18,13 @@ struct ip_range
     u8 cidr;
 } typedef ip_range_t;
 
+struct port_range
+{
+    unsigned int success;
+    u16 min;
+    u16 max;
+} typedef port_range_t;
+
 extern int cont;
 
 void print_help_menu();
@@ -22,4 +32,5 @@ void hdl_signal(int code);
 ip_range_t parse_ip_range(const char* ip);
 const char* get_protocol_str_by_id(int id);
 void print_tool_info();
-u64 get_boot_nano_time();
\ No newline at end of file
+u64 get_boot_nano_time();
+port_range_t parse_port_range(const char* range_str);
\ No newline at end of file