diff --git a/ReportToolz/floss.php b/ReportToolz/floss.php index 439bdb7..e6c74cf 100755 --- a/ReportToolz/floss.php +++ b/ReportToolz/floss.php @@ -42,7 +42,7 @@ die("[-] no path set\n"); // load vdb vulns -$vdbVulns = getDirContents($vulnDB); +$vdbVulns = getDirContents($vulnDB, '/\.json$/'); foreach($vdbVulns as $h => $i){ // remove begining of vdb path (keeps clean) $vdbVulns[$h] = str_replace($vulnDB, "", $i); } @@ -126,13 +126,18 @@ |_______________________________________________|\n"; -function getDirContents($path) { - $rii = new RecursiveIteratorIterator(new RecursiveDirectoryIterator($path)); +function getDirContents($dir, $filter = '', &$results = array()) { + $files = scandir($dir); - $files = array(); - foreach ($rii as $file) - if (!$file->isDir()) - $files[] = $file->getPathname(); + foreach($files as $key => $value){ + $path = realpath($dir.DIRECTORY_SEPARATOR.$value); - return $files; -} + if(!is_dir($path)) { + if(empty($filter) || preg_match($filter, $path)) $results[] = $path; + } elseif($value != "." && $value != "..") { + getDirContents($path, $filter, $results); + } + } + + return $results; +}