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)