def domain_fetch(self):
'''****域名爆破开始****'''
if self.options.get("domain_brute"):
self.update_task_field("status", "domain_brute")
t1 = time.time()
self.domain_brute()
elapse = time.time() - t1
self.update_services("domain_brute", elapse)
else:
domain_info = self.build_single_domain_info(self.base_domain)
if domain_info:
self.domain_info_list.append(domain_info)
self.save_domain_info_list([domain_info])
# ***RiskIQ查询****
if self.options.get("riskiq_search") and services.riskiq_quota() > 0:
self.update_task_field("status", "riskiq_search")
t1 = time.time()
self.riskiq_search()
elapse = time.time() - t1
self.update_services("riskiq_search", elapse)
# crt.sh 网站查询
if self.options.get("crtsh_search"):
self.update_task_field("status", "crtsh_search")
t1 = time.time()
self.crtsh_search()
elapse = time.time() - t1
self.update_services("crtsh_search", elapse)
if self.options.get("arl_search"):
self.update_task_field("status", "arl_search")
t1 = time.time()
self.arl_search()
elapse = time.time() - t1
self.update_services("arl_search", elapse)
'''***智能域名生成****'''
if self.options.get("alt_dns"):
self.update_task_field("status", "alt_dns")
t1 = time.time()
self.alt_dns()
elapse = time.time() - t1
self.update_services("alt_dns", elapse)
def run(self):
self.update_task_field("start_time", utils.curr_date())
'''****域名爆破开始****'''
if self.options.get("domain_brute"):
self.update_task_field("status", "domain_brute")
t1 = time.time()
self.domain_brute()
elapse = time.time() - t1
self.update_services("domain_brute", elapse)
else:
domain_info = self.build_single_domain_info(self.base_domain)
if domain_info:
self.domain_info_list.append(domain_info)
self.save_domain_info_list([domain_info])
'''***RiskIQ查询****'''
if services.riskiq_quota() > 0 and self.options.get("riskiq_search"):
self.update_task_field("status", "riskiq_search")
t1 = time.time()
self.riskiq_search()
elapse = time.time() - t1
self.update_services("riskiq_search", elapse)
if self.options.get("arl_search"):
self.update_task_field("status", "arl_search")
t1 = time.time()
self.arl_search()
elapse = time.time() - t1
self.update_services("arl_search", elapse)
'''***智能域名生成****'''
if self.options.get("alt_dns"):
self.update_task_field("status", "alt_dns")
t1 = time.time()
self.alt_dns()
elapse = time.time() - t1
self.update_services("alt_dns", elapse)
self.gen_ipv4_map()
'''***佛法证书域名关联****'''
if self.options.get("fofa_search"):
self.update_task_field("status", "fofa_search")
t1 = time.time()
self.fofa_search()
elapse = time.time() - t1
self.update_services("fofa_search", elapse)
'''***端口扫描开始***'''
if self.options.get("port_scan"):
self.update_task_field("status", "port_scan")
t1 = time.time()
self.port_scan()
elapse = time.time() - t1
self.update_services("port_scan", elapse)
'''***证书获取***'''
if self.options.get("ssl_cert"):
self.update_task_field("status", "ssl_cert")
t1 = time.time()
self.ssl_cert()
elapse = time.time() - t1
self.update_services("ssl_cert", elapse)
# 服务信息存储
if self.options.get("service_detection"):
self.save_service_info()
self.save_ip_info()
self.update_task_field("status", "find_site")
t1 = time.time()
self.find_site()
elapse = time.time() - t1
self.update_services("find_site", elapse)
'''***站点识别***'''
if self.options.get("site_identify"):
self.update_task_field("status", "site_identify")
t1 = time.time()
self.site_identify()
elapse = time.time() - t1
self.update_services("site_identify", elapse)
self.update_task_field("status", "fetch_site")
t1 = time.time()
self.fetch_site()
elapse = time.time() - t1
self.update_services("fetch_site", elapse)
'''***站点截图***'''
if self.options.get("site_capture"):
self.update_task_field("status", "site_capture")
t1 = time.time()
self.site_screenshot()
elapse = time.time() - t1
self.update_services("site_capture", elapse)
'''搜索引擎查询'''
if self.options.get("search_engines"):
self.update_task_field("status", "search_engines")
t1 = time.time()
self.search_engines()
elapse = time.time() - t1
self.update_services("search_engines", elapse)
'''站点爬虫'''
if self.options.get("site_spider"):
self.update_task_field("status", "site_spider")
t1 = time.time()
self.site_spider()
elapse = time.time() - t1
self.update_services("site_spider", elapse)
'''文件泄露'''
if self.options.get("file_leak"):
self.update_task_field("status", "file_leak")
t1 = time.time()
self.file_leak()
elapse = time.time() - t1
self.update_services("file_leak", elapse)
self.update_task_field("status", TaskStatus.DONE)
self.update_task_field("end_time", utils.curr_date())