]> git.lizzy.rs Git - rust.git/blob - src/librustc_typeck/collect.rs
Run `rustfmt --file-lines ...` for changes from previous commits.
[rust.git] / src / librustc_typeck / collect.rs
1 //! "Collection" is the process of determining the type and other external
2 //! details of each item in Rust. Collection is specifically concerned
3 //! with *inter-procedural* things -- for example, for a function
4 //! definition, collection will figure out the type and signature of the
5 //! function, but it will not visit the *body* of the function in any way,
6 //! nor examine type annotations on local variables (that's the job of
7 //! type *checking*).
8 //!
9 //! Collecting is ultimately defined by a bundle of queries that
10 //! inquire after various facts about the items in the crate (e.g.,
11 //! `type_of`, `generics_of`, `predicates_of`, etc). See the `provide` function
12 //! for the full set.
13 //!
14 //! At present, however, we do run collection across all items in the
15 //! crate as a kind of pass. This should eventually be factored away.
16
17 use crate::astconv::{AstConv, Bounds, SizedByDefault};
18 use crate::constrained_generic_params as cgp;
19 use crate::check::intrinsic::intrisic_operation_unsafety;
20 use crate::lint;
21 use crate::middle::resolve_lifetime as rl;
22 use crate::middle::weak_lang_items;
23 use rustc::mir::mono::Linkage;
24 use rustc::ty::query::Providers;
25 use rustc::ty::subst::{Subst, InternalSubsts};
26 use rustc::ty::util::Discr;
27 use rustc::ty::util::IntTypeExt;
28 use rustc::ty::subst::UnpackedKind;
29 use rustc::ty::{self, AdtKind, DefIdTree, ToPolyTraitRef, Ty, TyCtxt, Const};
30 use rustc::ty::{ReprOptions, ToPredicate};
31 use rustc::util::captures::Captures;
32 use rustc::util::nodemap::FxHashMap;
33 use rustc_target::spec::abi;
34
35 use syntax::ast;
36 use syntax::ast::{Ident, MetaItemKind};
37 use syntax::attr::{InlineAttr, OptimizeAttr, list_contains_name, mark_used};
38 use syntax::source_map::Spanned;
39 use syntax::feature_gate;
40 use syntax::symbol::{InternedString, kw, Symbol, sym};
41 use syntax_pos::{Span, DUMMY_SP};
42
43 use rustc::hir::def::{CtorKind, Res, DefKind};
44 use rustc::hir::Node;
45 use rustc::hir::def_id::{DefId, LOCAL_CRATE};
46 use rustc::hir::intravisit::{self, NestedVisitorMap, Visitor};
47 use rustc::hir::GenericParamKind;
48 use rustc::hir::{self, CodegenFnAttrFlags, CodegenFnAttrs, Unsafety};
49
50 use errors::{Applicability, DiagnosticId};
51
52 use std::iter;
53
54 struct OnlySelfBounds(bool);
55
56 ///////////////////////////////////////////////////////////////////////////
57 // Main entry point
58
59 fn collect_mod_item_types<'tcx>(tcx: TyCtxt<'tcx, 'tcx>, module_def_id: DefId) {
60     tcx.hir().visit_item_likes_in_module(
61         module_def_id,
62         &mut CollectItemTypesVisitor { tcx }.as_deep_visitor()
63     );
64 }
65
66 pub fn provide(providers: &mut Providers<'_>) {
67     *providers = Providers {
68         type_of,
69         generics_of,
70         predicates_of,
71         predicates_defined_on,
72         explicit_predicates_of,
73         super_predicates_of,
74         type_param_predicates,
75         trait_def,
76         adt_def,
77         fn_sig,
78         impl_trait_ref,
79         impl_polarity,
80         is_foreign_item,
81         static_mutability,
82         codegen_fn_attrs,
83         collect_mod_item_types,
84         ..*providers
85     };
86 }
87
88 ///////////////////////////////////////////////////////////////////////////
89
90 /// Context specific to some particular item. This is what implements
91 /// `AstConv`. It has information about the predicates that are defined
92 /// on the trait. Unfortunately, this predicate information is
93 /// available in various different forms at various points in the
94 /// process. So we can't just store a pointer to e.g., the AST or the
95 /// parsed ty form, we have to be more flexible. To this end, the
96 /// `ItemCtxt` is parameterized by a `DefId` that it uses to satisfy
97 /// `get_type_parameter_bounds` requests, drawing the information from
98 /// the AST (`hir::Generics`), recursively.
99 pub struct ItemCtxt<'tcx> {
100     tcx: TyCtxt<'tcx, 'tcx>,
101     item_def_id: DefId,
102 }
103
104 ///////////////////////////////////////////////////////////////////////////
105
106 struct CollectItemTypesVisitor<'tcx> {
107     tcx: TyCtxt<'tcx, 'tcx>,
108 }
109
110 impl Visitor<'tcx> for CollectItemTypesVisitor<'tcx> {
111     fn nested_visit_map<'this>(&'this mut self) -> NestedVisitorMap<'this, 'tcx> {
112         NestedVisitorMap::OnlyBodies(&self.tcx.hir())
113     }
114
115     fn visit_item(&mut self, item: &'tcx hir::Item) {
116         convert_item(self.tcx, item.hir_id);
117         intravisit::walk_item(self, item);
118     }
119
120     fn visit_generics(&mut self, generics: &'tcx hir::Generics) {
121         for param in &generics.params {
122             match param.kind {
123                 hir::GenericParamKind::Lifetime { .. } => {}
124                 hir::GenericParamKind::Type {
125                     default: Some(_), ..
126                 } => {
127                     let def_id = self.tcx.hir().local_def_id_from_hir_id(param.hir_id);
128                     self.tcx.type_of(def_id);
129                 }
130                 hir::GenericParamKind::Type { .. } => {}
131                 hir::GenericParamKind::Const { .. } => {
132                     let def_id = self.tcx.hir().local_def_id_from_hir_id(param.hir_id);
133                     self.tcx.type_of(def_id);
134                 }
135             }
136         }
137         intravisit::walk_generics(self, generics);
138     }
139
140     fn visit_expr(&mut self, expr: &'tcx hir::Expr) {
141         if let hir::ExprKind::Closure(..) = expr.node {
142             let def_id = self.tcx.hir().local_def_id_from_hir_id(expr.hir_id);
143             self.tcx.generics_of(def_id);
144             self.tcx.type_of(def_id);
145         }
146         intravisit::walk_expr(self, expr);
147     }
148
149     fn visit_trait_item(&mut self, trait_item: &'tcx hir::TraitItem) {
150         convert_trait_item(self.tcx, trait_item.hir_id);
151         intravisit::walk_trait_item(self, trait_item);
152     }
153
154     fn visit_impl_item(&mut self, impl_item: &'tcx hir::ImplItem) {
155         convert_impl_item(self.tcx, impl_item.hir_id);
156         intravisit::walk_impl_item(self, impl_item);
157     }
158 }
159
160 ///////////////////////////////////////////////////////////////////////////
161 // Utility types and common code for the above passes.
162
163 impl ItemCtxt<'tcx> {
164     pub fn new(tcx: TyCtxt<'tcx, 'tcx>, item_def_id: DefId) -> ItemCtxt<'tcx> {
165         ItemCtxt { tcx, item_def_id }
166     }
167
168     pub fn to_ty(&self, ast_ty: &hir::Ty) -> Ty<'tcx> {
169         AstConv::ast_ty_to_ty(self, ast_ty)
170     }
171 }
172
173 impl AstConv<'tcx, 'tcx> for ItemCtxt<'tcx> {
174     fn tcx(&self) -> TyCtxt<'tcx, 'tcx> {
175         self.tcx
176     }
177
178     fn get_type_parameter_bounds(&self, span: Span, def_id: DefId)
179                                  -> &'tcx ty::GenericPredicates<'tcx> {
180         self.tcx
181             .at(span)
182             .type_param_predicates((self.item_def_id, def_id))
183     }
184
185     fn re_infer(
186         &self,
187         _: Option<&ty::GenericParamDef>,
188         _: Span,
189     ) -> Option<ty::Region<'tcx>> {
190         None
191     }
192
193     fn ty_infer(&self, _: Option<&ty::GenericParamDef>, span: Span) -> Ty<'tcx> {
194         self.tcx().sess.struct_span_err_with_code(
195             span,
196             "the type placeholder `_` is not allowed within types on item signatures",
197             DiagnosticId::Error("E0121".into()),
198         ).span_label(span, "not allowed in type signatures")
199          .emit();
200
201         self.tcx().types.err
202     }
203
204     fn ct_infer(
205         &self,
206         _: Ty<'tcx>,
207         _: Option<&ty::GenericParamDef>,
208         span: Span,
209     ) -> &'tcx Const<'tcx> {
210         self.tcx().sess.struct_span_err_with_code(
211             span,
212             "the const placeholder `_` is not allowed within types on item signatures",
213             DiagnosticId::Error("E0121".into()),
214         ).span_label(span, "not allowed in type signatures")
215          .emit();
216
217         self.tcx().consts.err
218     }
219
220     fn projected_ty_from_poly_trait_ref(
221         &self,
222         span: Span,
223         item_def_id: DefId,
224         poly_trait_ref: ty::PolyTraitRef<'tcx>,
225     ) -> Ty<'tcx> {
226         if let Some(trait_ref) = poly_trait_ref.no_bound_vars() {
227             self.tcx().mk_projection(item_def_id, trait_ref.substs)
228         } else {
229             // no late-bound regions, we can just ignore the binder
230             span_err!(
231                 self.tcx().sess,
232                 span,
233                 E0212,
234                 "cannot extract an associated type from a higher-ranked trait bound \
235                  in this context"
236             );
237             self.tcx().types.err
238         }
239     }
240
241     fn normalize_ty(&self, _span: Span, ty: Ty<'tcx>) -> Ty<'tcx> {
242         // types in item signatures are not normalized, to avoid undue
243         // dependencies.
244         ty
245     }
246
247     fn set_tainted_by_errors(&self) {
248         // no obvious place to track this, so just let it go
249     }
250
251     fn record_ty(&self, _hir_id: hir::HirId, _ty: Ty<'tcx>, _span: Span) {
252         // no place to record types from signatures?
253     }
254 }
255
256 fn type_param_predicates<'tcx>(
257     tcx: TyCtxt<'tcx, 'tcx>,
258     (item_def_id, def_id): (DefId, DefId),
259 ) -> &'tcx ty::GenericPredicates<'tcx> {
260     use rustc::hir::*;
261
262     // In the AST, bounds can derive from two places. Either
263     // written inline like `<T : Foo>` or in a where clause like
264     // `where T : Foo`.
265
266     let param_id = tcx.hir().as_local_hir_id(def_id).unwrap();
267     let param_owner = tcx.hir().ty_param_owner(param_id);
268     let param_owner_def_id = tcx.hir().local_def_id_from_hir_id(param_owner);
269     let generics = tcx.generics_of(param_owner_def_id);
270     let index = generics.param_def_id_to_index[&def_id];
271     let ty = tcx.mk_ty_param(index, tcx.hir().ty_param_name(param_id).as_interned_str());
272
273     // Don't look for bounds where the type parameter isn't in scope.
274     let parent = if item_def_id == param_owner_def_id {
275         None
276     } else {
277         tcx.generics_of(item_def_id).parent
278     };
279
280     let result = parent.map_or(&tcx.common.empty_predicates, |parent| {
281         let icx = ItemCtxt::new(tcx, parent);
282         icx.get_type_parameter_bounds(DUMMY_SP, def_id)
283     });
284     let mut extend = None;
285
286     let item_hir_id = tcx.hir().as_local_hir_id(item_def_id).unwrap();
287     let ast_generics = match tcx.hir().get_by_hir_id(item_hir_id) {
288         Node::TraitItem(item) => &item.generics,
289
290         Node::ImplItem(item) => &item.generics,
291
292         Node::Item(item) => {
293             match item.node {
294                 ItemKind::Fn(.., ref generics, _)
295                 | ItemKind::Impl(_, _, _, ref generics, ..)
296                 | ItemKind::Ty(_, ref generics)
297                 | ItemKind::Existential(ExistTy {
298                     ref generics,
299                     impl_trait_fn: None,
300                     ..
301                 })
302                 | ItemKind::Enum(_, ref generics)
303                 | ItemKind::Struct(_, ref generics)
304                 | ItemKind::Union(_, ref generics) => generics,
305                 ItemKind::Trait(_, _, ref generics, ..) => {
306                     // Implied `Self: Trait` and supertrait bounds.
307                     if param_id == item_hir_id {
308                         let identity_trait_ref = ty::TraitRef::identity(tcx, item_def_id);
309                         extend = Some((identity_trait_ref.to_predicate(), item.span));
310                     }
311                     generics
312                 }
313                 _ => return result,
314             }
315         }
316
317         Node::ForeignItem(item) => match item.node {
318             ForeignItemKind::Fn(_, _, ref generics) => generics,
319             _ => return result,
320         },
321
322         _ => return result,
323     };
324
325     let icx = ItemCtxt::new(tcx, item_def_id);
326     let mut result = (*result).clone();
327     result.predicates.extend(extend.into_iter());
328     result.predicates
329           .extend(icx.type_parameter_bounds_in_generics(ast_generics, param_id, ty,
330                   OnlySelfBounds(true)));
331     tcx.arena.alloc(result)
332 }
333
334 impl ItemCtxt<'tcx> {
335     /// Finds bounds from `hir::Generics`. This requires scanning through the
336     /// AST. We do this to avoid having to convert *all* the bounds, which
337     /// would create artificial cycles. Instead we can only convert the
338     /// bounds for a type parameter `X` if `X::Foo` is used.
339     fn type_parameter_bounds_in_generics(
340         &self,
341         ast_generics: &hir::Generics,
342         param_id: hir::HirId,
343         ty: Ty<'tcx>,
344         only_self_bounds: OnlySelfBounds,
345     ) -> Vec<(ty::Predicate<'tcx>, Span)> {
346         let from_ty_params = ast_generics
347             .params
348             .iter()
349             .filter_map(|param| match param.kind {
350                 GenericParamKind::Type { .. } if param.hir_id == param_id => Some(&param.bounds),
351                 _ => None,
352             })
353             .flat_map(|bounds| bounds.iter())
354             .flat_map(|b| predicates_from_bound(self, ty, b));
355
356         let from_where_clauses = ast_generics
357             .where_clause
358             .predicates
359             .iter()
360             .filter_map(|wp| match *wp {
361                 hir::WherePredicate::BoundPredicate(ref bp) => Some(bp),
362                 _ => None,
363             })
364             .flat_map(|bp| {
365                 let bt = if is_param(self.tcx, &bp.bounded_ty, param_id) {
366                     Some(ty)
367                 } else if !only_self_bounds.0 {
368                     Some(self.to_ty(&bp.bounded_ty))
369                 } else {
370                     None
371                 };
372                 bp.bounds.iter().filter_map(move |b| bt.map(|bt| (bt, b)))
373             })
374             .flat_map(|(bt, b)| predicates_from_bound(self, bt, b));
375
376         from_ty_params.chain(from_where_clauses).collect()
377     }
378 }
379
380 /// Tests whether this is the AST for a reference to the type
381 /// parameter with ID `param_id`. We use this so as to avoid running
382 /// `ast_ty_to_ty`, because we want to avoid triggering an all-out
383 /// conversion of the type to avoid inducing unnecessary cycles.
384 fn is_param<'tcx>(tcx: TyCtxt<'tcx, 'tcx>, ast_ty: &hir::Ty, param_id: hir::HirId) -> bool {
385     if let hir::TyKind::Path(hir::QPath::Resolved(None, ref path)) = ast_ty.node {
386         match path.res {
387             Res::SelfTy(Some(def_id), None) | Res::Def(DefKind::TyParam, def_id) => {
388                 def_id == tcx.hir().local_def_id_from_hir_id(param_id)
389             }
390             _ => false,
391         }
392     } else {
393         false
394     }
395 }
396
397 fn convert_item<'tcx>(tcx: TyCtxt<'tcx, 'tcx>, item_id: hir::HirId) {
398     let it = tcx.hir().expect_item_by_hir_id(item_id);
399     debug!("convert: item {} with id {}", it.ident, it.hir_id);
400     let def_id = tcx.hir().local_def_id_from_hir_id(item_id);
401     match it.node {
402         // These don't define types.
403         hir::ItemKind::ExternCrate(_)
404         | hir::ItemKind::Use(..)
405         | hir::ItemKind::Mod(_)
406         | hir::ItemKind::GlobalAsm(_) => {}
407         hir::ItemKind::ForeignMod(ref foreign_mod) => {
408             for item in &foreign_mod.items {
409                 let def_id = tcx.hir().local_def_id_from_hir_id(item.hir_id);
410                 tcx.generics_of(def_id);
411                 tcx.type_of(def_id);
412                 tcx.predicates_of(def_id);
413                 if let hir::ForeignItemKind::Fn(..) = item.node {
414                     tcx.fn_sig(def_id);
415                 }
416             }
417         }
418         hir::ItemKind::Enum(ref enum_definition, _) => {
419             tcx.generics_of(def_id);
420             tcx.type_of(def_id);
421             tcx.predicates_of(def_id);
422             convert_enum_variant_types(tcx, def_id, &enum_definition.variants);
423         }
424         hir::ItemKind::Impl(..) => {
425             tcx.generics_of(def_id);
426             tcx.type_of(def_id);
427             tcx.impl_trait_ref(def_id);
428             tcx.predicates_of(def_id);
429         }
430         hir::ItemKind::Trait(..) => {
431             tcx.generics_of(def_id);
432             tcx.trait_def(def_id);
433             tcx.at(it.span).super_predicates_of(def_id);
434             tcx.predicates_of(def_id);
435         }
436         hir::ItemKind::TraitAlias(..) => {
437             tcx.generics_of(def_id);
438             tcx.at(it.span).super_predicates_of(def_id);
439             tcx.predicates_of(def_id);
440         }
441         hir::ItemKind::Struct(ref struct_def, _) | hir::ItemKind::Union(ref struct_def, _) => {
442             tcx.generics_of(def_id);
443             tcx.type_of(def_id);
444             tcx.predicates_of(def_id);
445
446             for f in struct_def.fields() {
447                 let def_id = tcx.hir().local_def_id_from_hir_id(f.hir_id);
448                 tcx.generics_of(def_id);
449                 tcx.type_of(def_id);
450                 tcx.predicates_of(def_id);
451             }
452
453             if let Some(ctor_hir_id) = struct_def.ctor_hir_id() {
454                 convert_variant_ctor(tcx, ctor_hir_id);
455             }
456         }
457
458         // Desugared from `impl Trait`, so visited by the function's return type.
459         hir::ItemKind::Existential(hir::ExistTy {
460             impl_trait_fn: Some(_),
461             ..
462         }) => {}
463
464         hir::ItemKind::Existential(..)
465         | hir::ItemKind::Ty(..)
466         | hir::ItemKind::Static(..)
467         | hir::ItemKind::Const(..)
468         | hir::ItemKind::Fn(..) => {
469             tcx.generics_of(def_id);
470             tcx.type_of(def_id);
471             tcx.predicates_of(def_id);
472             if let hir::ItemKind::Fn(..) = it.node {
473                 tcx.fn_sig(def_id);
474             }
475         }
476     }
477 }
478
479 fn convert_trait_item<'tcx>(tcx: TyCtxt<'tcx, 'tcx>, trait_item_id: hir::HirId) {
480     let trait_item = tcx.hir().expect_trait_item(trait_item_id);
481     let def_id = tcx.hir().local_def_id_from_hir_id(trait_item.hir_id);
482     tcx.generics_of(def_id);
483
484     match trait_item.node {
485         hir::TraitItemKind::Const(..)
486         | hir::TraitItemKind::Type(_, Some(_))
487         | hir::TraitItemKind::Method(..) => {
488             tcx.type_of(def_id);
489             if let hir::TraitItemKind::Method(..) = trait_item.node {
490                 tcx.fn_sig(def_id);
491             }
492         }
493
494         hir::TraitItemKind::Type(_, None) => {}
495     };
496
497     tcx.predicates_of(def_id);
498 }
499
500 fn convert_impl_item<'tcx>(tcx: TyCtxt<'tcx, 'tcx>, impl_item_id: hir::HirId) {
501     let def_id = tcx.hir().local_def_id_from_hir_id(impl_item_id);
502     tcx.generics_of(def_id);
503     tcx.type_of(def_id);
504     tcx.predicates_of(def_id);
505     if let hir::ImplItemKind::Method(..) = tcx.hir().expect_impl_item(impl_item_id).node {
506         tcx.fn_sig(def_id);
507     }
508 }
509
510 fn convert_variant_ctor<'tcx>(tcx: TyCtxt<'tcx, 'tcx>, ctor_id: hir::HirId) {
511     let def_id = tcx.hir().local_def_id_from_hir_id(ctor_id);
512     tcx.generics_of(def_id);
513     tcx.type_of(def_id);
514     tcx.predicates_of(def_id);
515 }
516
517 fn convert_enum_variant_types<'tcx>(
518     tcx: TyCtxt<'tcx, 'tcx>,
519     def_id: DefId,
520     variants: &[hir::Variant],
521 ) {
522     let def = tcx.adt_def(def_id);
523     let repr_type = def.repr.discr_type();
524     let initial = repr_type.initial_discriminant(tcx);
525     let mut prev_discr = None::<Discr<'tcx>>;
526
527     // fill the discriminant values and field types
528     for variant in variants {
529         let wrapped_discr = prev_discr.map_or(initial, |d| d.wrap_incr(tcx));
530         prev_discr = Some(
531             if let Some(ref e) = variant.node.disr_expr {
532                 let expr_did = tcx.hir().local_def_id_from_hir_id(e.hir_id);
533                 def.eval_explicit_discr(tcx, expr_did)
534             } else if let Some(discr) = repr_type.disr_incr(tcx, prev_discr) {
535                 Some(discr)
536             } else {
537                 struct_span_err!(
538                     tcx.sess,
539                     variant.span,
540                     E0370,
541                     "enum discriminant overflowed"
542                 ).span_label(
543                     variant.span,
544                     format!("overflowed on value after {}", prev_discr.unwrap()),
545                 ).note(&format!(
546                     "explicitly set `{} = {}` if that is desired outcome",
547                     variant.node.ident, wrapped_discr
548                 ))
549                 .emit();
550                 None
551             }.unwrap_or(wrapped_discr),
552         );
553
554         for f in variant.node.data.fields() {
555             let def_id = tcx.hir().local_def_id_from_hir_id(f.hir_id);
556             tcx.generics_of(def_id);
557             tcx.type_of(def_id);
558             tcx.predicates_of(def_id);
559         }
560
561         // Convert the ctor, if any. This also registers the variant as
562         // an item.
563         if let Some(ctor_hir_id) = variant.node.data.ctor_hir_id() {
564             convert_variant_ctor(tcx, ctor_hir_id);
565         }
566     }
567 }
568
569 fn convert_variant<'tcx>(
570     tcx: TyCtxt<'tcx, 'tcx>,
571     variant_did: Option<DefId>,
572     ctor_did: Option<DefId>,
573     ident: Ident,
574     discr: ty::VariantDiscr,
575     def: &hir::VariantData,
576     adt_kind: ty::AdtKind,
577     parent_did: DefId,
578 ) -> ty::VariantDef {
579     let mut seen_fields: FxHashMap<ast::Ident, Span> = Default::default();
580     let hir_id = tcx.hir().as_local_hir_id(variant_did.unwrap_or(parent_did)).unwrap();
581     let fields = def
582         .fields()
583         .iter()
584         .map(|f| {
585             let fid = tcx.hir().local_def_id_from_hir_id(f.hir_id);
586             let dup_span = seen_fields.get(&f.ident.modern()).cloned();
587             if let Some(prev_span) = dup_span {
588                 struct_span_err!(
589                     tcx.sess,
590                     f.span,
591                     E0124,
592                     "field `{}` is already declared",
593                     f.ident
594                 ).span_label(f.span, "field already declared")
595                  .span_label(prev_span, format!("`{}` first declared here", f.ident))
596                  .emit();
597             } else {
598                 seen_fields.insert(f.ident.modern(), f.span);
599             }
600
601             ty::FieldDef {
602                 did: fid,
603                 ident: f.ident,
604                 vis: ty::Visibility::from_hir(&f.vis, hir_id, tcx),
605             }
606         })
607         .collect();
608     let recovered = match def {
609         hir::VariantData::Struct(_, r) => *r,
610         _ => false,
611     };
612     ty::VariantDef::new(
613         tcx,
614         ident,
615         variant_did,
616         ctor_did,
617         discr,
618         fields,
619         CtorKind::from_hir(def),
620         adt_kind,
621         parent_did,
622         recovered,
623     )
624 }
625
626 fn adt_def<'tcx>(tcx: TyCtxt<'tcx, 'tcx>, def_id: DefId) -> &'tcx ty::AdtDef {
627     use rustc::hir::*;
628
629     let hir_id = tcx.hir().as_local_hir_id(def_id).unwrap();
630     let item = match tcx.hir().get_by_hir_id(hir_id) {
631         Node::Item(item) => item,
632         _ => bug!(),
633     };
634
635     let repr = ReprOptions::new(tcx, def_id);
636     let (kind, variants) = match item.node {
637         ItemKind::Enum(ref def, _) => {
638             let mut distance_from_explicit = 0;
639             let variants = def.variants
640                 .iter()
641                 .map(|v| {
642                     let variant_did = Some(tcx.hir().local_def_id_from_hir_id(v.node.id));
643                     let ctor_did = v.node.data.ctor_hir_id()
644                         .map(|hir_id| tcx.hir().local_def_id_from_hir_id(hir_id));
645
646                     let discr = if let Some(ref e) = v.node.disr_expr {
647                         distance_from_explicit = 0;
648                         ty::VariantDiscr::Explicit(tcx.hir().local_def_id_from_hir_id(e.hir_id))
649                     } else {
650                         ty::VariantDiscr::Relative(distance_from_explicit)
651                     };
652                     distance_from_explicit += 1;
653
654                     convert_variant(tcx, variant_did, ctor_did, v.node.ident, discr,
655                                     &v.node.data, AdtKind::Enum, def_id)
656                 })
657                 .collect();
658
659             (AdtKind::Enum, variants)
660         }
661         ItemKind::Struct(ref def, _) => {
662             let variant_did = None;
663             let ctor_did = def.ctor_hir_id()
664                 .map(|hir_id| tcx.hir().local_def_id_from_hir_id(hir_id));
665
666             let variants = std::iter::once(convert_variant(
667                 tcx, variant_did, ctor_did, item.ident, ty::VariantDiscr::Relative(0), def,
668                 AdtKind::Struct, def_id,
669             )).collect();
670
671             (AdtKind::Struct, variants)
672         }
673         ItemKind::Union(ref def, _) => {
674             let variant_did = None;
675             let ctor_did = def.ctor_hir_id()
676                 .map(|hir_id| tcx.hir().local_def_id_from_hir_id(hir_id));
677
678             let variants = std::iter::once(convert_variant(
679                 tcx, variant_did, ctor_did, item.ident, ty::VariantDiscr::Relative(0), def,
680                 AdtKind::Union, def_id,
681             )).collect();
682
683             (AdtKind::Union, variants)
684         },
685         _ => bug!(),
686     };
687     tcx.alloc_adt_def(def_id, kind, variants, repr)
688 }
689
690 /// Ensures that the super-predicates of the trait with a `DefId`
691 /// of `trait_def_id` are converted and stored. This also ensures that
692 /// the transitive super-predicates are converted.
693 fn super_predicates_of<'tcx>(
694     tcx: TyCtxt<'tcx, 'tcx>,
695     trait_def_id: DefId,
696 ) -> &'tcx ty::GenericPredicates<'tcx> {
697     debug!("super_predicates(trait_def_id={:?})", trait_def_id);
698     let trait_hir_id = tcx.hir().as_local_hir_id(trait_def_id).unwrap();
699
700     let item = match tcx.hir().get_by_hir_id(trait_hir_id) {
701         Node::Item(item) => item,
702         _ => bug!("trait_node_id {} is not an item", trait_hir_id),
703     };
704
705     let (generics, bounds) = match item.node {
706         hir::ItemKind::Trait(.., ref generics, ref supertraits, _) => (generics, supertraits),
707         hir::ItemKind::TraitAlias(ref generics, ref supertraits) => (generics, supertraits),
708         _ => span_bug!(item.span, "super_predicates invoked on non-trait"),
709     };
710
711     let icx = ItemCtxt::new(tcx, trait_def_id);
712
713     // Convert the bounds that follow the colon, e.g., `Bar + Zed` in `trait Foo: Bar + Zed`.
714     let self_param_ty = tcx.mk_self_type();
715     let superbounds1 = AstConv::compute_bounds(&icx, self_param_ty, bounds, SizedByDefault::No,
716         item.span);
717
718     let superbounds1 = superbounds1.predicates(tcx, self_param_ty);
719
720     // Convert any explicit superbounds in the where-clause,
721     // e.g., `trait Foo where Self: Bar`.
722     // In the case of trait aliases, however, we include all bounds in the where-clause,
723     // so e.g., `trait Foo = where u32: PartialEq<Self>` would include `u32: PartialEq<Self>`
724     // as one of its "superpredicates".
725     let is_trait_alias = tcx.is_trait_alias(trait_def_id);
726     let superbounds2 = icx.type_parameter_bounds_in_generics(
727         generics, item.hir_id, self_param_ty, OnlySelfBounds(!is_trait_alias));
728
729     // Combine the two lists to form the complete set of superbounds:
730     let superbounds: Vec<_> = superbounds1.into_iter().chain(superbounds2).collect();
731
732     // Now require that immediate supertraits are converted,
733     // which will, in turn, reach indirect supertraits.
734     for &(pred, span) in &superbounds {
735         debug!("superbound: {:?}", pred);
736         if let ty::Predicate::Trait(bound) = pred {
737             tcx.at(span).super_predicates_of(bound.def_id());
738         }
739     }
740
741     tcx.arena.alloc(ty::GenericPredicates {
742         parent: None,
743         predicates: superbounds,
744     })
745 }
746
747 fn trait_def<'tcx>(tcx: TyCtxt<'tcx, 'tcx>, def_id: DefId) -> &'tcx ty::TraitDef {
748     let hir_id = tcx.hir().as_local_hir_id(def_id).unwrap();
749     let item = tcx.hir().expect_item_by_hir_id(hir_id);
750
751     let (is_auto, unsafety) = match item.node {
752         hir::ItemKind::Trait(is_auto, unsafety, ..) => (is_auto == hir::IsAuto::Yes, unsafety),
753         hir::ItemKind::TraitAlias(..) => (false, hir::Unsafety::Normal),
754         _ => span_bug!(item.span, "trait_def_of_item invoked on non-trait"),
755     };
756
757     let paren_sugar = tcx.has_attr(def_id, sym::rustc_paren_sugar);
758     if paren_sugar && !tcx.features().unboxed_closures {
759         let mut err = tcx.sess.struct_span_err(
760             item.span,
761             "the `#[rustc_paren_sugar]` attribute is a temporary means of controlling \
762              which traits can use parenthetical notation",
763         );
764         help!(
765             &mut err,
766             "add `#![feature(unboxed_closures)]` to \
767              the crate attributes to use it"
768         );
769         err.emit();
770     }
771
772     let is_marker = tcx.has_attr(def_id, sym::marker);
773     let def_path_hash = tcx.def_path_hash(def_id);
774     let def = ty::TraitDef::new(def_id, unsafety, paren_sugar, is_auto, is_marker, def_path_hash);
775     tcx.arena.alloc(def)
776 }
777
778 fn has_late_bound_regions<'tcx>(tcx: TyCtxt<'tcx, 'tcx>, node: Node<'tcx>) -> Option<Span> {
779     struct LateBoundRegionsDetector<'tcx> {
780         tcx: TyCtxt<'tcx, 'tcx>,
781         outer_index: ty::DebruijnIndex,
782         has_late_bound_regions: Option<Span>,
783     }
784
785     impl Visitor<'tcx> for LateBoundRegionsDetector<'tcx> {
786         fn nested_visit_map<'this>(&'this mut self) -> NestedVisitorMap<'this, 'tcx> {
787             NestedVisitorMap::None
788         }
789
790         fn visit_ty(&mut self, ty: &'tcx hir::Ty) {
791             if self.has_late_bound_regions.is_some() {
792                 return;
793             }
794             match ty.node {
795                 hir::TyKind::BareFn(..) => {
796                     self.outer_index.shift_in(1);
797                     intravisit::walk_ty(self, ty);
798                     self.outer_index.shift_out(1);
799                 }
800                 _ => intravisit::walk_ty(self, ty),
801             }
802         }
803
804         fn visit_poly_trait_ref(
805             &mut self,
806             tr: &'tcx hir::PolyTraitRef,
807             m: hir::TraitBoundModifier,
808         ) {
809             if self.has_late_bound_regions.is_some() {
810                 return;
811             }
812             self.outer_index.shift_in(1);
813             intravisit::walk_poly_trait_ref(self, tr, m);
814             self.outer_index.shift_out(1);
815         }
816
817         fn visit_lifetime(&mut self, lt: &'tcx hir::Lifetime) {
818             if self.has_late_bound_regions.is_some() {
819                 return;
820             }
821
822             match self.tcx.named_region(lt.hir_id) {
823                 Some(rl::Region::Static) | Some(rl::Region::EarlyBound(..)) => {}
824                 Some(rl::Region::LateBound(debruijn, _, _))
825                 | Some(rl::Region::LateBoundAnon(debruijn, _)) if debruijn < self.outer_index => {}
826                 Some(rl::Region::LateBound(..))
827                 | Some(rl::Region::LateBoundAnon(..))
828                 | Some(rl::Region::Free(..))
829                 | None => {
830                     self.has_late_bound_regions = Some(lt.span);
831                 }
832             }
833         }
834     }
835
836     fn has_late_bound_regions<'tcx>(
837         tcx: TyCtxt<'tcx, 'tcx>,
838         generics: &'tcx hir::Generics,
839         decl: &'tcx hir::FnDecl,
840     ) -> Option<Span> {
841         let mut visitor = LateBoundRegionsDetector {
842             tcx,
843             outer_index: ty::INNERMOST,
844             has_late_bound_regions: None,
845         };
846         for param in &generics.params {
847             if let GenericParamKind::Lifetime { .. } = param.kind {
848                 if tcx.is_late_bound(param.hir_id) {
849                     return Some(param.span);
850                 }
851             }
852         }
853         visitor.visit_fn_decl(decl);
854         visitor.has_late_bound_regions
855     }
856
857     match node {
858         Node::TraitItem(item) => match item.node {
859             hir::TraitItemKind::Method(ref sig, _) => {
860                 has_late_bound_regions(tcx, &item.generics, &sig.decl)
861             }
862             _ => None,
863         },
864         Node::ImplItem(item) => match item.node {
865             hir::ImplItemKind::Method(ref sig, _) => {
866                 has_late_bound_regions(tcx, &item.generics, &sig.decl)
867             }
868             _ => None,
869         },
870         Node::ForeignItem(item) => match item.node {
871             hir::ForeignItemKind::Fn(ref fn_decl, _, ref generics) => {
872                 has_late_bound_regions(tcx, generics, fn_decl)
873             }
874             _ => None,
875         },
876         Node::Item(item) => match item.node {
877             hir::ItemKind::Fn(ref fn_decl, .., ref generics, _) => {
878                 has_late_bound_regions(tcx, generics, fn_decl)
879             }
880             _ => None,
881         },
882         _ => None,
883     }
884 }
885
886 fn generics_of<'tcx>(tcx: TyCtxt<'tcx, 'tcx>, def_id: DefId) -> &'tcx ty::Generics {
887     use rustc::hir::*;
888
889     let hir_id = tcx.hir().as_local_hir_id(def_id).unwrap();
890
891     let node = tcx.hir().get_by_hir_id(hir_id);
892     let parent_def_id = match node {
893         Node::ImplItem(_) | Node::TraitItem(_) | Node::Variant(_) |
894         Node::Ctor(..) | Node::Field(_) => {
895             let parent_id = tcx.hir().get_parent_item(hir_id);
896             Some(tcx.hir().local_def_id_from_hir_id(parent_id))
897         }
898         Node::Expr(&hir::Expr {
899             node: hir::ExprKind::Closure(..),
900             ..
901         }) => Some(tcx.closure_base_def_id(def_id)),
902         Node::Item(item) => match item.node {
903             ItemKind::Existential(hir::ExistTy { impl_trait_fn, .. }) => impl_trait_fn,
904             _ => None,
905         },
906         _ => None,
907     };
908
909     let mut opt_self = None;
910     let mut allow_defaults = false;
911
912     let no_generics = hir::Generics::empty();
913     let ast_generics = match node {
914         Node::TraitItem(item) => &item.generics,
915
916         Node::ImplItem(item) => &item.generics,
917
918         Node::Item(item) => {
919             match item.node {
920                 ItemKind::Fn(.., ref generics, _) | ItemKind::Impl(_, _, _, ref generics, ..) => {
921                     generics
922                 }
923
924                 ItemKind::Ty(_, ref generics)
925                 | ItemKind::Enum(_, ref generics)
926                 | ItemKind::Struct(_, ref generics)
927                 | ItemKind::Existential(hir::ExistTy { ref generics, .. })
928                 | ItemKind::Union(_, ref generics) => {
929                     allow_defaults = true;
930                     generics
931                 }
932
933                 ItemKind::Trait(_, _, ref generics, ..)
934                 | ItemKind::TraitAlias(ref generics, ..) => {
935                     // Add in the self type parameter.
936                     //
937                     // Something of a hack: use the node id for the trait, also as
938                     // the node id for the Self type parameter.
939                     let param_id = item.hir_id;
940
941                     opt_self = Some(ty::GenericParamDef {
942                         index: 0,
943                         name: kw::SelfUpper.as_interned_str(),
944                         def_id: tcx.hir().local_def_id_from_hir_id(param_id),
945                         pure_wrt_drop: false,
946                         kind: ty::GenericParamDefKind::Type {
947                             has_default: false,
948                             object_lifetime_default: rl::Set1::Empty,
949                             synthetic: None,
950                         },
951                     });
952
953                     allow_defaults = true;
954                     generics
955                 }
956
957                 _ => &no_generics,
958             }
959         }
960
961         Node::ForeignItem(item) => match item.node {
962             ForeignItemKind::Static(..) => &no_generics,
963             ForeignItemKind::Fn(_, _, ref generics) => generics,
964             ForeignItemKind::Type => &no_generics,
965         },
966
967         _ => &no_generics,
968     };
969
970     let has_self = opt_self.is_some();
971     let mut parent_has_self = false;
972     let mut own_start = has_self as u32;
973     let parent_count = parent_def_id.map_or(0, |def_id| {
974         let generics = tcx.generics_of(def_id);
975         assert_eq!(has_self, false);
976         parent_has_self = generics.has_self;
977         own_start = generics.count() as u32;
978         generics.parent_count + generics.params.len()
979     });
980
981     let mut params: Vec<_> = opt_self.into_iter().collect();
982
983     let early_lifetimes = early_bound_lifetimes_from_generics(tcx, ast_generics);
984     params.extend(
985         early_lifetimes
986             .enumerate()
987             .map(|(i, param)| ty::GenericParamDef {
988                 name: param.name.ident().as_interned_str(),
989                 index: own_start + i as u32,
990                 def_id: tcx.hir().local_def_id_from_hir_id(param.hir_id),
991                 pure_wrt_drop: param.pure_wrt_drop,
992                 kind: ty::GenericParamDefKind::Lifetime,
993             }),
994     );
995
996     let object_lifetime_defaults = tcx.object_lifetime_defaults(hir_id);
997
998     // Now create the real type parameters.
999     let type_start = own_start - has_self as u32 + params.len() as u32;
1000     let mut i = 0;
1001     params.extend(
1002         ast_generics
1003             .params
1004             .iter()
1005             .filter_map(|param| {
1006                 let kind = match param.kind {
1007                     GenericParamKind::Type {
1008                         ref default,
1009                         synthetic,
1010                         ..
1011                     } => {
1012                         if param.name.ident().name == kw::SelfUpper {
1013                             span_bug!(
1014                                 param.span,
1015                                 "`Self` should not be the name of a regular parameter"
1016                             );
1017                         }
1018
1019                         if !allow_defaults && default.is_some() {
1020                             if !tcx.features().default_type_parameter_fallback {
1021                                 tcx.lint_hir(
1022                                     lint::builtin::INVALID_TYPE_PARAM_DEFAULT,
1023                                     param.hir_id,
1024                                     param.span,
1025                                     &format!(
1026                                         "defaults for type parameters are only allowed in \
1027                                         `struct`, `enum`, `type`, or `trait` definitions."
1028                                     ),
1029                                 );
1030                             }
1031                         }
1032
1033                         ty::GenericParamDefKind::Type {
1034                             has_default: default.is_some(),
1035                             object_lifetime_default: object_lifetime_defaults
1036                                 .as_ref()
1037                                 .map_or(rl::Set1::Empty, |o| o[i]),
1038                             synthetic,
1039                         }
1040                     }
1041                     GenericParamKind::Const { .. } => {
1042                         if param.name.ident().name == kw::SelfUpper {
1043                             span_bug!(
1044                                 param.span,
1045                                 "`Self` should not be the name of a regular parameter",
1046                             );
1047                         }
1048
1049                         ty::GenericParamDefKind::Const
1050                     }
1051                     _ => return None,
1052                 };
1053
1054                 let param_def = ty::GenericParamDef {
1055                     index: type_start + i as u32,
1056                     name: param.name.ident().as_interned_str(),
1057                     def_id: tcx.hir().local_def_id_from_hir_id(param.hir_id),
1058                     pure_wrt_drop: param.pure_wrt_drop,
1059                     kind,
1060                 };
1061                 i += 1;
1062                 Some(param_def)
1063             })
1064     );
1065
1066     // provide junk type parameter defs - the only place that
1067     // cares about anything but the length is instantiation,
1068     // and we don't do that for closures.
1069     if let Node::Expr(&hir::Expr {
1070         node: hir::ExprKind::Closure(.., gen),
1071         ..
1072     }) = node
1073     {
1074         let dummy_args = if gen.is_some() {
1075             &["<yield_ty>", "<return_ty>", "<witness>"][..]
1076         } else {
1077             &["<closure_kind>", "<closure_signature>"][..]
1078         };
1079
1080         params.extend(
1081             dummy_args
1082                 .iter()
1083                 .enumerate()
1084                 .map(|(i, &arg)| ty::GenericParamDef {
1085                     index: type_start + i as u32,
1086                     name: InternedString::intern(arg),
1087                     def_id,
1088                     pure_wrt_drop: false,
1089                     kind: ty::GenericParamDefKind::Type {
1090                         has_default: false,
1091                         object_lifetime_default: rl::Set1::Empty,
1092                         synthetic: None,
1093                     },
1094                 }),
1095         );
1096
1097         if let Some(upvars) = tcx.upvars(def_id) {
1098             params.extend(upvars.iter().zip((dummy_args.len() as u32)..).map(|(_, i)| {
1099                 ty::GenericParamDef {
1100                     index: type_start + i,
1101                     name: InternedString::intern("<upvar>"),
1102                     def_id,
1103                     pure_wrt_drop: false,
1104                     kind: ty::GenericParamDefKind::Type {
1105                         has_default: false,
1106                         object_lifetime_default: rl::Set1::Empty,
1107                         synthetic: None,
1108                     },
1109                 }
1110             }));
1111         }
1112     }
1113
1114     let param_def_id_to_index = params
1115         .iter()
1116         .map(|param| (param.def_id, param.index))
1117         .collect();
1118
1119     tcx.arena.alloc(ty::Generics {
1120         parent: parent_def_id,
1121         parent_count,
1122         params,
1123         param_def_id_to_index,
1124         has_self: has_self || parent_has_self,
1125         has_late_bound_regions: has_late_bound_regions(tcx, node),
1126     })
1127 }
1128
1129 fn report_assoc_ty_on_inherent_impl<'tcx>(tcx: TyCtxt<'tcx, 'tcx>, span: Span) {
1130     span_err!(
1131         tcx.sess,
1132         span,
1133         E0202,
1134         "associated types are not yet supported in inherent impls (see #8995)"
1135     );
1136 }
1137
1138 fn type_of<'tcx>(tcx: TyCtxt<'tcx, 'tcx>, def_id: DefId) -> Ty<'tcx> {
1139     checked_type_of(tcx, def_id, true).unwrap()
1140 }
1141
1142 /// Same as [`type_of`] but returns [`Option`] instead of failing.
1143 ///
1144 /// If you want to fail anyway, you can set the `fail` parameter to true, but in this case,
1145 /// you'd better just call [`type_of`] directly.
1146 pub fn checked_type_of<'tcx>(
1147     tcx: TyCtxt<'tcx, 'tcx>,
1148     def_id: DefId,
1149     fail: bool,
1150 ) -> Option<Ty<'tcx>> {
1151     use rustc::hir::*;
1152
1153     let hir_id = match tcx.hir().as_local_hir_id(def_id) {
1154         Some(hir_id) => hir_id,
1155         None => {
1156             if !fail {
1157                 return None;
1158             }
1159             bug!("invalid node");
1160         }
1161     };
1162
1163     let icx = ItemCtxt::new(tcx, def_id);
1164
1165     Some(match tcx.hir().get_by_hir_id(hir_id) {
1166         Node::TraitItem(item) => match item.node {
1167             TraitItemKind::Method(..) => {
1168                 let substs = InternalSubsts::identity_for_item(tcx, def_id);
1169                 tcx.mk_fn_def(def_id, substs)
1170             }
1171             TraitItemKind::Const(ref ty, _) | TraitItemKind::Type(_, Some(ref ty)) => icx.to_ty(ty),
1172             TraitItemKind::Type(_, None) => {
1173                 if !fail {
1174                     return None;
1175                 }
1176                 span_bug!(item.span, "associated type missing default");
1177             }
1178         },
1179
1180         Node::ImplItem(item) => match item.node {
1181             ImplItemKind::Method(..) => {
1182                 let substs = InternalSubsts::identity_for_item(tcx, def_id);
1183                 tcx.mk_fn_def(def_id, substs)
1184             }
1185             ImplItemKind::Const(ref ty, _) => icx.to_ty(ty),
1186             ImplItemKind::Existential(_) => {
1187                 if tcx
1188                     .impl_trait_ref(tcx.hir().get_parent_did_by_hir_id(hir_id))
1189                     .is_none()
1190                 {
1191                     report_assoc_ty_on_inherent_impl(tcx, item.span);
1192                 }
1193
1194                 find_existential_constraints(tcx, def_id)
1195             }
1196             ImplItemKind::Type(ref ty) => {
1197                 if tcx
1198                     .impl_trait_ref(tcx.hir().get_parent_did_by_hir_id(hir_id))
1199                     .is_none()
1200                 {
1201                     report_assoc_ty_on_inherent_impl(tcx, item.span);
1202                 }
1203
1204                 icx.to_ty(ty)
1205             }
1206         },
1207
1208         Node::Item(item) => {
1209             match item.node {
1210                 ItemKind::Static(ref t, ..)
1211                 | ItemKind::Const(ref t, _)
1212                 | ItemKind::Ty(ref t, _)
1213                 | ItemKind::Impl(.., ref t, _) => icx.to_ty(t),
1214                 ItemKind::Fn(..) => {
1215                     let substs = InternalSubsts::identity_for_item(tcx, def_id);
1216                     tcx.mk_fn_def(def_id, substs)
1217                 }
1218                 ItemKind::Enum(..) | ItemKind::Struct(..) | ItemKind::Union(..) => {
1219                     let def = tcx.adt_def(def_id);
1220                     let substs = InternalSubsts::identity_for_item(tcx, def_id);
1221                     tcx.mk_adt(def, substs)
1222                 }
1223                 ItemKind::Existential(hir::ExistTy {
1224                     impl_trait_fn: None,
1225                     ..
1226                 }) => find_existential_constraints(tcx, def_id),
1227                 // Existential types desugared from `impl Trait`.
1228                 ItemKind::Existential(hir::ExistTy {
1229                     impl_trait_fn: Some(owner),
1230                     ..
1231                 }) => {
1232                     tcx.typeck_tables_of(owner)
1233                         .concrete_existential_types
1234                         .get(&def_id)
1235                         .map(|opaque| opaque.concrete_type)
1236                         .unwrap_or_else(|| {
1237                             // This can occur if some error in the
1238                             // owner fn prevented us from populating
1239                             // the `concrete_existential_types` table.
1240                             tcx.sess.delay_span_bug(
1241                                 DUMMY_SP,
1242                                 &format!(
1243                                     "owner {:?} has no existential type for {:?} in its tables",
1244                                     owner, def_id,
1245                                 ),
1246                             );
1247                             tcx.types.err
1248                         })
1249                 }
1250                 ItemKind::Trait(..)
1251                 | ItemKind::TraitAlias(..)
1252                 | ItemKind::Mod(..)
1253                 | ItemKind::ForeignMod(..)
1254                 | ItemKind::GlobalAsm(..)
1255                 | ItemKind::ExternCrate(..)
1256                 | ItemKind::Use(..) => {
1257                     if !fail {
1258                         return None;
1259                     }
1260                     span_bug!(
1261                         item.span,
1262                         "compute_type_of_item: unexpected item type: {:?}",
1263                         item.node
1264                     );
1265                 }
1266             }
1267         }
1268
1269         Node::ForeignItem(foreign_item) => match foreign_item.node {
1270             ForeignItemKind::Fn(..) => {
1271                 let substs = InternalSubsts::identity_for_item(tcx, def_id);
1272                 tcx.mk_fn_def(def_id, substs)
1273             }
1274             ForeignItemKind::Static(ref t, _) => icx.to_ty(t),
1275             ForeignItemKind::Type => tcx.mk_foreign(def_id),
1276         },
1277
1278         Node::Ctor(&ref def) | Node::Variant(&Spanned {
1279             node: hir::VariantKind { data: ref def, .. },
1280             ..
1281         }) => match *def {
1282             VariantData::Unit(..) | VariantData::Struct(..) => {
1283                 tcx.type_of(tcx.hir().get_parent_did_by_hir_id(hir_id))
1284             }
1285             VariantData::Tuple(..) => {
1286                 let substs = InternalSubsts::identity_for_item(tcx, def_id);
1287                 tcx.mk_fn_def(def_id, substs)
1288             }
1289         },
1290
1291         Node::Field(field) => icx.to_ty(&field.ty),
1292
1293         Node::Expr(&hir::Expr {
1294             node: hir::ExprKind::Closure(.., gen),
1295             ..
1296         }) => {
1297             if gen.is_some() {
1298                 return Some(tcx.typeck_tables_of(def_id).node_type(hir_id));
1299             }
1300
1301             let substs = ty::ClosureSubsts {
1302                 substs: InternalSubsts::identity_for_item(tcx, def_id),
1303             };
1304
1305             tcx.mk_closure(def_id, substs)
1306         }
1307
1308         Node::AnonConst(_) => {
1309             let parent_node = tcx.hir().get_by_hir_id(tcx.hir().get_parent_node_by_hir_id(hir_id));
1310             match parent_node {
1311                 Node::Ty(&hir::Ty {
1312                     node: hir::TyKind::Array(_, ref constant),
1313                     ..
1314                 })
1315                 | Node::Ty(&hir::Ty {
1316                     node: hir::TyKind::Typeof(ref constant),
1317                     ..
1318                 })
1319                 | Node::Expr(&hir::Expr {
1320                     node: ExprKind::Repeat(_, ref constant),
1321                     ..
1322                 }) if constant.hir_id == hir_id =>
1323                 {
1324                     tcx.types.usize
1325                 }
1326
1327                 Node::Variant(&Spanned {
1328                     node:
1329                         VariantKind {
1330                             disr_expr: Some(ref e),
1331                             ..
1332                         },
1333                     ..
1334                 }) if e.hir_id == hir_id =>
1335                 {
1336                     tcx.adt_def(tcx.hir().get_parent_did_by_hir_id(hir_id))
1337                         .repr
1338                         .discr_type()
1339                         .to_ty(tcx)
1340                 }
1341
1342                 Node::Ty(&hir::Ty { node: hir::TyKind::Path(_), .. }) |
1343                 Node::Expr(&hir::Expr { node: ExprKind::Struct(..), .. }) |
1344                 Node::Expr(&hir::Expr { node: ExprKind::Path(_), .. }) |
1345                 Node::TraitRef(..) => {
1346                     let path = match parent_node {
1347                         Node::Ty(&hir::Ty {
1348                             node: hir::TyKind::Path(QPath::Resolved(_, ref path)),
1349                             ..
1350                         })
1351                         | Node::Expr(&hir::Expr {
1352                             node: ExprKind::Path(QPath::Resolved(_, ref path)),
1353                             ..
1354                         }) => {
1355                             Some(&**path)
1356                         }
1357                         Node::Expr(&hir::Expr { node: ExprKind::Struct(ref path, ..), .. }) => {
1358                             if let QPath::Resolved(_, ref path) = **path {
1359                                 Some(&**path)
1360                             } else {
1361                                 None
1362                             }
1363                         }
1364                         Node::TraitRef(&hir::TraitRef { ref path, .. }) => Some(path),
1365                         _ => None,
1366                     };
1367
1368                     if let Some(path) = path {
1369                         let arg_index = path.segments.iter()
1370                             .filter_map(|seg| seg.args.as_ref())
1371                             .map(|generic_args| generic_args.args.as_ref())
1372                             .find_map(|args| {
1373                                 args.iter()
1374                                     .filter(|arg| arg.is_const())
1375                                     .enumerate()
1376                                     .filter(|(_, arg)| arg.id() == hir_id)
1377                                     .map(|(index, _)| index)
1378                                     .next()
1379                             })
1380                             .or_else(|| {
1381                                 if !fail {
1382                                     None
1383                                 } else {
1384                                     bug!("no arg matching AnonConst in path")
1385                                 }
1386                             })?;
1387
1388                         // We've encountered an `AnonConst` in some path, so we need to
1389                         // figure out which generic parameter it corresponds to and return
1390                         // the relevant type.
1391                         let generics = match path.res {
1392                             Res::Def(DefKind::Ctor(..), def_id) => {
1393                                 tcx.generics_of(tcx.parent(def_id).unwrap())
1394                             }
1395                             Res::Def(_, def_id) => tcx.generics_of(def_id),
1396                             Res::Err => return Some(tcx.types.err),
1397                             _ if !fail => return None,
1398                             res => {
1399                                 tcx.sess.delay_span_bug(
1400                                     DUMMY_SP,
1401                                     &format!(
1402                                         "unexpected const parent path def {:?}",
1403                                         res,
1404                                     ),
1405                                 );
1406                                 return Some(tcx.types.err);
1407                             }
1408                         };
1409
1410                         generics.params.iter()
1411                             .filter(|param| {
1412                                 if let ty::GenericParamDefKind::Const = param.kind {
1413                                     true
1414                                 } else {
1415                                     false
1416                                 }
1417                             })
1418                             .nth(arg_index)
1419                             .map(|param| tcx.type_of(param.def_id))
1420                             // This is no generic parameter associated with the arg. This is
1421                             // probably from an extra arg where one is not needed.
1422                             .unwrap_or(tcx.types.err)
1423                     } else {
1424                         if !fail {
1425                             return None;
1426                         }
1427                         tcx.sess.delay_span_bug(
1428                             DUMMY_SP,
1429                             &format!(
1430                                 "unexpected const parent path {:?}",
1431                                 parent_node,
1432                             ),
1433                         );
1434                         return Some(tcx.types.err);
1435                     }
1436                 }
1437
1438                 x => {
1439                     if !fail {
1440                         return None;
1441                     }
1442                     tcx.sess.delay_span_bug(
1443                         DUMMY_SP,
1444                         &format!(
1445                             "unexpected const parent in type_of_def_id(): {:?}", x
1446                         ),
1447                     );
1448                     tcx.types.err
1449                 }
1450             }
1451         }
1452
1453         Node::GenericParam(param) => match &param.kind {
1454             hir::GenericParamKind::Type { default: Some(ref ty), .. } |
1455             hir::GenericParamKind::Const { ref ty, .. } => {
1456                 icx.to_ty(ty)
1457             }
1458             x => {
1459                 if !fail {
1460                     return None;
1461                 }
1462                 bug!("unexpected non-type Node::GenericParam: {:?}", x)
1463             },
1464         },
1465
1466         x => {
1467             if !fail {
1468                 return None;
1469             }
1470             bug!("unexpected sort of node in type_of_def_id(): {:?}", x);
1471         }
1472     })
1473 }
1474
1475 fn find_existential_constraints<'tcx>(tcx: TyCtxt<'tcx, 'tcx>, def_id: DefId) -> Ty<'tcx> {
1476     use rustc::hir::{ImplItem, Item, TraitItem};
1477
1478     debug!("find_existential_constraints({:?})", def_id);
1479
1480     struct ConstraintLocator<'tcx> {
1481         tcx: TyCtxt<'tcx, 'tcx>,
1482         def_id: DefId,
1483         // (first found type span, actual type, mapping from the existential type's generic
1484         // parameters to the concrete type's generic parameters)
1485         //
1486         // The mapping is an index for each use site of a generic parameter in the concrete type
1487         //
1488         // The indices index into the generic parameters on the existential type.
1489         found: Option<(Span, Ty<'tcx>, Vec<usize>)>,
1490     }
1491
1492     impl ConstraintLocator<'tcx> {
1493         fn check(&mut self, def_id: DefId) {
1494             // Don't try to check items that cannot possibly constrain the type.
1495             if !self.tcx.has_typeck_tables(def_id) {
1496                 debug!(
1497                     "find_existential_constraints: no constraint for `{:?}` at `{:?}`: no tables",
1498                     self.def_id,
1499                     def_id,
1500                 );
1501                 return;
1502             }
1503             let ty = self
1504                 .tcx
1505                 .typeck_tables_of(def_id)
1506                 .concrete_existential_types
1507                 .get(&self.def_id);
1508             if let Some(ty::ResolvedOpaqueTy { concrete_type, substs }) = ty {
1509                 debug!(
1510                     "find_existential_constraints: found constraint for `{:?}` at `{:?}`: {:?}",
1511                     self.def_id,
1512                     def_id,
1513                     ty,
1514                 );
1515
1516                 // FIXME(oli-obk): trace the actual span from inference to improve errors.
1517                 let span = self.tcx.def_span(def_id);
1518                 // used to quickly look up the position of a generic parameter
1519                 let mut index_map: FxHashMap<ty::ParamTy, usize> = FxHashMap::default();
1520                 // Skipping binder is ok, since we only use this to find generic parameters and
1521                 // their positions.
1522                 for (idx, subst) in substs.iter().enumerate() {
1523                     if let UnpackedKind::Type(ty) = subst.unpack() {
1524                         if let ty::Param(p) = ty.sty {
1525                             if index_map.insert(p, idx).is_some() {
1526                                 // There was already an entry for `p`, meaning a generic parameter
1527                                 // was used twice.
1528                                 self.tcx.sess.span_err(
1529                                     span,
1530                                     &format!(
1531                                         "defining existential type use restricts existential \
1532                                          type by using the generic parameter `{}` twice",
1533                                         p.name
1534                                     ),
1535                                 );
1536                                 return;
1537                             }
1538                         } else {
1539                             self.tcx.sess.delay_span_bug(
1540                                 span,
1541                                 &format!(
1542                                     "non-defining exist ty use in defining scope: {:?}, {:?}",
1543                                     concrete_type, substs,
1544                                 ),
1545                             );
1546                         }
1547                     }
1548                 }
1549                 // Compute the index within the existential type for each generic parameter used in
1550                 // the concrete type.
1551                 let indices = concrete_type
1552                     .subst(self.tcx, substs)
1553                     .walk()
1554                     .filter_map(|t| match &t.sty {
1555                         ty::Param(p) => Some(*index_map.get(p).unwrap()),
1556                         _ => None,
1557                     }).collect();
1558                 let is_param = |ty: Ty<'_>| match ty.sty {
1559                     ty::Param(_) => true,
1560                     _ => false,
1561                 };
1562                 if !substs.types().all(is_param) {
1563                     self.tcx.sess.span_err(
1564                         span,
1565                         "defining existential type use does not fully define existential type",
1566                     );
1567                 } else if let Some((prev_span, prev_ty, ref prev_indices)) = self.found {
1568                     let mut ty = concrete_type.walk().fuse();
1569                     let mut p_ty = prev_ty.walk().fuse();
1570                     let iter_eq = (&mut ty).zip(&mut p_ty).all(|(t, p)| match (&t.sty, &p.sty) {
1571                         // Type parameters are equal to any other type parameter for the purpose of
1572                         // concrete type equality, as it is possible to obtain the same type just
1573                         // by passing matching parameters to a function.
1574                         (ty::Param(_), ty::Param(_)) => true,
1575                         _ => t == p,
1576                     });
1577                     if !iter_eq || ty.next().is_some() || p_ty.next().is_some() {
1578                         debug!("find_existential_constraints: span={:?}", span);
1579                         // Found different concrete types for the existential type.
1580                         let mut err = self.tcx.sess.struct_span_err(
1581                             span,
1582                             "concrete type differs from previous defining existential type use",
1583                         );
1584                         err.span_label(
1585                             span,
1586                             format!("expected `{}`, got `{}`", prev_ty, concrete_type),
1587                         );
1588                         err.span_note(prev_span, "previous use here");
1589                         err.emit();
1590                     } else if indices != *prev_indices {
1591                         // Found "same" concrete types, but the generic parameter order differs.
1592                         let mut err = self.tcx.sess.struct_span_err(
1593                             span,
1594                             "concrete type's generic parameters differ from previous defining use",
1595                         );
1596                         use std::fmt::Write;
1597                         let mut s = String::new();
1598                         write!(s, "expected [").unwrap();
1599                         let list = |s: &mut String, indices: &Vec<usize>| {
1600                             let mut indices = indices.iter().cloned();
1601                             if let Some(first) = indices.next() {
1602                                 write!(s, "`{}`", substs[first]).unwrap();
1603                                 for i in indices {
1604                                     write!(s, ", `{}`", substs[i]).unwrap();
1605                                 }
1606                             }
1607                         };
1608                         list(&mut s, prev_indices);
1609                         write!(s, "], got [").unwrap();
1610                         list(&mut s, &indices);
1611                         write!(s, "]").unwrap();
1612                         err.span_label(span, s);
1613                         err.span_note(prev_span, "previous use here");
1614                         err.emit();
1615                     }
1616                 } else {
1617                     self.found = Some((span, concrete_type, indices));
1618                 }
1619             } else {
1620                 debug!(
1621                     "find_existential_constraints: no constraint for `{:?}` at `{:?}`",
1622                     self.def_id,
1623                     def_id,
1624                 );
1625             }
1626         }
1627     }
1628
1629     impl<'tcx> intravisit::Visitor<'tcx> for ConstraintLocator<'tcx> {
1630         fn nested_visit_map<'this>(&'this mut self) -> intravisit::NestedVisitorMap<'this, 'tcx> {
1631             intravisit::NestedVisitorMap::All(&self.tcx.hir())
1632         }
1633         fn visit_item(&mut self, it: &'tcx Item) {
1634             let def_id = self.tcx.hir().local_def_id_from_hir_id(it.hir_id);
1635             // The existential type itself or its children are not within its reveal scope.
1636             if def_id != self.def_id {
1637                 self.check(def_id);
1638                 intravisit::walk_item(self, it);
1639             }
1640         }
1641         fn visit_impl_item(&mut self, it: &'tcx ImplItem) {
1642             let def_id = self.tcx.hir().local_def_id_from_hir_id(it.hir_id);
1643             // The existential type itself or its children are not within its reveal scope.
1644             if def_id != self.def_id {
1645                 self.check(def_id);
1646                 intravisit::walk_impl_item(self, it);
1647             }
1648         }
1649         fn visit_trait_item(&mut self, it: &'tcx TraitItem) {
1650             let def_id = self.tcx.hir().local_def_id_from_hir_id(it.hir_id);
1651             self.check(def_id);
1652             intravisit::walk_trait_item(self, it);
1653         }
1654     }
1655
1656     let hir_id = tcx.hir().as_local_hir_id(def_id).unwrap();
1657     let scope = tcx.hir()
1658         .get_defining_scope(hir_id)
1659         .expect("could not get defining scope");
1660     let mut locator = ConstraintLocator {
1661         def_id,
1662         tcx,
1663         found: None,
1664     };
1665
1666     debug!("find_existential_constraints: scope={:?}", scope);
1667
1668     if scope == hir::CRATE_HIR_ID {
1669         intravisit::walk_crate(&mut locator, tcx.hir().krate());
1670     } else {
1671         debug!("find_existential_constraints: scope={:?}", tcx.hir().get_by_hir_id(scope));
1672         match tcx.hir().get_by_hir_id(scope) {
1673             Node::Item(ref it) => intravisit::walk_item(&mut locator, it),
1674             Node::ImplItem(ref it) => intravisit::walk_impl_item(&mut locator, it),
1675             Node::TraitItem(ref it) => intravisit::walk_trait_item(&mut locator, it),
1676             other => bug!(
1677                 "{:?} is not a valid scope for an existential type item",
1678                 other
1679             ),
1680         }
1681     }
1682
1683     match locator.found {
1684         Some((_, ty, _)) => ty,
1685         None => {
1686             let span = tcx.def_span(def_id);
1687             tcx.sess.span_err(span, "could not find defining uses");
1688             tcx.types.err
1689         }
1690     }
1691 }
1692
1693 fn fn_sig<'tcx>(tcx: TyCtxt<'tcx, 'tcx>, def_id: DefId) -> ty::PolyFnSig<'tcx> {
1694     use rustc::hir::*;
1695     use rustc::hir::Node::*;
1696
1697     let hir_id = tcx.hir().as_local_hir_id(def_id).unwrap();
1698
1699     let icx = ItemCtxt::new(tcx, def_id);
1700
1701     match tcx.hir().get_by_hir_id(hir_id) {
1702         TraitItem(hir::TraitItem {
1703             node: TraitItemKind::Method(sig, _),
1704             ..
1705         })
1706         | ImplItem(hir::ImplItem {
1707             node: ImplItemKind::Method(sig, _),
1708             ..
1709         }) => AstConv::ty_of_fn(&icx, sig.header.unsafety, sig.header.abi, &sig.decl),
1710
1711         Item(hir::Item {
1712             node: ItemKind::Fn(decl, header, _, _),
1713             ..
1714         }) => AstConv::ty_of_fn(&icx, header.unsafety, header.abi, decl),
1715
1716         ForeignItem(&hir::ForeignItem {
1717             node: ForeignItemKind::Fn(ref fn_decl, _, _),
1718             ..
1719         }) => {
1720             let abi = tcx.hir().get_foreign_abi_by_hir_id(hir_id);
1721             compute_sig_of_foreign_fn_decl(tcx, def_id, fn_decl, abi)
1722         }
1723
1724         Ctor(data) | Variant(Spanned {
1725             node: hir::VariantKind { data, ..  },
1726             ..
1727         }) if data.ctor_hir_id().is_some() => {
1728             let ty = tcx.type_of(tcx.hir().get_parent_did_by_hir_id(hir_id));
1729             let inputs = data.fields()
1730                 .iter()
1731                 .map(|f| tcx.type_of(tcx.hir().local_def_id_from_hir_id(f.hir_id)));
1732             ty::Binder::bind(tcx.mk_fn_sig(
1733                 inputs,
1734                 ty,
1735                 false,
1736                 hir::Unsafety::Normal,
1737                 abi::Abi::Rust,
1738             ))
1739         }
1740
1741         Expr(&hir::Expr {
1742             node: hir::ExprKind::Closure(..),
1743             ..
1744         }) => {
1745             // Closure signatures are not like other function
1746             // signatures and cannot be accessed through `fn_sig`. For
1747             // example, a closure signature excludes the `self`
1748             // argument. In any case they are embedded within the
1749             // closure type as part of the `ClosureSubsts`.
1750             //
1751             // To get
1752             // the signature of a closure, you should use the
1753             // `closure_sig` method on the `ClosureSubsts`:
1754             //
1755             //    closure_substs.closure_sig(def_id, tcx)
1756             //
1757             // or, inside of an inference context, you can use
1758             //
1759             //    infcx.closure_sig(def_id, closure_substs)
1760             bug!("to get the signature of a closure, use `closure_sig()` not `fn_sig()`");
1761         }
1762
1763         x => {
1764             bug!("unexpected sort of node in fn_sig(): {:?}", x);
1765         }
1766     }
1767 }
1768
1769 fn impl_trait_ref<'tcx>(tcx: TyCtxt<'tcx, 'tcx>, def_id: DefId) -> Option<ty::TraitRef<'tcx>> {
1770     let icx = ItemCtxt::new(tcx, def_id);
1771
1772     let hir_id = tcx.hir().as_local_hir_id(def_id).unwrap();
1773     match tcx.hir().expect_item_by_hir_id(hir_id).node {
1774         hir::ItemKind::Impl(.., ref opt_trait_ref, _, _) => {
1775             opt_trait_ref.as_ref().map(|ast_trait_ref| {
1776                 let selfty = tcx.type_of(def_id);
1777                 AstConv::instantiate_mono_trait_ref(&icx, ast_trait_ref, selfty)
1778             })
1779         }
1780         _ => bug!(),
1781     }
1782 }
1783
1784 fn impl_polarity<'tcx>(tcx: TyCtxt<'tcx, 'tcx>, def_id: DefId) -> hir::ImplPolarity {
1785     let hir_id = tcx.hir().as_local_hir_id(def_id).unwrap();
1786     match tcx.hir().expect_item_by_hir_id(hir_id).node {
1787         hir::ItemKind::Impl(_, polarity, ..) => polarity,
1788         ref item => bug!("impl_polarity: {:?} not an impl", item),
1789     }
1790 }
1791
1792 /// Returns the early-bound lifetimes declared in this generics
1793 /// listing. For anything other than fns/methods, this is just all
1794 /// the lifetimes that are declared. For fns or methods, we have to
1795 /// screen out those that do not appear in any where-clauses etc using
1796 /// `resolve_lifetime::early_bound_lifetimes`.
1797 fn early_bound_lifetimes_from_generics<'a, 'tcx: 'a>(
1798     tcx: TyCtxt<'tcx, 'tcx>,
1799     generics: &'a hir::Generics,
1800 ) -> impl Iterator<Item = &'a hir::GenericParam> + Captures<'tcx> {
1801     generics
1802         .params
1803         .iter()
1804         .filter(move |param| match param.kind {
1805             GenericParamKind::Lifetime { .. } => {
1806                 !tcx.is_late_bound(param.hir_id)
1807             }
1808             _ => false,
1809         })
1810 }
1811
1812 /// Returns a list of type predicates for the definition with ID `def_id`, including inferred
1813 /// lifetime constraints. This includes all predicates returned by `explicit_predicates_of`, plus
1814 /// inferred constraints concerning which regions outlive other regions.
1815 fn predicates_defined_on<'tcx>(
1816     tcx: TyCtxt<'tcx, 'tcx>,
1817     def_id: DefId,
1818 ) -> &'tcx ty::GenericPredicates<'tcx> {
1819     debug!("predicates_defined_on({:?})", def_id);
1820     let mut result = tcx.explicit_predicates_of(def_id);
1821     debug!(
1822         "predicates_defined_on: explicit_predicates_of({:?}) = {:?}",
1823         def_id,
1824         result,
1825     );
1826     let inferred_outlives = tcx.inferred_outlives_of(def_id);
1827     if !inferred_outlives.is_empty() {
1828         let span = tcx.def_span(def_id);
1829         debug!(
1830             "predicates_defined_on: inferred_outlives_of({:?}) = {:?}",
1831             def_id,
1832             inferred_outlives,
1833         );
1834         let mut predicates = (*result).clone();
1835         predicates.predicates.extend(inferred_outlives.iter().map(|&p| (p, span)));
1836         result = tcx.arena.alloc(predicates);
1837     }
1838     debug!("predicates_defined_on({:?}) = {:?}", def_id, result);
1839     result
1840 }
1841
1842 /// Returns a list of all type predicates (explicit and implicit) for the definition with
1843 /// ID `def_id`. This includes all predicates returned by `predicates_defined_on`, plus
1844 /// `Self: Trait` predicates for traits.
1845 fn predicates_of<'tcx>(
1846     tcx: TyCtxt<'tcx, 'tcx>,
1847     def_id: DefId,
1848 ) -> &'tcx ty::GenericPredicates<'tcx> {
1849     let mut result = tcx.predicates_defined_on(def_id);
1850
1851     if tcx.is_trait(def_id) {
1852         // For traits, add `Self: Trait` predicate. This is
1853         // not part of the predicates that a user writes, but it
1854         // is something that one must prove in order to invoke a
1855         // method or project an associated type.
1856         //
1857         // In the chalk setup, this predicate is not part of the
1858         // "predicates" for a trait item. But it is useful in
1859         // rustc because if you directly (e.g.) invoke a trait
1860         // method like `Trait::method(...)`, you must naturally
1861         // prove that the trait applies to the types that were
1862         // used, and adding the predicate into this list ensures
1863         // that this is done.
1864         let span = tcx.def_span(def_id);
1865         let mut predicates = (*result).clone();
1866         predicates.predicates.push((ty::TraitRef::identity(tcx, def_id).to_predicate(), span));
1867         result = tcx.arena.alloc(predicates);
1868     }
1869     debug!("predicates_of(def_id={:?}) = {:?}", def_id, result);
1870     result
1871 }
1872
1873 /// Returns a list of user-specified type predicates for the definition with ID `def_id`.
1874 /// N.B., this does not include any implied/inferred constraints.
1875 fn explicit_predicates_of<'tcx>(
1876     tcx: TyCtxt<'tcx, 'tcx>,
1877     def_id: DefId,
1878 ) -> &'tcx ty::GenericPredicates<'tcx> {
1879     use rustc::hir::*;
1880     use rustc_data_structures::fx::FxHashSet;
1881
1882     debug!("explicit_predicates_of(def_id={:?})", def_id);
1883
1884     /// A data structure with unique elements, which preserves order of insertion.
1885     /// Preserving the order of insertion is important here so as not to break
1886     /// compile-fail UI tests.
1887     struct UniquePredicates<'tcx> {
1888         predicates: Vec<(ty::Predicate<'tcx>, Span)>,
1889         uniques: FxHashSet<(ty::Predicate<'tcx>, Span)>,
1890     }
1891
1892     impl<'tcx> UniquePredicates<'tcx> {
1893         fn new() -> Self {
1894             UniquePredicates {
1895                 predicates: vec![],
1896                 uniques: FxHashSet::default(),
1897             }
1898         }
1899
1900         fn push(&mut self, value: (ty::Predicate<'tcx>, Span)) {
1901             if self.uniques.insert(value) {
1902                 self.predicates.push(value);
1903             }
1904         }
1905
1906         fn extend<I: IntoIterator<Item = (ty::Predicate<'tcx>, Span)>>(&mut self, iter: I) {
1907             for value in iter {
1908                 self.push(value);
1909             }
1910         }
1911     }
1912
1913     let hir_id = match tcx.hir().as_local_hir_id(def_id) {
1914         Some(hir_id) => hir_id,
1915         None => return tcx.predicates_of(def_id),
1916     };
1917     let node = tcx.hir().get_by_hir_id(hir_id);
1918
1919     let mut is_trait = None;
1920     let mut is_default_impl_trait = None;
1921
1922     let icx = ItemCtxt::new(tcx, def_id);
1923     let no_generics = hir::Generics::empty();
1924     let empty_trait_items = HirVec::new();
1925
1926     let mut predicates = UniquePredicates::new();
1927
1928     let ast_generics = match node {
1929         Node::TraitItem(item) => &item.generics,
1930
1931         Node::ImplItem(item) => match item.node {
1932             ImplItemKind::Existential(ref bounds) => {
1933                 let substs = InternalSubsts::identity_for_item(tcx, def_id);
1934                 let opaque_ty = tcx.mk_opaque(def_id, substs);
1935
1936                 // Collect the bounds, i.e., the `A + B + 'c` in `impl A + B + 'c`.
1937                 let bounds = AstConv::compute_bounds(
1938                     &icx,
1939                     opaque_ty,
1940                     bounds,
1941                     SizedByDefault::Yes,
1942                     tcx.def_span(def_id),
1943                 );
1944
1945                 predicates.extend(bounds.predicates(tcx, opaque_ty));
1946                 &item.generics
1947             }
1948             _ => &item.generics,
1949         },
1950
1951         Node::Item(item) => {
1952             match item.node {
1953                 ItemKind::Impl(_, _, defaultness, ref generics, ..) => {
1954                     if defaultness.is_default() {
1955                         is_default_impl_trait = tcx.impl_trait_ref(def_id);
1956                     }
1957                     generics
1958                 }
1959                 ItemKind::Fn(.., ref generics, _)
1960                 | ItemKind::Ty(_, ref generics)
1961                 | ItemKind::Enum(_, ref generics)
1962                 | ItemKind::Struct(_, ref generics)
1963                 | ItemKind::Union(_, ref generics) => generics,
1964
1965                 ItemKind::Trait(_, _, ref generics, .., ref items) => {
1966                     is_trait = Some((ty::TraitRef::identity(tcx, def_id), items));
1967                     generics
1968                 }
1969                 ItemKind::TraitAlias(ref generics, _) => {
1970                     is_trait = Some((ty::TraitRef::identity(tcx, def_id), &empty_trait_items));
1971                     generics
1972                 }
1973                 ItemKind::Existential(ExistTy {
1974                     ref bounds,
1975                     impl_trait_fn,
1976                     ref generics,
1977                     origin: _,
1978                 }) => {
1979                     let substs = InternalSubsts::identity_for_item(tcx, def_id);
1980                     let opaque_ty = tcx.mk_opaque(def_id, substs);
1981
1982                     // Collect the bounds, i.e., the `A + B + 'c` in `impl A + B + 'c`.
1983                     let bounds = AstConv::compute_bounds(
1984                         &icx,
1985                         opaque_ty,
1986                         bounds,
1987                         SizedByDefault::Yes,
1988                         tcx.def_span(def_id),
1989                     );
1990
1991                     let bounds_predicates = bounds.predicates(tcx, opaque_ty);
1992                     if impl_trait_fn.is_some() {
1993                         // opaque types
1994                         return tcx.arena.alloc(ty::GenericPredicates {
1995                             parent: None,
1996                             predicates: bounds_predicates,
1997                         });
1998                     } else {
1999                         // named existential types
2000                         predicates.extend(bounds_predicates);
2001                         generics
2002                     }
2003                 }
2004
2005                 _ => &no_generics,
2006             }
2007         }
2008
2009         Node::ForeignItem(item) => match item.node {
2010             ForeignItemKind::Static(..) => &no_generics,
2011             ForeignItemKind::Fn(_, _, ref generics) => generics,
2012             ForeignItemKind::Type => &no_generics,
2013         },
2014
2015         _ => &no_generics,
2016     };
2017
2018     let generics = tcx.generics_of(def_id);
2019     let parent_count = generics.parent_count as u32;
2020     let has_own_self = generics.has_self && parent_count == 0;
2021
2022     // Below we'll consider the bounds on the type parameters (including `Self`)
2023     // and the explicit where-clauses, but to get the full set of predicates
2024     // on a trait we need to add in the supertrait bounds and bounds found on
2025     // associated types.
2026     if let Some((_trait_ref, _)) = is_trait {
2027         predicates.extend(tcx.super_predicates_of(def_id).predicates.iter().cloned());
2028     }
2029
2030     // In default impls, we can assume that the self type implements
2031     // the trait. So in:
2032     //
2033     //     default impl Foo for Bar { .. }
2034     //
2035     // we add a default where clause `Foo: Bar`. We do a similar thing for traits
2036     // (see below). Recall that a default impl is not itself an impl, but rather a
2037     // set of defaults that can be incorporated into another impl.
2038     if let Some(trait_ref) = is_default_impl_trait {
2039         predicates.push((trait_ref.to_poly_trait_ref().to_predicate(), tcx.def_span(def_id)));
2040     }
2041
2042     // Collect the region predicates that were declared inline as
2043     // well. In the case of parameters declared on a fn or method, we
2044     // have to be careful to only iterate over early-bound regions.
2045     let mut index = parent_count + has_own_self as u32;
2046     for param in early_bound_lifetimes_from_generics(tcx, ast_generics) {
2047         let region = tcx.mk_region(ty::ReEarlyBound(ty::EarlyBoundRegion {
2048             def_id: tcx.hir().local_def_id_from_hir_id(param.hir_id),
2049             index,
2050             name: param.name.ident().as_interned_str(),
2051         }));
2052         index += 1;
2053
2054         match param.kind {
2055             GenericParamKind::Lifetime { .. } => {
2056                 param.bounds.iter().for_each(|bound| match bound {
2057                     hir::GenericBound::Outlives(lt) => {
2058                         let bound = AstConv::ast_region_to_region(&icx, &lt, None);
2059                         let outlives = ty::Binder::bind(ty::OutlivesPredicate(region, bound));
2060                         predicates.push((outlives.to_predicate(), lt.span));
2061                     }
2062                     _ => bug!(),
2063                 });
2064             }
2065             _ => bug!(),
2066         }
2067     }
2068
2069     // Collect the predicates that were written inline by the user on each
2070     // type parameter (e.g., `<T: Foo>`).
2071     for param in &ast_generics.params {
2072         if let GenericParamKind::Type { .. } = param.kind {
2073             let name = param.name.ident().as_interned_str();
2074             let param_ty = ty::ParamTy::new(index, name).to_ty(tcx);
2075             index += 1;
2076
2077             let sized = SizedByDefault::Yes;
2078             let bounds = AstConv::compute_bounds(&icx, param_ty, &param.bounds, sized, param.span);
2079             predicates.extend(bounds.predicates(tcx, param_ty));
2080         }
2081     }
2082
2083     // Add in the bounds that appear in the where-clause.
2084     let where_clause = &ast_generics.where_clause;
2085     for predicate in &where_clause.predicates {
2086         match predicate {
2087             &hir::WherePredicate::BoundPredicate(ref bound_pred) => {
2088                 let ty = icx.to_ty(&bound_pred.bounded_ty);
2089
2090                 // Keep the type around in a dummy predicate, in case of no bounds.
2091                 // That way, `where Ty:` is not a complete noop (see #53696) and `Ty`
2092                 // is still checked for WF.
2093                 if bound_pred.bounds.is_empty() {
2094                     if let ty::Param(_) = ty.sty {
2095                         // This is a `where T:`, which can be in the HIR from the
2096                         // transformation that moves `?Sized` to `T`'s declaration.
2097                         // We can skip the predicate because type parameters are
2098                         // trivially WF, but also we *should*, to avoid exposing
2099                         // users who never wrote `where Type:,` themselves, to
2100                         // compiler/tooling bugs from not handling WF predicates.
2101                     } else {
2102                         let span = bound_pred.bounded_ty.span;
2103                         let predicate = ty::OutlivesPredicate(ty, tcx.mk_region(ty::ReEmpty));
2104                         predicates.push(
2105                             (ty::Predicate::TypeOutlives(ty::Binder::dummy(predicate)), span)
2106                         );
2107                     }
2108                 }
2109
2110                 for bound in bound_pred.bounds.iter() {
2111                     match bound {
2112                         &hir::GenericBound::Trait(ref poly_trait_ref, _) => {
2113                             let mut bounds = Bounds::default();
2114
2115                             let (trait_ref, _) = AstConv::instantiate_poly_trait_ref(
2116                                 &icx,
2117                                 poly_trait_ref,
2118                                 ty,
2119                                 &mut bounds,
2120                             );
2121
2122                             predicates.push((trait_ref.to_predicate(), poly_trait_ref.span));
2123                             predicates.extend(bounds.predicates(tcx, ty));
2124                         }
2125
2126                         &hir::GenericBound::Outlives(ref lifetime) => {
2127                             let region = AstConv::ast_region_to_region(&icx, lifetime, None);
2128                             let pred = ty::Binder::bind(ty::OutlivesPredicate(ty, region));
2129                             predicates.push((ty::Predicate::TypeOutlives(pred), lifetime.span))
2130                         }
2131                     }
2132                 }
2133             }
2134
2135             &hir::WherePredicate::RegionPredicate(ref region_pred) => {
2136                 let r1 = AstConv::ast_region_to_region(&icx, &region_pred.lifetime, None);
2137                 predicates.extend(region_pred.bounds.iter().map(|bound| {
2138                     let (r2, span) = match bound {
2139                         hir::GenericBound::Outlives(lt) => {
2140                             (AstConv::ast_region_to_region(&icx, lt, None), lt.span)
2141                         }
2142                         _ => bug!(),
2143                     };
2144                     let pred = ty::Binder::bind(ty::OutlivesPredicate(r1, r2));
2145
2146                     (ty::Predicate::RegionOutlives(pred), span)
2147                 }))
2148             }
2149
2150             &hir::WherePredicate::EqPredicate(..) => {
2151                 // FIXME(#20041)
2152             }
2153         }
2154     }
2155
2156     // Add predicates from associated type bounds.
2157     if let Some((self_trait_ref, trait_items)) = is_trait {
2158         predicates.extend(trait_items.iter().flat_map(|trait_item_ref| {
2159             let trait_item = tcx.hir().trait_item(trait_item_ref.id);
2160             let bounds = match trait_item.node {
2161                 hir::TraitItemKind::Type(ref bounds, _) => bounds,
2162                 _ => return Vec::new().into_iter()
2163             };
2164
2165             let assoc_ty =
2166                 tcx.mk_projection(tcx.hir().local_def_id_from_hir_id(trait_item.hir_id),
2167                     self_trait_ref.substs);
2168
2169             let bounds = AstConv::compute_bounds(
2170                 &ItemCtxt::new(tcx, def_id),
2171                 assoc_ty,
2172                 bounds,
2173                 SizedByDefault::Yes,
2174                 trait_item.span,
2175             );
2176
2177             bounds.predicates(tcx, assoc_ty).into_iter()
2178         }))
2179     }
2180
2181     let mut predicates = predicates.predicates;
2182
2183     // Subtle: before we store the predicates into the tcx, we
2184     // sort them so that predicates like `T: Foo<Item=U>` come
2185     // before uses of `U`.  This avoids false ambiguity errors
2186     // in trait checking. See `setup_constraining_predicates`
2187     // for details.
2188     if let Node::Item(&Item {
2189         node: ItemKind::Impl(..),
2190         ..
2191     }) = node
2192     {
2193         let self_ty = tcx.type_of(def_id);
2194         let trait_ref = tcx.impl_trait_ref(def_id);
2195         cgp::setup_constraining_predicates(
2196             tcx,
2197             &mut predicates,
2198             trait_ref,
2199             &mut cgp::parameters_for_impl(self_ty, trait_ref),
2200         );
2201     }
2202
2203     let result = tcx.arena.alloc(ty::GenericPredicates {
2204         parent: generics.parent,
2205         predicates,
2206     });
2207     debug!("explicit_predicates_of(def_id={:?}) = {:?}", def_id, result);
2208     result
2209 }
2210
2211 /// Converts a specific `GenericBound` from the AST into a set of
2212 /// predicates that apply to the self type. A vector is returned
2213 /// because this can be anywhere from zero predicates (`T: ?Sized` adds no
2214 /// predicates) to one (`T: Foo`) to many (`T: Bar<X=i32>` adds `T: Bar`
2215 /// and `<T as Bar>::X == i32`).
2216 fn predicates_from_bound<'tcx>(
2217     astconv: &dyn AstConv<'tcx, 'tcx>,
2218     param_ty: Ty<'tcx>,
2219     bound: &hir::GenericBound,
2220 ) -> Vec<(ty::Predicate<'tcx>, Span)> {
2221     match *bound {
2222         hir::GenericBound::Trait(ref tr, hir::TraitBoundModifier::None) => {
2223             let mut bounds = Bounds::default();
2224             let (pred, _) = astconv.instantiate_poly_trait_ref(tr, param_ty, &mut bounds);
2225             iter::once((pred.to_predicate(), tr.span))
2226                 .chain(bounds.predicates(astconv.tcx(), param_ty))
2227                 .collect()
2228         }
2229         hir::GenericBound::Outlives(ref lifetime) => {
2230             let region = astconv.ast_region_to_region(lifetime, None);
2231             let pred = ty::Binder::bind(ty::OutlivesPredicate(param_ty, region));
2232             vec![(ty::Predicate::TypeOutlives(pred), lifetime.span)]
2233         }
2234         hir::GenericBound::Trait(_, hir::TraitBoundModifier::Maybe) => vec![],
2235     }
2236 }
2237
2238 fn compute_sig_of_foreign_fn_decl<'tcx>(
2239     tcx: TyCtxt<'tcx, 'tcx>,
2240     def_id: DefId,
2241     decl: &hir::FnDecl,
2242     abi: abi::Abi,
2243 ) -> ty::PolyFnSig<'tcx> {
2244     let unsafety = if abi == abi::Abi::RustIntrinsic {
2245         intrisic_operation_unsafety(&*tcx.item_name(def_id).as_str())
2246     } else {
2247         hir::Unsafety::Unsafe
2248     };
2249     let fty = AstConv::ty_of_fn(&ItemCtxt::new(tcx, def_id), unsafety, abi, decl);
2250
2251     // Feature gate SIMD types in FFI, since I am not sure that the
2252     // ABIs are handled at all correctly. -huonw
2253     if abi != abi::Abi::RustIntrinsic
2254         && abi != abi::Abi::PlatformIntrinsic
2255         && !tcx.features().simd_ffi
2256     {
2257         let check = |ast_ty: &hir::Ty, ty: Ty<'_>| {
2258             if ty.is_simd() {
2259                 tcx.sess
2260                    .struct_span_err(
2261                        ast_ty.span,
2262                        &format!(
2263                            "use of SIMD type `{}` in FFI is highly experimental and \
2264                             may result in invalid code",
2265                            tcx.hir().hir_to_pretty_string(ast_ty.hir_id)
2266                        ),
2267                    )
2268                    .help("add #![feature(simd_ffi)] to the crate attributes to enable")
2269                    .emit();
2270             }
2271         };
2272         for (input, ty) in decl.inputs.iter().zip(*fty.inputs().skip_binder()) {
2273             check(&input, ty)
2274         }
2275         if let hir::Return(ref ty) = decl.output {
2276             check(&ty, *fty.output().skip_binder())
2277         }
2278     }
2279
2280     fty
2281 }
2282
2283 fn is_foreign_item<'tcx>(tcx: TyCtxt<'tcx, 'tcx>, def_id: DefId) -> bool {
2284     match tcx.hir().get_if_local(def_id) {
2285         Some(Node::ForeignItem(..)) => true,
2286         Some(_) => false,
2287         _ => bug!("is_foreign_item applied to non-local def-id {:?}", def_id),
2288     }
2289 }
2290
2291 fn static_mutability<'tcx>(tcx: TyCtxt<'tcx, 'tcx>, def_id: DefId) -> Option<hir::Mutability> {
2292     match tcx.hir().get_if_local(def_id) {
2293         Some(Node::Item(&hir::Item {
2294             node: hir::ItemKind::Static(_, mutbl, _), ..
2295         })) |
2296         Some(Node::ForeignItem( &hir::ForeignItem {
2297             node: hir::ForeignItemKind::Static(_, mutbl), ..
2298         })) => Some(mutbl),
2299         Some(_) => None,
2300         _ => bug!("static_mutability applied to non-local def-id {:?}", def_id),
2301     }
2302 }
2303
2304 fn from_target_feature(
2305     tcx: TyCtxt<'_, '_>,
2306     id: DefId,
2307     attr: &ast::Attribute,
2308     whitelist: &FxHashMap<String, Option<Symbol>>,
2309     target_features: &mut Vec<Symbol>,
2310 ) {
2311     let list = match attr.meta_item_list() {
2312         Some(list) => list,
2313         None => return,
2314     };
2315     let bad_item = |span| {
2316         let msg = "malformed `target_feature` attribute input";
2317         let code = "enable = \"..\"".to_owned();
2318         tcx.sess.struct_span_err(span, &msg)
2319             .span_suggestion(span, "must be of the form", code, Applicability::HasPlaceholders)
2320             .emit();
2321     };
2322     let rust_features = tcx.features();
2323     for item in list {
2324         // Only `enable = ...` is accepted in the meta-item list.
2325         if !item.check_name(sym::enable) {
2326             bad_item(item.span());
2327             continue;
2328         }
2329
2330         // Must be of the form `enable = "..."` (a string).
2331         let value = match item.value_str() {
2332             Some(value) => value,
2333             None => {
2334                 bad_item(item.span());
2335                 continue;
2336             }
2337         };
2338
2339         // We allow comma separation to enable multiple features.
2340         target_features.extend(value.as_str().split(',').filter_map(|feature| {
2341             // Only allow whitelisted features per platform.
2342             let feature_gate = match whitelist.get(feature) {
2343                 Some(g) => g,
2344                 None => {
2345                     let msg = format!(
2346                         "the feature named `{}` is not valid for this target",
2347                         feature
2348                     );
2349                     let mut err = tcx.sess.struct_span_err(item.span(), &msg);
2350                     err.span_label(
2351                         item.span(),
2352                         format!("`{}` is not valid for this target", feature),
2353                     );
2354                     if feature.starts_with("+") {
2355                         let valid = whitelist.contains_key(&feature[1..]);
2356                         if valid {
2357                             err.help("consider removing the leading `+` in the feature name");
2358                         }
2359                     }
2360                     err.emit();
2361                     return None;
2362                 }
2363             };
2364
2365             // Only allow features whose feature gates have been enabled.
2366             let allowed = match feature_gate.as_ref().map(|s| *s) {
2367                 Some(sym::arm_target_feature) => rust_features.arm_target_feature,
2368                 Some(sym::aarch64_target_feature) => rust_features.aarch64_target_feature,
2369                 Some(sym::hexagon_target_feature) => rust_features.hexagon_target_feature,
2370                 Some(sym::powerpc_target_feature) => rust_features.powerpc_target_feature,
2371                 Some(sym::mips_target_feature) => rust_features.mips_target_feature,
2372                 Some(sym::avx512_target_feature) => rust_features.avx512_target_feature,
2373                 Some(sym::mmx_target_feature) => rust_features.mmx_target_feature,
2374                 Some(sym::sse4a_target_feature) => rust_features.sse4a_target_feature,
2375                 Some(sym::tbm_target_feature) => rust_features.tbm_target_feature,
2376                 Some(sym::wasm_target_feature) => rust_features.wasm_target_feature,
2377                 Some(sym::cmpxchg16b_target_feature) => rust_features.cmpxchg16b_target_feature,
2378                 Some(sym::adx_target_feature) => rust_features.adx_target_feature,
2379                 Some(sym::movbe_target_feature) => rust_features.movbe_target_feature,
2380                 Some(sym::rtm_target_feature) => rust_features.rtm_target_feature,
2381                 Some(sym::f16c_target_feature) => rust_features.f16c_target_feature,
2382                 Some(name) => bug!("unknown target feature gate {}", name),
2383                 None => true,
2384             };
2385             if !allowed && id.is_local() {
2386                 feature_gate::emit_feature_err(
2387                     &tcx.sess.parse_sess,
2388                     feature_gate.unwrap(),
2389                     item.span(),
2390                     feature_gate::GateIssue::Language,
2391                     &format!("the target feature `{}` is currently unstable", feature),
2392                 );
2393             }
2394             Some(Symbol::intern(feature))
2395         }));
2396     }
2397 }
2398
2399 fn linkage_by_name<'tcx>(tcx: TyCtxt<'tcx, 'tcx>, def_id: DefId, name: &str) -> Linkage {
2400     use rustc::mir::mono::Linkage::*;
2401
2402     // Use the names from src/llvm/docs/LangRef.rst here. Most types are only
2403     // applicable to variable declarations and may not really make sense for
2404     // Rust code in the first place but whitelist them anyway and trust that
2405     // the user knows what s/he's doing. Who knows, unanticipated use cases
2406     // may pop up in the future.
2407     //
2408     // ghost, dllimport, dllexport and linkonce_odr_autohide are not supported
2409     // and don't have to be, LLVM treats them as no-ops.
2410     match name {
2411         "appending" => Appending,
2412         "available_externally" => AvailableExternally,
2413         "common" => Common,
2414         "extern_weak" => ExternalWeak,
2415         "external" => External,
2416         "internal" => Internal,
2417         "linkonce" => LinkOnceAny,
2418         "linkonce_odr" => LinkOnceODR,
2419         "private" => Private,
2420         "weak" => WeakAny,
2421         "weak_odr" => WeakODR,
2422         _ => {
2423             let span = tcx.hir().span_if_local(def_id);
2424             if let Some(span) = span {
2425                 tcx.sess.span_fatal(span, "invalid linkage specified")
2426             } else {
2427                 tcx.sess
2428                    .fatal(&format!("invalid linkage specified: {}", name))
2429             }
2430         }
2431     }
2432 }
2433
2434 fn codegen_fn_attrs<'tcx>(tcx: TyCtxt<'tcx, 'tcx>, id: DefId) -> CodegenFnAttrs {
2435     let attrs = tcx.get_attrs(id);
2436
2437     let mut codegen_fn_attrs = CodegenFnAttrs::new();
2438
2439     let whitelist = tcx.target_features_whitelist(LOCAL_CRATE);
2440
2441     let mut inline_span = None;
2442     for attr in attrs.iter() {
2443         if attr.check_name(sym::cold) {
2444             codegen_fn_attrs.flags |= CodegenFnAttrFlags::COLD;
2445         } else if attr.check_name(sym::rustc_allocator) {
2446             codegen_fn_attrs.flags |= CodegenFnAttrFlags::ALLOCATOR;
2447         } else if attr.check_name(sym::unwind) {
2448             codegen_fn_attrs.flags |= CodegenFnAttrFlags::UNWIND;
2449         } else if attr.check_name(sym::ffi_returns_twice) {
2450             if tcx.is_foreign_item(id) {
2451                 codegen_fn_attrs.flags |= CodegenFnAttrFlags::FFI_RETURNS_TWICE;
2452             } else {
2453                 // `#[ffi_returns_twice]` is only allowed `extern fn`s.
2454                 struct_span_err!(
2455                     tcx.sess,
2456                     attr.span,
2457                     E0724,
2458                     "`#[ffi_returns_twice]` may only be used on foreign functions"
2459                 ).emit();
2460             }
2461         } else if attr.check_name(sym::rustc_allocator_nounwind) {
2462             codegen_fn_attrs.flags |= CodegenFnAttrFlags::RUSTC_ALLOCATOR_NOUNWIND;
2463         } else if attr.check_name(sym::naked) {
2464             codegen_fn_attrs.flags |= CodegenFnAttrFlags::NAKED;
2465         } else if attr.check_name(sym::no_mangle) {
2466             codegen_fn_attrs.flags |= CodegenFnAttrFlags::NO_MANGLE;
2467         } else if attr.check_name(sym::rustc_std_internal_symbol) {
2468             codegen_fn_attrs.flags |= CodegenFnAttrFlags::RUSTC_STD_INTERNAL_SYMBOL;
2469         } else if attr.check_name(sym::no_debug) {
2470             codegen_fn_attrs.flags |= CodegenFnAttrFlags::NO_DEBUG;
2471         } else if attr.check_name(sym::used) {
2472             codegen_fn_attrs.flags |= CodegenFnAttrFlags::USED;
2473         } else if attr.check_name(sym::thread_local) {
2474             codegen_fn_attrs.flags |= CodegenFnAttrFlags::THREAD_LOCAL;
2475         } else if attr.check_name(sym::export_name) {
2476             if let Some(s) = attr.value_str() {
2477                 if s.as_str().contains("\0") {
2478                     // `#[export_name = ...]` will be converted to a null-terminated string,
2479                     // so it may not contain any null characters.
2480                     struct_span_err!(
2481                         tcx.sess,
2482                         attr.span,
2483                         E0648,
2484                         "`export_name` may not contain null characters"
2485                     ).emit();
2486                 }
2487                 codegen_fn_attrs.export_name = Some(s);
2488             }
2489         } else if attr.check_name(sym::target_feature) {
2490             if tcx.fn_sig(id).unsafety() == Unsafety::Normal {
2491                 let msg = "#[target_feature(..)] can only be applied to `unsafe` functions";
2492                 tcx.sess.struct_span_err(attr.span, msg)
2493                     .span_label(attr.span, "can only be applied to `unsafe` functions")
2494                     .span_label(tcx.def_span(id), "not an `unsafe` function")
2495                     .emit();
2496             }
2497             from_target_feature(
2498                 tcx,
2499                 id,
2500                 attr,
2501                 &whitelist,
2502                 &mut codegen_fn_attrs.target_features,
2503             );
2504         } else if attr.check_name(sym::linkage) {
2505             if let Some(val) = attr.value_str() {
2506                 codegen_fn_attrs.linkage = Some(linkage_by_name(tcx, id, &val.as_str()));
2507             }
2508         } else if attr.check_name(sym::link_section) {
2509             if let Some(val) = attr.value_str() {
2510                 if val.as_str().bytes().any(|b| b == 0) {
2511                     let msg = format!(
2512                         "illegal null byte in link_section \
2513                          value: `{}`",
2514                         &val
2515                     );
2516                     tcx.sess.span_err(attr.span, &msg);
2517                 } else {
2518                     codegen_fn_attrs.link_section = Some(val);
2519                 }
2520             }
2521         } else if attr.check_name(sym::link_name) {
2522             codegen_fn_attrs.link_name = attr.value_str();
2523         }
2524     }
2525
2526     codegen_fn_attrs.inline = attrs.iter().fold(InlineAttr::None, |ia, attr| {
2527         if attr.path != sym::inline {
2528             return ia;
2529         }
2530         match attr.meta().map(|i| i.node) {
2531             Some(MetaItemKind::Word) => {
2532                 mark_used(attr);
2533                 InlineAttr::Hint
2534             }
2535             Some(MetaItemKind::List(ref items)) => {
2536                 mark_used(attr);
2537                 inline_span = Some(attr.span);
2538                 if items.len() != 1 {
2539                     span_err!(
2540                         tcx.sess.diagnostic(),
2541                         attr.span,
2542                         E0534,
2543                         "expected one argument"
2544                     );
2545                     InlineAttr::None
2546                 } else if list_contains_name(&items[..], sym::always) {
2547                     InlineAttr::Always
2548                 } else if list_contains_name(&items[..], sym::never) {
2549                     InlineAttr::Never
2550                 } else {
2551                     span_err!(
2552                         tcx.sess.diagnostic(),
2553                         items[0].span(),
2554                         E0535,
2555                         "invalid argument"
2556                     );
2557
2558                     InlineAttr::None
2559                 }
2560             }
2561             Some(MetaItemKind::NameValue(_)) => ia,
2562             None => ia,
2563         }
2564     });
2565
2566     codegen_fn_attrs.optimize = attrs.iter().fold(OptimizeAttr::None, |ia, attr| {
2567         if attr.path != sym::optimize {
2568             return ia;
2569         }
2570         let err = |sp, s| span_err!(tcx.sess.diagnostic(), sp, E0722, "{}", s);
2571         match attr.meta().map(|i| i.node) {
2572             Some(MetaItemKind::Word) => {
2573                 err(attr.span, "expected one argument");
2574                 ia
2575             }
2576             Some(MetaItemKind::List(ref items)) => {
2577                 mark_used(attr);
2578                 inline_span = Some(attr.span);
2579                 if items.len() != 1 {
2580                     err(attr.span, "expected one argument");
2581                     OptimizeAttr::None
2582                 } else if list_contains_name(&items[..], sym::size) {
2583                     OptimizeAttr::Size
2584                 } else if list_contains_name(&items[..], sym::speed) {
2585                     OptimizeAttr::Speed
2586                 } else {
2587                     err(items[0].span(), "invalid argument");
2588                     OptimizeAttr::None
2589                 }
2590             }
2591             Some(MetaItemKind::NameValue(_)) => ia,
2592             None => ia,
2593         }
2594     });
2595
2596     // If a function uses #[target_feature] it can't be inlined into general
2597     // purpose functions as they wouldn't have the right target features
2598     // enabled. For that reason we also forbid #[inline(always)] as it can't be
2599     // respected.
2600     if codegen_fn_attrs.target_features.len() > 0 {
2601         if codegen_fn_attrs.inline == InlineAttr::Always {
2602             if let Some(span) = inline_span {
2603                 tcx.sess.span_err(
2604                     span,
2605                     "cannot use #[inline(always)] with \
2606                      #[target_feature]",
2607                 );
2608             }
2609         }
2610     }
2611
2612     // Weak lang items have the same semantics as "std internal" symbols in the
2613     // sense that they're preserved through all our LTO passes and only
2614     // strippable by the linker.
2615     //
2616     // Additionally weak lang items have predetermined symbol names.
2617     if tcx.is_weak_lang_item(id) {
2618         codegen_fn_attrs.flags |= CodegenFnAttrFlags::RUSTC_STD_INTERNAL_SYMBOL;
2619     }
2620     if let Some(name) = weak_lang_items::link_name(&attrs) {
2621         codegen_fn_attrs.export_name = Some(name);
2622         codegen_fn_attrs.link_name = Some(name);
2623     }
2624
2625     // Internal symbols to the standard library all have no_mangle semantics in
2626     // that they have defined symbol names present in the function name. This
2627     // also applies to weak symbols where they all have known symbol names.
2628     if codegen_fn_attrs.flags.contains(CodegenFnAttrFlags::RUSTC_STD_INTERNAL_SYMBOL) {
2629         codegen_fn_attrs.flags |= CodegenFnAttrFlags::NO_MANGLE;
2630     }
2631
2632     codegen_fn_attrs
2633 }