target_experiment.py uploads reproducer and binary in target_experiment.py

infra/build/functions/target_experiment.py

@@ -30,7 +30,7 @@
 
 def run_experiment(project_name, target_name, args, output_path,
                    build_output_path, upload_corpus_path, upload_coverage_path,
-                   experiment_name):
+                   experiment_name, upload_reproducer_path):
   config = build_project.Config(testing=True,
                                 test_image_suffix='',
                                 repo=build_project.DEFAULT_OSS_FUZZ_REPO,
@@ -74,8 +74,11 @@ def run_experiment(project_name, target_name, args, output_path,
   local_output_path = '/workspace/output.log'
   local_corpus_path_base = '/workspace/corpus'
   local_corpus_path = os.path.join(local_corpus_path_base, target_name)
+  default_target_path = os.path.join(build.out, target_name)
+  local_target_dir = os.path.join(build.out, 'target')
   local_corpus_zip_path = '/workspace/corpus/corpus.zip'
-  fuzzer_args = ' '.join(args)
+  local_artifact_path = os.path.join(build.out, 'artifacts/')
+  fuzzer_args = ' '.join(args + [f'-artifact_prefix={local_artifact_path}'])
 
   env = build_project.get_env(project_yaml['language'], build)
   env.append('RUN_FUZZER_MODE=batch')
@@ -90,6 +93,8 @@ def run_experiment(project_name, target_name, args, output_path,
           'bash',
           '-c',
           (f'mkdir -p {local_corpus_path} && '
+           f'mkdir -p {local_target_dir} && '
+           f'mkdir -p {local_artifact_path} && '
            f'run_fuzzer {target_name} {fuzzer_args} '
            f'|& tee {local_output_path} || true'),
       ]
@@ -115,6 +120,39 @@ def run_experiment(project_name, target_name, args, output_path,
       ],
   })
 
+  if upload_reproducer_path:
+    # Upload binary. First copy the binary directly from default_target_path,
+    # if that fails, find it under build out dir and copy to local_target_dir.
+    # If either succeeds, then upload it to bucket.
+    # If multiple files are found, suffix them and upload them all.
+    steps.append({
+        'name':
+            'gcr.io/cloud-builders/gsutil',
+        'entrypoint':
+            '/bin/bash',
+        'args': [
+            '-c',
+            (f'cp {default_target_path} {local_target_dir} 2>/dev/null || '
+             f'find {build.out} -type f -name {target_name} -exec bash -c '
+             f'\'cp "$0" "{local_target_dir}/$(echo "$0" | sed "s@/@_@g")"\' '
+             f'{{}} \\; && gsutil cp -r {local_target_dir} '
+             f'{upload_reproducer_path}/{target_name} || true'),
+        ],
+    })
+
+    # Upload reproducer.
+    steps.append({
+        'name':
+            'gcr.io/cloud-builders/gsutil',
+        'entrypoint':
+            '/bin/bash',
+        'args': [
+            '-c',
+            (f'gsutil -m cp -r {local_artifact_path} {upload_reproducer_path} '
+             '|| true'),
+        ],
+    })
+
   # Build for coverage.
   build = build_project.Build('libfuzzer', 'coverage', 'x86_64')
   env = build_project.get_env(project_yaml['language'], build)
@@ -213,6 +251,10 @@ def main():
   parser.add_argument('--upload_corpus',
                       required=True,
                       help='GCS location to upload corpus.')
+  parser.add_argument('--upload_reproducer',
+                      required=False,
+                      default='',
+                      help='GCS location to upload reproducer.')
   parser.add_argument('--upload_coverage',
                       required=True,
                       help='GCS location to upload coverage data.')
@@ -223,7 +265,8 @@ def main():
 
   run_experiment(args.project, args.target, args.args, args.upload_output_log,
                  args.upload_build_log, args.upload_corpus,
-                 args.upload_coverage, args.experiment_name)
+                 args.upload_coverage, args.experiment_name,
+                 args.upload_reproducer)
 
 
 if __name__ == '__main__':