def uses(self, cfg=None): # Avoid circular import. TODO: fix properly import arch import progdb """Return set of all registers used by this instruction. Function calls (and maybe SFUNCs) require special treatment.""" if self.op == "call": addr = self.args[0] uses = addr.regs() if isinstance(addr, ADDR): # Direct call with known address addr = addr.addr if addr in progdb.FUNC_DB and "params" in progdb.FUNC_DB[addr]: return uses | progdb.FUNC_DB[addr]["params"] # Indirect call or not params in funcdb # TODO: need to allow saving callsite info in funcdb return uses | arch.call_params(addr) if self.op == "return": if not self.args: return arch.ret_uses(cfg) uses = set() for a in self.args: for r in a.regs(): uses.add(r) if is_mem(self.dest): for r in self.dest.regs(): uses.add(r) return uses
def uses(self, cfg=None): # Avoid circular import. TODO: fix properly import arch import progdb """Return set of all registers used by this instruction. Function calls (and maybe SFUNCs) require special treatment.""" if self.op == "call": addr = self.args[0] uses = addr.regs() if isinstance(addr, ADDR): # Direct call with known address addr = addr.addr if addr in progdb.FUNC_DB and "params" in progdb.FUNC_DB[addr]: return uses | progdb.FUNC_DB[addr]["params"] # Indirect call or not params in funcdb # TODO: need to allow saving callsite info in funcdb return uses | arch.call_params(addr) if self.op == "return": if not self.args: return arch.ret_uses(cfg) uses = set() for a in self.args: for r in a.regs(): uses.add(r) if is_mem(self.dest): for r in self.dest.regs(): uses.add(r) return uses
def insert_params(cfg): entry_addr = cfg.entry() first_bblock = cfg[entry_addr]["val"] for i, reg in enumerate(sorted(list(arch.call_params(cfg.props["name"])))): first_bblock.items.insert( i, Inst(reg, "=", [REG("arg_" + reg.name)], addr=entry_addr + ".arg%d" % i))
def insert_params(cfg): entry_addr = cfg.entry() first_bblock = cfg[entry_addr]["val"] for i, reg in enumerate(sorted(list(arch.call_params(cfg.props["name"])))): first_bblock.items.insert(i, Inst(reg, "=", [REG("arg_" + reg.name)], addr=entry_addr + ".arg%d" % i))