mirror of
https://github.com/ivre/masscanned.git
synced 2025-10-02 06:38:21 +00:00
98 lines
2.6 KiB
Python
Executable file
98 lines
2.6 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
|
|
# This file is part of masscanned.
|
|
# Copyright 2021 - The IVRE project
|
|
#
|
|
# Masscanned is free software: you can redistribute it and/or modify it
|
|
# under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# Masscanned is distributed in the hope that it will be useful, but WITHOUT
|
|
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
|
|
# or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public
|
|
# License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with Masscanned. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
import logging
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
from time import sleep
|
|
from tempfile import NamedTemporaryFile
|
|
|
|
from scapy.config import conf
|
|
from scapy.interfaces import resolve_iface
|
|
from scapy.layers.tuntap import TunTapInterface
|
|
|
|
from src.all import test_all
|
|
from src.conf import IPV4_ADDR, IPV6_ADDR, MAC_ADDR, OUTDIR
|
|
|
|
|
|
def setup_logs():
|
|
ch = logging.StreamHandler()
|
|
ch.setFormatter(logging.Formatter("%(levelname)s\t%(message)s"))
|
|
ch.setLevel(logging.INFO)
|
|
log = logging.getLogger(__name__)
|
|
log.setLevel(logging.INFO)
|
|
log.addHandler(ch)
|
|
return log
|
|
|
|
|
|
LOG = setup_logs()
|
|
IFACE = "tap0"
|
|
conf.verb = 0
|
|
|
|
# prepare configuration file for masscanned
|
|
with NamedTemporaryFile(delete=False, mode="w") as ipfile:
|
|
ipfile.write(f"{IPV4_ADDR}\n")
|
|
ipfile.write(f"{IPV6_ADDR}\n")
|
|
|
|
# create test interface
|
|
tap = TunTapInterface(IFACE)
|
|
conf.iface = resolve_iface(IFACE)
|
|
|
|
# set interface
|
|
subprocess.check_call(["ip", "addr", "add", "dev", IFACE, "192.0.0.0/31"])
|
|
subprocess.check_call(["ip", "link", "set", IFACE, "up"])
|
|
subprocess.check_call(["ip", "route", "add", "1.2.3.4/32", "via", IPV4_ADDR])
|
|
conf.route.resync()
|
|
|
|
# start capture
|
|
tcpdump = subprocess.Popen(
|
|
["tcpdump", "-enli", IFACE, "-w", os.path.join(OUTDIR, "test_capture.pcap")]
|
|
)
|
|
# run masscanned
|
|
masscanned = subprocess.Popen(
|
|
[
|
|
"./target/debug/masscanned",
|
|
"-vvvvv",
|
|
"-i",
|
|
IFACE,
|
|
"-f",
|
|
ipfile.name,
|
|
"-a",
|
|
MAC_ADDR,
|
|
]
|
|
# if args in CLI, they are passed to masscanned
|
|
+ sys.argv[1:],
|
|
env=dict(os.environ, RUST_BACKTRACE="1"),
|
|
stdout=open("test/res/masscanned.stdout", "w"),
|
|
stderr=open("test/res/masscanned.stderr", "w"),
|
|
)
|
|
sleep(1)
|
|
|
|
try:
|
|
result = test_all(tap)
|
|
except AssertionError:
|
|
result = -1
|
|
|
|
# terminate masscanned
|
|
masscanned.kill()
|
|
masscanned.wait()
|
|
# terminate capture
|
|
tcpdump.kill()
|
|
tcpdump.wait()
|
|
sys.exit(result)
|