]> git.lizzy.rs Git - rust.git/blob - src/librustc/middle/cstore.rs
Implement the translation item collector.
[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     fn is_item_mir_available(&self, def: DefId) -> bool;
227
228     // This is basically a 1-based range of ints, which is a little
229     // silly - I may fix that.
230     fn crates(&self) -> Vec<ast::CrateNum>;
231     fn used_libraries(&self) -> Vec<(String, NativeLibraryKind)>;
232     fn used_link_args(&self) -> Vec<String>;
233
234     // utility functions
235     fn metadata_filename(&self) -> &str;
236     fn metadata_section_name(&self, target: &Target) -> &str;
237     fn encode_type(&self, tcx: &ty::ctxt<'tcx>, ty: Ty<'tcx>) -> Vec<u8>;
238     fn used_crates(&self, prefer: LinkagePreference) -> Vec<(ast::CrateNum, Option<PathBuf>)>;
239     fn used_crate_source(&self, cnum: ast::CrateNum) -> CrateSource;
240     fn extern_mod_stmt_cnum(&self, emod_id: ast::NodeId) -> Option<ast::CrateNum>;
241     fn encode_metadata(&self,
242                        tcx: &ty::ctxt<'tcx>,
243                        reexports: &def::ExportMap,
244                        item_symbols: &RefCell<NodeMap<String>>,
245                        link_meta: &LinkMeta,
246                        reachable: &NodeSet,
247                        mir_map: &NodeMap<Mir<'tcx>>,
248                        krate: &hir::Crate) -> Vec<u8>;
249     fn metadata_encoding_version(&self) -> &[u8];
250 }
251
252 impl InlinedItem {
253     pub fn visit<'ast,V>(&'ast self, visitor: &mut V)
254         where V: Visitor<'ast>
255     {
256         match *self {
257             InlinedItem::Item(ref i) => visitor.visit_item(&**i),
258             InlinedItem::Foreign(ref i) => visitor.visit_foreign_item(&**i),
259             InlinedItem::TraitItem(_, ref ti) => visitor.visit_trait_item(ti),
260             InlinedItem::ImplItem(_, ref ii) => visitor.visit_impl_item(ii),
261         }
262     }
263
264     pub fn visit_ids<O: IdVisitingOperation>(&self, operation: &mut O) {
265         let mut id_visitor = IdVisitor::new(operation);
266         self.visit(&mut id_visitor);
267     }
268 }
269
270 // FIXME: find a better place for this?
271 pub fn validate_crate_name(sess: Option<&Session>, s: &str, sp: Option<Span>) {
272     let say = |s: &str| {
273         match (sp, sess) {
274             (_, None) => panic!("{}", s),
275             (Some(sp), Some(sess)) => sess.span_err(sp, s),
276             (None, Some(sess)) => sess.err(s),
277         }
278     };
279     if s.is_empty() {
280         say("crate name must not be empty");
281     }
282     for c in s.chars() {
283         if c.is_alphanumeric() { continue }
284         if c == '_'  { continue }
285         say(&format!("invalid character `{}` in crate name: `{}`", c, s));
286     }
287     match sess {
288         Some(sess) => sess.abort_if_errors(),
289         None => {}
290     }
291 }
292
293 /// A dummy crate store that does not support any non-local crates,
294 /// for test purposes.
295 pub struct DummyCrateStore;
296 #[allow(unused_variables)]
297 impl<'tcx> CrateStore<'tcx> for DummyCrateStore {
298     // item info
299     fn stability(&self, def: DefId) -> Option<attr::Stability> { unimplemented!() }
300     fn deprecation(&self, def: DefId) -> Option<attr::Deprecation> { unimplemented!() }
301     fn closure_kind(&self, tcx: &ty::ctxt<'tcx>, def_id: DefId)
302                     -> ty::ClosureKind  { unimplemented!() }
303     fn closure_ty(&self, tcx: &ty::ctxt<'tcx>, def_id: DefId)
304                   -> ty::ClosureTy<'tcx>  { unimplemented!() }
305     fn item_variances(&self, def: DefId) -> ty::ItemVariances { unimplemented!() }
306     fn repr_attrs(&self, def: DefId) -> Vec<attr::ReprAttr> { unimplemented!() }
307     fn item_type(&self, tcx: &ty::ctxt<'tcx>, def: DefId)
308                  -> ty::TypeScheme<'tcx> { unimplemented!() }
309     fn item_path(&self, def: DefId) -> Vec<hir_map::PathElem> { unimplemented!() }
310     fn extern_item_path(&self, def: DefId) -> Vec<hir_map::PathElem> { unimplemented!() }
311     fn item_name(&self, def: DefId) -> ast::Name { unimplemented!() }
312     fn item_predicates(&self, tcx: &ty::ctxt<'tcx>, def: DefId)
313                        -> ty::GenericPredicates<'tcx> { unimplemented!() }
314     fn item_super_predicates(&self, tcx: &ty::ctxt<'tcx>, def: DefId)
315                              -> ty::GenericPredicates<'tcx> { unimplemented!() }
316     fn item_attrs(&self, def_id: DefId) -> Vec<ast::Attribute> { unimplemented!() }
317     fn item_symbol(&self, def: DefId) -> String { unimplemented!() }
318     fn trait_def(&self, tcx: &ty::ctxt<'tcx>, def: DefId)-> ty::TraitDef<'tcx>
319         { unimplemented!() }
320     fn adt_def(&self, tcx: &ty::ctxt<'tcx>, def: DefId) -> ty::AdtDefMaster<'tcx>
321         { unimplemented!() }
322     fn method_arg_names(&self, did: DefId) -> Vec<String> { unimplemented!() }
323     fn inherent_implementations_for_type(&self, def_id: DefId) -> Vec<DefId> { vec![] }
324
325     // trait info
326     fn implementations_of_trait(&self, def_id: DefId) -> Vec<DefId> { vec![] }
327     fn provided_trait_methods(&self, tcx: &ty::ctxt<'tcx>, def: DefId)
328                               -> Vec<Rc<ty::Method<'tcx>>> { unimplemented!() }
329     fn trait_item_def_ids(&self, def: DefId)
330                           -> Vec<ty::ImplOrTraitItemId> { unimplemented!() }
331
332     // impl info
333     fn impl_items(&self, impl_def_id: DefId) -> Vec<ty::ImplOrTraitItemId>
334         { unimplemented!() }
335     fn impl_trait_ref(&self, tcx: &ty::ctxt<'tcx>, def: DefId)
336                       -> Option<ty::TraitRef<'tcx>> { unimplemented!() }
337     fn impl_polarity(&self, def: DefId) -> Option<hir::ImplPolarity> { unimplemented!() }
338     fn custom_coerce_unsized_kind(&self, def: DefId)
339                                   -> Option<ty::adjustment::CustomCoerceUnsized>
340         { unimplemented!() }
341     fn associated_consts(&self, tcx: &ty::ctxt<'tcx>, def: DefId)
342                          -> Vec<Rc<ty::AssociatedConst<'tcx>>> { unimplemented!() }
343
344     // trait/impl-item info
345     fn trait_of_item(&self, tcx: &ty::ctxt<'tcx>, def_id: DefId)
346                      -> Option<DefId> { unimplemented!() }
347     fn impl_or_trait_item(&self, tcx: &ty::ctxt<'tcx>, def: DefId)
348                           -> ty::ImplOrTraitItem<'tcx> { unimplemented!() }
349
350     // flags
351     fn is_const_fn(&self, did: DefId) -> bool { unimplemented!() }
352     fn is_defaulted_trait(&self, did: DefId) -> bool { unimplemented!() }
353     fn is_impl(&self, did: DefId) -> bool { unimplemented!() }
354     fn is_default_impl(&self, impl_did: DefId) -> bool { unimplemented!() }
355     fn is_extern_fn(&self, tcx: &ty::ctxt<'tcx>, did: DefId) -> bool { unimplemented!() }
356     fn is_static(&self, did: DefId) -> bool { unimplemented!() }
357     fn is_static_method(&self, did: DefId) -> bool { unimplemented!() }
358     fn is_statically_included_foreign_item(&self, id: ast::NodeId) -> bool { false }
359     fn is_typedef(&self, did: DefId) -> bool { unimplemented!() }
360
361     // crate metadata
362     fn dylib_dependency_formats(&self, cnum: ast::CrateNum)
363                                     -> Vec<(ast::CrateNum, LinkagePreference)>
364         { unimplemented!() }
365     fn lang_items(&self, cnum: ast::CrateNum) -> Vec<(DefIndex, usize)>
366         { unimplemented!() }
367     fn missing_lang_items(&self, cnum: ast::CrateNum) -> Vec<lang_items::LangItem>
368         { unimplemented!() }
369     fn is_staged_api(&self, cnum: ast::CrateNum) -> bool { unimplemented!() }
370     fn is_explicitly_linked(&self, cnum: ast::CrateNum) -> bool { unimplemented!() }
371     fn is_allocator(&self, cnum: ast::CrateNum) -> bool { unimplemented!() }
372     fn crate_attrs(&self, cnum: ast::CrateNum) -> Vec<ast::Attribute>
373         { unimplemented!() }
374     fn crate_name(&self, cnum: ast::CrateNum) -> String { unimplemented!() }
375     fn crate_hash(&self, cnum: ast::CrateNum) -> Svh { unimplemented!() }
376     fn crate_struct_field_attrs(&self, cnum: ast::CrateNum)
377                                 -> FnvHashMap<DefId, Vec<ast::Attribute>>
378         { unimplemented!() }
379     fn plugin_registrar_fn(&self, cnum: ast::CrateNum) -> Option<DefId>
380         { unimplemented!() }
381     fn native_libraries(&self, cnum: ast::CrateNum) -> Vec<(NativeLibraryKind, String)>
382         { unimplemented!() }
383     fn reachable_ids(&self, cnum: ast::CrateNum) -> Vec<DefId> { unimplemented!() }
384
385     // resolve
386     fn def_path(&self, def: DefId) -> hir_map::DefPath { unimplemented!() }
387     fn variant_kind(&self, def_id: DefId) -> Option<VariantKind> { unimplemented!() }
388     fn struct_ctor_def_id(&self, struct_def_id: DefId) -> Option<DefId>
389         { unimplemented!() }
390     fn tuple_struct_definition_if_ctor(&self, did: DefId) -> Option<DefId>
391         { unimplemented!() }
392     fn struct_field_names(&self, def: DefId) -> Vec<ast::Name> { unimplemented!() }
393     fn item_children(&self, did: DefId) -> Vec<ChildItem> { unimplemented!() }
394     fn crate_top_level_items(&self, cnum: ast::CrateNum) -> Vec<ChildItem>
395         { unimplemented!() }
396
397     // misc. metadata
398     fn maybe_get_item_ast(&'tcx self, tcx: &ty::ctxt<'tcx>, def: DefId)
399                           -> FoundAst<'tcx> { unimplemented!() }
400     fn maybe_get_item_mir(&self, tcx: &ty::ctxt<'tcx>, def: DefId)
401                           -> Option<Mir<'tcx>> { unimplemented!() }
402     fn is_item_mir_available(&self, def: DefId) -> bool {
403         unimplemented!()
404     }
405
406     // This is basically a 1-based range of ints, which is a little
407     // silly - I may fix that.
408     fn crates(&self) -> Vec<ast::CrateNum> { vec![] }
409     fn used_libraries(&self) -> Vec<(String, NativeLibraryKind)> { vec![] }
410     fn used_link_args(&self) -> Vec<String> { vec![] }
411
412     // utility functions
413     fn metadata_filename(&self) -> &str { unimplemented!() }
414     fn metadata_section_name(&self, target: &Target) -> &str { unimplemented!() }
415     fn encode_type(&self, tcx: &ty::ctxt<'tcx>, ty: Ty<'tcx>) -> Vec<u8>
416         { unimplemented!() }
417     fn used_crates(&self, prefer: LinkagePreference) -> Vec<(ast::CrateNum, Option<PathBuf>)>
418         { vec![] }
419     fn used_crate_source(&self, cnum: ast::CrateNum) -> CrateSource { unimplemented!() }
420     fn extern_mod_stmt_cnum(&self, emod_id: ast::NodeId) -> Option<ast::CrateNum> { None }
421     fn encode_metadata(&self,
422                        tcx: &ty::ctxt<'tcx>,
423                        reexports: &def::ExportMap,
424                        item_symbols: &RefCell<NodeMap<String>>,
425                        link_meta: &LinkMeta,
426                        reachable: &NodeSet,
427                        mir_map: &NodeMap<Mir<'tcx>>,
428                        krate: &hir::Crate) -> Vec<u8> { vec![] }
429     fn metadata_encoding_version(&self) -> &[u8] { unimplemented!() }
430 }
431
432
433 /// Metadata encoding and decoding can make use of thread-local encoding and
434 /// decoding contexts. These allow implementers of serialize::Encodable and
435 /// Decodable to access information and datastructures that would otherwise not
436 /// be available to them. For example, we can automatically translate def-id and
437 /// span information during decoding because the decoding context knows which
438 /// crate the data is decoded from. Or it allows to make ty::Ty decodable
439 /// because the context has access to the ty::ctxt that is needed for creating
440 /// ty::Ty instances.
441 ///
442 /// Note, however, that this only works for RBML-based encoding and decoding at
443 /// the moment.
444 pub mod tls {
445     use rbml::opaque::Encoder as OpaqueEncoder;
446     use rbml::opaque::Decoder as OpaqueDecoder;
447     use serialize;
448     use std::mem;
449     use middle::ty::{self, Ty};
450     use middle::subst::Substs;
451     use middle::def_id::DefId;
452
453     pub trait EncodingContext<'tcx> {
454         fn tcx<'a>(&'a self) -> &'a ty::ctxt<'tcx>;
455         fn encode_ty(&self, encoder: &mut OpaqueEncoder, t: Ty<'tcx>);
456         fn encode_substs(&self, encoder: &mut OpaqueEncoder, substs: &Substs<'tcx>);
457     }
458
459     /// Marker type used for the scoped TLS slot.
460     /// The type context cannot be used directly because the scoped TLS
461     /// in libstd doesn't allow types generic over lifetimes.
462     struct TlsPayload;
463
464     scoped_thread_local!(static TLS_ENCODING: TlsPayload);
465
466     /// Execute f after pushing the given EncodingContext onto the TLS stack.
467     pub fn enter_encoding_context<'tcx, F, R>(ecx: &EncodingContext<'tcx>,
468                                               encoder: &mut OpaqueEncoder,
469                                               f: F) -> R
470         where F: FnOnce(&EncodingContext<'tcx>, &mut OpaqueEncoder) -> R
471     {
472         let tls_payload = (ecx as *const _, encoder as *mut _);
473         let tls_ptr = &tls_payload as *const _ as *const TlsPayload;
474         TLS_ENCODING.set(unsafe { &*tls_ptr }, || f(ecx, encoder))
475     }
476
477     /// Execute f with access to the thread-local encoding context and
478     /// rbml encoder. This function will panic if the encoder passed in and the
479     /// context encoder are not the same.
480     ///
481     /// Note that this method is 'practically' safe due to its checking that the
482     /// encoder passed in is the same as the one in TLS, but it would still be
483     /// possible to construct cases where the EncodingContext is exchanged
484     /// while the same encoder is used, thus working with a wrong context.
485     pub fn with_encoding_context<'tcx, E, F, R>(encoder: &mut E, f: F) -> R
486         where F: FnOnce(&EncodingContext<'tcx>, &mut OpaqueEncoder) -> R,
487               E: serialize::Encoder
488     {
489         unsafe {
490             unsafe_with_encoding_context(|ecx, tls_encoder| {
491                 assert!(encoder as *mut _ as usize == tls_encoder as *mut _ as usize);
492
493                 let ecx: &EncodingContext<'tcx> = mem::transmute(ecx);
494
495                 f(ecx, tls_encoder)
496             })
497         }
498     }
499
500     /// Execute f with access to the thread-local encoding context and
501     /// rbml encoder.
502     pub unsafe fn unsafe_with_encoding_context<F, R>(f: F) -> R
503         where F: FnOnce(&EncodingContext, &mut OpaqueEncoder) -> R
504     {
505         TLS_ENCODING.with(|tls| {
506             let tls_payload = (tls as *const TlsPayload)
507                                    as *mut (&EncodingContext, &mut OpaqueEncoder);
508             f((*tls_payload).0, (*tls_payload).1)
509         })
510     }
511
512     pub trait DecodingContext<'tcx> {
513         fn tcx<'a>(&'a self) -> &'a ty::ctxt<'tcx>;
514         fn decode_ty(&self, decoder: &mut OpaqueDecoder) -> ty::Ty<'tcx>;
515         fn decode_substs(&self, decoder: &mut OpaqueDecoder) -> Substs<'tcx>;
516         fn translate_def_id(&self, def_id: DefId) -> DefId;
517     }
518
519     scoped_thread_local!(static TLS_DECODING: TlsPayload);
520
521     /// Execute f after pushing the given DecodingContext onto the TLS stack.
522     pub fn enter_decoding_context<'tcx, F, R>(dcx: &DecodingContext<'tcx>,
523                                               decoder: &mut OpaqueDecoder,
524                                               f: F) -> R
525         where F: FnOnce(&DecodingContext<'tcx>, &mut OpaqueDecoder) -> R
526     {
527         let tls_payload = (dcx as *const _, decoder as *mut _);
528         let tls_ptr = &tls_payload as *const _ as *const TlsPayload;
529         TLS_DECODING.set(unsafe { &*tls_ptr }, || f(dcx, decoder))
530     }
531
532     /// Execute f with access to the thread-local decoding context and
533     /// rbml decoder. This function will panic if the decoder passed in and the
534     /// context decoder are not the same.
535     ///
536     /// Note that this method is 'practically' safe due to its checking that the
537     /// decoder passed in is the same as the one in TLS, but it would still be
538     /// possible to construct cases where the DecodingContext is exchanged
539     /// while the same decoder is used, thus working with a wrong context.
540     pub fn with_decoding_context<'decoder, 'tcx, D, F, R>(d: &'decoder mut D, f: F) -> R
541         where D: serialize::Decoder,
542               F: FnOnce(&DecodingContext<'tcx>,
543                         &mut OpaqueDecoder) -> R,
544               'tcx: 'decoder
545     {
546         unsafe {
547             unsafe_with_decoding_context(|dcx, decoder| {
548                 assert!((d as *mut _ as usize) == (decoder as *mut _ as usize));
549
550                 let dcx: &DecodingContext<'tcx> = mem::transmute(dcx);
551
552                 f(dcx, decoder)
553             })
554         }
555     }
556
557     /// Execute f with access to the thread-local decoding context and
558     /// rbml decoder.
559     pub unsafe fn unsafe_with_decoding_context<F, R>(f: F) -> R
560         where F: FnOnce(&DecodingContext, &mut OpaqueDecoder) -> R
561     {
562         TLS_DECODING.with(|tls| {
563             let tls_payload = (tls as *const TlsPayload)
564                                    as *mut (&DecodingContext, &mut OpaqueDecoder);
565             f((*tls_payload).0, (*tls_payload).1)
566         })
567     }
568 }