本文整理汇总了Python中triton.TritonContext.getConcreteRegisterValue方法的典型用法代码示例。如果您正苦于以下问题:Python TritonContext.getConcreteRegisterValue方法的具体用法?Python TritonContext.getConcreteRegisterValue怎么用?Python TritonContext.getConcreteRegisterValue使用的例子?那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类triton.TritonContext
的用法示例。
在下文中一共展示了TritonContext.getConcreteRegisterValue方法的4个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: TestX86ConcreteRegisterValue
# 需要导入模块: from triton import TritonContext [as 别名]
# 或者: from triton.TritonContext import getConcreteRegisterValue [as 别名]
class TestX86ConcreteRegisterValue(unittest.TestCase):
"""Testing the X86 concrete value api."""
def setUp(self):
"""Define the arch."""
self.Triton = TritonContext()
self.Triton.setArchitecture(ARCH.X86)
self.ar = self.Triton.getAllRegisters()
self.pr = self.Triton.getParentRegisters()
def test_all_registers(self):
"""Check all registers"""
self.assertEqual(len(self.ar), 110)
def test_parent_registers(self):
"""Check parent registers"""
self.assertEqual(len(self.pr), 89)
def test_set_get_concrete_value(self):
"""Check setting concrete values"""
for r in self.pr:
if r.getBitSize() == 32:
self.Triton.setConcreteRegisterValue(r, 0xdeadbeaf)
elif r.getBitSize() == 64:
self.Triton.setConcreteRegisterValue(r, 0xabcdef0123456789)
elif r.getBitSize() == 128:
self.Triton.setConcreteRegisterValue(r, 0xabcdef01234567899876543210fedcba)
elif r.getBitSize() == 256:
self.Triton.setConcreteRegisterValue(r, 0xabcdef01234567899876543210fedcbaabcdef01234567899876543210fedcba)
else:
pass
"""Check getting concrete values"""
for r in self.pr:
if r.getBitSize() == 32:
self.assertEqual(self.Triton.getConcreteRegisterValue(r), 0xdeadbeaf)
elif r.getBitSize() == 64:
self.assertEqual(self.Triton.getConcreteRegisterValue(r), 0xabcdef0123456789)
elif r.getBitSize() == 128:
self.assertEqual(self.Triton.getConcreteRegisterValue(r), 0xabcdef01234567899876543210fedcba)
elif r.getBitSize() == 256:
self.assertEqual(self.Triton.getConcreteRegisterValue(r), 0xabcdef01234567899876543210fedcbaabcdef01234567899876543210fedcba)
else:
pass
"""Set everything to zero"""
for r in self.ar:
self.Triton.setConcreteRegisterValue(r, 0)
"""Check if everything is equal to zero"""
for r in self.ar:
self.assertEqual(self.Triton.getConcreteRegisterValue(r), 0)
示例2: TestFlags
# 需要导入模块: from triton import TritonContext [as 别名]
# 或者: from triton.TritonContext import getConcreteRegisterValue [as 别名]
class TestFlags(unittest.TestCase):
"""Testing the concrete flag values."""
def setUp(self):
"""Define the arch."""
self.Triton = TritonContext()
self.Triton.setArchitecture(ARCH.X86_64)
def test_set_flags(self):
"""Check flags can be set in any order with a correct output result."""
registers = [REG.X86_64.ZF, REG.X86_64.AF, REG.X86_64.IF, REG.X86_64.CF,
REG.X86_64.DF, REG.X86_64.PF, REG.X86_64.SF, REG.X86_64.OF,
REG.X86_64.TF]
values = [0] * len(registers)
rand_registers = list(registers)
random.shuffle(rand_registers)
# Randomnly set flags registers and check result is the one expected
for reg in rand_registers:
self.Triton.setConcreteRegisterValue(self.Triton.getRegister(reg), 1)
values[registers.index(reg)] = 1
self.assertListEqual([self.Triton.getConcreteRegisterValue(self.Triton.getRegister(r)) for r in registers], values)
def test_unset_flags(self):
"""Check flags can be unset in any order with a correct result."""
registers = [REG.X86_64.ZF, REG.X86_64.AF, REG.X86_64.IF, REG.X86_64.CF,
REG.X86_64.DF, REG.X86_64.PF, REG.X86_64.SF, REG.X86_64.OF,
REG.X86_64.TF]
values = [1] * len(registers)
for reg in registers:
self.Triton.setConcreteRegisterValue(self.Triton.getRegister(reg), 1)
rand_registers = list(registers)
random.shuffle(rand_registers)
# Randomnly unset flags registers and check result is the one expected
for reg in rand_registers:
self.Triton.setConcreteRegisterValue(self.Triton.getRegister(reg), 0)
values[registers.index(reg)] = 0
self.assertListEqual([self.Triton.getConcreteRegisterValue(self.Triton.getRegister(r)) for r in registers], values)
示例3: hex
# 需要导入模块: from triton import TritonContext [as 别名]
# 或者: from triton.TritonContext import getConcreteRegisterValue [as 别名]
print inst
# Display symbolic expressions
for expr in inst.getSymbolicExpressions():
print '\t', expr
print
print 'Display emulated information'
print '~~~~~~~~~~~~~~~~~~~~~~~~~~~~'
write = inst.getOperands()[0].getAddress()
print 'Instruction :', inst.getDisassembly()
print 'Write at :', hex(write)
print 'Content :', hex(Triton.getConcreteMemoryValue(MemoryAccess(write+4, CPUSIZE.DWORD)))
print 'RAX value :', hex(Triton.getConcreteRegisterValue(Triton.registers.rax))
print 'RSI value :', hex(Triton.getConcreteRegisterValue(Triton.registers.rsi))
print 'RDI value :', hex(Triton.getConcreteRegisterValue(Triton.registers.rdi))
print
print 'Symbolic registers information'
print '~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~'
for k, v in Triton.getSymbolicRegisters().items():
print Triton.getRegister(k), v
print
print 'Symbolic memory information'
print '~~~~~~~~~~~~~~~~~~~~~~~~~~~'
for k, v in Triton.getSymbolicMemory().items():
print hex(k), v
示例4: TestSymbolic
# 需要导入模块: from triton import TritonContext [as 别名]
# 或者: from triton.TritonContext import getConcreteRegisterValue [as 别名]
class TestSymbolic(unittest.TestCase):
"""Testing the symbolic engine."""
def setUp(self):
"""Define the arch."""
self.Triton = TritonContext()
self.Triton.setArchitecture(ARCH.X86_64)
self.astCtxt = self.Triton.getAstContext()
def test_backup(self):
"""
Check Symbolics value are saved when engine is disable.
* Also check reseting a disable symbolic engines doesn't crash.
"""
inst = Instruction()
# update RAX
inst.setOpcode("\x48\xFF\xC0")
self.Triton.processing(inst)
self.assertEqual(self.Triton.getSymbolicRegisterValue(self.Triton.registers.rax), 1)
# This call triton::api.backupSymbolicEngine()
self.Triton.enableSymbolicEngine(False)
inst = Instruction()
# update RAX again
inst.setOpcode("\x48\xFF\xC0")
self.Triton.processing(inst)
self.assertEqual(self.Triton.getConcreteRegisterValue(self.Triton.registers.rax), 2, "concrete value is updated")
self.assertEqual(self.Triton.getSymbolicRegisterValue(self.Triton.registers.rax), 1)
self.assertEqual(self.Triton.getSymbolicRegisterValue(self.Triton.registers.rax), 1, "Symbolic value is not update")
# Try to reset engine after a backup to test if the bug #385 is fixed.
self.Triton.resetEngines()
def test_bind_expr_to_memory(self):
"""Check symbolic expression binded to memory can be retrieve."""
# Bind expr1 to 0x100
expr1 = self.Triton.newSymbolicExpression(self.astCtxt.bv(0x11, 8))
mem = MemoryAccess(0x100, CPUSIZE.BYTE)
self.Triton.assignSymbolicExpressionToMemory(expr1, mem)
# Get expr from memory
expr2 = self.Triton.getSymbolicExpressionFromId(self.Triton.getSymbolicMemoryId(0x100))
self.assertEqual(expr1.getAst().evaluate(), expr2.getAst().evaluate())
def test_bind_expr_to_multi_memory(self):
"""Check symbolic expression binded to multiple memory location."""
# Bind expr to multi memory location (0x100, 0x101, 0x102, 0x103)
expr1 = self.Triton.newSymbolicExpression(self.astCtxt.bv(0x11223344, 32))
mem = MemoryAccess(0x100, CPUSIZE.DWORD)
self.Triton.assignSymbolicExpressionToMemory(expr1, mem)
# Check we can get back the same values
expr2 = self.Triton.getSymbolicExpressionFromId(self.Triton.getSymbolicMemoryId(0x100))
expr3 = self.Triton.getSymbolicExpressionFromId(self.Triton.getSymbolicMemoryId(0x101))
expr4 = self.Triton.getSymbolicExpressionFromId(self.Triton.getSymbolicMemoryId(0x102))
expr5 = self.Triton.getSymbolicExpressionFromId(self.Triton.getSymbolicMemoryId(0x103))
self.assertEqual(expr2.getAst().evaluate(), 0x44)
self.assertEqual(expr3.getAst().evaluate(), 0x33)
self.assertEqual(expr4.getAst().evaluate(), 0x22)
self.assertEqual(expr5.getAst().evaluate(), 0x11)
self.assertEqual(self.Triton.getSymbolicMemoryValue(mem), 0x11223344)
def test_bind_expr_to_register(self):
"""Check symbolic expression binded to register."""
expr1 = self.Triton.newSymbolicExpression(self.astCtxt.bv(0x11223344, 64))
self.Triton.assignSymbolicExpressionToRegister(expr1, self.Triton.registers.rax)
self.assertEqual(self.Triton.getSymbolicRegisterValue(self.Triton.registers.rax), 0x11223344)
expr1 = self.Triton.newSymbolicExpression(self.astCtxt.bv(0x11223344, 32))
with self.assertRaises(Exception):
# Incorrect size
self.Triton.assignSymbolicExpressionToRegister(expr1, self.Triton.registers.rax)