Update exploithawk.c

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

exploithawk.c

@@ -21,6 +21,7 @@ typedef struct {
 Exploit results[MAX_RESULTS];
 int result_count = 0;
 pthread_mutex_t lock;
+int use_colors = 0;  // flag for color support
 
 const char *exploitDBPaths[] = {
     "/usr/share/exploitdb/files_exploits.csv",
@@ -155,6 +156,16 @@ void display_results() {
     cbreak();
     keypad(stdscr, TRUE);
 
+    if (has_colors()) {
+        start_color();
+        use_colors = 1;
+        // color pairs
+        init_pair(1, COLOR_YELLOW, COLOR_BLACK);  // source
+        init_pair(2, COLOR_CYAN, COLOR_BLACK);    // title
+        init_pair(3, COLOR_GREEN, COLOR_BLACK);   // path
+        init_pair(4, COLOR_MAGENTA, COLOR_BLACK); // header
+    }
+
     int ch, start = 0, rows, cols;
     getmaxyx(stdscr, rows, cols);
 
@@ -164,7 +175,11 @@ void display_results() {
 
     while (1) {
         clear();
+
+        if (use_colors) attron(COLOR_PAIR(4) | A_BOLD);
         mvprintw(0, 0, "| %-10s | %-*s | %-*s |", "Source", col_title, "Title", col_path, "Path");
+        if (use_colors) attroff(COLOR_PAIR(4) | A_BOLD);
+
         mvhline(1, 0, '-', cols);
 
         int printed_rows = 0;
@@ -176,14 +191,29 @@ void display_results() {
             int max_lines = title_count > path_count ? title_count : path_count;
 
             for (int l = 0; l < max_lines && printed_rows < rows - 3; l++) {
-                mvprintw(printed_rows + 2, 0, "| %-10.10s | %-*s | %-*s |",
-                         l == 0 ? results[i].source : "",  
-                         col_title, l < title_count ? title_lines[l] : "",
-                         col_path, l < path_count ? path_lines[l] : "");
+                int line_y = printed_rows + 2;
+
+                // Source column
+                if (use_colors) attron(COLOR_PAIR(1));
+                mvprintw(line_y, 0, "| %-10.10s ", l == 0 ? results[i].source : "");
+                if (use_colors) attroff(COLOR_PAIR(1));
+
+                // Title column
+                if (use_colors) attron(COLOR_PAIR(2));
+                mvprintw(line_y, 13, "| %-*s ", col_title, l < title_count ? title_lines[l] : "");
+                if (use_colors) attroff(COLOR_PAIR(2));
+
+                // Path column
+                if (use_colors) attron(COLOR_PAIR(3));
+                mvprintw(line_y, 15 + col_title, "| %-*s |", col_path, l < path_count ? path_lines[l] : "");
+                if (use_colors) attroff(COLOR_PAIR(3));
+
                 printed_rows++;
             }
         }
-        mvprintw(rows - 1, 0, "Use Up/Down arrows to scroll, q to quit. Showing %d of %d results", start+1, result_count);
+
+        mvprintw(rows - 1, 0, "Use Up/Down arrows to scroll, q to quit. Showing %d of %d results",
+                 start+1, result_count);
         refresh();
 
         ch = getch();