blob: 7f45542a1911b83f48fc0e2d115959d67b529966 [file] [log] [blame]
Yuke Liao506e8822017-12-04 16:52:541#!/usr/bin/python
2# Copyright 2017 The Chromium Authors. All rights reserved.
3# Use of this source code is governed by a BSD-style license that can be
4# found in the LICENSE file.
Abhishek Arya1ec832c2017-12-05 18:06:595"""This script helps to generate code coverage report.
Yuke Liao506e8822017-12-04 16:52:546
Abhishek Arya1ec832c2017-12-05 18:06:597 It uses Clang Source-based Code Coverage -
8 https://clang.llvm.org/docs/SourceBasedCodeCoverage.html
Yuke Liao506e8822017-12-04 16:52:549
Abhishek Arya16f059a2017-12-07 17:47:3210 In order to generate code coverage report, you need to first add
Yuke Liaoab9c44e2018-02-21 00:24:4011 "use_clang_coverage=true" and "is_component_build=false" GN flags to args.gn
12 file in your build output directory (e.g. out/coverage).
Yuke Liao506e8822017-12-04 16:52:5413
Abhishek Arya1ec832c2017-12-05 18:06:5914 Example usage:
15
Abhishek Arya16f059a2017-12-07 17:47:3216 gn gen out/coverage --args='use_clang_coverage=true is_component_build=false'
17 gclient runhooks
Abhishek Arya1ec832c2017-12-05 18:06:5918 python tools/code_coverage/coverage.py crypto_unittests url_unittests \\
Abhishek Arya16f059a2017-12-07 17:47:3219 -b out/coverage -o out/report -c 'out/coverage/crypto_unittests' \\
20 -c 'out/coverage/url_unittests --gtest_filter=URLParser.PathURL' \\
21 -f url/ -f crypto/
Abhishek Arya1ec832c2017-12-05 18:06:5922
Abhishek Arya16f059a2017-12-07 17:47:3223 The command above builds crypto_unittests and url_unittests targets and then
24 runs them with specified command line arguments. For url_unittests, it only
25 runs the test URLParser.PathURL. The coverage report is filtered to include
26 only files and sub-directories under url/ and crypto/ directories.
Abhishek Arya1ec832c2017-12-05 18:06:5927
Yuke Liao545db322018-02-15 17:12:0128 If you want to run tests that try to draw to the screen but don't have a
29 display connected, you can run tests in headless mode with xvfb.
30
31 Sample flow for running a test target with xvfb (e.g. unit_tests):
32
33 python tools/code_coverage/coverage.py unit_tests -b out/coverage \\
34 -o out/report -c 'python testing/xvfb.py out/coverage/unit_tests'
35
Abhishek Arya1ec832c2017-12-05 18:06:5936 If you are building a fuzz target, you need to add "use_libfuzzer=true" GN
37 flag as well.
38
39 Sample workflow for a fuzz target (e.g. pdfium_fuzzer):
40
Abhishek Arya16f059a2017-12-07 17:47:3241 python tools/code_coverage/coverage.py pdfium_fuzzer \\
42 -b out/coverage -o out/report \\
43 -c 'out/coverage/pdfium_fuzzer -runs=<runs> <corpus_dir>' \\
44 -f third_party/pdfium
Abhishek Arya1ec832c2017-12-05 18:06:5945
46 where:
47 <corpus_dir> - directory containing samples files for this format.
48 <runs> - number of times to fuzz target function. Should be 0 when you just
49 want to see the coverage on corpus and don't want to fuzz at all.
50
51 For more options, please refer to tools/code_coverage/coverage.py -h.
Yuke Liao8e209fe82018-04-18 20:36:3852
53 For an overview of how code coverage works in Chromium, please refer to
54 https://chromium.googlesource.com/chromium/src/+/master/docs/code_coverage.md
Yuke Liao506e8822017-12-04 16:52:5455"""
56
57from __future__ import print_function
58
59import sys
60
61import argparse
Yuke Liaoea228d02018-01-05 19:10:3362import json
Yuke Liao481d3482018-01-29 19:17:1063import logging
Yuke Liao506e8822017-12-04 16:52:5464import os
Yuke Liaob2926832018-03-02 17:34:2965import re
66import shlex
Max Moroz025d8952018-05-03 16:33:3467import shutil
Yuke Liao506e8822017-12-04 16:52:5468import subprocess
Yuke Liao506e8822017-12-04 16:52:5469import urllib2
70
Abhishek Arya1ec832c2017-12-05 18:06:5971sys.path.append(
72 os.path.join(
73 os.path.dirname(__file__), os.path.pardir, os.path.pardir, 'tools',
74 'clang', 'scripts'))
Yuke Liao506e8822017-12-04 16:52:5475import update as clang_update
76
Yuke Liaoea228d02018-01-05 19:10:3377sys.path.append(
78 os.path.join(
79 os.path.dirname(__file__), os.path.pardir, os.path.pardir,
80 'third_party'))
81import jinja2
82from collections import defaultdict
83
Yuke Liao506e8822017-12-04 16:52:5484# Absolute path to the root of the checkout.
Abhishek Arya1ec832c2017-12-05 18:06:5985SRC_ROOT_PATH = os.path.abspath(
86 os.path.join(os.path.dirname(__file__), os.path.pardir, os.path.pardir))
Yuke Liao506e8822017-12-04 16:52:5487
88# Absolute path to the code coverage tools binary.
89LLVM_BUILD_DIR = clang_update.LLVM_BUILD_DIR
Abhishek Arya1c97ea542018-05-10 03:53:1990LLVM_BIN_DIR = os.path.join(LLVM_BUILD_DIR, 'bin')
91LLVM_COV_PATH = os.path.join(LLVM_BIN_DIR, 'llvm-cov')
92LLVM_PROFDATA_PATH = os.path.join(LLVM_BIN_DIR, 'llvm-profdata')
Yuke Liao506e8822017-12-04 16:52:5493
94# Build directory, the value is parsed from command line arguments.
95BUILD_DIR = None
96
97# Output directory for generated artifacts, the value is parsed from command
98# line arguemnts.
99OUTPUT_DIR = None
100
101# Default number of jobs used to build when goma is configured and enabled.
102DEFAULT_GOMA_JOBS = 100
103
104# Name of the file extension for profraw data files.
105PROFRAW_FILE_EXTENSION = 'profraw'
106
107# Name of the final profdata file, and this file needs to be passed to
108# "llvm-cov" command in order to call "llvm-cov show" to inspect the
109# line-by-line coverage of specific files.
Max Moroz7c5354f2018-05-06 00:03:48110PROFDATA_FILE_NAME = os.extsep.join(['coverage', 'profdata'])
111
112# Name of the file with summary information generated by llvm-cov export.
113SUMMARY_FILE_NAME = os.extsep.join(['summary', 'json'])
Yuke Liao506e8822017-12-04 16:52:54114
115# Build arg required for generating code coverage data.
116CLANG_COVERAGE_BUILD_ARG = 'use_clang_coverage'
117
Yuke Liaoea228d02018-01-05 19:10:33118# The default name of the html coverage report for a directory.
119DIRECTORY_COVERAGE_HTML_REPORT_NAME = os.extsep.join(['report', 'html'])
120
Yuke Liaodd1ec0592018-02-02 01:26:37121# Name of the html index files for different views.
Yuke Liaodd1ec0592018-02-02 01:26:37122COMPONENT_VIEW_INDEX_FILE = os.extsep.join(['component_view_index', 'html'])
Max Moroz7c5354f2018-05-06 00:03:48123DIRECTORY_VIEW_INDEX_FILE = os.extsep.join(['directory_view_index', 'html'])
Yuke Liaodd1ec0592018-02-02 01:26:37124FILE_VIEW_INDEX_FILE = os.extsep.join(['file_view_index', 'html'])
Max Moroz7c5354f2018-05-06 00:03:48125INDEX_HTML_FILE = os.extsep.join(['index', 'html'])
126
127LOGS_DIR_NAME = 'logs'
Yuke Liaodd1ec0592018-02-02 01:26:37128
129# Used to extract a mapping between directories and components.
Abhishek Arya1c97ea542018-05-10 03:53:19130COMPONENT_MAPPING_URL = (
131 'https://storage.googleapis.com/chromium-owners/component_map.json')
Yuke Liaodd1ec0592018-02-02 01:26:37132
Yuke Liao80afff32018-03-07 01:26:20133# Caches the results returned by _GetBuildArgs, don't use this variable
134# directly, call _GetBuildArgs instead.
135_BUILD_ARGS = None
136
Abhishek Aryac19bc5ef2018-05-04 22:10:02137# Retry failed merges.
138MERGE_RETRIES = 3
139
Abhishek Aryad35de7e2018-05-10 22:23:04140# Message to guide user to file a bug when everything else fails.
141FILE_BUG_MESSAGE = (
142 'If it persists, please file a bug with the command you used, git revision '
143 'and args.gn config here: '
144 'https://bugs.chromium.org/p/chromium/issues/entry?'
145 'components=Tools%3ECodeCoverage')
146
Yuke Liaoea228d02018-01-05 19:10:33147
148class _CoverageSummary(object):
149 """Encapsulates coverage summary representation."""
150
Yuke Liaodd1ec0592018-02-02 01:26:37151 def __init__(self,
152 regions_total=0,
153 regions_covered=0,
154 functions_total=0,
155 functions_covered=0,
156 lines_total=0,
157 lines_covered=0):
Yuke Liaoea228d02018-01-05 19:10:33158 """Initializes _CoverageSummary object."""
159 self._summary = {
160 'regions': {
161 'total': regions_total,
162 'covered': regions_covered
163 },
164 'functions': {
165 'total': functions_total,
166 'covered': functions_covered
167 },
168 'lines': {
169 'total': lines_total,
170 'covered': lines_covered
171 }
172 }
173
174 def Get(self):
175 """Returns summary as a dictionary."""
176 return self._summary
177
178 def AddSummary(self, other_summary):
179 """Adds another summary to this one element-wise."""
180 for feature in self._summary:
181 self._summary[feature]['total'] += other_summary.Get()[feature]['total']
182 self._summary[feature]['covered'] += other_summary.Get()[feature][
183 'covered']
184
185
Yuke Liaodd1ec0592018-02-02 01:26:37186class _CoverageReportHtmlGenerator(object):
187 """Encapsulates coverage html report generation.
Yuke Liaoea228d02018-01-05 19:10:33188
Yuke Liaodd1ec0592018-02-02 01:26:37189 The generated html has a table that contains links to other coverage reports.
Yuke Liaoea228d02018-01-05 19:10:33190 """
191
Yuke Liaodd1ec0592018-02-02 01:26:37192 def __init__(self, output_path, table_entry_type):
193 """Initializes _CoverageReportHtmlGenerator object.
194
195 Args:
196 output_path: Path to the html report that will be generated.
197 table_entry_type: Type of the table entries to be displayed in the table
198 header. For example: 'Path', 'Component'.
199 """
Yuke Liaoea228d02018-01-05 19:10:33200 css_file_name = os.extsep.join(['style', 'css'])
Max Moroz7c5354f2018-05-06 00:03:48201 css_absolute_path = os.path.join(OUTPUT_DIR, css_file_name)
Yuke Liaoea228d02018-01-05 19:10:33202 assert os.path.exists(css_absolute_path), (
203 'css file doesn\'t exit. Please make sure "llvm-cov show -format=html" '
Abhishek Aryafb70b532018-05-06 17:47:40204 'is called first, and the css file is generated at: "%s".' %
Yuke Liaoea228d02018-01-05 19:10:33205 css_absolute_path)
206
207 self._css_absolute_path = css_absolute_path
Yuke Liaodd1ec0592018-02-02 01:26:37208 self._output_path = output_path
209 self._table_entry_type = table_entry_type
210
Yuke Liaoea228d02018-01-05 19:10:33211 self._table_entries = []
Yuke Liaod54030e2018-01-08 17:34:12212 self._total_entry = {}
Abhishek Arya302b67a2018-05-10 19:43:23213
214 source_dir = os.path.dirname(os.path.realpath(__file__))
215 template_dir = os.path.join(source_dir, 'html_templates')
Yuke Liaoea228d02018-01-05 19:10:33216
217 jinja_env = jinja2.Environment(
218 loader=jinja2.FileSystemLoader(template_dir), trim_blocks=True)
219 self._header_template = jinja_env.get_template('header.html')
220 self._table_template = jinja_env.get_template('table.html')
221 self._footer_template = jinja_env.get_template('footer.html')
Abhishek Arya302b67a2018-05-10 19:43:23222
Abhishek Arya865fffd2018-05-08 22:16:01223 self._style_overrides = open(
Abhishek Arya302b67a2018-05-10 19:43:23224 os.path.join(source_dir, 'static', 'css', 'style.css')).read()
Yuke Liaoea228d02018-01-05 19:10:33225
226 def AddLinkToAnotherReport(self, html_report_path, name, summary):
227 """Adds a link to another html report in this report.
228
229 The link to be added is assumed to be an entry in this directory.
230 """
Yuke Liaodd1ec0592018-02-02 01:26:37231 # Use relative paths instead of absolute paths to make the generated reports
232 # portable.
233 html_report_relative_path = _GetRelativePathToDirectoryOfFile(
234 html_report_path, self._output_path)
235
Yuke Liaod54030e2018-01-08 17:34:12236 table_entry = self._CreateTableEntryFromCoverageSummary(
Yuke Liaodd1ec0592018-02-02 01:26:37237 summary, html_report_relative_path, name,
Yuke Liaod54030e2018-01-08 17:34:12238 os.path.basename(html_report_path) ==
239 DIRECTORY_COVERAGE_HTML_REPORT_NAME)
240 self._table_entries.append(table_entry)
241
242 def CreateTotalsEntry(self, summary):
Yuke Liaoa785f4d32018-02-13 21:41:35243 """Creates an entry corresponds to the 'Totals' row in the html report."""
Yuke Liaod54030e2018-01-08 17:34:12244 self._total_entry = self._CreateTableEntryFromCoverageSummary(summary)
245
246 def _CreateTableEntryFromCoverageSummary(self,
247 summary,
248 href=None,
249 name=None,
250 is_dir=None):
251 """Creates an entry to display in the html report."""
Yuke Liaodd1ec0592018-02-02 01:26:37252 assert (href is None and name is None and is_dir is None) or (
253 href is not None and name is not None and is_dir is not None), (
254 'The only scenario when href or name or is_dir can be None is when '
Yuke Liaoa785f4d32018-02-13 21:41:35255 'creating an entry for the Totals row, and in that case, all three '
Yuke Liaodd1ec0592018-02-02 01:26:37256 'attributes must be None.')
257
Yuke Liaod54030e2018-01-08 17:34:12258 entry = {}
Yuke Liaodd1ec0592018-02-02 01:26:37259 if href is not None:
260 entry['href'] = href
261 if name is not None:
262 entry['name'] = name
263 if is_dir is not None:
264 entry['is_dir'] = is_dir
265
Yuke Liaoea228d02018-01-05 19:10:33266 summary_dict = summary.Get()
Yuke Liaod54030e2018-01-08 17:34:12267 for feature in summary_dict:
Yuke Liaodd1ec0592018-02-02 01:26:37268 if summary_dict[feature]['total'] == 0:
269 percentage = 0.0
270 else:
Yuke Liao0e4c8682018-04-18 21:06:59271 percentage = float(summary_dict[feature]
272 ['covered']) / summary_dict[feature]['total'] * 100
Yuke Liaoa785f4d32018-02-13 21:41:35273
Yuke Liaoea228d02018-01-05 19:10:33274 color_class = self._GetColorClass(percentage)
Yuke Liaod54030e2018-01-08 17:34:12275 entry[feature] = {
Yuke Liaoea228d02018-01-05 19:10:33276 'total': summary_dict[feature]['total'],
277 'covered': summary_dict[feature]['covered'],
Yuke Liaoa785f4d32018-02-13 21:41:35278 'percentage': '{:6.2f}'.format(percentage),
Yuke Liaoea228d02018-01-05 19:10:33279 'color_class': color_class
280 }
Yuke Liaod54030e2018-01-08 17:34:12281
Yuke Liaod54030e2018-01-08 17:34:12282 return entry
Yuke Liaoea228d02018-01-05 19:10:33283
284 def _GetColorClass(self, percentage):
285 """Returns the css color class based on coverage percentage."""
286 if percentage >= 0 and percentage < 80:
287 return 'red'
288 if percentage >= 80 and percentage < 100:
289 return 'yellow'
290 if percentage == 100:
291 return 'green'
292
Abhishek Aryafb70b532018-05-06 17:47:40293 assert False, 'Invalid coverage percentage: "%d".' % percentage
Yuke Liaoea228d02018-01-05 19:10:33294
Yuke Liao1b852fd2018-05-11 17:07:32295 def WriteHtmlCoverageReport(self, no_file_view):
Yuke Liaodd1ec0592018-02-02 01:26:37296 """Writes html coverage report.
Yuke Liaoea228d02018-01-05 19:10:33297
298 In the report, sub-directories are displayed before files and within each
299 category, entries are sorted alphabetically.
Yuke Liaoea228d02018-01-05 19:10:33300 """
301
302 def EntryCmp(left, right):
303 """Compare function for table entries."""
304 if left['is_dir'] != right['is_dir']:
305 return -1 if left['is_dir'] == True else 1
306
Yuke Liaodd1ec0592018-02-02 01:26:37307 return -1 if left['name'] < right['name'] else 1
Yuke Liaoea228d02018-01-05 19:10:33308
309 self._table_entries = sorted(self._table_entries, cmp=EntryCmp)
310
311 css_path = os.path.join(OUTPUT_DIR, os.extsep.join(['style', 'css']))
Max Moroz7c5354f2018-05-06 00:03:48312
313 directory_view_path = _GetDirectoryViewPath()
Yuke Liao1b852fd2018-05-11 17:07:32314 directory_view_href = _GetRelativePathToDirectoryOfFile(
315 directory_view_path, self._output_path)
Max Moroz7c5354f2018-05-06 00:03:48316 component_view_path = _GetComponentViewPath()
Yuke Liao1b852fd2018-05-11 17:07:32317 component_view_href = _GetRelativePathToDirectoryOfFile(
318 component_view_path, self._output_path)
319
320 # File view is optional in the report.
321 file_view_href = None
322 if not no_file_view:
323 file_view_path = _GetFileViewPath()
324 file_view_href = _GetRelativePathToDirectoryOfFile(
325 file_view_path, self._output_path)
Yuke Liaodd1ec0592018-02-02 01:26:37326
Yuke Liaoea228d02018-01-05 19:10:33327 html_header = self._header_template.render(
Yuke Liaodd1ec0592018-02-02 01:26:37328 css_path=_GetRelativePathToDirectoryOfFile(css_path, self._output_path),
Yuke Liao1b852fd2018-05-11 17:07:32329 directory_view_href=directory_view_href,
330 component_view_href=component_view_href,
331 file_view_href=file_view_href,
Abhishek Arya865fffd2018-05-08 22:16:01332 style_overrides=self._style_overrides)
Yuke Liaodd1ec0592018-02-02 01:26:37333
Yuke Liaod54030e2018-01-08 17:34:12334 html_table = self._table_template.render(
Yuke Liaodd1ec0592018-02-02 01:26:37335 entries=self._table_entries,
336 total_entry=self._total_entry,
337 table_entry_type=self._table_entry_type)
Yuke Liaoea228d02018-01-05 19:10:33338 html_footer = self._footer_template.render()
339
Yuke Liaodd1ec0592018-02-02 01:26:37340 with open(self._output_path, 'w') as html_file:
Yuke Liaoea228d02018-01-05 19:10:33341 html_file.write(html_header + html_table + html_footer)
342
Yuke Liao506e8822017-12-04 16:52:54343
Abhishek Arya64636af2018-05-04 14:42:13344def _ConfigureLogging(args):
345 """Configures logging settings for later use."""
346 log_level = logging.DEBUG if args.verbose else logging.INFO
347 log_format = '[%(asctime)s %(levelname)s] %(message)s'
348 log_file = args.log_file if args.log_file else None
349 logging.basicConfig(filename=log_file, level=log_level, format=log_format)
350
351
Max Morozd73e45f2018-04-24 18:32:47352def _GetSharedLibraries(binary_paths):
Abhishek Arya78120bc2018-05-07 20:53:54353 """Returns list of shared libraries used by specified binaries."""
354 logging.info('Finding shared libraries for targets (if any).')
355 shared_libraries = []
Max Morozd73e45f2018-04-24 18:32:47356 cmd = []
357 shared_library_re = None
358
359 if sys.platform.startswith('linux'):
360 cmd.extend(['ldd'])
Abhishek Arya64636af2018-05-04 14:42:13361 shared_library_re = re.compile(r'.*\.so\s=>\s(.*' + BUILD_DIR +
362 r'.*\.so)\s.*')
Max Morozd73e45f2018-04-24 18:32:47363 elif sys.platform.startswith('darwin'):
364 cmd.extend(['otool', '-L'])
365 shared_library_re = re.compile(r'\s+(@rpath/.*\.dylib)\s.*')
366 else:
Abhishek Aryafb70b532018-05-06 17:47:40367 assert False, 'Cannot detect shared libraries used by the given targets.'
Max Morozd73e45f2018-04-24 18:32:47368
369 assert shared_library_re is not None
370
371 cmd.extend(binary_paths)
372 output = subprocess.check_output(cmd)
373
374 for line in output.splitlines():
375 m = shared_library_re.match(line)
376 if not m:
377 continue
378
379 shared_library_path = m.group(1)
380 if sys.platform.startswith('darwin'):
381 # otool outputs "@rpath" macro instead of the dirname of the given binary.
382 shared_library_path = shared_library_path.replace('@rpath', BUILD_DIR)
383
Abhishek Arya78120bc2018-05-07 20:53:54384 if shared_library_path in shared_libraries:
385 continue
386
Max Morozd73e45f2018-04-24 18:32:47387 assert os.path.exists(shared_library_path), ('Shared library "%s" used by '
388 'the given target(s) does not '
389 'exist.' % shared_library_path)
390 with open(shared_library_path) as f:
391 data = f.read()
392
393 # Do not add non-instrumented libraries. Otherwise, llvm-cov errors outs.
394 if '__llvm_cov' in data:
Abhishek Arya78120bc2018-05-07 20:53:54395 shared_libraries.append(shared_library_path)
Max Morozd73e45f2018-04-24 18:32:47396
Abhishek Arya78120bc2018-05-07 20:53:54397 logging.debug('Found shared libraries (%d): %s.', len(shared_libraries),
398 shared_libraries)
399 logging.info('Finished finding shared libraries for targets.')
400 return shared_libraries
Max Morozd73e45f2018-04-24 18:32:47401
402
Yuke Liaoc60b2d02018-03-02 21:40:43403def _GetHostPlatform():
404 """Returns the host platform.
405
406 This is separate from the target platform/os that coverage is running for.
407 """
Abhishek Arya1ec832c2017-12-05 18:06:59408 if sys.platform == 'win32' or sys.platform == 'cygwin':
409 return 'win'
410 if sys.platform.startswith('linux'):
411 return 'linux'
412 else:
413 assert sys.platform == 'darwin'
414 return 'mac'
415
416
Abhishek Arya1c97ea542018-05-10 03:53:19417def _GetPathWithLLVMSymbolizerDir():
418 """Add llvm-symbolizer directory to path for symbolized stacks."""
419 path = os.getenv('PATH')
420 dirs = path.split(os.pathsep)
421 if LLVM_BIN_DIR in dirs:
422 return path
423
424 return path + os.pathsep + LLVM_BIN_DIR
425
426
Yuke Liaoc60b2d02018-03-02 21:40:43427def _GetTargetOS():
428 """Returns the target os specified in args.gn file.
429
430 Returns an empty string is target_os is not specified.
431 """
Yuke Liao80afff32018-03-07 01:26:20432 build_args = _GetBuildArgs()
Yuke Liaoc60b2d02018-03-02 21:40:43433 return build_args['target_os'] if 'target_os' in build_args else ''
434
435
Yuke Liaob2926832018-03-02 17:34:29436def _IsIOS():
Yuke Liaoa0c8c2f2018-02-28 20:14:10437 """Returns true if the target_os specified in args.gn file is ios"""
Yuke Liaoc60b2d02018-03-02 21:40:43438 return _GetTargetOS() == 'ios'
Yuke Liaoa0c8c2f2018-02-28 20:14:10439
440
Yuke Liao506e8822017-12-04 16:52:54441# TODO(crbug.com/759794): remove this function once tools get included to
442# Clang bundle:
443# https://chromium-review.googlesource.com/c/chromium/src/+/688221
444def DownloadCoverageToolsIfNeeded():
445 """Temporary solution to download llvm-profdata and llvm-cov tools."""
Abhishek Arya1ec832c2017-12-05 18:06:59446
Yuke Liaoc60b2d02018-03-02 21:40:43447 def _GetRevisionFromStampFile(stamp_file_path):
Yuke Liao506e8822017-12-04 16:52:54448 """Returns a pair of revision number by reading the build stamp file.
449
450 Args:
451 stamp_file_path: A path the build stamp file created by
452 tools/clang/scripts/update.py.
453 Returns:
454 A pair of integers represeting the main and sub revision respectively.
455 """
456 if not os.path.exists(stamp_file_path):
457 return 0, 0
458
459 with open(stamp_file_path) as stamp_file:
Yuke Liaoc60b2d02018-03-02 21:40:43460 stamp_file_line = stamp_file.readline()
461 if ',' in stamp_file_line:
462 package_version = stamp_file_line.rstrip().split(',')[0]
463 else:
464 package_version = stamp_file_line.rstrip()
Yuke Liao506e8822017-12-04 16:52:54465
Yuke Liaoc60b2d02018-03-02 21:40:43466 clang_revision_str, clang_sub_revision_str = package_version.split('-')
467 return int(clang_revision_str), int(clang_sub_revision_str)
Abhishek Arya1ec832c2017-12-05 18:06:59468
Yuke Liaoc60b2d02018-03-02 21:40:43469 host_platform = _GetHostPlatform()
Yuke Liao506e8822017-12-04 16:52:54470 clang_revision, clang_sub_revision = _GetRevisionFromStampFile(
Yuke Liaoc60b2d02018-03-02 21:40:43471 clang_update.STAMP_FILE)
Yuke Liao506e8822017-12-04 16:52:54472
473 coverage_revision_stamp_file = os.path.join(
474 os.path.dirname(clang_update.STAMP_FILE), 'cr_coverage_revision')
475 coverage_revision, coverage_sub_revision = _GetRevisionFromStampFile(
Yuke Liaoc60b2d02018-03-02 21:40:43476 coverage_revision_stamp_file)
Yuke Liao506e8822017-12-04 16:52:54477
Yuke Liaoea228d02018-01-05 19:10:33478 has_coverage_tools = (
479 os.path.exists(LLVM_COV_PATH) and os.path.exists(LLVM_PROFDATA_PATH))
Abhishek Arya16f059a2017-12-07 17:47:32480
Yuke Liaoea228d02018-01-05 19:10:33481 if (has_coverage_tools and coverage_revision == clang_revision and
Yuke Liao506e8822017-12-04 16:52:54482 coverage_sub_revision == clang_sub_revision):
483 # LLVM coverage tools are up to date, bail out.
Yuke Liaoc60b2d02018-03-02 21:40:43484 return
Yuke Liao506e8822017-12-04 16:52:54485
486 package_version = '%d-%d' % (clang_revision, clang_sub_revision)
487 coverage_tools_file = 'llvm-code-coverage-%s.tgz' % package_version
488
489 # The code bellow follows the code from tools/clang/scripts/update.py.
Yuke Liaoc60b2d02018-03-02 21:40:43490 if host_platform == 'mac':
Yuke Liao506e8822017-12-04 16:52:54491 coverage_tools_url = clang_update.CDS_URL + '/Mac/' + coverage_tools_file
Yuke Liaoc60b2d02018-03-02 21:40:43492 elif host_platform == 'linux':
Yuke Liao506e8822017-12-04 16:52:54493 coverage_tools_url = (
494 clang_update.CDS_URL + '/Linux_x64/' + coverage_tools_file)
Yuke Liaoc60b2d02018-03-02 21:40:43495 else:
496 assert host_platform == 'win'
497 coverage_tools_url = (clang_update.CDS_URL + '/Win/' + coverage_tools_file)
Yuke Liao506e8822017-12-04 16:52:54498
499 try:
500 clang_update.DownloadAndUnpack(coverage_tools_url,
501 clang_update.LLVM_BUILD_DIR)
Yuke Liao506e8822017-12-04 16:52:54502 with open(coverage_revision_stamp_file, 'w') as file_handle:
Yuke Liaoc60b2d02018-03-02 21:40:43503 file_handle.write('%s,%s' % (package_version, host_platform))
Yuke Liao506e8822017-12-04 16:52:54504 file_handle.write('\n')
505 except urllib2.URLError:
506 raise Exception(
507 'Failed to download coverage tools: %s.' % coverage_tools_url)
508
509
Yuke Liaodd1ec0592018-02-02 01:26:37510def _GeneratePerFileLineByLineCoverageInHtml(binary_paths, profdata_file_path,
Yuke Liao0e4c8682018-04-18 21:06:59511 filters, ignore_filename_regex):
Yuke Liao506e8822017-12-04 16:52:54512 """Generates per file line-by-line coverage in html using 'llvm-cov show'.
513
514 For a file with absolute path /a/b/x.cc, a html report is generated as:
515 OUTPUT_DIR/coverage/a/b/x.cc.html. An index html file is also generated as:
516 OUTPUT_DIR/index.html.
517
518 Args:
519 binary_paths: A list of paths to the instrumented binaries.
520 profdata_file_path: A path to the profdata file.
Yuke Liao66da1732017-12-05 22:19:42521 filters: A list of directories and files to get coverage for.
Yuke Liao506e8822017-12-04 16:52:54522 """
Yuke Liao506e8822017-12-04 16:52:54523 # llvm-cov show [options] -instr-profile PROFILE BIN [-object BIN,...]
524 # [[-object BIN]] [SOURCES]
525 # NOTE: For object files, the first one is specified as a positional argument,
526 # and the rest are specified as keyword argument.
Yuke Liao481d3482018-01-29 19:17:10527 logging.debug('Generating per file line by line coverage reports using '
Abhishek Aryafb70b532018-05-06 17:47:40528 '"llvm-cov show" command.')
Abhishek Arya1ec832c2017-12-05 18:06:59529 subprocess_cmd = [
530 LLVM_COV_PATH, 'show', '-format=html',
531 '-output-dir={}'.format(OUTPUT_DIR),
532 '-instr-profile={}'.format(profdata_file_path), binary_paths[0]
533 ]
534 subprocess_cmd.extend(
535 ['-object=' + binary_path for binary_path in binary_paths[1:]])
Yuke Liaob2926832018-03-02 17:34:29536 _AddArchArgumentForIOSIfNeeded(subprocess_cmd, len(binary_paths))
Yuke Liao66da1732017-12-05 22:19:42537 subprocess_cmd.extend(filters)
Yuke Liao0e4c8682018-04-18 21:06:59538 if ignore_filename_regex:
539 subprocess_cmd.append('-ignore-filename-regex=%s' % ignore_filename_regex)
540
Yuke Liao506e8822017-12-04 16:52:54541 subprocess.check_call(subprocess_cmd)
Max Moroz025d8952018-05-03 16:33:34542
543 # llvm-cov creates "coverage" subdir in the output dir. We would like to use
544 # the platform name instead, as it simplifies the report dir structure when
545 # the same report is generated for different platforms.
546 default_report_subdir_path = os.path.join(OUTPUT_DIR, 'coverage')
Max Moroz7c5354f2018-05-06 00:03:48547 platform_report_subdir_path = _GetCoverageReportRootDirPath()
548 _MergeTwoDirectories(default_report_subdir_path, platform_report_subdir_path)
Max Moroz025d8952018-05-03 16:33:34549
Abhishek Aryafb70b532018-05-06 17:47:40550 logging.debug('Finished running "llvm-cov show" command.')
Yuke Liao506e8822017-12-04 16:52:54551
552
Yuke Liaodd1ec0592018-02-02 01:26:37553def _GenerateFileViewHtmlIndexFile(per_file_coverage_summary):
554 """Generates html index file for file view."""
Max Moroz7c5354f2018-05-06 00:03:48555 file_view_index_file_path = _GetFileViewPath()
Yuke Liaodd1ec0592018-02-02 01:26:37556 logging.debug('Generating file view html index file as: "%s".',
557 file_view_index_file_path)
558 html_generator = _CoverageReportHtmlGenerator(file_view_index_file_path,
559 'Path')
560 totals_coverage_summary = _CoverageSummary()
Yuke Liaoea228d02018-01-05 19:10:33561
Yuke Liaodd1ec0592018-02-02 01:26:37562 for file_path in per_file_coverage_summary:
563 totals_coverage_summary.AddSummary(per_file_coverage_summary[file_path])
564
565 html_generator.AddLinkToAnotherReport(
566 _GetCoverageHtmlReportPathForFile(file_path),
567 os.path.relpath(file_path, SRC_ROOT_PATH),
568 per_file_coverage_summary[file_path])
569
570 html_generator.CreateTotalsEntry(totals_coverage_summary)
Yuke Liao1b852fd2018-05-11 17:07:32571 html_generator.WriteHtmlCoverageReport(no_file_view=False)
Yuke Liaodd1ec0592018-02-02 01:26:37572 logging.debug('Finished generating file view html index file.')
573
574
575def _CalculatePerDirectoryCoverageSummary(per_file_coverage_summary):
576 """Calculates per directory coverage summary."""
Abhishek Aryafb70b532018-05-06 17:47:40577 logging.debug('Calculating per-directory coverage summary.')
Yuke Liaodd1ec0592018-02-02 01:26:37578 per_directory_coverage_summary = defaultdict(lambda: _CoverageSummary())
579
Yuke Liaoea228d02018-01-05 19:10:33580 for file_path in per_file_coverage_summary:
581 summary = per_file_coverage_summary[file_path]
582 parent_dir = os.path.dirname(file_path)
Abhishek Aryafb70b532018-05-06 17:47:40583
Yuke Liaoea228d02018-01-05 19:10:33584 while True:
585 per_directory_coverage_summary[parent_dir].AddSummary(summary)
586
587 if parent_dir == SRC_ROOT_PATH:
588 break
589 parent_dir = os.path.dirname(parent_dir)
590
Abhishek Aryafb70b532018-05-06 17:47:40591 logging.debug('Finished calculating per-directory coverage summary.')
Yuke Liaodd1ec0592018-02-02 01:26:37592 return per_directory_coverage_summary
593
594
Yuke Liao1b852fd2018-05-11 17:07:32595def _GeneratePerDirectoryCoverageInHtml(
596 per_directory_coverage_summary, per_file_coverage_summary, no_file_view):
Yuke Liaodd1ec0592018-02-02 01:26:37597 """Generates per directory coverage breakdown in html."""
Abhishek Aryafb70b532018-05-06 17:47:40598 logging.debug('Writing per-directory coverage html reports.')
Yuke Liaoea228d02018-01-05 19:10:33599 for dir_path in per_directory_coverage_summary:
Yuke Liao1b852fd2018-05-11 17:07:32600 _GenerateCoverageInHtmlForDirectory(dir_path,
601 per_directory_coverage_summary,
602 per_file_coverage_summary, no_file_view)
Yuke Liaoea228d02018-01-05 19:10:33603
Abhishek Aryafb70b532018-05-06 17:47:40604 logging.debug('Finished writing per-directory coverage html reports.')
Yuke Liao481d3482018-01-29 19:17:10605
Yuke Liaoea228d02018-01-05 19:10:33606
607def _GenerateCoverageInHtmlForDirectory(
Yuke Liao1b852fd2018-05-11 17:07:32608 dir_path, per_directory_coverage_summary, per_file_coverage_summary,
609 no_file_view):
Yuke Liaoea228d02018-01-05 19:10:33610 """Generates coverage html report for a single directory."""
Yuke Liaodd1ec0592018-02-02 01:26:37611 html_generator = _CoverageReportHtmlGenerator(
612 _GetCoverageHtmlReportPathForDirectory(dir_path), 'Path')
Yuke Liaoea228d02018-01-05 19:10:33613
614 for entry_name in os.listdir(dir_path):
615 entry_path = os.path.normpath(os.path.join(dir_path, entry_name))
Yuke Liaoea228d02018-01-05 19:10:33616
Yuke Liaodd1ec0592018-02-02 01:26:37617 if entry_path in per_file_coverage_summary:
618 entry_html_report_path = _GetCoverageHtmlReportPathForFile(entry_path)
619 entry_coverage_summary = per_file_coverage_summary[entry_path]
620 elif entry_path in per_directory_coverage_summary:
621 entry_html_report_path = _GetCoverageHtmlReportPathForDirectory(
622 entry_path)
623 entry_coverage_summary = per_directory_coverage_summary[entry_path]
624 else:
Yuke Liaoc7e607142018-02-05 20:26:14625 # Any file without executable lines shouldn't be included into the report.
626 # For example, OWNER and README.md files.
Yuke Liaodd1ec0592018-02-02 01:26:37627 continue
Yuke Liaoea228d02018-01-05 19:10:33628
Yuke Liaodd1ec0592018-02-02 01:26:37629 html_generator.AddLinkToAnotherReport(entry_html_report_path,
630 os.path.basename(entry_path),
631 entry_coverage_summary)
Yuke Liaoea228d02018-01-05 19:10:33632
Yuke Liaod54030e2018-01-08 17:34:12633 html_generator.CreateTotalsEntry(per_directory_coverage_summary[dir_path])
Yuke Liao1b852fd2018-05-11 17:07:32634 html_generator.WriteHtmlCoverageReport(no_file_view)
Yuke Liaodd1ec0592018-02-02 01:26:37635
636
637def _GenerateDirectoryViewHtmlIndexFile():
638 """Generates the html index file for directory view.
639
640 Note that the index file is already generated under SRC_ROOT_PATH, so this
641 file simply redirects to it, and the reason of this extra layer is for
642 structural consistency with other views.
643 """
Max Moroz7c5354f2018-05-06 00:03:48644 directory_view_index_file_path = _GetDirectoryViewPath()
Yuke Liaodd1ec0592018-02-02 01:26:37645 logging.debug('Generating directory view html index file as: "%s".',
646 directory_view_index_file_path)
647 src_root_html_report_path = _GetCoverageHtmlReportPathForDirectory(
648 SRC_ROOT_PATH)
649 _WriteRedirectHtmlFile(directory_view_index_file_path,
650 src_root_html_report_path)
651 logging.debug('Finished generating directory view html index file.')
652
653
654def _CalculatePerComponentCoverageSummary(component_to_directories,
655 per_directory_coverage_summary):
656 """Calculates per component coverage summary."""
Abhishek Aryafb70b532018-05-06 17:47:40657 logging.debug('Calculating per-component coverage summary.')
Yuke Liaodd1ec0592018-02-02 01:26:37658 per_component_coverage_summary = defaultdict(lambda: _CoverageSummary())
659
660 for component in component_to_directories:
661 for directory in component_to_directories[component]:
662 absolute_directory_path = os.path.abspath(directory)
663 if absolute_directory_path in per_directory_coverage_summary:
664 per_component_coverage_summary[component].AddSummary(
665 per_directory_coverage_summary[absolute_directory_path])
666
Abhishek Aryafb70b532018-05-06 17:47:40667 logging.debug('Finished calculating per-component coverage summary.')
Yuke Liaodd1ec0592018-02-02 01:26:37668 return per_component_coverage_summary
669
670
671def _ExtractComponentToDirectoriesMapping():
672 """Returns a mapping from components to directories."""
673 component_mappings = json.load(urllib2.urlopen(COMPONENT_MAPPING_URL))
674 directory_to_component = component_mappings['dir-to-component']
675
676 component_to_directories = defaultdict(list)
Abhishek Arya8c3a1ce322018-05-13 04:14:01677 for directory in sorted(directory_to_component):
Yuke Liaodd1ec0592018-02-02 01:26:37678 component = directory_to_component[directory]
Abhishek Arya8c3a1ce322018-05-13 04:14:01679
680 # Check if we already added the parent directory of this directory. If yes,
681 # skip this sub-directory to avoid double-counting.
682 found_parent_directory = False
683 for component_directory in component_to_directories[component]:
684 if directory.startswith(component_directory + '/'):
685 found_parent_directory = True
686 break
687
688 if not found_parent_directory:
689 component_to_directories[component].append(directory)
Yuke Liaodd1ec0592018-02-02 01:26:37690
691 return component_to_directories
692
693
Yuke Liao1b852fd2018-05-11 17:07:32694def _GeneratePerComponentCoverageInHtml(
695 per_component_coverage_summary, component_to_directories,
696 per_directory_coverage_summary, no_file_view):
Yuke Liaodd1ec0592018-02-02 01:26:37697 """Generates per-component coverage reports in html."""
698 logging.debug('Writing per-component coverage html reports.')
699 for component in per_component_coverage_summary:
700 _GenerateCoverageInHtmlForComponent(
701 component, per_component_coverage_summary, component_to_directories,
Yuke Liao1b852fd2018-05-11 17:07:32702 per_directory_coverage_summary, no_file_view)
Yuke Liaodd1ec0592018-02-02 01:26:37703
704 logging.debug('Finished writing per-component coverage html reports.')
705
706
707def _GenerateCoverageInHtmlForComponent(
708 component_name, per_component_coverage_summary, component_to_directories,
Yuke Liao1b852fd2018-05-11 17:07:32709 per_directory_coverage_summary, no_file_view):
Yuke Liaodd1ec0592018-02-02 01:26:37710 """Generates coverage html report for a component."""
711 component_html_report_path = _GetCoverageHtmlReportPathForComponent(
712 component_name)
Yuke Liaoc7e607142018-02-05 20:26:14713 component_html_report_dir = os.path.dirname(component_html_report_path)
714 if not os.path.exists(component_html_report_dir):
715 os.makedirs(component_html_report_dir)
Yuke Liaodd1ec0592018-02-02 01:26:37716
717 html_generator = _CoverageReportHtmlGenerator(component_html_report_path,
718 'Path')
719
720 for dir_path in component_to_directories[component_name]:
721 dir_absolute_path = os.path.abspath(dir_path)
722 if dir_absolute_path not in per_directory_coverage_summary:
Yuke Liaoc7e607142018-02-05 20:26:14723 # Any directory without an excercised file shouldn't be included into the
724 # report.
Yuke Liaodd1ec0592018-02-02 01:26:37725 continue
726
727 html_generator.AddLinkToAnotherReport(
728 _GetCoverageHtmlReportPathForDirectory(dir_path),
729 os.path.relpath(dir_path, SRC_ROOT_PATH),
730 per_directory_coverage_summary[dir_absolute_path])
731
732 html_generator.CreateTotalsEntry(
733 per_component_coverage_summary[component_name])
Yuke Liao1b852fd2018-05-11 17:07:32734 html_generator.WriteHtmlCoverageReport(no_file_view)
Yuke Liaodd1ec0592018-02-02 01:26:37735
736
Yuke Liao1b852fd2018-05-11 17:07:32737def _GenerateComponentViewHtmlIndexFile(per_component_coverage_summary,
738 no_file_view):
Yuke Liaodd1ec0592018-02-02 01:26:37739 """Generates the html index file for component view."""
Max Moroz7c5354f2018-05-06 00:03:48740 component_view_index_file_path = _GetComponentViewPath()
Yuke Liaodd1ec0592018-02-02 01:26:37741 logging.debug('Generating component view html index file as: "%s".',
742 component_view_index_file_path)
743 html_generator = _CoverageReportHtmlGenerator(component_view_index_file_path,
744 'Component')
Yuke Liaodd1ec0592018-02-02 01:26:37745 for component in per_component_coverage_summary:
Yuke Liaodd1ec0592018-02-02 01:26:37746 html_generator.AddLinkToAnotherReport(
747 _GetCoverageHtmlReportPathForComponent(component), component,
748 per_component_coverage_summary[component])
749
Abhishek Aryaefbe1df2018-05-14 20:19:48750 # Do not create a totals row for the component view as the value is incorrect
751 # due to failure to account for UNKNOWN component and some paths belonging to
752 # multiple components.
753
Yuke Liao1b852fd2018-05-11 17:07:32754 html_generator.WriteHtmlCoverageReport(no_file_view)
Yuke Liaoc7e607142018-02-05 20:26:14755 logging.debug('Finished generating component view html index file.')
Yuke Liaoea228d02018-01-05 19:10:33756
757
Max Moroz7c5354f2018-05-06 00:03:48758def _MergeTwoDirectories(src_path, dst_path):
759 """Merge src_path directory into dst_path directory."""
760 for filename in os.listdir(src_path):
761 dst_path = os.path.join(dst_path, filename)
762 if os.path.exists(dst_path):
763 shutil.rmtree(dst_path)
764 os.rename(os.path.join(src_path, filename), dst_path)
765 shutil.rmtree(src_path)
766
767
Yuke Liaoea228d02018-01-05 19:10:33768def _OverwriteHtmlReportsIndexFile():
Yuke Liaodd1ec0592018-02-02 01:26:37769 """Overwrites the root index file to redirect to the default view."""
Max Moroz7c5354f2018-05-06 00:03:48770 html_index_file_path = _GetHtmlIndexPath()
771 directory_view_index_file_path = _GetDirectoryViewPath()
Yuke Liaodd1ec0592018-02-02 01:26:37772 _WriteRedirectHtmlFile(html_index_file_path, directory_view_index_file_path)
773
774
775def _WriteRedirectHtmlFile(from_html_path, to_html_path):
776 """Writes a html file that redirects to another html file."""
777 to_html_relative_path = _GetRelativePathToDirectoryOfFile(
778 to_html_path, from_html_path)
Yuke Liaoea228d02018-01-05 19:10:33779 content = ("""
780 <!DOCTYPE html>
781 <html>
782 <head>
783 <!-- HTML meta refresh URL redirection -->
784 <meta http-equiv="refresh" content="0; url=%s">
785 </head>
Yuke Liaodd1ec0592018-02-02 01:26:37786 </html>""" % to_html_relative_path)
787 with open(from_html_path, 'w') as f:
Yuke Liaoea228d02018-01-05 19:10:33788 f.write(content)
789
790
Max Moroz7c5354f2018-05-06 00:03:48791def _CleanUpOutputDir():
792 """Perform a cleanup of the output dir."""
793 # Remove the default index.html file produced by llvm-cov.
794 index_path = os.path.join(OUTPUT_DIR, INDEX_HTML_FILE)
795 if os.path.exists(index_path):
796 os.remove(index_path)
797
798
Yuke Liaodd1ec0592018-02-02 01:26:37799def _GetCoverageHtmlReportPathForFile(file_path):
800 """Given a file path, returns the corresponding html report path."""
Abhishek Aryafb70b532018-05-06 17:47:40801 assert os.path.isfile(file_path), '"%s" is not a file.' % file_path
Yuke Liaodd1ec0592018-02-02 01:26:37802 html_report_path = os.extsep.join([os.path.abspath(file_path), 'html'])
803
804 # '+' is used instead of os.path.join because both of them are absolute paths
805 # and os.path.join ignores the first path.
Yuke Liaoc7e607142018-02-05 20:26:14806 # TODO(crbug.com/809150): Think of a generic cross platform fix (Windows).
Yuke Liaodd1ec0592018-02-02 01:26:37807 return _GetCoverageReportRootDirPath() + html_report_path
808
809
810def _GetCoverageHtmlReportPathForDirectory(dir_path):
811 """Given a directory path, returns the corresponding html report path."""
Abhishek Aryafb70b532018-05-06 17:47:40812 assert os.path.isdir(dir_path), '"%s" is not a directory.' % dir_path
Yuke Liaodd1ec0592018-02-02 01:26:37813 html_report_path = os.path.join(
814 os.path.abspath(dir_path), DIRECTORY_COVERAGE_HTML_REPORT_NAME)
815
816 # '+' is used instead of os.path.join because both of them are absolute paths
817 # and os.path.join ignores the first path.
Yuke Liaoc7e607142018-02-05 20:26:14818 # TODO(crbug.com/809150): Think of a generic cross platform fix (Windows).
Yuke Liaodd1ec0592018-02-02 01:26:37819 return _GetCoverageReportRootDirPath() + html_report_path
820
821
822def _GetCoverageHtmlReportPathForComponent(component_name):
823 """Given a component, returns the corresponding html report path."""
824 component_file_name = component_name.lower().replace('>', '-')
825 html_report_name = os.extsep.join([component_file_name, 'html'])
826 return os.path.join(_GetCoverageReportRootDirPath(), 'components',
827 html_report_name)
828
829
830def _GetCoverageReportRootDirPath():
831 """The root directory that contains all generated coverage html reports."""
Max Moroz7c5354f2018-05-06 00:03:48832 return os.path.join(OUTPUT_DIR, _GetHostPlatform())
833
834
835def _GetComponentViewPath():
836 """Path to the HTML file for the component view."""
837 return os.path.join(_GetCoverageReportRootDirPath(),
838 COMPONENT_VIEW_INDEX_FILE)
839
840
841def _GetDirectoryViewPath():
842 """Path to the HTML file for the directory view."""
843 return os.path.join(_GetCoverageReportRootDirPath(),
844 DIRECTORY_VIEW_INDEX_FILE)
845
846
847def _GetFileViewPath():
848 """Path to the HTML file for the file view."""
849 return os.path.join(_GetCoverageReportRootDirPath(), FILE_VIEW_INDEX_FILE)
850
851
852def _GetLogsDirectoryPath():
853 """Path to the logs directory."""
854 return os.path.join(_GetCoverageReportRootDirPath(), LOGS_DIR_NAME)
855
856
857def _GetHtmlIndexPath():
858 """Path to the main HTML index file."""
859 return os.path.join(_GetCoverageReportRootDirPath(), INDEX_HTML_FILE)
860
861
862def _GetProfdataFilePath():
863 """Path to the resulting .profdata file."""
864 return os.path.join(_GetCoverageReportRootDirPath(), PROFDATA_FILE_NAME)
865
866
867def _GetSummaryFilePath():
868 """The JSON file that contains coverage summary written by llvm-cov export."""
869 return os.path.join(_GetCoverageReportRootDirPath(), SUMMARY_FILE_NAME)
Yuke Liaoea228d02018-01-05 19:10:33870
871
Yuke Liao506e8822017-12-04 16:52:54872def _CreateCoverageProfileDataForTargets(targets, commands, jobs_count=None):
873 """Builds and runs target to generate the coverage profile data.
874
875 Args:
876 targets: A list of targets to build with coverage instrumentation.
877 commands: A list of commands used to run the targets.
878 jobs_count: Number of jobs to run in parallel for building. If None, a
879 default value is derived based on CPUs availability.
880
881 Returns:
882 A relative path to the generated profdata file.
883 """
884 _BuildTargets(targets, jobs_count)
Abhishek Aryac19bc5ef2018-05-04 22:10:02885 target_profdata_file_paths = _GetTargetProfDataPathsByExecutingCommands(
Abhishek Arya1ec832c2017-12-05 18:06:59886 targets, commands)
Abhishek Aryac19bc5ef2018-05-04 22:10:02887 coverage_profdata_file_path = (
888 _CreateCoverageProfileDataFromTargetProfDataFiles(
889 target_profdata_file_paths))
Yuke Liao506e8822017-12-04 16:52:54890
Abhishek Aryac19bc5ef2018-05-04 22:10:02891 for target_profdata_file_path in target_profdata_file_paths:
892 os.remove(target_profdata_file_path)
Yuke Liaod4a9865202018-01-12 23:17:52893
Abhishek Aryac19bc5ef2018-05-04 22:10:02894 return coverage_profdata_file_path
Yuke Liao506e8822017-12-04 16:52:54895
896
897def _BuildTargets(targets, jobs_count):
898 """Builds target with Clang coverage instrumentation.
899
900 This function requires current working directory to be the root of checkout.
901
902 Args:
903 targets: A list of targets to build with coverage instrumentation.
904 jobs_count: Number of jobs to run in parallel for compilation. If None, a
905 default value is derived based on CPUs availability.
Yuke Liao506e8822017-12-04 16:52:54906 """
Abhishek Arya1ec832c2017-12-05 18:06:59907
Yuke Liao506e8822017-12-04 16:52:54908 def _IsGomaConfigured():
909 """Returns True if goma is enabled in the gn build args.
910
911 Returns:
912 A boolean indicates whether goma is configured for building or not.
913 """
Yuke Liao80afff32018-03-07 01:26:20914 build_args = _GetBuildArgs()
Yuke Liao506e8822017-12-04 16:52:54915 return 'use_goma' in build_args and build_args['use_goma'] == 'true'
916
Abhishek Aryafb70b532018-05-06 17:47:40917 logging.info('Building %s.', str(targets))
Yuke Liao506e8822017-12-04 16:52:54918 if jobs_count is None and _IsGomaConfigured():
919 jobs_count = DEFAULT_GOMA_JOBS
920
921 subprocess_cmd = ['ninja', '-C', BUILD_DIR]
922 if jobs_count is not None:
923 subprocess_cmd.append('-j' + str(jobs_count))
924
925 subprocess_cmd.extend(targets)
926 subprocess.check_call(subprocess_cmd)
Abhishek Aryafb70b532018-05-06 17:47:40927 logging.debug('Finished building %s.', str(targets))
Yuke Liao506e8822017-12-04 16:52:54928
929
Abhishek Aryac19bc5ef2018-05-04 22:10:02930def _GetTargetProfDataPathsByExecutingCommands(targets, commands):
Yuke Liao506e8822017-12-04 16:52:54931 """Runs commands and returns the relative paths to the profraw data files.
932
933 Args:
934 targets: A list of targets built with coverage instrumentation.
935 commands: A list of commands used to run the targets.
936
937 Returns:
938 A list of relative paths to the generated profraw data files.
939 """
Abhishek Aryafb70b532018-05-06 17:47:40940 logging.debug('Executing the test commands.')
Yuke Liao481d3482018-01-29 19:17:10941
Yuke Liao506e8822017-12-04 16:52:54942 # Remove existing profraw data files.
Max Moroz7c5354f2018-05-06 00:03:48943 for file_or_dir in os.listdir(_GetCoverageReportRootDirPath()):
Yuke Liao506e8822017-12-04 16:52:54944 if file_or_dir.endswith(PROFRAW_FILE_EXTENSION):
Max Moroz7c5354f2018-05-06 00:03:48945 os.remove(os.path.join(_GetCoverageReportRootDirPath(), file_or_dir))
946
947 # Ensure that logs directory exists.
948 if not os.path.exists(_GetLogsDirectoryPath()):
949 os.makedirs(_GetLogsDirectoryPath())
Yuke Liao506e8822017-12-04 16:52:54950
Abhishek Aryac19bc5ef2018-05-04 22:10:02951 profdata_file_paths = []
Yuke Liaoa0c8c2f2018-02-28 20:14:10952
Yuke Liaod4a9865202018-01-12 23:17:52953 # Run all test targets to generate profraw data files.
Yuke Liao506e8822017-12-04 16:52:54954 for target, command in zip(targets, commands):
Max Moroz7c5354f2018-05-06 00:03:48955 output_file_name = os.extsep.join([target + '_output', 'log'])
956 output_file_path = os.path.join(_GetLogsDirectoryPath(), output_file_name)
Yuke Liaoa0c8c2f2018-02-28 20:14:10957
Abhishek Aryac19bc5ef2018-05-04 22:10:02958 profdata_file_path = None
959 for _ in xrange(MERGE_RETRIES):
Abhishek Aryafb70b532018-05-06 17:47:40960 logging.info('Running command: "%s", the output is redirected to "%s".',
Abhishek Aryac19bc5ef2018-05-04 22:10:02961 command, output_file_path)
Yuke Liaoa0c8c2f2018-02-28 20:14:10962
Abhishek Aryac19bc5ef2018-05-04 22:10:02963 if _IsIOSCommand(command):
964 # On iOS platform, due to lack of write permissions, profraw files are
965 # generated outside of the OUTPUT_DIR, and the exact paths are contained
966 # in the output of the command execution.
967 output = _ExecuteIOSCommand(target, command)
968 else:
969 # On other platforms, profraw files are generated inside the OUTPUT_DIR.
970 output = _ExecuteCommand(target, command)
971
972 with open(output_file_path, 'w') as output_file:
973 output_file.write(output)
974
975 profraw_file_paths = []
976 if _IsIOS():
977 profraw_file_paths = _GetProfrawDataFileByParsingOutput(output)
978 else:
Max Moroz7c5354f2018-05-06 00:03:48979 for file_or_dir in os.listdir(_GetCoverageReportRootDirPath()):
Abhishek Aryac19bc5ef2018-05-04 22:10:02980 if file_or_dir.endswith(PROFRAW_FILE_EXTENSION):
Max Moroz7c5354f2018-05-06 00:03:48981 profraw_file_paths.append(
982 os.path.join(_GetCoverageReportRootDirPath(), file_or_dir))
Abhishek Aryac19bc5ef2018-05-04 22:10:02983
984 assert profraw_file_paths, (
Abhishek Aryafb70b532018-05-06 17:47:40985 'Running target "%s" failed to generate any profraw data file, '
Abhishek Aryad35de7e2018-05-10 22:23:04986 'please make sure the binary exists, is properly instrumented and '
987 'does not crash. %s' % (target, FILE_BUG_MESSAGE))
Abhishek Aryac19bc5ef2018-05-04 22:10:02988
989 try:
990 profdata_file_path = _CreateTargetProfDataFileFromProfRawFiles(
991 target, profraw_file_paths)
992 break
993 except Exception:
Abhishek Aryad35de7e2018-05-10 22:23:04994 logging.info('Retrying...')
Abhishek Aryac19bc5ef2018-05-04 22:10:02995 finally:
996 # Remove profraw files now so that they are not used in next iteration.
997 for profraw_file_path in profraw_file_paths:
998 os.remove(profraw_file_path)
999
1000 assert profdata_file_path, (
Abhishek Aryad35de7e2018-05-10 22:23:041001 'Failed to merge target "%s" profraw files after %d retries. %s' %
1002 (target, MERGE_RETRIES, FILE_BUG_MESSAGE))
Abhishek Aryac19bc5ef2018-05-04 22:10:021003 profdata_file_paths.append(profdata_file_path)
Yuke Liao506e8822017-12-04 16:52:541004
Abhishek Aryafb70b532018-05-06 17:47:401005 logging.debug('Finished executing the test commands.')
Yuke Liao481d3482018-01-29 19:17:101006
Abhishek Aryac19bc5ef2018-05-04 22:10:021007 return profdata_file_paths
Yuke Liao506e8822017-12-04 16:52:541008
1009
1010def _ExecuteCommand(target, command):
Yuke Liaoa0c8c2f2018-02-28 20:14:101011 """Runs a single command and generates a profraw data file."""
Yuke Liaod4a9865202018-01-12 23:17:521012 # Per Clang "Source-based Code Coverage" doc:
Yuke Liao27349c92018-03-22 21:10:011013 #
Max Morozd73e45f2018-04-24 18:32:471014 # "%p" expands out to the process ID. It's not used by this scripts due to:
1015 # 1) If a target program spawns too many processess, it may exhaust all disk
1016 # space available. For example, unit_tests writes thousands of .profraw
1017 # files each of size 1GB+.
1018 # 2) If a target binary uses shared libraries, coverage profile data for them
1019 # will be missing, resulting in incomplete coverage reports.
Yuke Liao27349c92018-03-22 21:10:011020 #
Yuke Liaod4a9865202018-01-12 23:17:521021 # "%Nm" expands out to the instrumented binary's signature. When this pattern
1022 # is specified, the runtime creates a pool of N raw profiles which are used
1023 # for on-line profile merging. The runtime takes care of selecting a raw
1024 # profile from the pool, locking it, and updating it before the program exits.
Yuke Liaod4a9865202018-01-12 23:17:521025 # N must be between 1 and 9. The merge pool specifier can only occur once per
1026 # filename pattern.
1027 #
Max Morozd73e45f2018-04-24 18:32:471028 # "%1m" is used when tests run in single process, such as fuzz targets.
Yuke Liao27349c92018-03-22 21:10:011029 #
Max Morozd73e45f2018-04-24 18:32:471030 # For other cases, "%4m" is chosen as it creates some level of parallelism,
1031 # but it's not too big to consume too much computing resource or disk space.
1032 profile_pattern_string = '%1m' if _IsFuzzerTarget(target) else '%4m'
Abhishek Arya1ec832c2017-12-05 18:06:591033 expected_profraw_file_name = os.extsep.join(
Yuke Liao27349c92018-03-22 21:10:011034 [target, profile_pattern_string, PROFRAW_FILE_EXTENSION])
Max Moroz7c5354f2018-05-06 00:03:481035 expected_profraw_file_path = os.path.join(_GetCoverageReportRootDirPath(),
Yuke Liao506e8822017-12-04 16:52:541036 expected_profraw_file_name)
Yuke Liao506e8822017-12-04 16:52:541037
Yuke Liaoa0c8c2f2018-02-28 20:14:101038 try:
Max Moroz7c5354f2018-05-06 00:03:481039 # Some fuzz targets or tests may write into stderr, redirect it as well.
Yuke Liaoa0c8c2f2018-02-28 20:14:101040 output = subprocess.check_output(
Yuke Liaob2926832018-03-02 17:34:291041 shlex.split(command),
Max Moroz7c5354f2018-05-06 00:03:481042 stderr=subprocess.STDOUT,
Abhishek Arya1c97ea542018-05-10 03:53:191043 env={
1044 'LLVM_PROFILE_FILE': expected_profraw_file_path,
1045 'PATH': _GetPathWithLLVMSymbolizerDir()
1046 })
Yuke Liaoa0c8c2f2018-02-28 20:14:101047 except subprocess.CalledProcessError as e:
1048 output = e.output
Abhishek Arya1c97ea542018-05-10 03:53:191049 logging.warning(
1050 'Command: "%s" exited with non-zero return code. Output:\n%s', command,
1051 output)
Yuke Liaoa0c8c2f2018-02-28 20:14:101052
1053 return output
1054
1055
Yuke Liao27349c92018-03-22 21:10:011056def _IsFuzzerTarget(target):
1057 """Returns true if the target is a fuzzer target."""
1058 build_args = _GetBuildArgs()
1059 use_libfuzzer = ('use_libfuzzer' in build_args and
1060 build_args['use_libfuzzer'] == 'true')
1061 return use_libfuzzer and target.endswith('_fuzzer')
1062
1063
Yuke Liaob2926832018-03-02 17:34:291064def _ExecuteIOSCommand(target, command):
Yuke Liaoa0c8c2f2018-02-28 20:14:101065 """Runs a single iOS command and generates a profraw data file.
1066
1067 iOS application doesn't have write access to folders outside of the app, so
1068 it's impossible to instruct the app to flush the profraw data file to the
1069 desired location. The profraw data file will be generated somewhere within the
1070 application's Documents folder, and the full path can be obtained by parsing
1071 the output.
1072 """
Yuke Liaob2926832018-03-02 17:34:291073 assert _IsIOSCommand(command)
1074
1075 # After running tests, iossim generates a profraw data file, it won't be
1076 # needed anyway, so dump it into the OUTPUT_DIR to avoid polluting the
1077 # checkout.
1078 iossim_profraw_file_path = os.path.join(
1079 OUTPUT_DIR, os.extsep.join(['iossim', PROFRAW_FILE_EXTENSION]))
Yuke Liaoa0c8c2f2018-02-28 20:14:101080
1081 try:
Yuke Liaob2926832018-03-02 17:34:291082 output = subprocess.check_output(
1083 shlex.split(command),
Abhishek Arya1c97ea542018-05-10 03:53:191084 env={
1085 'LLVM_PROFILE_FILE': iossim_profraw_file_path,
1086 'PATH': _GetPathWithLLVMSymbolizerDir()
1087 })
Yuke Liaoa0c8c2f2018-02-28 20:14:101088 except subprocess.CalledProcessError as e:
1089 # iossim emits non-zero return code even if tests run successfully, so
1090 # ignore the return code.
1091 output = e.output
1092
1093 return output
1094
1095
1096def _GetProfrawDataFileByParsingOutput(output):
1097 """Returns the path to the profraw data file obtained by parsing the output.
1098
1099 The output of running the test target has no format, but it is guaranteed to
1100 have a single line containing the path to the generated profraw data file.
1101 NOTE: This should only be called when target os is iOS.
1102 """
Yuke Liaob2926832018-03-02 17:34:291103 assert _IsIOS()
Yuke Liaoa0c8c2f2018-02-28 20:14:101104
Yuke Liaob2926832018-03-02 17:34:291105 output_by_lines = ''.join(output).splitlines()
1106 profraw_file_pattern = re.compile('.*Coverage data at (.*coverage\.profraw).')
Yuke Liaoa0c8c2f2018-02-28 20:14:101107
1108 for line in output_by_lines:
Yuke Liaob2926832018-03-02 17:34:291109 result = profraw_file_pattern.match(line)
1110 if result:
1111 return result.group(1)
Yuke Liaoa0c8c2f2018-02-28 20:14:101112
1113 assert False, ('No profraw data file was generated, did you call '
1114 'coverage_util::ConfigureCoverageReportPath() in test setup? '
1115 'Please refer to base/test/test_support_ios.mm for example.')
Yuke Liao506e8822017-12-04 16:52:541116
1117
Abhishek Aryac19bc5ef2018-05-04 22:10:021118def _CreateCoverageProfileDataFromTargetProfDataFiles(profdata_file_paths):
1119 """Returns a relative path to coverage profdata file by merging target
1120 profdata files.
Yuke Liao506e8822017-12-04 16:52:541121
1122 Args:
Abhishek Aryac19bc5ef2018-05-04 22:10:021123 profdata_file_paths: A list of relative paths to the profdata data files
1124 that are to be merged.
Yuke Liao506e8822017-12-04 16:52:541125
1126 Returns:
Abhishek Aryac19bc5ef2018-05-04 22:10:021127 A relative path to the merged coverage profdata file.
Yuke Liao506e8822017-12-04 16:52:541128
1129 Raises:
Abhishek Aryac19bc5ef2018-05-04 22:10:021130 CalledProcessError: An error occurred merging profdata files.
Yuke Liao506e8822017-12-04 16:52:541131 """
Abhishek Aryafb70b532018-05-06 17:47:401132 logging.info('Creating the coverage profile data file.')
1133 logging.debug('Merging target profraw files to create target profdata file.')
Max Moroz7c5354f2018-05-06 00:03:481134 profdata_file_path = _GetProfdataFilePath()
Yuke Liao506e8822017-12-04 16:52:541135 try:
Abhishek Arya1ec832c2017-12-05 18:06:591136 subprocess_cmd = [
1137 LLVM_PROFDATA_PATH, 'merge', '-o', profdata_file_path, '-sparse=true'
1138 ]
Abhishek Aryac19bc5ef2018-05-04 22:10:021139 subprocess_cmd.extend(profdata_file_paths)
1140 subprocess.check_call(subprocess_cmd)
1141 except subprocess.CalledProcessError as error:
Abhishek Aryad35de7e2018-05-10 22:23:041142 logging.error(
1143 'Failed to merge target profdata files to create coverage profdata. %s',
1144 FILE_BUG_MESSAGE)
Abhishek Aryac19bc5ef2018-05-04 22:10:021145 raise error
1146
Abhishek Aryafb70b532018-05-06 17:47:401147 logging.debug('Finished merging target profdata files.')
1148 logging.info('Code coverage profile data is created as: "%s".',
Abhishek Aryac19bc5ef2018-05-04 22:10:021149 profdata_file_path)
1150 return profdata_file_path
1151
1152
1153def _CreateTargetProfDataFileFromProfRawFiles(target, profraw_file_paths):
1154 """Returns a relative path to target profdata file by merging target
1155 profraw files.
1156
1157 Args:
1158 profraw_file_paths: A list of relative paths to the profdata data files
1159 that are to be merged.
1160
1161 Returns:
1162 A relative path to the merged coverage profdata file.
1163
1164 Raises:
1165 CalledProcessError: An error occurred merging profdata files.
1166 """
Abhishek Aryafb70b532018-05-06 17:47:401167 logging.info('Creating target profile data file.')
1168 logging.debug('Merging target profraw files to create target profdata file.')
Abhishek Aryac19bc5ef2018-05-04 22:10:021169 profdata_file_path = os.path.join(OUTPUT_DIR, '%s.profdata' % target)
1170
1171 try:
1172 subprocess_cmd = [
1173 LLVM_PROFDATA_PATH, 'merge', '-o', profdata_file_path, '-sparse=true'
1174 ]
Yuke Liao506e8822017-12-04 16:52:541175 subprocess_cmd.extend(profraw_file_paths)
1176 subprocess.check_call(subprocess_cmd)
1177 except subprocess.CalledProcessError as error:
Abhishek Aryad35de7e2018-05-10 22:23:041178 logging.error(
1179 'Failed to merge target profraw files to create target profdata.')
Yuke Liao506e8822017-12-04 16:52:541180 raise error
1181
Abhishek Aryafb70b532018-05-06 17:47:401182 logging.debug('Finished merging target profraw files.')
1183 logging.info('Target "%s" profile data is created as: "%s".', target,
Yuke Liao481d3482018-01-29 19:17:101184 profdata_file_path)
Yuke Liao506e8822017-12-04 16:52:541185 return profdata_file_path
1186
1187
Yuke Liao0e4c8682018-04-18 21:06:591188def _GeneratePerFileCoverageSummary(binary_paths, profdata_file_path, filters,
1189 ignore_filename_regex):
Yuke Liaoea228d02018-01-05 19:10:331190 """Generates per file coverage summary using "llvm-cov export" command."""
1191 # llvm-cov export [options] -instr-profile PROFILE BIN [-object BIN,...]
1192 # [[-object BIN]] [SOURCES].
1193 # NOTE: For object files, the first one is specified as a positional argument,
1194 # and the rest are specified as keyword argument.
Yuke Liao481d3482018-01-29 19:17:101195 logging.debug('Generating per-file code coverage summary using "llvm-cov '
Abhishek Aryafb70b532018-05-06 17:47:401196 'export -summary-only" command.')
Yuke Liaoea228d02018-01-05 19:10:331197 subprocess_cmd = [
1198 LLVM_COV_PATH, 'export', '-summary-only',
1199 '-instr-profile=' + profdata_file_path, binary_paths[0]
1200 ]
1201 subprocess_cmd.extend(
1202 ['-object=' + binary_path for binary_path in binary_paths[1:]])
Yuke Liaob2926832018-03-02 17:34:291203 _AddArchArgumentForIOSIfNeeded(subprocess_cmd, len(binary_paths))
Yuke Liaoea228d02018-01-05 19:10:331204 subprocess_cmd.extend(filters)
Yuke Liao0e4c8682018-04-18 21:06:591205 if ignore_filename_regex:
1206 subprocess_cmd.append('-ignore-filename-regex=%s' % ignore_filename_regex)
Yuke Liaoea228d02018-01-05 19:10:331207
Max Moroz7c5354f2018-05-06 00:03:481208 export_output = subprocess.check_output(subprocess_cmd)
1209
1210 # Write output on the disk to be used by code coverage bot.
1211 with open(_GetSummaryFilePath(), 'w') as f:
1212 f.write(export_output)
1213
1214 json_output = json.loads(export_output)
Yuke Liaoea228d02018-01-05 19:10:331215 assert len(json_output['data']) == 1
1216 files_coverage_data = json_output['data'][0]['files']
1217
1218 per_file_coverage_summary = {}
1219 for file_coverage_data in files_coverage_data:
1220 file_path = file_coverage_data['filename']
Abhishek Aryafb70b532018-05-06 17:47:401221 assert file_path.startswith(SRC_ROOT_PATH + os.sep), (
1222 'File path "%s" in coverage summary is outside source checkout.' %
1223 file_path)
Yuke Liaoea228d02018-01-05 19:10:331224
Abhishek Aryafb70b532018-05-06 17:47:401225 summary = file_coverage_data['summary']
Yuke Liaoea228d02018-01-05 19:10:331226 if summary['lines']['count'] == 0:
1227 continue
1228
1229 per_file_coverage_summary[file_path] = _CoverageSummary(
1230 regions_total=summary['regions']['count'],
1231 regions_covered=summary['regions']['covered'],
1232 functions_total=summary['functions']['count'],
1233 functions_covered=summary['functions']['covered'],
1234 lines_total=summary['lines']['count'],
1235 lines_covered=summary['lines']['covered'])
1236
Abhishek Aryafb70b532018-05-06 17:47:401237 logging.debug('Finished generating per-file code coverage summary.')
Yuke Liaoea228d02018-01-05 19:10:331238 return per_file_coverage_summary
1239
1240
Yuke Liaob2926832018-03-02 17:34:291241def _AddArchArgumentForIOSIfNeeded(cmd_list, num_archs):
1242 """Appends -arch arguments to the command list if it's ios platform.
1243
1244 iOS binaries are universal binaries, and require specifying the architecture
1245 to use, and one architecture needs to be specified for each binary.
1246 """
1247 if _IsIOS():
1248 cmd_list.extend(['-arch=x86_64'] * num_archs)
1249
1250
Yuke Liao506e8822017-12-04 16:52:541251def _GetBinaryPath(command):
1252 """Returns a relative path to the binary to be run by the command.
1253
Yuke Liao545db322018-02-15 17:12:011254 Currently, following types of commands are supported (e.g. url_unittests):
1255 1. Run test binary direcly: "out/coverage/url_unittests <arguments>"
1256 2. Use xvfb.
1257 2.1. "python testing/xvfb.py out/coverage/url_unittests <arguments>"
1258 2.2. "testing/xvfb.py out/coverage/url_unittests <arguments>"
Yuke Liao92107f02018-03-07 01:44:371259 3. Use iossim to run tests on iOS platform, please refer to testing/iossim.mm
1260 for its usage.
Yuke Liaoa0c8c2f2018-02-28 20:14:101261 3.1. "out/Coverage-iphonesimulator/iossim
Yuke Liao92107f02018-03-07 01:44:371262 <iossim_arguments> -c <app_arguments>
1263 out/Coverage-iphonesimulator/url_unittests.app"
1264
Yuke Liao545db322018-02-15 17:12:011265
Yuke Liao506e8822017-12-04 16:52:541266 Args:
1267 command: A command used to run a target.
1268
1269 Returns:
1270 A relative path to the binary.
1271 """
Yuke Liao545db322018-02-15 17:12:011272 xvfb_script_name = os.extsep.join(['xvfb', 'py'])
1273
Yuke Liaob2926832018-03-02 17:34:291274 command_parts = shlex.split(command)
Yuke Liao545db322018-02-15 17:12:011275 if os.path.basename(command_parts[0]) == 'python':
1276 assert os.path.basename(command_parts[1]) == xvfb_script_name, (
Abhishek Aryafb70b532018-05-06 17:47:401277 'This tool doesn\'t understand the command: "%s".' % command)
Yuke Liao545db322018-02-15 17:12:011278 return command_parts[2]
1279
1280 if os.path.basename(command_parts[0]) == xvfb_script_name:
1281 return command_parts[1]
1282
Yuke Liaob2926832018-03-02 17:34:291283 if _IsIOSCommand(command):
Yuke Liaoa0c8c2f2018-02-28 20:14:101284 # For a given application bundle, the binary resides in the bundle and has
1285 # the same name with the application without the .app extension.
Yuke Liao92107f02018-03-07 01:44:371286 app_path = command_parts[-1].rstrip(os.path.sep)
Yuke Liaoa0c8c2f2018-02-28 20:14:101287 app_name = os.path.splitext(os.path.basename(app_path))[0]
1288 return os.path.join(app_path, app_name)
1289
Yuke Liaob2926832018-03-02 17:34:291290 return command_parts[0]
Yuke Liao506e8822017-12-04 16:52:541291
1292
Yuke Liaob2926832018-03-02 17:34:291293def _IsIOSCommand(command):
Yuke Liaoa0c8c2f2018-02-28 20:14:101294 """Returns true if command is used to run tests on iOS platform."""
Yuke Liaob2926832018-03-02 17:34:291295 return os.path.basename(shlex.split(command)[0]) == 'iossim'
Yuke Liaoa0c8c2f2018-02-28 20:14:101296
1297
Yuke Liao95d13d72017-12-07 18:18:501298def _VerifyTargetExecutablesAreInBuildDirectory(commands):
1299 """Verifies that the target executables specified in the commands are inside
1300 the given build directory."""
Yuke Liao506e8822017-12-04 16:52:541301 for command in commands:
1302 binary_path = _GetBinaryPath(command)
Yuke Liao95d13d72017-12-07 18:18:501303 binary_absolute_path = os.path.abspath(os.path.normpath(binary_path))
Max Moroz7c5354f2018-05-06 00:03:481304 assert binary_absolute_path.startswith(BUILD_DIR), (
Yuke Liao95d13d72017-12-07 18:18:501305 'Target executable "%s" in command: "%s" is outside of '
1306 'the given build directory: "%s".' % (binary_path, command, BUILD_DIR))
Yuke Liao506e8822017-12-04 16:52:541307
1308
1309def _ValidateBuildingWithClangCoverage():
1310 """Asserts that targets are built with Clang coverage enabled."""
Yuke Liao80afff32018-03-07 01:26:201311 build_args = _GetBuildArgs()
Yuke Liao506e8822017-12-04 16:52:541312
1313 if (CLANG_COVERAGE_BUILD_ARG not in build_args or
1314 build_args[CLANG_COVERAGE_BUILD_ARG] != 'true'):
Abhishek Arya1ec832c2017-12-05 18:06:591315 assert False, ('\'{} = true\' is required in args.gn.'
1316 ).format(CLANG_COVERAGE_BUILD_ARG)
Yuke Liao506e8822017-12-04 16:52:541317
1318
Yuke Liaoc60b2d02018-03-02 21:40:431319def _ValidateCurrentPlatformIsSupported():
1320 """Asserts that this script suports running on the current platform"""
1321 target_os = _GetTargetOS()
1322 if target_os:
1323 current_platform = target_os
1324 else:
1325 current_platform = _GetHostPlatform()
1326
1327 assert current_platform in [
1328 'linux', 'mac', 'chromeos', 'ios'
1329 ], ('Coverage is only supported on linux, mac, chromeos and ios.')
1330
1331
Yuke Liao80afff32018-03-07 01:26:201332def _GetBuildArgs():
Yuke Liao506e8822017-12-04 16:52:541333 """Parses args.gn file and returns results as a dictionary.
1334
1335 Returns:
1336 A dictionary representing the build args.
1337 """
Yuke Liao80afff32018-03-07 01:26:201338 global _BUILD_ARGS
1339 if _BUILD_ARGS is not None:
1340 return _BUILD_ARGS
1341
1342 _BUILD_ARGS = {}
Yuke Liao506e8822017-12-04 16:52:541343 build_args_path = os.path.join(BUILD_DIR, 'args.gn')
1344 assert os.path.exists(build_args_path), ('"%s" is not a build directory, '
1345 'missing args.gn file.' % BUILD_DIR)
1346 with open(build_args_path) as build_args_file:
1347 build_args_lines = build_args_file.readlines()
1348
Yuke Liao506e8822017-12-04 16:52:541349 for build_arg_line in build_args_lines:
1350 build_arg_without_comments = build_arg_line.split('#')[0]
1351 key_value_pair = build_arg_without_comments.split('=')
1352 if len(key_value_pair) != 2:
1353 continue
1354
1355 key = key_value_pair[0].strip()
Yuke Liaoc60b2d02018-03-02 21:40:431356
1357 # Values are wrapped within a pair of double-quotes, so remove the leading
1358 # and trailing double-quotes.
1359 value = key_value_pair[1].strip().strip('"')
Yuke Liao80afff32018-03-07 01:26:201360 _BUILD_ARGS[key] = value
Yuke Liao506e8822017-12-04 16:52:541361
Yuke Liao80afff32018-03-07 01:26:201362 return _BUILD_ARGS
Yuke Liao506e8822017-12-04 16:52:541363
1364
Abhishek Arya16f059a2017-12-07 17:47:321365def _VerifyPathsAndReturnAbsolutes(paths):
1366 """Verifies that the paths specified in |paths| exist and returns absolute
1367 versions.
Yuke Liao66da1732017-12-05 22:19:421368
1369 Args:
1370 paths: A list of files or directories.
1371 """
Abhishek Arya16f059a2017-12-07 17:47:321372 absolute_paths = []
Yuke Liao66da1732017-12-05 22:19:421373 for path in paths:
Abhishek Arya16f059a2017-12-07 17:47:321374 absolute_path = os.path.join(SRC_ROOT_PATH, path)
1375 assert os.path.exists(absolute_path), ('Path: "%s" doesn\'t exist.' % path)
1376
1377 absolute_paths.append(absolute_path)
1378
1379 return absolute_paths
Yuke Liao66da1732017-12-05 22:19:421380
1381
Yuke Liaodd1ec0592018-02-02 01:26:371382def _GetRelativePathToDirectoryOfFile(target_path, base_path):
1383 """Returns a target path relative to the directory of base_path.
1384
1385 This method requires base_path to be a file, otherwise, one should call
1386 os.path.relpath directly.
1387 """
1388 assert os.path.dirname(base_path) != base_path, (
Yuke Liaoc7e607142018-02-05 20:26:141389 'Base path: "%s" is a directory, please call os.path.relpath directly.' %
Yuke Liaodd1ec0592018-02-02 01:26:371390 base_path)
Yuke Liaoc7e607142018-02-05 20:26:141391 base_dir = os.path.dirname(base_path)
1392 return os.path.relpath(target_path, base_dir)
Yuke Liaodd1ec0592018-02-02 01:26:371393
1394
Abhishek Arya64636af2018-05-04 14:42:131395def _GetBinaryPathsFromTargets(targets, build_dir):
1396 """Return binary paths from target names."""
1397 # FIXME: Derive output binary from target build definitions rather than
1398 # assuming that it is always the same name.
1399 binary_paths = []
1400 for target in targets:
1401 binary_path = os.path.join(build_dir, target)
1402 if _GetHostPlatform() == 'win':
1403 binary_path += '.exe'
1404
1405 if os.path.exists(binary_path):
1406 binary_paths.append(binary_path)
1407 else:
1408 logging.warning(
Abhishek Aryafb70b532018-05-06 17:47:401409 'Target binary "%s" not found in build directory, skipping.',
Abhishek Arya64636af2018-05-04 14:42:131410 os.path.basename(binary_path))
1411
1412 return binary_paths
1413
1414
Yuke Liao506e8822017-12-04 16:52:541415def _ParseCommandArguments():
1416 """Adds and parses relevant arguments for tool comands.
1417
1418 Returns:
1419 A dictionary representing the arguments.
1420 """
1421 arg_parser = argparse.ArgumentParser()
1422 arg_parser.usage = __doc__
1423
Abhishek Arya1ec832c2017-12-05 18:06:591424 arg_parser.add_argument(
1425 '-b',
1426 '--build-dir',
1427 type=str,
1428 required=True,
1429 help='The build directory, the path needs to be relative to the root of '
1430 'the checkout.')
Yuke Liao506e8822017-12-04 16:52:541431
Abhishek Arya1ec832c2017-12-05 18:06:591432 arg_parser.add_argument(
1433 '-o',
1434 '--output-dir',
1435 type=str,
1436 required=True,
1437 help='Output directory for generated artifacts.')
Yuke Liao506e8822017-12-04 16:52:541438
Abhishek Arya1ec832c2017-12-05 18:06:591439 arg_parser.add_argument(
1440 '-c',
1441 '--command',
1442 action='append',
Abhishek Arya64636af2018-05-04 14:42:131443 required=False,
Abhishek Arya1ec832c2017-12-05 18:06:591444 help='Commands used to run test targets, one test target needs one and '
1445 'only one command, when specifying commands, one should assume the '
Abhishek Arya64636af2018-05-04 14:42:131446 'current working directory is the root of the checkout. This option is '
1447 'incompatible with -p/--profdata-file option.')
1448
1449 arg_parser.add_argument(
1450 '-p',
1451 '--profdata-file',
1452 type=str,
1453 required=False,
1454 help='Path to profdata file to use for generating code coverage reports. '
1455 'This can be useful if you generated the profdata file seperately in '
1456 'your own test harness. This option is ignored if run command(s) are '
1457 'already provided above using -c/--command option.')
Yuke Liao506e8822017-12-04 16:52:541458
Abhishek Arya1ec832c2017-12-05 18:06:591459 arg_parser.add_argument(
Yuke Liao66da1732017-12-05 22:19:421460 '-f',
1461 '--filters',
1462 action='append',
Abhishek Arya16f059a2017-12-07 17:47:321463 required=False,
Yuke Liao66da1732017-12-05 22:19:421464 help='Directories or files to get code coverage for, and all files under '
1465 'the directories are included recursively.')
1466
1467 arg_parser.add_argument(
Yuke Liao0e4c8682018-04-18 21:06:591468 '-i',
1469 '--ignore-filename-regex',
1470 type=str,
1471 help='Skip source code files with file paths that match the given '
1472 'regular expression. For example, use -i=\'.*/out/.*|.*/third_party/.*\' '
1473 'to exclude files in third_party/ and out/ folders from the report.')
1474
1475 arg_parser.add_argument(
Yuke Liao1b852fd2018-05-11 17:07:321476 '--no-file-view',
1477 action='store_true',
1478 help='Don\'t generate the file view in the coverage report. When there '
1479 'are large number of html files, the file view becomes heavy and may '
1480 'cause the browser to freeze, and this argument comes handy.')
1481
1482 arg_parser.add_argument(
Abhishek Arya1ec832c2017-12-05 18:06:591483 '-j',
1484 '--jobs',
1485 type=int,
1486 default=None,
1487 help='Run N jobs to build in parallel. If not specified, a default value '
1488 'will be derived based on CPUs availability. Please refer to '
1489 '\'ninja -h\' for more details.')
Yuke Liao506e8822017-12-04 16:52:541490
Abhishek Arya1ec832c2017-12-05 18:06:591491 arg_parser.add_argument(
Yuke Liao481d3482018-01-29 19:17:101492 '-v',
1493 '--verbose',
1494 action='store_true',
1495 help='Prints additional output for diagnostics.')
1496
1497 arg_parser.add_argument(
1498 '-l', '--log_file', type=str, help='Redirects logs to a file.')
1499
1500 arg_parser.add_argument(
Abhishek Aryac19bc5ef2018-05-04 22:10:021501 'targets',
1502 nargs='+',
1503 help='The names of the test targets to run. If multiple run commands are '
1504 'specified using the -c/--command option, then the order of targets and '
1505 'commands must match, otherwise coverage generation will fail.')
Yuke Liao506e8822017-12-04 16:52:541506
1507 args = arg_parser.parse_args()
1508 return args
1509
1510
1511def Main():
1512 """Execute tool commands."""
Abhishek Arya64636af2018-05-04 14:42:131513 # Change directory to source root to aid in relative paths calculations.
1514 os.chdir(SRC_ROOT_PATH)
Abhishek Arya8a0751a2018-05-03 18:53:111515
Abhishek Arya64636af2018-05-04 14:42:131516 # Setup coverage binaries even when script is called with empty params. This
1517 # is used by coverage bot for initial setup.
Abhishek Arya8a0751a2018-05-03 18:53:111518 DownloadCoverageToolsIfNeeded()
1519
Yuke Liao506e8822017-12-04 16:52:541520 args = _ParseCommandArguments()
Abhishek Arya64636af2018-05-04 14:42:131521 _ConfigureLogging(args)
1522
Yuke Liao506e8822017-12-04 16:52:541523 global BUILD_DIR
Max Moroz7c5354f2018-05-06 00:03:481524 BUILD_DIR = os.path.abspath(args.build_dir)
Yuke Liao506e8822017-12-04 16:52:541525 global OUTPUT_DIR
Max Moroz7c5354f2018-05-06 00:03:481526 OUTPUT_DIR = os.path.abspath(args.output_dir)
Yuke Liao506e8822017-12-04 16:52:541527
Abhishek Arya64636af2018-05-04 14:42:131528 assert args.command or args.profdata_file, (
1529 'Need to either provide commands to run using -c/--command option OR '
1530 'provide prof-data file as input using -p/--profdata-file option.')
Yuke Liaoc60b2d02018-03-02 21:40:431531
Abhishek Arya64636af2018-05-04 14:42:131532 assert not args.command or (len(args.targets) == len(args.command)), (
1533 'Number of targets must be equal to the number of test commands.')
Yuke Liaoc60b2d02018-03-02 21:40:431534
Abhishek Arya1ec832c2017-12-05 18:06:591535 assert os.path.exists(BUILD_DIR), (
Abhishek Aryafb70b532018-05-06 17:47:401536 'Build directory: "%s" doesn\'t exist. '
1537 'Please run "gn gen" to generate.' % BUILD_DIR)
Abhishek Arya64636af2018-05-04 14:42:131538
Yuke Liaoc60b2d02018-03-02 21:40:431539 _ValidateCurrentPlatformIsSupported()
Yuke Liao506e8822017-12-04 16:52:541540 _ValidateBuildingWithClangCoverage()
Abhishek Arya16f059a2017-12-07 17:47:321541
1542 absolute_filter_paths = []
Yuke Liao66da1732017-12-05 22:19:421543 if args.filters:
Abhishek Arya16f059a2017-12-07 17:47:321544 absolute_filter_paths = _VerifyPathsAndReturnAbsolutes(args.filters)
Yuke Liao66da1732017-12-05 22:19:421545
Max Moroz7c5354f2018-05-06 00:03:481546 if not os.path.exists(_GetCoverageReportRootDirPath()):
1547 os.makedirs(_GetCoverageReportRootDirPath())
Yuke Liao506e8822017-12-04 16:52:541548
Abhishek Arya64636af2018-05-04 14:42:131549 # Get profdate file and list of binary paths.
1550 if args.command:
1551 # A list of commands are provided. Run them to generate profdata file, and
1552 # create a list of binary paths from parsing commands.
1553 _VerifyTargetExecutablesAreInBuildDirectory(args.command)
1554 profdata_file_path = _CreateCoverageProfileDataForTargets(
1555 args.targets, args.command, args.jobs)
1556 binary_paths = [_GetBinaryPath(command) for command in args.command]
1557 else:
1558 # An input prof-data file is already provided. Just calculate binary paths.
1559 profdata_file_path = args.profdata_file
1560 binary_paths = _GetBinaryPathsFromTargets(args.targets, args.build_dir)
Yuke Liaoea228d02018-01-05 19:10:331561
Abhishek Arya78120bc2018-05-07 20:53:541562 binary_paths.extend(_GetSharedLibraries(binary_paths))
1563
Yuke Liao481d3482018-01-29 19:17:101564 logging.info('Generating code coverage report in html (this can take a while '
Abhishek Aryafb70b532018-05-06 17:47:401565 'depending on size of target!).')
Yuke Liaodd1ec0592018-02-02 01:26:371566 per_file_coverage_summary = _GeneratePerFileCoverageSummary(
Yuke Liao0e4c8682018-04-18 21:06:591567 binary_paths, profdata_file_path, absolute_filter_paths,
1568 args.ignore_filename_regex)
Yuke Liaodd1ec0592018-02-02 01:26:371569 _GeneratePerFileLineByLineCoverageInHtml(binary_paths, profdata_file_path,
Yuke Liao0e4c8682018-04-18 21:06:591570 absolute_filter_paths,
1571 args.ignore_filename_regex)
Yuke Liao1b852fd2018-05-11 17:07:321572 if not args.no_file_view:
1573 _GenerateFileViewHtmlIndexFile(per_file_coverage_summary)
Yuke Liaodd1ec0592018-02-02 01:26:371574
1575 per_directory_coverage_summary = _CalculatePerDirectoryCoverageSummary(
1576 per_file_coverage_summary)
1577 _GeneratePerDirectoryCoverageInHtml(per_directory_coverage_summary,
Yuke Liao1b852fd2018-05-11 17:07:321578 per_file_coverage_summary,
1579 args.no_file_view)
Yuke Liaodd1ec0592018-02-02 01:26:371580 _GenerateDirectoryViewHtmlIndexFile()
1581
1582 component_to_directories = _ExtractComponentToDirectoriesMapping()
1583 per_component_coverage_summary = _CalculatePerComponentCoverageSummary(
1584 component_to_directories, per_directory_coverage_summary)
Yuke Liao1b852fd2018-05-11 17:07:321585 _GeneratePerComponentCoverageInHtml(
1586 per_component_coverage_summary, component_to_directories,
1587 per_directory_coverage_summary, args.no_file_view)
1588 _GenerateComponentViewHtmlIndexFile(per_component_coverage_summary,
1589 args.no_file_view)
Yuke Liaoea228d02018-01-05 19:10:331590
1591 # The default index file is generated only for the list of source files, needs
Yuke Liaodd1ec0592018-02-02 01:26:371592 # to overwrite it to display per directory coverage view by default.
Yuke Liaoea228d02018-01-05 19:10:331593 _OverwriteHtmlReportsIndexFile()
Max Moroz7c5354f2018-05-06 00:03:481594 _CleanUpOutputDir()
Yuke Liaoea228d02018-01-05 19:10:331595
Max Moroz7c5354f2018-05-06 00:03:481596 html_index_file_path = 'file://' + os.path.abspath(_GetHtmlIndexPath())
Abhishek Aryafb70b532018-05-06 17:47:401597 logging.info('Index file for html report is generated as: "%s".',
Yuke Liao481d3482018-01-29 19:17:101598 html_index_file_path)
Yuke Liao506e8822017-12-04 16:52:541599
Abhishek Arya1ec832c2017-12-05 18:06:591600
Yuke Liao506e8822017-12-04 16:52:541601if __name__ == '__main__':
1602 sys.exit(Main())