def main(): root = "https://www.shiyanlou.com/" domain = common.w8urlparse(root) threadNum = 10 output = outputer.outputer() # CDN Check print "CDN check...." iscdn = True try: msg, iscdn = fun_until.checkCDN(root) output.add("cdn", msg) output.build_html(domain) print msg except: print "[Error]:CDN check error" if iscdn: #IP Ports Scan ip = common.gethostbyname(root) print "IP:", ip print "START Port Scan:" pp = PortScan.PortScan(ip) pp.work() output.build_html(domain) # DIR Fuzz dd = webdir.webdir(root, threadNum) dd.work() dd.output() output.build_html(domain) #webcms ww = webcms.webcms(root, threadNum) ww.run() output.build_html(domain) #spider w8 = SpiderMain(root, threadNum) w8.craw()
#!/usr/bin/env python # __author__= 'w8ay' import os import sys import Queue import requests import threading from lib.core import outputer output = outputer.outputer() class webdir: def __init__(self, root, threadNum): self.root = root self.threadNum = threadNum self.headers = { 'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_7_3) AppleWebKit/535.20 (KHTML, like Gecko) Chrome/19.0.1036.7 Safari/535.20', 'Referer': 'http://www.shiyanlou.com', 'Cookie': 'whoami=w8ay', } self.task = Queue.Queue() self.s_list = [] filename = os.path.join(sys.path[0], "data", "dir.txt") for line in open(filename): self.task.put(root + line.strip()) def checkdir(self, url): status_code = 0 try:
def main(): checkEnvironment() #检测环境 common.setPaths(modulePath()) # 为一些目录和文件设置了绝对路径 #目标url,一些参数设置 parser = argparse.ArgumentParser(description="Ajatar scan must a url") parser.add_argument("-u", "--url", help="url") args = parser.parse_args() root = args.url #root = "http://www.imufe.edu.cn/" #判断是否为Windows if IS_WIN: #Windows界面颜色 windowsColorInit() #Banner() try: configFileParser(os.path.join(paths.Ajatar_ROOT_PATH, "config.conf")) #线程数 threadNum = initOption(args) except Exception as e: raise e #拆解url 得到netloc domain = common.Ajurlparse(root) #输出报告对象 output = outputer.outputer() # CDN Check print "CDN check...." iscdn = True try: msg, iscdn = fun_until.checkCDN(root) #获取数据生成报告 output.add("cdn", msg) #build html output.build_html(domain) print msg except: print "[Error]:CDN check error" if iscdn: #IP Ports Scan #获取ip ip = common.gethostbyname(root) print "IP:", ip print "Start Port Scan:" #ip端口扫描 pp = PortScan.PortScan(ip) pp.work() output.build_html(domain) # DIR Fuzz dd = webdir.webdir(root, threadNum) dd.work() dd.output() output.build_html(domain) #webcms ww = webcms.webcms(root, threadNum) ww.run() output.build_html(domain) #spider Aj = SpiderMain(root, threadNum) Aj.craw()