0
Files
src/tools/code_coverage/merge_all_profdata.py
Julia Hansbrough 2f7fe7135b [fuzz coverage] Remote coverage.py dependency from merge_all_fuzzers.py.
The profdata file that coverage.py *exports* is different than what it
uses to generate coverage locally, causing a divergence in behavior of
the local coverage script and the recipe run for coverage in automation.

Plus, coverage.py is far too heavyweight for what we need anyway.

Replace coverage.py with a simple llvm-profdata merge command.

Change-Id: Icd8c8f52abb214399395e4d82a34f71fa2bc1329
Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/4657425
Reviewed-by: Prakhar Asthana <pasthana@google.com>
Commit-Queue: Julia Hansbrough <flowerhack@google.com>
Cr-Commit-Position: refs/heads/main@{#1166687}
2023-07-06 18:39:14 +00:00

62 lines
1.8 KiB
Python
Executable File

#!/usr/bin/env python3
# Copyright 2023 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
""" Merge all the profdata files in PROFDATA_DIR, utilizing the binaries in
BINARIES_DIR, to create a unified coverage report in REPORT_DIR. (If no
REPORT_DIR is provided, defaults to `out/report`.)
* Example usage: merge_all_profdata.py --profdata-dir [PROFDATA_DIR]
--binaries-dir [BINARIES_DIR] [--report-dir [REPORT_DIR]]
"""
import argparse
import logging
import os
import subprocess
def _ParseCommandArguments():
"""Adds and parses relevant arguments for tool commands.
Returns:
A dictionary representing the arguments.
"""
arg_parser = argparse.ArgumentParser()
arg_parser.usage = __doc__
arg_parser.add_argument('--llvm-profdata',
required=True,
type=str,
help='Location of LLVM profdata tool')
arg_parser.add_argument('--profdata-dir',
required=True,
type=str,
help='Directory in which profdata files are stored.')
arg_parser.add_argument(
'--outfile',
type=str,
required=True,
help=('Directory where the coverage report should go. '
'Default is out/report.'))
args = arg_parser.parse_args()
return args
args = _ParseCommandArguments()
targets_to_cover = []
for profdata_file in os.listdir(args.profdata_dir):
targets_to_cover.append(os.path.join(args.profdata_dir, profdata_file))
subprocess_cmd = [args.llvm_profdata, 'merge', '-o', args.outfile]
for target in targets_to_cover:
subprocess_cmd.append(target)
try:
subprocess.check_call(subprocess_cmd)
except:
logging.error("An error occured while merging the profdata.")
exit(1)