]> git.lizzy.rs Git - rust.git/blob - compiler/rustc_mir/src/transform/coverage/debug.rs
clippy::useless_format
[rust.git] / compiler / rustc_mir / src / transform / 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::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::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 crate::util::generic_graphviz::GraphvizWriter;
115 use crate::util::pretty;
116 use crate::util::spanview::{self, SpanViewable};
117
118 use rustc_data_structures::fx::FxHashMap;
119 use rustc_middle::mir::coverage::*;
120 use rustc_middle::mir::{self, BasicBlock, TerminatorKind};
121 use rustc_middle::ty::TyCtxt;
122 use rustc_span::Span;
123
124 use std::iter;
125 use std::lazy::SyncOnceCell;
126
127 pub const NESTED_INDENT: &str = "    ";
128
129 const RUSTC_COVERAGE_DEBUG_OPTIONS: &str = "RUSTC_COVERAGE_DEBUG_OPTIONS";
130
131 pub(super) fn debug_options<'a>() -> &'a DebugOptions {
132     static DEBUG_OPTIONS: SyncOnceCell<DebugOptions> = SyncOnceCell::new();
133
134     &DEBUG_OPTIONS.get_or_init(DebugOptions::from_env)
135 }
136
137 /// Parses and maintains coverage-specific debug options captured from the environment variable
138 /// "RUSTC_COVERAGE_DEBUG_OPTIONS", if set.
139 #[derive(Debug, Clone)]
140 pub(super) struct DebugOptions {
141     pub allow_unused_expressions: bool,
142     counter_format: ExpressionFormat,
143 }
144
145 impl DebugOptions {
146     fn from_env() -> Self {
147         let mut allow_unused_expressions = true;
148         let mut counter_format = ExpressionFormat::default();
149
150         if let Ok(env_debug_options) = std::env::var(RUSTC_COVERAGE_DEBUG_OPTIONS) {
151             for setting_str in env_debug_options.replace(" ", "").replace("-", "_").split(',') {
152                 let (option, value) = match setting_str.split_once('=') {
153                     None => (setting_str, None),
154                     Some((k, v)) => (k, Some(v)),
155                 };
156                 match option {
157                     "allow_unused_expressions" => {
158                         allow_unused_expressions = bool_option_val(option, value);
159                         debug!(
160                             "{} env option `allow_unused_expressions` is set to {}",
161                             RUSTC_COVERAGE_DEBUG_OPTIONS, allow_unused_expressions
162                         );
163                     }
164                     "counter_format" => {
165                         match value {
166                             None => {
167                                 bug!(
168                                     "`{}` option in environment variable {} requires one or more \
169                                     plus-separated choices (a non-empty subset of \
170                                     `id+block+operation`)",
171                                     option,
172                                     RUSTC_COVERAGE_DEBUG_OPTIONS
173                                 );
174                             }
175                             Some(val) => {
176                                 counter_format = counter_format_option_val(val);
177                                 debug!(
178                                     "{} env option `counter_format` is set to {:?}",
179                                     RUSTC_COVERAGE_DEBUG_OPTIONS, counter_format
180                                 );
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
195         Self { allow_unused_expressions, counter_format }
196     }
197 }
198
199 fn bool_option_val(option: &str, some_strval: Option<&str>) -> bool {
200     if let Some(val) = some_strval {
201         if vec!["yes", "y", "on", "true"].contains(&val) {
202             true
203         } else if vec!["no", "n", "off", "false"].contains(&val) {
204             false
205         } else {
206             bug!(
207                 "Unsupported value `{}` for option `{}` in environment variable {}",
208                 option,
209                 val,
210                 RUSTC_COVERAGE_DEBUG_OPTIONS
211             )
212         }
213     } else {
214         true
215     }
216 }
217
218 fn counter_format_option_val(strval: &str) -> ExpressionFormat {
219     let mut counter_format = ExpressionFormat { id: false, block: false, operation: false };
220     let components = strval.splitn(3, '+');
221     for component in components {
222         match component {
223             "id" => counter_format.id = true,
224             "block" => counter_format.block = true,
225             "operation" => counter_format.operation = true,
226             _ => bug!(
227                 "Unsupported counter_format choice `{}` in environment variable {}",
228                 component,
229                 RUSTC_COVERAGE_DEBUG_OPTIONS
230             ),
231         }
232     }
233     counter_format
234 }
235
236 #[derive(Debug, Clone)]
237 struct ExpressionFormat {
238     id: bool,
239     block: bool,
240     operation: bool,
241 }
242
243 impl Default for ExpressionFormat {
244     fn default() -> Self {
245         Self { id: false, block: true, operation: true }
246     }
247 }
248
249 /// If enabled, this struct maintains a map from `CoverageKind` IDs (as `ExpressionOperandId`) to
250 /// the `CoverageKind` data and optional label (normally, the counter's associated
251 /// `BasicCoverageBlock` format string, if any).
252 ///
253 /// Use `format_counter` to convert one of these `CoverageKind` counters to a debug output string,
254 /// as directed by the `DebugOptions`. This allows the format of counter labels in logs and dump
255 /// files (including the `CoverageGraph` graphviz file) to be changed at runtime, via environment
256 /// variable.
257 ///
258 /// `DebugCounters` supports a recursive rendering of `Expression` counters, so they can be
259 /// presented as nested expressions such as `(bcb3 - (bcb0 + bcb1))`.
260 pub(super) struct DebugCounters {
261     some_counters: Option<FxHashMap<ExpressionOperandId, DebugCounter>>,
262 }
263
264 impl DebugCounters {
265     pub fn new() -> Self {
266         Self { some_counters: None }
267     }
268
269     pub fn enable(&mut self) {
270         debug_assert!(!self.is_enabled());
271         self.some_counters.replace(FxHashMap::default());
272     }
273
274     pub fn is_enabled(&self) -> bool {
275         self.some_counters.is_some()
276     }
277
278     pub fn add_counter(&mut self, counter_kind: &CoverageKind, some_block_label: Option<String>) {
279         if let Some(counters) = &mut self.some_counters {
280             let id: ExpressionOperandId = match *counter_kind {
281                 CoverageKind::Counter { id, .. } => id.into(),
282                 CoverageKind::Expression { id, .. } => id.into(),
283                 _ => bug!(
284                     "the given `CoverageKind` is not an counter or expression: {:?}",
285                     counter_kind
286                 ),
287             };
288             counters
289                 .try_insert(id, DebugCounter::new(counter_kind.clone(), some_block_label))
290                 .expect("attempt to add the same counter_kind to DebugCounters more than once");
291         }
292     }
293
294     pub fn some_block_label(&self, operand: ExpressionOperandId) -> Option<&String> {
295         self.some_counters.as_ref().map_or(None, |counters| {
296             counters
297                 .get(&operand)
298                 .map_or(None, |debug_counter| debug_counter.some_block_label.as_ref())
299         })
300     }
301
302     pub fn format_counter(&self, counter_kind: &CoverageKind) -> String {
303         match *counter_kind {
304             CoverageKind::Counter { .. } => {
305                 format!("Counter({})", self.format_counter_kind(counter_kind))
306             }
307             CoverageKind::Expression { .. } => {
308                 format!("Expression({})", self.format_counter_kind(counter_kind))
309             }
310             CoverageKind::Unreachable { .. } => "Unreachable".to_owned(),
311         }
312     }
313
314     fn format_counter_kind(&self, counter_kind: &CoverageKind) -> String {
315         let counter_format = &debug_options().counter_format;
316         if let CoverageKind::Expression { id, lhs, op, rhs } = *counter_kind {
317             if counter_format.operation {
318                 return format!(
319                     "{}{} {} {}",
320                     if counter_format.id || self.some_counters.is_none() {
321                         format!("#{} = ", id.index())
322                     } else {
323                         String::new()
324                     },
325                     self.format_operand(lhs),
326                     if op == Op::Add { "+" } else { "-" },
327                     self.format_operand(rhs),
328                 );
329             }
330         }
331
332         let id: ExpressionOperandId = match *counter_kind {
333             CoverageKind::Counter { id, .. } => id.into(),
334             CoverageKind::Expression { id, .. } => id.into(),
335             _ => {
336                 bug!("the given `CoverageKind` is not an counter or expression: {:?}", counter_kind)
337             }
338         };
339         if self.some_counters.is_some() && (counter_format.block || !counter_format.id) {
340             let counters = self.some_counters.as_ref().unwrap();
341             if let Some(DebugCounter { some_block_label: Some(block_label), .. }) =
342                 counters.get(&id)
343             {
344                 return if counter_format.id {
345                     format!("{}#{}", block_label, id.index())
346                 } else {
347                     block_label.to_string()
348                 };
349             }
350         }
351         format!("#{}", id.index())
352     }
353
354     fn format_operand(&self, operand: ExpressionOperandId) -> String {
355         if operand.index() == 0 {
356             return String::from("0");
357         }
358         if let Some(counters) = &self.some_counters {
359             if let Some(DebugCounter { counter_kind, some_block_label }) = counters.get(&operand) {
360                 if let CoverageKind::Expression { .. } = counter_kind {
361                     if let Some(block_label) = some_block_label {
362                         if debug_options().counter_format.block {
363                             return format!(
364                                 "{}:({})",
365                                 block_label,
366                                 self.format_counter_kind(counter_kind)
367                             );
368                         }
369                     }
370                     return format!("({})", self.format_counter_kind(counter_kind));
371                 }
372                 return self.format_counter_kind(counter_kind).to_string();
373             }
374         }
375         format!("#{}", operand.index().to_string())
376     }
377 }
378
379 /// A non-public support class to `DebugCounters`.
380 #[derive(Debug)]
381 struct DebugCounter {
382     counter_kind: CoverageKind,
383     some_block_label: Option<String>,
384 }
385
386 impl DebugCounter {
387     fn new(counter_kind: CoverageKind, some_block_label: Option<String>) -> Self {
388         Self { counter_kind, some_block_label }
389     }
390 }
391
392 /// If enabled, this data structure captures additional debugging information used when generating
393 /// a Graphviz (.dot file) representation of the `CoverageGraph`, for debugging purposes.
394 pub(super) struct GraphvizData {
395     some_bcb_to_coverage_spans_with_counters:
396         Option<FxHashMap<BasicCoverageBlock, Vec<(CoverageSpan, CoverageKind)>>>,
397     some_bcb_to_dependency_counters: Option<FxHashMap<BasicCoverageBlock, Vec<CoverageKind>>>,
398     some_edge_to_counter: Option<FxHashMap<(BasicCoverageBlock, BasicBlock), CoverageKind>>,
399 }
400
401 impl GraphvizData {
402     pub fn new() -> Self {
403         Self {
404             some_bcb_to_coverage_spans_with_counters: None,
405             some_bcb_to_dependency_counters: None,
406             some_edge_to_counter: None,
407         }
408     }
409
410     pub fn enable(&mut self) {
411         debug_assert!(!self.is_enabled());
412         self.some_bcb_to_coverage_spans_with_counters = Some(FxHashMap::default());
413         self.some_bcb_to_dependency_counters = Some(FxHashMap::default());
414         self.some_edge_to_counter = Some(FxHashMap::default());
415     }
416
417     pub fn is_enabled(&self) -> bool {
418         self.some_bcb_to_coverage_spans_with_counters.is_some()
419     }
420
421     pub fn add_bcb_coverage_span_with_counter(
422         &mut self,
423         bcb: BasicCoverageBlock,
424         coverage_span: &CoverageSpan,
425         counter_kind: &CoverageKind,
426     ) {
427         if let Some(bcb_to_coverage_spans_with_counters) =
428             self.some_bcb_to_coverage_spans_with_counters.as_mut()
429         {
430             bcb_to_coverage_spans_with_counters
431                 .entry(bcb)
432                 .or_insert_with(Vec::new)
433                 .push((coverage_span.clone(), counter_kind.clone()));
434         }
435     }
436
437     pub fn get_bcb_coverage_spans_with_counters(
438         &self,
439         bcb: BasicCoverageBlock,
440     ) -> Option<&Vec<(CoverageSpan, CoverageKind)>> {
441         if let Some(bcb_to_coverage_spans_with_counters) =
442             self.some_bcb_to_coverage_spans_with_counters.as_ref()
443         {
444             bcb_to_coverage_spans_with_counters.get(&bcb)
445         } else {
446             None
447         }
448     }
449
450     pub fn add_bcb_dependency_counter(
451         &mut self,
452         bcb: BasicCoverageBlock,
453         counter_kind: &CoverageKind,
454     ) {
455         if let Some(bcb_to_dependency_counters) = self.some_bcb_to_dependency_counters.as_mut() {
456             bcb_to_dependency_counters
457                 .entry(bcb)
458                 .or_insert_with(Vec::new)
459                 .push(counter_kind.clone());
460         }
461     }
462
463     pub fn get_bcb_dependency_counters(
464         &self,
465         bcb: BasicCoverageBlock,
466     ) -> Option<&Vec<CoverageKind>> {
467         if let Some(bcb_to_dependency_counters) = self.some_bcb_to_dependency_counters.as_ref() {
468             bcb_to_dependency_counters.get(&bcb)
469         } else {
470             None
471         }
472     }
473
474     pub fn set_edge_counter(
475         &mut self,
476         from_bcb: BasicCoverageBlock,
477         to_bb: BasicBlock,
478         counter_kind: &CoverageKind,
479     ) {
480         if let Some(edge_to_counter) = self.some_edge_to_counter.as_mut() {
481             edge_to_counter
482                 .try_insert((from_bcb, to_bb), counter_kind.clone())
483                 .expect("invalid attempt to insert more than one edge counter for the same edge");
484         }
485     }
486
487     pub fn get_edge_counter(
488         &self,
489         from_bcb: BasicCoverageBlock,
490         to_bb: BasicBlock,
491     ) -> Option<&CoverageKind> {
492         if let Some(edge_to_counter) = self.some_edge_to_counter.as_ref() {
493             edge_to_counter.get(&(from_bcb, to_bb))
494         } else {
495             None
496         }
497     }
498 }
499
500 /// If enabled, this struct captures additional data used to track whether expressions were used,
501 /// directly or indirectly, to compute the coverage counts for all `CoverageSpan`s, and any that are
502 /// _not_ used are retained in the `unused_expressions` Vec, to be included in debug output (logs
503 /// and/or a `CoverageGraph` graphviz output).
504 pub(super) struct UsedExpressions {
505     some_used_expression_operands:
506         Option<FxHashMap<ExpressionOperandId, Vec<InjectedExpressionId>>>,
507     some_unused_expressions:
508         Option<Vec<(CoverageKind, Option<BasicCoverageBlock>, BasicCoverageBlock)>>,
509 }
510
511 impl UsedExpressions {
512     pub fn new() -> Self {
513         Self { some_used_expression_operands: None, some_unused_expressions: None }
514     }
515
516     pub fn enable(&mut self) {
517         debug_assert!(!self.is_enabled());
518         self.some_used_expression_operands = Some(FxHashMap::default());
519         self.some_unused_expressions = Some(Vec::new());
520     }
521
522     pub fn is_enabled(&self) -> bool {
523         self.some_used_expression_operands.is_some()
524     }
525
526     pub fn add_expression_operands(&mut self, expression: &CoverageKind) {
527         if let Some(used_expression_operands) = self.some_used_expression_operands.as_mut() {
528             if let CoverageKind::Expression { id, lhs, rhs, .. } = *expression {
529                 used_expression_operands.entry(lhs).or_insert_with(Vec::new).push(id);
530                 used_expression_operands.entry(rhs).or_insert_with(Vec::new).push(id);
531             }
532         }
533     }
534
535     pub fn expression_is_used(&self, expression: &CoverageKind) -> bool {
536         if let Some(used_expression_operands) = self.some_used_expression_operands.as_ref() {
537             used_expression_operands.contains_key(&expression.as_operand_id())
538         } else {
539             false
540         }
541     }
542
543     pub fn add_unused_expression_if_not_found(
544         &mut self,
545         expression: &CoverageKind,
546         edge_from_bcb: Option<BasicCoverageBlock>,
547         target_bcb: BasicCoverageBlock,
548     ) {
549         if let Some(used_expression_operands) = self.some_used_expression_operands.as_ref() {
550             if !used_expression_operands.contains_key(&expression.as_operand_id()) {
551                 self.some_unused_expressions.as_mut().unwrap().push((
552                     expression.clone(),
553                     edge_from_bcb,
554                     target_bcb,
555                 ));
556             }
557         }
558     }
559
560     /// Return the list of unused counters (if any) as a tuple with the counter (`CoverageKind`),
561     /// optional `from_bcb` (if it was an edge counter), and `target_bcb`.
562     pub fn get_unused_expressions(
563         &self,
564     ) -> Vec<(CoverageKind, Option<BasicCoverageBlock>, BasicCoverageBlock)> {
565         if let Some(unused_expressions) = self.some_unused_expressions.as_ref() {
566             unused_expressions.clone()
567         } else {
568             Vec::new()
569         }
570     }
571
572     /// If enabled, validate that every BCB or edge counter not directly associated with a coverage
573     /// span is at least indirectly associated (it is a dependency of a BCB counter that _is_
574     /// associated with a coverage span).
575     pub fn validate(
576         &mut self,
577         bcb_counters_without_direct_coverage_spans: &Vec<(
578             Option<BasicCoverageBlock>,
579             BasicCoverageBlock,
580             CoverageKind,
581         )>,
582     ) {
583         if self.is_enabled() {
584             let mut not_validated = bcb_counters_without_direct_coverage_spans
585                 .iter()
586                 .map(|(_, _, counter_kind)| counter_kind)
587                 .collect::<Vec<_>>();
588             let mut validating_count = 0;
589             while not_validated.len() != validating_count {
590                 let to_validate = not_validated.split_off(0);
591                 validating_count = to_validate.len();
592                 for counter_kind in to_validate {
593                     if self.expression_is_used(counter_kind) {
594                         self.add_expression_operands(counter_kind);
595                     } else {
596                         not_validated.push(counter_kind);
597                     }
598                 }
599             }
600         }
601     }
602
603     pub fn alert_on_unused_expressions(&self, debug_counters: &DebugCounters) {
604         if let Some(unused_expressions) = self.some_unused_expressions.as_ref() {
605             for (counter_kind, edge_from_bcb, target_bcb) in unused_expressions {
606                 let unused_counter_message = if let Some(from_bcb) = edge_from_bcb.as_ref() {
607                     format!(
608                         "non-coverage edge counter found without a dependent expression, in \
609                         {:?}->{:?}; counter={}",
610                         from_bcb,
611                         target_bcb,
612                         debug_counters.format_counter(&counter_kind),
613                     )
614                 } else {
615                     format!(
616                         "non-coverage counter found without a dependent expression, in {:?}; \
617                         counter={}",
618                         target_bcb,
619                         debug_counters.format_counter(&counter_kind),
620                     )
621                 };
622
623                 if debug_options().allow_unused_expressions {
624                     debug!("WARNING: {}", unused_counter_message);
625                 } else {
626                     bug!("{}", unused_counter_message);
627                 }
628             }
629         }
630     }
631 }
632
633 /// Generates the MIR pass `CoverageSpan`-specific spanview dump file.
634 pub(super) fn dump_coverage_spanview(
635     tcx: TyCtxt<'tcx>,
636     mir_body: &mir::Body<'tcx>,
637     basic_coverage_blocks: &CoverageGraph,
638     pass_name: &str,
639     body_span: Span,
640     coverage_spans: &Vec<CoverageSpan>,
641 ) {
642     let mir_source = mir_body.source;
643     let def_id = mir_source.def_id();
644
645     let span_viewables = span_viewables(tcx, mir_body, basic_coverage_blocks, &coverage_spans);
646     let mut file = pretty::create_dump_file(tcx, "html", None, pass_name, &0, mir_source)
647         .expect("Unexpected error creating MIR spanview HTML file");
648     let crate_name = tcx.crate_name(def_id.krate);
649     let item_name = tcx.def_path(def_id).to_filename_friendly_no_crate();
650     let title = format!("{}.{} - Coverage Spans", crate_name, item_name);
651     spanview::write_document(tcx, body_span, span_viewables, &title, &mut file)
652         .expect("Unexpected IO error dumping coverage spans as HTML");
653 }
654
655 /// Converts the computed `BasicCoverageBlockData`s into `SpanViewable`s.
656 fn span_viewables(
657     tcx: TyCtxt<'tcx>,
658     mir_body: &mir::Body<'tcx>,
659     basic_coverage_blocks: &CoverageGraph,
660     coverage_spans: &Vec<CoverageSpan>,
661 ) -> Vec<SpanViewable> {
662     let mut span_viewables = Vec::new();
663     for coverage_span in coverage_spans {
664         let tooltip = coverage_span.format_coverage_statements(tcx, mir_body);
665         let CoverageSpan { span, bcb, .. } = coverage_span;
666         let bcb_data = &basic_coverage_blocks[*bcb];
667         let id = bcb_data.id();
668         let leader_bb = bcb_data.leader_bb();
669         span_viewables.push(SpanViewable { bb: leader_bb, span: *span, id, tooltip });
670     }
671     span_viewables
672 }
673
674 /// Generates the MIR pass coverage-specific graphviz dump file.
675 pub(super) fn dump_coverage_graphviz(
676     tcx: TyCtxt<'tcx>,
677     mir_body: &mir::Body<'tcx>,
678     pass_name: &str,
679     basic_coverage_blocks: &CoverageGraph,
680     debug_counters: &DebugCounters,
681     graphviz_data: &GraphvizData,
682     intermediate_expressions: &Vec<CoverageKind>,
683     debug_used_expressions: &UsedExpressions,
684 ) {
685     let mir_source = mir_body.source;
686     let def_id = mir_source.def_id();
687     let node_content = |bcb| {
688         bcb_to_string_sections(
689             tcx,
690             mir_body,
691             debug_counters,
692             &basic_coverage_blocks[bcb],
693             graphviz_data.get_bcb_coverage_spans_with_counters(bcb),
694             graphviz_data.get_bcb_dependency_counters(bcb),
695             // intermediate_expressions are injected into the mir::START_BLOCK, so
696             // include them in the first BCB.
697             if bcb.index() == 0 { Some(&intermediate_expressions) } else { None },
698         )
699     };
700     let edge_labels = |from_bcb| {
701         let from_bcb_data = &basic_coverage_blocks[from_bcb];
702         let from_terminator = from_bcb_data.terminator(mir_body);
703         let mut edge_labels = from_terminator.kind.fmt_successor_labels();
704         edge_labels.retain(|label| label != "unreachable");
705         let edge_counters = from_terminator
706             .successors()
707             .map(|&successor_bb| graphviz_data.get_edge_counter(from_bcb, successor_bb));
708         iter::zip(&edge_labels, edge_counters)
709             .map(|(label, some_counter)| {
710                 if let Some(counter) = some_counter {
711                     format!("{}\n{}", label, debug_counters.format_counter(counter))
712                 } else {
713                     label.to_string()
714                 }
715             })
716             .collect::<Vec<_>>()
717     };
718     let graphviz_name = format!("Cov_{}_{}", def_id.krate.index(), def_id.index.index());
719     let mut graphviz_writer =
720         GraphvizWriter::new(basic_coverage_blocks, &graphviz_name, node_content, edge_labels);
721     let unused_expressions = debug_used_expressions.get_unused_expressions();
722     if unused_expressions.len() > 0 {
723         graphviz_writer.set_graph_label(&format!(
724             "Unused expressions:\n  {}",
725             unused_expressions
726                 .as_slice()
727                 .iter()
728                 .map(|(counter_kind, edge_from_bcb, target_bcb)| {
729                     if let Some(from_bcb) = edge_from_bcb.as_ref() {
730                         format!(
731                             "{:?}->{:?}: {}",
732                             from_bcb,
733                             target_bcb,
734                             debug_counters.format_counter(&counter_kind),
735                         )
736                     } else {
737                         format!(
738                             "{:?}: {}",
739                             target_bcb,
740                             debug_counters.format_counter(&counter_kind),
741                         )
742                     }
743                 })
744                 .collect::<Vec<_>>()
745                 .join("\n  ")
746         ));
747     }
748     let mut file = pretty::create_dump_file(tcx, "dot", None, pass_name, &0, mir_source)
749         .expect("Unexpected error creating BasicCoverageBlock graphviz DOT file");
750     graphviz_writer
751         .write_graphviz(tcx, &mut file)
752         .expect("Unexpected error writing BasicCoverageBlock graphviz DOT file");
753 }
754
755 fn bcb_to_string_sections(
756     tcx: TyCtxt<'tcx>,
757     mir_body: &mir::Body<'tcx>,
758     debug_counters: &DebugCounters,
759     bcb_data: &BasicCoverageBlockData,
760     some_coverage_spans_with_counters: Option<&Vec<(CoverageSpan, CoverageKind)>>,
761     some_dependency_counters: Option<&Vec<CoverageKind>>,
762     some_intermediate_expressions: Option<&Vec<CoverageKind>>,
763 ) -> Vec<String> {
764     let len = bcb_data.basic_blocks.len();
765     let mut sections = Vec::new();
766     if let Some(collect_intermediate_expressions) = some_intermediate_expressions {
767         sections.push(
768             collect_intermediate_expressions
769                 .iter()
770                 .map(|expression| {
771                     format!("Intermediate {}", debug_counters.format_counter(expression))
772                 })
773                 .collect::<Vec<_>>()
774                 .join("\n"),
775         );
776     }
777     if let Some(coverage_spans_with_counters) = some_coverage_spans_with_counters {
778         sections.push(
779             coverage_spans_with_counters
780                 .iter()
781                 .map(|(covspan, counter)| {
782                     format!(
783                         "{} at {}",
784                         debug_counters.format_counter(counter),
785                         covspan.format(tcx, mir_body)
786                     )
787                 })
788                 .collect::<Vec<_>>()
789                 .join("\n"),
790         );
791     }
792     if let Some(dependency_counters) = some_dependency_counters {
793         sections.push(format!(
794             "Non-coverage counters:\n  {}",
795             dependency_counters
796                 .iter()
797                 .map(|counter| debug_counters.format_counter(counter))
798                 .collect::<Vec<_>>()
799                 .join("  \n"),
800         ));
801     }
802     if let Some(counter_kind) = &bcb_data.counter_kind {
803         sections.push(format!("{:?}", counter_kind));
804     }
805     let non_term_blocks = bcb_data.basic_blocks[0..len - 1]
806         .iter()
807         .map(|&bb| format!("{:?}: {}", bb, term_type(&mir_body[bb].terminator().kind)))
808         .collect::<Vec<_>>();
809     if non_term_blocks.len() > 0 {
810         sections.push(non_term_blocks.join("\n"));
811     }
812     sections.push(format!(
813         "{:?}: {}",
814         bcb_data.basic_blocks.last().unwrap(),
815         term_type(&bcb_data.terminator(mir_body).kind)
816     ));
817     sections
818 }
819
820 /// Returns a simple string representation of a `TerminatorKind` variant, independent of any
821 /// values it might hold.
822 pub(super) fn term_type(kind: &TerminatorKind<'tcx>) -> &'static str {
823     match kind {
824         TerminatorKind::Goto { .. } => "Goto",
825         TerminatorKind::SwitchInt { .. } => "SwitchInt",
826         TerminatorKind::Resume => "Resume",
827         TerminatorKind::Abort => "Abort",
828         TerminatorKind::Return => "Return",
829         TerminatorKind::Unreachable => "Unreachable",
830         TerminatorKind::Drop { .. } => "Drop",
831         TerminatorKind::DropAndReplace { .. } => "DropAndReplace",
832         TerminatorKind::Call { .. } => "Call",
833         TerminatorKind::Assert { .. } => "Assert",
834         TerminatorKind::Yield { .. } => "Yield",
835         TerminatorKind::GeneratorDrop => "GeneratorDrop",
836         TerminatorKind::FalseEdge { .. } => "FalseEdge",
837         TerminatorKind::FalseUnwind { .. } => "FalseUnwind",
838         TerminatorKind::InlineAsm { .. } => "InlineAsm",
839     }
840 }