def value(self, value):
"""
Set the operand to the specified value within the cpu_context.
"""
# If we are writing to an immediate, I believe they want to write to the memory at the immediate.
# TODO: Should we fail instead?
if self.is_immediate:
offset = self.value
if idaapi.is_loaded(offset):
self._cpu_context.mem_write(offset, value)
return
if self.is_register:
# Convert the value from string to integer...
if isinstance(value, str):
value = utils.struct_unpack(value)
# On 64-bit, the destination register must be set to 0 first (per documentation)
# TODO: Check if this happens regardless of the source size
if idc.__EA64__ and self.width == 4: # Only do this for 32-bit setting
reg64 = utils.convert_reg(self.text, 8)
self._cpu_context.registers[reg64] = 0
self._cpu_context.registers[self.text] = value
return
# TODO: Determine if this is still necessary.
# FS, GS (at least) registers are identified as memory addresses. We need to identify them as registers
# and handle them as such
if self.type == idc.o_mem:
if "fs" in self.text:
self._cpu_context.registers.fs = value
return
elif "gs" in self.text:
self._cpu_context.registers.gs = value
return
if self.is_memory_reference:
# For data written to the frame or memory, this data MUST be a byte string.
if numpy.issubdtype(type(value), numpy.integer):
value = utils.struct_pack(value, width=self.width)
self._cpu_context.mem_write(self.addr, value)
return
raise FunctionTracingError('Invalid operand type: {}'.format(
self.type),
ip=self.ip)
def value(self, value):
try:
logger.debug("0x%X -> %s", value, self.text)
except TypeError:
logger.debug("%r -> %s", value, self.text)
# TODO: Determine if this is still necessary.
# FS, GS (at least) registers are identified as memory addresses. We need to identify them as registers
# and handle them as such
if self.type == idc.o_mem:
if "fs" in self.text:
self._cpu_context.registers.fs = value
return
elif "gs" in self.text:
self._cpu_context.registers.gs = value
return
# On 64-bit, the destination register must be set to 0 first (per documentation)
# TODO: Check if this happens regardless of the source size
if self.is_register and idc.__EA64__ and self.width == 4: # Only do this for 32-bit setting
reg64 = utils.convert_reg(self.text, 8)
self._cpu_context.registers[reg64] = 0
super(x86_64Operand, self.__class__).value.__set__(self, value)