def setupBatchReports(self):
self.batch = True
self.batchSession = "BATCH-{0}".format(
time.strftime("%y-%m-%d_%H-%M-%S"))
self.batchDirectoryPath = FileUtils.build_path(self.savePath,
"reports",
self.batchSession)
if not FileUtils.exists(self.batchDirectoryPath):
FileUtils.create_directory(self.batchDirectoryPath)
if not FileUtils.exists(self.batchDirectoryPath):
self.output.error("Couldn't create batch folder at {}".format(
self.batchDirectoryPath))
sys.exit(1)
if FileUtils.can_write(self.batchDirectoryPath):
FileUtils.create_directory(self.batchDirectoryPath)
targetsFile = FileUtils.build_path(self.batchDirectoryPath,
"TARGETS.txt")
FileUtils.write_lines(targetsFile, self.urlList)
else:
self.output.error("Couldn't create batch folder at {}".format(
self.batchDirectoryPath))
sys.exit(1)
def validate_path(self, path):
if not FileUtils.exists(path):
self.output.error("{} does not exist".format(path))
exit(1)
if FileUtils.exists(path) and not FileUtils.is_dir(path):
self.output.error(
"{} is a file, should be a directory".format(path))
exit(1)
if not FileUtils.can_write(path):
self.output.error("Directory {} is not writable".format(path))
exit(1)
return True
def __init__(self, script_path, arguments, output):
global VERSION
program_banner = (open(
FileUtils.build_path(script_path, "lib", "controller",
"banner.txt")).read().format(**VERSION))
self.directories = Queue()
self.script_path = script_path
self.exit = False
self.arguments = arguments
self.output = output
self.savePath = self.script_path
self.doneDirs = []
if arguments.raw_file:
# Overwrite python-requests default headers
default_headers = {
"User-Agent": None,
"Accept-Encoding": None,
"Accept": None,
}
_raw = Raw(arguments.raw_file, arguments.scheme)
self.urlList = [_raw.url()]
self.httpmethod = _raw.method()
self.data = _raw.data()
self.headers = {**default_headers, **_raw.headers()}
self.cookie = _raw.cookie()
self.useragent = _raw.user_agent()
else:
default_headers = {
"User-Agent":
"Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/87.0.4280.88 Safari/537.36",
"Accept-Language": "*",
"Accept-Encoding": "*",
"Keep-Alive": "300",
"Cache-Control": "max-age=0",
}
self.urlList = list(filter(None, dict.fromkeys(arguments.urlList)))
self.httpmethod = arguments.httpmethod.lower()
self.data = arguments.data
self.headers = {**default_headers, **arguments.headers}
self.cookie = arguments.cookie
self.useragent = arguments.useragent
self.recursion_depth = arguments.recursion_depth
if arguments.saveHome:
savePath = self.getSavePath()
if not FileUtils.exists(savePath):
FileUtils.create_directory(savePath)
if FileUtils.exists(savePath) and not FileUtils.is_dir(savePath):
self.output.error(
"Cannot use {} because it's a file. Should be a directory".
format(savePath))
exit(1)
if not FileUtils.can_write(savePath):
self.output.error(
"Directory {} is not writable".format(savePath))
exit(1)
logs = FileUtils.build_path(savePath, "logs")
if not FileUtils.exists(logs):
FileUtils.create_directory(logs)
reports = FileUtils.build_path(savePath, "reports")
if not FileUtils.exists(reports):
FileUtils.create_directory(reports)
self.savePath = savePath
self.reportsPath = FileUtils.build_path(self.savePath, "logs")
self.blacklists = self.getBlacklists()
self.includeStatusCodes = arguments.includeStatusCodes
self.excludeStatusCodes = arguments.excludeStatusCodes
self.excludeSizes = arguments.excludeSizes
self.excludeTexts = arguments.excludeTexts
self.excludeRegexps = arguments.excludeRegexps
self.excludeRedirects = arguments.excludeRedirects
self.recursive = arguments.recursive
self.minimumResponseSize = arguments.minimumResponseSize
self.maximumResponseSize = arguments.maximumResponseSize
self.scanSubdirs = arguments.scanSubdirs
self.excludeSubdirs = (arguments.excludeSubdirs
if arguments.excludeSubdirs else [])
self.dictionary = Dictionary(
arguments.wordlist, arguments.extensions, arguments.suffixes,
arguments.prefixes, arguments.lowercase, arguments.uppercase,
arguments.capitalization, arguments.forceExtensions,
arguments.excludeExtensions, arguments.noExtension,
arguments.onlySelected)
self.allJobs = len(self.scanSubdirs) if self.scanSubdirs else 1
self.currentJob = 0
self.errorLog = None
self.errorLogPath = None
self.threadsLock = Lock()
self.batch = False
self.batchSession = None
self.skip429 = False
self.output.header(program_banner)
self.printConfig()
self.setupErrorLogs()
self.output.errorLogFile(self.errorLogPath)
if arguments.autoSave and len(self.urlList) > 1:
self.setupBatchReports()
self.output.newLine("\nAutoSave path: {0}".format(
self.batchDirectoryPath))
if arguments.useRandomAgents:
self.randomAgents = FileUtils.get_lines(
FileUtils.build_path(script_path, "db", "user-agents.txt"))
try:
for url in self.urlList:
try:
gc.collect()
self.reportManager = ReportManager()
self.currentUrl = url if url.endswith("/") else url + "/"
self.output.setTarget(self.currentUrl,
self.arguments.scheme)
try:
self.requester = Requester(
url,
cookie=self.cookie,
useragent=self.useragent,
maxPool=arguments.threadsCount,
maxRetries=arguments.maxRetries,
timeout=arguments.timeout,
ip=arguments.ip,
proxy=arguments.proxy,
proxylist=arguments.proxylist,
redirect=arguments.redirect,
requestByHostname=arguments.requestByHostname,
httpmethod=self.httpmethod,
data=self.data,
scheme=arguments.scheme,
)
for key, value in self.headers.items():
self.requester.setHeader(key, value)
self.requester.request("")
except RequestException as e:
self.output.error(e.args[0]["message"])
raise SkipTargetInterrupt
if arguments.useRandomAgents:
self.requester.setRandomAgents(self.randomAgents)
# Initialize directories Queue with start Path
self.basePath = self.requester.basePath
if self.scanSubdirs:
for subdir in self.scanSubdirs:
self.directories.put(subdir)
else:
self.directories.put("")
self.setupReports(self.requester)
matchCallbacks = [self.matchCallback]
notFoundCallbacks = [self.notFoundCallback]
errorCallbacks = [self.errorCallback, self.appendErrorLog]
self.fuzzer = Fuzzer(
self.requester,
self.dictionary,
testFailPath=arguments.testFailPath,
threads=arguments.threadsCount,
delay=arguments.delay,
matchCallbacks=matchCallbacks,
notFoundCallbacks=notFoundCallbacks,
errorCallbacks=errorCallbacks,
)
try:
self.prepare()
except RequestException as e:
self.output.error("Fatal error during scanning: " +
e.args[0]["message"])
raise SkipTargetInterrupt
except SkipTargetInterrupt:
continue
except KeyboardInterrupt:
self.output.error("\nCanceled by the user")
exit(0)
finally:
if not self.errorLog.closed:
self.errorLog.close()
self.reportManager.close()
self.output.warning("\nTask Completed")
def setupReports(self, requester):
if self.arguments.autoSave:
basePath = requester.basePath
basePath = basePath.replace(os.path.sep, ".")[:-1]
fileName = None
directoryPath = None
if self.batch:
fileName = requester.host
directoryPath = self.batchDirectoryPath
else:
fileName = ('{}_'.format(basePath))
fileName += time.strftime('%y-%m-%d_%H-%M-%S')
fileName += ".{0}".format(self.arguments.autoSaveFormat)
directoryPath = FileUtils.build_path(self.savePath, 'reports',
requester.host)
outputFile = FileUtils.build_path(directoryPath, fileName)
self.output.outputFile(outputFile)
if FileUtils.exists(outputFile):
i = 2
while FileUtils.exists(outputFile + "_" + str(i)):
i += 1
outputFile += "_" + str(i)
if not FileUtils.exists(directoryPath):
FileUtils.create_directory(directoryPath)
if not FileUtils.exists(directoryPath):
self.output.error(
"Couldn't create the reports folder at {}".format(
directoryPath))
sys.exit(1)
if FileUtils.can_write(directoryPath):
report = None
if self.arguments.autoSaveFormat == "simple":
report = SimpleReport(
requester.host,
requester.port,
requester.protocol,
requester.basePath,
outputFile,
self.batch,
)
elif self.arguments.autoSaveFormat == "json":
report = JSONReport(
requester.host,
requester.port,
requester.protocol,
requester.basePath,
outputFile,
self.batch,
)
elif self.arguments.autoSaveFormat == "xml":
report = XMLReport(
requester.host,
requester.port,
requester.protocol,
requester.basePath,
outputFile,
self.batch,
)
elif self.arguments.autoSaveFormat == "md":
report = MarkdownReport(
requester.host,
requester.port,
requester.protocol,
requester.basePath,
outputFile,
self.batch,
)
elif self.arguments.autoSaveFormat == "csv":
report = CSVReport(
requester.host,
requester.port,
requester.protocol,
requester.basePath,
outputFile,
self.batch,
)
else:
report = PlainTextReport(requester.host, requester.port,
requester.protocol,
requester.basePath, outputFile,
self.batch)
self.reportManager.addOutput(report)
else:
self.output.error(
"Can't write reports to {}".format(directoryPath))
sys.exit(1)
# TODO: format, refactor code
if self.arguments.simpleOutputFile:
self.reportManager.addOutput(
SimpleReport(requester.host, requester.port,
requester.protocol, requester.basePath,
self.arguments.simpleOutputFile, self.batch))
if self.arguments.plainTextOutputFile:
self.reportManager.addOutput(
PlainTextReport(requester.host, requester.port,
requester.protocol, requester.basePath,
self.arguments.plainTextOutputFile,
self.batch))
if self.arguments.jsonOutputFile:
self.reportManager.addOutput(
JSONReport(requester.host, requester.port, requester.protocol,
requester.basePath, self.arguments.jsonOutputFile,
self.batch))
if self.arguments.xmlOutputFile:
self.reportManager.addOutput(
XMLReport(requester.host, requester.port, requester.protocol,
requester.basePath, self.arguments.xmlOutputFile,
self.batch))
if self.arguments.markdownOutputFile:
self.reportManager.addOutput(
MarkdownReport(requester.host, requester.port,
requester.protocol, requester.basePath,
self.arguments.markdownOutputFile, self.batch))
if self.arguments.csvOutputFile:
self.reportManager.addOutput(
CSVReport(requester.host, requester.port, requester.protocol,
requester.basePath, self.arguments.csvOutputFile,
self.batch))
def __init__(self, script_path, arguments, output):
global VERSION
program_banner = (open(
FileUtils.build_path(script_path, "lib", "controller",
"banner.txt")).read().format(**VERSION))
self.directories = Queue()
self.script_path = script_path
self.exit = False
self.arguments = arguments
self.output = output
self.savePath = self.script_path
self.doneDirs = []
self.urlList = list(filter(None,
dict.fromkeys(self.arguments.urlList)))
self.recursive_level_max = self.arguments.recursive_level_max
if self.arguments.httpmethod.lower() not in [
"get", "head", "post", "put", "patch", "options", "delete",
"trace", "debug", "connect"
]:
self.output.error("Invalid HTTP method")
exit(1)
self.httpmethod = self.arguments.httpmethod.lower()
if self.arguments.saveHome:
savePath = self.getSavePath()
if not FileUtils.exists(savePath):
FileUtils.create_directory(savePath)
if FileUtils.exists(savePath) and not FileUtils.is_dir(savePath):
self.output.error(
"Cannot use {} because is a file. Should be a directory".
format(savePath))
exit(1)
if not FileUtils.can_write(savePath):
self.output.error(
"Directory {} is not writable".format(savePath))
exit(1)
logs = FileUtils.build_path(savePath, "logs")
if not FileUtils.exists(logs):
FileUtils.create_directory(logs)
reports = FileUtils.build_path(savePath, "reports")
if not FileUtils.exists(reports):
FileUtils.create_directory(reports)
self.savePath = savePath
self.reportsPath = FileUtils.build_path(self.savePath, "logs")
self.blacklists = self.getBlacklists()
self.includeStatusCodes = self.arguments.includeStatusCodes
self.excludeStatusCodes = self.arguments.excludeStatusCodes
self.excludeSizes = self.arguments.excludeSizes
self.excludeTexts = self.arguments.excludeTexts
self.excludeRegexps = self.arguments.excludeRegexps
self.recursive = self.arguments.recursive
self.minimumResponseSize = self.arguments.minimumResponseSize
self.maximumResponseSize = self.arguments.maximumResponseSize
self.scanSubdirs = (list(dict.fromkeys(arguments.scanSubdirs))
if arguments.scanSubdirs else [])
self.excludeSubdirs = (arguments.excludeSubdirs
if arguments.excludeSubdirs else [])
self.dictionary = Dictionary(
self.arguments.wordlist, self.arguments.extensions,
self.arguments.suffixes, self.arguments.prefixes,
self.arguments.lowercase, self.arguments.uppercase,
self.arguments.capitalization, self.arguments.forceExtensions,
self.arguments.noDotExtensions, self.arguments.excludeExtensions,
self.arguments.noExtension, self.arguments.onlySelected)
self.allJobs = len(
self.urlList) * (len(self.scanSubdirs) if self.scanSubdirs else 1)
self.currentJob = 0
self.errorLog = None
self.errorLogPath = None
self.threadsLock = Lock()
self.batch = False
self.batchSession = None
self.output.header(program_banner)
self.printConfig()
self.setupErrorLogs()
self.output.errorLogFile(self.errorLogPath)
if self.arguments.autoSave and len(self.urlList) > 1:
self.setupBatchReports()
self.output.newLine("\nAutoSave path: {0}".format(
self.batchDirectoryPath))
if self.arguments.useRandomAgents:
self.randomAgents = FileUtils.get_lines(
FileUtils.build_path(script_path, "db", "user-agents.txt"))
try:
for url in self.urlList:
try:
gc.collect()
self.reportManager = ReportManager()
self.currentUrl = url
self.output.setTarget(self.currentUrl)
try:
self.requester = Requester(
url,
cookie=self.arguments.cookie,
useragent=self.arguments.useragent,
maxPool=self.arguments.threadsCount,
maxRetries=self.arguments.maxRetries,
delay=self.arguments.delay,
timeout=self.arguments.timeout,
ip=self.arguments.ip,
proxy=self.arguments.proxy,
proxylist=self.arguments.proxylist,
redirect=self.arguments.redirect,
requestByHostname=self.arguments.requestByHostname,
httpmethod=self.httpmethod,
data=self.arguments.data,
)
self.requester.request("")
except RequestException as e:
self.output.error(e.args[0]["message"])
raise SkipTargetInterrupt
if self.arguments.useRandomAgents:
self.requester.setRandomAgents(self.randomAgents)
for key, value in arguments.headers.items():
self.requester.setHeader(key, value)
# Initialize directories Queue with start Path
self.basePath = self.requester.basePath
if self.scanSubdirs:
for subdir in self.scanSubdirs:
self.directories.put(subdir)
else:
self.directories.put("")
self.setupReports(self.requester)
matchCallbacks = [self.matchCallback]
notFoundCallbacks = [self.notFoundCallback]
errorCallbacks = [self.errorCallback, self.appendErrorLog]
self.fuzzer = Fuzzer(
self.requester,
self.dictionary,
testFailPath=self.arguments.testFailPath,
threads=self.arguments.threadsCount,
matchCallbacks=matchCallbacks,
notFoundCallbacks=notFoundCallbacks,
errorCallbacks=errorCallbacks,
)
try:
self.wait()
except RequestException as e:
self.output.error(
"Fatal error during site scanning: " +
e.args[0]["message"])
raise SkipTargetInterrupt
except SkipTargetInterrupt:
continue
except KeyboardInterrupt:
self.output.error("\nCanceled by the user")
exit(0)
finally:
if not self.errorLog.closed:
self.errorLog.close()
self.reportManager.close()
self.output.warning("\nTask Completed")
