1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
|
# Copyright 2014 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Creates IPC fuzzer packages suitable for uploading to ClusterFuzz. Stores
the packages into chrome build directory. See fuzzer_list below for the list of
fuzzers.
"""
from __future__ import print_function
import argparse
import distutils.archive_util
import os
import shutil
import sys
import tempfile
FUZZER_LIST = [
'ipc_fuzzer_mut',
'ipc_fuzzer_gen',
]
class CFPackageBuilder:
def __init__(self):
self.fuzzer_list = FUZZER_LIST
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',
help='output directory under src/ directory')
parser.add_argument(
'--build-type',
dest='build_type',
default='Release',
help='Debug vs. Release build')
self.args = parser.parse_args()
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)
src_dir = os.path.abspath(src_dir)
out_dir = os.path.join(src_dir, self.args.out_dir)
self.build_dir = os.path.join(out_dir, self.args.build_type)
def switch_to_temp_work_directory(self):
self.old_cwd = os.getcwd()
self.work_dir = tempfile.mkdtemp()
os.chdir(self.work_dir)
def remove_temp_work_directory(self):
os.chdir(self.old_cwd)
shutil.rmtree(self.work_dir)
def build_package(self, fuzzer):
os.makedirs(fuzzer)
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)
def main(self):
self.parse_arguments()
self.set_application_paths()
self.switch_to_temp_work_directory()
for fuzzer in self.fuzzer_list:
self.build_package(fuzzer)
self.remove_temp_work_directory()
return 0
if __name__ == '__main__':
builder = CFPackageBuilder()
sys.exit(builder.main())
|
