def test_get_analyzed_r2pipe_from_input_with_none(self):
with self.assertRaises(Exception) as context:
r2pu.get_analyzed_r2pipe_from_input()
self.assertTrue('Inside empty session' in str(context.exception))
def test_get_call_xref_list_to_function_offset_with_offset(self):
r2 = r2pu.get_analyzed_r2pipe_from_input(TEST_FILE)
self.assertTrue(
len(r2pu.get_call_xref_list_to_function_offset(r2, 0x401020)) == 3)
r2.quit()
def test_get_args_count_to_function_offset_with_offset(self):
r2 = r2pu.get_analyzed_r2pipe_from_input(TEST_FILE)
self.assertTrue(
int(r2pu.get_args_count_to_function_offset(r2, 0x40102a)) == 2)
r2.quit()
def test_get_analyzed_r2pipe_from_input_with_pipe(self):
r2 = r2pipe.open(TEST_FILE)
r2 = r2pu.get_analyzed_r2pipe_from_input(r2)
self.assertTrue(int(r2.cmd('aflc')) > 0)
r2.quit()
def test_get_function_start_from_offset_with_offset(self):
r2 = r2pu.get_analyzed_r2pipe_from_input(TEST_FILE)
self.assertTrue(
hex(r2pu.get_function_start_from_offset(r2, 0x40100a)) ==
'0x401000')
r2.quit()
def test_get_funcj_list(self):
r2 = r2pu.get_analyzed_r2pipe_from_input(TEST_FILE)
funcj_list = r2pu.get_funcj_list(r2)
self.assertTrue(len(funcj_list) > 0)
r2.quit()
def test_get_args_count_to_function_offset_with_none(self):
r2 = r2pu.get_analyzed_r2pipe_from_input(TEST_FILE)
# Address 0x401020 is the 'add' function that takes 2 args
r2.cmd("s 0x401020")
self.assertTrue(int(r2pu.get_args_count_to_function_offset(r2)) == 2)
r2.quit()
def test_get_function_start_from_offset_with_none(self):
r2 = r2pu.get_analyzed_r2pipe_from_input(TEST_FILE)
# Address 0x40100a belongs to fcn.401000
r2.cmd("s 0x40100a")
self.assertTrue(
hex(r2pu.get_function_start_from_offset(r2)) == '0x401000')
r2.quit()
