| OLD | NEW |
|---|
| (Empty) |
| 1 #!/usr/bin/env python | |
| 2 # Copyright 2014 The Chromium Authors. All rights reserved. | |
| 3 # Use of this source code is governed by a BSD-style license that can be | |
| 4 # found in the LICENSE file. | |
| 5 | |
| 6 """Utility functions used by Generational and Mutational ClusterFuzz | |
| 7 fuzzers.""" | |
| 8 | |
| 9 import argparse | |
| 10 import os | |
| 11 import random | |
| 12 import string | |
| 13 import sys | |
| 14 import tempfile | |
| 15 | |
| 16 APP_PATH_KEY = 'APP_PATH' | |
| 17 FLAGS_PREFIX = 'flags-' | |
| 18 FUZZ_PREFIX = 'fuzz-' | |
| 19 IPC_FUZZER_APPLICATION = 'ipc_fuzzer' | |
| 20 IPC_REPLAY_APPLICATION = 'ipc_fuzzer_replay' | |
| 21 IPCDUMP_EXTENSION = '.ipcdump' | |
| 22 LAUNCH_PREFIXES = [ | |
| 23 '--gpu-launcher', | |
| 24 '--plugin-launcher', | |
| 25 '--ppapi-plugin-launcher', | |
| 26 '--renderer-cmd-prefix', | |
| 27 '--utility-cmd-prefix', | |
| 28 ] | |
| 29 | |
| 30 def application_name_for_platform(application_name): | |
| 31 """Return application name for current platform.""" | |
| 32 if platform() == 'WINDOWS': | |
| 33 return application_name + '.exe' | |
| 34 return application_name | |
| 35 | |
| 36 def create_flags_file(ipcdump_testcase_path, ipc_replay_application_path): | |
| 37 """Create a flags file to add launch prefix to application command line.""" | |
| 38 random_launch_prefix = random.choice(LAUNCH_PREFIXES) | |
| 39 file_content = '%s=%s' % (random_launch_prefix, ipc_replay_application_path) | |
| 40 | |
| 41 flags_file_path = ipcdump_testcase_path.replace(FUZZ_PREFIX, FLAGS_PREFIX) | |
| 42 file_handle = open(flags_file_path, 'w') | |
| 43 file_handle.write(file_content) | |
| 44 file_handle.close() | |
| 45 | |
| 46 def create_temp_file(): | |
| 47 """Create a temporary file.""" | |
| 48 temp_file = tempfile.NamedTemporaryFile(delete=False) | |
| 49 temp_file.close() | |
| 50 return temp_file.name | |
| 51 | |
| 52 def get_fuzzer_application_name(): | |
| 53 """Get the application name for the fuzzer binary.""" | |
| 54 return application_name_for_platform(IPC_FUZZER_APPLICATION) | |
| 55 | |
| 56 def get_replay_application_name(): | |
| 57 """Get the application name for the replay binary.""" | |
| 58 return application_name_for_platform(IPC_REPLAY_APPLICATION) | |
| 59 | |
| 60 def parse_arguments(): | |
| 61 """Parse fuzzer arguments.""" | |
| 62 parser = argparse.ArgumentParser() | |
| 63 parser.add_argument('--input_dir') | |
| 64 parser.add_argument('--output_dir') | |
| 65 parser.add_argument('--no_of_files', type=int) | |
| 66 args = parser.parse_args(); | |
| 67 if (not args.input_dir or | |
| 68 not args.output_dir or | |
| 69 not args.no_of_files): | |
| 70 parser.print_help() | |
| 71 sys.exit(1) | |
| 72 | |
| 73 return args | |
| 74 | |
| 75 def random_id(size=16, chars=string.ascii_lowercase): | |
| 76 """Return a random id string, default 16 characters long.""" | |
| 77 return ''.join(random.choice(chars) for _ in range(size)) | |
| 78 | |
| 79 def random_ipcdump_testcase_path(ipcdump_directory): | |
| 80 """Return a random ipc testcase path.""" | |
| 81 return os.path.join( | |
| 82 ipcdump_directory, | |
| 83 '%s%s%s' % (FUZZ_PREFIX, random_id(), IPCDUMP_EXTENSION)) | |
| 84 | |
| 85 def platform(): | |
| 86 """Return running platform.""" | |
| 87 if sys.platform.startswith('win'): | |
| 88 return 'WINDOWS' | |
| 89 if sys.platform.startswith('linux'): | |
| 90 return 'LINUX' | |
| 91 if sys.platform == 'darwin': | |
| 92 return 'MAC' | |
| 93 | |
| 94 assert False, 'Unknown platform' | |
| 95 | |
| 96 def get_application_path(): | |
| 97 """Return chrome application path.""" | |
| 98 if APP_PATH_KEY not in os.environ: | |
| 99 sys.exit( | |
| 100 'Environment variable %s should be set to chrome path.' % APP_PATH_KEY) | |
| 101 | |
| 102 return os.environ[APP_PATH_KEY] | |
| OLD | NEW |
|---|
Chromium Code Reviews
Issue 1025483002:
Restructure the ipc_fuzzer directory. (Closed)
Base URL: https://chromium.googlesource.com/chromium/src.git@master