1 //! the rustc crate store interface. This also includes types that
2 //! are *mostly* used as a part of that interface, but these should
3 //! probably get a better home if someone can find one.
8 use rustc_ast
::expand
::allocator
::AllocatorKind
;
9 use rustc_data_structures
::svh
::Svh
;
10 use rustc_data_structures
::sync
::{self, MetadataRef}
;
11 use rustc_hir
::def
::DefKind
;
12 use rustc_hir
::def_id
::{CrateNum, DefId, LOCAL_CRATE}
;
13 use rustc_hir
::definitions
::{DefKey, DefPath, DefPathHash}
;
14 use rustc_macros
::HashStable
;
15 use rustc_session
::search_paths
::PathKind
;
16 use rustc_session
::utils
::NativeLibKind
;
17 use rustc_session
::CrateDisambiguator
;
18 use rustc_span
::symbol
::Symbol
;
20 use rustc_target
::spec
::Target
;
23 use std
::path
::{Path, PathBuf}
;
25 // lonely orphan structs and enums looking for a better home
27 /// Where a crate came from on the local filesystem. One of these three options
29 #[derive(PartialEq, Clone, Debug, HashStable, Encodable, Decodable)]
30 pub struct CrateSource
{
31 pub dylib
: Option
<(PathBuf
, PathKind
)>,
32 pub rlib
: Option
<(PathBuf
, PathKind
)>,
33 pub rmeta
: Option
<(PathBuf
, PathKind
)>,
37 pub fn paths(&self) -> impl Iterator
<Item
= &PathBuf
> {
38 self.dylib
.iter().chain(self.rlib
.iter()).chain(self.rmeta
.iter()).map(|p
| &p
.0)
42 #[derive(Encodable, Decodable, Copy, Clone, Ord, PartialOrd, Eq, PartialEq, Debug)]
44 pub enum CrateDepKind
{
45 /// A dependency that is only used for its macros.
47 /// A dependency that is always injected into the dependency list and so
48 /// doesn't need to be linked to an rlib, e.g., the injected allocator.
50 /// A dependency that is required by an rlib version of this crate.
51 /// Ordinary `extern crate`s result in `Explicit` dependencies.
56 pub fn macros_only(self) -> bool
{
58 CrateDepKind
::MacrosOnly
=> true,
59 CrateDepKind
::Implicit
| CrateDepKind
::Explicit
=> false,
64 #[derive(PartialEq, Clone, Debug, Encodable, Decodable)]
72 pub fn is_some(&self) -> bool
{
73 matches
!(self, LibSource
::Some(_
))
76 pub fn option(&self) -> Option
<PathBuf
> {
78 LibSource
::Some(ref p
) => Some(p
.clone()),
79 LibSource
::MetadataOnly
| LibSource
::None
=> None
,
84 #[derive(Copy, Debug, PartialEq, Clone, Encodable, Decodable, HashStable)]
85 pub enum LinkagePreference
{
90 #[derive(Clone, Debug, Encodable, Decodable, HashStable)]
91 pub struct NativeLib
{
92 pub kind
: NativeLibKind
,
93 pub name
: Option
<Symbol
>,
94 pub cfg
: Option
<ast
::MetaItem
>,
95 pub foreign_module
: Option
<DefId
>,
96 pub wasm_import_module
: Option
<Symbol
>,
99 #[derive(Clone, TyEncodable, TyDecodable, HashStable)]
100 pub struct ForeignModule
{
101 pub foreign_items
: Vec
<DefId
>,
105 #[derive(Copy, Clone, Debug, HashStable)]
106 pub struct ExternCrate
{
107 pub src
: ExternCrateSource
,
109 /// span of the extern crate that caused this to be loaded
112 /// Number of links to reach the extern;
113 /// used to select the extern with the shortest path
116 /// Crate that depends on this crate
117 pub dependency_of
: CrateNum
,
121 /// If true, then this crate is the crate named by the extern
122 /// crate referenced above. If false, then this crate is a dep
124 pub fn is_direct(&self) -> bool
{
125 self.dependency_of
== LOCAL_CRATE
128 pub fn rank(&self) -> impl PartialOrd
{
130 // - direct extern crate to indirect
131 // - shorter paths to longer
132 (self.is_direct(), !self.path_len
)
136 #[derive(Copy, Clone, Debug, HashStable)]
137 pub enum ExternCrateSource
{
138 /// Crate is loaded by `extern crate`.
140 /// def_id of the item in the current crate that caused
141 /// this crate to be loaded; note that there could be multiple
145 /// Crate is implicitly loaded by a path resolving through extern prelude.
149 #[derive(Encodable, Decodable)]
150 pub struct EncodedMetadata
{
151 pub raw_data
: Vec
<u8>,
154 impl EncodedMetadata
{
155 pub fn new() -> EncodedMetadata
{
156 EncodedMetadata { raw_data: Vec::new() }
160 /// The backend's way to give the crate store access to the metadata in a library.
161 /// Note that it returns the raw metadata bytes stored in the library file, whether
162 /// it is compressed, uncompressed, some weird mix, etc.
163 /// rmeta files are backend independent and not handled here.
165 /// At the time of this writing, there is only one backend and one way to store
166 /// metadata in library -- this trait just serves to decouple rustc_metadata from
167 /// the archive reader, which depends on LLVM.
168 pub trait MetadataLoader
{
169 fn get_rlib_metadata(&self, target
: &Target
, filename
: &Path
) -> Result
<MetadataRef
, String
>;
170 fn get_dylib_metadata(&self, target
: &Target
, filename
: &Path
) -> Result
<MetadataRef
, String
>;
173 pub type MetadataLoaderDyn
= dyn MetadataLoader
+ Sync
;
175 /// A store of Rust crates, through which their metadata can be accessed.
177 /// Note that this trait should probably not be expanding today. All new
178 /// functionality should be driven through queries instead!
180 /// If you find a method on this trait named `{name}_untracked` it signifies
181 /// that it's *not* tracked for dependency information throughout compilation
182 /// (it'd break incremental compilation) and should only be called pre-HIR (e.g.
184 pub trait CrateStore
{
185 fn as_any(&self) -> &dyn Any
;
188 fn def_key(&self, def
: DefId
) -> DefKey
;
189 fn def_kind(&self, def
: DefId
) -> DefKind
;
190 fn def_path(&self, def
: DefId
) -> DefPath
;
191 fn def_path_hash(&self, def
: DefId
) -> DefPathHash
;
192 fn def_path_hash_to_def_id(
199 // "queries" used in resolve that aren't tracked for incremental compilation
200 fn crate_name_untracked(&self, cnum
: CrateNum
) -> Symbol
;
201 fn crate_is_private_dep_untracked(&self, cnum
: CrateNum
) -> bool
;
202 fn crate_disambiguator_untracked(&self, cnum
: CrateNum
) -> CrateDisambiguator
;
203 fn crate_hash_untracked(&self, cnum
: CrateNum
) -> Svh
;
205 // This is basically a 1-based range of ints, which is a little
206 // silly - I may fix that.
207 fn crates_untracked(&self) -> Vec
<CrateNum
>;
210 fn encode_metadata(&self, tcx
: TyCtxt
<'_
>) -> EncodedMetadata
;
211 fn metadata_encoding_version(&self) -> &[u8];
212 fn allocator_kind(&self) -> Option
<AllocatorKind
>;
215 pub type CrateStoreDyn
= dyn CrateStore
+ sync
::Sync
;
217 // This method is used when generating the command line to pass through to
218 // system linker. The linker expects undefined symbols on the left of the
219 // command line to be defined in libraries on the right, not the other way
220 // around. For more info, see some comments in the add_used_library function
223 // In order to get this left-to-right dependency ordering, we perform a
224 // topological sort of all crates putting the leaves at the right-most
226 pub fn used_crates(tcx
: TyCtxt
<'_
>, prefer
: LinkagePreference
) -> Vec
<(CrateNum
, LibSource
)> {
232 if tcx
.dep_kind(cnum
).macros_only() {
235 let source
= tcx
.used_crate_source(cnum
);
236 let path
= match prefer
{
237 LinkagePreference
::RequireDynamic
=> source
.dylib
.clone().map(|p
| p
.0),
238 LinkagePreference
::RequireStatic
=> source
.rlib
.clone().map(|p
| p
.0),
240 let path
= match path
{
241 Some(p
) => LibSource
::Some(p
),
243 if source
.rmeta
.is_some() {
244 LibSource
::MetadataOnly
252 .collect
::<Vec
<_
>>();
253 let mut ordering
= tcx
.postorder_cnums(LOCAL_CRATE
).to_owned();
255 libs
.sort_by_cached_key(|&(a
, _
)| ordering
.iter().position(|x
| *x
== a
));