def _init(self):
timestamp = str(time.time())
self.target_file = log_file_path.joinpath(
"tmp_{}.log".format(timestamp))
command = "-v -sn -PS -n --min-hostgroup {} --min-parallelism {} -iL {}"
self.command = command.format(nmap_min_hostgroup, nmap_min_parallelism,
self.target_file)
with open(self.target_file, "w") as f_obj:
f_obj.write("\n".join(self.ip_list))
def masscan_scan(self):
'''masscan 探测端口'''
logger.info("start masscan port scan...")
ports = "11-65535" if self.all_ports else ",".join(map(
str, self.ports))
timestamp = str(time.time())
target_file = log_file_path.joinpath("target_{}.log".format(timestamp))
result_file = log_file_path.joinpath("result_{}.log".format(timestamp))
f_obj = open(result_file, "w")
f_obj.close()
with open(target_file, "w") as f:
f.write("\n".join(self.ip_list))
if platform.system() == 'Linux' or platform.system() == 'Darwin':
masscan = masscan_path.joinpath("masscan")
if platform.system() == 'Windows':
masscan = masscan_path.joinpath("masscan.exe")
try:
command = "{} -sS -v -Pn -n -p{} -iL {} -oL {} --randomize-hosts --rate={}"
command = command.format(masscan, ports, target_file, result_file,
self.rate)
logger.info(command)
# , preexec_fn=os.setgid, stdout=PIPE,
p = Popen(command, shell=True, stderr=STDOUT)
# print("状态：", p.poll())
# print("开启进程的pid", p.pid)
# print("所属进程组的pid", os.getpgid(p.pid))
# time.sleep(90)
p.communicate()
except KeyboardInterrupt:
if os.path.exists(target_file):
os.remove(target_file)
if os.path.exists(result_file):
os.remove(result_file)
time.sleep(11)
if os.path.exists("paused.conf"):
os.remove("paused.conf")
# os.killpg(os.getpgid(p.pid), 9)
logger.error("User aborted.")
exit(0)
try:
lines = [line.strip() for line in open(result_file).readlines()]
for line in lines[1:-1]:
logger.info(line)
ip = line.split()[3]
port = int(line.split()[2])
if ip in self.open_list:
self.open_list[ip].append(port)
else:
self.open_list[ip] = [port]
except Exception as e:
logger.error(e)
finally:
if os.path.exists(target_file):
os.remove(target_file)
if os.path.exists(result_file):
os.remove(result_file)
return self.open_list
try:
from config import common_port
from config import wooyun_top100_web_port
from config import masscan_path
from config import log_file_path
from config import log_level
from libs.mylog import MyLog
except ModuleNotFoundError:
from Rpscan.config import common_port
from Rpscan.config import wooyun_top100_web_port
from Rpscan.config import masscan_path
from Rpscan.config import log_file_path
from Rpscan.config import log_level
from Rpscan.libs.mylog import MyLog
log_file = log_file_path.joinpath("{}.log".format(
time.strftime("%Y-%m-%d", time.localtime())))
logger = MyLog(loglevel=log_level, logger_name='port scan', logfile=log_file)
class PortScan(object):
"""端口扫描"""
def __init__(self, ip_list, all_ports=False, rate=2000):
super(PortScan, self).__init__()
self.ip_list = ip_list
self.rate = rate
self.all_ports = all_ports
self.open_list = {}
self.ports = sorted(list(set(wooyun_top100_web_port + common_port)))
logger.info("[*] PortScan...")
async def async_port_check(self, semaphore, ip_port):