src/tools/compiletest/src/util.rs

   1 use crate::common::Config;
   2 use std::env;
   3 use std::ffi::OsStr;
   4 use std::path::PathBuf;
   5 use std::process::Command;
   6
   7 use tracing::*;
   8
   9 #[cfg(test)]
  10 mod tests;
  11
  12 pub const ASAN_SUPPORTED_TARGETS: &[&str] = &[
  13     "aarch64-apple-darwin",
  14     "aarch64-fuchsia",
  15     "aarch64-linux-android",
  16     "aarch64-unknown-linux-gnu",
  17     "arm-linux-androideabi",
  18     "armv7-linux-androideabi",
  19     "i686-linux-android",
  20     "i686-unknown-linux-gnu",
  21     "x86_64-apple-darwin",
  22     "x86_64-fuchsia",
  23     "x86_64-linux-android",
  24     "x86_64-unknown-freebsd",
  25     "x86_64-unknown-linux-gnu",
  26 ];
  27
  28 // FIXME(rcvalle): More targets are likely supported.
  29 pub const CFI_SUPPORTED_TARGETS: &[&str] = &[
  30     "aarch64-apple-darwin",
  31     "aarch64-fuchsia",
  32     "aarch64-linux-android",
  33     "aarch64-unknown-freebsd",
  34     "aarch64-unknown-linux-gnu",
  35     "x86_64-apple-darwin",
  36     "x86_64-fuchsia",
  37     "x86_64-pc-solaris",
  38     "x86_64-unknown-freebsd",
  39     "x86_64-unknown-illumos",
  40     "x86_64-unknown-linux-gnu",
  41     "x86_64-unknown-linux-musl",
  42     "x86_64-unknown-netbsd",
  43 ];
  44
  45 pub const LSAN_SUPPORTED_TARGETS: &[&str] = &[
  46     // FIXME: currently broken, see #88132
  47     // "aarch64-apple-darwin",
  48     "aarch64-unknown-linux-gnu",
  49     "x86_64-apple-darwin",
  50     "x86_64-unknown-linux-gnu",
  51 ];
  52
  53 pub const MSAN_SUPPORTED_TARGETS: &[&str] =
  54     &["aarch64-unknown-linux-gnu", "x86_64-unknown-freebsd", "x86_64-unknown-linux-gnu"];
  55
  56 pub const TSAN_SUPPORTED_TARGETS: &[&str] = &[
  57     "aarch64-apple-darwin",
  58     "aarch64-unknown-linux-gnu",
  59     "x86_64-apple-darwin",
  60     "x86_64-unknown-freebsd",
  61     "x86_64-unknown-linux-gnu",
  62 ];
  63
  64 pub const HWASAN_SUPPORTED_TARGETS: &[&str] =
  65     &["aarch64-linux-android", "aarch64-unknown-linux-gnu"];
  66
  67 pub const MEMTAG_SUPPORTED_TARGETS: &[&str] =
  68     &["aarch64-linux-android", "aarch64-unknown-linux-gnu"];
  69
  70 pub const SHADOWCALLSTACK_SUPPORTED_TARGETS: &[&str] = &["aarch64-linux-android"];
  71
  72 pub fn make_new_path(path: &str) -> String {
  73     assert!(cfg!(windows));
  74     // Windows just uses PATH as the library search path, so we have to
  75     // maintain the current value while adding our own
  76     match env::var(lib_path_env_var()) {
  77         Ok(curr) => format!("{}{}{}", path, path_div(), curr),
  78         Err(..) => path.to_owned(),
  79     }
  80 }
  81
  82 pub fn lib_path_env_var() -> &'static str {
  83     "PATH"
  84 }
  85 fn path_div() -> &'static str {
  86     ";"
  87 }
  88
  89 pub fn logv(config: &Config, s: String) {
  90     debug!("{}", s);
  91     if config.verbose {
  92         println!("{}", s);
  93     }
  94 }
  95
  96 pub trait PathBufExt {
  97     /// Append an extension to the path, even if it already has one.
  98     fn with_extra_extension<S: AsRef<OsStr>>(&self, extension: S) -> PathBuf;
  99 }
 100
 101 impl PathBufExt for PathBuf {
 102     fn with_extra_extension<S: AsRef<OsStr>>(&self, extension: S) -> PathBuf {
 103         if extension.as_ref().is_empty() {
 104             self.clone()
 105         } else {
 106             let mut fname = self.file_name().unwrap().to_os_string();
 107             if !extension.as_ref().to_str().unwrap().starts_with('.') {
 108                 fname.push(".");
 109             }
 110             fname.push(extension);
 111             self.with_file_name(fname)
 112         }
 113     }
 114 }
 115
 116 /// The name of the environment variable that holds dynamic library locations.
 117 pub fn dylib_env_var() -> &'static str {
 118     if cfg!(windows) {
 119         "PATH"
 120     } else if cfg!(target_os = "macos") {
 121         "DYLD_LIBRARY_PATH"
 122     } else if cfg!(target_os = "haiku") {
 123         "LIBRARY_PATH"
 124     } else {
 125         "LD_LIBRARY_PATH"
 126     }
 127 }
 128
 129 /// Adds a list of lookup paths to `cmd`'s dynamic library lookup path.
 130 /// If the dylib_path_var is already set for this cmd, the old value will be overwritten!
 131 pub fn add_dylib_path(cmd: &mut Command, paths: impl Iterator<Item = impl Into<PathBuf>>) {
 132     let path_env = env::var_os(dylib_env_var());
 133     let old_paths = path_env.as_ref().map(env::split_paths);
 134     let new_paths = paths.map(Into::into).chain(old_paths.into_iter().flatten());
 135     cmd.env(dylib_env_var(), env::join_paths(new_paths).unwrap());
 136 }