def test_if(self):
program = Program([
Assign("foo", Value(UInt32(10))),
IF(AddOp(Value(UInt32(0)), Value(UInt32(1))), Value(UInt32(2)), Value(UInt32(3))),
Assign("foo", AddOp(Var("foo"), Value(UInt32(10)))),
Assign("blah", Value(UInt32(0)))
])
context = self.build_context(program)
self.interpreter.run(context)
self.assertEqual(UInt32(20), context.resolve_name("foo").value)
def test_bed(self):
the_input = GetInput([UInt32(3), UInt32(1)])
program = Program([
Assign("X", MulOp(Value(UInt32(2)), the_input)),
IF(EQ(SubOp(Var("X"), AddOp(Value(UInt32(3)), Value(UInt32(2)))), Value(UInt32(15))), Value(UInt32(2)),
Value(UInt32(3))),
Assign("Y", AddOp(Value(UInt32(3)), Var("X"))),
IF(GT(Var("Y"), SubOp(the_input, Value(UInt32(20)))), Value(UInt32(4)), Value(UInt32(5)))
])
self.interpreter.run(a_context().with_program(program).build())
print str(self.interpreter.constraints)
print repr(self.interpreter.constraints)
def test_store(self):
mem_address = UInt32(0x1000)
program = Program([
Store(Value(mem_address), AddOp(Value(UInt32(10)), Value(UInt32(20))))
])
context = self.build_context(program)
self.interpreter.run(context)
self.assertEqual(UInt32(30), context.get_mem_value(mem_address).value)
def test_load(self):
mem_address = UInt32(0x1000)
program = Program([
Store(Value(mem_address), AddOp(Value(UInt32(10)), Value(UInt32(20)))),
Assign("foo", Load(Value(mem_address)))
])
context = self.build_context(program)
self.interpreter.run(context)
self.assertEqual(UInt32(30), context.resolve_name("foo").value)
def test_get_input_assign(self):
program = Program([
Assign('foo', GetInput([UInt32(1), UInt32(2), UInt32(3), UInt32(4)])),
Assign("blah", AddOp(Var("foo"), Value(UInt32(1))))
])
context = self.build_context(program)
result = self.interpreter.run(context)
self.assertEqual(UInt32(1), result.resolve_name("foo").value)
self.assertEqual(UInt32(2), result.resolve_name("blah").value)
def test_input_var(self):
program = Program([
Assign("foo", GetInput([UInt32(0)])),
Assign("blah", AddOp(Var("foo"), Value(UInt32(1))))
])
context = a_context().with_program(program).build()
self.interpreter.run(context)
self.assertTrue(context.resolve_name("foo").isTainted())
self.assertTrue(context.resolve_name("blah").isTainted())
def test_assign(self):
program = Program([
Assign('foo', AddOp(Value(UInt32(10)), Value(UInt32(20))))
])
result = self.interpreter.run(self.build_context(program))
self.assertEqual(UInt32(30), result.resolve_name("foo").value)
def test_bin_op(self):
program = Program([
Assign("foo", MulOp(Value(UInt32(3)), AddOp(Value(UInt32(2)), Value(UInt32(3)))))
])
result = self.interpreter.run(self.build_context(program))
self.assertEqual(UInt32(15), result.resolve_name("foo").value)