1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071 |
- #!/usr/bin/env python3
- # -*- coding: utf-8 -*-
- """
- @Author: hywell
- @Email: hywell.28@gmail.com
- @Blog: iassas.com
- @Date: 2019/10/16 22:24
- """
- from lib.core.data import logger
- from lib.parse.ip import parseTarget
- from lib.core.setting import ASYNC_NUM, CONF, TARGETS, PROCESS_NUM, RESULT
- def TargetRegister(targets):
- target_result = []
- for target in targets:
- parse_targets = parseTarget(target)
- for parse_target in parse_targets:
- ip = parse_target["ip"]
- domain = parse_target["domain"]
- url = parse_target["url"]
- if ip:
- RESULT[ip] = {}
- TARGETS.IP.append(ip)
- target_result.append([1, ip, None])
- elif domain:
- RESULT[domain] = {"subDomain": {"open": [], "close": []}, "status": ""}
- TARGETS.DOMAIN.append(domain)
- target_result.append([2, domain, None])
- for sub in CONF.dns_sub:
- target_result.append([2, "%s.%s" % (sub, domain), domain])
- elif url:
- RESULT[url] = {"dic": [], "status": ""}
- TARGETS.URL.append(url)
- target_result.append([3, url, None])
- for dir in CONF.dir:
- if "/" in dir:
- target_result.append([3, url + dir, url])
- else:
- target_result.append([3, url + "/" + dir, url])
- else:
- logger.info("%s is end" % domain if domain else ip)
- return target_result
- def InitRegister(level):
- CONF.async_num = ASYNC_NUM
- CONF.base_nums = None
- CONF.dir = []
- CONF.dns_servers = []
- CONF.dns_sub = []
- CONF.level = level
- CONF.levels = {"1": "", "2": "", "3": ""}
- CONF.process_num = PROCESS_NUM
- logger.info("[AWIScan] Configuration has been initialized")
- TARGETS.IP = []
- TARGETS.DOMAIN = []
- TARGETS.URL = []
- TARGETS.END.ip = []
- TARGETS.END.domain = []
- TARGETS.END.url = []
- TARGETS.ERROR.ip = []
- TARGETS.ERROR.domain = []
- TARGETS.ERROR.url = []
- logger.info("[AWIScan] Target information has been initialized")
- def initOptions(level):
- InitRegister(level)
|