1 | #!/usr/bin/python
|
---|
2 | #
|
---|
3 | # Copyright 2016-2018 The OpenSSL Project Authors. All Rights Reserved.
|
---|
4 | #
|
---|
5 | # Licensed under the OpenSSL license (the "License"). You may not use
|
---|
6 | # this file except in compliance with the License. You can obtain a copy
|
---|
7 | # in the file LICENSE in the source distribution or at
|
---|
8 | # https://www.openssl.org/source/license.html
|
---|
9 |
|
---|
10 | """Fuzzing helper, creates and uses corpus/crash directories.
|
---|
11 |
|
---|
12 | fuzzer.py <fuzzer> <extra fuzzer arguments>
|
---|
13 | """
|
---|
14 |
|
---|
15 | import os
|
---|
16 | import subprocess
|
---|
17 | import sys
|
---|
18 |
|
---|
19 | FUZZER = sys.argv[1]
|
---|
20 |
|
---|
21 | THIS_DIR = os.path.abspath(os.path.dirname(__file__))
|
---|
22 | CORPORA_DIR = os.path.abspath(os.path.join(THIS_DIR, "corpora"))
|
---|
23 |
|
---|
24 | FUZZER_DIR = os.path.abspath(os.path.join(CORPORA_DIR, FUZZER))
|
---|
25 | if not os.path.isdir(FUZZER_DIR):
|
---|
26 | os.mkdir(FUZZER_DIR)
|
---|
27 |
|
---|
28 | corpora = []
|
---|
29 |
|
---|
30 | def _create(d):
|
---|
31 | dd = os.path.abspath(os.path.join(CORPORA_DIR, d))
|
---|
32 | if not os.path.isdir(dd):
|
---|
33 | os.mkdir(dd)
|
---|
34 | corpora.append(dd)
|
---|
35 |
|
---|
36 | def _add(d):
|
---|
37 | dd = os.path.abspath(os.path.join(CORPORA_DIR, d))
|
---|
38 | if os.path.isdir(dd):
|
---|
39 | corpora.append(dd)
|
---|
40 |
|
---|
41 | def main():
|
---|
42 | _create(FUZZER)
|
---|
43 | _create(FUZZER + "-crash")
|
---|
44 | _add(FUZZER + "-seed")
|
---|
45 |
|
---|
46 | cmd = ([os.path.abspath(os.path.join(THIS_DIR, FUZZER))] + sys.argv[2:]
|
---|
47 | + ["-artifact_prefix=" + corpora[1] + "/"] + corpora)
|
---|
48 | print(" ".join(cmd))
|
---|
49 | subprocess.call(cmd)
|
---|
50 |
|
---|
51 | if __name__ == "__main__":
|
---|
52 | main()
|
---|