def _perform_vex_stmt_Exit(self, guard_bundle, target_bundle, jumpkind):
guard, guard_deps = guard_bundle
target, target_deps = target_bundle
if o.TRACK_JMP_ACTIONS in self.state.options:
guard_ao = SimActionObject(guard, deps=guard_deps, state=self.state)
target_ao = SimActionObject(target, deps=target_deps, state=self.state)
self.state.history.add_action(SimActionExit(self.state, target=target_ao, condition=guard_ao, exit_type=SimActionExit.CONDITIONAL))
super()._perform_vex_stmt_Exit(guard, target, jumpkind)
def _perform_vex_stmt_Put(self, offset_bundle, data_bundle, **kwargs):
offset, offset_deps = offset_bundle
data, data_deps = data_bundle
# track the put
if o.TRACK_REGISTER_ACTIONS in self.state.options:
data_ao = SimActionObject(data, deps=data_deps, state=self.state)
size_ao = SimActionObject(len(data))
a = SimActionData(self.state, SimActionData.REG, SimActionData.WRITE, addr=offset, data=data_ao, size=size_ao)
self.state.history.add_action(a)
else:
a = None
super()._perform_vex_stmt_Put(offset, data, action=a, **kwargs)
def _perform_vex_defaultexit(self, target_bundle, jumpkind):
if target_bundle is not None:
target, target_deps = target_bundle
if o.TRACK_JMP_ACTIONS in self.state.options:
target_ao = SimActionObject(target, deps=target_deps, state=self.state)
self.state.history.add_action(SimActionExit(self.state, target_ao, exit_type=SimActionExit.DEFAULT))
else:
target = None
super()._perform_vex_defaultexit(target, jumpkind)
def _perform_vex_expr_Op(self, op, args):
exprs, deps = zip(*args)
result = super()._perform_vex_expr_Op(op, exprs)
if o.TRACK_OP_ACTIONS in self.state.options:
action_objects = [SimActionObject(arg, deps=dep, state=self.state) for arg, dep in args]
r = SimActionOperation(self.state, op, action_objects, result)
self.state.history.add_action(r)
result_deps = frozenset((r,))
else:
result_deps = frozenset().union(*deps)
return result, result_deps
def _perform_vex_stmt_Store(self, addr_bundle, data_bundle, end, condition=None, **kwargs):
addr, addr_deps = addr_bundle
data, data_deps = data_bundle
if condition is not None:
condition, condition_deps = condition
else:
condition_deps = None
# track the write
if o.TRACK_MEMORY_ACTIONS in self.state.options and addr_deps is not None:
data_ao = SimActionObject(data, deps=data_deps, state=self.state)
addr_ao = SimActionObject(addr, deps=addr_deps, state=self.state)
size_ao = SimActionObject(len(data))
cond_ao = SimActionObject(condition, deps=condition_deps, state=self.state) if condition_deps is not None else None
a = SimActionData(self.state, SimActionData.MEM, SimActionData.WRITE, data=data_ao, size=size_ao, addr=addr_ao, condition=cond_ao)
self.state.history.add_action(a)
else:
a = None
super()._perform_vex_stmt_Store(addr, data, end, action=a, condition=condition, **kwargs)
def _perform_vex_expr_Load(self, addr_bundle, ty, end, condition=None, **kwargs):
addr, addr_deps = addr_bundle
if condition is not None:
condition, condition_deps = condition
else:
condition_deps = None
result = super()._perform_vex_expr_Load(addr, ty, end, condition=condition, **kwargs)
if o.TRACK_MEMORY_ACTIONS in self.state.options:
addr_ao = SimActionObject(addr, deps=addr_deps, state=self.state)
condition_ao = SimActionObject(condition, deps=condition_deps, state=self.state) \
if condition is not None else None
r = SimActionData(self.state, self.state.memory.id, SimActionData.READ, addr=addr_ao,
size=pyvex.get_type_size(ty), data=result, condition=condition_ao)
self.state.history.add_action(r)
a = frozenset((r,))
else:
a = frozenset()
return result, a
def _perform_vex_expr_Get(self, offset_bundle, ty, **kwargs):
offset, offset_deps = offset_bundle
result = super()._perform_vex_expr_Get(offset, ty, **kwargs)
if o.TRACK_REGISTER_ACTIONS in self.state.options:
offset_ao = SimActionObject(offset, deps=offset_deps, state=self.state)
r = SimActionData(self.state, self.state.registers.id, SimActionData.READ, addr=offset_ao,
size=pyvex.get_type_size(ty), data=result
)
self.state.history.add_action(r)
a = frozenset((r,))
else:
a = frozenset()
return result, a
def _perform_vex_expr_Load(self, addr_bundle, ty, end, **kwargs):
addr, addr_deps = addr_bundle
result = super()._perform_vex_expr_Load(addr, ty, end, **kwargs)
if o.TRACK_MEMORY_ACTIONS in self.state.options:
addr_ao = SimActionObject(addr, deps=addr_deps, state=self.state)
r = SimActionData(self.state,
self.state.memory.id,
SimActionData.READ,
addr=addr_ao,
size=pyvex.get_type_size(ty),
data=result)
self.state.history.add_action(r)
a = (r, )
else:
a = ()
return result, a