]> git.lizzy.rs Git - rust.git/blob - src/tools/compiletest/src/runtest.rs
Rollup merge of #64139 - Mark-Simulacrum:strip-legacy-proc-macro, r=petrochenkov
[rust.git] / src / tools / compiletest / src / runtest.rs
1 // ignore-tidy-filelength
2
3 use crate::common::{CompareMode, PassMode};
4 use crate::common::{expected_output_path, UI_EXTENSIONS, UI_FIXED, UI_STDERR, UI_STDOUT};
5 use crate::common::{UI_RUN_STDERR, UI_RUN_STDOUT};
6 use crate::common::{output_base_dir, output_base_name, output_testname_unique};
7 use crate::common::{Codegen, CodegenUnits, Rustdoc};
8 use crate::common::{DebugInfoCdb, DebugInfoGdbLldb, DebugInfoGdb, DebugInfoLldb};
9 use crate::common::{CompileFail, Pretty, RunFail, RunPassValgrind};
10 use crate::common::{Config, TestPaths};
11 use crate::common::{Incremental, MirOpt, RunMake, Ui, JsDocTest, Assembly};
12 use diff;
13 use crate::errors::{self, Error, ErrorKind};
14 use crate::header::TestProps;
15 use crate::json;
16 use regex::{Captures, Regex};
17 use rustfix::{apply_suggestions, get_suggestions_from_json, Filter};
18 use crate::util::{logv, PathBufExt};
19
20 use std::collections::hash_map::DefaultHasher;
21 use std::collections::{HashMap, HashSet, VecDeque};
22 use std::env;
23 use std::ffi::{OsStr, OsString};
24 use std::fmt;
25 use std::fs::{self, create_dir_all, File, OpenOptions};
26 use std::hash::{Hash, Hasher};
27 use std::io::prelude::*;
28 use std::io::{self, BufReader};
29 use std::path::{Path, PathBuf};
30 use std::process::{Child, Command, ExitStatus, Output, Stdio};
31 use std::str;
32
33 use lazy_static::lazy_static;
34 use log::*;
35
36 use crate::extract_gdb_version;
37 use crate::is_android_gdb_target;
38
39 #[cfg(test)]
40 mod tests;
41
42 #[cfg(windows)]
43 fn disable_error_reporting<F: FnOnce() -> R, R>(f: F) -> R {
44     use std::sync::Mutex;
45     const SEM_NOGPFAULTERRORBOX: u32 = 0x0002;
46     extern "system" {
47         fn SetErrorMode(mode: u32) -> u32;
48     }
49
50     lazy_static! {
51         static ref LOCK: Mutex<()> = { Mutex::new(()) };
52     }
53     // Error mode is a global variable, so lock it so only one thread will change it
54     let _lock = LOCK.lock().unwrap();
55
56     // Tell Windows to not show any UI on errors (such as terminating abnormally).
57     // This is important for running tests, since some of them use abnormal
58     // termination by design. This mode is inherited by all child processes.
59     unsafe {
60         let old_mode = SetErrorMode(SEM_NOGPFAULTERRORBOX); // read inherited flags
61         SetErrorMode(old_mode | SEM_NOGPFAULTERRORBOX);
62         let r = f();
63         SetErrorMode(old_mode);
64         r
65     }
66 }
67
68 #[cfg(not(windows))]
69 fn disable_error_reporting<F: FnOnce() -> R, R>(f: F) -> R {
70     f()
71 }
72
73 /// The name of the environment variable that holds dynamic library locations.
74 pub fn dylib_env_var() -> &'static str {
75     if cfg!(windows) {
76         "PATH"
77     } else if cfg!(target_os = "macos") {
78         "DYLD_LIBRARY_PATH"
79     } else if cfg!(target_os = "haiku") {
80         "LIBRARY_PATH"
81     } else {
82         "LD_LIBRARY_PATH"
83     }
84 }
85
86 /// The platform-specific library name
87 pub fn get_lib_name(lib: &str, dylib: bool) -> String {
88     // In some casess (e.g. MUSL), we build a static
89     // library, rather than a dynamic library.
90     // In this case, the only path we can pass
91     // with '--extern-meta' is the '.lib' file
92     if !dylib {
93         return format!("lib{}.rlib", lib);
94     }
95
96     if cfg!(windows) {
97         format!("{}.dll", lib)
98     } else if cfg!(target_os = "macos") {
99         format!("lib{}.dylib", lib)
100     } else {
101         format!("lib{}.so", lib)
102     }
103 }
104
105 #[derive(Debug, PartialEq)]
106 pub enum DiffLine {
107     Context(String),
108     Expected(String),
109     Resulting(String),
110 }
111
112 #[derive(Debug, PartialEq)]
113 pub struct Mismatch {
114     pub line_number: u32,
115     pub lines: Vec<DiffLine>,
116 }
117
118 impl Mismatch {
119     fn new(line_number: u32) -> Mismatch {
120         Mismatch {
121             line_number: line_number,
122             lines: Vec::new(),
123         }
124     }
125 }
126
127 // Produces a diff between the expected output and actual output.
128 pub fn make_diff(expected: &str, actual: &str, context_size: usize) -> Vec<Mismatch> {
129     let mut line_number = 1;
130     let mut context_queue: VecDeque<&str> = VecDeque::with_capacity(context_size);
131     let mut lines_since_mismatch = context_size + 1;
132     let mut results = Vec::new();
133     let mut mismatch = Mismatch::new(0);
134
135     for result in diff::lines(expected, actual) {
136         match result {
137             diff::Result::Left(str) => {
138                 if lines_since_mismatch >= context_size && lines_since_mismatch > 0 {
139                     results.push(mismatch);
140                     mismatch = Mismatch::new(line_number - context_queue.len() as u32);
141                 }
142
143                 while let Some(line) = context_queue.pop_front() {
144                     mismatch.lines.push(DiffLine::Context(line.to_owned()));
145                 }
146
147                 mismatch.lines.push(DiffLine::Expected(str.to_owned()));
148                 line_number += 1;
149                 lines_since_mismatch = 0;
150             }
151             diff::Result::Right(str) => {
152                 if lines_since_mismatch >= context_size && lines_since_mismatch > 0 {
153                     results.push(mismatch);
154                     mismatch = Mismatch::new(line_number - context_queue.len() as u32);
155                 }
156
157                 while let Some(line) = context_queue.pop_front() {
158                     mismatch.lines.push(DiffLine::Context(line.to_owned()));
159                 }
160
161                 mismatch.lines.push(DiffLine::Resulting(str.to_owned()));
162                 lines_since_mismatch = 0;
163             }
164             diff::Result::Both(str, _) => {
165                 if context_queue.len() >= context_size {
166                     let _ = context_queue.pop_front();
167                 }
168
169                 if lines_since_mismatch < context_size {
170                     mismatch.lines.push(DiffLine::Context(str.to_owned()));
171                 } else if context_size > 0 {
172                     context_queue.push_back(str);
173                 }
174
175                 line_number += 1;
176                 lines_since_mismatch += 1;
177             }
178         }
179     }
180
181     results.push(mismatch);
182     results.remove(0);
183
184     results
185 }
186
187 pub fn run(config: Config, testpaths: &TestPaths, revision: Option<&str>) {
188     match &*config.target {
189         "arm-linux-androideabi"
190         | "armv7-linux-androideabi"
191         | "thumbv7neon-linux-androideabi"
192         | "aarch64-linux-android" => {
193             if !config.adb_device_status {
194                 panic!("android device not available");
195             }
196         }
197
198         _ => {
199             // android has its own gdb handling
200             if config.mode == DebugInfoGdb && config.gdb.is_none() {
201                 panic!("gdb not available but debuginfo gdb debuginfo test requested");
202             }
203         }
204     }
205
206     if config.verbose {
207         // We're going to be dumping a lot of info. Start on a new line.
208         print!("\n\n");
209     }
210     debug!("running {:?}", testpaths.file.display());
211     let props = TestProps::from_file(&testpaths.file, revision, &config);
212
213     let cx = TestCx {
214         config: &config,
215         props: &props,
216         testpaths,
217         revision: revision,
218     };
219     create_dir_all(&cx.output_base_dir()).unwrap();
220
221     if config.mode == Incremental {
222         // Incremental tests are special because they cannot be run in
223         // parallel.
224         assert!(
225             !props.revisions.is_empty(),
226             "Incremental tests require revisions."
227         );
228         cx.init_incremental_test();
229         for revision in &props.revisions {
230             let revision_props = TestProps::from_file(&testpaths.file, Some(revision), &config);
231             let rev_cx = TestCx {
232                 config: &config,
233                 props: &revision_props,
234                 testpaths,
235                 revision: Some(revision),
236             };
237             rev_cx.run_revision();
238         }
239     } else {
240         cx.run_revision();
241     }
242
243     cx.create_stamp();
244 }
245
246 pub fn compute_stamp_hash(config: &Config) -> String {
247     let mut hash = DefaultHasher::new();
248     config.stage_id.hash(&mut hash);
249
250     if config.mode == DebugInfoCdb {
251         config.cdb.hash(&mut hash);
252     }
253
254     if config.mode == DebugInfoGdb || config.mode == DebugInfoGdbLldb {
255         match config.gdb {
256             None => env::var_os("PATH").hash(&mut hash),
257             Some(ref s) if s.is_empty() => env::var_os("PATH").hash(&mut hash),
258             Some(ref s) => s.hash(&mut hash),
259         };
260     }
261
262     if config.mode == DebugInfoLldb || config.mode == DebugInfoGdbLldb {
263         env::var_os("PATH").hash(&mut hash);
264         env::var_os("PYTHONPATH").hash(&mut hash);
265     }
266
267     if let Ui | Incremental | Pretty = config.mode {
268         config.force_pass_mode.hash(&mut hash);
269     }
270
271     format!("{:x}", hash.finish())
272 }
273
274 struct TestCx<'test> {
275     config: &'test Config,
276     props: &'test TestProps,
277     testpaths: &'test TestPaths,
278     revision: Option<&'test str>,
279 }
280
281 struct DebuggerCommands {
282     commands: Vec<String>,
283     check_lines: Vec<String>,
284     breakpoint_lines: Vec<usize>,
285 }
286
287 enum ReadFrom {
288     Path,
289     Stdin(String),
290 }
291
292 enum TestOutput {
293     Compile,
294     Run,
295 }
296
297 impl<'test> TestCx<'test> {
298     /// Code executed for each revision in turn (or, if there are no
299     /// revisions, exactly once, with revision == None).
300     fn run_revision(&self) {
301         match self.config.mode {
302             CompileFail => self.run_cfail_test(),
303             RunFail => self.run_rfail_test(),
304             RunPassValgrind => self.run_valgrind_test(),
305             Pretty => self.run_pretty_test(),
306             DebugInfoGdbLldb => {
307                 self.run_debuginfo_gdb_test();
308                 self.run_debuginfo_lldb_test();
309             },
310             DebugInfoCdb => self.run_debuginfo_cdb_test(),
311             DebugInfoGdb => self.run_debuginfo_gdb_test(),
312             DebugInfoLldb => self.run_debuginfo_lldb_test(),
313             Codegen => self.run_codegen_test(),
314             Rustdoc => self.run_rustdoc_test(),
315             CodegenUnits => self.run_codegen_units_test(),
316             Incremental => self.run_incremental_test(),
317             RunMake => self.run_rmake_test(),
318             Ui => self.run_ui_test(),
319             MirOpt => self.run_mir_opt_test(),
320             Assembly => self.run_assembly_test(),
321             JsDocTest => self.run_js_doc_test(),
322         }
323     }
324
325     fn pass_mode(&self) -> Option<PassMode> {
326         self.props.pass_mode(self.config)
327     }
328
329     fn should_run_successfully(&self) -> bool {
330         let pass_mode = self.pass_mode();
331         match self.config.mode {
332             Ui => pass_mode == Some(PassMode::Run),
333             mode => panic!("unimplemented for mode {:?}", mode),
334         }
335     }
336
337     fn should_compile_successfully(&self) -> bool {
338         match self.config.mode {
339             CompileFail => false,
340             JsDocTest => true,
341             Ui => self.pass_mode().is_some(),
342             Incremental => {
343                 let revision = self.revision
344                     .expect("incremental tests require a list of revisions");
345                 if revision.starts_with("rpass") || revision.starts_with("rfail") {
346                     true
347                 } else if revision.starts_with("cfail") {
348                     // FIXME: would be nice if incremental revs could start with "cpass"
349                     self.pass_mode().is_some()
350                 } else {
351                     panic!("revision name must begin with rpass, rfail, or cfail");
352                 }
353             }
354             mode => panic!("unimplemented for mode {:?}", mode),
355         }
356     }
357
358     fn check_if_test_should_compile(&self, proc_res: &ProcRes) {
359         if self.should_compile_successfully() {
360             if !proc_res.status.success() {
361                 self.fatal_proc_rec("test compilation failed although it shouldn't!", proc_res);
362             }
363         } else {
364             if proc_res.status.success() {
365                 self.fatal_proc_rec(
366                     &format!("{} test compiled successfully!", self.config.mode)[..],
367                     proc_res,
368                 );
369             }
370
371             self.check_correct_failure_status(proc_res);
372         }
373     }
374
375     fn run_cfail_test(&self) {
376         let proc_res = self.compile_test();
377         self.check_if_test_should_compile(&proc_res);
378         self.check_no_compiler_crash(&proc_res);
379
380         let output_to_check = self.get_output(&proc_res);
381         let expected_errors = errors::load_errors(&self.testpaths.file, self.revision);
382         if !expected_errors.is_empty() {
383             if !self.props.error_patterns.is_empty() {
384                 self.fatal("both error pattern and expected errors specified");
385             }
386             self.check_expected_errors(expected_errors, &proc_res);
387         } else {
388             self.check_error_patterns(&output_to_check, &proc_res);
389         }
390
391         self.check_forbid_output(&output_to_check, &proc_res);
392     }
393
394     fn run_rfail_test(&self) {
395         let proc_res = self.compile_test();
396
397         if !proc_res.status.success() {
398             self.fatal_proc_rec("compilation failed!", &proc_res);
399         }
400
401         let proc_res = self.exec_compiled_test();
402
403         // The value our Makefile configures valgrind to return on failure
404         const VALGRIND_ERR: i32 = 100;
405         if proc_res.status.code() == Some(VALGRIND_ERR) {
406             self.fatal_proc_rec("run-fail test isn't valgrind-clean!", &proc_res);
407         }
408
409         let output_to_check = self.get_output(&proc_res);
410         self.check_correct_failure_status(&proc_res);
411         self.check_error_patterns(&output_to_check, &proc_res);
412     }
413
414     fn get_output(&self, proc_res: &ProcRes) -> String {
415         if self.props.check_stdout {
416             format!("{}{}", proc_res.stdout, proc_res.stderr)
417         } else {
418             proc_res.stderr.clone()
419         }
420     }
421
422     fn check_correct_failure_status(&self, proc_res: &ProcRes) {
423         let expected_status = Some(self.props.failure_status);
424         let received_status = proc_res.status.code();
425
426         if expected_status != received_status {
427             self.fatal_proc_rec(
428                 &format!(
429                     "Error: expected failure status ({:?}) but received status {:?}.",
430                     expected_status, received_status
431                 ),
432                 proc_res,
433             );
434         }
435     }
436
437     fn run_rpass_test(&self) {
438         let proc_res = self.compile_test();
439
440         if !proc_res.status.success() {
441             self.fatal_proc_rec("compilation failed!", &proc_res);
442         }
443
444         // FIXME(#41968): Move this check to tidy?
445         let expected_errors = errors::load_errors(&self.testpaths.file, self.revision);
446         assert!(
447             expected_errors.is_empty(),
448             "run-pass tests with expected warnings should be moved to ui/"
449         );
450
451         let proc_res = self.exec_compiled_test();
452         if !proc_res.status.success() {
453             self.fatal_proc_rec("test run failed!", &proc_res);
454         }
455     }
456
457     fn run_valgrind_test(&self) {
458         assert!(self.revision.is_none(), "revisions not relevant here");
459
460         if self.config.valgrind_path.is_none() {
461             assert!(!self.config.force_valgrind);
462             return self.run_rpass_test();
463         }
464
465         let mut proc_res = self.compile_test();
466
467         if !proc_res.status.success() {
468             self.fatal_proc_rec("compilation failed!", &proc_res);
469         }
470
471         let mut new_config = self.config.clone();
472         new_config.runtool = new_config.valgrind_path.clone();
473         let new_cx = TestCx {
474             config: &new_config,
475             ..*self
476         };
477         proc_res = new_cx.exec_compiled_test();
478
479         if !proc_res.status.success() {
480             self.fatal_proc_rec("test run failed!", &proc_res);
481         }
482     }
483
484     fn run_pretty_test(&self) {
485         if self.props.pp_exact.is_some() {
486             logv(self.config, "testing for exact pretty-printing".to_owned());
487         } else {
488             logv(
489                 self.config,
490                 "testing for converging pretty-printing".to_owned(),
491             );
492         }
493
494         let rounds = match self.props.pp_exact {
495             Some(_) => 1,
496             None => 2,
497         };
498
499         let src = fs::read_to_string(&self.testpaths.file).unwrap();
500         let mut srcs = vec![src];
501
502         let mut round = 0;
503         while round < rounds {
504             logv(
505                 self.config,
506                 format!(
507                     "pretty-printing round {} revision {:?}",
508                     round, self.revision
509                 ),
510             );
511             let read_from = if round == 0 {
512                 ReadFrom::Path
513             } else {
514                 ReadFrom::Stdin(srcs[round].to_owned())
515             };
516
517             let proc_res = self.print_source(read_from,
518                                              &self.props.pretty_mode);
519             if !proc_res.status.success() {
520                 self.fatal_proc_rec(
521                     &format!(
522                         "pretty-printing failed in round {} revision {:?}",
523                         round, self.revision
524                     ),
525                     &proc_res,
526                 );
527             }
528
529             let ProcRes { stdout, .. } = proc_res;
530             srcs.push(stdout);
531             round += 1;
532         }
533
534         let mut expected = match self.props.pp_exact {
535             Some(ref file) => {
536                 let filepath = self.testpaths.file.parent().unwrap().join(file);
537                 fs::read_to_string(&filepath).unwrap()
538             }
539             None => srcs[srcs.len() - 2].clone(),
540         };
541         let mut actual = srcs[srcs.len() - 1].clone();
542
543         if self.props.pp_exact.is_some() {
544             // Now we have to care about line endings
545             let cr = "\r".to_owned();
546             actual = actual.replace(&cr, "").to_owned();
547             expected = expected.replace(&cr, "").to_owned();
548         }
549
550         self.compare_source(&expected, &actual);
551
552         // If we're only making sure that the output matches then just stop here
553         if self.props.pretty_compare_only {
554             return;
555         }
556
557         // Finally, let's make sure it actually appears to remain valid code
558         let proc_res = self.typecheck_source(actual);
559         if !proc_res.status.success() {
560             self.fatal_proc_rec("pretty-printed source does not typecheck", &proc_res);
561         }
562
563         if !self.props.pretty_expanded {
564             return;
565         }
566
567         // additionally, run `--pretty expanded` and try to build it.
568         let proc_res = self.print_source(ReadFrom::Path, "expanded");
569         if !proc_res.status.success() {
570             self.fatal_proc_rec("pretty-printing (expanded) failed", &proc_res);
571         }
572
573         let ProcRes {
574             stdout: expanded_src,
575             ..
576         } = proc_res;
577         let proc_res = self.typecheck_source(expanded_src);
578         if !proc_res.status.success() {
579             self.fatal_proc_rec(
580                 "pretty-printed source (expanded) does not typecheck",
581                 &proc_res,
582             );
583         }
584     }
585
586     fn print_source(&self, read_from: ReadFrom, pretty_type: &str) -> ProcRes {
587         let aux_dir = self.aux_output_dir_name();
588         let input: &str = match read_from {
589             ReadFrom::Stdin(_) => "-",
590             ReadFrom::Path => self.testpaths.file.to_str().unwrap(),
591         };
592
593         let mut rustc = Command::new(&self.config.rustc_path);
594         rustc
595             .arg(input)
596             .args(&["-Z", &format!("unpretty={}", pretty_type)])
597             .args(&["--target", &self.config.target])
598             .arg("-L")
599             .arg(&aux_dir)
600             .args(&self.props.compile_flags)
601             .envs(self.props.exec_env.clone());
602         self.maybe_add_external_args(&mut rustc,
603                                      self.split_maybe_args(&self.config.target_rustcflags));
604
605         let src = match read_from {
606             ReadFrom::Stdin(src) => Some(src),
607             ReadFrom::Path => None
608         };
609
610         self.compose_and_run(
611             rustc,
612             self.config.compile_lib_path.to_str().unwrap(),
613             Some(aux_dir.to_str().unwrap()),
614             src,
615         )
616     }
617
618     fn compare_source(&self, expected: &str, actual: &str) {
619         if expected != actual {
620             self.fatal(&format!(
621                 "pretty-printed source does not match expected source\n\
622                  expected:\n\
623                  ------------------------------------------\n\
624                  {}\n\
625                  ------------------------------------------\n\
626                  actual:\n\
627                  ------------------------------------------\n\
628                  {}\n\
629                  ------------------------------------------\n\
630                  \n",
631                 expected, actual)
632             );
633         }
634     }
635
636     fn set_revision_flags(&self, cmd: &mut Command) {
637         if let Some(revision) = self.revision {
638             // Normalize revisions to be lowercase and replace `-`s with `_`s.
639             // Otherwise the `--cfg` flag is not valid.
640             let normalized_revision = revision.to_lowercase().replace("-", "_");
641             cmd.args(&["--cfg", &normalized_revision]);
642         }
643     }
644
645     fn typecheck_source(&self, src: String) -> ProcRes {
646         let mut rustc = Command::new(&self.config.rustc_path);
647
648         let out_dir = self.output_base_name().with_extension("pretty-out");
649         let _ = fs::remove_dir_all(&out_dir);
650         create_dir_all(&out_dir).unwrap();
651
652         let target = if self.props.force_host {
653             &*self.config.host
654         } else {
655             &*self.config.target
656         };
657
658         let aux_dir = self.aux_output_dir_name();
659
660         rustc
661             .arg("-")
662             .arg("-Zno-codegen")
663             .arg("--out-dir")
664             .arg(&out_dir)
665             .arg(&format!("--target={}", target))
666             .arg("-L")
667             .arg(&self.config.build_base)
668             .arg("-L")
669             .arg(aux_dir);
670         self.set_revision_flags(&mut rustc);
671         self.maybe_add_external_args(&mut rustc,
672                                      self.split_maybe_args(&self.config.target_rustcflags));
673         rustc.args(&self.props.compile_flags);
674
675         self.compose_and_run_compiler(rustc, Some(src))
676     }
677
678     fn run_debuginfo_cdb_test(&self) {
679         assert!(self.revision.is_none(), "revisions not relevant here");
680
681         let config = Config {
682             target_rustcflags: self.cleanup_debug_info_options(&self.config.target_rustcflags),
683             host_rustcflags: self.cleanup_debug_info_options(&self.config.host_rustcflags),
684             mode: DebugInfoCdb,
685             ..self.config.clone()
686         };
687
688         let test_cx = TestCx {
689             config: &config,
690             ..*self
691         };
692
693         test_cx.run_debuginfo_cdb_test_no_opt();
694     }
695
696     fn run_debuginfo_cdb_test_no_opt(&self) {
697         // compile test file (it should have 'compile-flags:-g' in the header)
698         let compile_result = self.compile_test();
699         if !compile_result.status.success() {
700             self.fatal_proc_rec("compilation failed!", &compile_result);
701         }
702
703         let exe_file = self.make_exe_name();
704
705         let prefixes = {
706             static PREFIXES: &'static [&'static str] = &["cdb", "cdbg"];
707             // No "native rust support" variation for CDB yet.
708             PREFIXES
709         };
710
711         // Parse debugger commands etc from test files
712         let DebuggerCommands {
713             commands,
714             check_lines,
715             breakpoint_lines,
716             ..
717         } = self.parse_debugger_commands(prefixes);
718
719         // https://docs.microsoft.com/en-us/windows-hardware/drivers/debugger/debugger-commands
720         let mut script_str = String::with_capacity(2048);
721         script_str.push_str("version\n"); // List CDB (and more) version info in test output
722         script_str.push_str(".nvlist\n"); // List loaded `*.natvis` files, bulk of custom MSVC debug
723
724         // Set breakpoints on every line that contains the string "#break"
725         let source_file_name = self.testpaths.file.file_name().unwrap().to_string_lossy();
726         for line in &breakpoint_lines {
727             script_str.push_str(&format!(
728                 "bp `{}:{}`\n",
729                 source_file_name, line
730             ));
731         }
732
733         // Append the other `cdb-command:`s
734         for line in &commands {
735             script_str.push_str(line);
736             script_str.push_str("\n");
737         }
738
739         script_str.push_str("\nqq\n"); // Quit the debugger (including remote debugger, if any)
740
741         // Write the script into a file
742         debug!("script_str = {}", script_str);
743         self.dump_output_file(&script_str, "debugger.script");
744         let debugger_script = self.make_out_name("debugger.script");
745
746         let cdb_path = &self.config.cdb.as_ref().unwrap();
747         let mut cdb = Command::new(cdb_path);
748         cdb
749             .arg("-lines") // Enable source line debugging.
750             .arg("-cf").arg(&debugger_script)
751             .arg(&exe_file);
752
753         let debugger_run_result = self.compose_and_run(
754             cdb,
755             self.config.run_lib_path.to_str().unwrap(),
756             None, // aux_path
757             None  // input
758         );
759
760         if !debugger_run_result.status.success() {
761             self.fatal_proc_rec("Error while running CDB", &debugger_run_result);
762         }
763
764         self.check_debugger_output(&debugger_run_result, &check_lines);
765     }
766
767     fn run_debuginfo_gdb_test(&self) {
768         assert!(self.revision.is_none(), "revisions not relevant here");
769
770         let config = Config {
771             target_rustcflags: self.cleanup_debug_info_options(&self.config.target_rustcflags),
772             host_rustcflags: self.cleanup_debug_info_options(&self.config.host_rustcflags),
773             mode: DebugInfoGdb,
774             ..self.config.clone()
775         };
776
777         let test_cx = TestCx {
778             config: &config,
779             ..*self
780         };
781
782         test_cx.run_debuginfo_gdb_test_no_opt();
783     }
784
785     fn run_debuginfo_gdb_test_no_opt(&self) {
786         let prefixes = if self.config.gdb_native_rust {
787             // GDB with Rust
788             static PREFIXES: &'static [&'static str] = &["gdb", "gdbr"];
789             println!("NOTE: compiletest thinks it is using GDB with native rust support");
790             PREFIXES
791         } else {
792             // Generic GDB
793             static PREFIXES: &'static [&'static str] = &["gdb", "gdbg"];
794             println!("NOTE: compiletest thinks it is using GDB without native rust support");
795             PREFIXES
796         };
797
798         let DebuggerCommands {
799             commands,
800             check_lines,
801             breakpoint_lines,
802         } = self.parse_debugger_commands(prefixes);
803         let mut cmds = commands.join("\n");
804
805         // compile test file (it should have 'compile-flags:-g' in the header)
806         let compiler_run_result = self.compile_test();
807         if !compiler_run_result.status.success() {
808             self.fatal_proc_rec("compilation failed!", &compiler_run_result);
809         }
810
811         let exe_file = self.make_exe_name();
812
813         let debugger_run_result;
814         if is_android_gdb_target(&self.config.target) {
815             cmds = cmds.replace("run", "continue");
816
817             let tool_path = match self.config.android_cross_path.to_str() {
818                 Some(x) => x.to_owned(),
819                 None => self.fatal("cannot find android cross path"),
820             };
821
822             // write debugger script
823             let mut script_str = String::with_capacity(2048);
824             script_str.push_str(&format!("set charset {}\n", Self::charset()));
825             script_str.push_str(&format!("set sysroot {}\n", tool_path));
826             script_str.push_str(&format!("file {}\n", exe_file.to_str().unwrap()));
827             script_str.push_str("target remote :5039\n");
828             script_str.push_str(&format!(
829                 "set solib-search-path \
830                  ./{}/stage2/lib/rustlib/{}/lib/\n",
831                 self.config.host, self.config.target
832             ));
833             for line in &breakpoint_lines {
834                 script_str.push_str(
835                     &format!(
836                         "break {:?}:{}\n",
837                         self.testpaths.file.file_name().unwrap().to_string_lossy(),
838                         *line
839                     )[..],
840                 );
841             }
842             script_str.push_str(&cmds);
843             script_str.push_str("\nquit\n");
844
845             debug!("script_str = {}", script_str);
846             self.dump_output_file(&script_str, "debugger.script");
847
848             let adb_path = &self.config.adb_path;
849
850             Command::new(adb_path)
851                 .arg("push")
852                 .arg(&exe_file)
853                 .arg(&self.config.adb_test_dir)
854                 .status()
855                 .expect(&format!("failed to exec `{:?}`", adb_path));
856
857             Command::new(adb_path)
858                 .args(&["forward", "tcp:5039", "tcp:5039"])
859                 .status()
860                 .expect(&format!("failed to exec `{:?}`", adb_path));
861
862             let adb_arg = format!(
863                 "export LD_LIBRARY_PATH={}; \
864                  gdbserver{} :5039 {}/{}",
865                 self.config.adb_test_dir.clone(),
866                 if self.config.target.contains("aarch64") {
867                     "64"
868                 } else {
869                     ""
870                 },
871                 self.config.adb_test_dir.clone(),
872                 exe_file.file_name().unwrap().to_str().unwrap()
873             );
874
875             debug!("adb arg: {}", adb_arg);
876             let mut adb = Command::new(adb_path)
877                 .args(&["shell", &adb_arg])
878                 .stdout(Stdio::piped())
879                 .stderr(Stdio::inherit())
880                 .spawn()
881                 .expect(&format!("failed to exec `{:?}`", adb_path));
882
883             // Wait for the gdbserver to print out "Listening on port ..."
884             // at which point we know that it's started and then we can
885             // execute the debugger below.
886             let mut stdout = BufReader::new(adb.stdout.take().unwrap());
887             let mut line = String::new();
888             loop {
889                 line.truncate(0);
890                 stdout.read_line(&mut line).unwrap();
891                 if line.starts_with("Listening on port 5039") {
892                     break;
893                 }
894             }
895             drop(stdout);
896
897             let mut debugger_script = OsString::from("-command=");
898             debugger_script.push(self.make_out_name("debugger.script"));
899             let debugger_opts: &[&OsStr] = &[
900                 "-quiet".as_ref(),
901                 "-batch".as_ref(),
902                 "-nx".as_ref(),
903                 &debugger_script,
904             ];
905
906             let gdb_path = self.config.gdb.as_ref().unwrap();
907             let Output {
908                 status,
909                 stdout,
910                 stderr,
911             } = Command::new(&gdb_path)
912                 .args(debugger_opts)
913                 .output()
914                 .expect(&format!("failed to exec `{:?}`", gdb_path));
915             let cmdline = {
916                 let mut gdb = Command::new(&format!("{}-gdb", self.config.target));
917                 gdb.args(debugger_opts);
918                 let cmdline = self.make_cmdline(&gdb, "");
919                 logv(self.config, format!("executing {}", cmdline));
920                 cmdline
921             };
922
923             debugger_run_result = ProcRes {
924                 status,
925                 stdout: String::from_utf8(stdout).unwrap(),
926                 stderr: String::from_utf8(stderr).unwrap(),
927                 cmdline,
928             };
929             if adb.kill().is_err() {
930                 println!("Adb process is already finished.");
931             }
932         } else {
933             let rust_src_root = self
934                 .config
935                 .find_rust_src_root()
936                 .expect("Could not find Rust source root");
937             let rust_pp_module_rel_path = Path::new("./src/etc");
938             let rust_pp_module_abs_path = rust_src_root
939                 .join(rust_pp_module_rel_path)
940                 .to_str()
941                 .unwrap()
942                 .to_owned();
943             // write debugger script
944             let mut script_str = String::with_capacity(2048);
945             script_str.push_str(&format!("set charset {}\n", Self::charset()));
946             script_str.push_str("show version\n");
947
948             match self.config.gdb_version {
949                 Some(version) => {
950                     println!(
951                         "NOTE: compiletest thinks it is using GDB version {}",
952                         version
953                     );
954
955                     if version > extract_gdb_version("7.4").unwrap() {
956                         // Add the directory containing the pretty printers to
957                         // GDB's script auto loading safe path
958                         script_str.push_str(&format!(
959                             "add-auto-load-safe-path {}\n",
960                             rust_pp_module_abs_path.replace(r"\", r"\\")
961                         ));
962                     }
963                 }
964                 _ => {
965                     println!(
966                         "NOTE: compiletest does not know which version of \
967                          GDB it is using"
968                     );
969                 }
970             }
971
972             // The following line actually doesn't have to do anything with
973             // pretty printing, it just tells GDB to print values on one line:
974             script_str.push_str("set print pretty off\n");
975
976             // Add the pretty printer directory to GDB's source-file search path
977             script_str.push_str(&format!("directory {}\n", rust_pp_module_abs_path));
978
979             // Load the target executable
980             script_str.push_str(&format!(
981                 "file {}\n",
982                 exe_file.to_str().unwrap().replace(r"\", r"\\")
983             ));
984
985             // Force GDB to print values in the Rust format.
986             if self.config.gdb_native_rust {
987                 script_str.push_str("set language rust\n");
988             }
989
990             // Add line breakpoints
991             for line in &breakpoint_lines {
992                 script_str.push_str(&format!(
993                     "break '{}':{}\n",
994                     self.testpaths.file.file_name().unwrap().to_string_lossy(),
995                     *line
996                 ));
997             }
998
999             script_str.push_str(&cmds);
1000             script_str.push_str("\nquit\n");
1001
1002             debug!("script_str = {}", script_str);
1003             self.dump_output_file(&script_str, "debugger.script");
1004
1005             let mut debugger_script = OsString::from("-command=");
1006             debugger_script.push(self.make_out_name("debugger.script"));
1007
1008             let debugger_opts: &[&OsStr] = &[
1009                 "-quiet".as_ref(),
1010                 "-batch".as_ref(),
1011                 "-nx".as_ref(),
1012                 &debugger_script,
1013             ];
1014
1015             let mut gdb = Command::new(self.config.gdb.as_ref().unwrap());
1016             gdb.args(debugger_opts)
1017                 .env("PYTHONPATH", rust_pp_module_abs_path);
1018
1019             debugger_run_result = self.compose_and_run(
1020                 gdb,
1021                 self.config.run_lib_path.to_str().unwrap(),
1022                 None,
1023                 None,
1024             );
1025         }
1026
1027         if !debugger_run_result.status.success() {
1028             self.fatal_proc_rec("gdb failed to execute", &debugger_run_result);
1029         }
1030
1031         self.check_debugger_output(&debugger_run_result, &check_lines);
1032     }
1033
1034     fn run_debuginfo_lldb_test(&self) {
1035         assert!(self.revision.is_none(), "revisions not relevant here");
1036
1037         if self.config.lldb_python_dir.is_none() {
1038             self.fatal("Can't run LLDB test because LLDB's python path is not set.");
1039         }
1040
1041         let config = Config {
1042             target_rustcflags: self.cleanup_debug_info_options(&self.config.target_rustcflags),
1043             host_rustcflags: self.cleanup_debug_info_options(&self.config.host_rustcflags),
1044             mode: DebugInfoLldb,
1045             ..self.config.clone()
1046         };
1047
1048         let test_cx = TestCx {
1049             config: &config,
1050             ..*self
1051         };
1052
1053         test_cx.run_debuginfo_lldb_test_no_opt();
1054     }
1055
1056     fn run_debuginfo_lldb_test_no_opt(&self) {
1057         // compile test file (it should have 'compile-flags:-g' in the header)
1058         let compile_result = self.compile_test();
1059         if !compile_result.status.success() {
1060             self.fatal_proc_rec("compilation failed!", &compile_result);
1061         }
1062
1063         let exe_file = self.make_exe_name();
1064
1065         match self.config.lldb_version {
1066             Some(ref version) => {
1067                 println!(
1068                     "NOTE: compiletest thinks it is using LLDB version {}",
1069                     version
1070                 );
1071             }
1072             _ => {
1073                 println!(
1074                     "NOTE: compiletest does not know which version of \
1075                      LLDB it is using"
1076                 );
1077             }
1078         }
1079
1080         let prefixes = if self.config.lldb_native_rust {
1081             static PREFIXES: &'static [&'static str] = &["lldb", "lldbr"];
1082             println!("NOTE: compiletest thinks it is using LLDB with native rust support");
1083             PREFIXES
1084         } else {
1085             static PREFIXES: &'static [&'static str] = &["lldb", "lldbg"];
1086             println!("NOTE: compiletest thinks it is using LLDB without native rust support");
1087             PREFIXES
1088         };
1089
1090         // Parse debugger commands etc from test files
1091         let DebuggerCommands {
1092             commands,
1093             check_lines,
1094             breakpoint_lines,
1095             ..
1096         } = self.parse_debugger_commands(prefixes);
1097
1098         // Write debugger script:
1099         // We don't want to hang when calling `quit` while the process is still running
1100         let mut script_str = String::from("settings set auto-confirm true\n");
1101
1102         // Make LLDB emit its version, so we have it documented in the test output
1103         script_str.push_str("version\n");
1104
1105         // Switch LLDB into "Rust mode"
1106         let rust_src_root = self
1107             .config
1108             .find_rust_src_root()
1109             .expect("Could not find Rust source root");
1110         let rust_pp_module_rel_path = Path::new("./src/etc/lldb_rust_formatters.py");
1111         let rust_pp_module_abs_path = rust_src_root
1112             .join(rust_pp_module_rel_path)
1113             .to_str()
1114             .unwrap()
1115             .to_owned();
1116
1117         script_str
1118             .push_str(&format!("command script import {}\n", &rust_pp_module_abs_path[..])[..]);
1119         script_str.push_str("type summary add --no-value ");
1120         script_str.push_str("--python-function lldb_rust_formatters.print_val ");
1121         script_str.push_str("-x \".*\" --category Rust\n");
1122         script_str.push_str("type category enable Rust\n");
1123
1124         // Set breakpoints on every line that contains the string "#break"
1125         let source_file_name = self.testpaths.file.file_name().unwrap().to_string_lossy();
1126         for line in &breakpoint_lines {
1127             script_str.push_str(&format!(
1128                 "breakpoint set --file '{}' --line {}\n",
1129                 source_file_name, line
1130             ));
1131         }
1132
1133         // Append the other commands
1134         for line in &commands {
1135             script_str.push_str(line);
1136             script_str.push_str("\n");
1137         }
1138
1139         // Finally, quit the debugger
1140         script_str.push_str("\nquit\n");
1141
1142         // Write the script into a file
1143         debug!("script_str = {}", script_str);
1144         self.dump_output_file(&script_str, "debugger.script");
1145         let debugger_script = self.make_out_name("debugger.script");
1146
1147         // Let LLDB execute the script via lldb_batchmode.py
1148         let debugger_run_result = self.run_lldb(&exe_file, &debugger_script, &rust_src_root);
1149
1150         if !debugger_run_result.status.success() {
1151             self.fatal_proc_rec("Error while running LLDB", &debugger_run_result);
1152         }
1153
1154         self.check_debugger_output(&debugger_run_result, &check_lines);
1155     }
1156
1157     fn run_lldb(
1158         &self,
1159         test_executable: &Path,
1160         debugger_script: &Path,
1161         rust_src_root: &Path,
1162     ) -> ProcRes {
1163         // Prepare the lldb_batchmode which executes the debugger script
1164         let lldb_script_path = rust_src_root.join("src/etc/lldb_batchmode.py");
1165         self.cmd2procres(
1166             Command::new(&self.config.lldb_python)
1167                 .arg(&lldb_script_path)
1168                 .arg(test_executable)
1169                 .arg(debugger_script)
1170                 .env("PYTHONPATH", self.config.lldb_python_dir.as_ref().unwrap()),
1171         )
1172     }
1173
1174     fn cmd2procres(&self, cmd: &mut Command) -> ProcRes {
1175         let (status, out, err) = match cmd.output() {
1176             Ok(Output {
1177                 status,
1178                 stdout,
1179                 stderr,
1180             }) => (
1181                 status,
1182                 String::from_utf8(stdout).unwrap(),
1183                 String::from_utf8(stderr).unwrap(),
1184             ),
1185             Err(e) => self.fatal(&format!(
1186                 "Failed to setup Python process for \
1187                  LLDB script: {}",
1188                 e
1189             )),
1190         };
1191
1192         self.dump_output(&out, &err);
1193         ProcRes {
1194             status,
1195             stdout: out,
1196             stderr: err,
1197             cmdline: format!("{:?}", cmd),
1198         }
1199     }
1200
1201     fn parse_debugger_commands(&self, debugger_prefixes: &[&str]) -> DebuggerCommands {
1202         let directives = debugger_prefixes
1203             .iter()
1204             .map(|prefix| (format!("{}-command", prefix), format!("{}-check", prefix)))
1205             .collect::<Vec<_>>();
1206
1207         let mut breakpoint_lines = vec![];
1208         let mut commands = vec![];
1209         let mut check_lines = vec![];
1210         let mut counter = 1;
1211         let reader = BufReader::new(File::open(&self.testpaths.file).unwrap());
1212         for line in reader.lines() {
1213             match line {
1214                 Ok(line) => {
1215                     let line = if line.starts_with("//") {
1216                         line[2..].trim_start()
1217                     } else {
1218                         line.as_str()
1219                     };
1220
1221                     if line.contains("#break") {
1222                         breakpoint_lines.push(counter);
1223                     }
1224
1225                     for &(ref command_directive, ref check_directive) in &directives {
1226                         self.config
1227                             .parse_name_value_directive(&line, command_directive)
1228                             .map(|cmd| commands.push(cmd));
1229
1230                         self.config
1231                             .parse_name_value_directive(&line, check_directive)
1232                             .map(|cmd| check_lines.push(cmd));
1233                     }
1234                 }
1235                 Err(e) => self.fatal(&format!("Error while parsing debugger commands: {}", e)),
1236             }
1237             counter += 1;
1238         }
1239
1240         DebuggerCommands {
1241             commands,
1242             check_lines,
1243             breakpoint_lines,
1244         }
1245     }
1246
1247     fn cleanup_debug_info_options(&self, options: &Option<String>) -> Option<String> {
1248         if options.is_none() {
1249             return None;
1250         }
1251
1252         // Remove options that are either unwanted (-O) or may lead to duplicates due to RUSTFLAGS.
1253         let options_to_remove = ["-O".to_owned(), "-g".to_owned(), "--debuginfo".to_owned()];
1254         let new_options = self
1255             .split_maybe_args(options)
1256             .into_iter()
1257             .filter(|x| !options_to_remove.contains(x))
1258             .collect::<Vec<String>>();
1259
1260         Some(new_options.join(" "))
1261     }
1262
1263     fn maybe_add_external_args(&self, cmd: &mut Command, args: Vec<String>) {
1264         // Filter out the arguments that should not be added by runtest here.
1265         //
1266         // Notable use-cases are: do not add our optimisation flag if
1267         // `compile-flags: -Copt-level=x` and similar for debug-info level as well.
1268         const OPT_FLAGS: &[&str] = &["-O", "-Copt-level=", /*-C<space>*/"opt-level="];
1269         const DEBUG_FLAGS: &[&str] = &["-g", "-Cdebuginfo=", /*-C<space>*/"debuginfo="];
1270
1271         // FIXME: ideally we would "just" check the `cmd` itself, but it does not allow inspecting
1272         // its arguments. They need to be collected separately. For now I cannot be bothered to
1273         // implement this the "right" way.
1274         let have_opt_flag = self.props.compile_flags.iter().any(|arg| {
1275             OPT_FLAGS.iter().any(|f| arg.starts_with(f))
1276         });
1277         let have_debug_flag = self.props.compile_flags.iter().any(|arg| {
1278             DEBUG_FLAGS.iter().any(|f| arg.starts_with(f))
1279         });
1280
1281         for arg in args {
1282             if OPT_FLAGS.iter().any(|f| arg.starts_with(f)) && have_opt_flag {
1283                 continue;
1284             }
1285             if DEBUG_FLAGS.iter().any(|f| arg.starts_with(f)) && have_debug_flag {
1286                 continue;
1287             }
1288             cmd.arg(arg);
1289         }
1290     }
1291
1292     fn check_debugger_output(&self, debugger_run_result: &ProcRes, check_lines: &[String]) {
1293         let num_check_lines = check_lines.len();
1294
1295         let mut check_line_index = 0;
1296         for line in debugger_run_result.stdout.lines() {
1297             if check_line_index >= num_check_lines {
1298                 break;
1299             }
1300
1301             if check_single_line(line, &(check_lines[check_line_index])[..]) {
1302                 check_line_index += 1;
1303             }
1304         }
1305         if check_line_index != num_check_lines && num_check_lines > 0 {
1306             self.fatal_proc_rec(
1307                 &format!(
1308                     "line not found in debugger output: {}",
1309                     check_lines[check_line_index]
1310                 ),
1311                 debugger_run_result,
1312             );
1313         }
1314
1315         fn check_single_line(line: &str, check_line: &str) -> bool {
1316             // Allow check lines to leave parts unspecified (e.g., uninitialized
1317             // bits in the  wrong case of an enum) with the notation "[...]".
1318             let line = line.trim();
1319             let check_line = check_line.trim();
1320             let can_start_anywhere = check_line.starts_with("[...]");
1321             let can_end_anywhere = check_line.ends_with("[...]");
1322
1323             let check_fragments: Vec<&str> = check_line
1324                 .split("[...]")
1325                 .filter(|frag| !frag.is_empty())
1326                 .collect();
1327             if check_fragments.is_empty() {
1328                 return true;
1329             }
1330
1331             let (mut rest, first_fragment) = if can_start_anywhere {
1332                 match line.find(check_fragments[0]) {
1333                     Some(pos) => (&line[pos + check_fragments[0].len()..], 1),
1334                     None => return false,
1335                 }
1336             } else {
1337                 (line, 0)
1338             };
1339
1340             for current_fragment in &check_fragments[first_fragment..] {
1341                 match rest.find(current_fragment) {
1342                     Some(pos) => {
1343                         rest = &rest[pos + current_fragment.len()..];
1344                     }
1345                     None => return false,
1346                 }
1347             }
1348
1349             if !can_end_anywhere && !rest.is_empty() {
1350                 return false;
1351             }
1352
1353             true
1354         }
1355     }
1356
1357     fn check_error_patterns(&self, output_to_check: &str, proc_res: &ProcRes) {
1358         debug!("check_error_patterns");
1359         if self.props.error_patterns.is_empty() {
1360             if self.pass_mode().is_some() {
1361                 return;
1362             } else {
1363                 self.fatal(&format!(
1364                     "no error pattern specified in {:?}",
1365                     self.testpaths.file.display()
1366                 ));
1367             }
1368         }
1369
1370         let mut missing_patterns: Vec<String> = Vec::new();
1371
1372         for pattern in &self.props.error_patterns {
1373             if output_to_check.contains(pattern.trim()) {
1374                 debug!("found error pattern {}", pattern);
1375             } else {
1376                 missing_patterns.push(pattern.to_string());
1377             }
1378         }
1379
1380         if missing_patterns.is_empty() {
1381             return;
1382         }
1383
1384         if missing_patterns.len() == 1 {
1385             self.fatal_proc_rec(
1386                 &format!("error pattern '{}' not found!", missing_patterns[0]),
1387                 proc_res,
1388             );
1389         } else {
1390             for pattern in missing_patterns {
1391                 self.error(&format!("error pattern '{}' not found!", pattern));
1392             }
1393             self.fatal_proc_rec("multiple error patterns not found", proc_res);
1394         }
1395     }
1396
1397     fn check_no_compiler_crash(&self, proc_res: &ProcRes) {
1398         match proc_res.status.code() {
1399             Some(101) => self.fatal_proc_rec("compiler encountered internal error", proc_res),
1400             None => self.fatal_proc_rec("compiler terminated by signal", proc_res),
1401             _ => (),
1402         }
1403     }
1404
1405     fn check_forbid_output(&self, output_to_check: &str, proc_res: &ProcRes) {
1406         for pat in &self.props.forbid_output {
1407             if output_to_check.contains(pat) {
1408                 self.fatal_proc_rec("forbidden pattern found in compiler output", proc_res);
1409             }
1410         }
1411     }
1412
1413     fn check_expected_errors(&self, expected_errors: Vec<errors::Error>, proc_res: &ProcRes) {
1414         debug!("check_expected_errors: expected_errors={:?} proc_res.status={:?}",
1415                expected_errors, proc_res.status);
1416         if proc_res.status.success()
1417             && expected_errors
1418                 .iter()
1419                 .any(|x| x.kind == Some(ErrorKind::Error))
1420         {
1421             self.fatal_proc_rec("process did not return an error status", proc_res);
1422         }
1423
1424         // On Windows, keep all '\' path separators to match the paths reported in the JSON output
1425         // from the compiler
1426         let os_file_name = self.testpaths.file.display().to_string();
1427
1428         // on windows, translate all '\' path separators to '/'
1429         let file_name = format!("{}", self.testpaths.file.display()).replace(r"\", "/");
1430
1431         // If the testcase being checked contains at least one expected "help"
1432         // message, then we'll ensure that all "help" messages are expected.
1433         // Otherwise, all "help" messages reported by the compiler will be ignored.
1434         // This logic also applies to "note" messages.
1435         let expect_help = expected_errors
1436             .iter()
1437             .any(|ee| ee.kind == Some(ErrorKind::Help));
1438         let expect_note = expected_errors
1439             .iter()
1440             .any(|ee| ee.kind == Some(ErrorKind::Note));
1441
1442         // Parse the JSON output from the compiler and extract out the messages.
1443         let actual_errors = json::parse_output(&os_file_name, &proc_res.stderr, proc_res);
1444         let mut unexpected = Vec::new();
1445         let mut found = vec![false; expected_errors.len()];
1446         for actual_error in &actual_errors {
1447             let opt_index = expected_errors.iter().enumerate().position(
1448                 |(index, expected_error)| {
1449                     !found[index] && actual_error.line_num == expected_error.line_num
1450                         && (expected_error.kind.is_none()
1451                             || actual_error.kind == expected_error.kind)
1452                         && actual_error.msg.contains(&expected_error.msg)
1453                 },
1454             );
1455
1456             match opt_index {
1457                 Some(index) => {
1458                     // found a match, everybody is happy
1459                     assert!(!found[index]);
1460                     found[index] = true;
1461                 }
1462
1463                 None => {
1464                     if self.is_unexpected_compiler_message(actual_error, expect_help, expect_note) {
1465                         self.error(&format!(
1466                             "{}:{}: unexpected {}: '{}'",
1467                             file_name,
1468                             actual_error.line_num,
1469                             actual_error
1470                                 .kind
1471                                 .as_ref()
1472                                 .map_or(String::from("message"), |k| k.to_string()),
1473                             actual_error.msg
1474                         ));
1475                         unexpected.push(actual_error);
1476                     }
1477                 }
1478             }
1479         }
1480
1481         let mut not_found = Vec::new();
1482         // anything not yet found is a problem
1483         for (index, expected_error) in expected_errors.iter().enumerate() {
1484             if !found[index] {
1485                 self.error(&format!(
1486                     "{}:{}: expected {} not found: {}",
1487                     file_name,
1488                     expected_error.line_num,
1489                     expected_error
1490                         .kind
1491                         .as_ref()
1492                         .map_or("message".into(), |k| k.to_string()),
1493                     expected_error.msg
1494                 ));
1495                 not_found.push(expected_error);
1496             }
1497         }
1498
1499         if !unexpected.is_empty() || !not_found.is_empty() {
1500             self.error(&format!(
1501                 "{} unexpected errors found, {} expected errors not found",
1502                 unexpected.len(),
1503                 not_found.len()
1504             ));
1505             println!("status: {}\ncommand: {}", proc_res.status, proc_res.cmdline);
1506             if !unexpected.is_empty() {
1507                 println!("unexpected errors (from JSON output): {:#?}\n", unexpected);
1508             }
1509             if !not_found.is_empty() {
1510                 println!("not found errors (from test file): {:#?}\n", not_found);
1511             }
1512             panic!();
1513         }
1514     }
1515
1516     /// Returns `true` if we should report an error about `actual_error`,
1517     /// which did not match any of the expected error. We always require
1518     /// errors/warnings to be explicitly listed, but only require
1519     /// helps/notes if there are explicit helps/notes given.
1520     fn is_unexpected_compiler_message(
1521         &self,
1522         actual_error: &Error,
1523         expect_help: bool,
1524         expect_note: bool,
1525     ) -> bool {
1526         match actual_error.kind {
1527             Some(ErrorKind::Help) => expect_help,
1528             Some(ErrorKind::Note) => expect_note,
1529             Some(ErrorKind::Error) | Some(ErrorKind::Warning) => true,
1530             Some(ErrorKind::Suggestion) | None => false,
1531         }
1532     }
1533
1534     fn compile_test(&self) -> ProcRes {
1535         // Only use `make_exe_name` when the test ends up being executed.
1536         let will_execute = match self.config.mode {
1537             Ui => self.should_run_successfully(),
1538             Incremental => self.revision.unwrap().starts_with("r"),
1539             RunFail | RunPassValgrind | MirOpt |
1540             DebugInfoCdb | DebugInfoGdbLldb | DebugInfoGdb | DebugInfoLldb => true,
1541             _ => false,
1542         };
1543         let output_file = if will_execute {
1544             TargetLocation::ThisFile(self.make_exe_name())
1545         } else {
1546             TargetLocation::ThisDirectory(self.output_base_dir())
1547         };
1548
1549         let mut rustc = self.make_compile_args(&self.testpaths.file, output_file);
1550
1551         rustc.arg("-L").arg(&self.aux_output_dir_name());
1552
1553         match self.config.mode {
1554             CompileFail | Ui => {
1555                 // compile-fail and ui tests tend to have tons of unused code as
1556                 // it's just testing various pieces of the compile, but we don't
1557                 // want to actually assert warnings about all this code. Instead
1558                 // let's just ignore unused code warnings by defaults and tests
1559                 // can turn it back on if needed.
1560                 if !self.config.src_base.ends_with("rustdoc-ui") {
1561                     rustc.args(&["-A", "unused"]);
1562                 }
1563             }
1564             _ => {}
1565         }
1566
1567         self.compose_and_run_compiler(rustc, None)
1568     }
1569
1570     fn document(&self, out_dir: &Path) -> ProcRes {
1571         if self.props.build_aux_docs {
1572             for rel_ab in &self.props.aux_builds {
1573                 let aux_testpaths = self.compute_aux_test_paths(rel_ab);
1574                 let aux_props =
1575                     self.props
1576                         .from_aux_file(&aux_testpaths.file, self.revision, self.config);
1577                 let aux_cx = TestCx {
1578                     config: self.config,
1579                     props: &aux_props,
1580                     testpaths: &aux_testpaths,
1581                     revision: self.revision,
1582                 };
1583                 // Create the directory for the stdout/stderr files.
1584                 create_dir_all(aux_cx.output_base_dir()).unwrap();
1585                 let auxres = aux_cx.document(out_dir);
1586                 if !auxres.status.success() {
1587                     return auxres;
1588                 }
1589             }
1590         }
1591
1592         let aux_dir = self.aux_output_dir_name();
1593
1594         let rustdoc_path = self
1595             .config
1596             .rustdoc_path
1597             .as_ref()
1598             .expect("--rustdoc-path passed");
1599         let mut rustdoc = Command::new(rustdoc_path);
1600
1601         rustdoc
1602             .arg("-L")
1603             .arg(self.config.run_lib_path.to_str().unwrap())
1604             .arg("-L")
1605             .arg(aux_dir)
1606             .arg("-o")
1607             .arg(out_dir)
1608             .arg(&self.testpaths.file)
1609             .args(&self.props.compile_flags);
1610
1611         if let Some(ref linker) = self.config.linker {
1612             rustdoc.arg(format!("-Clinker={}", linker));
1613         }
1614
1615         self.compose_and_run_compiler(rustdoc, None)
1616     }
1617
1618     fn exec_compiled_test(&self) -> ProcRes {
1619         let env = &self.props.exec_env;
1620
1621         let proc_res = match &*self.config.target {
1622             // This is pretty similar to below, we're transforming:
1623             //
1624             //      program arg1 arg2
1625             //
1626             // into
1627             //
1628             //      remote-test-client run program:support-lib.so arg1 arg2
1629             //
1630             // The test-client program will upload `program` to the emulator
1631             // along with all other support libraries listed (in this case
1632             // `support-lib.so`. It will then execute the program on the
1633             // emulator with the arguments specified (in the environment we give
1634             // the process) and then report back the same result.
1635             _ if self.config.remote_test_client.is_some() => {
1636                 let aux_dir = self.aux_output_dir_name();
1637                 let ProcArgs { mut prog, args } = self.make_run_args();
1638                 if let Ok(entries) = aux_dir.read_dir() {
1639                     for entry in entries {
1640                         let entry = entry.unwrap();
1641                         if !entry.path().is_file() {
1642                             continue;
1643                         }
1644                         prog.push_str(":");
1645                         prog.push_str(entry.path().to_str().unwrap());
1646                     }
1647                 }
1648                 let mut test_client =
1649                     Command::new(self.config.remote_test_client.as_ref().unwrap());
1650                 test_client
1651                     .args(&["run", &prog])
1652                     .args(args)
1653                     .envs(env.clone());
1654                 self.compose_and_run(
1655                     test_client,
1656                     self.config.run_lib_path.to_str().unwrap(),
1657                     Some(aux_dir.to_str().unwrap()),
1658                     None,
1659                 )
1660             }
1661             _ if self.config.target.contains("vxworks") => {
1662                 let aux_dir = self.aux_output_dir_name();
1663                 let ProcArgs { prog, args } = self.make_run_args();
1664                 let mut wr_run = Command::new("wr-run");
1665                 wr_run.args(&[&prog]).args(args).envs(env.clone());
1666                 self.compose_and_run(
1667                     wr_run,
1668                     self.config.run_lib_path.to_str().unwrap(),
1669                     Some(aux_dir.to_str().unwrap()),
1670                     None,
1671                 )
1672             }
1673             _ => {
1674                 let aux_dir = self.aux_output_dir_name();
1675                 let ProcArgs { prog, args } = self.make_run_args();
1676                 let mut program = Command::new(&prog);
1677                 program
1678                     .args(args)
1679                     .current_dir(&self.output_base_dir())
1680                     .envs(env.clone());
1681                 self.compose_and_run(
1682                     program,
1683                     self.config.run_lib_path.to_str().unwrap(),
1684                     Some(aux_dir.to_str().unwrap()),
1685                     None,
1686                 )
1687             }
1688         };
1689
1690         if proc_res.status.success() {
1691             // delete the executable after running it to save space.
1692             // it is ok if the deletion failed.
1693             let _ = fs::remove_file(self.make_exe_name());
1694         }
1695
1696         proc_res
1697     }
1698
1699     /// For each `aux-build: foo/bar` annotation, we check to find the
1700     /// file in a `auxiliary` directory relative to the test itself.
1701     fn compute_aux_test_paths(&self, rel_ab: &str) -> TestPaths {
1702         let test_ab = self
1703             .testpaths
1704             .file
1705             .parent()
1706             .expect("test file path has no parent")
1707             .join("auxiliary")
1708             .join(rel_ab);
1709         if !test_ab.exists() {
1710             self.fatal(&format!(
1711                 "aux-build `{}` source not found",
1712                 test_ab.display()
1713             ))
1714         }
1715
1716         TestPaths {
1717             file: test_ab,
1718             relative_dir: self
1719                 .testpaths
1720                 .relative_dir
1721                 .join(self.output_testname_unique())
1722                 .join("auxiliary")
1723                 .join(rel_ab)
1724                 .parent()
1725                 .expect("aux-build path has no parent")
1726                 .to_path_buf(),
1727         }
1728     }
1729
1730     fn is_vxworks_pure_static(&self) -> bool {
1731         if self.config.target.contains("vxworks") {
1732             match env::var("RUST_VXWORKS_TEST_DYLINK") {
1733                 Ok(s) => s != "1",
1734                 _ => true
1735             }
1736         } else {
1737             false
1738         }
1739     }
1740
1741     fn is_vxworks_pure_dynamic(&self) -> bool {
1742         self.config.target.contains("vxworks") && !self.is_vxworks_pure_static()
1743     }
1744
1745     fn compose_and_run_compiler(&self, mut rustc: Command, input: Option<String>) -> ProcRes {
1746         let aux_dir = self.aux_output_dir_name();
1747
1748         if !self.props.aux_builds.is_empty() {
1749             let _ = fs::remove_dir_all(&aux_dir);
1750             create_dir_all(&aux_dir).unwrap();
1751         }
1752
1753         // Use a Vec instead of a HashMap to preserve original order
1754         let mut extern_priv = self.props.extern_private.clone();
1755
1756         let mut add_extern_priv = |priv_dep: &str, dylib: bool| {
1757             let lib_name = get_lib_name(priv_dep, dylib);
1758             rustc
1759                 .arg("--extern-private")
1760                 .arg(format!("{}={}", priv_dep, aux_dir.join(lib_name).to_str().unwrap()));
1761         };
1762
1763         for rel_ab in &self.props.aux_builds {
1764             let aux_testpaths = self.compute_aux_test_paths(rel_ab);
1765             let aux_props =
1766                 self.props
1767                     .from_aux_file(&aux_testpaths.file, self.revision, self.config);
1768             let aux_output = TargetLocation::ThisDirectory(self.aux_output_dir_name());
1769             let aux_cx = TestCx {
1770                 config: self.config,
1771                 props: &aux_props,
1772                 testpaths: &aux_testpaths,
1773                 revision: self.revision,
1774             };
1775             // Create the directory for the stdout/stderr files.
1776             create_dir_all(aux_cx.output_base_dir()).unwrap();
1777             let mut aux_rustc = aux_cx.make_compile_args(&aux_testpaths.file, aux_output);
1778
1779             let (dylib, crate_type) = if aux_props.no_prefer_dynamic {
1780                 (true, None)
1781             } else if self.config.target.contains("cloudabi")
1782                 || self.config.target.contains("emscripten")
1783                 || (self.config.target.contains("musl")
1784                     && !aux_props.force_host
1785                     && !self.config.host.contains("musl"))
1786                 || self.config.target.contains("wasm32")
1787                 || self.config.target.contains("nvptx")
1788                 || self.is_vxworks_pure_static()
1789             {
1790                 // We primarily compile all auxiliary libraries as dynamic libraries
1791                 // to avoid code size bloat and large binaries as much as possible
1792                 // for the test suite (otherwise including libstd statically in all
1793                 // executables takes up quite a bit of space).
1794                 //
1795                 // For targets like MUSL or Emscripten, however, there is no support for
1796                 // dynamic libraries so we just go back to building a normal library. Note,
1797                 // however, that for MUSL if the library is built with `force_host` then
1798                 // it's ok to be a dylib as the host should always support dylibs.
1799                 (false, Some("lib"))
1800             } else {
1801                 (true, Some("dylib"))
1802             };
1803
1804             let trimmed = rel_ab.trim_end_matches(".rs").to_string();
1805
1806             // Normally, every 'extern-private' has a correspodning 'aux-build'
1807             // entry. If so, we remove it from our list of private crates,
1808             // and add an '--extern-private' flag to rustc
1809             if extern_priv.remove_item(&trimmed).is_some() {
1810                 add_extern_priv(&trimmed, dylib);
1811             }
1812
1813             if let Some(crate_type) = crate_type {
1814                 aux_rustc.args(&["--crate-type", crate_type]);
1815             }
1816
1817             aux_rustc.arg("-L").arg(&aux_dir);
1818
1819             let auxres = aux_cx.compose_and_run(
1820                 aux_rustc,
1821                 aux_cx.config.compile_lib_path.to_str().unwrap(),
1822                 Some(aux_dir.to_str().unwrap()),
1823                 None,
1824             );
1825             if !auxres.status.success() {
1826                 self.fatal_proc_rec(
1827                     &format!(
1828                         "auxiliary build of {:?} failed to compile: ",
1829                         aux_testpaths.file.display()
1830                     ),
1831                     &auxres,
1832                 );
1833             }
1834         }
1835
1836         // Add any '--extern-private' entries without a matching
1837         // 'aux-build'
1838         for private_lib in extern_priv {
1839             add_extern_priv(&private_lib, true);
1840         }
1841
1842         self.props.unset_rustc_env.clone()
1843             .iter()
1844             .fold(&mut rustc, |rustc, v| rustc.env_remove(v));
1845         rustc.envs(self.props.rustc_env.clone());
1846         self.compose_and_run(
1847             rustc,
1848             self.config.compile_lib_path.to_str().unwrap(),
1849             Some(aux_dir.to_str().unwrap()),
1850             input,
1851         )
1852     }
1853
1854     fn compose_and_run(
1855         &self,
1856         mut command: Command,
1857         lib_path: &str,
1858         aux_path: Option<&str>,
1859         input: Option<String>,
1860     ) -> ProcRes {
1861         let cmdline = {
1862             let cmdline = self.make_cmdline(&command, lib_path);
1863             logv(self.config, format!("executing {}", cmdline));
1864             cmdline
1865         };
1866
1867         command
1868             .stdout(Stdio::piped())
1869             .stderr(Stdio::piped())
1870             .stdin(Stdio::piped());
1871
1872         // Need to be sure to put both the lib_path and the aux path in the dylib
1873         // search path for the child.
1874         let mut path = env::split_paths(&env::var_os(dylib_env_var()).unwrap_or(OsString::new()))
1875             .collect::<Vec<_>>();
1876         if let Some(p) = aux_path {
1877             path.insert(0, PathBuf::from(p))
1878         }
1879         path.insert(0, PathBuf::from(lib_path));
1880
1881         // Add the new dylib search path var
1882         let newpath = env::join_paths(&path).unwrap();
1883         command.env(dylib_env_var(), newpath);
1884
1885         let mut child = disable_error_reporting(|| command.spawn())
1886             .expect(&format!("failed to exec `{:?}`", &command));
1887         if let Some(input) = input {
1888             child
1889                 .stdin
1890                 .as_mut()
1891                 .unwrap()
1892                 .write_all(input.as_bytes())
1893                 .unwrap();
1894         }
1895
1896         let Output {
1897             status,
1898             stdout,
1899             stderr,
1900         } = read2_abbreviated(child).expect("failed to read output");
1901
1902         let result = ProcRes {
1903             status,
1904             stdout: String::from_utf8_lossy(&stdout).into_owned(),
1905             stderr: String::from_utf8_lossy(&stderr).into_owned(),
1906             cmdline,
1907         };
1908
1909         self.dump_output(&result.stdout, &result.stderr);
1910
1911         result
1912     }
1913
1914     fn make_compile_args(
1915         &self,
1916         input_file: &Path,
1917         output_file: TargetLocation,
1918     ) -> Command {
1919         let is_rustdoc = self.config.src_base.ends_with("rustdoc-ui") ||
1920                          self.config.src_base.ends_with("rustdoc-js");
1921         let mut rustc = if !is_rustdoc {
1922             Command::new(&self.config.rustc_path)
1923         } else {
1924             Command::new(
1925                 &self
1926                     .config
1927                     .rustdoc_path
1928                     .clone()
1929                     .expect("no rustdoc built yet"),
1930             )
1931         };
1932         // FIXME Why is -L here?
1933         rustc.arg(input_file); //.arg("-L").arg(&self.config.build_base);
1934
1935         // Use a single thread for efficiency and a deterministic error message order
1936         rustc.arg("-Zthreads=1");
1937
1938         // Optionally prevent default --target if specified in test compile-flags.
1939         let custom_target = self
1940             .props
1941             .compile_flags
1942             .iter()
1943             .any(|x| x.starts_with("--target"));
1944
1945         if !custom_target {
1946             let target = if self.props.force_host {
1947                 &*self.config.host
1948             } else {
1949                 &*self.config.target
1950             };
1951
1952             rustc.arg(&format!("--target={}", target));
1953         }
1954         self.set_revision_flags(&mut rustc);
1955
1956         if !is_rustdoc {
1957             if let Some(ref incremental_dir) = self.props.incremental_dir {
1958                 rustc.args(&["-C", &format!("incremental={}", incremental_dir.display())]);
1959                 rustc.args(&["-Z", "incremental-verify-ich"]);
1960                 rustc.args(&["-Z", "incremental-queries"]);
1961             }
1962
1963             if self.config.mode == CodegenUnits {
1964                 rustc.args(&["-Z", "human_readable_cgu_names"]);
1965             }
1966         }
1967
1968         match self.config.mode {
1969             CompileFail | Incremental => {
1970                 // If we are extracting and matching errors in the new
1971                 // fashion, then you want JSON mode. Old-skool error
1972                 // patterns still match the raw compiler output.
1973                 if self.props.error_patterns.is_empty() {
1974                     rustc.args(&["--error-format", "json"]);
1975                 }
1976                 if !self.props.disable_ui_testing_normalization {
1977                     rustc.arg("-Zui-testing");
1978                 }
1979             }
1980             Ui => {
1981                 if !self
1982                     .props
1983                     .compile_flags
1984                     .iter()
1985                     .any(|s| s.starts_with("--error-format"))
1986                 {
1987                     rustc.args(&["--error-format", "json"]);
1988                 }
1989                 if !self.props.disable_ui_testing_normalization {
1990                     rustc.arg("-Zui-testing");
1991                 }
1992             }
1993             MirOpt => {
1994                 rustc.args(&[
1995                     "-Zdump-mir=all",
1996                     "-Zmir-opt-level=3",
1997                     "-Zdump-mir-exclude-pass-number",
1998                 ]);
1999
2000                 let mir_dump_dir = self.get_mir_dump_dir();
2001                 let _ = fs::remove_dir_all(&mir_dump_dir);
2002                 create_dir_all(mir_dump_dir.as_path()).unwrap();
2003                 let mut dir_opt = "-Zdump-mir-dir=".to_string();
2004                 dir_opt.push_str(mir_dump_dir.to_str().unwrap());
2005                 debug!("dir_opt: {:?}", dir_opt);
2006
2007                 rustc.arg(dir_opt);
2008             }
2009             RunFail | RunPassValgrind | Pretty | DebugInfoCdb | DebugInfoGdbLldb | DebugInfoGdb
2010             | DebugInfoLldb | Codegen | Rustdoc | RunMake | CodegenUnits | JsDocTest | Assembly => {
2011                 // do not use JSON output
2012             }
2013         }
2014
2015         if let Some(PassMode::Check) = self.pass_mode() {
2016             rustc.args(&["--emit", "metadata"]);
2017         }
2018
2019         if !is_rustdoc {
2020             if self.config.target == "wasm32-unknown-unknown"
2021                 || self.is_vxworks_pure_static() {
2022                 // rustc.arg("-g"); // get any backtrace at all on errors
2023             } else if !self.props.no_prefer_dynamic {
2024                 rustc.args(&["-C", "prefer-dynamic"]);
2025             }
2026         }
2027
2028         match output_file {
2029             TargetLocation::ThisFile(path) => {
2030                 rustc.arg("-o").arg(path);
2031             }
2032             TargetLocation::ThisDirectory(path) => {
2033                 if is_rustdoc {
2034                     // `rustdoc` uses `-o` for the output directory.
2035                     rustc.arg("-o").arg(path);
2036                 } else {
2037                     rustc.arg("--out-dir").arg(path);
2038                 }
2039             }
2040         }
2041
2042         match self.config.compare_mode {
2043             Some(CompareMode::Nll) => {
2044                 rustc.args(&["-Zborrowck=mir"]);
2045             }
2046             Some(CompareMode::Polonius) => {
2047                 rustc.args(&["-Zpolonius", "-Zborrowck=mir"]);
2048             }
2049             None => {}
2050         }
2051
2052         if self.props.force_host {
2053             self.maybe_add_external_args(&mut rustc,
2054                                          self.split_maybe_args(&self.config.host_rustcflags));
2055         } else {
2056             self.maybe_add_external_args(&mut rustc,
2057                                          self.split_maybe_args(&self.config.target_rustcflags));
2058             if !is_rustdoc {
2059                 if let Some(ref linker) = self.config.linker {
2060                     rustc.arg(format!("-Clinker={}", linker));
2061                 }
2062             }
2063         }
2064
2065         // Use dynamic musl for tests because static doesn't allow creating dylibs
2066         if self.config.host.contains("musl")
2067             || self.is_vxworks_pure_dynamic() {
2068             rustc.arg("-Ctarget-feature=-crt-static");
2069         }
2070
2071         rustc.args(&self.props.compile_flags);
2072
2073         rustc
2074     }
2075
2076     fn make_exe_name(&self) -> PathBuf {
2077         // Using a single letter here to keep the path length down for
2078         // Windows.  Some test names get very long.  rustc creates `rcgu`
2079         // files with the module name appended to it which can more than
2080         // double the length.
2081         let mut f = self.output_base_dir().join("a");
2082         // FIXME: This is using the host architecture exe suffix, not target!
2083         if self.config.target.contains("emscripten") {
2084             f = f.with_extra_extension("js");
2085         } else if self.config.target.contains("wasm32") {
2086             f = f.with_extra_extension("wasm");
2087         } else if !env::consts::EXE_SUFFIX.is_empty() {
2088             f = f.with_extra_extension(env::consts::EXE_SUFFIX);
2089         }
2090         f
2091     }
2092
2093     fn make_run_args(&self) -> ProcArgs {
2094         // If we've got another tool to run under (valgrind),
2095         // then split apart its command
2096         let mut args = self.split_maybe_args(&self.config.runtool);
2097
2098         // If this is emscripten, then run tests under nodejs
2099         if self.config.target.contains("emscripten") {
2100             if let Some(ref p) = self.config.nodejs {
2101                 args.push(p.clone());
2102             } else {
2103                 self.fatal("no NodeJS binary found (--nodejs)");
2104             }
2105         // If this is otherwise wasm, then run tests under nodejs with our
2106         // shim
2107         } else if self.config.target.contains("wasm32") {
2108             if let Some(ref p) = self.config.nodejs {
2109                 args.push(p.clone());
2110             } else {
2111                 self.fatal("no NodeJS binary found (--nodejs)");
2112             }
2113
2114             let src = self.config.src_base
2115                 .parent().unwrap() // chop off `ui`
2116                 .parent().unwrap() // chop off `test`
2117                 .parent().unwrap(); // chop off `src`
2118             args.push(src.join("src/etc/wasm32-shim.js").display().to_string());
2119         }
2120
2121         let exe_file = self.make_exe_name();
2122
2123         // FIXME (#9639): This needs to handle non-utf8 paths
2124         args.push(exe_file.to_str().unwrap().to_owned());
2125
2126         // Add the arguments in the run_flags directive
2127         args.extend(self.split_maybe_args(&self.props.run_flags));
2128
2129         let prog = args.remove(0);
2130         ProcArgs { prog, args }
2131     }
2132
2133     fn split_maybe_args(&self, argstr: &Option<String>) -> Vec<String> {
2134         match *argstr {
2135             Some(ref s) => s
2136                 .split(' ')
2137                 .filter_map(|s| {
2138                     if s.chars().all(|c| c.is_whitespace()) {
2139                         None
2140                     } else {
2141                         Some(s.to_owned())
2142                     }
2143                 })
2144                 .collect(),
2145             None => Vec::new(),
2146         }
2147     }
2148
2149     fn make_cmdline(&self, command: &Command, libpath: &str) -> String {
2150         use crate::util;
2151
2152         // Linux and mac don't require adjusting the library search path
2153         if cfg!(unix) {
2154             format!("{:?}", command)
2155         } else {
2156             // Build the LD_LIBRARY_PATH variable as it would be seen on the command line
2157             // for diagnostic purposes
2158             fn lib_path_cmd_prefix(path: &str) -> String {
2159                 format!(
2160                     "{}=\"{}\"",
2161                     util::lib_path_env_var(),
2162                     util::make_new_path(path)
2163                 )
2164             }
2165
2166             format!("{} {:?}", lib_path_cmd_prefix(libpath), command)
2167         }
2168     }
2169
2170     fn dump_output(&self, out: &str, err: &str) {
2171         let revision = if let Some(r) = self.revision {
2172             format!("{}.", r)
2173         } else {
2174             String::new()
2175         };
2176
2177         self.dump_output_file(out, &format!("{}out", revision));
2178         self.dump_output_file(err, &format!("{}err", revision));
2179         self.maybe_dump_to_stdout(out, err);
2180     }
2181
2182     fn dump_output_file(&self, out: &str, extension: &str) {
2183         let outfile = self.make_out_name(extension);
2184         fs::write(&outfile, out).unwrap();
2185     }
2186
2187     /// Creates a filename for output with the given extension.
2188     /// E.g., `/.../testname.revision.mode/testname.extension`.
2189     fn make_out_name(&self, extension: &str) -> PathBuf {
2190         self.output_base_name().with_extension(extension)
2191     }
2192
2193     /// Gets the directory where auxiliary files are written.
2194     /// E.g., `/.../testname.revision.mode/auxiliary/`.
2195     fn aux_output_dir_name(&self) -> PathBuf {
2196         self.output_base_dir()
2197             .join("auxiliary")
2198             .with_extra_extension(self.config.mode.disambiguator())
2199     }
2200
2201     /// Generates a unique name for the test, such as `testname.revision.mode`.
2202     fn output_testname_unique(&self) -> PathBuf {
2203         output_testname_unique(self.config, self.testpaths, self.safe_revision())
2204     }
2205
2206     /// The revision, ignored for incremental compilation since it wants all revisions in
2207     /// the same directory.
2208     fn safe_revision(&self) -> Option<&str> {
2209         if self.config.mode == Incremental {
2210             None
2211         } else {
2212             self.revision
2213         }
2214     }
2215
2216     /// Gets the absolute path to the directory where all output for the given
2217     /// test/revision should reside.
2218     /// E.g., `/path/to/build/host-triple/test/ui/relative/testname.revision.mode/`.
2219     fn output_base_dir(&self) -> PathBuf {
2220         output_base_dir(self.config, self.testpaths, self.safe_revision())
2221     }
2222
2223     /// Gets the absolute path to the base filename used as output for the given
2224     /// test/revision.
2225     /// E.g., `/.../relative/testname.revision.mode/testname`.
2226     fn output_base_name(&self) -> PathBuf {
2227         output_base_name(self.config, self.testpaths, self.safe_revision())
2228     }
2229
2230     fn maybe_dump_to_stdout(&self, out: &str, err: &str) {
2231         if self.config.verbose {
2232             println!("------{}------------------------------", "stdout");
2233             println!("{}", out);
2234             println!("------{}------------------------------", "stderr");
2235             println!("{}", err);
2236             println!("------------------------------------------");
2237         }
2238     }
2239
2240     fn error(&self, err: &str) {
2241         match self.revision {
2242             Some(rev) => println!("\nerror in revision `{}`: {}", rev, err),
2243             None => println!("\nerror: {}", err),
2244         }
2245     }
2246
2247     fn fatal(&self, err: &str) -> ! {
2248         self.error(err);
2249         error!("fatal error, panic: {:?}", err);
2250         panic!("fatal error");
2251     }
2252
2253     fn fatal_proc_rec(&self, err: &str, proc_res: &ProcRes) -> ! {
2254         self.error(err);
2255         proc_res.fatal(None);
2256     }
2257
2258     // codegen tests (using FileCheck)
2259
2260     fn compile_test_and_save_ir(&self) -> ProcRes {
2261         let aux_dir = self.aux_output_dir_name();
2262
2263         let output_file = TargetLocation::ThisDirectory(self.output_base_dir());
2264         let mut rustc = self.make_compile_args(&self.testpaths.file, output_file);
2265         rustc.arg("-L").arg(aux_dir).arg("--emit=llvm-ir");
2266
2267         self.compose_and_run_compiler(rustc, None)
2268     }
2269
2270     fn compile_test_and_save_assembly(&self) -> (ProcRes, PathBuf) {
2271         // This works with both `--emit asm` (as default output name for the assembly)
2272         // and `ptx-linker` because the latter can write output at requested location.
2273         let output_path = self.output_base_name().with_extension("s");
2274
2275         let output_file = TargetLocation::ThisFile(output_path.clone());
2276         let mut rustc = self.make_compile_args(&self.testpaths.file, output_file);
2277
2278         rustc.arg("-L").arg(self.aux_output_dir_name());
2279
2280         match self.props.assembly_output.as_ref().map(AsRef::as_ref) {
2281             Some("emit-asm") => {
2282                 rustc.arg("--emit=asm");
2283             }
2284
2285             Some("ptx-linker") => {
2286                 // No extra flags needed.
2287             }
2288
2289             Some(_) => self.fatal("unknown 'assembly-output' header"),
2290             None => self.fatal("missing 'assembly-output' header"),
2291         }
2292
2293         (self.compose_and_run_compiler(rustc, None), output_path)
2294     }
2295
2296     fn verify_with_filecheck(&self, output: &Path) -> ProcRes {
2297         let mut filecheck = Command::new(self.config.llvm_filecheck.as_ref().unwrap());
2298         filecheck
2299             .arg("--input-file")
2300             .arg(output)
2301             .arg(&self.testpaths.file);
2302         // It would be more appropriate to make most of the arguments configurable through
2303         // a comment-attribute similar to `compile-flags`. For example, --check-prefixes is a very
2304         // useful flag.
2305         //
2306         // For now, though…
2307         if let Some(rev) = self.revision {
2308             let prefixes = format!("CHECK,{}", rev);
2309             filecheck.args(&["--check-prefixes", &prefixes]);
2310         }
2311         self.compose_and_run(filecheck, "", None, None)
2312     }
2313
2314     fn run_codegen_test(&self) {
2315         if self.config.llvm_filecheck.is_none() {
2316             self.fatal("missing --llvm-filecheck");
2317         }
2318
2319         let proc_res = self.compile_test_and_save_ir();
2320         if !proc_res.status.success() {
2321             self.fatal_proc_rec("compilation failed!", &proc_res);
2322         }
2323
2324         let output_path = self.output_base_name().with_extension("ll");
2325         let proc_res = self.verify_with_filecheck(&output_path);
2326         if !proc_res.status.success() {
2327             self.fatal_proc_rec("verification with 'FileCheck' failed", &proc_res);
2328         }
2329     }
2330
2331     fn run_assembly_test(&self) {
2332         if self.config.llvm_filecheck.is_none() {
2333             self.fatal("missing --llvm-filecheck");
2334         }
2335
2336         let (proc_res, output_path) = self.compile_test_and_save_assembly();
2337         if !proc_res.status.success() {
2338             self.fatal_proc_rec("compilation failed!", &proc_res);
2339         }
2340
2341         let proc_res = self.verify_with_filecheck(&output_path);
2342         if !proc_res.status.success() {
2343             self.fatal_proc_rec("verification with 'FileCheck' failed", &proc_res);
2344         }
2345     }
2346
2347     fn charset() -> &'static str {
2348         // FreeBSD 10.1 defaults to GDB 6.1.1 which doesn't support "auto" charset
2349         if cfg!(target_os = "freebsd") {
2350             "ISO-8859-1"
2351         } else {
2352             "UTF-8"
2353         }
2354     }
2355
2356     fn run_rustdoc_test(&self) {
2357         assert!(self.revision.is_none(), "revisions not relevant here");
2358
2359         let out_dir = self.output_base_dir();
2360         let _ = fs::remove_dir_all(&out_dir);
2361         create_dir_all(&out_dir).unwrap();
2362
2363         let proc_res = self.document(&out_dir);
2364         if !proc_res.status.success() {
2365             self.fatal_proc_rec("rustdoc failed!", &proc_res);
2366         }
2367
2368         if self.props.check_test_line_numbers_match {
2369             self.check_rustdoc_test_option(proc_res);
2370         } else {
2371             let root = self.config.find_rust_src_root().unwrap();
2372             let res = self.cmd2procres(
2373                 Command::new(&self.config.docck_python)
2374                     .arg(root.join("src/etc/htmldocck.py"))
2375                     .arg(out_dir)
2376                     .arg(&self.testpaths.file),
2377             );
2378             if !res.status.success() {
2379                 self.fatal_proc_rec("htmldocck failed!", &res);
2380             }
2381         }
2382     }
2383
2384     fn get_lines<P: AsRef<Path>>(
2385         &self,
2386         path: &P,
2387         mut other_files: Option<&mut Vec<String>>,
2388     ) -> Vec<usize> {
2389         let content = fs::read_to_string(&path).unwrap();
2390         let mut ignore = false;
2391         content
2392             .lines()
2393             .enumerate()
2394             .filter_map(|(line_nb, line)| {
2395                 if (line.trim_start().starts_with("pub mod ")
2396                     || line.trim_start().starts_with("mod "))
2397                     && line.ends_with(';')
2398                 {
2399                     if let Some(ref mut other_files) = other_files {
2400                         other_files.push(line.rsplit("mod ").next().unwrap().replace(";", ""));
2401                     }
2402                     None
2403                 } else {
2404                     let sline = line.split("///").last().unwrap_or("");
2405                     let line = sline.trim_start();
2406                     if line.starts_with("```") {
2407                         if ignore {
2408                             ignore = false;
2409                             None
2410                         } else {
2411                             ignore = true;
2412                             Some(line_nb + 1)
2413                         }
2414                     } else {
2415                         None
2416                     }
2417                 }
2418             })
2419             .collect()
2420     }
2421
2422     fn check_rustdoc_test_option(&self, res: ProcRes) {
2423         let mut other_files = Vec::new();
2424         let mut files: HashMap<String, Vec<usize>> = HashMap::new();
2425         let cwd = env::current_dir().unwrap();
2426         files.insert(
2427             self.testpaths
2428                 .file
2429                 .strip_prefix(&cwd)
2430                 .unwrap_or(&self.testpaths.file)
2431                 .to_str()
2432                 .unwrap()
2433                 .replace('\\', "/"),
2434             self.get_lines(&self.testpaths.file, Some(&mut other_files)),
2435         );
2436         for other_file in other_files {
2437             let mut path = self.testpaths.file.clone();
2438             path.set_file_name(&format!("{}.rs", other_file));
2439             files.insert(
2440                 path.strip_prefix(&cwd)
2441                     .unwrap_or(&path)
2442                     .to_str()
2443                     .unwrap()
2444                     .replace('\\', "/"),
2445                 self.get_lines(&path, None),
2446             );
2447         }
2448
2449         let mut tested = 0;
2450         for _ in res
2451             .stdout
2452             .split('\n')
2453             .filter(|s| s.starts_with("test "))
2454             .inspect(|s| {
2455                 let tmp: Vec<&str> = s.split(" - ").collect();
2456                 if tmp.len() == 2 {
2457                     let path = tmp[0].rsplit("test ").next().unwrap();
2458                     if let Some(ref mut v) = files.get_mut(&path.replace('\\', "/")) {
2459                         tested += 1;
2460                         let mut iter = tmp[1].split("(line ");
2461                         iter.next();
2462                         let line = iter
2463                             .next()
2464                             .unwrap_or(")")
2465                             .split(')')
2466                             .next()
2467                             .unwrap_or("0")
2468                             .parse()
2469                             .unwrap_or(0);
2470                         if let Ok(pos) = v.binary_search(&line) {
2471                             v.remove(pos);
2472                         } else {
2473                             self.fatal_proc_rec(
2474                                 &format!("Not found doc test: \"{}\" in \"{}\":{:?}", s, path, v),
2475                                 &res,
2476                             );
2477                         }
2478                     }
2479                 }
2480             }) {}
2481         if tested == 0 {
2482             self.fatal_proc_rec(&format!("No test has been found... {:?}", files), &res);
2483         } else {
2484             for (entry, v) in &files {
2485                 if !v.is_empty() {
2486                     self.fatal_proc_rec(
2487                         &format!(
2488                             "Not found test at line{} \"{}\":{:?}",
2489                             if v.len() > 1 { "s" } else { "" },
2490                             entry,
2491                             v
2492                         ),
2493                         &res,
2494                     );
2495                 }
2496             }
2497         }
2498     }
2499
2500     fn run_codegen_units_test(&self) {
2501         assert!(self.revision.is_none(), "revisions not relevant here");
2502
2503         let proc_res = self.compile_test();
2504
2505         if !proc_res.status.success() {
2506             self.fatal_proc_rec("compilation failed!", &proc_res);
2507         }
2508
2509         self.check_no_compiler_crash(&proc_res);
2510
2511         const PREFIX: &'static str = "MONO_ITEM ";
2512         const CGU_MARKER: &'static str = "@@";
2513
2514         let actual: Vec<MonoItem> = proc_res
2515             .stdout
2516             .lines()
2517             .filter(|line| line.starts_with(PREFIX))
2518             .map(|line| str_to_mono_item(line, true))
2519             .collect();
2520
2521         let expected: Vec<MonoItem> = errors::load_errors(&self.testpaths.file, None)
2522             .iter()
2523             .map(|e| str_to_mono_item(&e.msg[..], false))
2524             .collect();
2525
2526         let mut missing = Vec::new();
2527         let mut wrong_cgus = Vec::new();
2528
2529         for expected_item in &expected {
2530             let actual_item_with_same_name = actual.iter().find(|ti| ti.name == expected_item.name);
2531
2532             if let Some(actual_item) = actual_item_with_same_name {
2533                 if !expected_item.codegen_units.is_empty() &&
2534                    // Also check for codegen units
2535                    expected_item.codegen_units != actual_item.codegen_units
2536                 {
2537                     wrong_cgus.push((expected_item.clone(), actual_item.clone()));
2538                 }
2539             } else {
2540                 missing.push(expected_item.string.clone());
2541             }
2542         }
2543
2544         let unexpected: Vec<_> = actual
2545             .iter()
2546             .filter(|acgu| !expected.iter().any(|ecgu| acgu.name == ecgu.name))
2547             .map(|acgu| acgu.string.clone())
2548             .collect();
2549
2550         if !missing.is_empty() {
2551             missing.sort();
2552
2553             println!("\nThese items should have been contained but were not:\n");
2554
2555             for item in &missing {
2556                 println!("{}", item);
2557             }
2558
2559             println!("\n");
2560         }
2561
2562         if !unexpected.is_empty() {
2563             let sorted = {
2564                 let mut sorted = unexpected.clone();
2565                 sorted.sort();
2566                 sorted
2567             };
2568
2569             println!("\nThese items were contained but should not have been:\n");
2570
2571             for item in sorted {
2572                 println!("{}", item);
2573             }
2574
2575             println!("\n");
2576         }
2577
2578         if !wrong_cgus.is_empty() {
2579             wrong_cgus.sort_by_key(|pair| pair.0.name.clone());
2580             println!("\nThe following items were assigned to wrong codegen units:\n");
2581
2582             for &(ref expected_item, ref actual_item) in &wrong_cgus {
2583                 println!("{}", expected_item.name);
2584                 println!(
2585                     "  expected: {}",
2586                     codegen_units_to_str(&expected_item.codegen_units)
2587                 );
2588                 println!(
2589                     "  actual:   {}",
2590                     codegen_units_to_str(&actual_item.codegen_units)
2591                 );
2592                 println!("");
2593             }
2594         }
2595
2596         if !(missing.is_empty() && unexpected.is_empty() && wrong_cgus.is_empty()) {
2597             panic!();
2598         }
2599
2600         #[derive(Clone, Eq, PartialEq)]
2601         struct MonoItem {
2602             name: String,
2603             codegen_units: HashSet<String>,
2604             string: String,
2605         }
2606
2607         // [MONO_ITEM] name [@@ (cgu)+]
2608         fn str_to_mono_item(s: &str, cgu_has_crate_disambiguator: bool) -> MonoItem {
2609             let s = if s.starts_with(PREFIX) {
2610                 (&s[PREFIX.len()..]).trim()
2611             } else {
2612                 s.trim()
2613             };
2614
2615             let full_string = format!("{}{}", PREFIX, s);
2616
2617             let parts: Vec<&str> = s
2618                 .split(CGU_MARKER)
2619                 .map(str::trim)
2620                 .filter(|s| !s.is_empty())
2621                 .collect();
2622
2623             let name = parts[0].trim();
2624
2625             let cgus = if parts.len() > 1 {
2626                 let cgus_str = parts[1];
2627
2628                 cgus_str
2629                     .split(' ')
2630                     .map(str::trim)
2631                     .filter(|s| !s.is_empty())
2632                     .map(|s| {
2633                         if cgu_has_crate_disambiguator {
2634                             remove_crate_disambiguator_from_cgu(s)
2635                         } else {
2636                             s.to_string()
2637                         }
2638                     })
2639                     .collect()
2640             } else {
2641                 HashSet::new()
2642             };
2643
2644             MonoItem {
2645                 name: name.to_owned(),
2646                 codegen_units: cgus,
2647                 string: full_string,
2648             }
2649         }
2650
2651         fn codegen_units_to_str(cgus: &HashSet<String>) -> String {
2652             let mut cgus: Vec<_> = cgus.iter().collect();
2653             cgus.sort();
2654
2655             let mut string = String::new();
2656             for cgu in cgus {
2657                 string.push_str(&cgu[..]);
2658                 string.push_str(" ");
2659             }
2660
2661             string
2662         }
2663
2664         // Given a cgu-name-prefix of the form <crate-name>.<crate-disambiguator> or
2665         // the form <crate-name1>.<crate-disambiguator1>-in-<crate-name2>.<crate-disambiguator2>,
2666         // remove all crate-disambiguators.
2667         fn remove_crate_disambiguator_from_cgu(cgu: &str) -> String {
2668             lazy_static! {
2669                 static ref RE: Regex = Regex::new(
2670                     r"^[^\.]+(?P<d1>\.[[:alnum:]]+)(-in-[^\.]+(?P<d2>\.[[:alnum:]]+))?"
2671                 ).unwrap();
2672             }
2673
2674             let captures = RE.captures(cgu).unwrap_or_else(|| {
2675                 panic!("invalid cgu name encountered: {}", cgu)
2676             });
2677
2678             let mut new_name = cgu.to_owned();
2679
2680             if let Some(d2) = captures.name("d2") {
2681                 new_name.replace_range(d2.start() .. d2.end(), "");
2682             }
2683
2684             let d1 = captures.name("d1").unwrap();
2685             new_name.replace_range(d1.start() .. d1.end(), "");
2686
2687             new_name
2688         }
2689     }
2690
2691     fn init_incremental_test(&self) {
2692         // (See `run_incremental_test` for an overview of how incremental tests work.)
2693
2694         // Before any of the revisions have executed, create the
2695         // incremental workproduct directory.  Delete any old
2696         // incremental work products that may be there from prior
2697         // runs.
2698         let incremental_dir = self.incremental_dir();
2699         if incremental_dir.exists() {
2700             // Canonicalizing the path will convert it to the //?/ format
2701             // on Windows, which enables paths longer than 260 character
2702             let canonicalized = incremental_dir.canonicalize().unwrap();
2703             fs::remove_dir_all(canonicalized).unwrap();
2704         }
2705         fs::create_dir_all(&incremental_dir).unwrap();
2706
2707         if self.config.verbose {
2708             print!(
2709                 "init_incremental_test: incremental_dir={}",
2710                 incremental_dir.display()
2711             );
2712         }
2713     }
2714
2715     fn run_incremental_test(&self) {
2716         // Basic plan for a test incremental/foo/bar.rs:
2717         // - load list of revisions rpass1, cfail2, rpass3
2718         //   - each should begin with `rpass`, `cfail`, or `rfail`
2719         //   - if `rpass`, expect compile and execution to succeed
2720         //   - if `cfail`, expect compilation to fail
2721         //   - if `rfail`, expect execution to fail
2722         // - create a directory build/foo/bar.incremental
2723         // - compile foo/bar.rs with -Z incremental=.../foo/bar.incremental and -C rpass1
2724         //   - because name of revision starts with "rpass", expect success
2725         // - compile foo/bar.rs with -Z incremental=.../foo/bar.incremental and -C cfail2
2726         //   - because name of revision starts with "cfail", expect an error
2727         //   - load expected errors as usual, but filter for those that end in `[rfail2]`
2728         // - compile foo/bar.rs with -Z incremental=.../foo/bar.incremental and -C rpass3
2729         //   - because name of revision starts with "rpass", expect success
2730         // - execute build/foo/bar.exe and save output
2731         //
2732         // FIXME -- use non-incremental mode as an oracle? That doesn't apply
2733         // to #[rustc_dirty] and clean tests I guess
2734
2735         let revision = self
2736             .revision
2737             .expect("incremental tests require a list of revisions");
2738
2739         // Incremental workproduct directory should have already been created.
2740         let incremental_dir = self.incremental_dir();
2741         assert!(
2742             incremental_dir.exists(),
2743             "init_incremental_test failed to create incremental dir"
2744         );
2745
2746         // Add an extra flag pointing at the incremental directory.
2747         let mut revision_props = self.props.clone();
2748         revision_props.incremental_dir = Some(incremental_dir);
2749
2750         let revision_cx = TestCx {
2751             config: self.config,
2752             props: &revision_props,
2753             testpaths: self.testpaths,
2754             revision: self.revision,
2755         };
2756
2757         if self.config.verbose {
2758             print!(
2759                 "revision={:?} revision_props={:#?}",
2760                 revision, revision_props
2761             );
2762         }
2763
2764         if revision.starts_with("rpass") {
2765             revision_cx.run_rpass_test();
2766         } else if revision.starts_with("rfail") {
2767             revision_cx.run_rfail_test();
2768         } else if revision.starts_with("cfail") {
2769             revision_cx.run_cfail_test();
2770         } else {
2771             revision_cx.fatal("revision name must begin with rpass, rfail, or cfail");
2772         }
2773     }
2774
2775     /// Directory where incremental work products are stored.
2776     fn incremental_dir(&self) -> PathBuf {
2777         self.output_base_name().with_extension("inc")
2778     }
2779
2780     fn run_rmake_test(&self) {
2781         let cwd = env::current_dir().unwrap();
2782         let src_root = self
2783             .config
2784             .src_base
2785             .parent()
2786             .unwrap()
2787             .parent()
2788             .unwrap()
2789             .parent()
2790             .unwrap();
2791         let src_root = cwd.join(&src_root);
2792
2793         let tmpdir = cwd.join(self.output_base_name());
2794         if tmpdir.exists() {
2795             self.aggressive_rm_rf(&tmpdir).unwrap();
2796         }
2797         create_dir_all(&tmpdir).unwrap();
2798
2799         let host = &self.config.host;
2800         let make = if host.contains("dragonfly")
2801             || host.contains("freebsd")
2802             || host.contains("netbsd")
2803             || host.contains("openbsd")
2804         {
2805             "gmake"
2806         } else {
2807             "make"
2808         };
2809
2810         let mut cmd = Command::new(make);
2811         cmd.current_dir(&self.testpaths.file)
2812             .stdout(Stdio::piped())
2813             .stderr(Stdio::piped())
2814             .env("TARGET", &self.config.target)
2815             .env("PYTHON", &self.config.docck_python)
2816             .env("S", src_root)
2817             .env("RUST_BUILD_STAGE", &self.config.stage_id)
2818             .env("RUSTC", cwd.join(&self.config.rustc_path))
2819             .env("TMPDIR", &tmpdir)
2820             .env("LD_LIB_PATH_ENVVAR", dylib_env_var())
2821             .env("HOST_RPATH_DIR", cwd.join(&self.config.compile_lib_path))
2822             .env("TARGET_RPATH_DIR", cwd.join(&self.config.run_lib_path))
2823             .env("LLVM_COMPONENTS", &self.config.llvm_components)
2824             .env("LLVM_CXXFLAGS", &self.config.llvm_cxxflags)
2825
2826             // We for sure don't want these tests to run in parallel, so make
2827             // sure they don't have access to these vars if we run via `make`
2828             // at the top level
2829             .env_remove("MAKEFLAGS")
2830             .env_remove("MFLAGS")
2831             .env_remove("CARGO_MAKEFLAGS");
2832
2833         if let Some(ref rustdoc) = self.config.rustdoc_path {
2834             cmd.env("RUSTDOC", cwd.join(rustdoc));
2835         }
2836
2837         if let Some(ref node) = self.config.nodejs {
2838             cmd.env("NODE", node);
2839         }
2840
2841         if let Some(ref linker) = self.config.linker {
2842             cmd.env("RUSTC_LINKER", linker);
2843         }
2844
2845         if let Some(ref clang) = self.config.run_clang_based_tests_with {
2846             cmd.env("CLANG", clang);
2847         }
2848
2849         if let Some(ref filecheck) = self.config.llvm_filecheck {
2850             cmd.env("LLVM_FILECHECK", filecheck);
2851         }
2852
2853         if let Some(ref llvm_bin_dir) = self.config.llvm_bin_dir {
2854             cmd.env("LLVM_BIN_DIR", llvm_bin_dir);
2855         }
2856
2857         // We don't want RUSTFLAGS set from the outside to interfere with
2858         // compiler flags set in the test cases:
2859         cmd.env_remove("RUSTFLAGS");
2860
2861         // Use dynamic musl for tests because static doesn't allow creating dylibs
2862         if self.config.host.contains("musl") {
2863             cmd.env("RUSTFLAGS", "-Ctarget-feature=-crt-static")
2864                 .env("IS_MUSL_HOST", "1");
2865         }
2866
2867         if self.config.target.contains("msvc") && self.config.cc != "" {
2868             // We need to pass a path to `lib.exe`, so assume that `cc` is `cl.exe`
2869             // and that `lib.exe` lives next to it.
2870             let lib = Path::new(&self.config.cc).parent().unwrap().join("lib.exe");
2871
2872             // MSYS doesn't like passing flags of the form `/foo` as it thinks it's
2873             // a path and instead passes `C:\msys64\foo`, so convert all
2874             // `/`-arguments to MSVC here to `-` arguments.
2875             let cflags = self
2876                 .config
2877                 .cflags
2878                 .split(' ')
2879                 .map(|s| s.replace("/", "-"))
2880                 .collect::<Vec<_>>()
2881                 .join(" ");
2882
2883             cmd.env("IS_MSVC", "1")
2884                 .env("IS_WINDOWS", "1")
2885                 .env("MSVC_LIB", format!("'{}' -nologo", lib.display()))
2886                 .env("CC", format!("'{}' {}", self.config.cc, cflags))
2887                 .env("CXX", format!("'{}'", &self.config.cxx));
2888         } else {
2889             cmd.env("CC", format!("{} {}", self.config.cc, self.config.cflags))
2890                 .env("CXX", format!("{} {}", self.config.cxx, self.config.cflags))
2891                 .env("AR", &self.config.ar);
2892
2893             if self.config.target.contains("windows") {
2894                 cmd.env("IS_WINDOWS", "1");
2895             }
2896         }
2897
2898         let output = cmd
2899             .spawn()
2900             .and_then(read2_abbreviated)
2901             .expect("failed to spawn `make`");
2902         if !output.status.success() {
2903             let res = ProcRes {
2904                 status: output.status,
2905                 stdout: String::from_utf8_lossy(&output.stdout).into_owned(),
2906                 stderr: String::from_utf8_lossy(&output.stderr).into_owned(),
2907                 cmdline: format!("{:?}", cmd),
2908             };
2909             self.fatal_proc_rec("make failed", &res);
2910         }
2911     }
2912
2913     fn aggressive_rm_rf(&self, path: &Path) -> io::Result<()> {
2914         for e in path.read_dir()? {
2915             let entry = e?;
2916             let path = entry.path();
2917             if entry.file_type()?.is_dir() {
2918                 self.aggressive_rm_rf(&path)?;
2919             } else {
2920                 // Remove readonly files as well on windows (by default we can't)
2921                 fs::remove_file(&path).or_else(|e| {
2922                     if cfg!(windows) && e.kind() == io::ErrorKind::PermissionDenied {
2923                         let mut meta = entry.metadata()?.permissions();
2924                         meta.set_readonly(false);
2925                         fs::set_permissions(&path, meta)?;
2926                         fs::remove_file(&path)
2927                     } else {
2928                         Err(e)
2929                     }
2930                 })?;
2931             }
2932         }
2933         fs::remove_dir(path)
2934     }
2935
2936     fn run_js_doc_test(&self) {
2937         if let Some(nodejs) = &self.config.nodejs {
2938             let out_dir = self.output_base_dir();
2939
2940             self.document(&out_dir);
2941
2942             let root = self.config.find_rust_src_root().unwrap();
2943             let res = self.cmd2procres(
2944                 Command::new(&nodejs)
2945                     .arg(root.join("src/tools/rustdoc-js/tester.js"))
2946                     .arg(out_dir.parent().expect("no parent"))
2947                     .arg(&self.testpaths.file.file_stem().expect("couldn't get file stem")),
2948             );
2949             if !res.status.success() {
2950                 self.fatal_proc_rec("rustdoc-js test failed!", &res);
2951             }
2952         } else {
2953             self.fatal("no nodeJS");
2954         }
2955     }
2956
2957     fn load_compare_outputs(&self, proc_res: &ProcRes,
2958         output_kind: TestOutput, explicit_format: bool) -> usize {
2959
2960         let (stderr_kind, stdout_kind) = match output_kind {
2961             TestOutput::Compile => (UI_STDERR, UI_STDOUT),
2962             TestOutput::Run => (UI_RUN_STDERR, UI_RUN_STDOUT)
2963         };
2964
2965         let expected_stderr = self.load_expected_output(stderr_kind);
2966         let expected_stdout = self.load_expected_output(stdout_kind);
2967
2968         let normalized_stdout = match output_kind {
2969             TestOutput::Run if self.config.remote_test_client.is_some() => {
2970                 // When tests are run using the remote-test-client, the string
2971                 // 'uploaded "$TEST_BUILD_DIR/<test_executable>, waiting for result"'
2972                 // is printed to stdout by the client and then captured in the ProcRes,
2973                 // so it needs to be removed when comparing the run-pass test execution output
2974                 lazy_static! {
2975                     static ref REMOTE_TEST_RE: Regex = Regex::new(
2976                         "^uploaded \"\\$TEST_BUILD_DIR(/[[:alnum:]_\\-]+)+\", waiting for result\n"
2977                     ).unwrap();
2978                 }
2979                 REMOTE_TEST_RE.replace(
2980                     &self.normalize_output(&proc_res.stdout, &self.props.normalize_stdout),
2981                     ""
2982                 ).to_string()
2983             }
2984             _ => self.normalize_output(&proc_res.stdout, &self.props.normalize_stdout)
2985         };
2986
2987         let stderr = if explicit_format {
2988             proc_res.stderr.clone()
2989         } else {
2990             json::extract_rendered(&proc_res.stderr)
2991         };
2992
2993         let normalized_stderr = self.normalize_output(&stderr, &self.props.normalize_stderr);
2994         let mut errors = 0;
2995         match output_kind {
2996             TestOutput::Compile => {
2997                 if !self.props.dont_check_compiler_stdout {
2998                     errors += self.compare_output("stdout", &normalized_stdout, &expected_stdout);
2999                 }
3000                 if !self.props.dont_check_compiler_stderr {
3001                     errors += self.compare_output("stderr", &normalized_stderr, &expected_stderr);
3002                 }
3003             }
3004             TestOutput::Run => {
3005                 errors += self.compare_output(stdout_kind, &normalized_stdout, &expected_stdout);
3006                 errors += self.compare_output(stderr_kind, &normalized_stderr, &expected_stderr);
3007             }
3008         }
3009         errors
3010     }
3011
3012     fn run_ui_test(&self) {
3013         // if the user specified a format in the ui test
3014         // print the output to the stderr file, otherwise extract
3015         // the rendered error messages from json and print them
3016         let explicit = self
3017             .props
3018             .compile_flags
3019             .iter()
3020             .any(|s| s.contains("--error-format"));
3021         let proc_res = self.compile_test();
3022         self.check_if_test_should_compile(&proc_res);
3023
3024         let expected_fixed = self.load_expected_output(UI_FIXED);
3025
3026         let modes_to_prune = vec![CompareMode::Nll];
3027         self.prune_duplicate_outputs(&modes_to_prune);
3028
3029         let mut errors = self.load_compare_outputs(&proc_res, TestOutput::Compile, explicit);
3030
3031         if self.config.compare_mode.is_some() {
3032             // don't test rustfix with nll right now
3033         } else if self.config.rustfix_coverage {
3034             // Find out which tests have `MachineApplicable` suggestions but are missing
3035             // `run-rustfix` or `run-rustfix-only-machine-applicable` headers.
3036             //
3037             // This will return an empty `Vec` in case the executed test file has a
3038             // `compile-flags: --error-format=xxxx` header with a value other than `json`.
3039             let suggestions = get_suggestions_from_json(
3040                 &proc_res.stderr,
3041                 &HashSet::new(),
3042                 Filter::MachineApplicableOnly
3043             ).unwrap_or_default();
3044             if suggestions.len() > 0
3045                 && !self.props.run_rustfix
3046                 && !self.props.rustfix_only_machine_applicable {
3047                     let mut coverage_file_path = self.config.build_base.clone();
3048                     coverage_file_path.push("rustfix_missing_coverage.txt");
3049                     debug!("coverage_file_path: {}", coverage_file_path.display());
3050
3051                     let mut file = OpenOptions::new()
3052                         .create(true)
3053                         .append(true)
3054                         .open(coverage_file_path.as_path())
3055                         .expect("could not create or open file");
3056
3057                     if let Err(_) = writeln!(file, "{}", self.testpaths.file.display()) {
3058                         panic!("couldn't write to {}", coverage_file_path.display());
3059                     }
3060             }
3061         } else if self.props.run_rustfix {
3062             // Apply suggestions from rustc to the code itself
3063             let unfixed_code = self
3064                 .load_expected_output_from_path(&self.testpaths.file)
3065                 .unwrap();
3066             let suggestions = get_suggestions_from_json(
3067                 &proc_res.stderr,
3068                 &HashSet::new(),
3069                 if self.props.rustfix_only_machine_applicable {
3070                     Filter::MachineApplicableOnly
3071                 } else {
3072                     Filter::Everything
3073                 },
3074             ).unwrap();
3075             let fixed_code = apply_suggestions(&unfixed_code, &suggestions).expect(&format!(
3076                 "failed to apply suggestions for {:?} with rustfix",
3077                 self.testpaths.file
3078             ));
3079
3080             errors += self.compare_output("fixed", &fixed_code, &expected_fixed);
3081         } else if !expected_fixed.is_empty() {
3082             panic!(
3083                 "the `// run-rustfix` directive wasn't found but a `*.fixed` \
3084                  file was found"
3085             );
3086         }
3087
3088         if errors > 0 {
3089             println!("To update references, rerun the tests and pass the `--bless` flag");
3090             let relative_path_to_file = self
3091                 .testpaths
3092                 .relative_dir
3093                 .join(self.testpaths.file.file_name().unwrap());
3094             println!(
3095                 "To only update this specific test, also pass `--test-args {}`",
3096                 relative_path_to_file.display(),
3097             );
3098             self.fatal_proc_rec(
3099                 &format!("{} errors occurred comparing output.", errors),
3100                 &proc_res,
3101             );
3102         }
3103
3104         let expected_errors = errors::load_errors(&self.testpaths.file, self.revision);
3105
3106         if self.should_run_successfully() {
3107             let proc_res = self.exec_compiled_test();
3108             let run_output_errors = if self.props.check_run_results {
3109                 self.load_compare_outputs(&proc_res, TestOutput::Run, explicit)
3110             } else {
3111                 0
3112             };
3113             if run_output_errors > 0 {
3114                 self.fatal_proc_rec(
3115                     &format!("{} errors occured comparing run output.", run_output_errors),
3116                     &proc_res,
3117                 );
3118             }
3119             if !proc_res.status.success() {
3120                 self.fatal_proc_rec("test run failed!", &proc_res);
3121             }
3122         }
3123
3124         debug!("run_ui_test: explicit={:?} config.compare_mode={:?} expected_errors={:?} \
3125                proc_res.status={:?} props.error_patterns={:?}",
3126                explicit, self.config.compare_mode, expected_errors, proc_res.status,
3127                self.props.error_patterns);
3128         if !explicit && self.config.compare_mode.is_none() {
3129             if !proc_res.status.success() {
3130                 if !self.props.error_patterns.is_empty() {
3131                     // "// error-pattern" comments
3132                     self.check_error_patterns(&proc_res.stderr, &proc_res);
3133                 } else {
3134                     // "//~ERROR comments"
3135                     self.check_expected_errors(expected_errors, &proc_res);
3136                 }
3137             }
3138         }
3139
3140         if self.props.run_rustfix && self.config.compare_mode.is_none() {
3141             // And finally, compile the fixed code and make sure it both
3142             // succeeds and has no diagnostics.
3143             let mut rustc = self.make_compile_args(
3144                 &self.testpaths.file.with_extension(UI_FIXED),
3145                 TargetLocation::ThisFile(self.make_exe_name()),
3146             );
3147             rustc.arg("-L").arg(&self.aux_output_dir_name());
3148             let res = self.compose_and_run_compiler(rustc, None);
3149             if !res.status.success() {
3150                 self.fatal_proc_rec("failed to compile fixed code", &res);
3151             }
3152             if !res.stderr.is_empty() && !self.props.rustfix_only_machine_applicable {
3153                 self.fatal_proc_rec("fixed code is still producing diagnostics", &res);
3154             }
3155         }
3156     }
3157
3158     fn run_mir_opt_test(&self) {
3159         let proc_res = self.compile_test();
3160
3161         if !proc_res.status.success() {
3162             self.fatal_proc_rec("compilation failed!", &proc_res);
3163         }
3164
3165         let proc_res = self.exec_compiled_test();
3166
3167         if !proc_res.status.success() {
3168             self.fatal_proc_rec("test run failed!", &proc_res);
3169         }
3170         self.check_mir_dump();
3171     }
3172
3173     fn check_mir_dump(&self) {
3174         let test_file_contents = fs::read_to_string(&self.testpaths.file).unwrap();
3175         if let Some(idx) = test_file_contents.find("// END RUST SOURCE") {
3176             let (_, tests_text) = test_file_contents.split_at(idx + "// END_RUST SOURCE".len());
3177             let tests_text_str = String::from(tests_text);
3178             let mut curr_test: Option<&str> = None;
3179             let mut curr_test_contents = vec![ExpectedLine::Elision];
3180             for l in tests_text_str.lines() {
3181                 debug!("line: {:?}", l);
3182                 if l.starts_with("// START ") {
3183                     let (_, t) = l.split_at("// START ".len());
3184                     curr_test = Some(t);
3185                 } else if l.starts_with("// END") {
3186                     let (_, t) = l.split_at("// END ".len());
3187                     if Some(t) != curr_test {
3188                         panic!("mismatched START END test name");
3189                     }
3190                     self.compare_mir_test_output(curr_test.unwrap(), &curr_test_contents);
3191                     curr_test = None;
3192                     curr_test_contents.clear();
3193                     curr_test_contents.push(ExpectedLine::Elision);
3194                 } else if l.is_empty() {
3195                     // ignore
3196                 } else if l.starts_with("//") && l.split_at("//".len()).1.trim() == "..." {
3197                     curr_test_contents.push(ExpectedLine::Elision)
3198                 } else if l.starts_with("// ") {
3199                     let (_, test_content) = l.split_at("// ".len());
3200                     curr_test_contents.push(ExpectedLine::Text(test_content));
3201                 }
3202             }
3203         }
3204     }
3205
3206     fn check_mir_test_timestamp(&self, test_name: &str, output_file: &Path) {
3207         let t = |file| fs::metadata(file).unwrap().modified().unwrap();
3208         let source_file = &self.testpaths.file;
3209         let output_time = t(output_file);
3210         let source_time = t(source_file);
3211         if source_time > output_time {
3212             debug!(
3213                 "source file time: {:?} output file time: {:?}",
3214                 source_time, output_time
3215             );
3216             panic!(
3217                 "test source file `{}` is newer than potentially stale output file `{}`.",
3218                 source_file.display(),
3219                 test_name
3220             );
3221         }
3222     }
3223
3224     fn compare_mir_test_output(&self, test_name: &str, expected_content: &[ExpectedLine<&str>]) {
3225         let mut output_file = PathBuf::new();
3226         output_file.push(self.get_mir_dump_dir());
3227         output_file.push(test_name);
3228         debug!("comparing the contests of: {:?}", output_file);
3229         debug!("with: {:?}", expected_content);
3230         if !output_file.exists() {
3231             panic!(
3232                 "Output file `{}` from test does not exist",
3233                 output_file.into_os_string().to_string_lossy()
3234             );
3235         }
3236         self.check_mir_test_timestamp(test_name, &output_file);
3237
3238         let dumped_string = fs::read_to_string(&output_file).unwrap();
3239         let mut dumped_lines = dumped_string
3240             .lines()
3241             .map(|l| nocomment_mir_line(l))
3242             .filter(|l| !l.is_empty());
3243         let mut expected_lines = expected_content
3244             .iter()
3245             .filter(|&l| {
3246                 if let &ExpectedLine::Text(l) = l {
3247                     !l.is_empty()
3248                 } else {
3249                     true
3250                 }
3251             })
3252             .peekable();
3253
3254         let compare = |expected_line, dumped_line| {
3255             let e_norm = normalize_mir_line(expected_line);
3256             let d_norm = normalize_mir_line(dumped_line);
3257             debug!("found: {:?}", d_norm);
3258             debug!("expected: {:?}", e_norm);
3259             e_norm == d_norm
3260         };
3261
3262         let error = |expected_line, extra_msg| {
3263             let normalize_all = dumped_string
3264                 .lines()
3265                 .map(nocomment_mir_line)
3266                 .filter(|l| !l.is_empty())
3267                 .collect::<Vec<_>>()
3268                 .join("\n");
3269             let f = |l: &ExpectedLine<_>| match l {
3270                 &ExpectedLine::Elision => "... (elided)".into(),
3271                 &ExpectedLine::Text(t) => t,
3272             };
3273             let expected_content = expected_content
3274                 .iter()
3275                 .map(|l| f(l))
3276                 .collect::<Vec<_>>()
3277                 .join("\n");
3278             panic!(
3279                 "Did not find expected line, error: {}\n\
3280                  Expected Line: {:?}\n\
3281                  Test Name: {}\n\
3282                  Expected:\n{}\n\
3283                  Actual:\n{}",
3284                 extra_msg, expected_line, test_name, expected_content, normalize_all
3285             );
3286         };
3287
3288         // We expect each non-empty line to appear consecutively, non-consecutive lines
3289         // must be separated by at least one Elision
3290         let mut start_block_line = None;
3291         while let Some(dumped_line) = dumped_lines.next() {
3292             match expected_lines.next() {
3293                 Some(&ExpectedLine::Text(expected_line)) => {
3294                     let normalized_expected_line = normalize_mir_line(expected_line);
3295                     if normalized_expected_line.contains(":{") {
3296                         start_block_line = Some(expected_line);
3297                     }
3298
3299                     if !compare(expected_line, dumped_line) {
3300                         error!("{:?}", start_block_line);
3301                         error(
3302                             expected_line,
3303                             format!(
3304                                 "Mismatch in lines\n\
3305                                  Current block: {}\n\
3306                                  Actual Line: {:?}",
3307                                 start_block_line.unwrap_or("None"),
3308                                 dumped_line
3309                             ),
3310                         );
3311                     }
3312                 }
3313                 Some(&ExpectedLine::Elision) => {
3314                     // skip any number of elisions in a row.
3315                     while let Some(&&ExpectedLine::Elision) = expected_lines.peek() {
3316                         expected_lines.next();
3317                     }
3318                     if let Some(&ExpectedLine::Text(expected_line)) = expected_lines.next() {
3319                         let mut found = compare(expected_line, dumped_line);
3320                         if found {
3321                             continue;
3322                         }
3323                         while let Some(dumped_line) = dumped_lines.next() {
3324                             found = compare(expected_line, dumped_line);
3325                             if found {
3326                                 break;
3327                             }
3328                         }
3329                         if !found {
3330                             error(expected_line, "ran out of mir dump to match against".into());
3331                         }
3332                     }
3333                 }
3334                 None => {}
3335             }
3336         }
3337     }
3338
3339     fn get_mir_dump_dir(&self) -> PathBuf {
3340         let mut mir_dump_dir = PathBuf::from(self.config.build_base.as_path());
3341         debug!("input_file: {:?}", self.testpaths.file);
3342         mir_dump_dir.push(&self.testpaths.relative_dir);
3343         mir_dump_dir.push(self.testpaths.file.file_stem().unwrap());
3344         mir_dump_dir
3345     }
3346
3347     fn normalize_output(&self, output: &str, custom_rules: &[(String, String)]) -> String {
3348         let cflags = self.props.compile_flags.join(" ");
3349         let json = cflags.contains("--error-format json")
3350             || cflags.contains("--error-format pretty-json")
3351             || cflags.contains("--error-format=json")
3352             || cflags.contains("--error-format=pretty-json");
3353
3354         let mut normalized = output.to_string();
3355
3356         let mut normalize_path = |from: &Path, to: &str| {
3357             let mut from = from.display().to_string();
3358             if json {
3359                 from = from.replace("\\", "\\\\");
3360             }
3361             normalized = normalized.replace(&from, to);
3362         };
3363
3364         let parent_dir = self.testpaths.file.parent().unwrap();
3365         normalize_path(parent_dir, "$DIR");
3366
3367         // Paths into the libstd/libcore
3368         let src_dir = self.config.src_base.parent().unwrap().parent().unwrap();
3369         normalize_path(src_dir, "$SRC_DIR");
3370
3371         // Paths into the build directory
3372         let test_build_dir = &self.config.build_base;
3373         let parent_build_dir = test_build_dir.parent().unwrap().parent().unwrap().parent().unwrap();
3374
3375         // eg. /home/user/rust/build/x86_64-unknown-linux-gnu/test/ui
3376         normalize_path(test_build_dir, "$TEST_BUILD_DIR");
3377         // eg. /home/user/rust/build
3378         normalize_path(parent_build_dir, "$BUILD_DIR");
3379
3380         // Paths into lib directory.
3381         normalize_path(&parent_build_dir.parent().unwrap().join("lib"), "$LIB_DIR");
3382
3383         if json {
3384             // escaped newlines in json strings should be readable
3385             // in the stderr files. There's no point int being correct,
3386             // since only humans process the stderr files.
3387             // Thus we just turn escaped newlines back into newlines.
3388             normalized = normalized.replace("\\n", "\n");
3389         }
3390
3391         // If there are `$SRC_DIR` normalizations with line and column numbers, then replace them
3392         // with placeholders as we do not want tests needing updated when compiler source code
3393         // changes.
3394         // eg. $SRC_DIR/libcore/mem.rs:323:14 becomes $SRC_DIR/libcore/mem.rs:LL:COL
3395         normalized = Regex::new("SRC_DIR(.+):\\d+:\\d+").unwrap()
3396             .replace_all(&normalized, "SRC_DIR$1:LL:COL").into_owned();
3397
3398         normalized = Self::normalize_platform_differences(&normalized);
3399         normalized = normalized.replace("\t", "\\t"); // makes tabs visible
3400
3401         // Remove test annotations like `//~ ERROR text` from the output,
3402         // since they duplicate actual errors and make the output hard to read.
3403         normalized = Regex::new("\\s*//(\\[.*\\])?~.*").unwrap()
3404             .replace_all(&normalized, "").into_owned();
3405
3406         for rule in custom_rules {
3407             let re = Regex::new(&rule.0).expect("bad regex in custom normalization rule");
3408             normalized = re.replace_all(&normalized, &rule.1[..]).into_owned();
3409         }
3410         normalized
3411     }
3412
3413     /// Normalize output differences across platforms. Generally changes Windows output to be more
3414     /// Unix-like.
3415     ///
3416     /// Replaces backslashes in paths with forward slashes, and replaces CRLF line endings
3417     /// with LF.
3418     fn normalize_platform_differences(output: &str) -> String {
3419         lazy_static! {
3420             /// Used to find Windows paths.
3421             ///
3422             /// It's not possible to detect paths in the error messages generally, but this is a
3423             /// decent enough heuristic.
3424             static ref PATH_BACKSLASH_RE: Regex = Regex::new(r#"(?x)
3425                 (?:
3426                   # Match paths that don't include spaces.
3427                   (?:\\[\pL\pN\.\-_']+)+\.\pL+
3428                 |
3429                   # If the path starts with a well-known root, then allow spaces.
3430                   \$(?:DIR|SRC_DIR|TEST_BUILD_DIR|BUILD_DIR|LIB_DIR)(?:\\[\pL\pN\.\-_' ]+)+
3431                 )"#
3432             ).unwrap();
3433         }
3434
3435         let output = output.replace(r"\\", r"\");
3436
3437         PATH_BACKSLASH_RE.replace_all(&output, |caps: &Captures<'_>| {
3438             println!("{}", &caps[0]);
3439             caps[0].replace(r"\", "/")
3440         }).replace("\r\n", "\n")
3441     }
3442
3443     fn expected_output_path(&self, kind: &str) -> PathBuf {
3444         let mut path = expected_output_path(
3445             &self.testpaths,
3446             self.revision,
3447             &self.config.compare_mode,
3448             kind,
3449         );
3450
3451         if !path.exists() {
3452             if let Some(CompareMode::Polonius) = self.config.compare_mode {
3453                 path = expected_output_path(
3454                     &self.testpaths,
3455                     self.revision,
3456                     &Some(CompareMode::Nll),
3457                     kind,
3458                 );
3459             }
3460         }
3461
3462         if !path.exists() {
3463             path = expected_output_path(&self.testpaths, self.revision, &None, kind);
3464         }
3465
3466         path
3467     }
3468
3469     fn load_expected_output(&self, kind: &str) -> String {
3470         let path = self.expected_output_path(kind);
3471         if path.exists() {
3472             match self.load_expected_output_from_path(&path) {
3473                 Ok(x) => x,
3474                 Err(x) => self.fatal(&x),
3475             }
3476         } else {
3477             String::new()
3478         }
3479     }
3480
3481     fn load_expected_output_from_path(&self, path: &Path) -> Result<String, String> {
3482         fs::read_to_string(path).map_err(|err| {
3483             format!("failed to load expected output from `{}`: {}", path.display(), err)
3484         })
3485     }
3486
3487     fn delete_file(&self, file: &PathBuf) {
3488         if let Err(e) = fs::remove_file(file) {
3489             self.fatal(&format!(
3490                 "failed to delete `{}`: {}",
3491                 file.display(),
3492                 e,
3493             ));
3494         }
3495     }
3496
3497     fn compare_output(&self, kind: &str, actual: &str, expected: &str) -> usize {
3498         if actual == expected {
3499             return 0;
3500         }
3501
3502         if !self.config.bless {
3503             if expected.is_empty() {
3504                 println!("normalized {}:\n{}\n", kind, actual);
3505             } else {
3506                 println!("diff of {}:\n", kind);
3507                 let diff_results = make_diff(expected, actual, 3);
3508                 for result in diff_results {
3509                     let mut line_number = result.line_number;
3510                     for line in result.lines {
3511                         match line {
3512                             DiffLine::Expected(e) => {
3513                                 println!("-\t{}", e);
3514                                 line_number += 1;
3515                             }
3516                             DiffLine::Context(c) => {
3517                                 println!("{}\t{}", line_number, c);
3518                                 line_number += 1;
3519                             }
3520                             DiffLine::Resulting(r) => {
3521                                 println!("+\t{}", r);
3522                             }
3523                         }
3524                     }
3525                     println!("");
3526                 }
3527             }
3528         }
3529
3530         let mode = self.config.compare_mode.as_ref().map_or("", |m| m.to_str());
3531         let output_file = self
3532             .output_base_name()
3533             .with_extra_extension(self.revision.unwrap_or(""))
3534             .with_extra_extension(mode)
3535             .with_extra_extension(kind);
3536
3537         let mut files = vec![output_file];
3538         if self.config.bless {
3539             files.push(expected_output_path(
3540                 self.testpaths,
3541                 self.revision,
3542                 &self.config.compare_mode,
3543                 kind,
3544             ));
3545         }
3546
3547         for output_file in &files {
3548             if actual.is_empty() {
3549                 self.delete_file(output_file);
3550             } else if let Err(err) = fs::write(&output_file, &actual) {
3551                 self.fatal(&format!(
3552                     "failed to write {} to `{}`: {}",
3553                     kind,
3554                     output_file.display(),
3555                     err,
3556                 ));
3557             }
3558         }
3559
3560         println!("\nThe actual {0} differed from the expected {0}.", kind);
3561         for output_file in files {
3562             println!("Actual {} saved to {}", kind, output_file.display());
3563         }
3564         if self.config.bless {
3565             0
3566         } else {
3567             1
3568         }
3569     }
3570
3571     fn prune_duplicate_output(&self, mode: CompareMode, kind: &str, canon_content: &str) {
3572         let examined_path = expected_output_path(
3573             &self.testpaths,
3574             self.revision,
3575             &Some(mode),
3576             kind,
3577         );
3578
3579         let examined_content = self
3580             .load_expected_output_from_path(&examined_path)
3581             .unwrap_or_else(|_| String::new());
3582
3583         if examined_path.exists() && canon_content == &examined_content {
3584             self.delete_file(&examined_path);
3585         }
3586     }
3587
3588     fn prune_duplicate_outputs(&self, modes: &[CompareMode]) {
3589         if self.config.bless {
3590             for kind in UI_EXTENSIONS {
3591                 let canon_comparison_path = expected_output_path(
3592                     &self.testpaths,
3593                     self.revision,
3594                     &None,
3595                     kind,
3596                 );
3597
3598                 if let Ok(canon) = self.load_expected_output_from_path(&canon_comparison_path) {
3599                     for mode in modes {
3600                         self.prune_duplicate_output(mode.clone(), kind, &canon);
3601                     }
3602                 }
3603             }
3604         }
3605     }
3606
3607     fn create_stamp(&self) {
3608         let stamp = crate::stamp(&self.config, self.testpaths, self.revision);
3609         fs::write(&stamp, compute_stamp_hash(&self.config)).unwrap();
3610     }
3611 }
3612
3613 struct ProcArgs {
3614     prog: String,
3615     args: Vec<String>,
3616 }
3617
3618 pub struct ProcRes {
3619     status: ExitStatus,
3620     stdout: String,
3621     stderr: String,
3622     cmdline: String,
3623 }
3624
3625 impl ProcRes {
3626     pub fn fatal(&self, err: Option<&str>) -> ! {
3627         if let Some(e) = err {
3628             println!("\nerror: {}", e);
3629         }
3630         print!(
3631             "\
3632              status: {}\n\
3633              command: {}\n\
3634              stdout:\n\
3635              ------------------------------------------\n\
3636              {}\n\
3637              ------------------------------------------\n\
3638              stderr:\n\
3639              ------------------------------------------\n\
3640              {}\n\
3641              ------------------------------------------\n\
3642              \n",
3643             self.status, self.cmdline,
3644             json::extract_rendered(&self.stdout),
3645             json::extract_rendered(&self.stderr),
3646         );
3647         // Use resume_unwind instead of panic!() to prevent a panic message + backtrace from
3648         // compiletest, which is unnecessary noise.
3649         std::panic::resume_unwind(Box::new(()));
3650     }
3651 }
3652
3653 enum TargetLocation {
3654     ThisFile(PathBuf),
3655     ThisDirectory(PathBuf),
3656 }
3657
3658 #[derive(Clone, PartialEq, Eq)]
3659 enum ExpectedLine<T: AsRef<str>> {
3660     Elision,
3661     Text(T),
3662 }
3663
3664 impl<T> fmt::Debug for ExpectedLine<T>
3665 where
3666     T: AsRef<str> + fmt::Debug,
3667 {
3668     fn fmt(&self, formatter: &mut fmt::Formatter<'_>) -> fmt::Result {
3669         if let &ExpectedLine::Text(ref t) = self {
3670             write!(formatter, "{:?}", t)
3671         } else {
3672             write!(formatter, "\"...\" (Elision)")
3673         }
3674     }
3675 }
3676
3677 fn normalize_mir_line(line: &str) -> String {
3678     nocomment_mir_line(line).replace(char::is_whitespace, "")
3679 }
3680
3681 fn nocomment_mir_line(line: &str) -> &str {
3682     if let Some(idx) = line.find("//") {
3683         let (l, _) = line.split_at(idx);
3684         l.trim_end()
3685     } else {
3686         line
3687     }
3688 }
3689
3690 fn read2_abbreviated(mut child: Child) -> io::Result<Output> {
3691     use crate::read2::read2;
3692     use std::mem::replace;
3693
3694     const HEAD_LEN: usize = 160 * 1024;
3695     const TAIL_LEN: usize = 256 * 1024;
3696
3697     enum ProcOutput {
3698         Full(Vec<u8>),
3699         Abbreviated {
3700             head: Vec<u8>,
3701             skipped: usize,
3702             tail: Box<[u8]>,
3703         },
3704     }
3705
3706     impl ProcOutput {
3707         fn extend(&mut self, data: &[u8]) {
3708             let new_self = match *self {
3709                 ProcOutput::Full(ref mut bytes) => {
3710                     bytes.extend_from_slice(data);
3711                     let new_len = bytes.len();
3712                     if new_len <= HEAD_LEN + TAIL_LEN {
3713                         return;
3714                     }
3715                     let tail = bytes.split_off(new_len - TAIL_LEN).into_boxed_slice();
3716                     let head = replace(bytes, Vec::new());
3717                     let skipped = new_len - HEAD_LEN - TAIL_LEN;
3718                     ProcOutput::Abbreviated {
3719                         head,
3720                         skipped,
3721                         tail,
3722                     }
3723                 }
3724                 ProcOutput::Abbreviated {
3725                     ref mut skipped,
3726                     ref mut tail,
3727                     ..
3728                 } => {
3729                     *skipped += data.len();
3730                     if data.len() <= TAIL_LEN {
3731                         tail[..data.len()].copy_from_slice(data);
3732                         tail.rotate_left(data.len());
3733                     } else {
3734                         tail.copy_from_slice(&data[(data.len() - TAIL_LEN)..]);
3735                     }
3736                     return;
3737                 }
3738             };
3739             *self = new_self;
3740         }
3741
3742         fn into_bytes(self) -> Vec<u8> {
3743             match self {
3744                 ProcOutput::Full(bytes) => bytes,
3745                 ProcOutput::Abbreviated {
3746                     mut head,
3747                     skipped,
3748                     tail,
3749                 } => {
3750                     write!(&mut head, "\n\n<<<<<< SKIPPED {} BYTES >>>>>>\n\n", skipped).unwrap();
3751                     head.extend_from_slice(&tail);
3752                     head
3753                 }
3754             }
3755         }
3756     }
3757
3758     let mut stdout = ProcOutput::Full(Vec::new());
3759     let mut stderr = ProcOutput::Full(Vec::new());
3760
3761     drop(child.stdin.take());
3762     read2(
3763         child.stdout.take().unwrap(),
3764         child.stderr.take().unwrap(),
3765         &mut |is_stdout, data, _| {
3766             if is_stdout { &mut stdout } else { &mut stderr }.extend(data);
3767             data.clear();
3768         },
3769     )?;
3770     let status = child.wait()?;
3771
3772     Ok(Output {
3773         status,
3774         stdout: stdout.into_bytes(),
3775         stderr: stderr.into_bytes(),
3776     })
3777 }