inital release
This commit is contained in:
GitHub
parent
4fdaf0542b
commit
a45b04af15
|
|
@ -0,0 +1,248 @@
|
|||
import os
|
||||
import struct
|
||||
import array
|
||||
|
||||
lua_opcode_types = [
|
||||
"ABC", "ABx", "ABC", "ABC",
|
||||
"ABC", "ABx", "ABC", "ABx",
|
||||
"ABC", "ABC", "ABC", "ABC",
|
||||
"ABC", "ABC", "ABC", "ABC",
|
||||
"ABC", "ABC", "ABC", "ABC",
|
||||
"ABC", "ABC", "AsBx", "ABC",
|
||||
"ABC", "ABC", "ABC", "ABC",
|
||||
"ABC", "ABC", "ABC", "AsBx",
|
||||
"AsBx", "ABC", "ABC", "ABC",
|
||||
"ABx", "ABC"
|
||||
]
|
||||
|
||||
lua_opcode_names = [
|
||||
"MOVE", "LOADK", "LOADBOOL", "LOADNIL",
|
||||
"GETUPVAL", "GETGLOBAL", "GETTABLE", "SETGLOBAL",
|
||||
"SETUPVAL", "SETTABLE", "NEWTABLE", "SELF",
|
||||
"ADD", "SUB", "MUL", "DIV",
|
||||
"MOD", "POW", "UNM", "NOT",
|
||||
"LEN", "CONCAT", "JMP", "EQ",
|
||||
"LT", "LE", "TEST", "TESTSET",
|
||||
"CALL", "TAILCALL", "RETURN", "FORLOOP",
|
||||
"FORPREP", "TFORLOOP", "SETLIST", "CLOSE",
|
||||
"CLOSURE", "VARARG"
|
||||
]
|
||||
|
||||
# at [p]osition size of k
|
||||
def get_bits(num, p, k):
|
||||
# convert number into binary first
|
||||
binary = bin(num)
|
||||
|
||||
# remove first two characters
|
||||
binary = binary[2:]
|
||||
|
||||
# fill in missing bits
|
||||
for i in range(32 - len(binary)):
|
||||
binary = '0' + binary
|
||||
|
||||
end = len(binary) - p + 1
|
||||
start = len(binary) - k + 1
|
||||
|
||||
# extract k bit sub-string
|
||||
kBitSubStr = binary[start : end]
|
||||
|
||||
# convert extracted sub-string into decimal again
|
||||
return (int(kBitSubStr,2) % 256)
|
||||
|
||||
class LuaCompiler:
|
||||
def __init__(self):
|
||||
self.luac = "luac5.1"
|
||||
self.o_flag = "-o"
|
||||
self.temp_out = "out.luac"
|
||||
self.chunks = []
|
||||
self.index = 0
|
||||
|
||||
|
||||
def get_byte(self):
|
||||
b = self.bytecode[self.index]
|
||||
self.index = self.index + 1
|
||||
return b
|
||||
|
||||
def get_int32(self):
|
||||
i = 0
|
||||
if (self.big_endian):
|
||||
i = int.from_bytes(self.bytecode[self.index:self.index+4], byteorder='big', signed=False)
|
||||
else:
|
||||
i = int.from_bytes(self.bytecode[self.index:self.index+4], byteorder='little', signed=False)
|
||||
self.index = self.index + self.int_size
|
||||
return i
|
||||
|
||||
def get_int(self):
|
||||
i = 0
|
||||
if (self.big_endian):
|
||||
i = int.from_bytes(self.bytecode[self.index:self.index+self.int_size], byteorder='big', signed=False)
|
||||
else:
|
||||
i = int.from_bytes(self.bytecode[self.index:self.index+self.int_size], byteorder='little', signed=False)
|
||||
self.index = self.index + self.int_size
|
||||
return i
|
||||
|
||||
def get_size_t(self):
|
||||
s = ''
|
||||
if (self.big_endian):
|
||||
s = int.from_bytes(self.bytecode[self.index:self.index+self.size_t], byteorder='big', signed=False)
|
||||
else:
|
||||
s = int.from_bytes(self.bytecode[self.index:self.index+self.size_t], byteorder='little', signed=False)
|
||||
self.index = self.index + self.size_t
|
||||
return s
|
||||
|
||||
def get_float(self):
|
||||
f = struct.unpack('<f', bytearray(self.bytecode[self.index:self.index+8]))
|
||||
self.index = self.index + 8
|
||||
return f
|
||||
|
||||
def get_string(self, size):
|
||||
if (size == None):
|
||||
size = self.get_size_t()
|
||||
if (size == 0):
|
||||
return None
|
||||
|
||||
s = "".join(chr(x) for x in self.bytecode[self.index:self.index+size])
|
||||
self.index = self.index + size
|
||||
return s
|
||||
|
||||
def decode_chunk(self):
|
||||
chunk = {
|
||||
'INSTRUCTIONS': {},
|
||||
'CONSTANTS': {},
|
||||
'PROTOTYPES': {}
|
||||
}
|
||||
|
||||
chunk['NAME'] = self.get_string(None)
|
||||
chunk['FIRST_LINE'] = self.get_int()
|
||||
chunk['LAST_LINE'] = self.get_int()
|
||||
|
||||
chunk['UPVALUES'] = self.get_byte()
|
||||
chunk['ARGUMENTS'] = self.get_byte()
|
||||
chunk['VARG'] = self.get_byte()
|
||||
chunk['STACK'] = self.get_byte()
|
||||
|
||||
if (not chunk['NAME'] == None):
|
||||
chunk['NAME'] = chunk['NAME'][1:-1]
|
||||
|
||||
# parse instructions
|
||||
print("** DECODING INSTRUCTIONS")
|
||||
|
||||
num = self.get_int()
|
||||
for i in range(num):
|
||||
instruction = {
|
||||
# opcode = opcode number;
|
||||
# type = [ABC, ABx, AsBx]
|
||||
# A, B, C, Bx, or sBx depending on type
|
||||
};
|
||||
|
||||
data = self.get_int32();
|
||||
opcode = get_bits(data, 1, 6);
|
||||
tp = lua_opcode_types[opcode];
|
||||
|
||||
instruction['OPCODE'] = opcode
|
||||
instruction['TYPE'] = tp
|
||||
instruction['A'] = get_bits(data, 7, 14)
|
||||
|
||||
if instruction['TYPE'] == "ABC":
|
||||
instruction['B'] = get_bits(data, 24, 32);
|
||||
instruction['C'] = get_bits(data, 15, 23);
|
||||
elif instruction['TYPE'] == "ABx":
|
||||
instruction['Bx'] = get_bits(data, 15, 32);
|
||||
elif instruction['TYPE'] == "AsBx":
|
||||
instruction['sBx'] = get_bits(data, 15, 32) - 131071;
|
||||
|
||||
chunk['INSTRUCTIONS'][i] = instruction
|
||||
|
||||
print(lua_opcode_names[opcode], instruction)
|
||||
|
||||
# get constants
|
||||
print("** DECODING CONSTANTS")
|
||||
|
||||
num = self.get_int();
|
||||
for i in range(num):
|
||||
constant = {
|
||||
# type = constant type;
|
||||
# data = constant data;
|
||||
};
|
||||
constant['TYPE'] = self.get_byte()
|
||||
|
||||
if constant['TYPE'] == 1:
|
||||
constant['DATA'] = (self.get_byte() != 0);
|
||||
elif constant['TYPE'] == 3:
|
||||
constant['DATA'] = self.get_float();
|
||||
elif constant['TYPE'] == 4:
|
||||
constant['DATA'] = self.get_string(None)[:-1];
|
||||
|
||||
print(constant)
|
||||
|
||||
chunk['CONSTANTS'][i] = constant
|
||||
|
||||
# parse protos
|
||||
|
||||
print("** DECODING PROTOS")
|
||||
|
||||
num = self.get_int();
|
||||
for i in range(num):
|
||||
chunk['PROTOTYPES'][i] = self.decode_chunk()
|
||||
|
||||
# debug stuff
|
||||
print("** DECODING DEBUG SYMBOLS")
|
||||
|
||||
# line numbers
|
||||
num = self.get_int();
|
||||
for i in range(num):
|
||||
self.get_int32()
|
||||
|
||||
# locals
|
||||
num = self.get_int();
|
||||
for i in range(num):
|
||||
self.get_string(None)[:-1] # local name
|
||||
self.get_int32() # local start PC
|
||||
self.get_int32() # local end PC
|
||||
|
||||
# upvalues
|
||||
num = self.get_int();
|
||||
for i in range(num):
|
||||
self.get_string(None) # upvalue name
|
||||
|
||||
return chunk
|
||||
|
||||
def decode_rawbytecode(self, rawbytecode):
|
||||
# bytecode sanity checks
|
||||
if not rawbytecode[0:4] == b'\x1bLua':
|
||||
print("Lua Bytecode expected!")
|
||||
exit(0)
|
||||
|
||||
bytecode = array.array('b', rawbytecode)
|
||||
return self.decode_bytecode(bytecode)
|
||||
|
||||
def decode_bytecode(self, bytecode):
|
||||
self.bytecode = bytecode
|
||||
|
||||
# alligns index lol
|
||||
self.index = 4
|
||||
|
||||
self.vm_version = self.get_byte()
|
||||
self.offical_bytecode = self.get_byte()
|
||||
self.big_endian = (self.get_byte() == 0)
|
||||
self.int_size = self.get_byte()
|
||||
self.size_t = self.get_byte()
|
||||
self.thing = self.get_string(3)
|
||||
|
||||
|
||||
print("Lua VM version: ", hex(self.vm_version))
|
||||
print("Big Endian: ", self.big_endian)
|
||||
print("int_size: ", self.int_size)
|
||||
print("size_t: ", self.size_t)
|
||||
print(self.thing)
|
||||
|
||||
#print(self.bytecode)
|
||||
return self.decode_chunk()
|
||||
|
||||
def compileC(self, luafile):
|
||||
os.system(self.luac + " " + self.o_flag + " " + self.temp_out + " " + luafile)
|
||||
|
||||
with open(self.temp_out, 'rb') as luac_file:
|
||||
bytecode = luac_file.read()
|
||||
return self.decode_rawbytecode(bytecode)
|
||||
|
||||
Loading…
Reference in New Issue