def launch_execute_file(self):
working_dir = os.path.join(self.DEFAULT_ROOT_TMP_DIR, self.DEFAULT_ROOT_OUTPUT_DIR, self.fuzzer_name)
cmd_format = "cd %s; bash %s 2>&1 > %s"
execution_log_dir = os.path.join(self.working_dir, self.DEFAULT_ROOT_LOG_DIR, self.fuzzer_name)
if os.path.exists(execution_log_dir) is False:
os.makedirs(execution_log_dir)
execution_log_path = os.path.join(execution_log_dir, self.DEFAULT_EXEC_LOG_NAME)
cmd_str = cmd_format % (working_dir, self.DEFAULT_LAUNCH_SCRIPT_FILE_NAME, execution_log_path)
adb_obj = AdbWrapper()
adb_obj.adb_forward(local="tcp:2828", remote="tcp:2828")
os.system(cmd_str)
def launch_execute_file(self):
working_dir = os.path.join(self.DEFAULT_ROOT_TMP_DIR,
self.DEFAULT_ROOT_OUTPUT_DIR,
self.fuzzer_name)
cmd_format = "cd %s; bash %s 2>&1 > %s"
execution_log_dir = os.path.join(self.working_dir,
self.DEFAULT_ROOT_LOG_DIR,
self.fuzzer_name)
if os.path.exists(execution_log_dir) is False:
os.makedirs(execution_log_dir)
execution_log_path = os.path.join(execution_log_dir,
self.DEFAULT_EXEC_LOG_NAME)
cmd_str = cmd_format % (working_dir,
self.DEFAULT_LAUNCH_SCRIPT_FILE_NAME,
execution_log_path)
adb_obj = AdbWrapper()
adb_obj.adb_forward(local="tcp:2828", remote="tcp:2828")
os.system(cmd_str)
def test_forward_remove(self):
"""
test remove a forward
"""
str_ret = textwrap.dedent('')
self.mock_obj.communicate.return_value = [str_ret, None]
self.mock_popen.return_value = self.mock_obj
ret = AdbWrapper.adb_forward(command='remove', local='tcp:2828')
self.assertTrue(ret, 'The result should be True.')
def test_forward_remove_no_dest(self):
"""
test remove a forward by wrong argv
"""
str_ret = textwrap.dedent('')
self.mock_obj.communicate.return_value = [str_ret, None]
self.mock_popen.return_value = self.mock_obj
ret = AdbWrapper.adb_forward(command='remove', remote='tcp:2828')
self.assertFalse(ret, 'The result should be False.')
def test_forward_list(self):
"""
test forward --list
"""
str_ret = textwrap.dedent("""\
e481d888 tcp:49662 tcp:2828
e479da98 tcp:57988 tcp:2828
e47cd830 tcp:45784 tcp:2828
e47cd887 tcp:41946 tcp:2828
7ed3ca11 tcp:39961 tcp:2828
""")
expected_ret = {'e481d888': {'source': 'tcp:49662', 'dest': 'tcp:2828'},
'e479da98': {'source': 'tcp:57988', 'dest': 'tcp:2828'},
'e47cd830': {'source': 'tcp:45784', 'dest': 'tcp:2828'},
'e47cd887': {'source': 'tcp:41946', 'dest': 'tcp:2828'},
'7ed3ca11': {'source': 'tcp:39961', 'dest': 'tcp:2828'}}
self.mock_obj.communicate.return_value = [str_ret, None]
self.mock_popen.return_value = self.mock_obj
fwd_list = AdbWrapper.adb_forward(command='list')
self.assertEqual(fwd_list, expected_ret,
'The result should be {}, not {}.'.format(expected_ret, fwd_list))
