def test_mockup_function5(self):
arch = mozz.system.architecture()
if arch != mozz.system.ARCH_X86_64:
return #this test not yet supported on non x86_64 systems
s = mozz.Session(abs_path(__file__, "function_test.bin"))
s.set_calling_convention(native_convention())
result = {
'count': 0,
'params_correct': [0, 0, 0],
'default_case': 0,
'checkpoint3': 0
}
@s.at_addr("default_case")
def default_case(host):
result['default_case'] = 1
@s.at_function("checkpoint3", TCInt32, TCInt32)
def checkpoint3(host, fn_ctx, n, status, status2):
result['checkpoint3'] = 1
self.assertEqual(status.value(), 0x8000)
self.assertEqual(status2.value(), 345987)
@s.mockup_function("test_function", TCInt32, TCInt8, TCInt16, TCUInt32,
TCInt32, Pointer64, Pointer64, Pointer64)
def test_function(host, fn_ctx, n, a, b, c, d, e, \
buf1, buf2, output_var):
host.log("test_function(%d):" % n)
host.log(" a=%d" % a.value())
host.log(" b=%d" % b.value())
host.log(" c=%d" % c.value())
host.log(" d=%d" % d.value())
host.log(" e=%d" % e.value())
host.log(" buf1=%x:%r" % (buf1.value(), buf1.string(host)))
host.log(" buf2=%x:%r" % (buf2.value(), buf2.string(host)))
host.log(" output_var=%x" % output_var.value())
result['count'] += 1
if n < 3 and self.check_params(n, host, a, b, c, d, e, buf1, buf2):
result['params_correct'][n] = 1
if n == 2:
d.set(0x98989898)
e.set(0)
c.set(0)
a.set(0)
return None # dont do mockup, execute function
else:
return 0
mozz.run_session(s)
self.assertEqual(result['count'], 3)
for i in range(3):
print("check params #%d" % i)
self.assertEqual(result['params_correct'][i], 1)
self.assertEqual(result['checkpoint3'], 1)
self.assertEqual(result['default_case'], 1)
def test_mockup_function4(self):
arch = mozz.system.architecture()
if arch != mozz.system.ARCH_X86_64:
return #this test not yet supported on non x86_64 systems
s = mozz.Session(abs_path(__file__, "function_test.bin"))
s.set_calling_convention(native_convention())
result = {
'count': 0,
'params_correct': [0,0,0],
'default_case': 0
}
@s.at_addr("default_case")
def default_case(host):
result['default_case'] = 1
@s.mockup_function(
"test_function",
TCInt32, TCInt8, TCInt16, TCUInt32, TCInt32,
Pointer64, Pointer64, Pointer64
)
def test_function(host, fn_ctx, n, a, b, c, d, e, \
buf1, buf2, output_var):
host.log("test_function(%d):" % n)
host.log(" a=%d" % a.value())
host.log(" b=%d" % b.value())
host.log(" c=%d" % c.value())
host.log(" d=%d" % d.value())
host.log(" e=%d" % e.value())
host.log(" buf1=%x:%r" % (buf1.value(), buf1.string(host)))
host.log(" buf2=%x:%r" % (buf2.value(), buf2.string(host)))
host.log(" output_var=%x" % output_var.value())
result['count'] += 1
if n < 3 and self.check_params(n,host,a,b,c,d,e,buf1,buf2):
result['params_correct'][n] = 1
if n == 2:
output_var.deref(host, TCInt32).set(0xc0bb13)
return 4567
else:
return 0
mozz.run_session(s)
self.assertEqual(result['count'], 3)
for i in range(3):
print("check params #%d" % i)
self.assertEqual(result['params_correct'][i], 1)
self.assertEqual(result['default_case'], 0)
def test_mockup_function3(self):
arch = mozz.system.architecture()
if arch != mozz.system.ARCH_X86_64:
return #this test not yet supported on non x86_64 systems
s = mozz.Session(abs_path(__file__, "function_test.bin"))
s.set_calling_convention(native_convention())
result = {'count': 0, 'params_correct': [0, 0, 0], 'default_case': 0}
@s.at_addr("default_case")
def default_case(host):
result['default_case'] = 1
@s.mockup_function("test_function", TCInt32, TCInt8, TCInt16, TCUInt32,
TCInt32, Pointer64, Pointer64, Pointer64)
def test_function(host, fn_ctx, n, a, b, c, d, e, \
buf1, buf2, output_var):
host.log("test_function(%d):" % n)
host.log(" a=%d" % a.value())
host.log(" b=%d" % b.value())
host.log(" c=%d" % c.value())
host.log(" d=%d" % d.value())
host.log(" e=%d" % e.value())
host.log(" buf1=%x:%r" % (buf1.value(), buf1.string(host)))
host.log(" buf2=%x:%r" % (buf2.value(), buf2.string(host)))
host.log(" output_var=%x" % output_var.value())
result['count'] += 1
if n < 3 and self.check_params(n, host, a, b, c, d, e, buf1, buf2):
result['params_correct'][n] = 1
if n == 1:
output_var.deref(host, TCUInt32).set(0xc001f00d)
return 84739
else:
return 0
mozz.run_session(s)
self.assertEqual(result['count'], 2)
for i in range(2):
print("check params #%d" % i)
self.assertEqual(result['params_correct'][i], 1)
self.assertEqual(result['default_case'], 0)