1 // Copyright 2012-2014 The Rust Project Developers. See the COPYRIGHT
2 // file at the top-level directory of this distribution and at
3 // http://rust-lang.org/COPYRIGHT.
5 // Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
6 // http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
7 // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
8 // option. This file may not be copied, modified, or distributed
9 // except according to those terms.
11 #![allow(non_camel_case_types)]
13 pub use self::FileMatch
::*;
15 use rustc_data_structures
::fx
::FxHashSet
;
19 use std
::path
::{Path, PathBuf}
;
21 use session
::search_paths
::{SearchPaths, PathKind}
;
22 use rustc_fs_util
::fix_windows_verbatim_for_gcc
;
24 #[derive(Copy, Clone)]
30 // A module for searching for libraries
32 pub struct FileSearch
<'a
> {
33 pub sysroot
: &'a Path
,
34 pub search_paths
: &'a SearchPaths
,
39 impl<'a
> FileSearch
<'a
> {
40 pub fn for_each_lib_search_path
<F
>(&self, mut f
: F
) where
41 F
: FnMut(&Path
, PathKind
)
43 let mut visited_dirs
= FxHashSet
::default();
44 visited_dirs
.reserve(self.search_paths
.paths
.len() + 1);
45 for (path
, kind
) in self.search_paths
.iter(self.kind
) {
47 visited_dirs
.insert(path
.to_path_buf());
50 debug
!("filesearch: searching lib path");
51 let tlib_path
= make_target_lib_path(self.sysroot
,
53 if !visited_dirs
.contains(&tlib_path
) {
54 f(&tlib_path
, PathKind
::All
);
57 visited_dirs
.insert(tlib_path
);
60 pub fn get_lib_path(&self) -> PathBuf
{
61 make_target_lib_path(self.sysroot
, self.triple
)
64 pub fn search
<F
>(&self, mut pick
: F
)
65 where F
: FnMut(&Path
, PathKind
) -> FileMatch
67 self.for_each_lib_search_path(|lib_search_path
, kind
| {
68 debug
!("searching {}", lib_search_path
.display());
69 let files
= match fs
::read_dir(lib_search_path
) {
73 let files
= files
.filter_map(|p
| p
.ok().map(|s
| s
.path()))
75 fn is_rlib(p
: &Path
) -> bool
{
76 p
.extension() == Some("rlib".as_ref())
78 // Reading metadata out of rlibs is faster, and if we find both
79 // an rlib and a dylib we only read one of the files of
80 // metadata, so in the name of speed, bring all rlib files to
81 // the front of the search list.
82 let files1
= files
.iter().filter(|p
| is_rlib(p
));
83 let files2
= files
.iter().filter(|p
| !is_rlib(p
));
84 for path
in files1
.chain(files2
) {
85 debug
!("testing {}", path
.display());
86 let maybe_picked
= pick(path
, kind
);
89 debug
!("picked {}", path
.display());
92 debug
!("rejected {}", path
.display());
99 pub fn new(sysroot
: &'a Path
,
101 search_paths
: &'a SearchPaths
,
102 kind
: PathKind
) -> FileSearch
<'a
> {
103 debug
!("using sysroot = {}, triple = {}", sysroot
.display(), triple
);
112 // Returns a list of directories where target-specific dylibs might be located.
113 pub fn get_dylib_search_paths(&self) -> Vec
<PathBuf
> {
114 let mut paths
= Vec
::new();
115 self.for_each_lib_search_path(|lib_search_path
, _
| {
116 paths
.push(lib_search_path
.to_path_buf());
121 // Returns a list of directories where target-specific tool binaries are located.
122 pub fn get_tools_search_paths(&self) -> Vec
<PathBuf
> {
123 let mut p
= PathBuf
::from(self.sysroot
);
124 p
.push(find_libdir(self.sysroot
).as_ref());
125 p
.push(RUST_LIB_DIR
);
126 p
.push(&self.triple
);
132 pub fn relative_target_lib_path(sysroot
: &Path
, target_triple
: &str) -> PathBuf
{
133 let mut p
= PathBuf
::from(find_libdir(sysroot
).as_ref());
134 assert
!(p
.is_relative());
135 p
.push(RUST_LIB_DIR
);
136 p
.push(target_triple
);
141 fn make_target_lib_path(sysroot
: &Path
,
142 target_triple
: &str) -> PathBuf
{
143 sysroot
.join(&relative_target_lib_path(sysroot
, target_triple
))
146 pub fn get_or_default_sysroot() -> PathBuf
{
147 // Follow symlinks. If the resolved path is relative, make it absolute.
148 fn canonicalize(path
: Option
<PathBuf
>) -> Option
<PathBuf
> {
149 path
.and_then(|path
| {
150 match fs
::canonicalize(&path
) {
151 // See comments on this target function, but the gist is that
152 // gcc chokes on verbatim paths which fs::canonicalize generates
153 // so we try to avoid those kinds of paths.
154 Ok(canon
) => Some(fix_windows_verbatim_for_gcc(&canon
)),
155 Err(e
) => bug
!("failed to get realpath: {}", e
),
160 match env
::current_exe() {
162 match canonicalize(Some(exe
)) {
163 Some(mut p
) => { p.pop(); p.pop(); p }
,
164 None
=> bug
!("can't determine value for sysroot")
167 Err(ref e
) => panic
!(format
!("failed to get current_exe: {}", e
))
171 // The name of the directory rustc expects libraries to be located.
172 fn find_libdir(sysroot
: &Path
) -> Cow
<'
static, str> {
173 // FIXME: This is a quick hack to make the rustc binary able to locate
174 // Rust libraries in Linux environments where libraries might be installed
175 // to lib64/lib32. This would be more foolproof by basing the sysroot off
176 // of the directory where librustc is located, rather than where the rustc
178 // If --libdir is set during configuration to the value other than
179 // "lib" (i.e. non-default), this value is used (see issue #16552).
181 #[cfg(target_pointer_width = "64")]
182 const PRIMARY_LIB_DIR
: &'
static str = "lib64";
184 #[cfg(target_pointer_width = "32")]
185 const PRIMARY_LIB_DIR
: &'
static str = "lib32";
187 const SECONDARY_LIB_DIR
: &'
static str = "lib";
189 match option_env
!("CFG_LIBDIR_RELATIVE") {
190 Some(libdir
) if libdir
!= "lib" => libdir
.into(),
191 _
=> if sysroot
.join(PRIMARY_LIB_DIR
).join(RUST_LIB_DIR
).exists() {
192 PRIMARY_LIB_DIR
.into()
194 SECONDARY_LIB_DIR
.into()
199 // The name of rustc's own place to organize libraries.
200 // Used to be "rustc", now the default is "rustlib"
201 const RUST_LIB_DIR
: &'
static str = "rustlib";