forked from mia/Aegisub
429 lines
13 KiB
Lua
429 lines
13 KiB
Lua
----------------------------------------------------------------------------
|
|
-- LuaJIT MIPS disassembler module.
|
|
--
|
|
-- Copyright (C) 2005-2017 Mike Pall. All rights reserved.
|
|
-- Released under the MIT/X license. See Copyright Notice in luajit.h
|
|
----------------------------------------------------------------------------
|
|
-- This is a helper module used by the LuaJIT machine code dumper module.
|
|
--
|
|
-- It disassembles all standard MIPS32R1/R2 instructions.
|
|
-- Default mode is big-endian, but see: dis_mipsel.lua
|
|
------------------------------------------------------------------------------
|
|
|
|
local type = type
|
|
local byte, format = string.byte, string.format
|
|
local match, gmatch = string.match, string.gmatch
|
|
local concat = table.concat
|
|
local bit = require("bit")
|
|
local band, bor, tohex = bit.band, bit.bor, bit.tohex
|
|
local lshift, rshift, arshift = bit.lshift, bit.rshift, bit.arshift
|
|
|
|
------------------------------------------------------------------------------
|
|
-- Primary and extended opcode maps
|
|
------------------------------------------------------------------------------
|
|
|
|
local map_movci = { shift = 16, mask = 1, [0] = "movfDSC", "movtDSC", }
|
|
local map_srl = { shift = 21, mask = 1, [0] = "srlDTA", "rotrDTA", }
|
|
local map_srlv = { shift = 6, mask = 1, [0] = "srlvDTS", "rotrvDTS", }
|
|
|
|
local map_special = {
|
|
shift = 0, mask = 63,
|
|
[0] = { shift = 0, mask = -1, [0] = "nop", _ = "sllDTA" },
|
|
map_movci, map_srl, "sraDTA",
|
|
"sllvDTS", false, map_srlv, "sravDTS",
|
|
"jrS", "jalrD1S", "movzDST", "movnDST",
|
|
"syscallY", "breakY", false, "sync",
|
|
"mfhiD", "mthiS", "mfloD", "mtloS",
|
|
false, false, false, false,
|
|
"multST", "multuST", "divST", "divuST",
|
|
false, false, false, false,
|
|
"addDST", "addu|moveDST0", "subDST", "subu|neguDS0T",
|
|
"andDST", "orDST", "xorDST", "nor|notDST0",
|
|
false, false, "sltDST", "sltuDST",
|
|
false, false, false, false,
|
|
"tgeSTZ", "tgeuSTZ", "tltSTZ", "tltuSTZ",
|
|
"teqSTZ", false, "tneSTZ",
|
|
}
|
|
|
|
local map_special2 = {
|
|
shift = 0, mask = 63,
|
|
[0] = "maddST", "madduST", "mulDST", false,
|
|
"msubST", "msubuST",
|
|
[32] = "clzDS", [33] = "cloDS",
|
|
[63] = "sdbbpY",
|
|
}
|
|
|
|
local map_bshfl = {
|
|
shift = 6, mask = 31,
|
|
[2] = "wsbhDT",
|
|
[16] = "sebDT",
|
|
[24] = "sehDT",
|
|
}
|
|
|
|
local map_special3 = {
|
|
shift = 0, mask = 63,
|
|
[0] = "extTSAK", [4] = "insTSAL",
|
|
[32] = map_bshfl,
|
|
[59] = "rdhwrTD",
|
|
}
|
|
|
|
local map_regimm = {
|
|
shift = 16, mask = 31,
|
|
[0] = "bltzSB", "bgezSB", "bltzlSB", "bgezlSB",
|
|
false, false, false, false,
|
|
"tgeiSI", "tgeiuSI", "tltiSI", "tltiuSI",
|
|
"teqiSI", false, "tneiSI", false,
|
|
"bltzalSB", "bgezalSB", "bltzallSB", "bgezallSB",
|
|
false, false, false, false,
|
|
false, false, false, false,
|
|
false, false, false, "synciSO",
|
|
}
|
|
|
|
local map_cop0 = {
|
|
shift = 25, mask = 1,
|
|
[0] = {
|
|
shift = 21, mask = 15,
|
|
[0] = "mfc0TDW", [4] = "mtc0TDW",
|
|
[10] = "rdpgprDT",
|
|
[11] = { shift = 5, mask = 1, [0] = "diT0", "eiT0", },
|
|
[14] = "wrpgprDT",
|
|
}, {
|
|
shift = 0, mask = 63,
|
|
[1] = "tlbr", [2] = "tlbwi", [6] = "tlbwr", [8] = "tlbp",
|
|
[24] = "eret", [31] = "deret",
|
|
[32] = "wait",
|
|
},
|
|
}
|
|
|
|
local map_cop1s = {
|
|
shift = 0, mask = 63,
|
|
[0] = "add.sFGH", "sub.sFGH", "mul.sFGH", "div.sFGH",
|
|
"sqrt.sFG", "abs.sFG", "mov.sFG", "neg.sFG",
|
|
"round.l.sFG", "trunc.l.sFG", "ceil.l.sFG", "floor.l.sFG",
|
|
"round.w.sFG", "trunc.w.sFG", "ceil.w.sFG", "floor.w.sFG",
|
|
false,
|
|
{ shift = 16, mask = 1, [0] = "movf.sFGC", "movt.sFGC" },
|
|
"movz.sFGT", "movn.sFGT",
|
|
false, "recip.sFG", "rsqrt.sFG", false,
|
|
false, false, false, false,
|
|
false, false, false, false,
|
|
false, "cvt.d.sFG", false, false,
|
|
"cvt.w.sFG", "cvt.l.sFG", "cvt.ps.sFGH", false,
|
|
false, false, false, false,
|
|
false, false, false, false,
|
|
"c.f.sVGH", "c.un.sVGH", "c.eq.sVGH", "c.ueq.sVGH",
|
|
"c.olt.sVGH", "c.ult.sVGH", "c.ole.sVGH", "c.ule.sVGH",
|
|
"c.sf.sVGH", "c.ngle.sVGH", "c.seq.sVGH", "c.ngl.sVGH",
|
|
"c.lt.sVGH", "c.nge.sVGH", "c.le.sVGH", "c.ngt.sVGH",
|
|
}
|
|
|
|
local map_cop1d = {
|
|
shift = 0, mask = 63,
|
|
[0] = "add.dFGH", "sub.dFGH", "mul.dFGH", "div.dFGH",
|
|
"sqrt.dFG", "abs.dFG", "mov.dFG", "neg.dFG",
|
|
"round.l.dFG", "trunc.l.dFG", "ceil.l.dFG", "floor.l.dFG",
|
|
"round.w.dFG", "trunc.w.dFG", "ceil.w.dFG", "floor.w.dFG",
|
|
false,
|
|
{ shift = 16, mask = 1, [0] = "movf.dFGC", "movt.dFGC" },
|
|
"movz.dFGT", "movn.dFGT",
|
|
false, "recip.dFG", "rsqrt.dFG", false,
|
|
false, false, false, false,
|
|
false, false, false, false,
|
|
"cvt.s.dFG", false, false, false,
|
|
"cvt.w.dFG", "cvt.l.dFG", false, false,
|
|
false, false, false, false,
|
|
false, false, false, false,
|
|
"c.f.dVGH", "c.un.dVGH", "c.eq.dVGH", "c.ueq.dVGH",
|
|
"c.olt.dVGH", "c.ult.dVGH", "c.ole.dVGH", "c.ule.dVGH",
|
|
"c.df.dVGH", "c.ngle.dVGH", "c.deq.dVGH", "c.ngl.dVGH",
|
|
"c.lt.dVGH", "c.nge.dVGH", "c.le.dVGH", "c.ngt.dVGH",
|
|
}
|
|
|
|
local map_cop1ps = {
|
|
shift = 0, mask = 63,
|
|
[0] = "add.psFGH", "sub.psFGH", "mul.psFGH", false,
|
|
false, "abs.psFG", "mov.psFG", "neg.psFG",
|
|
false, false, false, false,
|
|
false, false, false, false,
|
|
false,
|
|
{ shift = 16, mask = 1, [0] = "movf.psFGC", "movt.psFGC" },
|
|
"movz.psFGT", "movn.psFGT",
|
|
false, false, false, false,
|
|
false, false, false, false,
|
|
false, false, false, false,
|
|
"cvt.s.puFG", false, false, false,
|
|
false, false, false, false,
|
|
"cvt.s.plFG", false, false, false,
|
|
"pll.psFGH", "plu.psFGH", "pul.psFGH", "puu.psFGH",
|
|
"c.f.psVGH", "c.un.psVGH", "c.eq.psVGH", "c.ueq.psVGH",
|
|
"c.olt.psVGH", "c.ult.psVGH", "c.ole.psVGH", "c.ule.psVGH",
|
|
"c.psf.psVGH", "c.ngle.psVGH", "c.pseq.psVGH", "c.ngl.psVGH",
|
|
"c.lt.psVGH", "c.nge.psVGH", "c.le.psVGH", "c.ngt.psVGH",
|
|
}
|
|
|
|
local map_cop1w = {
|
|
shift = 0, mask = 63,
|
|
[32] = "cvt.s.wFG", [33] = "cvt.d.wFG",
|
|
}
|
|
|
|
local map_cop1l = {
|
|
shift = 0, mask = 63,
|
|
[32] = "cvt.s.lFG", [33] = "cvt.d.lFG",
|
|
}
|
|
|
|
local map_cop1bc = {
|
|
shift = 16, mask = 3,
|
|
[0] = "bc1fCB", "bc1tCB", "bc1flCB", "bc1tlCB",
|
|
}
|
|
|
|
local map_cop1 = {
|
|
shift = 21, mask = 31,
|
|
[0] = "mfc1TG", false, "cfc1TG", "mfhc1TG",
|
|
"mtc1TG", false, "ctc1TG", "mthc1TG",
|
|
map_cop1bc, false, false, false,
|
|
false, false, false, false,
|
|
map_cop1s, map_cop1d, false, false,
|
|
map_cop1w, map_cop1l, map_cop1ps,
|
|
}
|
|
|
|
local map_cop1x = {
|
|
shift = 0, mask = 63,
|
|
[0] = "lwxc1FSX", "ldxc1FSX", false, false,
|
|
false, "luxc1FSX", false, false,
|
|
"swxc1FSX", "sdxc1FSX", false, false,
|
|
false, "suxc1FSX", false, "prefxMSX",
|
|
false, false, false, false,
|
|
false, false, false, false,
|
|
false, false, false, false,
|
|
false, false, "alnv.psFGHS", false,
|
|
"madd.sFRGH", "madd.dFRGH", false, false,
|
|
false, false, "madd.psFRGH", false,
|
|
"msub.sFRGH", "msub.dFRGH", false, false,
|
|
false, false, "msub.psFRGH", false,
|
|
"nmadd.sFRGH", "nmadd.dFRGH", false, false,
|
|
false, false, "nmadd.psFRGH", false,
|
|
"nmsub.sFRGH", "nmsub.dFRGH", false, false,
|
|
false, false, "nmsub.psFRGH", false,
|
|
}
|
|
|
|
local map_pri = {
|
|
[0] = map_special, map_regimm, "jJ", "jalJ",
|
|
"beq|beqz|bST00B", "bne|bnezST0B", "blezSB", "bgtzSB",
|
|
"addiTSI", "addiu|liTS0I", "sltiTSI", "sltiuTSI",
|
|
"andiTSU", "ori|liTS0U", "xoriTSU", "luiTU",
|
|
map_cop0, map_cop1, false, map_cop1x,
|
|
"beql|beqzlST0B", "bnel|bnezlST0B", "blezlSB", "bgtzlSB",
|
|
false, false, false, false,
|
|
map_special2, false, false, map_special3,
|
|
"lbTSO", "lhTSO", "lwlTSO", "lwTSO",
|
|
"lbuTSO", "lhuTSO", "lwrTSO", false,
|
|
"sbTSO", "shTSO", "swlTSO", "swTSO",
|
|
false, false, "swrTSO", "cacheNSO",
|
|
"llTSO", "lwc1HSO", "lwc2TSO", "prefNSO",
|
|
false, "ldc1HSO", "ldc2TSO", false,
|
|
"scTSO", "swc1HSO", "swc2TSO", false,
|
|
false, "sdc1HSO", "sdc2TSO", false,
|
|
}
|
|
|
|
------------------------------------------------------------------------------
|
|
|
|
local map_gpr = {
|
|
[0] = "r0", "r1", "r2", "r3", "r4", "r5", "r6", "r7",
|
|
"r8", "r9", "r10", "r11", "r12", "r13", "r14", "r15",
|
|
"r16", "r17", "r18", "r19", "r20", "r21", "r22", "r23",
|
|
"r24", "r25", "r26", "r27", "r28", "sp", "r30", "ra",
|
|
}
|
|
|
|
------------------------------------------------------------------------------
|
|
|
|
-- Output a nicely formatted line with an opcode and operands.
|
|
local function putop(ctx, text, operands)
|
|
local pos = ctx.pos
|
|
local extra = ""
|
|
if ctx.rel then
|
|
local sym = ctx.symtab[ctx.rel]
|
|
if sym then extra = "\t->"..sym end
|
|
end
|
|
if ctx.hexdump > 0 then
|
|
ctx.out(format("%08x %s %-7s %s%s\n",
|
|
ctx.addr+pos, tohex(ctx.op), text, concat(operands, ", "), extra))
|
|
else
|
|
ctx.out(format("%08x %-7s %s%s\n",
|
|
ctx.addr+pos, text, concat(operands, ", "), extra))
|
|
end
|
|
ctx.pos = pos + 4
|
|
end
|
|
|
|
-- Fallback for unknown opcodes.
|
|
local function unknown(ctx)
|
|
return putop(ctx, ".long", { "0x"..tohex(ctx.op) })
|
|
end
|
|
|
|
local function get_be(ctx)
|
|
local pos = ctx.pos
|
|
local b0, b1, b2, b3 = byte(ctx.code, pos+1, pos+4)
|
|
return bor(lshift(b0, 24), lshift(b1, 16), lshift(b2, 8), b3)
|
|
end
|
|
|
|
local function get_le(ctx)
|
|
local pos = ctx.pos
|
|
local b0, b1, b2, b3 = byte(ctx.code, pos+1, pos+4)
|
|
return bor(lshift(b3, 24), lshift(b2, 16), lshift(b1, 8), b0)
|
|
end
|
|
|
|
-- Disassemble a single instruction.
|
|
local function disass_ins(ctx)
|
|
local op = ctx:get()
|
|
local operands = {}
|
|
local last = nil
|
|
ctx.op = op
|
|
ctx.rel = nil
|
|
|
|
local opat = map_pri[rshift(op, 26)]
|
|
while type(opat) ~= "string" do
|
|
if not opat then return unknown(ctx) end
|
|
opat = opat[band(rshift(op, opat.shift), opat.mask)] or opat._
|
|
end
|
|
local name, pat = match(opat, "^([a-z0-9_.]*)(.*)")
|
|
local altname, pat2 = match(pat, "|([a-z0-9_.|]*)(.*)")
|
|
if altname then pat = pat2 end
|
|
|
|
for p in gmatch(pat, ".") do
|
|
local x = nil
|
|
if p == "S" then
|
|
x = map_gpr[band(rshift(op, 21), 31)]
|
|
elseif p == "T" then
|
|
x = map_gpr[band(rshift(op, 16), 31)]
|
|
elseif p == "D" then
|
|
x = map_gpr[band(rshift(op, 11), 31)]
|
|
elseif p == "F" then
|
|
x = "f"..band(rshift(op, 6), 31)
|
|
elseif p == "G" then
|
|
x = "f"..band(rshift(op, 11), 31)
|
|
elseif p == "H" then
|
|
x = "f"..band(rshift(op, 16), 31)
|
|
elseif p == "R" then
|
|
x = "f"..band(rshift(op, 21), 31)
|
|
elseif p == "A" then
|
|
x = band(rshift(op, 6), 31)
|
|
elseif p == "M" then
|
|
x = band(rshift(op, 11), 31)
|
|
elseif p == "N" then
|
|
x = band(rshift(op, 16), 31)
|
|
elseif p == "C" then
|
|
x = band(rshift(op, 18), 7)
|
|
if x == 0 then x = nil end
|
|
elseif p == "K" then
|
|
x = band(rshift(op, 11), 31) + 1
|
|
elseif p == "L" then
|
|
x = band(rshift(op, 11), 31) - last + 1
|
|
elseif p == "I" then
|
|
x = arshift(lshift(op, 16), 16)
|
|
elseif p == "U" then
|
|
x = band(op, 0xffff)
|
|
elseif p == "O" then
|
|
local disp = arshift(lshift(op, 16), 16)
|
|
operands[#operands] = format("%d(%s)", disp, last)
|
|
elseif p == "X" then
|
|
local index = map_gpr[band(rshift(op, 16), 31)]
|
|
operands[#operands] = format("%s(%s)", index, last)
|
|
elseif p == "B" then
|
|
x = ctx.addr + ctx.pos + arshift(lshift(op, 16), 16)*4 + 4
|
|
ctx.rel = x
|
|
x = "0x"..tohex(x)
|
|
elseif p == "J" then
|
|
x = band(ctx.addr + ctx.pos, 0xf0000000) + band(op, 0x03ffffff)*4
|
|
ctx.rel = x
|
|
x = "0x"..tohex(x)
|
|
elseif p == "V" then
|
|
x = band(rshift(op, 8), 7)
|
|
if x == 0 then x = nil end
|
|
elseif p == "W" then
|
|
x = band(op, 7)
|
|
if x == 0 then x = nil end
|
|
elseif p == "Y" then
|
|
x = band(rshift(op, 6), 0x000fffff)
|
|
if x == 0 then x = nil end
|
|
elseif p == "Z" then
|
|
x = band(rshift(op, 6), 1023)
|
|
if x == 0 then x = nil end
|
|
elseif p == "0" then
|
|
if last == "r0" or last == 0 then
|
|
local n = #operands
|
|
operands[n] = nil
|
|
last = operands[n-1]
|
|
if altname then
|
|
local a1, a2 = match(altname, "([^|]*)|(.*)")
|
|
if a1 then name, altname = a1, a2
|
|
else name = altname end
|
|
end
|
|
end
|
|
elseif p == "1" then
|
|
if last == "ra" then
|
|
operands[#operands] = nil
|
|
end
|
|
else
|
|
assert(false)
|
|
end
|
|
if x then operands[#operands+1] = x; last = x end
|
|
end
|
|
|
|
return putop(ctx, name, operands)
|
|
end
|
|
|
|
------------------------------------------------------------------------------
|
|
|
|
-- Disassemble a block of code.
|
|
local function disass_block(ctx, ofs, len)
|
|
if not ofs then ofs = 0 end
|
|
local stop = len and ofs+len or #ctx.code
|
|
stop = stop - stop % 4
|
|
ctx.pos = ofs - ofs % 4
|
|
ctx.rel = nil
|
|
while ctx.pos < stop do disass_ins(ctx) end
|
|
end
|
|
|
|
-- Extended API: create a disassembler context. Then call ctx:disass(ofs, len).
|
|
local function create_(code, addr, out)
|
|
local ctx = {}
|
|
ctx.code = code
|
|
ctx.addr = addr or 0
|
|
ctx.out = out or io.write
|
|
ctx.symtab = {}
|
|
ctx.disass = disass_block
|
|
ctx.hexdump = 8
|
|
ctx.get = get_be
|
|
return ctx
|
|
end
|
|
|
|
local function create_el_(code, addr, out)
|
|
local ctx = create_(code, addr, out)
|
|
ctx.get = get_le
|
|
return ctx
|
|
end
|
|
|
|
-- Simple API: disassemble code (a string) at address and output via out.
|
|
local function disass_(code, addr, out)
|
|
create_(code, addr, out):disass()
|
|
end
|
|
|
|
local function disass_el_(code, addr, out)
|
|
create_el_(code, addr, out):disass()
|
|
end
|
|
|
|
-- Return register name for RID.
|
|
local function regname_(r)
|
|
if r < 32 then return map_gpr[r] end
|
|
return "f"..(r-32)
|
|
end
|
|
|
|
-- Public module functions.
|
|
module(...)
|
|
|
|
create = create_
|
|
create_el = create_el_
|
|
disass = disass_
|
|
disass_el = disass_el_
|
|
regname = regname_
|
|
|