lp: added support for OP_TEST

lparser.py

@@ -231,7 +231,8 @@ class LuaDecomp:
     def __emitOperand(self, a: int, b: str, c: str, op: str) -> None:
         self.__setReg(a, "(" + b + op + c + ")")
 
-    def __compJmp(self, op: str):
+    # handles conditional jumps
+    def __condJmp(self, op: str, rkBC: bool = True):
         instr = self.__getCurrInstr()
         jmpType = "if"
         scopeStart = "then"
@@ -254,7 +255,13 @@ class LuaDecomp:
             self.__addExpr("%s not " % jmpType)
         else:
             self.__addExpr("%s " % jmpType)
+
+        # write actual comparison
+        if rkBC:
             self.__addExpr(self.__readRK(instr.B) + op + self.__readRK(instr.C) + " ")
+        else: # just testing rkB
+            self.__addExpr(op + self.__readRK(instr.B))
+
         self.pc += 1 # skip next instr
         if scopeStart:
             self.__startScope("%s " % scopeStart, self.pc - 1, jmp)
@@ -368,11 +375,16 @@ class LuaDecomp:
         elif instr.opcode == Opcodes.JMP:
             pass
         elif instr.opcode == Opcodes.EQ:
-            self.__compJmp(" == ")
+            self.__condJmp(" == ")
         elif instr.opcode == Opcodes.LT:
-            self.__compJmp(" < ")
+            self.__condJmp(" < ")
         elif instr.opcode == Opcodes.LE:
-            self.__compJmp(" <= ")
+            self.__condJmp(" <= ")
+        elif instr.opcode == Opcodes.TEST:
+            if instr.C == 0:
+                self.__condJmp("", False)
+            else:
+                self.__condJmp("not ", False)
         elif instr.opcode == Opcodes.CALL:
             preStr = ""
             callStr = ""

lundump.py

@@ -1,7 +1,7 @@
 '''
     l(un)dump.py
 
-    A Lua5.1 cross-platform bytecode deserializer. This module pulls int and size_t sizes from the
+    A Lua5.1 cross-platform bytecode deserializer && serializer. This module pulls int and size_t sizes from the
     chunk header, meaning it should be able to deserialize lua bytecode dumps from most platforms,
     regardless of the host machine.
 
@@ -9,11 +9,9 @@
     as well as read the lundump.c source file from the Lua5.1 source.
 '''
 
-from multiprocessing.spawn import get_executable
 import struct
 import array
 from enum import IntEnum, Enum, auto
-from typing_extensions import Self
 
 class InstructionType(Enum):
     ABC = auto(),