Update exploithawk.c

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

exploithawk.c

@@ -17,6 +17,8 @@
 #define MAX_TITLE 512
 #define MAX_SOURCE 32
 #define TITLE_WRAP 50  // preferred width for title column
+#define MAX_THREADS 64
+#define MAX_THREADS (int)(sizeof(threads)/sizeof(threads[0]))
 
 typedef struct {
     char source[MAX_SOURCE];
@@ -344,17 +346,35 @@ int main(int argc, char *argv[]) {
     int num_threads = 0;
     pthread_t threads[64];
 
-    // CSV threads
-    for (int i = 0; i < (int)(sizeof(exploitDBPaths)/sizeof(exploitDBPaths[0])); i++)
-        pthread_create(&threads[num_threads++], NULL, load_csv_thread, (void*)exploitDBPaths[i]);
-
-    // Directory threads
-    for (int i = 0; i < (int)(sizeof(exploitDirs)/sizeof(exploitDirs[0])); i++)
-        pthread_create(&threads[num_threads++], NULL, scan_dir_thread, (void*)exploitDirs[i]);
-
-    // Join all threads
-    for (int i = 0; i < num_threads; i++)
-        pthread_join(threads[i], NULL);
+    // CSV threads 
+    for (int i = 0; i < (int)(sizeof(exploitDBPaths)/sizeof(exploitDBPaths[0])); i++) {
+        if (num_threads >= MAX_THREADS) break;
+           int rc = pthread_create(&threads[num_threads], NULL, load_csv_thread, (void*)exploitDBPaths[i]);
+        if (rc != 0) {
+            fprintf(stderr, "pthread_create failed for CSV path %s: %d\n", exploitDBPaths[i], rc);
+        } else {
+          num_threads++;
+          }
+     }
+
+     // Directory threads 
+     for (int i = 0; i < (int)(sizeof(exploitDirs)/sizeof(exploitDirs[0])); i++) {
+         if (num_threads >= MAX_THREADS) break;
+            int rc = pthread_create(&threads[num_threads], NULL, scan_dir_thread, (void*)exploitDirs[i]);
+         if (rc != 0) {
+            fprintf(stderr, "pthread_create failed for dir %s: %d\n", exploitDirs[i], rc);
+         } else {
+           num_threads++;
+           }
+      }
+
+       // Join exactly the created threads 
+       for (int i = 0; i < num_threads; i++) {
+           int rc = pthread_join(threads[i], NULL);
+           if (rc != 0) {
+              fprintf(stderr, "pthread_join failed on thread %d: %d\n", i, rc);
+           }
+        }
 
     display_results();
     pthread_mutex_destroy(&lock);