Mirror/masscanned
1
0
Fork 0
mirror of https://github.com/ivre/masscanned.git synced 2025-10-02 06:38:21 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Clean-up Python test script

Browse source
This commit is contained in:
Pierre Lalet 2021-12-08 23:35:26 +01:00
parent 12aa60b848
commit 9fb050188d
1 changed files with 44 additions and 31 deletions
Download patch file Download diff file Expand all files Collapse all files

75
test/test_masscanned.py
View file

@ -16,56 +16,68 @@
# You should have received a copy of the GNU General Public License # You should have received a copy of the GNU General Public License
# along with Masscanned. If not, see <http://www.gnu.org/licenses/>. # along with Masscanned. If not, see <http://www.gnu.org/licenses/>.
from scapy.all import *
from time import sleep
from tempfile import _get_candidate_names as gen_tmp_filename
from tempfile import gettempdir
import subprocess
import logging import logging
import sys
import os 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.all import test_all
from src.conf import * from src.conf import *
# if args in CLI, they are passed to masscanned def setup_logs():
if len(sys.argv) > 1: ch = logging.StreamHandler()
args = " ".join(sys.argv[1:]) ch.setFormatter(logging.Formatter("%(levelname)s\t%(message)s"))
else: ch.setLevel(logging.INFO)
args = "" log = logging.getLogger(__name__)
log.setLevel(logging.INFO)
fmt = logging.Formatter("%(levelname)s\t%(message)s") log.addHandler(ch)
ch = logging.StreamHandler() return log
ch.setFormatter(fmt)
ch.setLevel(logging.INFO)
LOG = logging.getLogger(__name__)
LOG.setLevel(logging.INFO)
LOG.addHandler(ch)
LOG = setup_logs()
IFACE = "tap0" IFACE = "tap0"
conf.verb = 0 conf.verb = 0
# prepare configuration file for masscanned # prepare configuration file for masscanned
ipfile = os.path.join(gettempdir(), next(gen_tmp_filename())) with NamedTemporaryFile(delete=False, mode="w") as ipfile:
with open(ipfile, "w") as f: ipfile.write(f"{IPV4_ADDR}\n")
f.write("{}\n".format(IPV4_ADDR)) ipfile.write(f"{IPV6_ADDR}\n")
f.write("{}\n".format(IPV6_ADDR))
# create test interface # create test interface
tap = TunTapInterface(IFACE) tap = TunTapInterface(IFACE)
conf.iface = resolve_iface(IFACE) conf.iface = resolve_iface(IFACE)
# set interface # set interface
subprocess.run("ip a a dev {} 192.0.0.2".format(conf.iface), shell=True) subprocess.check_call(["ip", "addr", "add", "dev", IFACE, "192.0.0.2"])
subprocess.run("ip link set {} up".format(conf.iface), shell=True) subprocess.check_call(["ip", "link", "set", IFACE, "up"])
# start capture # start capture
tcpdump = subprocess.Popen("tcpdump -enli {} -w {}".format(conf.iface, os.path.join(OUTDIR, "test_capture.pcap")), shell=True, tcpdump = subprocess.Popen(
stdin=None, stdout=None, stderr=None, close_fds=True) ["tcpdump", "-enli", IFACE, "-w", os.path.join(OUTDIR, "test_capture.pcap")]
)
# run masscanned # run masscanned
masscanned = subprocess.Popen("RUST_BACKTRACE=1 ./target/debug/masscanned -vvvvv -i {} -f {} -a {} {}".format(conf.iface, ipfile, MAC_ADDR, args), shell=True, masscanned = subprocess.Popen(
stdin=None, stdout=open("test/res/masscanned.stdout", "w"), stderr=open("test/res/masscanned.stderr", "w"), close_fds=True) [
"./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) sleep(1)
try: try:
@ -75,7 +87,8 @@ except AssertionError:
# terminate masscanned # terminate masscanned
masscanned.kill() masscanned.kill()
masscanned.wait()
# terminate capture # terminate capture
sleep(2)
tcpdump.kill() tcpdump.kill()
tcpdump.wait()
sys.exit(result) sys.exit(result)