mirror of
https://github.com/shmilylty/OneForAll.git
synced 2026-08-26 04:47:48 +08:00
shmilylty
207 lines
9.0 KiB
Python
207 lines
9.0 KiB
Python
# coding=utf-8
|
||
"""
|
||
OneForAll配置
|
||
"""
|
||
import os
|
||
import sys
|
||
import pathlib
|
||
from loguru import logger
|
||
|
||
# 路径设置
|
||
oneforall_relpath = pathlib.Path(__file__).parent # oneforall代码相对路径
|
||
oneforall_abspath = oneforall_relpath.resolve() # oneforall代码绝对路径
|
||
oneforall_module_path = oneforall_relpath.joinpath('modules') # oneforall模块目录
|
||
data_storage_path = oneforall_relpath.joinpath('data') # 数据存放目录
|
||
result_save_path = oneforall_relpath.joinpath('results') # 结果保存目录
|
||
|
||
# 模块设置
|
||
save_module_result = True # 保存模块中各脚本结果(默认True)
|
||
enable_all_module = True # 启用所有模块(默认True)
|
||
enable_partial_module = [] # 启用部分模块 必须禁用enable_all_module才能生效
|
||
# 只使用ask和baidu搜索引擎收集子域
|
||
# enable_partial_module = [('modules.search', 'ask')
|
||
# ('modules.search', 'baidu')]
|
||
|
||
|
||
# 爆破模块设置
|
||
enable_brute_module = False # 使用爆破模块(默认禁用)
|
||
enable_verify_subdomain = True # 验证子域有效性(默认True)
|
||
enable_wildcard_check = True # 开启泛解析检测 会去掉泛解析的子域
|
||
# 爆破时使用的进程数(根据系统中CPU数量情况设置 不宜大于CPU数量 默认为系统中的CPU数量)
|
||
brute_process_num = os.cpu_count()
|
||
brute_coroutine_num = 128 # 爆破时每个进程下的协程数(不宜大于1000)
|
||
# 爆破所使用的字典路径 默认data/subdomains.txt
|
||
brute_wordlist_path = data_storage_path.joinpath('subdomains.txt')
|
||
brute_task_segment = 500
|
||
# 参数segment的设置受CPU性能，网络带宽，运营商限制等限制，默认500个子域为一任务组，
|
||
# 当你觉得你的环境不受以上因素影响，当前爆破速度较慢，那么强烈建议根据字典大小调整大小：
|
||
# 十万字典建议设置为5000，百万字典设置为50000
|
||
enable_recursive_brute = False # 是否使用递归爆破(默认禁用)
|
||
brute_recursive_depth = 2 # 递归爆破深度(默认2层)
|
||
# 爆破下一层子域所使用的字典路径 默认data/next_subdomains.txt
|
||
recursive_namelist_path = data_storage_path.joinpath('next_subdomains.txt')
|
||
enable_fuzz = False # 是否使用fuzz模式枚举域名
|
||
fuzz_rule = '' # fuzz域名的正则 示例：[a-z][0-9] 第一位是字母 第二位是数字
|
||
ips_appear_maximum = 10 # 同一IP集合出现次数超过10认为是泛解析
|
||
|
||
# 代理设置
|
||
enable_proxy = False # 是否使用代理(全局开关)
|
||
proxy_all_module = False # 代理所有模块
|
||
proxy_partial_module = ['GoogleQuery', 'AskSearch', 'DuckDuckGoSearch',
|
||
'GoogleAPISearch', 'GoogleSearch', 'YahooSearch',
|
||
'YandexSearch', 'CrossDomainXml',
|
||
'ContentSecurityPolicy'] # 代理自定义的模块
|
||
proxy_pool = [{'http': 'http://127.0.0.1:1080',
|
||
'https': 'https://127.0.0.1:1080'}] # 代理池
|
||
# proxy_pool = [{'http': 'socks5://127.0.0.1:10808',
|
||
# 'https': 'socks5://127.0.0.1:10808'}] # 代理池
|
||
|
||
|
||
# 网络请求设置
|
||
enable_fake_header = True # 启用伪造请求头
|
||
request_delay = 1 # 请求时延
|
||
request_timeout = 60 # 请求超时
|
||
request_verify = True # 请求SSL验证
|
||
|
||
# 搜索模块设置
|
||
enable_recursive_search = False # 递归搜索子域
|
||
search_recursive_times = 2 # 递归搜索层数
|
||
|
||
# DNS解析设置
|
||
resolver_nameservers = [
|
||
'119.29.29.29', '182.254.116.116', # DNSPod
|
||
'180.76.76.76', # Baidu DNS
|
||
'223.5.5.5', '223.6.6.6', # AliDNS
|
||
'114.114.114.114', '114.114.115.115' # 114DNS
|
||
# '8.8.8.8', '8.8.4.4', # Google DNS
|
||
# '1.0.0.1', '1.1.1.1' # CloudFlare DNS
|
||
# '208.67.222.222', '208.67.220.220' # OpenDNS
|
||
] # 指定查询的DNS域名服务器
|
||
resolver_timeout = 5.0 # 解析超时时间
|
||
resolver_lifetime = 30.0 # 解析存活时间
|
||
limit_resolve_conn = 50 # 限制同一时间解析的数量(默认50)
|
||
|
||
# http探测设置
|
||
small_ports = {80, 443}
|
||
medium_ports = {80, 443, 8000, 8080, 8443} # 默认使用
|
||
large_ports = {80, 81, 443, 591, 2082, 2087, 2095, 2096, 3000, 8000, 8001,
|
||
8008, 8080, 8083, 8443, 8834, 8888}
|
||
xlarge_ports = {80, 81, 300, 443, 591, 593, 832, 981, 1010, 1311, 2082,
|
||
2087, 2095, 2096, 2480, 3000, 3128, 3333, 4243, 4567, 4711,
|
||
4712, 4993, 5000, 5104, 5108, 5800, 6543, 7000, 7396, 7474,
|
||
8000, 8001, 8008, 8014, 8042, 8069, 8080, 8081, 8088, 8090,
|
||
8091, 8118, 8123, 8172, 8222, 8243, 8280, 8281, 8333, 8443,
|
||
8500, 8834, 8880, 8888, 8983, 9000, 9043, 9060, 9080, 9090,
|
||
9091, 9200, 9443, 9800, 9981, 12443, 16080, 18091, 18092,
|
||
20720, 28017}
|
||
ports = {'small': small_ports, 'medium': medium_ports,
|
||
'large': large_ports, 'xlarge': xlarge_ports}
|
||
verify_ssl = False
|
||
# aiohttp 支持 HTTP/HTTPS形式的代理
|
||
get_proxy = None # proxy="http://user:pass@some.proxy.com"
|
||
get_timeout = 120 # http请求探测总超时时间 None或者0则表示不检测超时
|
||
get_redirects = True # 允许请求跳转
|
||
fake_header = True # 使用伪造请求头
|
||
# 限制同一时间打开的连接数(默认None，根据系统不同设置，Windows系统400 其他系统800)
|
||
limit_open_conn = None
|
||
# 限制同一时间在同一个端点((host, port, is_ssl) 3者都一样的情况)打开的连接数
|
||
limit_per_host = 0 # 默认0表示不限制
|
||
|
||
|
||
# 模块API配置
|
||
# Censys可以免费注册获取API：https://censys.io/api
|
||
censys_api_id = ''
|
||
censys_api_secret = ''
|
||
|
||
# Binaryedge可以免费注册获取API：https://app.binaryedge.io/account/api
|
||
# 免费的API有效期只有1个月，到期之后可以再次生成，每月可以查询250次。
|
||
binaryedge_api = ''
|
||
|
||
# Chinaz可以免费注册获取API：http://api.chinaz.com/ApiDetails/Alexa
|
||
chinaz_api = ''
|
||
|
||
# Bing可以免费注册获取API：https://azure.microsoft.com/zh-cn/services/
|
||
# cognitive-services/bing-web-search-api/#web-json
|
||
bing_api_id = ''
|
||
bing_api_key = ''
|
||
|
||
# SecurityTrails可以免费注册获取API：https://securitytrails.com/corp/api
|
||
securitytrails_api = ''
|
||
|
||
# https://fofa.so/api
|
||
fofa_api_email = '' # fofa用户邮箱
|
||
fofa_api_key = '' # fofa用户key
|
||
|
||
# Google可以免费注册获取API:
|
||
# https://developers.google.com/custom-search/v1/overview
|
||
# 免费的API只能查询前100条结果
|
||
google_api_key = '' # Google API搜索key
|
||
google_api_cx = '' # Google API搜索cx
|
||
|
||
# https://api.passivetotal.org/api/docs/
|
||
riskiq_api_username = ''
|
||
riskiq_api_key = ''
|
||
|
||
# Shodan可以免费注册获取API: https://account.shodan.io/register
|
||
# 免费的API限速1秒查询1次
|
||
shodan_api_key = ''
|
||
# ThreatBook API 查询子域名需要收费 https://x.threatbook.cn/nodev4/vb4/myAPI
|
||
threatbook_api_key = ''
|
||
|
||
# VirusTotal可以免费注册获取API: https://developers.virustotal.com/reference
|
||
virustotal_api_key = ''
|
||
|
||
# https://www.zoomeye.org/doc?channel=api
|
||
zoomeye_api_username = ''
|
||
zoomeye_api_password = ''
|
||
|
||
# Certdb可以免费注册获取API: https://spyse.com/
|
||
certdb_api_token = ''
|
||
|
||
# https://www.circl.lu/services/passive-dns/
|
||
circl_api_username = ''
|
||
circl_api_password = ''
|
||
|
||
# https://www.dnsdb.info/
|
||
dnsdb_api_key = ''
|
||
|
||
# ipv4info可以免费注册获取API: http://ipv4info.com/tools/api/
|
||
# 免费的API有效期只有2天，到期之后可以再次生成，每天可以查询50次。
|
||
ipv4info_api_key = ''
|
||
|
||
subdomains_common = {'i', 'w', 'm', 'en', 'us', 'zh', 'w3', 'app', 'bbs',
|
||
'web', 'www', 'job', 'docs', 'news', 'blog', 'data',
|
||
'help', 'live', 'mall', 'blogs', 'files', 'forum',
|
||
'store', 'mobile'}
|
||
|
||
# 日志配置
|
||
# 终端日志输出格式
|
||
stdout_fmt = '<cyan>{time:HH:mm:ss,SSS}</cyan> ' \
|
||
'[<level>{level: <5}</level>] ' \
|
||
'<blue>{module}</blue>:<cyan>{line}</cyan> - ' \
|
||
'<level>{message}</level>'
|
||
# 日志文件记录格式
|
||
logfile_fmt = '<light-green>{time:YYYY-MM-DD HH:mm:ss,SSS}</light-green> ' \
|
||
'[<level>{level: <5}</level>] ' \
|
||
'<cyan>{process.name}({process.id})</cyan>:' \
|
||
'<cyan>{thread.name: <10}({thread.id: <5})</cyan> | ' \
|
||
'<blue>{module}</blue>.<blue>{function}</blue>:' \
|
||
'<blue>{line}</blue> - <level>{message}</level>'
|
||
|
||
log_path = result_save_path.joinpath('oneforall.log')
|
||
|
||
logger.remove()
|
||
logger.level(name='TRACE', no=5, color='<cyan><bold>', icon='✏️')
|
||
logger.level(name='DEBUG', no=10, color='<blue><bold>', icon='🐞 ')
|
||
logger.level(name='INFOR', no=20, color='<green><bold>', icon='ℹ️')
|
||
logger.level(name='ALERT', no=30, color='<yellow><bold>', icon='⚠️')
|
||
logger.level(name='ERROR', no=40, color='<red><bold>', icon='❌️')
|
||
logger.level(name='FATAL', no=50, color='<RED><bold>', icon='☠️')
|
||
|
||
if not os.environ.get('PYTHONIOENCODING'): # 设置编码
|
||
os.environ['PYTHONIOENCODING'] = 'utf-8'
|
||
|
||
logger.add(sys.stderr, level='INFOR', format=stdout_fmt, enqueue=True)
|
||
logger.add(log_path, level='DEBUG', format=logfile_fmt, enqueue=True,
|
||
encoding='utf-8')
|