]> git.lizzy.rs Git - rust.git/blob - src/librustc_metadata/locator.rs
Read in rmeta crates
[rust.git] / src / librustc_metadata / locator.rs
1 // Copyright 2012-2015 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.
4 //
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.
10
11 //! Finds crate binaries and loads their metadata
12 //!
13 //! Might I be the first to welcome you to a world of platform differences,
14 //! version requirements, dependency graphs, conflicting desires, and fun! This
15 //! is the major guts (along with metadata::creader) of the compiler for loading
16 //! crates and resolving dependencies. Let's take a tour!
17 //!
18 //! # The problem
19 //!
20 //! Each invocation of the compiler is immediately concerned with one primary
21 //! problem, to connect a set of crates to resolved crates on the filesystem.
22 //! Concretely speaking, the compiler follows roughly these steps to get here:
23 //!
24 //! 1. Discover a set of `extern crate` statements.
25 //! 2. Transform these directives into crate names. If the directive does not
26 //!    have an explicit name, then the identifier is the name.
27 //! 3. For each of these crate names, find a corresponding crate on the
28 //!    filesystem.
29 //!
30 //! Sounds easy, right? Let's walk into some of the nuances.
31 //!
32 //! ## Transitive Dependencies
33 //!
34 //! Let's say we've got three crates: A, B, and C. A depends on B, and B depends
35 //! on C. When we're compiling A, we primarily need to find and locate B, but we
36 //! also end up needing to find and locate C as well.
37 //!
38 //! The reason for this is that any of B's types could be composed of C's types,
39 //! any function in B could return a type from C, etc. To be able to guarantee
40 //! that we can always typecheck/translate any function, we have to have
41 //! complete knowledge of the whole ecosystem, not just our immediate
42 //! dependencies.
43 //!
44 //! So now as part of the "find a corresponding crate on the filesystem" step
45 //! above, this involves also finding all crates for *all upstream
46 //! dependencies*. This includes all dependencies transitively.
47 //!
48 //! ## Rlibs and Dylibs
49 //!
50 //! The compiler has two forms of intermediate dependencies. These are dubbed
51 //! rlibs and dylibs for the static and dynamic variants, respectively. An rlib
52 //! is a rustc-defined file format (currently just an ar archive) while a dylib
53 //! is a platform-defined dynamic library. Each library has a metadata somewhere
54 //! inside of it.
55 //!
56 //! A third kind of dependency is an rmeta file. These are rlibs, which contain
57 //! metadata, but no code. To a first approximation, these are treated in the
58 //! same way as rlibs. Where there is both an rlib and an rmeta file, the rlib
59 //! gets priority (even if the rmeta file is newer). An rmeta file is only
60 //! useful for checking a downstream crate, attempting to link one will cause an
61 //! error.
62 //!
63 //! When translating a crate name to a crate on the filesystem, we all of a
64 //! sudden need to take into account both rlibs and dylibs! Linkage later on may
65 //! use either one of these files, as each has their pros/cons. The job of crate
66 //! loading is to discover what's possible by finding all candidates.
67 //!
68 //! Most parts of this loading systems keep the dylib/rlib as just separate
69 //! variables.
70 //!
71 //! ## Where to look?
72 //!
73 //! We can't exactly scan your whole hard drive when looking for dependencies,
74 //! so we need to places to look. Currently the compiler will implicitly add the
75 //! target lib search path ($prefix/lib/rustlib/$target/lib) to any compilation,
76 //! and otherwise all -L flags are added to the search paths.
77 //!
78 //! ## What criterion to select on?
79 //!
80 //! This a pretty tricky area of loading crates. Given a file, how do we know
81 //! whether it's the right crate? Currently, the rules look along these lines:
82 //!
83 //! 1. Does the filename match an rlib/dylib pattern? That is to say, does the
84 //!    filename have the right prefix/suffix?
85 //! 2. Does the filename have the right prefix for the crate name being queried?
86 //!    This is filtering for files like `libfoo*.rlib` and such.
87 //! 3. Is the file an actual rust library? This is done by loading the metadata
88 //!    from the library and making sure it's actually there.
89 //! 4. Does the name in the metadata agree with the name of the library?
90 //! 5. Does the target in the metadata agree with the current target?
91 //! 6. Does the SVH match? (more on this later)
92 //!
93 //! If the file answers `yes` to all these questions, then the file is
94 //! considered as being *candidate* for being accepted. It is illegal to have
95 //! more than two candidates as the compiler has no method by which to resolve
96 //! this conflict. Additionally, rlib/dylib candidates are considered
97 //! separately.
98 //!
99 //! After all this has happened, we have 1 or two files as candidates. These
100 //! represent the rlib/dylib file found for a library, and they're returned as
101 //! being found.
102 //!
103 //! ### What about versions?
104 //!
105 //! A lot of effort has been put forth to remove versioning from the compiler.
106 //! There have been forays in the past to have versioning baked in, but it was
107 //! largely always deemed insufficient to the point that it was recognized that
108 //! it's probably something the compiler shouldn't do anyway due to its
109 //! complicated nature and the state of the half-baked solutions.
110 //!
111 //! With a departure from versioning, the primary criterion for loading crates
112 //! is just the name of a crate. If we stopped here, it would imply that you
113 //! could never link two crates of the same name from different sources
114 //! together, which is clearly a bad state to be in.
115 //!
116 //! To resolve this problem, we come to the next section!
117 //!
118 //! # Expert Mode
119 //!
120 //! A number of flags have been added to the compiler to solve the "version
121 //! problem" in the previous section, as well as generally enabling more
122 //! powerful usage of the crate loading system of the compiler. The goal of
123 //! these flags and options are to enable third-party tools to drive the
124 //! compiler with prior knowledge about how the world should look.
125 //!
126 //! ## The `--extern` flag
127 //!
128 //! The compiler accepts a flag of this form a number of times:
129 //!
130 //! ```text
131 //! --extern crate-name=path/to/the/crate.rlib
132 //! ```
133 //!
134 //! This flag is basically the following letter to the compiler:
135 //!
136 //! > Dear rustc,
137 //! >
138 //! > When you are attempting to load the immediate dependency `crate-name`, I
139 //! > would like you to assume that the library is located at
140 //! > `path/to/the/crate.rlib`, and look nowhere else. Also, please do not
141 //! > assume that the path I specified has the name `crate-name`.
142 //!
143 //! This flag basically overrides most matching logic except for validating that
144 //! the file is indeed a rust library. The same `crate-name` can be specified
145 //! twice to specify the rlib/dylib pair.
146 //!
147 //! ## Enabling "multiple versions"
148 //!
149 //! This basically boils down to the ability to specify arbitrary packages to
150 //! the compiler. For example, if crate A wanted to use Bv1 and Bv2, then it
151 //! would look something like:
152 //!
153 //! ```ignore
154 //! extern crate b1;
155 //! extern crate b2;
156 //!
157 //! fn main() {}
158 //! ```
159 //!
160 //! and the compiler would be invoked as:
161 //!
162 //! ```text
163 //! rustc a.rs --extern b1=path/to/libb1.rlib --extern b2=path/to/libb2.rlib
164 //! ```
165 //!
166 //! In this scenario there are two crates named `b` and the compiler must be
167 //! manually driven to be informed where each crate is.
168 //!
169 //! ## Frobbing symbols
170 //!
171 //! One of the immediate problems with linking the same library together twice
172 //! in the same problem is dealing with duplicate symbols. The primary way to
173 //! deal with this in rustc is to add hashes to the end of each symbol.
174 //!
175 //! In order to force hashes to change between versions of a library, if
176 //! desired, the compiler exposes an option `-C metadata=foo`, which is used to
177 //! initially seed each symbol hash. The string `foo` is prepended to each
178 //! string-to-hash to ensure that symbols change over time.
179 //!
180 //! ## Loading transitive dependencies
181 //!
182 //! Dealing with same-named-but-distinct crates is not just a local problem, but
183 //! one that also needs to be dealt with for transitive dependencies. Note that
184 //! in the letter above `--extern` flags only apply to the *local* set of
185 //! dependencies, not the upstream transitive dependencies. Consider this
186 //! dependency graph:
187 //!
188 //! ```text
189 //! A.1   A.2
190 //! |     |
191 //! |     |
192 //! B     C
193 //!  \   /
194 //!   \ /
195 //!    D
196 //! ```
197 //!
198 //! In this scenario, when we compile `D`, we need to be able to distinctly
199 //! resolve `A.1` and `A.2`, but an `--extern` flag cannot apply to these
200 //! transitive dependencies.
201 //!
202 //! Note that the key idea here is that `B` and `C` are both *already compiled*.
203 //! That is, they have already resolved their dependencies. Due to unrelated
204 //! technical reasons, when a library is compiled, it is only compatible with
205 //! the *exact same* version of the upstream libraries it was compiled against.
206 //! We use the "Strict Version Hash" to identify the exact copy of an upstream
207 //! library.
208 //!
209 //! With this knowledge, we know that `B` and `C` will depend on `A` with
210 //! different SVH values, so we crawl the normal `-L` paths looking for
211 //! `liba*.rlib` and filter based on the contained SVH.
212 //!
213 //! In the end, this ends up not needing `--extern` to specify upstream
214 //! transitive dependencies.
215 //!
216 //! # Wrapping up
217 //!
218 //! That's the general overview of loading crates in the compiler, but it's by
219 //! no means all of the necessary details. Take a look at the rest of
220 //! metadata::locator or metadata::creader for all the juicy details!
221
222 use cstore::MetadataBlob;
223 use creader::Library;
224 use schema::{METADATA_HEADER, rustc_version};
225
226 use rustc::hir::svh::Svh;
227 use rustc::session::{config, Session};
228 use rustc::session::filesearch::{FileSearch, FileMatches, FileDoesntMatch};
229 use rustc::session::search_paths::PathKind;
230 use rustc::util::common;
231 use rustc::util::nodemap::FxHashMap;
232
233 use rustc_llvm as llvm;
234 use rustc_llvm::{False, ObjectFile, mk_section_iter};
235 use rustc_llvm::archive_ro::ArchiveRO;
236 use errors::DiagnosticBuilder;
237 use syntax_pos::Span;
238 use rustc_back::target::Target;
239
240 use std::cmp;
241 use std::fmt;
242 use std::fs;
243 use std::io;
244 use std::path::{Path, PathBuf};
245 use std::ptr;
246 use std::slice;
247 use std::time::Instant;
248
249 use flate;
250
251 pub struct CrateMismatch {
252     path: PathBuf,
253     got: String,
254 }
255
256 pub struct Context<'a> {
257     pub sess: &'a Session,
258     pub span: Span,
259     pub ident: &'a str,
260     pub crate_name: &'a str,
261     pub hash: Option<&'a Svh>,
262     // points to either self.sess.target.target or self.sess.host, must match triple
263     pub target: &'a Target,
264     pub triple: &'a str,
265     pub filesearch: FileSearch<'a>,
266     pub root: &'a Option<CratePaths>,
267     pub rejected_via_hash: Vec<CrateMismatch>,
268     pub rejected_via_triple: Vec<CrateMismatch>,
269     pub rejected_via_kind: Vec<CrateMismatch>,
270     pub rejected_via_version: Vec<CrateMismatch>,
271     pub should_match_name: bool,
272     pub is_proc_macro: Option<bool>,
273 }
274
275 pub struct ArchiveMetadata {
276     _archive: ArchiveRO,
277     // points into self._archive
278     data: *const [u8],
279 }
280
281 pub struct CratePaths {
282     pub ident: String,
283     pub dylib: Option<PathBuf>,
284     pub rlib: Option<PathBuf>,
285     pub rmeta: Option<PathBuf>,
286 }
287
288 pub const METADATA_FILENAME: &'static str = "rust.metadata.bin";
289
290 #[derive(Copy, Clone, PartialEq)]
291 enum CrateFlavor {
292     Rlib,
293     Rmeta,
294     Dylib,
295 }
296
297 impl fmt::Display for CrateFlavor {
298     fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
299         f.write_str(match *self {
300             CrateFlavor::Rlib => "rlib",
301             CrateFlavor::Rmeta => "rmeta",
302             CrateFlavor::Dylib => "dylib",
303         })
304     }
305 }
306
307 impl CratePaths {
308     fn paths(&self) -> Vec<PathBuf> {
309         self.dylib.iter().chain(self.rlib.iter()).chain(self.rmeta.iter()).cloned().collect()
310     }
311 }
312
313 impl<'a> Context<'a> {
314     pub fn maybe_load_library_crate(&mut self) -> Option<Library> {
315         self.find_library_crate()
316     }
317
318     pub fn load_library_crate(&mut self) -> Library {
319         self.find_library_crate().unwrap_or_else(|| self.report_errs())
320     }
321
322     pub fn report_errs(&mut self) -> ! {
323         let add = match self.root {
324             &None => String::new(),
325             &Some(ref r) => format!(" which `{}` depends on", r.ident),
326         };
327         let mut err = if !self.rejected_via_hash.is_empty() {
328             struct_span_err!(self.sess,
329                              self.span,
330                              E0460,
331                              "found possibly newer version of crate `{}`{}",
332                              self.ident,
333                              add)
334         } else if !self.rejected_via_triple.is_empty() {
335             struct_span_err!(self.sess,
336                              self.span,
337                              E0461,
338                              "couldn't find crate `{}` with expected target triple {}{}",
339                              self.ident,
340                              self.triple,
341                              add)
342         } else if !self.rejected_via_kind.is_empty() {
343             struct_span_err!(self.sess,
344                              self.span,
345                              E0462,
346                              "found staticlib `{}` instead of rlib or dylib{}",
347                              self.ident,
348                              add)
349         } else if !self.rejected_via_version.is_empty() {
350             struct_span_err!(self.sess,
351                              self.span,
352                              E0514,
353                              "found crate `{}` compiled by an incompatible version of rustc{}",
354                              self.ident,
355                              add)
356         } else {
357             let mut err = struct_span_err!(self.sess,
358                                            self.span,
359                                            E0463,
360                                            "can't find crate for `{}`{}",
361                                            self.ident,
362                                            add);
363
364             if (self.ident == "std" || self.ident == "core")
365                 && self.triple != config::host_triple() {
366                 err.note(&format!("the `{}` target may not be installed", self.triple));
367             }
368             err.span_label(self.span, &format!("can't find crate"));
369             err
370         };
371
372         if !self.rejected_via_triple.is_empty() {
373             let mismatches = self.rejected_via_triple.iter();
374             for (i, &CrateMismatch { ref path, ref got }) in mismatches.enumerate() {
375                 err.note(&format!("crate `{}`, path #{}, triple {}: {}",
376                                   self.ident,
377                                   i + 1,
378                                   got,
379                                   path.display()));
380             }
381         }
382         if !self.rejected_via_hash.is_empty() {
383             err.note("perhaps that crate needs to be recompiled?");
384             let mismatches = self.rejected_via_hash.iter();
385             for (i, &CrateMismatch { ref path, .. }) in mismatches.enumerate() {
386                 err.note(&format!("crate `{}` path #{}: {}", self.ident, i + 1, path.display()));
387             }
388             match self.root {
389                 &None => {}
390                 &Some(ref r) => {
391                     for (i, path) in r.paths().iter().enumerate() {
392                         err.note(&format!("crate `{}` path #{}: {}",
393                                           r.ident,
394                                           i + 1,
395                                           path.display()));
396                     }
397                 }
398             }
399         }
400         if !self.rejected_via_kind.is_empty() {
401             err.help("please recompile that crate using --crate-type lib");
402             let mismatches = self.rejected_via_kind.iter();
403             for (i, &CrateMismatch { ref path, .. }) in mismatches.enumerate() {
404                 err.note(&format!("crate `{}` path #{}: {}", self.ident, i + 1, path.display()));
405             }
406         }
407         if !self.rejected_via_version.is_empty() {
408             err.help(&format!("please recompile that crate using this compiler ({})",
409                               rustc_version()));
410             let mismatches = self.rejected_via_version.iter();
411             for (i, &CrateMismatch { ref path, ref got }) in mismatches.enumerate() {
412                 err.note(&format!("crate `{}` path #{}: {} compiled by {:?}",
413                                   self.ident,
414                                   i + 1,
415                                   path.display(),
416                                   got));
417             }
418         }
419
420         err.emit();
421         self.sess.abort_if_errors();
422         unreachable!();
423     }
424
425     fn find_library_crate(&mut self) -> Option<Library> {
426         // If an SVH is specified, then this is a transitive dependency that
427         // must be loaded via -L plus some filtering.
428         if self.hash.is_none() {
429             self.should_match_name = false;
430             if let Some(s) = self.sess.opts.externs.get(self.crate_name) {
431                 return self.find_commandline_library(s.iter());
432             }
433             self.should_match_name = true;
434         }
435
436         let dypair = self.dylibname();
437         let staticpair = self.staticlibname();
438
439         // want: crate_name.dir_part() + prefix + crate_name.file_part + "-"
440         let dylib_prefix = format!("{}{}", dypair.0, self.crate_name);
441         let rlib_prefix = format!("lib{}", self.crate_name);
442         let staticlib_prefix = format!("{}{}", staticpair.0, self.crate_name);
443
444         let mut candidates = FxHashMap();
445         let mut staticlibs = vec![];
446
447         // First, find all possible candidate rlibs and dylibs purely based on
448         // the name of the files themselves. We're trying to match against an
449         // exact crate name and a possibly an exact hash.
450         //
451         // During this step, we can filter all found libraries based on the
452         // name and id found in the crate id (we ignore the path portion for
453         // filename matching), as well as the exact hash (if specified). If we
454         // end up having many candidates, we must look at the metadata to
455         // perform exact matches against hashes/crate ids. Note that opening up
456         // the metadata is where we do an exact match against the full contents
457         // of the crate id (path/name/id).
458         //
459         // The goal of this step is to look at as little metadata as possible.
460         self.filesearch.search(|path, kind| {
461             let file = match path.file_name().and_then(|s| s.to_str()) {
462                 None => return FileDoesntMatch,
463                 Some(file) => file,
464             };
465             let (hash, found_kind) = if file.starts_with(&rlib_prefix[..]) && file.ends_with(".rlib") {
466                 (&file[(rlib_prefix.len())..(file.len() - ".rlib".len())], CrateFlavor::Rlib)
467             } else if file.starts_with(&rlib_prefix[..]) && file.ends_with(".rmeta") {
468                 (&file[(rlib_prefix.len())..(file.len() - ".rmeta".len())], CrateFlavor::Rmeta)
469             } else if file.starts_with(&dylib_prefix) &&
470                                          file.ends_with(&dypair.1) {
471                 (&file[(dylib_prefix.len())..(file.len() - dypair.1.len())], CrateFlavor::Dylib)
472             } else {
473                 if file.starts_with(&staticlib_prefix[..]) && file.ends_with(&staticpair.1) {
474                     staticlibs.push(CrateMismatch {
475                         path: path.to_path_buf(),
476                         got: "static".to_string(),
477                     });
478                 }
479                 return FileDoesntMatch;
480             };
481             info!("lib candidate: {}", path.display());
482
483             let hash_str = hash.to_string();
484             let slot = candidates.entry(hash_str)
485                 .or_insert_with(|| (FxHashMap(), FxHashMap(), FxHashMap()));
486             let (ref mut rlibs, ref mut rmetas, ref mut dylibs) = *slot;
487             fs::canonicalize(path)
488                 .map(|p| {
489                     match found_kind {
490                         CrateFlavor::Rlib => { rlibs.insert(p, kind); }
491                         CrateFlavor::Rmeta => { rmetas.insert(p, kind); }
492                         CrateFlavor::Dylib => { dylibs.insert(p, kind); }
493                     }
494                     FileMatches
495                 })
496                 .unwrap_or(FileDoesntMatch)
497         });
498         self.rejected_via_kind.extend(staticlibs);
499
500         // We have now collected all known libraries into a set of candidates
501         // keyed of the filename hash listed. For each filename, we also have a
502         // list of rlibs/dylibs that apply. Here, we map each of these lists
503         // (per hash), to a Library candidate for returning.
504         //
505         // A Library candidate is created if the metadata for the set of
506         // libraries corresponds to the crate id and hash criteria that this
507         // search is being performed for.
508         let mut libraries = FxHashMap();
509         for (_hash, (rlibs, rmetas, dylibs)) in candidates {
510             let mut slot = None;
511             let rlib = self.extract_one(rlibs, CrateFlavor::Rlib, &mut slot);
512             let rmeta = self.extract_one(rmetas, CrateFlavor::Rmeta, &mut slot);
513             let dylib = self.extract_one(dylibs, CrateFlavor::Dylib, &mut slot);
514             if let Some((h, m)) = slot {
515                 libraries.insert(h,
516                                  Library {
517                                      dylib: dylib,
518                                      rlib: rlib,
519                                      rmeta: rmeta,
520                                      metadata: m,
521                                  });
522             }
523         }
524
525         // Having now translated all relevant found hashes into libraries, see
526         // what we've got and figure out if we found multiple candidates for
527         // libraries or not.
528         match libraries.len() {
529             0 => None,
530             1 => Some(libraries.into_iter().next().unwrap().1),
531             _ => {
532                 let mut err = struct_span_err!(self.sess,
533                                                self.span,
534                                                E0464,
535                                                "multiple matching crates for `{}`",
536                                                self.crate_name);
537                 err.note("candidates:");
538                 for (_, lib) in libraries {
539                     if let Some((ref p, _)) = lib.dylib {
540                         err.note(&format!("path: {}", p.display()));
541                     }
542                     if let Some((ref p, _)) = lib.rlib {
543                         err.note(&format!("path: {}", p.display()));
544                     }
545                     note_crate_name(&mut err, &lib.metadata.get_root().name);
546                 }
547                 err.emit();
548                 None
549             }
550         }
551     }
552
553     // Attempts to extract *one* library from the set `m`. If the set has no
554     // elements, `None` is returned. If the set has more than one element, then
555     // the errors and notes are emitted about the set of libraries.
556     //
557     // With only one library in the set, this function will extract it, and then
558     // read the metadata from it if `*slot` is `None`. If the metadata couldn't
559     // be read, it is assumed that the file isn't a valid rust library (no
560     // errors are emitted).
561     fn extract_one(&mut self,
562                    m: FxHashMap<PathBuf, PathKind>,
563                    flavor: CrateFlavor,
564                    slot: &mut Option<(Svh, MetadataBlob)>)
565                    -> Option<(PathBuf, PathKind)> {
566         let mut ret: Option<(PathBuf, PathKind)> = None;
567         let mut error = 0;
568
569         if slot.is_some() {
570             // FIXME(#10786): for an optimization, we only read one of the
571             //                libraries' metadata sections. In theory we should
572             //                read both, but reading dylib metadata is quite
573             //                slow.
574             if m.is_empty() {
575                 return None;
576             } else if m.len() == 1 {
577                 return Some(m.into_iter().next().unwrap());
578             }
579         }
580
581         let mut err: Option<DiagnosticBuilder> = None;
582         for (lib, kind) in m {
583             info!("{} reading metadata from: {}", flavor, lib.display());
584             let (hash, metadata) = match get_metadata_section(self.target, flavor, &lib) {
585                 Ok(blob) => {
586                     if let Some(h) = self.crate_matches(&blob, &lib) {
587                         (h, blob)
588                     } else {
589                         info!("metadata mismatch");
590                         continue;
591                     }
592                 }
593                 Err(err) => {
594                     info!("no metadata found: {}", err);
595                     continue;
596                 }
597             };
598             // If we see multiple hashes, emit an error about duplicate candidates.
599             if slot.as_ref().map_or(false, |s| s.0 != hash) {
600                 let mut e = struct_span_err!(self.sess,
601                                              self.span,
602                                              E0465,
603                                              "multiple {} candidates for `{}` found",
604                                              flavor,
605                                              self.crate_name);
606                 e.span_note(self.span,
607                             &format!(r"candidate #1: {}",
608                                      ret.as_ref()
609                                          .unwrap()
610                                          .0
611                                          .display()));
612                 if let Some(ref mut e) = err {
613                     e.emit();
614                 }
615                 err = Some(e);
616                 error = 1;
617                 *slot = None;
618             }
619             if error > 0 {
620                 error += 1;
621                 err.as_mut().unwrap().span_note(self.span,
622                                                 &format!(r"candidate #{}: {}",
623                                                          error,
624                                                          lib.display()));
625                 continue;
626             }
627             *slot = Some((hash, metadata));
628             ret = Some((lib, kind));
629         }
630
631         if error > 0 {
632             err.unwrap().emit();
633             None
634         } else {
635             ret
636         }
637     }
638
639     fn crate_matches(&mut self, metadata: &MetadataBlob, libpath: &Path) -> Option<Svh> {
640         let root = metadata.get_root();
641         if let Some(is_proc_macro) = self.is_proc_macro {
642             if root.macro_derive_registrar.is_some() != is_proc_macro {
643                 return None;
644             }
645         }
646
647         let rustc_version = rustc_version();
648         if root.rustc_version != rustc_version {
649             info!("Rejecting via version: expected {} got {}",
650                   rustc_version,
651                   root.rustc_version);
652             self.rejected_via_version.push(CrateMismatch {
653                 path: libpath.to_path_buf(),
654                 got: root.rustc_version,
655             });
656             return None;
657         }
658
659         if self.should_match_name {
660             if self.crate_name != root.name {
661                 info!("Rejecting via crate name");
662                 return None;
663             }
664         }
665
666         if root.triple != self.triple {
667             info!("Rejecting via crate triple: expected {} got {}",
668                   self.triple,
669                   root.triple);
670             self.rejected_via_triple.push(CrateMismatch {
671                 path: libpath.to_path_buf(),
672                 got: root.triple,
673             });
674             return None;
675         }
676
677         if let Some(myhash) = self.hash {
678             if *myhash != root.hash {
679                 info!("Rejecting via hash: expected {} got {}", *myhash, root.hash);
680                 self.rejected_via_hash.push(CrateMismatch {
681                     path: libpath.to_path_buf(),
682                     got: myhash.to_string(),
683                 });
684                 return None;
685             }
686         }
687
688         Some(root.hash)
689     }
690
691
692     // Returns the corresponding (prefix, suffix) that files need to have for
693     // dynamic libraries
694     fn dylibname(&self) -> (String, String) {
695         let t = &self.target;
696         (t.options.dll_prefix.clone(), t.options.dll_suffix.clone())
697     }
698
699     // Returns the corresponding (prefix, suffix) that files need to have for
700     // static libraries
701     fn staticlibname(&self) -> (String, String) {
702         let t = &self.target;
703         (t.options.staticlib_prefix.clone(), t.options.staticlib_suffix.clone())
704     }
705
706     fn find_commandline_library<'b, LOCS>(&mut self, locs: LOCS) -> Option<Library>
707         where LOCS: Iterator<Item = &'b String>
708     {
709         // First, filter out all libraries that look suspicious. We only accept
710         // files which actually exist that have the correct naming scheme for
711         // rlibs/dylibs.
712         let sess = self.sess;
713         let dylibname = self.dylibname();
714         let mut rlibs = FxHashMap();
715         let mut rmetas = FxHashMap();
716         let mut dylibs = FxHashMap();
717         {
718             let locs = locs.map(|l| PathBuf::from(l)).filter(|loc| {
719                 if !loc.exists() {
720                     sess.err(&format!("extern location for {} does not exist: {}",
721                                       self.crate_name,
722                                       loc.display()));
723                     return false;
724                 }
725                 let file = match loc.file_name().and_then(|s| s.to_str()) {
726                     Some(file) => file,
727                     None => {
728                         sess.err(&format!("extern location for {} is not a file: {}",
729                                           self.crate_name,
730                                           loc.display()));
731                         return false;
732                     }
733                 };
734                 if file.starts_with("lib") && file.ends_with(".rlib") {
735                     return true;
736                 } else {
737                     let (ref prefix, ref suffix) = dylibname;
738                     if file.starts_with(&prefix[..]) && file.ends_with(&suffix[..]) {
739                         return true;
740                     }
741                 }
742                 sess.struct_err(&format!("extern location for {} is of an unknown type: {}",
743                                          self.crate_name,
744                                          loc.display()))
745                     .help(&format!("file name should be lib*.rlib or {}*.{}",
746                                    dylibname.0,
747                                    dylibname.1))
748                     .emit();
749                 false
750             });
751
752             // Now that we have an iterator of good candidates, make sure
753             // there's at most one rlib and at most one dylib.
754             for loc in locs {
755                 if loc.file_name().unwrap().to_str().unwrap().ends_with(".rlib") {
756                     rlibs.insert(fs::canonicalize(&loc).unwrap(), PathKind::ExternFlag);
757                 } else if loc.file_name().unwrap().to_str().unwrap().ends_with(".rmeta") {
758                     rmetas.insert(fs::canonicalize(&loc).unwrap(), PathKind::ExternFlag);
759                 } else {
760                     dylibs.insert(fs::canonicalize(&loc).unwrap(), PathKind::ExternFlag);
761                 }
762             }
763         };
764
765         // Extract the rlib/dylib pair.
766         let mut slot = None;
767         let rlib = self.extract_one(rlibs, CrateFlavor::Rlib, &mut slot);
768         let rmeta = self.extract_one(rmetas, CrateFlavor::Rmeta, &mut slot);
769         let dylib = self.extract_one(dylibs, CrateFlavor::Dylib, &mut slot);
770
771         if rlib.is_none() && rmeta.is_none() && dylib.is_none() {
772             return None;
773         }
774         match slot {
775             Some((_, metadata)) => {
776                 Some(Library {
777                     dylib: dylib,
778                     rlib: rlib,
779                     rmeta: rmeta,
780                     metadata: metadata,
781                 })
782             }
783             None => None,
784         }
785     }
786 }
787
788 pub fn note_crate_name(err: &mut DiagnosticBuilder, name: &str) {
789     err.note(&format!("crate name: {}", name));
790 }
791
792 impl ArchiveMetadata {
793     fn new(ar: ArchiveRO) -> Option<ArchiveMetadata> {
794         let data = {
795             let section = ar.iter()
796                 .filter_map(|s| s.ok())
797                 .find(|sect| sect.name() == Some(METADATA_FILENAME));
798             match section {
799                 Some(s) => s.data() as *const [u8],
800                 None => {
801                     debug!("didn't find '{}' in the archive", METADATA_FILENAME);
802                     return None;
803                 }
804             }
805         };
806
807         Some(ArchiveMetadata {
808             _archive: ar,
809             data: data,
810         })
811     }
812
813     pub fn as_slice<'a>(&'a self) -> &'a [u8] {
814         unsafe { &*self.data }
815     }
816 }
817
818 fn verify_decompressed_encoding_version(blob: &MetadataBlob,
819                                         filename: &Path)
820                                         -> Result<(), String> {
821     if !blob.is_compatible() {
822         Err((format!("incompatible metadata version found: '{}'",
823                      filename.display())))
824     } else {
825         Ok(())
826     }
827 }
828
829 // Just a small wrapper to time how long reading metadata takes.
830 fn get_metadata_section(target: &Target,
831                         flavor: CrateFlavor,
832                         filename: &Path)
833                         -> Result<MetadataBlob, String> {
834     let start = Instant::now();
835     let ret = get_metadata_section_imp(target, flavor, filename);
836     info!("reading {:?} => {:?}",
837           filename.file_name().unwrap(),
838           start.elapsed());
839     return ret;
840 }
841
842 fn get_metadata_section_imp(target: &Target,
843                             flavor: CrateFlavor,
844                             filename: &Path)
845                             -> Result<MetadataBlob, String> {
846     if !filename.exists() {
847         return Err(format!("no such file: '{}'", filename.display()));
848     }
849     if flavor == CrateFlavor::Rlib || flavor == CrateFlavor::Rmeta {
850         // Use ArchiveRO for speed here, it's backed by LLVM and uses mmap
851         // internally to read the file. We also avoid even using a memcpy by
852         // just keeping the archive along while the metadata is in use.
853         let archive = match ArchiveRO::open(filename) {
854             Some(ar) => ar,
855             None => {
856                 debug!("llvm didn't like `{}`", filename.display());
857                 return Err(format!("failed to read rlib metadata: '{}'", filename.display()));
858             }
859         };
860         return match ArchiveMetadata::new(archive).map(|ar| MetadataBlob::Archive(ar)) {
861             None => Err(format!("failed to read rlib metadata: '{}'", filename.display())),
862             Some(blob) => {
863                 verify_decompressed_encoding_version(&blob, filename)?;
864                 Ok(blob)
865             }
866         };
867     }
868     unsafe {
869         let buf = common::path2cstr(filename);
870         let mb = llvm::LLVMRustCreateMemoryBufferWithContentsOfFile(buf.as_ptr());
871         if mb as isize == 0 {
872             return Err(format!("error reading library: '{}'", filename.display()));
873         }
874         let of = match ObjectFile::new(mb) {
875             Some(of) => of,
876             _ => {
877                 return Err((format!("provided path not an object file: '{}'", filename.display())))
878             }
879         };
880         let si = mk_section_iter(of.llof);
881         while llvm::LLVMIsSectionIteratorAtEnd(of.llof, si.llsi) == False {
882             let mut name_buf = ptr::null();
883             let name_len = llvm::LLVMRustGetSectionName(si.llsi, &mut name_buf);
884             let name = slice::from_raw_parts(name_buf as *const u8, name_len as usize).to_vec();
885             let name = String::from_utf8(name).unwrap();
886             debug!("get_metadata_section: name {}", name);
887             if read_meta_section_name(target) == name {
888                 let cbuf = llvm::LLVMGetSectionContents(si.llsi);
889                 let csz = llvm::LLVMGetSectionSize(si.llsi) as usize;
890                 let cvbuf: *const u8 = cbuf as *const u8;
891                 let vlen = METADATA_HEADER.len();
892                 debug!("checking {} bytes of metadata-version stamp", vlen);
893                 let minsz = cmp::min(vlen, csz);
894                 let buf0 = slice::from_raw_parts(cvbuf, minsz);
895                 let version_ok = buf0 == METADATA_HEADER;
896                 if !version_ok {
897                     return Err((format!("incompatible metadata version found: '{}'",
898                                         filename.display())));
899                 }
900
901                 let cvbuf1 = cvbuf.offset(vlen as isize);
902                 debug!("inflating {} bytes of compressed metadata", csz - vlen);
903                 let bytes = slice::from_raw_parts(cvbuf1, csz - vlen);
904                 match flate::inflate_bytes(bytes) {
905                     Ok(inflated) => {
906                         let blob = MetadataBlob::Inflated(inflated);
907                         verify_decompressed_encoding_version(&blob, filename)?;
908                         return Ok(blob);
909                     }
910                     Err(_) => {}
911                 }
912             }
913             llvm::LLVMMoveToNextSection(si.llsi);
914         }
915         Err(format!("metadata not found: '{}'", filename.display()))
916     }
917 }
918
919 pub fn meta_section_name(target: &Target) -> &'static str {
920     // Historical note:
921     //
922     // When using link.exe it was seen that the section name `.note.rustc`
923     // was getting shortened to `.note.ru`, and according to the PE and COFF
924     // specification:
925     //
926     // > Executable images do not use a string table and do not support
927     // > section names longer than 8 characters
928     //
929     // https://msdn.microsoft.com/en-us/library/windows/hardware/gg463119.aspx
930     //
931     // As a result, we choose a slightly shorter name! As to why
932     // `.note.rustc` works on MinGW, that's another good question...
933
934     if target.options.is_like_osx {
935         "__DATA,.rustc"
936     } else {
937         ".rustc"
938     }
939 }
940
941 pub fn read_meta_section_name(_target: &Target) -> &'static str {
942     ".rustc"
943 }
944
945 // A diagnostic function for dumping crate metadata to an output stream
946 pub fn list_file_metadata(target: &Target, path: &Path, out: &mut io::Write) -> io::Result<()> {
947     let filename = path.file_name().unwrap().to_str().unwrap();
948     let flavor = if filename.ends_with(".rlib") {
949         CrateFlavor::Rlib
950     } else if filename.ends_with(".rmeta") {
951         CrateFlavor::Rmeta
952     } else {
953         CrateFlavor::Dylib
954     };
955     match get_metadata_section(target, flavor, path) {
956         Ok(metadata) => metadata.list_crate_metadata(out),
957         Err(msg) => write!(out, "{}\n", msg),
958     }
959 }