From 0125ec2c572288a27db074f8de3895dc172e4c6c Mon Sep 17 00:00:00 2001
From: qux-bbb <1147635419@qq.com>
Date: Sat, 3 Aug 2024 20:07:50 +0800
Subject: [PATCH] patterns: Added Lua 5.2 bytecode pattern (#284)

---
 README.md                                 |   1 +
 patterns/lua52.hexpat                     | 110 ++++++++++++++++++++++
 tests/patterns/test_data/lua52.hexpat.lua | Bin 0 -> 727 bytes
 3 files changed, 111 insertions(+)
 create mode 100644 patterns/lua52.hexpat
 create mode 100644 tests/patterns/test_data/lua52.hexpat.lua

diff --git a/README.md b/README.md
index b2baab27..7a3a952c 100644
--- a/README.md
+++ b/README.md
@@ -80,6 +80,7 @@ Everything will immediately show up in ImHex's Content Store and gets bundled wi
 | Java Class | `application/x-java-applet` | [`patterns/java_class.hexpat`](patterns/java_class.hexpat) | Java Class files |
 | JPEG | `image/jpeg` | [`patterns/jpeg.hexpat`](patterns/jpeg.hexpat) | JPEG Image Format |
 | Lua 5.1 | | [`patterns/lua51.hexpat`](patterns/lua51.hexpat) | Lua 5.1 bytecode |
+| Lua 5.2 | | [`patterns/lua52.hexpat`](patterns/lua52.hexpat) | Lua 5.2 bytecode |
 | Lua 5.4 | | [`patterns/lua54.hexpat`](patterns/lua54.hexpat) | Lua 5.4 bytecode |
 | LCE Savefile | | [`patterns/lcesave.hexpat`](patterns/lcesave.hexpat) | Minecraft Legacy Console Edition save file |
 | Mach-O | `application/x-mach-binary` | [`patterns/macho.hexpat`](patterns/macho.hexpat) | Mach-O executable |
diff --git a/patterns/lua52.hexpat b/patterns/lua52.hexpat
new file mode 100644
index 00000000..c25ad643
--- /dev/null
+++ b/patterns/lua52.hexpat
@@ -0,0 +1,110 @@
+#pragma description Lua 5.2 bytecode
+
+import std.io;
+import std.mem;
+
+namespace impl {
+    fn format_LuaString(auto string) {
+       if (string.size == 0) {
+           return "None";
+        }
+        return std::format("\"{}\"", string.data);
+    };
+    
+    fn format_Constant(auto constant) {
+       return constant.data;
+    };
+    
+    fn format_Version(auto ver) {
+       return std::format("Ver. {}.{}", ver.major, ver.minor);
+    };
+}
+using LuaFunction;
+
+bitfield Version {
+    minor : 4;
+    major : 4;
+} [[format("impl::format_Version")]];
+
+struct LuaBinaryHeader {
+    char magic[4];
+    Version version_number;
+    u8 format_version;
+    u8 endianness;
+    u8 size_of_int;
+    u8 size_of_size_t;
+    u8 size_of_Instruction;
+    u8 lua_Number;
+    u8 is_lua_Number_integral;
+    char LUAC_TAIL[6];
+};
+
+struct LuaString {
+    u64 size;
+    if (size > 0) {
+        char data[size];
+    }
+}[[format("impl::format_LuaString")]];
+
+struct LuaConstant {
+    u8 type;
+    if (type == 0) {  // LUA_TNIL
+        // NULL
+    } else if (type == 1) {  // LUA_TBOOLEAN
+        u8 data;
+    } else if (type == 3) {  // LUA_TNUMBER
+        double data;
+    } else if (type == 4) {  // LUA_TSTRING
+        LuaString data;
+    }
+}[[format("impl::format_Constant")]];
+
+struct LuaUpvalue {
+    u8 instack;
+    u8 idx;
+};
+
+struct Vector<T> {
+    u32 size;
+    T values[size];
+};
+
+struct LocalVar {
+    LuaString varname;
+    u32 startpc;
+    u32 endpc;
+};
+
+struct LuaDebugInfo {
+    LuaString source;
+    Vector<u32> lineInfo;
+    Vector<LocalVar> local_vars;
+    Vector<LuaString> upvalues;
+};
+
+struct LuaConstants {
+    Vector<LuaConstant> constants;
+    u32 sizep;  // size of proto
+    if (sizep > 0) {
+        LuaFunction protos[sizep];
+    }
+};
+
+struct LuaFunction {
+    u32 line_defined;
+    u32 last_line_defined;
+    u8 number_of_parameters;
+    u8 is_vararg;
+    u8 maxstacksize;
+    Vector<u32> code;
+    LuaConstants luaConstants;
+    Vector<LuaUpvalue> upvalues;
+    LuaDebugInfo debugInfo;
+};
+
+struct LuaFile {
+    LuaBinaryHeader header;
+    LuaFunction func;
+};
+
+LuaFile file @ 0;
diff --git a/tests/patterns/test_data/lua52.hexpat.lua b/tests/patterns/test_data/lua52.hexpat.lua
new file mode 100644
index 0000000000000000000000000000000000000000..beb51d5d88b51a26a15cfa2a29ea88c76bc95925
GIT binary patch
literal 727
zcmZuv%TB^T6g>mA)cQ72gT}zdo$(u{7+o0`F5DS1Xcdy77$`C9hWPja6TheWW!^i}
z0#<HvGIQVeo|&VovATiq0xv-Qb0a(qQHZEC=o(0Ad?RrQiDL=jBSK9^PB#z%%+&fb
z=MS>z)#@g)@oQWIc=ORhXnx$8eZRY!EgN&<Vm^GFBa&5XYrtTW>@=5?v>Ufa=@6cu
z8@jR9gSq-V>?P^na2+S`YI!*2_L3)fUani`V)10$N7Xq4IcwXWq^hIidEVoInyxR-
z(0myT^}{@(qiD+2qOF>g235LuA48CTk8nT`ZVRS(DuIklUm3(5(akheNS|$8TGQ4(
zZBhi6VJRL^rtv5}(_;lDY<~${YhZ=bND|eL?S&J#Sm9`2T;E!QSW=xo33JR;7U1Q}
h;JFI$r(g;TY|)*uonwbi#_j@mS|+Ss0;&Ip{RKBTPr3jA

literal 0
HcmV?d00001