From ec8011a6e4d34f6c109c3c1649f5b1c5b7a0699f Mon Sep 17 00:00:00 2001
From: Pierre Lalet <pierre.lalet@ssi.gouv.fr>
Date: Mon, 13 Dec 2021 19:30:40 +0100
Subject: [PATCH] Tests: add p0f logs

---
 test/test_masscanned.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/test/test_masscanned.py b/test/test_masscanned.py
index 11c75dd..13a8f18 100755
--- a/test/test_masscanned.py
+++ b/test/test_masscanned.py
@@ -54,6 +54,7 @@ if HAS_IVRE:
     ZEEK_PASSIVERECON = bool(os.environ.get("USE_ZEEK"))
 else:
     ZEEK_PASSIVERECON = False
+P0F = bool(os.environ.get("USE_P0F"))
 conf.verb = 0
 
 # prepare configuration file for masscanned
@@ -98,6 +99,12 @@ if ZEEK_PASSIVERECON:
         stdout=open("test/res/zeek_passiverecon.stdout", "w"),
         stderr=open("test/res/zeek_passiverecon.stderr", "w"),
     )
+if P0F:
+    p0f = subprocess.Popen(
+        ["p0f", "-i", IFACE, "-o", "test/res/p0f_log.txt"],
+        stdout=open("test/res/p0f.stdout", "w"),
+        stderr=open("test/res/p0f.stderr", "w"),
+    )
 # run masscanned
 masscanned = subprocess.Popen(
     [
@@ -133,4 +140,7 @@ if TCPDUMP:
 if ZEEK_PASSIVERECON:
     zeek.kill()
     zeek.wait()
+if P0F:
+    p0f.kill()
+    p0f.wait()
 sys.exit(result)