def print(self): for l in self.token_lines: for (string, col, is_bold) in l: if self.gctx.color: if col != 0: string = color(string, col) if is_bold: string = bold(string) print_no_end(string) print()
def print(self): for l in self.token_lines: for (string, col, is_bold) in l: if self.ctx.color: if col != 0: string = color(string, col) if is_bold: string = bold(string) print_no_end(string) print()
def __exec_help(self, args): for name in self.COMMANDS_ALPHA: cmd = self.COMMANDS[name] if cmd.callback_exec is not None: self.rl.print(color(name, 2)) self.rl.print(" ") for i, line in enumerate(cmd.desc): if i > 0: self.rl.print(self.TAB) self.rl.print(line) self.rl.print("\n")
def __thread(self): while True: # While true read incomming messages msg = read(self.sock) try: # and try to decode it as a normal # message. # By Decrypting it msg = decode(aes(self.key).decrypt(msg)) # And formating it with colors if msg.split(" ").pop() == "Disconnected": print(color(msg, "yellow")) else: # And charecters print(color(msg.split(":")[0] + COLOR_SEP, "blue") + " " + color(':'.join(msg.split(":")[1:]), "green")) # if and exception occurs except Exception as e: # decode message msg = msg.decode("utf8") # and print it print(color(msg, "red"))
def main(self): try: # key exhange write(self.keys.pub, self.sock) self.key = self.keys.decrypt(read(self.sock)).decode("utf8") write(aes(self.key).encrypt(encode(self.name)), self.sock) # start thread threading.Thread(target=self.__thread).start() # and while True while True: # send messages self.SendMsg(self.GetMsg()) except Exception as e: # if any error occurs # print it. it's probaly from # the self.__thread print(color(e, "red"))
def __print_inst(self, i, tab=0, prefix=""): def get_inst_str(): nonlocal i return "%s %s" % (i.mnemonic, i.op_str) print_tabbed_no_end(color_addr(i.address), tab) if is_ret(i): print_no_end(color_retcall(get_inst_str())) return if is_call(i): print_no_end(color_retcall(i.mnemonic) + " ") self.print_operand(i, 0, hexa=True) return # Here we can have conditional jump with the option --dump if is_jump(i): if i.operands[0].type != X86_OP_IMM: print_no_end(i.mnemonic + " ") self.print_operand(i, 0) if is_uncond_jump(i) and self.ctx.comments: print_comment_no_end(" # STOPPED") return try: addr = i.operands[0].value.imm print_no_end(i.mnemonic + " " + color(hex(addr), self.ctx.addr_color[addr])) except KeyError: print_no_end(i.mnemonic + " " + hex(addr)) return modified = False if i.id in INST_CHECK: if (i.id == X86_INS_OR and i.operands[1].type == X86_OP_IMM and i.operands[1].value.imm == -1): self.print_operand(i, 0) print_no_end(" = -1") elif (i.id == X86_INS_AND and i.operands[1].type == X86_OP_IMM and i.operands[1].value.imm == 0): self.print_operand(i, 0) print_no_end(" = 0") elif (all(op.type == X86_OP_REG for op in i.operands) and len(set(op.value.reg for op in i.operands)) == 1 and i.id == X86_INS_XOR): self.print_operand(i, 0) print_no_end(" = 0") elif i.id == X86_INS_INC or i.id == X86_INS_DEC: self.print_operand(i, 0) print_no_end(inst_symbol(i)) elif i.id == X86_INS_LEA: self.print_operand(i, 0) print_no_end(" = &(") self.print_operand(i, 1) print_no_end(")") elif i.id == X86_INS_IMUL: if len(i.operands) == 3: self.print_operand(i, 0) print_no_end(" = ") self.print_operand(i, 1) print_no_end(" " + inst_symbol(i).rstrip('=') + " ") self.print_operand(i, 2) elif len(i.operands) == 2: self.print_operand(i, 0) print_no_end(" " + inst_symbol(i) + " ") self.print_operand(i, 1) elif len(i.operands) == 1: sz = i.operands[0].size if sz == 1: print_no_end("ax = al * ") elif sz == 2: print_no_end("dx:ax = ax * ") elif sz == 4: print_no_end("edx:eax = eax * ") elif sz == 8: print_no_end("rdx:rax = rax * ") self.print_operand(i, 0) else: self.print_operand(i, 0) print_no_end(" " + inst_symbol(i) + " ") self.print_operand(i, 1) modified = True elif i.id == X86_INS_CDQE: print_no_end("rax = eax") modified = True elif i.id == X86_INS_IDIV: print_no_end('eax = edx:eax / ') self.print_operand(i, 0) print_no_end('; edx = edx:eax % ') self.print_operand(i, 0) modified = True elif i.id == X86_INS_MUL: lut = {1: ("al", "ax"), 2: ("ax", "dx:ax"), 4: ("eax", "edx:eax"), 8: ("rax", "rdx:rax")} src, dst = lut[i.operands[0].size] print_no_end('{0} = {1} * '.format(dst, src)) self.print_operand(i, 0) modified = True elif i.id == X86_INS_NOT: self.print_operand(i, 0) print_no_end(' ^= -1') modified = True elif i.id == X86_INS_SCASB and i.prefix[0] == X86_PREFIX_REPNE: print_no_end('while (') self.print_operand(i, 1) print_no_end(' != ') self.print_operand(i, 0) print_no_end(') { ') self.print_operand(i, 1, show_deref=False) print_no_end('++; cx--; } ') self.print_operand(i, 1, show_deref=False) print_no_end('++; cx--;') 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 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: self.print_inst(i2, tab, "# ") print_tabbed_no_end(color_addr(i.real_inst[0].address), tab) print(i.pseudo) return if prefix == "# ": if self.ctx.comments: print_comment_no_end(prefix, tab) print_no_end(color_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_tabbed_no_end(color_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) + " ") 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 and not self.ctx.dump: 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()
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): if i.operands[0].type != ARM_OP_IMM: print_no_end(i.mnemonic + " ") print_no_end(i.op_str) if is_uncond_jump(i) and self.ctx.comments: print_comment_no_end(" # STOPPED") print() return try: addr = i.operands[0].value.imm print(i.mnemonic + " " + color(hex(addr), self.ctx.addr_color[addr])) except KeyError: print(i.mnemonic + " " + hex(addr)) return modified = False if i.id in LDR_CHECK: self.print_operand(i, 0) print_no_end(" = (") print_no_end(color_type(LDR_TYPE[i.id])) print_no_end(") ") self.print_operand(i, 1) modified = True elif i.id in STR_CHECK: self.print_operand(i, 1) print_no_end(" = (") print_no_end(color_type(STR_TYPE[i.id])) print_no_end(") ") self.print_operand(i, 0) modified = True elif i.id in INST_CHECK: self.print_operand(i, 0) if i.id == ARM_INS_CMP: print_no_end(" " + inst_symbol(i) + " ") self.print_operand(i, 1) else: print_no_end(" = ") self.print_operand(i, 1) if len(i.operands) == 3: 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.update_flags and i.id != ARM_INS_CMP and i.id != ARM_INS_TST: print_no_end(color_type(" (FLAGS)")) if modified and self.ctx.comments: print_comment_no_end(" # " + get_inst_str()) print()
def print_addr(addr): c = addr_color.get(addr) print(hex(addr) if c is None else color(hex(addr), c))
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)) 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_tabbed_no_end(color_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) + " ") self.print_operand(i, 0, hexa=True) print() return # Here we can have conditional jump with the option --dump if is_jump(i): if i.operands[0].type != ARM_OP_IMM: print_no_end(i.mnemonic + " ") print_no_end(i.op_str) if is_uncond_jump( i) and self.ctx.comments and not self.ctx.dump: print_comment_no_end(" # STOPPED") print() return try: addr = i.operands[0].value.imm print(i.mnemonic + " " + color(hex(addr), self.ctx.addr_color[addr])) except KeyError: print(i.mnemonic + " " + hex(addr)) return modified = False if i.id in LDR_CHECK: self.print_operand(i, 0) print_no_end(" = (") print_no_end(color_type(LDR_TYPE[i.id])) print_no_end(") ") self.print_operand(i, 1) modified = True elif i.id in STR_CHECK: self.print_operand(i, 1) print_no_end(" = (") print_no_end(color_type(STR_TYPE[i.id])) print_no_end(") ") self.print_operand(i, 0) modified = True elif i.id in INST_CHECK: self.print_operand(i, 0) if i.id == ARM_INS_CMP: print_no_end(" " + inst_symbol(i) + " ") self.print_operand(i, 1) else: print_no_end(" = ") self.print_operand(i, 1) if len(i.operands) == 3: 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.update_flags and i.id != ARM_INS_CMP and i.id != ARM_INS_TST: print_no_end(color_type(" (FLAGS)")) if modified and self.ctx.comments: print_comment_no_end(" # " + get_inst_str()) print()
def __print_inst(self, i, tab=0, prefix=""): def get_inst_str(): nonlocal i return "%s %s" % (i.mnemonic, i.op_str) def print_rep_begin(): nonlocal tab if i.prefix[0] in REP_PREFIX: print_tabbed_no_end(color_keyword("while"), tab) # TODO: for 16 and 32 bits print_no_end(" (!rcx)") print(") {") tab += 1 def print_rep_end(): nonlocal tab if i.prefix[0] in REP_PREFIX: print() print_tabbed_no_end(color_addr(i.address), tab) print("rcx--") if i.prefix[0] == X86_PREFIX_REPNE: print_tabbed_no_end(color_keyword("if"), tab) print_no_end(" (!Z) ") print(color_keyword("break")) tab -= 1 print_tabbed_no_end("}", tab) print_rep_begin() print_tabbed_no_end(color_addr(i.address), tab) self.print_bytes(i) if is_ret(i): print_no_end(color_retcall(get_inst_str())) return if is_call(i): print_no_end(color_retcall(i.mnemonic) + " ") self.print_operand(i, 0, hexa=True) return # Here we can have conditional jump with the option --dump if is_jump(i): if i.operands[0].type != X86_OP_IMM: print_no_end(i.mnemonic + " ") self.print_operand(i, 0) if is_uncond_jump(i) and self.ctx.comments and not self.ctx.dump: print_comment_no_end(" # STOPPED") return try: addr = i.operands[0].value.imm print_no_end(i.mnemonic + " " + color(hex(addr), self.ctx.addr_color[addr])) except KeyError: print_no_end(i.mnemonic + " " + hex(addr)) return modified = False if i.id in INST_CHECK: if (i.id == X86_INS_OR and i.operands[1].type == X86_OP_IMM and i.operands[1].value.imm == -1): self.print_operand(i, 0) print_no_end(" = -1") elif (i.id == X86_INS_AND and i.operands[1].type == X86_OP_IMM and i.operands[1].value.imm == 0): self.print_operand(i, 0) print_no_end(" = 0") elif (all(op.type == X86_OP_REG for op in i.operands) and len(set(op.value.reg for op in i.operands)) == 1 and i.id == X86_INS_XOR): self.print_operand(i, 0) print_no_end(" = 0") elif i.id == X86_INS_INC or i.id == X86_INS_DEC: self.print_operand(i, 0) print_no_end(inst_symbol(i)) elif i.id == X86_INS_LEA: self.print_operand(i, 0) print_no_end(" = &(") self.print_operand(i, 1) print_no_end(")") elif i.id == X86_INS_MOVZX: self.print_operand(i, 0) print_no_end(" = (zero ext) ") self.print_operand(i, 1) elif i.id == X86_INS_IMUL: if len(i.operands) == 3: self.print_operand(i, 0) print_no_end(" = ") self.print_operand(i, 1) print_no_end(" " + inst_symbol(i).rstrip('=') + " ") self.print_operand(i, 2) elif len(i.operands) == 2: self.print_operand(i, 0) print_no_end(" " + inst_symbol(i) + " ") self.print_operand(i, 1) elif len(i.operands) == 1: sz = i.operands[0].size if sz == 1: print_no_end("ax = al * ") elif sz == 2: print_no_end("dx:ax = ax * ") elif sz == 4: print_no_end("edx:eax = eax * ") elif sz == 8: print_no_end("rdx:rax = rax * ") self.print_operand(i, 0) else: self.print_operand(i, 0) print_no_end(" " + inst_symbol(i) + " ") self.print_operand(i, 1) modified = True elif i.id == X86_INS_CDQE: print_no_end("rax = eax") modified = True elif i.id == X86_INS_IDIV: print_no_end('eax = edx:eax / ') self.print_operand(i, 0) print_no_end('; edx = edx:eax % ') self.print_operand(i, 0) modified = True elif i.id == X86_INS_MUL: lut = {1: ("al", "ax"), 2: ("ax", "dx:ax"), 4: ("eax", "edx:eax"), 8: ("rax", "rdx:rax")} src, dst = lut[i.operands[0].size] print_no_end('{0} = {1} * '.format(dst, src)) self.print_operand(i, 0) modified = True elif i.id == X86_INS_NOT: self.print_operand(i, 0) print_no_end(' ^= -1') modified = True elif i.id in INST_SCAS: self.print_operand(i, 0) print_no_end(" cmp ") self.print_operand(i, 1) print() print_tabbed_no_end(color_addr(i.address), tab) self.print_operand(i, 1, show_deref=False) print_no_end(" += D") modified = True elif i.id in INST_STOS: self.print_operand(i, 0) print_no_end(" = ") self.print_operand(i, 1) print() print_tabbed_no_end(color_addr(i.address), tab) self.print_operand(i, 0, show_deref=False) print_no_end(" += D") modified = True elif i.id in INST_LODS: self.print_operand(i, 0) print_no_end(" = ") self.print_operand(i, 1) print() print_tabbed_no_end(color_addr(i.address), tab) self.print_operand(i, 1, show_deref=False) print_no_end(" += D") modified = True elif i.id in INST_CMPS: self.print_operand(i, 0) print_no_end(" cmp ") self.print_operand(i, 1) print() print_tabbed_no_end(color_addr(i.address), tab) self.print_operand(i, 0, show_deref=False) print(" += D") print_tabbed_no_end(color_addr(i.address), tab) self.print_operand(i, 1, show_deref=False) print_no_end("' += D") modified = True elif i.id in INST_MOVS: self.print_operand(i, 0) print_no_end(" = ") self.print_operand(i, 1) print() print_tabbed_no_end(color_addr(i.address), tab) self.print_operand(i, 0, show_deref=False) print(" += D") print_tabbed_no_end(color_addr(i.address), tab) self.print_operand(i, 1, show_deref=False) print_no_end(" += D") 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 print_rep_end() 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: self.print_inst(i2, tab, "# ") print_tabbed_no_end(color_addr(i.real_inst[0].address), tab) print(i.pseudo) return if prefix == "# ": if self.ctx.comments: print_comment_no_end(prefix, tab) print_no_end(color_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_tabbed_no_end(color_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) + " ") 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 and not self.ctx.dump: 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()
def __print_inst(self, i, tab=0, prefix=""): def get_inst_str(): nonlocal i return "%s %s" % (i.mnemonic, i.op_str) print_tabbed_no_end(color_addr(i.address), tab) self.print_bytes(i) if is_ret(i): print_no_end(color_retcall(get_inst_str())) return if is_call(i): print_no_end(color_retcall(i.mnemonic) + " ") self.print_operand(i, 0, hexa=True) return # Here we can have conditional jump with the option --dump if is_jump(i): if i.operands[0].type != X86_OP_IMM: print_no_end(i.mnemonic + " ") self.print_operand(i, 0) if is_uncond_jump( i) and self.ctx.comments and not self.ctx.dump: print_comment_no_end(" # STOPPED") return try: addr = i.operands[0].value.imm print_no_end(i.mnemonic + " " + color(hex(addr), self.ctx.addr_color[addr])) except KeyError: print_no_end(i.mnemonic + " " + hex(addr)) return modified = False if i.id in INST_CHECK: if (i.id == X86_INS_OR and i.operands[1].type == X86_OP_IMM and i.operands[1].value.imm == -1): self.print_operand(i, 0) print_no_end(" = -1") elif (i.id == X86_INS_AND and i.operands[1].type == X86_OP_IMM and i.operands[1].value.imm == 0): self.print_operand(i, 0) print_no_end(" = 0") elif (all(op.type == X86_OP_REG for op in i.operands) and len(set(op.value.reg for op in i.operands)) == 1 and i.id == X86_INS_XOR): self.print_operand(i, 0) print_no_end(" = 0") elif i.id == X86_INS_INC or i.id == X86_INS_DEC: self.print_operand(i, 0) print_no_end(inst_symbol(i)) elif i.id == X86_INS_LEA: self.print_operand(i, 0) print_no_end(" = &(") self.print_operand(i, 1) print_no_end(")") elif i.id == X86_INS_IMUL: if len(i.operands) == 3: self.print_operand(i, 0) print_no_end(" = ") self.print_operand(i, 1) print_no_end(" " + inst_symbol(i).rstrip('=') + " ") self.print_operand(i, 2) elif len(i.operands) == 2: self.print_operand(i, 0) print_no_end(" " + inst_symbol(i) + " ") self.print_operand(i, 1) elif len(i.operands) == 1: sz = i.operands[0].size if sz == 1: print_no_end("ax = al * ") elif sz == 2: print_no_end("dx:ax = ax * ") elif sz == 4: print_no_end("edx:eax = eax * ") elif sz == 8: print_no_end("rdx:rax = rax * ") self.print_operand(i, 0) else: self.print_operand(i, 0) print_no_end(" " + inst_symbol(i) + " ") self.print_operand(i, 1) modified = True elif i.id == X86_INS_CDQE: print_no_end("rax = eax") modified = True elif i.id == X86_INS_IDIV: print_no_end('eax = edx:eax / ') self.print_operand(i, 0) print_no_end('; edx = edx:eax % ') self.print_operand(i, 0) modified = True elif i.id == X86_INS_MUL: lut = { 1: ("al", "ax"), 2: ("ax", "dx:ax"), 4: ("eax", "edx:eax"), 8: ("rax", "rdx:rax") } src, dst = lut[i.operands[0].size] print_no_end('{0} = {1} * '.format(dst, src)) self.print_operand(i, 0) modified = True elif i.id == X86_INS_NOT: self.print_operand(i, 0) print_no_end(' ^= -1') modified = True elif i.id == X86_INS_SCASB and i.prefix[0] == X86_PREFIX_REPNE: print_no_end('while (') self.print_operand(i, 1) print_no_end(' != ') self.print_operand(i, 0) print_no_end(') { ') self.print_operand(i, 1, show_deref=False) print_no_end('++; cx--; } ') self.print_operand(i, 1, show_deref=False) print_no_end('++; cx--;') 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 return modified
from lib.colors import color, yellow print color("123", 31)