61 lines
2.3 KiB
Python
61 lines
2.3 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
# @Author : jeffzhang
|
|
# @Time : 18-5-15
|
|
# @File : start.py
|
|
# @Desc : ""
|
|
|
|
import os
|
|
import sys
|
|
parent_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
|
sys.path.insert(0, parent_dir)
|
|
from instance import config_name
|
|
from fuxi.views.lib.mongo_db import connectiondb, db_name_conf
|
|
|
|
tasks_db = db_name_conf()['tasks_db']
|
|
asset_db = db_name_conf()['asset_db']
|
|
server_db = db_name_conf()['server_db']
|
|
subdomain_db = db_name_conf()['subdomain_db']
|
|
vul_db = db_name_conf()['vul_db']
|
|
plugin_db = db_name_conf()['plugin_db']
|
|
config_db = db_name_conf()['config_db']
|
|
|
|
|
|
def config():
|
|
connectiondb(config_db).drop()
|
|
subdomain_dict = []
|
|
subdomain_dict_path = os.getcwd() + '/tests/domain.dict'
|
|
try:
|
|
with open(subdomain_dict_path) as file_read:
|
|
for i in file_read:
|
|
subdomain_dict.append(i.strip())
|
|
except Exception as e:
|
|
print(e)
|
|
subdomain_dict = ['www', 'mail', 'test']
|
|
config_data = {
|
|
'poc_thread': 50,
|
|
'discovery_thread': 52,
|
|
'subdomain_thread': 53,
|
|
'port_thread': 54,
|
|
'config_name': config_name,
|
|
'poc_frequency': 15,
|
|
'port_list': [20, 21, 22, 23, 80, 81, 443, 445, 544, 873, 1080, 1433, 1434, 1521, 2100, 3306, 3389, 4440, 5671,
|
|
5672, 5900, 5984, 6379, 7001, 8080, 8081, 8089, 8888, 9090, 9200, 11211, 15672, 27017, 50070],
|
|
'subdomain_dict_2': subdomain_dict,
|
|
'subdomain_dict_3': ['www', 'mail', 'test'],
|
|
'username_dict': ['admin', 'root', 'administrators'],
|
|
'password_dict': ['123456', 'password', '12345678', 'admin', 'admin123'],
|
|
'auth_tester_thread': 100,
|
|
'discovery_time': "11:00:00",
|
|
'auth_service': ['asterisk', 'cisco', 'cisco-enable', 'cvs', 'firebird', 'ftp', 'ftps', 'http-proxy',
|
|
'http-proxy-urlenum', 'icq', 'imap', 'irc', 'ldap2', 'mssql', 'mysql', 'nntp',
|
|
'oracle-listener', 'oracle-sid', 'pcanywhere', 'pcnfs', 'pop3', 'postgres', 'rdp', 'redis',
|
|
'rexec', 'rlogin', 'rsh', 's7-300', 'sip', 'smb', 'smtp', 'smtp-enum', 'snmp', 'socks5',
|
|
'ssh', 'sshkey', 'svn', 'teamspeak', 'telnet', 'vmauthd', 'vnc', 'xmpp'],
|
|
}
|
|
connectiondb(config_db).insert_one(config_data)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
config()
|