| Index: tools/ipc_fuzzer/mutate/cf_package_builder.py
|
| diff --git a/tools/ipc_fuzzer/mutate/cf_package_builder.py b/tools/ipc_fuzzer/mutate/cf_package_builder.py
|
| index 608cea13d5110a314a304ee5c21bc68d6f2ee2a1..1fc98d186934c3c2f52aaa2107c2fb693ed39cde 100755
|
| --- a/tools/ipc_fuzzer/mutate/cf_package_builder.py
|
| +++ b/tools/ipc_fuzzer/mutate/cf_package_builder.py
|
| @@ -24,7 +24,7 @@ class CFPackageBuilder:
|
| def __init__(self):
|
| self.fuzzer_list = FUZZER_LIST
|
|
|
| - def parse_args(self):
|
| + def parse_arguments(self):
|
| desc = 'Builder of IPC fuzzer packages for ClusterFuzz'
|
| parser = argparse.ArgumentParser(description=desc)
|
| parser.add_argument('--out-dir', dest='out_dir', default='out',
|
| @@ -33,7 +33,7 @@ class CFPackageBuilder:
|
| help='Debug vs. Release build')
|
| self.args = parser.parse_args()
|
|
|
| - def get_paths(self):
|
| + def set_application_paths(self):
|
| script_path = os.path.realpath(__file__)
|
| self.mutate_dir = os.path.dirname(script_path)
|
| src_dir = os.path.join(self.mutate_dir, os.pardir, os.pardir, os.pardir)
|
| @@ -41,12 +41,12 @@ class CFPackageBuilder:
|
| out_dir = os.path.join(src_dir, self.args.out_dir)
|
| self.build_dir = os.path.join(out_dir, self.args.build_type)
|
|
|
| - def enter_tmp_workdir(self):
|
| + def switch_to_temp_work_directory(self):
|
| self.old_cwd = os.getcwd()
|
| self.work_dir = tempfile.mkdtemp()
|
| os.chdir(self.work_dir)
|
|
|
| - def rm_tmp_workdir(self):
|
| + def remove_temp_work_directory(self):
|
| os.chdir(self.old_cwd)
|
| shutil.rmtree(self.work_dir)
|
|
|
| @@ -55,19 +55,23 @@ class CFPackageBuilder:
|
| fuzzer_src_path = os.path.join(self.mutate_dir, fuzzer + '.py')
|
| fuzzer_dst_path = os.path.join(fuzzer, 'run.py')
|
| shutil.copyfile(fuzzer_src_path, fuzzer_dst_path)
|
| + utils_src_path = os.path.join(self.mutate_dir, 'utils.py')
|
| + utils_dst_path = os.path.join(fuzzer, 'utils.py')
|
| + shutil.copyfile(utils_src_path, utils_dst_path)
|
| distutils.archive_util.make_zipfile(fuzzer, fuzzer)
|
| package_name = fuzzer + '.zip'
|
| shutil.copy(package_name, self.build_dir)
|
| final_package_path = os.path.join(self.build_dir, package_name)
|
| - print 'Built %s' % final_package_path
|
| + print 'Built %s.' % final_package_path
|
|
|
| def main(self):
|
| - self.parse_args()
|
| - self.get_paths()
|
| - self.enter_tmp_workdir()
|
| + self.parse_arguments()
|
| + self.set_application_paths()
|
| + self.switch_to_temp_work_directory()
|
| for fuzzer in self.fuzzer_list:
|
| self.build_package(fuzzer)
|
| - self.rm_tmp_workdir()
|
| + self.remove_temp_work_directory()
|
| +
|
| return 0
|
|
|
| if __name__ == "__main__":
|
|
|