277 lines
8.1 KiB
Python
277 lines
8.1 KiB
Python
|
#!/usr/bin/env python2
|
||
|
#
|
||
|
# Utility to dump bytecode into a human readable form.
|
||
|
#
|
||
|
|
||
|
import os
|
||
|
import sys
|
||
|
import optparse
|
||
|
import struct
|
||
|
import yaml
|
||
|
|
||
|
script_path = sys.path[0] # http://stackoverflow.com/questions/4934806/how-can-i-find-scripts-directory-with-python
|
||
|
|
||
|
ops = None
|
||
|
|
||
|
def decode_string(buf, off):
|
||
|
strlen, = struct.unpack('>L', buf[off:off + 4])
|
||
|
off += 4
|
||
|
strdata = buf[off:off + strlen]
|
||
|
off += strlen
|
||
|
|
||
|
return off, strdata
|
||
|
|
||
|
def sanitize_string(val):
|
||
|
# Don't try to UTF-8 decode, just escape non-printable ASCII.
|
||
|
def f(c):
|
||
|
if ord(c) < 0x20 or ord(c) > 0x7e or c in '\'"':
|
||
|
return '\\x%02x' % ord(c)
|
||
|
else:
|
||
|
return c
|
||
|
|
||
|
return "'" + ''.join(map(f, val)) + "'"
|
||
|
|
||
|
def decode_sanitize_string(buf, off):
|
||
|
off, val = decode_string(buf, off)
|
||
|
return off, sanitize_string(val)
|
||
|
|
||
|
def dump_ins(ins, x):
|
||
|
global ops
|
||
|
|
||
|
if ops is None:
|
||
|
return ''
|
||
|
|
||
|
pc = x / 4
|
||
|
args = []
|
||
|
|
||
|
op = ops[ins & 0xff]
|
||
|
comments = []
|
||
|
|
||
|
if 'args' in op:
|
||
|
for j in xrange(len(op['args'])):
|
||
|
A = (ins >> 8) & 0xff
|
||
|
B = (ins >> 16) & 0xff
|
||
|
C = (ins >> 24) & 0xff
|
||
|
BC = (ins >> 16) & 0xffff
|
||
|
ABC = (ins >> 8) & 0xffffff
|
||
|
|
||
|
Bconst = ins & 0x1
|
||
|
Cconst = ins & 0x2
|
||
|
|
||
|
if op['args'][j] == 'A_R':
|
||
|
args.append('r' + str(A))
|
||
|
|
||
|
elif op['args'][j] == 'A_RI':
|
||
|
args.append('r' + str(A) + '(indirect)')
|
||
|
|
||
|
elif op['args'][j] == 'A_C':
|
||
|
args.append('c' + str(A))
|
||
|
|
||
|
elif op['args'][j] == 'A_H':
|
||
|
args.append(hex(A))
|
||
|
|
||
|
elif op['args'][j] == 'A_I':
|
||
|
args.append(str(A))
|
||
|
|
||
|
elif op['args'][j] == 'A_B':
|
||
|
args.append('true' if A else 'false')
|
||
|
|
||
|
elif op['args'][j] == 'B_RC':
|
||
|
args.append('c' if Bconst else 'r' + str(B))
|
||
|
|
||
|
elif op['args'][j] == 'B_R':
|
||
|
args.append('r' + str(B))
|
||
|
|
||
|
elif op['args'][j] == 'B_RI':
|
||
|
args.append('r' + str(B) + '(indirect)')
|
||
|
|
||
|
elif op['args'][j] == 'B_C':
|
||
|
args.append('c' + str(B))
|
||
|
|
||
|
elif op['args'][j] == 'B_H':
|
||
|
args.append(hex(B))
|
||
|
|
||
|
elif op['args'][j] == 'B_I':
|
||
|
args.append(str(B))
|
||
|
|
||
|
elif op['args'][j] == 'C_RC':
|
||
|
args.append('c' if Cconst else 'r' + str(C))
|
||
|
|
||
|
elif op['args'][j] == 'C_R':
|
||
|
args.append('r' + str(C))
|
||
|
|
||
|
elif op['args'][j] == 'C_RI':
|
||
|
args.append('r' + str(C) + '(indirect)')
|
||
|
|
||
|
elif op['args'][j] == 'C_C':
|
||
|
args.append('c' + str(C))
|
||
|
|
||
|
elif op['args'][j] == 'C_H':
|
||
|
args.append(hex(C))
|
||
|
|
||
|
elif op['args'][j] == 'C_I':
|
||
|
args.append(str(C))
|
||
|
|
||
|
elif op['args'][j] == 'BC_R':
|
||
|
args.append('r' + str(BC))
|
||
|
|
||
|
elif op['args'][j] == 'BC_C':
|
||
|
args.append('c' + str(BC))
|
||
|
|
||
|
elif op['args'][j] == 'BC_H':
|
||
|
args.append(hex(BC))
|
||
|
|
||
|
elif op['args'][j] == 'BC_I':
|
||
|
args.append(str(BC))
|
||
|
|
||
|
elif op['args'][j] == 'ABC_H':
|
||
|
args.append(hex(ABC))
|
||
|
|
||
|
elif op['args'][j] == 'ABC_I':
|
||
|
args.append(str(ABC))
|
||
|
|
||
|
elif op['args'][j] == 'BC_LDINT':
|
||
|
args.append(hex(BC - (1 << 15)))
|
||
|
|
||
|
elif op['args'][j] == 'BC_LDINTX':
|
||
|
args.append(hex(BC))
|
||
|
elif op['args'][j] == 'ABC_JUMP':
|
||
|
pc_add = ABC - (1 << 23) + 1
|
||
|
pc_dst = pc + pc_add
|
||
|
args.append(str(pc_dst) + ' (' + ('+' if pc_add >= 0 else '') + str(pc_add) + ')')
|
||
|
else:
|
||
|
args.append('?')
|
||
|
|
||
|
if 'flags' in op:
|
||
|
for f in op['flags']:
|
||
|
if ins & f['mask']:
|
||
|
comments.append(f['name'])
|
||
|
|
||
|
if len(args) > 0:
|
||
|
res = '%-12s %s' % (op['name'], ', '.join(args))
|
||
|
else:
|
||
|
res = op['name']
|
||
|
if len(comments) > 0:
|
||
|
res = '%-44s ; %s' % (res, ', '.join(comments))
|
||
|
|
||
|
return res
|
||
|
|
||
|
def dump_function(buf, off, ind):
|
||
|
count_inst, count_const, count_funcs = struct.unpack('>LLL', buf[off:off + 12])
|
||
|
off += 12
|
||
|
print('%sInstructions: %d' % (ind, count_inst))
|
||
|
print('%sConstants: %d' % (ind, count_const))
|
||
|
print('%sInner functions: %d' % (ind, count_funcs))
|
||
|
|
||
|
# Line numbers present, assuming debugger support; otherwise 0.
|
||
|
nregs, nargs, start_line, end_line = struct.unpack('>HHLL', buf[off:off + 12])
|
||
|
off += 12
|
||
|
print('%sNregs: %d' % (ind, nregs))
|
||
|
print('%sNargs: %d' % (ind, nargs))
|
||
|
print('%sStart line number: %d' % (ind, start_line))
|
||
|
print('%sEnd line number: %d' % (ind, end_line))
|
||
|
|
||
|
compfunc_flags, = struct.unpack('>L', buf[off:off + 4])
|
||
|
off += 4
|
||
|
print('%sduk_hcompiledfunction flags: 0x%08x' % (ind, compfunc_flags))
|
||
|
|
||
|
for i in xrange(count_inst):
|
||
|
ins, = struct.unpack('>L', buf[off:off + 4])
|
||
|
off += 4
|
||
|
code = dump_ins(ins, i)
|
||
|
print('%s %06d: %08lx %s' % (ind, i, ins, code))
|
||
|
|
||
|
print('%sConstants:' % ind)
|
||
|
for i in xrange(count_const):
|
||
|
const_type, = struct.unpack('B', buf[off:off + 1])
|
||
|
off += 1
|
||
|
|
||
|
if const_type == 0x00:
|
||
|
off, strdata = decode_sanitize_string(buf, off)
|
||
|
print('%s %06d: %s' % (ind, i, strdata))
|
||
|
elif const_type == 0x01:
|
||
|
num, = struct.unpack('>d', buf[off:off + 8])
|
||
|
off += 8
|
||
|
print('%s %06d: %f' % (ind, i, num))
|
||
|
else:
|
||
|
raise Exception('invalid constant type: %d' % const_type)
|
||
|
|
||
|
for i in xrange(count_funcs):
|
||
|
print('%sInner function %d:' % (ind, i))
|
||
|
off = dump_function(buf, off, ind + ' ')
|
||
|
|
||
|
val, = struct.unpack('>L', buf[off:off + 4])
|
||
|
off += 4
|
||
|
print('%s.length: %d' % (ind, val))
|
||
|
off, val = decode_sanitize_string(buf, off)
|
||
|
print('%s.name: %s' % (ind, val))
|
||
|
off, val = decode_sanitize_string(buf, off)
|
||
|
print('%s.fileName: %s' % (ind, val))
|
||
|
off, val = decode_string(buf, off) # actually a buffer
|
||
|
print('%s._Pc2line: %s' % (ind, val.encode('hex')))
|
||
|
|
||
|
while True:
|
||
|
off, name = decode_string(buf, off)
|
||
|
if name == '':
|
||
|
break
|
||
|
name = sanitize_string(name)
|
||
|
val, = struct.unpack('>L', buf[off:off + 4])
|
||
|
off += 4
|
||
|
print('%s_Varmap[%s] = %d' % (ind, name, val))
|
||
|
|
||
|
num_formals, = struct.unpack('>L', buf[off:off + 4])
|
||
|
off += 4
|
||
|
if num_formals != 0xffffffff:
|
||
|
print('%s_Formals: %d formal arguments' % (ind, num_formals))
|
||
|
for idx in xrange(num_formals):
|
||
|
off, name = decode_string(buf, off)
|
||
|
name = sanitize_string(name)
|
||
|
print('%s_Formals[%d] = %s' % (ind, idx, name))
|
||
|
else:
|
||
|
print('%s_Formals: absent' % ind)
|
||
|
|
||
|
return off
|
||
|
|
||
|
def dump_bytecode(buf, off, ind):
|
||
|
sig, = struct.unpack('B', buf[off:off + 1])
|
||
|
print('%sSignature byte: 0x%02x' % (ind, sig))
|
||
|
off += 1
|
||
|
if sig == 0xff:
|
||
|
raise Exception('pre-Duktape 2.2 0xFF signature byte (signature byte is 0xBF since Duktape 2.2)')
|
||
|
if sig != 0xbf:
|
||
|
raise Exception('invalid signature byte: %d' % sig)
|
||
|
|
||
|
off = dump_function(buf, off, ind + ' ')
|
||
|
|
||
|
return off
|
||
|
|
||
|
def main():
|
||
|
global ops
|
||
|
|
||
|
for ops_path in [ '.',
|
||
|
os.path.join('..', 'debugger'),
|
||
|
script_path,
|
||
|
os.path.join(script_path, '..', 'debugger') ]:
|
||
|
fn = os.path.join(ops_path, 'duk_opcodes.yaml')
|
||
|
if os.path.isfile(fn):
|
||
|
with open(fn) as f:
|
||
|
ops = yaml.load(f)['opcodes']
|
||
|
|
||
|
if ops is None:
|
||
|
print('WARN: duk_opcodes.yaml not found, unable do dump opcodes!')
|
||
|
|
||
|
parser = optparse.OptionParser()
|
||
|
parser.add_option('--hex-decode', dest='hex_decode', default=False, action='store_true',
|
||
|
help='Input file is ASCII hex encoded, decode before dump')
|
||
|
(opts, args) = parser.parse_args()
|
||
|
|
||
|
with open(args[0], 'rb') as f:
|
||
|
d = f.read()
|
||
|
if opts.hex_decode:
|
||
|
d = d.strip()
|
||
|
d = d.decode('hex')
|
||
|
dump_bytecode(d, 0, '')
|
||
|
|
||
|
if __name__ == '__main__':
|
||
|
main()
|