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())