def assign_colors(ctx, ast):
if isinstance(ast, Ast_Branch):
for n in ast.nodes:
if isinstance(n, list):
if is_uncond_jump(
n[0]) and n[0].operands[0].type == MIPS_OP_IMM:
nxt = ctx.gph.link_out[n[0].address][BRANCH_NEXT]
pick_color(nxt)
else: # ast
assign_colors(ctx, n)
elif isinstance(ast, Ast_IfGoto) or isinstance(ast, Ast_Goto):
pick_color(ast.addr_jump)
elif isinstance(ast, Ast_Ifelse):
assign_colors(ctx, ast.br_next_jump)
assign_colors(ctx, ast.br_next)
elif isinstance(ast, Ast_Loop):
assign_colors(ctx, ast.branch)
if ast.epilog != None:
assign_colors(ctx, ast.epilog)
elif isinstance(ast, Ast_If_cond):
assign_colors(ctx, ast.br)
def assign_colors(ctx, ast):
if isinstance(ast, Ast_Branch):
for n in ast.nodes:
if isinstance(n, list):
if is_uncond_jump(n[0]) and n[0].operands[0].type == MIPS_OP_IMM:
nxt = ctx.gph.link_out[n[0].address][BRANCH_NEXT]
pick_color(nxt)
else: # ast
assign_colors(ctx, n)
elif isinstance(ast, Ast_IfGoto) or isinstance(ast, Ast_Goto):
pick_color(ast.addr_jump)
elif isinstance(ast, Ast_Ifelse):
assign_colors(ctx, ast.br_next_jump)
assign_colors(ctx, ast.br_next)
elif isinstance(ast, Ast_Loop):
assign_colors(ctx, ast.branch)
elif isinstance(ast, Ast_If_cond):
assign_colors(ctx, ast.br)
def print_inst(self, i, tab=0, prefix=""):
def get_inst_str():
nonlocal i
return "%s %s" % (i.mnemonic, i.op_str)
if isinstance(i, NopInst):
return
if isinstance(i, PseudoInst):
for i2 in i.real_inst_list:
self.print_inst(i2, tab, "# ")
print_label_and_addr(i.real_inst_list[0].address, tab)
print(i.pseudo)
return
if i.address in self.ctx.dis.previous_comments:
for comm in self.ctx.dis.previous_comments[i.address]:
print_tabbed(color_intern_comment("; %s" % comm), tab)
if prefix == "# ":
if self.ctx.comments:
if i.address in self.ctx.labels:
print_label(i.address, tab)
print()
print_comment_no_end(prefix, tab)
print_addr(i.address)
self.print_bytes(i, True)
print_comment(get_inst_str())
return
if i.address in self.ctx.all_fused_inst:
return
if self.is_symbol(i.address):
print_tabbed_no_end("", tab)
self.print_symbol(i.address)
print()
print_label_and_addr(i.address, tab)
self.print_bytes(i)
if is_ret(i):
print(color_retcall(get_inst_str()))
return
if is_call(i):
print_no_end(color_retcall(i.mnemonic) + " ")
modified = self.print_operand(i, 0, hexa=True)
if modified and self.ctx.comments:
print_comment_no_end(" # " + get_inst_str())
print()
return
# Here we can have conditional jump with the option --dump
if is_jump(i):
print_no_end(i.mnemonic + " ")
if i.operands[-1].type != MIPS_OP_IMM:
print_no_end(i.op_str)
if is_uncond_jump(i) and self.ctx.comments and not self.ctx.dump \
and not i.address in self.ctx.dis.jmptables:
print_comment_no_end(" # STOPPED")
print()
return
for num in range(len(i.operands)-1):
self.print_operand(i, num)
print_no_end(", ")
addr = i.operands[0].value.imm
if addr in self.ctx.addr_color:
print_label_or_addr(addr, -1, False)
else:
print_no_end(hex(addr))
print()
return
modified = False
if i.id in LD_CHECK:
self.print_operand(i, 0)
print_no_end(" = (")
print_no_end(color_type(LD_TYPE[i.id]))
print_no_end(") ")
self.print_operand(i, 1)
modified = True
elif i.id in ST_CHECK:
self.print_operand(i, 1)
print_no_end(" = (")
print_no_end(color_type(ST_TYPE[i.id]))
print_no_end(") ")
self.print_operand(i, 0)
modified = True
elif i.id in INST_CHECK:
if i.id == MIPS_INS_LUI:
print_no_end("(load upper) ")
self.print_operand(i, 0)
print_no_end(" = ")
self.print_operand(i, 1)
elif i.id == MIPS_INS_MOVE:
self.print_operand(i, 0)
print_no_end(" = ")
if i.operands[1].value.reg == MIPS_REG_ZERO:
print_no_end("0")
else:
self.print_operand(i, 1)
else:
self.print_operand(i, 0)
if i.operands[0].type == i.operands[1].type == MIPS_OP_REG and \
i.operands[0].value.reg == i.operands[1].value.reg:
print_no_end(" " + inst_symbol(i) + "= ")
else:
print_no_end(" = ")
self.print_operand(i, 1)
print_no_end(" " + inst_symbol(i) + " ")
self.print_operand(i, 2)
modified = True
else:
print_no_end("%s " % i.mnemonic)
if len(i.operands) > 0:
modified = self.print_operand(i, 0)
k = 1
while k < len(i.operands):
print_no_end(", ")
modified |= self.print_operand(i, k)
k += 1
if i.address in self.ctx.dis.inline_comments:
print_no_end(color_intern_comment(" ; "))
print_no_end(color_intern_comment(self.ctx.dis.inline_comments[i.address]))
if modified and self.ctx.comments:
print_comment_no_end(" # " + get_inst_str())
print()
def _sub_asm_inst(self, i, tab=0, prefix=""):
if is_ret(i):
self._retcall(self.get_inst_str(i))
return False
if is_call(i):
self._retcall(i.mnemonic)
self._add(" ")
if self.ctx.sectionsname:
op = i.operands[0]
if op.type == MIPS_OP_IMM:
s = self.binary.get_section(op.value.imm)
if s is not None:
self._add("(")
self._section(s.name)
self._add(") ")
self._operand(i, 0, hexa=True, force_dont_print_data=True)
return False
# Here we can have conditional jump with the option --dump
if is_jump(i):
if len(i.operands) == 0:
self._add(i.mnemonic)
return False
self._add(i.mnemonic + " ")
for num in range(len(i.operands)-1):
self._operand(i, num)
self._add(", ")
if i.operands[-1].type != MIPS_OP_IMM:
self._operand(i, -1, force_dont_print_data=True)
self.inst_end_here()
if is_uncond_jump(i) and self.ctx.comments and not self.ctx.dump \
and not i.address in self.ctx.dis.jmptables:
self._add(" ")
self._comment("# STOPPED")
return False
self._operand(i, -1, hexa=True, force_dont_print_data=True)
return False
modified = False
if i.id in LD_CHECK:
self._operand(i, 0)
self._add(" = (")
self._type(LD_TYPE[i.id])
self._add(") ")
self._operand(i, 1)
modified = True
elif i.id in ST_CHECK:
self._operand(i, 1)
self._add(" = (")
self._type(ST_TYPE[i.id])
self._add(") ")
self._operand(i, 0)
modified = True
elif i.id in INST_CHECK:
if i.id == MIPS_INS_LUI:
self._add("(load upper) ")
self._operand(i, 0)
self._add(" = ")
self._operand(i, 1)
elif i.id == MIPS_INS_MOVE:
self._operand(i, 0)
self._add(" = ")
if i.operands[1].value.reg == MIPS_REG_ZERO:
self._add("0")
else:
self._operand(i, 1)
else:
self._operand(i, 0)
if i.operands[0].type == i.operands[1].type == MIPS_OP_REG and \
i.operands[0].value.reg == i.operands[1].value.reg:
self._add(" " + inst_symbol(i) + "= ")
else:
self._add(" = ")
self._operand(i, 1)
self._add(" " + inst_symbol(i) + " ")
self._operand(i, 2)
modified = True
else:
self._add("%s " % i.mnemonic)
if len(i.operands) > 0:
modified = self._operand(i, 0)
k = 1
while k < len(i.operands):
self._add(", ")
modified |= self._operand(i, k)
k += 1
return modified
def print_inst(self, i, tab=0, prefix=""):
def get_inst_str():
nonlocal i
return "%s %s" % (i.mnemonic, i.op_str)
if isinstance(i, NopInst):
return
if isinstance(i, PseudoInst):
for i2 in i.real_inst_list:
self.print_inst(i2, tab, "# ")
print_label_and_addr(i.real_inst_list[0].address, tab)
print(i.pseudo)
return
if i.address in self.ctx.dis.previous_comments:
for comm in self.ctx.dis.previous_comments[i.address]:
print_tabbed(color_intern_comment("; %s" % comm), tab)
if prefix == "# ":
if self.ctx.comments:
if i.address in self.ctx.labels:
print_label(i.address, tab)
print()
print_comment_no_end(prefix, tab)
print_addr(i.address)
self.print_bytes(i, True)
print_comment(get_inst_str())
return
if i.address in self.ctx.all_fused_inst:
return
if self.is_symbol(i.address):
print_tabbed_no_end("", tab)
self.print_symbol(i.address)
print()
print_label_and_addr(i.address, tab)
self.print_bytes(i)
if is_ret(i):
print(color_retcall(get_inst_str()))
return
if is_call(i):
print_no_end(color_retcall(i.mnemonic) + " ")
modified = self.print_operand(i, 0, hexa=True)
if modified and self.ctx.comments:
print_comment_no_end(" # " + get_inst_str())
print()
return
# Here we can have conditional jump with the option --dump
if is_jump(i):
print_no_end(i.mnemonic + " ")
if i.operands[-1].type != MIPS_OP_IMM:
print_no_end(i.op_str)
if is_uncond_jump(i) and self.ctx.comments and not self.ctx.dump \
and not i.address in self.ctx.dis.jmptables:
print_comment_no_end(" # STOPPED")
print()
return
for num in range(len(i.operands) - 1):
self.print_operand(i, num)
print_no_end(", ")
addr = i.operands[0].value.imm
if addr in self.ctx.addr_color:
print_label_or_addr(addr, -1, False)
else:
print_no_end(hex(addr))
print()
return
modified = False
if i.id in LD_CHECK:
self.print_operand(i, 0)
print_no_end(" = (")
print_no_end(color_type(LD_TYPE[i.id]))
print_no_end(") ")
self.print_operand(i, 1)
modified = True
elif i.id in ST_CHECK:
self.print_operand(i, 1)
print_no_end(" = (")
print_no_end(color_type(ST_TYPE[i.id]))
print_no_end(") ")
self.print_operand(i, 0)
modified = True
elif i.id in INST_CHECK:
if i.id == MIPS_INS_LUI:
print_no_end("(load upper) ")
self.print_operand(i, 0)
print_no_end(" = ")
self.print_operand(i, 1)
elif i.id == MIPS_INS_MOVE:
self.print_operand(i, 0)
print_no_end(" = ")
if i.operands[1].value.reg == MIPS_REG_ZERO:
print_no_end("0")
else:
self.print_operand(i, 1)
else:
self.print_operand(i, 0)
if i.operands[0].type == i.operands[1].type == MIPS_OP_REG and \
i.operands[0].value.reg == i.operands[1].value.reg:
print_no_end(" " + inst_symbol(i) + "= ")
else:
print_no_end(" = ")
self.print_operand(i, 1)
print_no_end(" " + inst_symbol(i) + " ")
self.print_operand(i, 2)
modified = True
else:
print_no_end("%s " % i.mnemonic)
if len(i.operands) > 0:
modified = self.print_operand(i, 0)
k = 1
while k < len(i.operands):
print_no_end(", ")
modified |= self.print_operand(i, k)
k += 1
if i.address in self.ctx.dis.inline_comments:
print_no_end(color_intern_comment(" ; "))
print_no_end(
color_intern_comment(self.ctx.dis.inline_comments[i.address]))
if modified and self.ctx.comments:
print_comment_no_end(" # " + get_inst_str())
print()
def _sub_asm_inst(self, i, tab=0, prefix=""):
if is_ret(i):
self._retcall(self.get_inst_str(i))
return False
if is_call(i):
self._retcall(i.mnemonic)
self._add(" ")
if self.gctx.sectionsname:
op = i.operands[0]
if op.type == MIPS_OP_IMM:
s = self._binary.get_section(op.value.imm)
if s is not None:
self._add("(")
self._section(s.name)
self._add(") ")
self._operand(i, 0, hexa=True, force_dont_print_data=True)
return False
# Here we can have conditional jump with the option --dump
if is_jump(i):
if len(i.operands) == 0:
self._add(i.mnemonic)
return False
self._add(i.mnemonic + " ")
for num in range(len(i.operands) - 1):
self._operand(i, num)
self._add(", ")
if i.operands[-1].type != MIPS_OP_IMM:
self._operand(i, -1, force_dont_print_data=True)
self.inst_end_here()
if is_uncond_jump(i) and not self.ctx.is_dump \
and not i.address in self._dis.jmptables:
self._add(" ")
self._comment("# STOPPED")
return False
self._operand(i, -1, hexa=True, force_dont_print_data=True)
return False
modified = False
if self.gctx.capstone_string:
if i.id in LD_CHECK:
self._operand(i, 0)
self._add(" = (")
self._type(LD_TYPE[i.id])
self._add(") ")
self._operand(i, 1)
modified = True
elif i.id in ST_CHECK:
self._operand(i, 1)
self._add(" = (")
self._type(ST_TYPE[i.id])
self._add(") ")
self._operand(i, 0)
modified = True
elif i.id in INST_CHECK:
if i.id == MIPS_INS_LUI:
self._add("(load upper) ")
self._operand(i, 0)
self._add(" = ")
self._operand(i, 1)
elif i.id == MIPS_INS_MOVE:
self._operand(i, 0)
self._add(" = ")
if i.operands[1].value.reg == MIPS_REG_ZERO:
self._add("0")
else:
self._operand(i, 1)
else:
self._operand(i, 0)
if i.operands[0].type == i.operands[1].type == MIPS_OP_REG and \
i.operands[0].value.reg == i.operands[1].value.reg:
self._add(" " + inst_symbol(i) + "= ")
else:
self._add(" = ")
self._operand(i, 1)
self._add(" " + inst_symbol(i) + " ")
self._operand(i, 2)
modified = True
if not modified:
self._add("%s " % i.mnemonic)
if len(i.operands) > 0:
modified = self._operand(i, 0)
k = 1
while k < len(i.operands):
self._add(", ")
modified |= self._operand(i, k)
k += 1
return modified
def _sub_asm_inst(self, i, tab=0, prefix=""):
self._label_and_address(i.address, tab)
self._bytes(i)
if is_ret(i):
self._retcall(self.get_inst_str(i))
return False
if is_call(i):
self._retcall(i.mnemonic)
self._add(" ")
self._operand(i, 0, hexa=True, force_dont_print_data=True)
return False
# Here we can have conditional jump with the option --dump
if is_jump(i):
if len(i.operands) == 0:
self._add(i.mnemonic)
return False
self._add(i.mnemonic + " ")
for num in range(len(i.operands) - 1):
self._operand(i, num)
self._add(", ")
if i.operands[-1].type != MIPS_OP_IMM:
self._operand(i, -1, force_dont_print_data=True)
self.inst_end_here()
if is_uncond_jump(i) and self.ctx.comments and not self.ctx.dump \
and not i.address in self.ctx.dis.jmptables:
self._add(" ")
self._comment("# STOPPED")
return False
addr = i.operands[-1].value.imm
if addr in self.ctx.addr_color:
self._label_or_address(addr, -1, False)
else:
self._add(hex(addr))
return False
modified = False
if i.id in LD_CHECK:
self._operand(i, 0)
self._add(" = (")
self._type(LD_TYPE[i.id])
self._add(") ")
self._operand(i, 1)
modified = True
elif i.id in ST_CHECK:
self._operand(i, 1)
self._add(" = (")
self._type(ST_TYPE[i.id])
self._add(") ")
self._operand(i, 0)
modified = True
elif i.id in INST_CHECK:
if i.id == MIPS_INS_LUI:
self._add("(load upper) ")
self._operand(i, 0)
self._add(" = ")
self._operand(i, 1)
elif i.id == MIPS_INS_MOVE:
self._operand(i, 0)
self._add(" = ")
if i.operands[1].value.reg == MIPS_REG_ZERO:
self._add("0")
else:
self._operand(i, 1)
else:
self._operand(i, 0)
if i.operands[0].type == i.operands[1].type == MIPS_OP_REG and \
i.operands[0].value.reg == i.operands[1].value.reg:
self._add(" " + inst_symbol(i) + "= ")
else:
self._add(" = ")
self._operand(i, 1)
self._add(" " + inst_symbol(i) + " ")
self._operand(i, 2)
modified = True
else:
self._add("%s " % i.mnemonic)
if len(i.operands) > 0:
modified = self._operand(i, 0)
k = 1
while k < len(i.operands):
self._add(", ")
modified |= self._operand(i, k)
k += 1
return modified
def print_inst(self, i, tab=0, prefix=""):
def get_inst_str():
nonlocal i
return "%s %s" % (i.mnemonic, i.op_str)
if prefix == "# ":
if self.ctx.comments:
print_comment_no_end(prefix, tab)
print_no_end(color_addr(i.address))
print_comment(get_inst_str())
return
if i.address in self.ctx.all_fused_inst:
return
if i.address != self.ctx.addr and \
i.address in self.ctx.dis.binary.reverse_symbols:
print_tabbed_no_end("", tab)
self.print_symbol(i.address)
print()
print_tabbed_no_end(color_addr(i.address), tab)
if is_ret(i):
print(color_retcall(get_inst_str()))
return
if is_call(i):
print_no_end(color_retcall(i.mnemonic) + " ")
self.print_operand(i, 0, hexa=True)
print()
return
# Here we can have conditional jump with the option --dump
if is_jump(i):
print_no_end(i.mnemonic + " ")
if i.operands[-1].type != MIPS_OP_IMM:
print_no_end(i.op_str)
if is_uncond_jump(i) and self.ctx.comments:
print_comment_no_end(" # STOPPED")
print()
return
for num in range(len(i.operands)-1):
self.print_operand(i, num)
print_no_end(", ")
addr = i.operands[-1].value.imm
if addr in self.ctx.addr_color:
print(color(hex(addr), self.ctx.addr_color[addr]))
else:
print(hex(addr))
return
modified = False
if i.id in LD_CHECK:
self.print_operand(i, 0)
print_no_end(" = (")
print_no_end(color_type(LD_TYPE[i.id]))
print_no_end(") ")
self.print_operand(i, 1)
modified = True
elif i.id in ST_CHECK:
self.print_operand(i, 1)
print_no_end(" = (")
print_no_end(color_type(ST_TYPE[i.id]))
print_no_end(") ")
self.print_operand(i, 0)
modified = True
elif i.id in INST_CHECK:
if i.id == MIPS_INS_LUI:
print_no_end("(load upper) ")
self.print_operand(i, 0)
print_no_end(" = ")
self.print_operand(i, 1)
elif i.id == MIPS_INS_MOVE:
self.print_operand(i, 0)
print_no_end(" = ")
if i.operands[1].value.reg == MIPS_REG_ZERO:
print_no_end("0")
else:
self.print_operand(i, 1)
else:
self.print_operand(i, 0)
if i.operands[0].type == i.operands[1].type == MIPS_OP_REG and \
i.operands[0].value.reg == i.operands[1].value.reg:
print_no_end(" " + inst_symbol(i) + "= ")
else:
print_no_end(" = ")
self.print_operand(i, 1)
print_no_end(" " + inst_symbol(i) + " ")
self.print_operand(i, 2)
modified = True
else:
print_no_end("%s " % i.mnemonic)
if len(i.operands) > 0:
modified = self.print_operand(i, 0)
k = 1
while k < len(i.operands):
print_no_end(", ")
modified |= self.print_operand(i, k)
k += 1
if modified and self.ctx.comments:
print_comment_no_end(" # " + get_inst_str())
print()