]> git.lizzy.rs Git - rust.git/blob - src/librustc_metadata/creader.rs
When encountering an Item in a pat context, point at the item def
[rust.git] / src / librustc_metadata / creader.rs
1 //! Validates all used crates and extern libraries and loads their metadata
2
3 use crate::locator::{CrateLocator, CratePaths};
4 use crate::rmeta::{CrateDep, CrateMetadata, CrateNumMap, CrateRoot, MetadataBlob};
5
6 use rustc::hir::map::Definitions;
7 use rustc::middle::cstore::DepKind;
8 use rustc::middle::cstore::{CrateSource, ExternCrate, ExternCrateSource, MetadataLoaderDyn};
9 use rustc::session::config;
10 use rustc::session::search_paths::PathKind;
11 use rustc::session::{CrateDisambiguator, Session};
12 use rustc::ty::TyCtxt;
13 use rustc_ast::ast;
14 use rustc_ast::attr;
15 use rustc_ast::expand::allocator::{global_allocator_spans, AllocatorKind};
16 use rustc_data_structures::svh::Svh;
17 use rustc_data_structures::sync::Lrc;
18 use rustc_errors::struct_span_err;
19 use rustc_expand::base::SyntaxExtension;
20 use rustc_hir::def_id::{CrateNum, LOCAL_CRATE};
21 use rustc_index::vec::IndexVec;
22 use rustc_span::edition::Edition;
23 use rustc_span::symbol::{sym, Symbol};
24 use rustc_span::{Span, DUMMY_SP};
25 use rustc_target::spec::{PanicStrategy, TargetTriple};
26
27 use log::{debug, info, log_enabled};
28 use proc_macro::bridge::client::ProcMacro;
29 use std::path::Path;
30 use std::{cmp, fs};
31
32 #[derive(Clone)]
33 pub struct CStore {
34     metas: IndexVec<CrateNum, Option<Lrc<CrateMetadata>>>,
35     injected_panic_runtime: Option<CrateNum>,
36     /// This crate needs an allocator and either provides it itself, or finds it in a dependency.
37     /// If the above is true, then this field denotes the kind of the found allocator.
38     allocator_kind: Option<AllocatorKind>,
39     /// This crate has a `#[global_allocator]` item.
40     has_global_allocator: bool,
41 }
42
43 pub struct CrateLoader<'a> {
44     // Immutable configuration.
45     sess: &'a Session,
46     metadata_loader: &'a MetadataLoaderDyn,
47     local_crate_name: Symbol,
48     // Mutable output.
49     cstore: CStore,
50 }
51
52 pub enum LoadedMacro {
53     MacroDef(ast::Item, Edition),
54     ProcMacro(SyntaxExtension),
55 }
56
57 crate struct Library {
58     pub source: CrateSource,
59     pub metadata: MetadataBlob,
60 }
61
62 enum LoadResult {
63     Previous(CrateNum),
64     Loaded(Library),
65 }
66
67 enum LoadError<'a> {
68     LocatorError(CrateLocator<'a>),
69 }
70
71 impl<'a> LoadError<'a> {
72     fn report(self) -> ! {
73         match self {
74             LoadError::LocatorError(locator) => locator.report_errs(),
75         }
76     }
77 }
78
79 fn dump_crates(cstore: &CStore) {
80     info!("resolved crates:");
81     cstore.iter_crate_data(|cnum, data| {
82         info!("  name: {}", data.name());
83         info!("  cnum: {}", cnum);
84         info!("  hash: {}", data.hash());
85         info!("  reqd: {:?}", data.dep_kind());
86         let CrateSource { dylib, rlib, rmeta } = data.source();
87         dylib.as_ref().map(|dl| info!("  dylib: {}", dl.0.display()));
88         rlib.as_ref().map(|rl| info!("   rlib: {}", rl.0.display()));
89         rmeta.as_ref().map(|rl| info!("   rmeta: {}", rl.0.display()));
90     });
91 }
92
93 impl CStore {
94     crate fn from_tcx(tcx: TyCtxt<'_>) -> &CStore {
95         tcx.cstore_as_any().downcast_ref::<CStore>().expect("`tcx.cstore` is not a `CStore`")
96     }
97
98     fn alloc_new_crate_num(&mut self) -> CrateNum {
99         self.metas.push(None);
100         CrateNum::new(self.metas.len() - 1)
101     }
102
103     crate fn get_crate_data(&self, cnum: CrateNum) -> &CrateMetadata {
104         self.metas[cnum]
105             .as_ref()
106             .unwrap_or_else(|| panic!("Failed to get crate data for {:?}", cnum))
107     }
108
109     fn set_crate_data(&mut self, cnum: CrateNum, data: CrateMetadata) {
110         assert!(self.metas[cnum].is_none(), "Overwriting crate metadata entry");
111         self.metas[cnum] = Some(Lrc::new(data));
112     }
113
114     crate fn iter_crate_data(&self, mut f: impl FnMut(CrateNum, &CrateMetadata)) {
115         for (cnum, data) in self.metas.iter_enumerated() {
116             if let Some(data) = data {
117                 f(cnum, data);
118             }
119         }
120     }
121
122     fn push_dependencies_in_postorder(&self, deps: &mut Vec<CrateNum>, cnum: CrateNum) {
123         if !deps.contains(&cnum) {
124             let data = self.get_crate_data(cnum);
125             for &dep in data.dependencies().iter() {
126                 if dep != cnum {
127                     self.push_dependencies_in_postorder(deps, dep);
128                 }
129             }
130
131             deps.push(cnum);
132         }
133     }
134
135     crate fn crate_dependencies_in_postorder(&self, cnum: CrateNum) -> Vec<CrateNum> {
136         let mut deps = Vec::new();
137         if cnum == LOCAL_CRATE {
138             self.iter_crate_data(|cnum, _| self.push_dependencies_in_postorder(&mut deps, cnum));
139         } else {
140             self.push_dependencies_in_postorder(&mut deps, cnum);
141         }
142         deps
143     }
144
145     fn crate_dependencies_in_reverse_postorder(&self, cnum: CrateNum) -> Vec<CrateNum> {
146         let mut deps = self.crate_dependencies_in_postorder(cnum);
147         deps.reverse();
148         deps
149     }
150
151     crate fn injected_panic_runtime(&self) -> Option<CrateNum> {
152         self.injected_panic_runtime
153     }
154
155     crate fn allocator_kind(&self) -> Option<AllocatorKind> {
156         self.allocator_kind
157     }
158
159     crate fn has_global_allocator(&self) -> bool {
160         self.has_global_allocator
161     }
162 }
163
164 impl<'a> CrateLoader<'a> {
165     pub fn new(
166         sess: &'a Session,
167         metadata_loader: &'a MetadataLoaderDyn,
168         local_crate_name: &str,
169     ) -> Self {
170         CrateLoader {
171             sess,
172             metadata_loader,
173             local_crate_name: Symbol::intern(local_crate_name),
174             cstore: CStore {
175                 // We add an empty entry for LOCAL_CRATE (which maps to zero) in
176                 // order to make array indices in `metas` match with the
177                 // corresponding `CrateNum`. This first entry will always remain
178                 // `None`.
179                 metas: IndexVec::from_elem_n(None, 1),
180                 injected_panic_runtime: None,
181                 allocator_kind: None,
182                 has_global_allocator: false,
183             },
184         }
185     }
186
187     pub fn cstore(&self) -> &CStore {
188         &self.cstore
189     }
190
191     pub fn into_cstore(self) -> CStore {
192         self.cstore
193     }
194
195     fn existing_match(&self, name: Symbol, hash: Option<Svh>, kind: PathKind) -> Option<CrateNum> {
196         let mut ret = None;
197         self.cstore.iter_crate_data(|cnum, data| {
198             if data.name() != name {
199                 return;
200             }
201
202             match hash {
203                 Some(hash) if hash == data.hash() => {
204                     ret = Some(cnum);
205                     return;
206                 }
207                 Some(..) => return,
208                 None => {}
209             }
210
211             // When the hash is None we're dealing with a top-level dependency
212             // in which case we may have a specification on the command line for
213             // this library. Even though an upstream library may have loaded
214             // something of the same name, we have to make sure it was loaded
215             // from the exact same location as well.
216             //
217             // We're also sure to compare *paths*, not actual byte slices. The
218             // `source` stores paths which are normalized which may be different
219             // from the strings on the command line.
220             let source = self.cstore.get_crate_data(cnum).source();
221             if let Some(entry) = self.sess.opts.externs.get(&name.as_str()) {
222                 // Only use `--extern crate_name=path` here, not `--extern crate_name`.
223                 if let Some(mut files) = entry.files() {
224                     if files.any(|l| {
225                         let l = fs::canonicalize(l).ok();
226                         source.dylib.as_ref().map(|p| &p.0) == l.as_ref()
227                             || source.rlib.as_ref().map(|p| &p.0) == l.as_ref()
228                     }) {
229                         ret = Some(cnum);
230                     }
231                 }
232                 return;
233             }
234
235             // Alright, so we've gotten this far which means that `data` has the
236             // right name, we don't have a hash, and we don't have a --extern
237             // pointing for ourselves. We're still not quite yet done because we
238             // have to make sure that this crate was found in the crate lookup
239             // path (this is a top-level dependency) as we don't want to
240             // implicitly load anything inside the dependency lookup path.
241             let prev_kind = source
242                 .dylib
243                 .as_ref()
244                 .or(source.rlib.as_ref())
245                 .or(source.rmeta.as_ref())
246                 .expect("No sources for crate")
247                 .1;
248             if kind.matches(prev_kind) {
249                 ret = Some(cnum);
250             }
251         });
252         return ret;
253     }
254
255     fn verify_no_symbol_conflicts(&self, span: Span, root: &CrateRoot<'_>) {
256         // Check for (potential) conflicts with the local crate
257         if self.local_crate_name == root.name()
258             && self.sess.local_crate_disambiguator() == root.disambiguator()
259         {
260             struct_span_err!(
261                 self.sess,
262                 span,
263                 E0519,
264                 "the current crate is indistinguishable from one of its \
265                          dependencies: it has the same crate-name `{}` and was \
266                          compiled with the same `-C metadata` arguments. This \
267                          will result in symbol conflicts between the two.",
268                 root.name()
269             )
270             .emit()
271         }
272
273         // Check for conflicts with any crate loaded so far
274         self.cstore.iter_crate_data(|_, other| {
275             if other.name() == root.name() && // same crate-name
276                other.disambiguator() == root.disambiguator() &&  // same crate-disambiguator
277                other.hash() != root.hash()
278             {
279                 // but different SVH
280                 struct_span_err!(
281                     self.sess,
282                     span,
283                     E0523,
284                     "found two different crates with name `{}` that are \
285                          not distinguished by differing `-C metadata`. This \
286                          will result in symbol conflicts between the two.",
287                     root.name()
288                 )
289                 .emit();
290             }
291         });
292     }
293
294     fn register_crate(
295         &mut self,
296         host_lib: Option<Library>,
297         root: Option<&CratePaths>,
298         span: Span,
299         lib: Library,
300         dep_kind: DepKind,
301         name: Symbol,
302     ) -> CrateNum {
303         let _prof_timer = self.sess.prof.generic_activity("metadata_register_crate");
304
305         let Library { source, metadata } = lib;
306         let crate_root = metadata.get_root();
307         let host_hash = host_lib.as_ref().map(|lib| lib.metadata.get_root().hash());
308         self.verify_no_symbol_conflicts(span, &crate_root);
309
310         let private_dep =
311             self.sess.opts.externs.get(&name.as_str()).map(|e| e.is_private_dep).unwrap_or(false);
312
313         info!("register crate `{}` (private_dep = {})", crate_root.name(), private_dep);
314
315         // Claim this crate number and cache it
316         let cnum = self.cstore.alloc_new_crate_num();
317
318         // Maintain a reference to the top most crate.
319         // Stash paths for top-most crate locally if necessary.
320         let crate_paths;
321         let root = if let Some(root) = root {
322             root
323         } else {
324             crate_paths = CratePaths::new(crate_root.name(), source.clone());
325             &crate_paths
326         };
327
328         let cnum_map = self.resolve_crate_deps(root, &crate_root, &metadata, cnum, span, dep_kind);
329
330         let raw_proc_macros = if crate_root.is_proc_macro_crate() {
331             let temp_root;
332             let (dlsym_source, dlsym_root) = match &host_lib {
333                 Some(host_lib) => (&host_lib.source, {
334                     temp_root = host_lib.metadata.get_root();
335                     &temp_root
336                 }),
337                 None => (&source, &crate_root),
338             };
339             let dlsym_dylib = dlsym_source.dylib.as_ref().expect("no dylib for a proc-macro crate");
340             Some(self.dlsym_proc_macros(&dlsym_dylib.0, dlsym_root.disambiguator(), span))
341         } else {
342             None
343         };
344
345         self.cstore.set_crate_data(
346             cnum,
347             CrateMetadata::new(
348                 self.sess,
349                 metadata,
350                 crate_root,
351                 raw_proc_macros,
352                 cnum,
353                 cnum_map,
354                 dep_kind,
355                 source,
356                 private_dep,
357                 host_hash,
358             ),
359         );
360
361         cnum
362     }
363
364     fn load_proc_macro<'b>(
365         &self,
366         locator: &mut CrateLocator<'b>,
367         path_kind: PathKind,
368     ) -> Option<(LoadResult, Option<Library>)>
369     where
370         'a: 'b,
371     {
372         // Use a new crate locator so trying to load a proc macro doesn't affect the error
373         // message we emit
374         let mut proc_macro_locator = locator.clone();
375
376         // Try to load a proc macro
377         proc_macro_locator.is_proc_macro = Some(true);
378
379         // Load the proc macro crate for the target
380         let (locator, target_result) = if self.sess.opts.debugging_opts.dual_proc_macros {
381             proc_macro_locator.reset();
382             let result = match self.load(&mut proc_macro_locator)? {
383                 LoadResult::Previous(cnum) => return Some((LoadResult::Previous(cnum), None)),
384                 LoadResult::Loaded(library) => Some(LoadResult::Loaded(library)),
385             };
386             locator.hash = locator.host_hash;
387             // Use the locator when looking for the host proc macro crate, as that is required
388             // so we want it to affect the error message
389             (locator, result)
390         } else {
391             (&mut proc_macro_locator, None)
392         };
393
394         // Load the proc macro crate for the host
395
396         locator.reset();
397         locator.is_proc_macro = Some(true);
398         locator.target = &self.sess.host;
399         locator.triple = TargetTriple::from_triple(config::host_triple());
400         locator.filesearch = self.sess.host_filesearch(path_kind);
401
402         let host_result = self.load(locator)?;
403
404         Some(if self.sess.opts.debugging_opts.dual_proc_macros {
405             let host_result = match host_result {
406                 LoadResult::Previous(..) => {
407                     panic!("host and target proc macros must be loaded in lock-step")
408                 }
409                 LoadResult::Loaded(library) => library,
410             };
411             (target_result.unwrap(), Some(host_result))
412         } else {
413             (host_result, None)
414         })
415     }
416
417     fn resolve_crate<'b>(
418         &'b mut self,
419         name: Symbol,
420         span: Span,
421         dep_kind: DepKind,
422         dep: Option<(&'b CratePaths, &'b CrateDep)>,
423     ) -> CrateNum {
424         self.maybe_resolve_crate(name, span, dep_kind, dep).unwrap_or_else(|err| err.report())
425     }
426
427     fn maybe_resolve_crate<'b>(
428         &'b mut self,
429         name: Symbol,
430         span: Span,
431         mut dep_kind: DepKind,
432         dep: Option<(&'b CratePaths, &'b CrateDep)>,
433     ) -> Result<CrateNum, LoadError<'b>> {
434         info!("resolving crate `{}`", name);
435         let (root, hash, host_hash, extra_filename, path_kind) = match dep {
436             Some((root, dep)) => (
437                 Some(root),
438                 Some(dep.hash),
439                 dep.host_hash,
440                 Some(&dep.extra_filename[..]),
441                 PathKind::Dependency,
442             ),
443             None => (None, None, None, None, PathKind::Crate),
444         };
445         let result = if let Some(cnum) = self.existing_match(name, hash, path_kind) {
446             (LoadResult::Previous(cnum), None)
447         } else {
448             info!("falling back to a load");
449             let mut locator = CrateLocator::new(
450                 self.sess,
451                 self.metadata_loader,
452                 name,
453                 hash,
454                 host_hash,
455                 extra_filename,
456                 false, // is_host
457                 path_kind,
458                 span,
459                 root,
460                 Some(false), // is_proc_macro
461             );
462
463             self.load(&mut locator)
464                 .map(|r| (r, None))
465                 .or_else(|| {
466                     dep_kind = DepKind::MacrosOnly;
467                     self.load_proc_macro(&mut locator, path_kind)
468                 })
469                 .ok_or_else(move || LoadError::LocatorError(locator))?
470         };
471
472         match result {
473             (LoadResult::Previous(cnum), None) => {
474                 let data = self.cstore.get_crate_data(cnum);
475                 if data.is_proc_macro_crate() {
476                     dep_kind = DepKind::MacrosOnly;
477                 }
478                 data.update_dep_kind(|data_dep_kind| cmp::max(data_dep_kind, dep_kind));
479                 Ok(cnum)
480             }
481             (LoadResult::Loaded(library), host_library) => {
482                 Ok(self.register_crate(host_library, root, span, library, dep_kind, name))
483             }
484             _ => panic!(),
485         }
486     }
487
488     fn load(&self, locator: &mut CrateLocator<'_>) -> Option<LoadResult> {
489         let library = locator.maybe_load_library_crate()?;
490
491         // In the case that we're loading a crate, but not matching
492         // against a hash, we could load a crate which has the same hash
493         // as an already loaded crate. If this is the case prevent
494         // duplicates by just using the first crate.
495         //
496         // Note that we only do this for target triple crates, though, as we
497         // don't want to match a host crate against an equivalent target one
498         // already loaded.
499         let root = library.metadata.get_root();
500         if locator.triple == self.sess.opts.target_triple {
501             let mut result = LoadResult::Loaded(library);
502             self.cstore.iter_crate_data(|cnum, data| {
503                 if data.name() == root.name() && root.hash() == data.hash() {
504                     assert!(locator.hash.is_none());
505                     info!("load success, going to previous cnum: {}", cnum);
506                     result = LoadResult::Previous(cnum);
507                 }
508             });
509             Some(result)
510         } else {
511             Some(LoadResult::Loaded(library))
512         }
513     }
514
515     fn update_extern_crate(&self, cnum: CrateNum, extern_crate: ExternCrate) {
516         let cmeta = self.cstore.get_crate_data(cnum);
517         if cmeta.update_extern_crate(extern_crate) {
518             // Propagate the extern crate info to dependencies if it was updated.
519             let extern_crate = ExternCrate { dependency_of: cnum, ..extern_crate };
520             for &dep_cnum in cmeta.dependencies().iter() {
521                 self.update_extern_crate(dep_cnum, extern_crate);
522             }
523         }
524     }
525
526     // Go through the crate metadata and load any crates that it references
527     fn resolve_crate_deps(
528         &mut self,
529         root: &CratePaths,
530         crate_root: &CrateRoot<'_>,
531         metadata: &MetadataBlob,
532         krate: CrateNum,
533         span: Span,
534         dep_kind: DepKind,
535     ) -> CrateNumMap {
536         debug!("resolving deps of external crate");
537         if crate_root.is_proc_macro_crate() {
538             return CrateNumMap::new();
539         }
540
541         // The map from crate numbers in the crate we're resolving to local crate numbers.
542         // We map 0 and all other holes in the map to our parent crate. The "additional"
543         // self-dependencies should be harmless.
544         std::iter::once(krate)
545             .chain(crate_root.decode_crate_deps(metadata).map(|dep| {
546                 info!(
547                     "resolving dep crate {} hash: `{}` extra filename: `{}`",
548                     dep.name, dep.hash, dep.extra_filename
549                 );
550                 let dep_kind = match dep_kind {
551                     DepKind::MacrosOnly => DepKind::MacrosOnly,
552                     _ => dep.kind,
553                 };
554                 self.resolve_crate(dep.name, span, dep_kind, Some((root, &dep)))
555             }))
556             .collect()
557     }
558
559     fn dlsym_proc_macros(
560         &self,
561         path: &Path,
562         disambiguator: CrateDisambiguator,
563         span: Span,
564     ) -> &'static [ProcMacro] {
565         use crate::dynamic_lib::DynamicLibrary;
566         use std::env;
567
568         // Make sure the path contains a / or the linker will search for it.
569         let path = env::current_dir().unwrap().join(path);
570         let lib = match DynamicLibrary::open(Some(&path)) {
571             Ok(lib) => lib,
572             Err(err) => self.sess.span_fatal(span, &err),
573         };
574
575         let sym = self.sess.generate_proc_macro_decls_symbol(disambiguator);
576         let decls = unsafe {
577             let sym = match lib.symbol(&sym) {
578                 Ok(f) => f,
579                 Err(err) => self.sess.span_fatal(span, &err),
580             };
581             *(sym as *const &[ProcMacro])
582         };
583
584         // Intentionally leak the dynamic library. We can't ever unload it
585         // since the library can make things that will live arbitrarily long.
586         std::mem::forget(lib);
587
588         decls
589     }
590
591     fn inject_panic_runtime(&mut self, krate: &ast::Crate) {
592         // If we're only compiling an rlib, then there's no need to select a
593         // panic runtime, so we just skip this section entirely.
594         let any_non_rlib =
595             self.sess.crate_types.borrow().iter().any(|ct| *ct != config::CrateType::Rlib);
596         if !any_non_rlib {
597             info!("panic runtime injection skipped, only generating rlib");
598             return;
599         }
600
601         // If we need a panic runtime, we try to find an existing one here. At
602         // the same time we perform some general validation of the DAG we've got
603         // going such as ensuring everything has a compatible panic strategy.
604         //
605         // The logic for finding the panic runtime here is pretty much the same
606         // as the allocator case with the only addition that the panic strategy
607         // compilation mode also comes into play.
608         let desired_strategy = self.sess.panic_strategy();
609         let mut runtime_found = false;
610         let mut needs_panic_runtime = attr::contains_name(&krate.attrs, sym::needs_panic_runtime);
611
612         self.cstore.iter_crate_data(|cnum, data| {
613             needs_panic_runtime = needs_panic_runtime || data.needs_panic_runtime();
614             if data.is_panic_runtime() {
615                 // Inject a dependency from all #![needs_panic_runtime] to this
616                 // #![panic_runtime] crate.
617                 self.inject_dependency_if(cnum, "a panic runtime", &|data| {
618                     data.needs_panic_runtime()
619                 });
620                 runtime_found = runtime_found || data.dep_kind() == DepKind::Explicit;
621             }
622         });
623
624         // If an explicitly linked and matching panic runtime was found, or if
625         // we just don't need one at all, then we're done here and there's
626         // nothing else to do.
627         if !needs_panic_runtime || runtime_found {
628             return;
629         }
630
631         // By this point we know that we (a) need a panic runtime and (b) no
632         // panic runtime was explicitly linked. Here we just load an appropriate
633         // default runtime for our panic strategy and then inject the
634         // dependencies.
635         //
636         // We may resolve to an already loaded crate (as the crate may not have
637         // been explicitly linked prior to this) and we may re-inject
638         // dependencies again, but both of those situations are fine.
639         //
640         // Also note that we have yet to perform validation of the crate graph
641         // in terms of everyone has a compatible panic runtime format, that's
642         // performed later as part of the `dependency_format` module.
643         let name = match desired_strategy {
644             PanicStrategy::Unwind => Symbol::intern("panic_unwind"),
645             PanicStrategy::Abort => Symbol::intern("panic_abort"),
646         };
647         info!("panic runtime not found -- loading {}", name);
648
649         let cnum = self.resolve_crate(name, DUMMY_SP, DepKind::Implicit, None);
650         let data = self.cstore.get_crate_data(cnum);
651
652         // Sanity check the loaded crate to ensure it is indeed a panic runtime
653         // and the panic strategy is indeed what we thought it was.
654         if !data.is_panic_runtime() {
655             self.sess.err(&format!("the crate `{}` is not a panic runtime", name));
656         }
657         if data.panic_strategy() != desired_strategy {
658             self.sess.err(&format!(
659                 "the crate `{}` does not have the panic \
660                                     strategy `{}`",
661                 name,
662                 desired_strategy.desc()
663             ));
664         }
665
666         self.cstore.injected_panic_runtime = Some(cnum);
667         self.inject_dependency_if(cnum, "a panic runtime", &|data| data.needs_panic_runtime());
668     }
669
670     fn inject_profiler_runtime(&mut self) {
671         if self.sess.opts.debugging_opts.profile || self.sess.opts.cg.profile_generate.enabled() {
672             info!("loading profiler");
673
674             let name = Symbol::intern("profiler_builtins");
675             let cnum = self.resolve_crate(name, DUMMY_SP, DepKind::Implicit, None);
676             let data = self.cstore.get_crate_data(cnum);
677
678             // Sanity check the loaded crate to ensure it is indeed a profiler runtime
679             if !data.is_profiler_runtime() {
680                 self.sess.err("the crate `profiler_builtins` is not a profiler runtime");
681             }
682         }
683     }
684
685     fn inject_allocator_crate(&mut self, krate: &ast::Crate) {
686         self.cstore.has_global_allocator = match &*global_allocator_spans(krate) {
687             [span1, span2, ..] => {
688                 self.sess
689                     .struct_span_err(*span2, "cannot define multiple global allocators")
690                     .span_label(*span2, "cannot define a new global allocator")
691                     .span_label(*span1, "previous global allocator defined here")
692                     .emit();
693                 true
694             }
695             spans => !spans.is_empty(),
696         };
697
698         // Check to see if we actually need an allocator. This desire comes
699         // about through the `#![needs_allocator]` attribute and is typically
700         // written down in liballoc.
701         let mut needs_allocator = attr::contains_name(&krate.attrs, sym::needs_allocator);
702         self.cstore.iter_crate_data(|_, data| {
703             needs_allocator = needs_allocator || data.needs_allocator();
704         });
705         if !needs_allocator {
706             return;
707         }
708
709         // At this point we've determined that we need an allocator. Let's see
710         // if our compilation session actually needs an allocator based on what
711         // we're emitting.
712         let all_rlib = self.sess.crate_types.borrow().iter().all(|ct| match *ct {
713             config::CrateType::Rlib => true,
714             _ => false,
715         });
716         if all_rlib {
717             return;
718         }
719
720         // Ok, we need an allocator. Not only that but we're actually going to
721         // create an artifact that needs one linked in. Let's go find the one
722         // that we're going to link in.
723         //
724         // First up we check for global allocators. Look at the crate graph here
725         // and see what's a global allocator, including if we ourselves are a
726         // global allocator.
727         let mut global_allocator =
728             self.cstore.has_global_allocator.then(|| Symbol::intern("this crate"));
729         self.cstore.iter_crate_data(|_, data| {
730             if !data.has_global_allocator() {
731                 return;
732             }
733             match global_allocator {
734                 Some(other_crate) => {
735                     self.sess.err(&format!(
736                         "the `#[global_allocator]` in {} \
737                                             conflicts with global \
738                                             allocator in: {}",
739                         other_crate,
740                         data.name()
741                     ));
742                 }
743                 None => global_allocator = Some(data.name()),
744             }
745         });
746         if global_allocator.is_some() {
747             self.cstore.allocator_kind = Some(AllocatorKind::Global);
748             return;
749         }
750
751         // Ok we haven't found a global allocator but we still need an
752         // allocator. At this point our allocator request is typically fulfilled
753         // by the standard library, denoted by the `#![default_lib_allocator]`
754         // attribute.
755         let mut has_default = attr::contains_name(&krate.attrs, sym::default_lib_allocator);
756         self.cstore.iter_crate_data(|_, data| {
757             if data.has_default_lib_allocator() {
758                 has_default = true;
759             }
760         });
761
762         if !has_default {
763             self.sess.err(
764                 "no global memory allocator found but one is \
765                            required; link to std or \
766                            add `#[global_allocator]` to a static item \
767                            that implements the GlobalAlloc trait.",
768             );
769         }
770         self.cstore.allocator_kind = Some(AllocatorKind::Default);
771     }
772
773     fn inject_dependency_if(
774         &self,
775         krate: CrateNum,
776         what: &str,
777         needs_dep: &dyn Fn(&CrateMetadata) -> bool,
778     ) {
779         // don't perform this validation if the session has errors, as one of
780         // those errors may indicate a circular dependency which could cause
781         // this to stack overflow.
782         if self.sess.has_errors() {
783             return;
784         }
785
786         // Before we inject any dependencies, make sure we don't inject a
787         // circular dependency by validating that this crate doesn't
788         // transitively depend on any crates satisfying `needs_dep`.
789         for dep in self.cstore.crate_dependencies_in_reverse_postorder(krate) {
790             let data = self.cstore.get_crate_data(dep);
791             if needs_dep(&data) {
792                 self.sess.err(&format!(
793                     "the crate `{}` cannot depend \
794                                         on a crate that needs {}, but \
795                                         it depends on `{}`",
796                     self.cstore.get_crate_data(krate).name(),
797                     what,
798                     data.name()
799                 ));
800             }
801         }
802
803         // All crates satisfying `needs_dep` do not explicitly depend on the
804         // crate provided for this compile, but in order for this compilation to
805         // be successfully linked we need to inject a dependency (to order the
806         // crates on the command line correctly).
807         self.cstore.iter_crate_data(|cnum, data| {
808             if !needs_dep(data) {
809                 return;
810             }
811
812             info!("injecting a dep from {} to {}", cnum, krate);
813             data.add_dependency(krate);
814         });
815     }
816
817     pub fn postprocess(&mut self, krate: &ast::Crate) {
818         self.inject_profiler_runtime();
819         self.inject_allocator_crate(krate);
820         self.inject_panic_runtime(krate);
821
822         if log_enabled!(log::Level::Info) {
823             dump_crates(&self.cstore);
824         }
825     }
826
827     pub fn process_extern_crate(
828         &mut self,
829         item: &ast::Item,
830         definitions: &Definitions,
831     ) -> CrateNum {
832         match item.kind {
833             ast::ItemKind::ExternCrate(orig_name) => {
834                 debug!(
835                     "resolving extern crate stmt. ident: {} orig_name: {:?}",
836                     item.ident, orig_name
837                 );
838                 let name = match orig_name {
839                     Some(orig_name) => {
840                         crate::validate_crate_name(
841                             Some(self.sess),
842                             &orig_name.as_str(),
843                             Some(item.span),
844                         );
845                         orig_name
846                     }
847                     None => item.ident.name,
848                 };
849                 let dep_kind = if attr::contains_name(&item.attrs, sym::no_link) {
850                     DepKind::MacrosOnly
851                 } else {
852                     DepKind::Explicit
853                 };
854
855                 let cnum = self.resolve_crate(name, item.span, dep_kind, None);
856
857                 let def_id = definitions.opt_local_def_id(item.id).unwrap();
858                 let path_len = definitions.def_path(def_id.index).data.len();
859                 self.update_extern_crate(
860                     cnum,
861                     ExternCrate {
862                         src: ExternCrateSource::Extern(def_id),
863                         span: item.span,
864                         path_len,
865                         dependency_of: LOCAL_CRATE,
866                     },
867                 );
868                 cnum
869             }
870             _ => bug!(),
871         }
872     }
873
874     pub fn process_path_extern(&mut self, name: Symbol, span: Span) -> CrateNum {
875         let cnum = self.resolve_crate(name, span, DepKind::Explicit, None);
876
877         self.update_extern_crate(
878             cnum,
879             ExternCrate {
880                 src: ExternCrateSource::Path,
881                 span,
882                 // to have the least priority in `update_extern_crate`
883                 path_len: usize::max_value(),
884                 dependency_of: LOCAL_CRATE,
885             },
886         );
887
888         cnum
889     }
890
891     pub fn maybe_process_path_extern(&mut self, name: Symbol, span: Span) -> Option<CrateNum> {
892         self.maybe_resolve_crate(name, span, DepKind::Explicit, None).ok()
893     }
894 }