Update exploithawk.c

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

exploithawk.c

@@ -53,6 +53,12 @@ const char *exploitDirs[] = {
     "/var/exploits"
 };
 
+const char *exploitDBRoots[] = {
+    "/usr/share/exploitdb",
+    "/var/lib/exploitdb",
+    "/opt/exploitdb"
+};
+
 char search_term[128];
 
 
@@ -112,7 +118,18 @@ void *load_csv_thread(void *arg) {
         if (col >= 3 && matches_search(cols[2], search_term)) {
             trim_newline(cols[1]);
             trim_newline(cols[2]);
-            add_result("CSV", cols[2], cols[1]);
+            if (col >= 3 && matches_search(cols[2], search_term)) {
+                char fullpath[MAX_PATH] = "";
+                // Find first existing ExploitDB root
+                for (int r = 0; r < sizeof(exploitDBRoots)/sizeof(exploitDBRoots[0]); r++) {
+                    snprintf(fullpath, sizeof(fullpath), "%s/%s", exploitDBRoots[r], cols[1]);
+                    if (access(fullpath, F_OK) == 0) {  // file exists
+                        add_result("ExploitDB", cols[2], fullpath);
+                        break;
+                    }
+                }
+            }
+
         }
     }
     fclose(f);