Update exploithawk.c

Author: s-r-e-e-r-a-j

exploithawk.c

@@ -165,26 +165,20 @@ void *scan_dir_thread(void *arg) {
     return NULL;
 }
 
-int wrap_text(const char *text, char lines[][MAX_TITLE], int width) {
-    if (!text) {
-        lines[0][0] = '\0';
-        return 1;
-    }
-    int len = strlen(text);
-    if (len == 0) {
-        lines[0][0] = '\0';
-        return 1;
-    }
-    int count = 0;
-    int i = 0;
-    while (i < len && count < 10) {
-        int chunk = (len - i > width) ? width : len - i;
-        strncpy(lines[count], text + i, chunk);
-        lines[count][chunk] = '\0';
-        count++;
-        i += chunk;
+int wrap_text_generic(const char *text, void *lines, int rows, size_t row_size, int width) {
+    if (!text || rows <= 0 || row_size < 2) return 1;
+    if (width > (int)row_size - 1) width = (int)row_size - 1;
+    if (width <= 0) { ((char*)lines)[0] = '\0'; return 1; }
+
+    int len = strlen(text), i = 0, count = 0;
+    while (i < len && count < rows) {
+        int chunk = (len - i > width) ? width : (len - i);
+        char *dest = (char*)lines + (size_t)count * row_size;
+        memcpy(dest, text + i, chunk);
+        dest[chunk] = '\0';
+        count++; i += chunk;
     }
-    return (count == 0) ? 1 : count;
+    return count ? count : 1;
 }
 
 void display_results() {
@@ -251,8 +245,8 @@ void display_results() {
         for (int i = start; i < result_count && printed_rows < rows - 3; i++) {
             char title_lines[10][MAX_TITLE];
             char path_lines[10][MAX_PATH];
-            int title_count = wrap_text(results[i].title, title_lines, content_title);
-            int path_count  = wrap_text(results[i].path,  path_lines,  content_path);
+            int title_count = wrap_text_generic(results[i].title, title_lines, 10, sizeof(title_lines[0]), content_title);
+            int path_count  = wrap_text_generic(results[i].path,  path_lines,  10, sizeof(path_lines[0]),  content_path);
             int max_lines = (title_count > path_count) ? title_count : path_count;
 
             for (int l = 0; l < max_lines && printed_rows < rows - 3; l++) {