]> git.lizzy.rs Git - rust.git/blob - src/librustc/middle/cstore.rs
The war on abort_if_errors
[rust.git] / src / librustc / middle / cstore.rs
1 // Copyright 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 // Copyright 2015 The Rust Project Developers. See the COPYRIGHT
12 // file at the top-level directory of this distribution and at
13 // http://rust-lang.org/COPYRIGHT.
14 //
15 // Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
16 // http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
17 // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
18 // option. This file may not be copied, modified, or distributed
19 // except according to those terms.
20
21 // the rustc crate store interface. This also includes types that
22 // are *mostly* used as a part of that interface, but these should
23 // probably get a better home if someone can find one.
24
25 use back::svh::Svh;
26 use front::map as hir_map;
27 use middle::def::{self, Def};
28 use middle::lang_items;
29 use middle::ty::{self, Ty, VariantKind};
30 use middle::def_id::{DefId, DefIndex};
31 use mir::repr::Mir;
32 use session::Session;
33 use session::search_paths::PathKind;
34 use util::nodemap::{FnvHashMap, NodeMap, NodeSet};
35 use std::any::Any;
36 use std::cell::RefCell;
37 use std::rc::Rc;
38 use std::path::PathBuf;
39 use syntax::ast;
40 use syntax::ast_util::{IdVisitingOperation};
41 use syntax::attr;
42 use syntax::codemap::Span;
43 use syntax::ptr::P;
44 use rustc_back::target::Target;
45 use rustc_front::hir;
46 use rustc_front::intravisit::Visitor;
47 use rustc_front::util::IdVisitor;
48
49 pub use self::DefLike::{DlDef, DlField, DlImpl};
50 pub use self::NativeLibraryKind::{NativeStatic, NativeFramework, NativeUnknown};
51
52 // lonely orphan structs and enums looking for a better home
53
54 #[derive(Clone, Debug)]
55 pub struct LinkMeta {
56     pub crate_name: String,
57     pub crate_hash: Svh,
58 }
59
60 // Where a crate came from on the local filesystem. One of these two options
61 // must be non-None.
62 #[derive(PartialEq, Clone, Debug)]
63 pub struct CrateSource {
64     pub dylib: Option<(PathBuf, PathKind)>,
65     pub rlib: Option<(PathBuf, PathKind)>,
66     pub cnum: ast::CrateNum,
67 }
68
69 #[derive(Copy, Debug, PartialEq, Clone)]
70 pub enum LinkagePreference {
71     RequireDynamic,
72     RequireStatic,
73 }
74
75 enum_from_u32! {
76     #[derive(Copy, Clone, PartialEq)]
77     pub enum NativeLibraryKind {
78         NativeStatic,    // native static library (.a archive)
79         NativeFramework, // OSX-specific
80         NativeUnknown,   // default way to specify a dynamic library
81     }
82 }
83
84 // Something that a name can resolve to.
85 #[derive(Copy, Clone, Debug)]
86 pub enum DefLike {
87     DlDef(Def),
88     DlImpl(DefId),
89     DlField
90 }
91
92 /// The data we save and restore about an inlined item or method.  This is not
93 /// part of the AST that we parse from a file, but it becomes part of the tree
94 /// that we trans.
95 #[derive(Clone, PartialEq, Eq, RustcEncodable, RustcDecodable, Hash, Debug)]
96 pub enum InlinedItem {
97     Item(P<hir::Item>),
98     TraitItem(DefId /* impl id */, P<hir::TraitItem>),
99     ImplItem(DefId /* impl id */, P<hir::ImplItem>),
100     Foreign(P<hir::ForeignItem>),
101 }
102
103 /// A borrowed version of `hir::InlinedItem`.
104 #[derive(Clone, Copy, PartialEq, Eq, Hash, Debug)]
105 pub enum InlinedItemRef<'a> {
106     Item(&'a hir::Item),
107     TraitItem(DefId, &'a hir::TraitItem),
108     ImplItem(DefId, &'a hir::ImplItem),
109     Foreign(&'a hir::ForeignItem)
110 }
111
112 /// Item definitions in the currently-compiled crate would have the CrateNum
113 /// LOCAL_CRATE in their DefId.
114 pub const LOCAL_CRATE: ast::CrateNum = 0;
115
116 pub struct ChildItem {
117     pub def: DefLike,
118     pub name: ast::Name,
119     pub vis: hir::Visibility
120 }
121
122 pub enum FoundAst<'ast> {
123     Found(&'ast InlinedItem),
124     FoundParent(DefId, &'ast InlinedItem),
125     NotFound,
126 }
127
128 /// A store of Rust crates, through with their metadata
129 /// can be accessed.
130 ///
131 /// The `: Any` bound is a temporary measure that allows access
132 /// to the backing `rustc_metadata::cstore::CStore` object. It
133 /// will be removed in the near future - if you need to access
134 /// internal APIs, please tell us.
135 pub trait CrateStore<'tcx> : Any {
136     // item info
137     fn stability(&self, def: DefId) -> Option<attr::Stability>;
138     fn deprecation(&self, def: DefId) -> Option<attr::Deprecation>;
139     fn closure_kind(&self, tcx: &ty::ctxt<'tcx>, def_id: DefId)
140                     -> ty::ClosureKind;
141     fn closure_ty(&self, tcx: &ty::ctxt<'tcx>, def_id: DefId)
142                   -> ty::ClosureTy<'tcx>;
143     fn item_variances(&self, def: DefId) -> ty::ItemVariances;
144     fn repr_attrs(&self, def: DefId) -> Vec<attr::ReprAttr>;
145     fn item_type(&self, tcx: &ty::ctxt<'tcx>, def: DefId)
146                  -> ty::TypeScheme<'tcx>;
147     fn item_path(&self, def: DefId) -> Vec<hir_map::PathElem>;
148     fn extern_item_path(&self, def: DefId) -> Vec<hir_map::PathElem>;
149     fn item_name(&self, def: DefId) -> ast::Name;
150     fn item_predicates(&self, tcx: &ty::ctxt<'tcx>, def: DefId)
151                        -> ty::GenericPredicates<'tcx>;
152     fn item_super_predicates(&self, tcx: &ty::ctxt<'tcx>, def: DefId)
153                              -> ty::GenericPredicates<'tcx>;
154     fn item_attrs(&self, def_id: DefId) -> Vec<ast::Attribute>;
155     fn item_symbol(&self, def: DefId) -> String;
156     fn trait_def(&self, tcx: &ty::ctxt<'tcx>, def: DefId)-> ty::TraitDef<'tcx>;
157     fn adt_def(&self, tcx: &ty::ctxt<'tcx>, def: DefId) -> ty::AdtDefMaster<'tcx>;
158     fn method_arg_names(&self, did: DefId) -> Vec<String>;
159     fn inherent_implementations_for_type(&self, def_id: DefId) -> Vec<DefId>;
160
161     // trait info
162     fn implementations_of_trait(&self, def_id: DefId) -> Vec<DefId>;
163     fn provided_trait_methods(&self, tcx: &ty::ctxt<'tcx>, def: DefId)
164                               -> Vec<Rc<ty::Method<'tcx>>>;
165     fn trait_item_def_ids(&self, def: DefId)
166                           -> Vec<ty::ImplOrTraitItemId>;
167
168     // impl info
169     fn impl_items(&self, impl_def_id: DefId) -> Vec<ty::ImplOrTraitItemId>;
170     fn impl_trait_ref(&self, tcx: &ty::ctxt<'tcx>, def: DefId)
171                       -> Option<ty::TraitRef<'tcx>>;
172     fn impl_polarity(&self, def: DefId) -> Option<hir::ImplPolarity>;
173     fn custom_coerce_unsized_kind(&self, def: DefId)
174                                   -> Option<ty::adjustment::CustomCoerceUnsized>;
175     fn associated_consts(&self, tcx: &ty::ctxt<'tcx>, def: DefId)
176                          -> Vec<Rc<ty::AssociatedConst<'tcx>>>;
177
178     // trait/impl-item info
179     fn trait_of_item(&self, tcx: &ty::ctxt<'tcx>, def_id: DefId)
180                      -> Option<DefId>;
181     fn impl_or_trait_item(&self, tcx: &ty::ctxt<'tcx>, def: DefId)
182                           -> ty::ImplOrTraitItem<'tcx>;
183
184     // flags
185     fn is_const_fn(&self, did: DefId) -> bool;
186     fn is_defaulted_trait(&self, did: DefId) -> bool;
187     fn is_impl(&self, did: DefId) -> bool;
188     fn is_default_impl(&self, impl_did: DefId) -> bool;
189     fn is_extern_fn(&self, tcx: &ty::ctxt<'tcx>, did: DefId) -> bool;
190     fn is_static(&self, did: DefId) -> bool;
191     fn is_static_method(&self, did: DefId) -> bool;
192     fn is_statically_included_foreign_item(&self, id: ast::NodeId) -> bool;
193     fn is_typedef(&self, did: DefId) -> bool;
194
195     // crate metadata
196     fn dylib_dependency_formats(&self, cnum: ast::CrateNum)
197                                     -> Vec<(ast::CrateNum, LinkagePreference)>;
198     fn lang_items(&self, cnum: ast::CrateNum) -> Vec<(DefIndex, usize)>;
199     fn missing_lang_items(&self, cnum: ast::CrateNum) -> Vec<lang_items::LangItem>;
200     fn is_staged_api(&self, cnum: ast::CrateNum) -> bool;
201     fn is_explicitly_linked(&self, cnum: ast::CrateNum) -> bool;
202     fn is_allocator(&self, cnum: ast::CrateNum) -> bool;
203     fn crate_attrs(&self, cnum: ast::CrateNum) -> Vec<ast::Attribute>;
204     fn crate_name(&self, cnum: ast::CrateNum) -> String;
205     fn crate_hash(&self, cnum: ast::CrateNum) -> Svh;
206     fn crate_struct_field_attrs(&self, cnum: ast::CrateNum)
207                                 -> FnvHashMap<DefId, Vec<ast::Attribute>>;
208     fn plugin_registrar_fn(&self, cnum: ast::CrateNum) -> Option<DefId>;
209     fn native_libraries(&self, cnum: ast::CrateNum) -> Vec<(NativeLibraryKind, String)>;
210     fn reachable_ids(&self, cnum: ast::CrateNum) -> Vec<DefId>;
211
212     // resolve
213     fn def_path(&self, def: DefId) -> hir_map::DefPath;
214     fn variant_kind(&self, def_id: DefId) -> Option<VariantKind>;
215     fn struct_ctor_def_id(&self, struct_def_id: DefId) -> Option<DefId>;
216     fn tuple_struct_definition_if_ctor(&self, did: DefId) -> Option<DefId>;
217     fn struct_field_names(&self, def: DefId) -> Vec<ast::Name>;
218     fn item_children(&self, did: DefId) -> Vec<ChildItem>;
219     fn crate_top_level_items(&self, cnum: ast::CrateNum) -> Vec<ChildItem>;
220
221     // misc. metadata
222     fn maybe_get_item_ast(&'tcx self, tcx: &ty::ctxt<'tcx>, def: DefId)
223                           -> FoundAst<'tcx>;
224     fn maybe_get_item_mir(&self, tcx: &ty::ctxt<'tcx>, def: DefId)
225                           -> Option<Mir<'tcx>>;
226     // This is basically a 1-based range of ints, which is a little
227     // silly - I may fix that.
228     fn crates(&self) -> Vec<ast::CrateNum>;
229     fn used_libraries(&self) -> Vec<(String, NativeLibraryKind)>;
230     fn used_link_args(&self) -> Vec<String>;
231
232     // utility functions
233     fn metadata_filename(&self) -> &str;
234     fn metadata_section_name(&self, target: &Target) -> &str;
235     fn encode_type(&self, tcx: &ty::ctxt<'tcx>, ty: Ty<'tcx>) -> Vec<u8>;
236     fn used_crates(&self, prefer: LinkagePreference) -> Vec<(ast::CrateNum, Option<PathBuf>)>;
237     fn used_crate_source(&self, cnum: ast::CrateNum) -> CrateSource;
238     fn extern_mod_stmt_cnum(&self, emod_id: ast::NodeId) -> Option<ast::CrateNum>;
239     fn encode_metadata(&self,
240                        tcx: &ty::ctxt<'tcx>,
241                        reexports: &def::ExportMap,
242                        item_symbols: &RefCell<NodeMap<String>>,
243                        link_meta: &LinkMeta,
244                        reachable: &NodeSet,
245                        mir_map: &NodeMap<Mir<'tcx>>,
246                        krate: &hir::Crate) -> Vec<u8>;
247     fn metadata_encoding_version(&self) -> &[u8];
248 }
249
250 impl InlinedItem {
251     pub fn visit<'ast,V>(&'ast self, visitor: &mut V)
252         where V: Visitor<'ast>
253     {
254         match *self {
255             InlinedItem::Item(ref i) => visitor.visit_item(&**i),
256             InlinedItem::Foreign(ref i) => visitor.visit_foreign_item(&**i),
257             InlinedItem::TraitItem(_, ref ti) => visitor.visit_trait_item(ti),
258             InlinedItem::ImplItem(_, ref ii) => visitor.visit_impl_item(ii),
259         }
260     }
261
262     pub fn visit_ids<O: IdVisitingOperation>(&self, operation: &mut O) {
263         let mut id_visitor = IdVisitor::new(operation);
264         self.visit(&mut id_visitor);
265     }
266 }
267
268 // FIXME: find a better place for this?
269 pub fn validate_crate_name(sess: Option<&Session>, s: &str, sp: Option<Span>) {
270     let say = |s: &str| {
271         match (sp, sess) {
272             (_, None) => panic!("{}", s),
273             (Some(sp), Some(sess)) => sess.span_fatal(sp, s),
274             (None, Some(sess)) => sess.fatal(s),
275         }
276     };
277     if s.is_empty() {
278         say("crate name must not be empty");
279     }
280     for c in s.chars() {
281         if c.is_alphanumeric() { continue }
282         if c == '_'  { continue }
283         say(&format!("invalid character `{}` in crate name: `{}`", c, s));
284     }
285 }
286
287 /// A dummy crate store that does not support any non-local crates,
288 /// for test purposes.
289 pub struct DummyCrateStore;
290 #[allow(unused_variables)]
291 impl<'tcx> CrateStore<'tcx> for DummyCrateStore {
292     // item info
293     fn stability(&self, def: DefId) -> Option<attr::Stability> { unimplemented!() }
294     fn deprecation(&self, def: DefId) -> Option<attr::Deprecation> { unimplemented!() }
295     fn closure_kind(&self, tcx: &ty::ctxt<'tcx>, def_id: DefId)
296                     -> ty::ClosureKind  { unimplemented!() }
297     fn closure_ty(&self, tcx: &ty::ctxt<'tcx>, def_id: DefId)
298                   -> ty::ClosureTy<'tcx>  { unimplemented!() }
299     fn item_variances(&self, def: DefId) -> ty::ItemVariances { unimplemented!() }
300     fn repr_attrs(&self, def: DefId) -> Vec<attr::ReprAttr> { unimplemented!() }
301     fn item_type(&self, tcx: &ty::ctxt<'tcx>, def: DefId)
302                  -> ty::TypeScheme<'tcx> { unimplemented!() }
303     fn item_path(&self, def: DefId) -> Vec<hir_map::PathElem> { unimplemented!() }
304     fn extern_item_path(&self, def: DefId) -> Vec<hir_map::PathElem> { unimplemented!() }
305     fn item_name(&self, def: DefId) -> ast::Name { unimplemented!() }
306     fn item_predicates(&self, tcx: &ty::ctxt<'tcx>, def: DefId)
307                        -> ty::GenericPredicates<'tcx> { unimplemented!() }
308     fn item_super_predicates(&self, tcx: &ty::ctxt<'tcx>, def: DefId)
309                              -> ty::GenericPredicates<'tcx> { unimplemented!() }
310     fn item_attrs(&self, def_id: DefId) -> Vec<ast::Attribute> { unimplemented!() }
311     fn item_symbol(&self, def: DefId) -> String { unimplemented!() }
312     fn trait_def(&self, tcx: &ty::ctxt<'tcx>, def: DefId)-> ty::TraitDef<'tcx>
313         { unimplemented!() }
314     fn adt_def(&self, tcx: &ty::ctxt<'tcx>, def: DefId) -> ty::AdtDefMaster<'tcx>
315         { unimplemented!() }
316     fn method_arg_names(&self, did: DefId) -> Vec<String> { unimplemented!() }
317     fn inherent_implementations_for_type(&self, def_id: DefId) -> Vec<DefId> { vec![] }
318
319     // trait info
320     fn implementations_of_trait(&self, def_id: DefId) -> Vec<DefId> { vec![] }
321     fn provided_trait_methods(&self, tcx: &ty::ctxt<'tcx>, def: DefId)
322                               -> Vec<Rc<ty::Method<'tcx>>> { unimplemented!() }
323     fn trait_item_def_ids(&self, def: DefId)
324                           -> Vec<ty::ImplOrTraitItemId> { unimplemented!() }
325
326     // impl info
327     fn impl_items(&self, impl_def_id: DefId) -> Vec<ty::ImplOrTraitItemId>
328         { unimplemented!() }
329     fn impl_trait_ref(&self, tcx: &ty::ctxt<'tcx>, def: DefId)
330                       -> Option<ty::TraitRef<'tcx>> { unimplemented!() }
331     fn impl_polarity(&self, def: DefId) -> Option<hir::ImplPolarity> { unimplemented!() }
332     fn custom_coerce_unsized_kind(&self, def: DefId)
333                                   -> Option<ty::adjustment::CustomCoerceUnsized>
334         { unimplemented!() }
335     fn associated_consts(&self, tcx: &ty::ctxt<'tcx>, def: DefId)
336                          -> Vec<Rc<ty::AssociatedConst<'tcx>>> { unimplemented!() }
337
338     // trait/impl-item info
339     fn trait_of_item(&self, tcx: &ty::ctxt<'tcx>, def_id: DefId)
340                      -> Option<DefId> { unimplemented!() }
341     fn impl_or_trait_item(&self, tcx: &ty::ctxt<'tcx>, def: DefId)
342                           -> ty::ImplOrTraitItem<'tcx> { unimplemented!() }
343
344     // flags
345     fn is_const_fn(&self, did: DefId) -> bool { unimplemented!() }
346     fn is_defaulted_trait(&self, did: DefId) -> bool { unimplemented!() }
347     fn is_impl(&self, did: DefId) -> bool { unimplemented!() }
348     fn is_default_impl(&self, impl_did: DefId) -> bool { unimplemented!() }
349     fn is_extern_fn(&self, tcx: &ty::ctxt<'tcx>, did: DefId) -> bool { unimplemented!() }
350     fn is_static(&self, did: DefId) -> bool { unimplemented!() }
351     fn is_static_method(&self, did: DefId) -> bool { unimplemented!() }
352     fn is_statically_included_foreign_item(&self, id: ast::NodeId) -> bool { false }
353     fn is_typedef(&self, did: DefId) -> bool { unimplemented!() }
354
355     // crate metadata
356     fn dylib_dependency_formats(&self, cnum: ast::CrateNum)
357                                     -> Vec<(ast::CrateNum, LinkagePreference)>
358         { unimplemented!() }
359     fn lang_items(&self, cnum: ast::CrateNum) -> Vec<(DefIndex, usize)>
360         { unimplemented!() }
361     fn missing_lang_items(&self, cnum: ast::CrateNum) -> Vec<lang_items::LangItem>
362         { unimplemented!() }
363     fn is_staged_api(&self, cnum: ast::CrateNum) -> bool { unimplemented!() }
364     fn is_explicitly_linked(&self, cnum: ast::CrateNum) -> bool { unimplemented!() }
365     fn is_allocator(&self, cnum: ast::CrateNum) -> bool { unimplemented!() }
366     fn crate_attrs(&self, cnum: ast::CrateNum) -> Vec<ast::Attribute>
367         { unimplemented!() }
368     fn crate_name(&self, cnum: ast::CrateNum) -> String { unimplemented!() }
369     fn crate_hash(&self, cnum: ast::CrateNum) -> Svh { unimplemented!() }
370     fn crate_struct_field_attrs(&self, cnum: ast::CrateNum)
371                                 -> FnvHashMap<DefId, Vec<ast::Attribute>>
372         { unimplemented!() }
373     fn plugin_registrar_fn(&self, cnum: ast::CrateNum) -> Option<DefId>
374         { unimplemented!() }
375     fn native_libraries(&self, cnum: ast::CrateNum) -> Vec<(NativeLibraryKind, String)>
376         { unimplemented!() }
377     fn reachable_ids(&self, cnum: ast::CrateNum) -> Vec<DefId> { unimplemented!() }
378
379     // resolve
380     fn def_path(&self, def: DefId) -> hir_map::DefPath { unimplemented!() }
381     fn variant_kind(&self, def_id: DefId) -> Option<VariantKind> { unimplemented!() }
382     fn struct_ctor_def_id(&self, struct_def_id: DefId) -> Option<DefId>
383         { unimplemented!() }
384     fn tuple_struct_definition_if_ctor(&self, did: DefId) -> Option<DefId>
385         { unimplemented!() }
386     fn struct_field_names(&self, def: DefId) -> Vec<ast::Name> { unimplemented!() }
387     fn item_children(&self, did: DefId) -> Vec<ChildItem> { unimplemented!() }
388     fn crate_top_level_items(&self, cnum: ast::CrateNum) -> Vec<ChildItem>
389         { unimplemented!() }
390
391     // misc. metadata
392     fn maybe_get_item_ast(&'tcx self, tcx: &ty::ctxt<'tcx>, def: DefId)
393                           -> FoundAst<'tcx> { unimplemented!() }
394     fn maybe_get_item_mir(&self, tcx: &ty::ctxt<'tcx>, def: DefId)
395                           -> Option<Mir<'tcx>> { unimplemented!() }
396
397     // This is basically a 1-based range of ints, which is a little
398     // silly - I may fix that.
399     fn crates(&self) -> Vec<ast::CrateNum> { vec![] }
400     fn used_libraries(&self) -> Vec<(String, NativeLibraryKind)> { vec![] }
401     fn used_link_args(&self) -> Vec<String> { vec![] }
402
403     // utility functions
404     fn metadata_filename(&self) -> &str { unimplemented!() }
405     fn metadata_section_name(&self, target: &Target) -> &str { unimplemented!() }
406     fn encode_type(&self, tcx: &ty::ctxt<'tcx>, ty: Ty<'tcx>) -> Vec<u8>
407         { unimplemented!() }
408     fn used_crates(&self, prefer: LinkagePreference) -> Vec<(ast::CrateNum, Option<PathBuf>)>
409         { vec![] }
410     fn used_crate_source(&self, cnum: ast::CrateNum) -> CrateSource { unimplemented!() }
411     fn extern_mod_stmt_cnum(&self, emod_id: ast::NodeId) -> Option<ast::CrateNum> { None }
412     fn encode_metadata(&self,
413                        tcx: &ty::ctxt<'tcx>,
414                        reexports: &def::ExportMap,
415                        item_symbols: &RefCell<NodeMap<String>>,
416                        link_meta: &LinkMeta,
417                        reachable: &NodeSet,
418                        mir_map: &NodeMap<Mir<'tcx>>,
419                        krate: &hir::Crate) -> Vec<u8> { vec![] }
420     fn metadata_encoding_version(&self) -> &[u8] { unimplemented!() }
421 }
422
423
424 /// Metadata encoding and decoding can make use of thread-local encoding and
425 /// decoding contexts. These allow implementers of serialize::Encodable and
426 /// Decodable to access information and datastructures that would otherwise not
427 /// be available to them. For example, we can automatically translate def-id and
428 /// span information during decoding because the decoding context knows which
429 /// crate the data is decoded from. Or it allows to make ty::Ty decodable
430 /// because the context has access to the ty::ctxt that is needed for creating
431 /// ty::Ty instances.
432 ///
433 /// Note, however, that this only works for RBML-based encoding and decoding at
434 /// the moment.
435 pub mod tls {
436     use rbml::opaque::Encoder as OpaqueEncoder;
437     use rbml::opaque::Decoder as OpaqueDecoder;
438     use serialize;
439     use std::mem;
440     use middle::ty::{self, Ty};
441     use middle::subst::Substs;
442     use middle::def_id::DefId;
443
444     pub trait EncodingContext<'tcx> {
445         fn tcx<'a>(&'a self) -> &'a ty::ctxt<'tcx>;
446         fn encode_ty(&self, encoder: &mut OpaqueEncoder, t: Ty<'tcx>);
447         fn encode_substs(&self, encoder: &mut OpaqueEncoder, substs: &Substs<'tcx>);
448     }
449
450     /// Marker type used for the scoped TLS slot.
451     /// The type context cannot be used directly because the scoped TLS
452     /// in libstd doesn't allow types generic over lifetimes.
453     struct TlsPayload;
454
455     scoped_thread_local!(static TLS_ENCODING: TlsPayload);
456
457     /// Execute f after pushing the given EncodingContext onto the TLS stack.
458     pub fn enter_encoding_context<'tcx, F, R>(ecx: &EncodingContext<'tcx>,
459                                               encoder: &mut OpaqueEncoder,
460                                               f: F) -> R
461         where F: FnOnce(&EncodingContext<'tcx>, &mut OpaqueEncoder) -> R
462     {
463         let tls_payload = (ecx as *const _, encoder as *mut _);
464         let tls_ptr = &tls_payload as *const _ as *const TlsPayload;
465         TLS_ENCODING.set(unsafe { &*tls_ptr }, || f(ecx, encoder))
466     }
467
468     /// Execute f with access to the thread-local encoding context and
469     /// rbml encoder. This function will panic if the encoder passed in and the
470     /// context encoder are not the same.
471     ///
472     /// Note that this method is 'practically' safe due to its checking that the
473     /// encoder passed in is the same as the one in TLS, but it would still be
474     /// possible to construct cases where the EncodingContext is exchanged
475     /// while the same encoder is used, thus working with a wrong context.
476     pub fn with_encoding_context<'tcx, E, F, R>(encoder: &mut E, f: F) -> R
477         where F: FnOnce(&EncodingContext<'tcx>, &mut OpaqueEncoder) -> R,
478               E: serialize::Encoder
479     {
480         unsafe {
481             unsafe_with_encoding_context(|ecx, tls_encoder| {
482                 assert!(encoder as *mut _ as usize == tls_encoder as *mut _ as usize);
483
484                 let ecx: &EncodingContext<'tcx> = mem::transmute(ecx);
485
486                 f(ecx, tls_encoder)
487             })
488         }
489     }
490
491     /// Execute f with access to the thread-local encoding context and
492     /// rbml encoder.
493     pub unsafe fn unsafe_with_encoding_context<F, R>(f: F) -> R
494         where F: FnOnce(&EncodingContext, &mut OpaqueEncoder) -> R
495     {
496         TLS_ENCODING.with(|tls| {
497             let tls_payload = (tls as *const TlsPayload)
498                                    as *mut (&EncodingContext, &mut OpaqueEncoder);
499             f((*tls_payload).0, (*tls_payload).1)
500         })
501     }
502
503     pub trait DecodingContext<'tcx> {
504         fn tcx<'a>(&'a self) -> &'a ty::ctxt<'tcx>;
505         fn decode_ty(&self, decoder: &mut OpaqueDecoder) -> ty::Ty<'tcx>;
506         fn decode_substs(&self, decoder: &mut OpaqueDecoder) -> Substs<'tcx>;
507         fn translate_def_id(&self, def_id: DefId) -> DefId;
508     }
509
510     scoped_thread_local!(static TLS_DECODING: TlsPayload);
511
512     /// Execute f after pushing the given DecodingContext onto the TLS stack.
513     pub fn enter_decoding_context<'tcx, F, R>(dcx: &DecodingContext<'tcx>,
514                                               decoder: &mut OpaqueDecoder,
515                                               f: F) -> R
516         where F: FnOnce(&DecodingContext<'tcx>, &mut OpaqueDecoder) -> R
517     {
518         let tls_payload = (dcx as *const _, decoder as *mut _);
519         let tls_ptr = &tls_payload as *const _ as *const TlsPayload;
520         TLS_DECODING.set(unsafe { &*tls_ptr }, || f(dcx, decoder))
521     }
522
523     /// Execute f with access to the thread-local decoding context and
524     /// rbml decoder. This function will panic if the decoder passed in and the
525     /// context decoder are not the same.
526     ///
527     /// Note that this method is 'practically' safe due to its checking that the
528     /// decoder passed in is the same as the one in TLS, but it would still be
529     /// possible to construct cases where the DecodingContext is exchanged
530     /// while the same decoder is used, thus working with a wrong context.
531     pub fn with_decoding_context<'decoder, 'tcx, D, F, R>(d: &'decoder mut D, f: F) -> R
532         where D: serialize::Decoder,
533               F: FnOnce(&DecodingContext<'tcx>,
534                         &mut OpaqueDecoder) -> R,
535               'tcx: 'decoder
536     {
537         unsafe {
538             unsafe_with_decoding_context(|dcx, decoder| {
539                 assert!((d as *mut _ as usize) == (decoder as *mut _ as usize));
540
541                 let dcx: &DecodingContext<'tcx> = mem::transmute(dcx);
542
543                 f(dcx, decoder)
544             })
545         }
546     }
547
548     /// Execute f with access to the thread-local decoding context and
549     /// rbml decoder.
550     pub unsafe fn unsafe_with_decoding_context<F, R>(f: F) -> R
551         where F: FnOnce(&DecodingContext, &mut OpaqueDecoder) -> R
552     {
553         TLS_DECODING.with(|tls| {
554             let tls_payload = (tls as *const TlsPayload)
555                                    as *mut (&DecodingContext, &mut OpaqueDecoder);
556             f((*tls_payload).0, (*tls_payload).1)
557         })
558     }
559 }