compiler/rustc_session/src/search_paths.rs

   1 use crate::filesearch::make_target_lib_path;
   2 use crate::{config, early_error};
   3 use std::path::{Path, PathBuf};
   4
   5 #[derive(Clone, Debug)]
   6 pub struct SearchPath {
   7     pub kind: PathKind,
   8     pub dir: PathBuf,
   9     pub files: Vec<SearchPathFile>,
  10 }
  11
  12 // The obvious implementation of `SearchPath::files` is a `Vec<PathBuf>`. But
  13 // it is searched repeatedly by `find_library_crate`, and the searches involve
  14 // checking the prefix and suffix of the filename of each `PathBuf`. This is
  15 // doable, but very slow, because it involves calls to `file_name` and
  16 // `extension` that are themselves slow.
  17 //
  18 // This type augments the `PathBuf` with an `Option<String>` containing the
  19 // `PathBuf`'s filename. The prefix and suffix checking is much faster on the
  20 // `Option<String>` than the `PathBuf`. (It's an `Option` because
  21 // `Path::file_name` can fail; if that happens then all subsequent checking
  22 // will also fail, which is fine.)
  23 #[derive(Clone, Debug)]
  24 pub struct SearchPathFile {
  25     pub path: PathBuf,
  26     pub file_name_str: Option<String>,
  27 }
  28
  29 impl SearchPathFile {
  30     fn new(path: PathBuf) -> SearchPathFile {
  31         let file_name_str = path.file_name().and_then(|f| f.to_str()).map(|s| s.to_string());
  32         SearchPathFile { path, file_name_str }
  33     }
  34 }
  35
  36 #[derive(PartialEq, Clone, Copy, Debug, Hash, Eq, Encodable, Decodable)]
  37 pub enum PathKind {
  38     Native,
  39     Crate,
  40     Dependency,
  41     Framework,
  42     ExternFlag,
  43     All,
  44 }
  45
  46 rustc_data_structures::impl_stable_hash_via_hash!(PathKind);
  47
  48 impl PathKind {
  49     pub fn matches(&self, kind: PathKind) -> bool {
  50         match (self, kind) {
  51             (PathKind::All, _) | (_, PathKind::All) => true,
  52             _ => *self == kind,
  53         }
  54     }
  55 }
  56
  57 impl SearchPath {
  58     pub fn from_cli_opt(path: &str, output: config::ErrorOutputType) -> Self {
  59         let (kind, path) = if let Some(stripped) = path.strip_prefix("native=") {
  60             (PathKind::Native, stripped)
  61         } else if let Some(stripped) = path.strip_prefix("crate=") {
  62             (PathKind::Crate, stripped)
  63         } else if let Some(stripped) = path.strip_prefix("dependency=") {
  64             (PathKind::Dependency, stripped)
  65         } else if let Some(stripped) = path.strip_prefix("framework=") {
  66             (PathKind::Framework, stripped)
  67         } else if let Some(stripped) = path.strip_prefix("all=") {
  68             (PathKind::All, stripped)
  69         } else {
  70             (PathKind::All, path)
  71         };
  72         if path.is_empty() {
  73             early_error(output, "empty search path given via `-L`");
  74         }
  75
  76         let dir = PathBuf::from(path);
  77         Self::new(kind, dir)
  78     }
  79
  80     pub fn from_sysroot_and_triple(sysroot: &Path, triple: &str) -> Self {
  81         Self::new(PathKind::All, make_target_lib_path(sysroot, triple))
  82     }
  83
  84     fn new(kind: PathKind, dir: PathBuf) -> Self {
  85         // Get the files within the directory.
  86         let files = match std::fs::read_dir(&dir) {
  87             Ok(files) => files
  88                 .filter_map(|e| e.ok().map(|e| SearchPathFile::new(e.path())))
  89                 .collect::<Vec<_>>(),
  90             Err(..) => vec![],
  91         };
  92
  93         SearchPath { kind, dir, files }
  94     }
  95 }