Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(77)

Side by Side Diff: testing/libfuzzer/archive_corpus.py

Issue 2520983002: [libfuzzer] Turn warnings from zipfile into errors while archiving seed corpus. (Closed)
Patch Set: Rebase. Created 4 years ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
« no previous file with comments | « no previous file | no next file » | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
1 #!/usr/bin/python2 1 #!/usr/bin/python2
2 # 2 #
3 # Copyright 2016 The Chromium Authors. All rights reserved. 3 # Copyright 2016 The Chromium Authors. All rights reserved.
4 # Use of this source code is governed by a BSD-style license that can be 4 # Use of this source code is governed by a BSD-style license that can be
5 # found in the LICENSE file. 5 # found in the LICENSE file.
6 6
7 """Archive corpus file into zip and generate .d depfile. 7 """Archive corpus file into zip and generate .d depfile.
8 8
9 Invoked by GN from fuzzer_test.gni. 9 Invoked by GN from fuzzer_test.gni.
10 """ 10 """
11 11
12 from __future__ import print_function 12 from __future__ import print_function
13 import argparse 13 import argparse
14 import os 14 import os
15 import sys 15 import sys
16 import warnings
16 import zipfile 17 import zipfile
17 18
18 19
19 def main(): 20 def main():
20 parser = argparse.ArgumentParser(description="Generate fuzzer config.") 21 parser = argparse.ArgumentParser(description="Generate fuzzer config.")
21 parser.add_argument('--corpus', required=True) 22 parser.add_argument('--corpus', required=True)
22 parser.add_argument('--output', required=True) 23 parser.add_argument('--output', required=True)
23 parser.add_argument('--fuzzer', required=True) 24 parser.add_argument('--fuzzer', required=True)
24 args = parser.parse_args() 25 args = parser.parse_args()
25 26
26 corpus_files = [] 27 corpus_files = []
27 28
28 for (dirpath, _, filenames) in os.walk(args.corpus): 29 for (dirpath, _, filenames) in os.walk(args.corpus):
29 for filename in filenames: 30 for filename in filenames:
30 full_filename = os.path.join(dirpath, filename) 31 full_filename = os.path.join(dirpath, filename)
31 corpus_files.append(full_filename) 32 corpus_files.append(full_filename)
32 33
33 with zipfile.ZipFile(args.output, 'w') as z: 34 with zipfile.ZipFile(args.output, 'w') as z:
34 for corpus_file in corpus_files: 35 # Turn warnings into errors to interrupt the build: crbug.com/653920.
35 z.write(corpus_file, os.path.basename(corpus_file)) 36 with warnings.catch_warnings():
37 warnings.simplefilter("error")
38 for i, corpus_file in enumerate(corpus_files):
39 # To avoid duplication of filenames inside the archive, use numbers.
40 arcname = '%016d' % i
mmoroz 2016/11/29 13:53:41 I thought to use subdirectory_name + '_' + os.path
41 z.write(corpus_file, arcname)
36 42
37 43
38 if __name__ == '__main__': 44 if __name__ == '__main__':
39 main() 45 main()
OLDNEW
« no previous file with comments | « no previous file | no next file » | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698