blob: 986b45731cf15fbb893552b17ff16f795166beb3 [file] [log] [blame]
#!/usr/bin/env python2.7
# Copyright 2020 The Fuchsia Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
import argparse
import os
import sys
def main():
parser = argparse.ArgumentParser("Create a manifest for a seed corpus")
parser.add_argument("--corpus", help="Path to corpus directory")
parser.add_argument("--depfile", help="Path to depfile", required=True)
parser.add_argument("--fuzzer", help="Name of fuzzer", required=True)
parser.add_argument("--manifest", help="Output manifest", required=True)
parser.add_argument("--unittest", help="Generated unit test", required=True)
args = parser.parse_args()
# Enumerate the corpus as a list of (srcpath, dstpath) pairs
elements = sorted(os.listdir(args.corpus)) if args.corpus else []
elements = [
os.path.join(args.corpus, element),
os.path.join('data', args.fuzzer, 'corpus', element))
for element in elements
# The manifest will be included in the package manifest to include corpus
# elements as resources.
with open(args.manifest, 'wb') as manifest:
for srcpath, dstpath in elements:
manifest.write('{}={}\n'.format(dstpath, srcpath))
# The 'unittest' argument is used to generate a source file that plugs into
# //src/lib/fuzzing/cpp/ to create a unit/regression test that
# exercises each corpus element.
with open(args.unittest, 'wb') as unittest:
// Autogenerated by //build/fuzzing/
#include <string>
#include <vector>
#include "src/lib/fuzzing/cpp/fuzzer_test.h"
namespace fuzzing {
std::vector<std::string> GetCorpus() {
return std::vector<std::string>{
for srcpath, dstpath in elements:
unittest.write(' "pkg/{}",\n'.format(dstpath))
} // namespace fuzzing
# The depfile is used by GN to trigger a re-packaging when the corpus
# changes.
with open(args.depfile, 'wb') as depfile:
depfile.write(args.manifest + ':')
for srcpath, dstpath in elements:
depfile.write(' ' + os.path.abspath(srcpath))
return 0
if __name__ == '__main__':