]> git.lizzy.rs Git - rust.git/blob - compiler/rustc_mir_transform/src/coverage/debug.rs
Auto merge of #92353 - Kobzol:doc-attr-lists-gat, r=GuillaumeGomez
[rust.git] / compiler / rustc_mir_transform / src / coverage / debug.rs
1 //! The `InstrumentCoverage` MIR pass implementation includes debugging tools and options
2 //! to help developers understand and/or improve the analysis and instrumentation of a MIR.
3 //!
4 //! To enable coverage, include the rustc command line option:
5 //!
6 //!   * `-Z instrument-coverage`
7 //!
8 //! MIR Dump Files, with additional `CoverageGraph` graphviz and `CoverageSpan` spanview
9 //! ------------------------------------------------------------------------------------
10 //!
11 //! Additional debugging options include:
12 //!
13 //!   * `-Z dump-mir=InstrumentCoverage` - Generate `.mir` files showing the state of the MIR,
14 //!     before and after the `InstrumentCoverage` pass, for each compiled function.
15 //!
16 //!   * `-Z dump-mir-graphviz` - If `-Z dump-mir` is also enabled for the current MIR node path,
17 //!     each MIR dump is accompanied by a before-and-after graphical view of the MIR, in Graphviz
18 //!     `.dot` file format (which can be visually rendered as a graph using any of a number of free
19 //!     Graphviz viewers and IDE extensions).
20 //!
21 //!     For the `InstrumentCoverage` pass, this option also enables generation of an additional
22 //!     Graphviz `.dot` file for each function, rendering the `CoverageGraph`: the control flow
23 //!     graph (CFG) of `BasicCoverageBlocks` (BCBs), as nodes, internally labeled to show the
24 //!     `CoverageSpan`-based MIR elements each BCB represents (`BasicBlock`s, `Statement`s and
25 //!     `Terminator`s), assigned coverage counters and/or expressions, and edge counters, as needed.
26 //!
27 //!     (Note the additional option, `-Z graphviz-dark-mode`, can be added, to change the rendered
28 //!     output from its default black-on-white background to a dark color theme, if desired.)
29 //!
30 //!   * `-Z dump-mir-spanview` - If `-Z dump-mir` is also enabled for the current MIR node path,
31 //!     each MIR dump is accompanied by a before-and-after `.html` document showing the function's
32 //!     original source code, highlighted by it's MIR spans, at the `statement`-level (by default),
33 //!     `terminator` only, or encompassing span for the `Terminator` plus all `Statement`s, in each
34 //!     `block` (`BasicBlock`).
35 //!
36 //!     For the `InstrumentCoverage` pass, this option also enables generation of an additional
37 //!     spanview `.html` file for each function, showing the aggregated `CoverageSpan`s that will
38 //!     require counters (or counter expressions) for accurate coverage analysis.
39 //!
40 //! Debug Logging
41 //! -------------
42 //!
43 //! The `InstrumentCoverage` pass includes debug logging messages at various phases and decision
44 //! points, which can be enabled via environment variable:
45 //!
46 //! ```shell
47 //! RUSTC_LOG=rustc_mir_transform::transform::coverage=debug
48 //! ```
49 //!
50 //! Other module paths with coverage-related debug logs may also be of interest, particularly for
51 //! debugging the coverage map data, injected as global variables in the LLVM IR (during rustc's
52 //! code generation pass). For example:
53 //!
54 //! ```shell
55 //! RUSTC_LOG=rustc_mir_transform::transform::coverage,rustc_codegen_ssa::coverageinfo,rustc_codegen_llvm::coverageinfo=debug
56 //! ```
57 //!
58 //! Coverage Debug Options
59 //! ---------------------------------
60 //!
61 //! Additional debugging options can be enabled using the environment variable:
62 //!
63 //! ```shell
64 //! RUSTC_COVERAGE_DEBUG_OPTIONS=<options>
65 //! ```
66 //!
67 //! These options are comma-separated, and specified in the format `option-name=value`. For example:
68 //!
69 //! ```shell
70 //! $ RUSTC_COVERAGE_DEBUG_OPTIONS=counter-format=id+operation,allow-unused-expressions=yes cargo build
71 //! ```
72 //!
73 //! Coverage debug options include:
74 //!
75 //!   * `allow-unused-expressions=yes` or `no` (default: `no`)
76 //!
77 //!     The `InstrumentCoverage` algorithms _should_ only create and assign expressions to a
78 //!     `BasicCoverageBlock`, or an incoming edge, if that expression is either (a) required to
79 //!     count a `CoverageSpan`, or (b) a dependency of some other required counter expression.
80 //!
81 //!     If an expression is generated that does not map to a `CoverageSpan` or dependency, this
82 //!     probably indicates there was a bug in the algorithm that creates and assigns counters
83 //!     and expressions.
84 //!
85 //!     When this kind of bug is encountered, the rustc compiler will panic by default. Setting:
86 //!     `allow-unused-expressions=yes` will log a warning message instead of panicking (effectively
87 //!     ignoring the unused expressions), which may be helpful when debugging the root cause of
88 //!     the problem.
89 //!
90 //!   * `counter-format=<choices>`, where `<choices>` can be any plus-separated combination of `id`,
91 //!     `block`, and/or `operation` (default: `block+operation`)
92 //!
93 //!     This option effects both the `CoverageGraph` (graphviz `.dot` files) and debug logging, when
94 //!     generating labels for counters and expressions.
95 //!
96 //!     Depending on the values and combinations, counters can be labeled by:
97 //!
98 //!       * `id` - counter or expression ID (ascending counter IDs, starting at 1, or descending
99 //!         expression IDs, starting at `u32:MAX`)
100 //!       * `block` - the `BasicCoverageBlock` label (for example, `bcb0`) or edge label (for
101 //!         example `bcb0->bcb1`), for counters or expressions assigned to count a
102 //!         `BasicCoverageBlock` or edge. Intermediate expressions (not directly associated with
103 //!         a BCB or edge) will be labeled by their expression ID, unless `operation` is also
104 //!         specified.
105 //!       * `operation` - applied to expressions only, labels include the left-hand-side counter
106 //!         or expression label (lhs operand), the operator (`+` or `-`), and the right-hand-side
107 //!         counter or expression (rhs operand). Expression operand labels are generated
108 //!         recursively, generating labels with nested operations, enclosed in parentheses
109 //!         (for example: `bcb2 + (bcb0 - bcb1)`).
110
111 use super::graph::{BasicCoverageBlock, BasicCoverageBlockData, CoverageGraph};
112 use super::spans::CoverageSpan;
113
114 use itertools::Itertools;
115 use rustc_middle::mir::create_dump_file;
116 use rustc_middle::mir::generic_graphviz::GraphvizWriter;
117 use rustc_middle::mir::spanview::{self, SpanViewable};
118
119 use rustc_data_structures::fx::FxHashMap;
120 use rustc_middle::mir::coverage::*;
121 use rustc_middle::mir::{self, BasicBlock, TerminatorKind};
122 use rustc_middle::ty::TyCtxt;
123 use rustc_span::Span;
124
125 use std::iter;
126 use std::lazy::SyncOnceCell;
127
128 pub const NESTED_INDENT: &str = "    ";
129
130 const RUSTC_COVERAGE_DEBUG_OPTIONS: &str = "RUSTC_COVERAGE_DEBUG_OPTIONS";
131
132 pub(super) fn debug_options<'a>() -> &'a DebugOptions {
133     static DEBUG_OPTIONS: SyncOnceCell<DebugOptions> = SyncOnceCell::new();
134
135     &DEBUG_OPTIONS.get_or_init(DebugOptions::from_env)
136 }
137
138 /// Parses and maintains coverage-specific debug options captured from the environment variable
139 /// "RUSTC_COVERAGE_DEBUG_OPTIONS", if set.
140 #[derive(Debug, Clone)]
141 pub(super) struct DebugOptions {
142     pub allow_unused_expressions: bool,
143     counter_format: ExpressionFormat,
144 }
145
146 impl DebugOptions {
147     fn from_env() -> Self {
148         let mut allow_unused_expressions = true;
149         let mut counter_format = ExpressionFormat::default();
150
151         if let Ok(env_debug_options) = std::env::var(RUSTC_COVERAGE_DEBUG_OPTIONS) {
152             for setting_str in env_debug_options.replace(' ', "").replace('-', "_").split(',') {
153                 let (option, value) = match setting_str.split_once('=') {
154                     None => (setting_str, None),
155                     Some((k, v)) => (k, Some(v)),
156                 };
157                 match option {
158                     "allow_unused_expressions" => {
159                         allow_unused_expressions = bool_option_val(option, value);
160                         debug!(
161                             "{} env option `allow_unused_expressions` is set to {}",
162                             RUSTC_COVERAGE_DEBUG_OPTIONS, allow_unused_expressions
163                         );
164                     }
165                     "counter_format" => {
166                         match value {
167                             None => {
168                                 bug!(
169                                     "`{}` option in environment variable {} requires one or more \
170                                     plus-separated choices (a non-empty subset of \
171                                     `id+block+operation`)",
172                                     option,
173                                     RUSTC_COVERAGE_DEBUG_OPTIONS
174                                 );
175                             }
176                             Some(val) => {
177                                 counter_format = counter_format_option_val(val);
178                                 debug!(
179                                     "{} env option `counter_format` is set to {:?}",
180                                     RUSTC_COVERAGE_DEBUG_OPTIONS, counter_format
181                                 );
182                             }
183                         };
184                     }
185                     _ => bug!(
186                         "Unsupported setting `{}` in environment variable {}",
187                         option,
188                         RUSTC_COVERAGE_DEBUG_OPTIONS
189                     ),
190                 };
191             }
192         }
193
194         Self { allow_unused_expressions, counter_format }
195     }
196 }
197
198 fn bool_option_val(option: &str, some_strval: Option<&str>) -> bool {
199     if let Some(val) = some_strval {
200         if vec!["yes", "y", "on", "true"].contains(&val) {
201             true
202         } else if vec!["no", "n", "off", "false"].contains(&val) {
203             false
204         } else {
205             bug!(
206                 "Unsupported value `{}` for option `{}` in environment variable {}",
207                 option,
208                 val,
209                 RUSTC_COVERAGE_DEBUG_OPTIONS
210             )
211         }
212     } else {
213         true
214     }
215 }
216
217 fn counter_format_option_val(strval: &str) -> ExpressionFormat {
218     let mut counter_format = ExpressionFormat { id: false, block: false, operation: false };
219     let components = strval.splitn(3, '+');
220     for component in components {
221         match component {
222             "id" => counter_format.id = true,
223             "block" => counter_format.block = true,
224             "operation" => counter_format.operation = true,
225             _ => bug!(
226                 "Unsupported counter_format choice `{}` in environment variable {}",
227                 component,
228                 RUSTC_COVERAGE_DEBUG_OPTIONS
229             ),
230         }
231     }
232     counter_format
233 }
234
235 #[derive(Debug, Clone)]
236 struct ExpressionFormat {
237     id: bool,
238     block: bool,
239     operation: bool,
240 }
241
242 impl Default for ExpressionFormat {
243     fn default() -> Self {
244         Self { id: false, block: true, operation: true }
245     }
246 }
247
248 /// If enabled, this struct maintains a map from `CoverageKind` IDs (as `ExpressionOperandId`) to
249 /// the `CoverageKind` data and optional label (normally, the counter's associated
250 /// `BasicCoverageBlock` format string, if any).
251 ///
252 /// Use `format_counter` to convert one of these `CoverageKind` counters to a debug output string,
253 /// as directed by the `DebugOptions`. This allows the format of counter labels in logs and dump
254 /// files (including the `CoverageGraph` graphviz file) to be changed at runtime, via environment
255 /// variable.
256 ///
257 /// `DebugCounters` supports a recursive rendering of `Expression` counters, so they can be
258 /// presented as nested expressions such as `(bcb3 - (bcb0 + bcb1))`.
259 pub(super) struct DebugCounters {
260     some_counters: Option<FxHashMap<ExpressionOperandId, DebugCounter>>,
261 }
262
263 impl DebugCounters {
264     pub fn new() -> Self {
265         Self { some_counters: None }
266     }
267
268     pub fn enable(&mut self) {
269         debug_assert!(!self.is_enabled());
270         self.some_counters.replace(FxHashMap::default());
271     }
272
273     pub fn is_enabled(&self) -> bool {
274         self.some_counters.is_some()
275     }
276
277     pub fn add_counter(&mut self, counter_kind: &CoverageKind, some_block_label: Option<String>) {
278         if let Some(counters) = &mut self.some_counters {
279             let id: ExpressionOperandId = match *counter_kind {
280                 CoverageKind::Counter { id, .. } => id.into(),
281                 CoverageKind::Expression { id, .. } => id.into(),
282                 _ => bug!(
283                     "the given `CoverageKind` is not an counter or expression: {:?}",
284                     counter_kind
285                 ),
286             };
287             counters
288                 .try_insert(id, DebugCounter::new(counter_kind.clone(), some_block_label))
289                 .expect("attempt to add the same counter_kind to DebugCounters more than once");
290         }
291     }
292
293     pub fn some_block_label(&self, operand: ExpressionOperandId) -> Option<&String> {
294         self.some_counters.as_ref().map_or(None, |counters| {
295             counters
296                 .get(&operand)
297                 .map_or(None, |debug_counter| debug_counter.some_block_label.as_ref())
298         })
299     }
300
301     pub fn format_counter(&self, counter_kind: &CoverageKind) -> String {
302         match *counter_kind {
303             CoverageKind::Counter { .. } => {
304                 format!("Counter({})", self.format_counter_kind(counter_kind))
305             }
306             CoverageKind::Expression { .. } => {
307                 format!("Expression({})", self.format_counter_kind(counter_kind))
308             }
309             CoverageKind::Unreachable { .. } => "Unreachable".to_owned(),
310         }
311     }
312
313     fn format_counter_kind(&self, counter_kind: &CoverageKind) -> String {
314         let counter_format = &debug_options().counter_format;
315         if let CoverageKind::Expression { id, lhs, op, rhs } = *counter_kind {
316             if counter_format.operation {
317                 return format!(
318                     "{}{} {} {}",
319                     if counter_format.id || self.some_counters.is_none() {
320                         format!("#{} = ", id.index())
321                     } else {
322                         String::new()
323                     },
324                     self.format_operand(lhs),
325                     if op == Op::Add { "+" } else { "-" },
326                     self.format_operand(rhs),
327                 );
328             }
329         }
330
331         let id: ExpressionOperandId = match *counter_kind {
332             CoverageKind::Counter { id, .. } => id.into(),
333             CoverageKind::Expression { id, .. } => id.into(),
334             _ => {
335                 bug!("the given `CoverageKind` is not an counter or expression: {:?}", counter_kind)
336             }
337         };
338         if self.some_counters.is_some() && (counter_format.block || !counter_format.id) {
339             let counters = self.some_counters.as_ref().unwrap();
340             if let Some(DebugCounter { some_block_label: Some(block_label), .. }) =
341                 counters.get(&id)
342             {
343                 return if counter_format.id {
344                     format!("{}#{}", block_label, id.index())
345                 } else {
346                     block_label.to_string()
347                 };
348             }
349         }
350         format!("#{}", id.index())
351     }
352
353     fn format_operand(&self, operand: ExpressionOperandId) -> String {
354         if operand.index() == 0 {
355             return String::from("0");
356         }
357         if let Some(counters) = &self.some_counters {
358             if let Some(DebugCounter { counter_kind, some_block_label }) = counters.get(&operand) {
359                 if let CoverageKind::Expression { .. } = counter_kind {
360                     if let Some(block_label) = some_block_label {
361                         if debug_options().counter_format.block {
362                             return format!(
363                                 "{}:({})",
364                                 block_label,
365                                 self.format_counter_kind(counter_kind)
366                             );
367                         }
368                     }
369                     return format!("({})", self.format_counter_kind(counter_kind));
370                 }
371                 return self.format_counter_kind(counter_kind);
372             }
373         }
374         format!("#{}", operand.index())
375     }
376 }
377
378 /// A non-public support class to `DebugCounters`.
379 #[derive(Debug)]
380 struct DebugCounter {
381     counter_kind: CoverageKind,
382     some_block_label: Option<String>,
383 }
384
385 impl DebugCounter {
386     fn new(counter_kind: CoverageKind, some_block_label: Option<String>) -> Self {
387         Self { counter_kind, some_block_label }
388     }
389 }
390
391 /// If enabled, this data structure captures additional debugging information used when generating
392 /// a Graphviz (.dot file) representation of the `CoverageGraph`, for debugging purposes.
393 pub(super) struct GraphvizData {
394     some_bcb_to_coverage_spans_with_counters:
395         Option<FxHashMap<BasicCoverageBlock, Vec<(CoverageSpan, CoverageKind)>>>,
396     some_bcb_to_dependency_counters: Option<FxHashMap<BasicCoverageBlock, Vec<CoverageKind>>>,
397     some_edge_to_counter: Option<FxHashMap<(BasicCoverageBlock, BasicBlock), CoverageKind>>,
398 }
399
400 impl GraphvizData {
401     pub fn new() -> Self {
402         Self {
403             some_bcb_to_coverage_spans_with_counters: None,
404             some_bcb_to_dependency_counters: None,
405             some_edge_to_counter: None,
406         }
407     }
408
409     pub fn enable(&mut self) {
410         debug_assert!(!self.is_enabled());
411         self.some_bcb_to_coverage_spans_with_counters = Some(FxHashMap::default());
412         self.some_bcb_to_dependency_counters = Some(FxHashMap::default());
413         self.some_edge_to_counter = Some(FxHashMap::default());
414     }
415
416     pub fn is_enabled(&self) -> bool {
417         self.some_bcb_to_coverage_spans_with_counters.is_some()
418     }
419
420     pub fn add_bcb_coverage_span_with_counter(
421         &mut self,
422         bcb: BasicCoverageBlock,
423         coverage_span: &CoverageSpan,
424         counter_kind: &CoverageKind,
425     ) {
426         if let Some(bcb_to_coverage_spans_with_counters) =
427             self.some_bcb_to_coverage_spans_with_counters.as_mut()
428         {
429             bcb_to_coverage_spans_with_counters
430                 .entry(bcb)
431                 .or_insert_with(Vec::new)
432                 .push((coverage_span.clone(), counter_kind.clone()));
433         }
434     }
435
436     pub fn get_bcb_coverage_spans_with_counters(
437         &self,
438         bcb: BasicCoverageBlock,
439     ) -> Option<&Vec<(CoverageSpan, CoverageKind)>> {
440         if let Some(bcb_to_coverage_spans_with_counters) =
441             self.some_bcb_to_coverage_spans_with_counters.as_ref()
442         {
443             bcb_to_coverage_spans_with_counters.get(&bcb)
444         } else {
445             None
446         }
447     }
448
449     pub fn add_bcb_dependency_counter(
450         &mut self,
451         bcb: BasicCoverageBlock,
452         counter_kind: &CoverageKind,
453     ) {
454         if let Some(bcb_to_dependency_counters) = self.some_bcb_to_dependency_counters.as_mut() {
455             bcb_to_dependency_counters
456                 .entry(bcb)
457                 .or_insert_with(Vec::new)
458                 .push(counter_kind.clone());
459         }
460     }
461
462     pub fn get_bcb_dependency_counters(
463         &self,
464         bcb: BasicCoverageBlock,
465     ) -> Option<&Vec<CoverageKind>> {
466         if let Some(bcb_to_dependency_counters) = self.some_bcb_to_dependency_counters.as_ref() {
467             bcb_to_dependency_counters.get(&bcb)
468         } else {
469             None
470         }
471     }
472
473     pub fn set_edge_counter(
474         &mut self,
475         from_bcb: BasicCoverageBlock,
476         to_bb: BasicBlock,
477         counter_kind: &CoverageKind,
478     ) {
479         if let Some(edge_to_counter) = self.some_edge_to_counter.as_mut() {
480             edge_to_counter
481                 .try_insert((from_bcb, to_bb), counter_kind.clone())
482                 .expect("invalid attempt to insert more than one edge counter for the same edge");
483         }
484     }
485
486     pub fn get_edge_counter(
487         &self,
488         from_bcb: BasicCoverageBlock,
489         to_bb: BasicBlock,
490     ) -> Option<&CoverageKind> {
491         if let Some(edge_to_counter) = self.some_edge_to_counter.as_ref() {
492             edge_to_counter.get(&(from_bcb, to_bb))
493         } else {
494             None
495         }
496     }
497 }
498
499 /// If enabled, this struct captures additional data used to track whether expressions were used,
500 /// directly or indirectly, to compute the coverage counts for all `CoverageSpan`s, and any that are
501 /// _not_ used are retained in the `unused_expressions` Vec, to be included in debug output (logs
502 /// and/or a `CoverageGraph` graphviz output).
503 pub(super) struct UsedExpressions {
504     some_used_expression_operands:
505         Option<FxHashMap<ExpressionOperandId, Vec<InjectedExpressionId>>>,
506     some_unused_expressions:
507         Option<Vec<(CoverageKind, Option<BasicCoverageBlock>, BasicCoverageBlock)>>,
508 }
509
510 impl UsedExpressions {
511     pub fn new() -> Self {
512         Self { some_used_expression_operands: None, some_unused_expressions: None }
513     }
514
515     pub fn enable(&mut self) {
516         debug_assert!(!self.is_enabled());
517         self.some_used_expression_operands = Some(FxHashMap::default());
518         self.some_unused_expressions = Some(Vec::new());
519     }
520
521     pub fn is_enabled(&self) -> bool {
522         self.some_used_expression_operands.is_some()
523     }
524
525     pub fn add_expression_operands(&mut self, expression: &CoverageKind) {
526         if let Some(used_expression_operands) = self.some_used_expression_operands.as_mut() {
527             if let CoverageKind::Expression { id, lhs, rhs, .. } = *expression {
528                 used_expression_operands.entry(lhs).or_insert_with(Vec::new).push(id);
529                 used_expression_operands.entry(rhs).or_insert_with(Vec::new).push(id);
530             }
531         }
532     }
533
534     pub fn expression_is_used(&self, expression: &CoverageKind) -> bool {
535         if let Some(used_expression_operands) = self.some_used_expression_operands.as_ref() {
536             used_expression_operands.contains_key(&expression.as_operand_id())
537         } else {
538             false
539         }
540     }
541
542     pub fn add_unused_expression_if_not_found(
543         &mut self,
544         expression: &CoverageKind,
545         edge_from_bcb: Option<BasicCoverageBlock>,
546         target_bcb: BasicCoverageBlock,
547     ) {
548         if let Some(used_expression_operands) = self.some_used_expression_operands.as_ref() {
549             if !used_expression_operands.contains_key(&expression.as_operand_id()) {
550                 self.some_unused_expressions.as_mut().unwrap().push((
551                     expression.clone(),
552                     edge_from_bcb,
553                     target_bcb,
554                 ));
555             }
556         }
557     }
558
559     /// Return the list of unused counters (if any) as a tuple with the counter (`CoverageKind`),
560     /// optional `from_bcb` (if it was an edge counter), and `target_bcb`.
561     pub fn get_unused_expressions(
562         &self,
563     ) -> Vec<(CoverageKind, Option<BasicCoverageBlock>, BasicCoverageBlock)> {
564         if let Some(unused_expressions) = self.some_unused_expressions.as_ref() {
565             unused_expressions.clone()
566         } else {
567             Vec::new()
568         }
569     }
570
571     /// If enabled, validate that every BCB or edge counter not directly associated with a coverage
572     /// span is at least indirectly associated (it is a dependency of a BCB counter that _is_
573     /// associated with a coverage span).
574     pub fn validate(
575         &mut self,
576         bcb_counters_without_direct_coverage_spans: &Vec<(
577             Option<BasicCoverageBlock>,
578             BasicCoverageBlock,
579             CoverageKind,
580         )>,
581     ) {
582         if self.is_enabled() {
583             let mut not_validated = bcb_counters_without_direct_coverage_spans
584                 .iter()
585                 .map(|(_, _, counter_kind)| counter_kind)
586                 .collect::<Vec<_>>();
587             let mut validating_count = 0;
588             while not_validated.len() != validating_count {
589                 let to_validate = not_validated.split_off(0);
590                 validating_count = to_validate.len();
591                 for counter_kind in to_validate {
592                     if self.expression_is_used(counter_kind) {
593                         self.add_expression_operands(counter_kind);
594                     } else {
595                         not_validated.push(counter_kind);
596                     }
597                 }
598             }
599         }
600     }
601
602     pub fn alert_on_unused_expressions(&self, debug_counters: &DebugCounters) {
603         if let Some(unused_expressions) = self.some_unused_expressions.as_ref() {
604             for (counter_kind, edge_from_bcb, target_bcb) in unused_expressions {
605                 let unused_counter_message = if let Some(from_bcb) = edge_from_bcb.as_ref() {
606                     format!(
607                         "non-coverage edge counter found without a dependent expression, in \
608                         {:?}->{:?}; counter={}",
609                         from_bcb,
610                         target_bcb,
611                         debug_counters.format_counter(&counter_kind),
612                     )
613                 } else {
614                     format!(
615                         "non-coverage counter found without a dependent expression, in {:?}; \
616                         counter={}",
617                         target_bcb,
618                         debug_counters.format_counter(&counter_kind),
619                     )
620                 };
621
622                 if debug_options().allow_unused_expressions {
623                     debug!("WARNING: {}", unused_counter_message);
624                 } else {
625                     bug!("{}", unused_counter_message);
626                 }
627             }
628         }
629     }
630 }
631
632 /// Generates the MIR pass `CoverageSpan`-specific spanview dump file.
633 pub(super) fn dump_coverage_spanview<'tcx>(
634     tcx: TyCtxt<'tcx>,
635     mir_body: &mir::Body<'tcx>,
636     basic_coverage_blocks: &CoverageGraph,
637     pass_name: &str,
638     body_span: Span,
639     coverage_spans: &Vec<CoverageSpan>,
640 ) {
641     let mir_source = mir_body.source;
642     let def_id = mir_source.def_id();
643
644     let span_viewables = span_viewables(tcx, mir_body, basic_coverage_blocks, &coverage_spans);
645     let mut file = create_dump_file(tcx, "html", None, pass_name, &0, mir_source)
646         .expect("Unexpected error creating MIR spanview HTML file");
647     let crate_name = tcx.crate_name(def_id.krate);
648     let item_name = tcx.def_path(def_id).to_filename_friendly_no_crate();
649     let title = format!("{}.{} - Coverage Spans", crate_name, item_name);
650     spanview::write_document(tcx, body_span, span_viewables, &title, &mut file)
651         .expect("Unexpected IO error dumping coverage spans as HTML");
652 }
653
654 /// Converts the computed `BasicCoverageBlockData`s into `SpanViewable`s.
655 fn span_viewables<'tcx>(
656     tcx: TyCtxt<'tcx>,
657     mir_body: &mir::Body<'tcx>,
658     basic_coverage_blocks: &CoverageGraph,
659     coverage_spans: &Vec<CoverageSpan>,
660 ) -> Vec<SpanViewable> {
661     let mut span_viewables = Vec::new();
662     for coverage_span in coverage_spans {
663         let tooltip = coverage_span.format_coverage_statements(tcx, mir_body);
664         let CoverageSpan { span, bcb, .. } = coverage_span;
665         let bcb_data = &basic_coverage_blocks[*bcb];
666         let id = bcb_data.id();
667         let leader_bb = bcb_data.leader_bb();
668         span_viewables.push(SpanViewable { bb: leader_bb, span: *span, id, tooltip });
669     }
670     span_viewables
671 }
672
673 /// Generates the MIR pass coverage-specific graphviz dump file.
674 pub(super) fn dump_coverage_graphviz<'tcx>(
675     tcx: TyCtxt<'tcx>,
676     mir_body: &mir::Body<'tcx>,
677     pass_name: &str,
678     basic_coverage_blocks: &CoverageGraph,
679     debug_counters: &DebugCounters,
680     graphviz_data: &GraphvizData,
681     intermediate_expressions: &Vec<CoverageKind>,
682     debug_used_expressions: &UsedExpressions,
683 ) {
684     let mir_source = mir_body.source;
685     let def_id = mir_source.def_id();
686     let node_content = |bcb| {
687         bcb_to_string_sections(
688             tcx,
689             mir_body,
690             debug_counters,
691             &basic_coverage_blocks[bcb],
692             graphviz_data.get_bcb_coverage_spans_with_counters(bcb),
693             graphviz_data.get_bcb_dependency_counters(bcb),
694             // intermediate_expressions are injected into the mir::START_BLOCK, so
695             // include them in the first BCB.
696             if bcb.index() == 0 { Some(&intermediate_expressions) } else { None },
697         )
698     };
699     let edge_labels = |from_bcb| {
700         let from_bcb_data = &basic_coverage_blocks[from_bcb];
701         let from_terminator = from_bcb_data.terminator(mir_body);
702         let mut edge_labels = from_terminator.kind.fmt_successor_labels();
703         edge_labels.retain(|label| label != "unreachable");
704         let edge_counters = from_terminator
705             .successors()
706             .map(|&successor_bb| graphviz_data.get_edge_counter(from_bcb, successor_bb));
707         iter::zip(&edge_labels, edge_counters)
708             .map(|(label, some_counter)| {
709                 if let Some(counter) = some_counter {
710                     format!("{}\n{}", label, debug_counters.format_counter(counter))
711                 } else {
712                     label.to_string()
713                 }
714             })
715             .collect::<Vec<_>>()
716     };
717     let graphviz_name = format!("Cov_{}_{}", def_id.krate.index(), def_id.index.index());
718     let mut graphviz_writer =
719         GraphvizWriter::new(basic_coverage_blocks, &graphviz_name, node_content, edge_labels);
720     let unused_expressions = debug_used_expressions.get_unused_expressions();
721     if unused_expressions.len() > 0 {
722         graphviz_writer.set_graph_label(&format!(
723             "Unused expressions:\n  {}",
724             unused_expressions
725                 .as_slice()
726                 .iter()
727                 .map(|(counter_kind, edge_from_bcb, target_bcb)| {
728                     if let Some(from_bcb) = edge_from_bcb.as_ref() {
729                         format!(
730                             "{:?}->{:?}: {}",
731                             from_bcb,
732                             target_bcb,
733                             debug_counters.format_counter(&counter_kind),
734                         )
735                     } else {
736                         format!(
737                             "{:?}: {}",
738                             target_bcb,
739                             debug_counters.format_counter(&counter_kind),
740                         )
741                     }
742                 })
743                 .join("\n  ")
744         ));
745     }
746     let mut file = create_dump_file(tcx, "dot", None, pass_name, &0, mir_source)
747         .expect("Unexpected error creating BasicCoverageBlock graphviz DOT file");
748     graphviz_writer
749         .write_graphviz(tcx, &mut file)
750         .expect("Unexpected error writing BasicCoverageBlock graphviz DOT file");
751 }
752
753 fn bcb_to_string_sections<'tcx>(
754     tcx: TyCtxt<'tcx>,
755     mir_body: &mir::Body<'tcx>,
756     debug_counters: &DebugCounters,
757     bcb_data: &BasicCoverageBlockData,
758     some_coverage_spans_with_counters: Option<&Vec<(CoverageSpan, CoverageKind)>>,
759     some_dependency_counters: Option<&Vec<CoverageKind>>,
760     some_intermediate_expressions: Option<&Vec<CoverageKind>>,
761 ) -> Vec<String> {
762     let len = bcb_data.basic_blocks.len();
763     let mut sections = Vec::new();
764     if let Some(collect_intermediate_expressions) = some_intermediate_expressions {
765         sections.push(
766             collect_intermediate_expressions
767                 .iter()
768                 .map(|expression| {
769                     format!("Intermediate {}", debug_counters.format_counter(expression))
770                 })
771                 .join("\n"),
772         );
773     }
774     if let Some(coverage_spans_with_counters) = some_coverage_spans_with_counters {
775         sections.push(
776             coverage_spans_with_counters
777                 .iter()
778                 .map(|(covspan, counter)| {
779                     format!(
780                         "{} at {}",
781                         debug_counters.format_counter(counter),
782                         covspan.format(tcx, mir_body)
783                     )
784                 })
785                 .join("\n"),
786         );
787     }
788     if let Some(dependency_counters) = some_dependency_counters {
789         sections.push(format!(
790             "Non-coverage counters:\n  {}",
791             dependency_counters
792                 .iter()
793                 .map(|counter| debug_counters.format_counter(counter))
794                 .join("  \n"),
795         ));
796     }
797     if let Some(counter_kind) = &bcb_data.counter_kind {
798         sections.push(format!("{:?}", counter_kind));
799     }
800     let non_term_blocks = bcb_data.basic_blocks[0..len - 1]
801         .iter()
802         .map(|&bb| format!("{:?}: {}", bb, term_type(&mir_body[bb].terminator().kind)))
803         .collect::<Vec<_>>();
804     if non_term_blocks.len() > 0 {
805         sections.push(non_term_blocks.join("\n"));
806     }
807     sections.push(format!(
808         "{:?}: {}",
809         bcb_data.basic_blocks.last().unwrap(),
810         term_type(&bcb_data.terminator(mir_body).kind)
811     ));
812     sections
813 }
814
815 /// Returns a simple string representation of a `TerminatorKind` variant, independent of any
816 /// values it might hold.
817 pub(super) fn term_type(kind: &TerminatorKind<'_>) -> &'static str {
818     match kind {
819         TerminatorKind::Goto { .. } => "Goto",
820         TerminatorKind::SwitchInt { .. } => "SwitchInt",
821         TerminatorKind::Resume => "Resume",
822         TerminatorKind::Abort => "Abort",
823         TerminatorKind::Return => "Return",
824         TerminatorKind::Unreachable => "Unreachable",
825         TerminatorKind::Drop { .. } => "Drop",
826         TerminatorKind::DropAndReplace { .. } => "DropAndReplace",
827         TerminatorKind::Call { .. } => "Call",
828         TerminatorKind::Assert { .. } => "Assert",
829         TerminatorKind::Yield { .. } => "Yield",
830         TerminatorKind::GeneratorDrop => "GeneratorDrop",
831         TerminatorKind::FalseEdge { .. } => "FalseEdge",
832         TerminatorKind::FalseUnwind { .. } => "FalseUnwind",
833         TerminatorKind::InlineAsm { .. } => "InlineAsm",
834     }
835 }