def info_collect():
for domain in SrcList().select_un_scan_src_list(sudomain_scan_size):
src_id = domain['src_id']
domain_id = domain['id']
url = domain['url']
domain = domain['url'][2:]
domains = AllDomain(domain).run()
print(len(domains.keys()))
info = SiteInfo([x for x in domains.keys()]).run()
Database().insert_subdomain(domains, info, domain_id, src_id)
SrcList().update_scan_time(url)
def setting(page=1):
src_list = SrcList().select_src_list(page)
return render_template('setting.html',
mode="setting",
page=page,
src_list=src_list,
max_page=max_src // item_size + 1)
def delete(id, mode):
if mode == 'src':
SrcList().delete(id)
if mode == 'fingerprint':
Rules().delete(id)
else:
Database().delete(id, mode)
return redirect(request.referrer)
def __init__(self):
self.src_scan_list = SrcList().select_un_scan_src_list()
self.src_scan_list = [{
'id': 155,
'name': 'bilibili bilisrc',
'src_id': 34,
'url': "*.biligame.com",
'scan_time': "2018-12-17"
}]
def index(page=1):
src_list = SrcList().select_recent_src_list(page)
return render_template('index.html',
mode="index",
page=page,
max_page=max_src // item_size + 1,
src_list=src_list,
max_domain=max_domain,
max_port=max_port,
max_sendir=max_sendir,
max_fingerprint=max_fingerprint,
max_vul=max_vul)
def detail(domain_id):
# 域名ID
src_name = SrcList().select_src_by_id(domain_id)
domain_number = d.count('subdomain', domain_id)
port_number = d.count('port', domain_id)
sendir_number = d.count('sendir', domain_id)
vul_number = d.count('vul', domain_id)
return render_template('detail.html',
domain_id=domain_id,
src_name=src_name[2:],
domain_number=domain_number,
port_number=port_number,
sendir_number=sendir_number,
vul_number=vul_number)
def delete(id, mode):
if mode == 'src':
SrcList().delete(id)
if mode == 'fingerprint':
r.delete(id)
return redirect(request.referrer)
from web.gitleak import gitleak
app = Flask(__name__)
app.register_blueprint(gitleak)
# TODO 每次刷新
d = Database()
max_domain = d.count('subdomain')
max_port = d.count('port')
max_sendir = d.count('sendir')
max_vul = d.count('vul')
r = Rules()
max_fingerprint = r.count()
max_src = SrcList().count()
@app.route('/')
@app.route('/index')
@app.route('/index/<int:page>')
def index(page=1):
src_list = SrcList().select_recent_src_list(page)
return render_template('index.html',
mode="index",
page=page,
max_page=max_src // item_size + 1,
src_list=src_list,
max_domain=max_domain,
max_port=max_port,
max_sendir=max_sendir,