#!/usr/bin/python

import sys
import os
import subprocess

if len(sys.argv) != 3:
print "Usage: dirbust.py <target url> <scan name>"
sys.exit(0)

url = str(sys.argv[1])
name = str(sys.argv[2])
folders = ["/usr/share/dirb/wordlists", "/usr/share/dirb/wordlists/vulns"]

found = []
print "INFO: Starting dirb scan for " + url
for folder in folders:
for filename in os.listdir(folder):

outfile = " -o " + "results/exam/" + name + "_dirb_" + filename
DIRBSCAN = "dirb %s %s/%s %s -S -r" % (url, folder, filename, outfile)
try:
results = subprocess.check_output(DIRBSCAN, shell=True)
resultarr = results.split("\n")
for line in resultarr:
if "+" in line:
if line not in found:
found.append(line)
except:
pass

try:
if found[0] != "":
print "[*] Dirb found the following items..."
for item in found:
print " " + item
except:
print "INFO: No items found during dirb scan of " + url