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
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
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.
17 use crate::astconv::{AstConv, Bounds, SizedByDefault};
18 use crate::constrained_generic_params as cgp;
19 use crate::check::intrinsic::intrisic_operation_unsafety;
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;
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};
43 use rustc::hir::def::{CtorKind, Res, DefKind};
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};
50 use errors::{Applicability, DiagnosticId};
54 struct OnlySelfBounds(bool);
56 ///////////////////////////////////////////////////////////////////////////
59 fn collect_mod_item_types<'tcx>(tcx: TyCtxt<'tcx, 'tcx>, module_def_id: DefId) {
60 tcx.hir().visit_item_likes_in_module(
62 &mut CollectItemTypesVisitor { tcx }.as_deep_visitor()
66 pub fn provide(providers: &mut Providers<'_>) {
67 *providers = Providers {
71 predicates_defined_on,
72 explicit_predicates_of,
74 type_param_predicates,
83 collect_mod_item_types,
88 ///////////////////////////////////////////////////////////////////////////
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>,
104 ///////////////////////////////////////////////////////////////////////////
106 struct CollectItemTypesVisitor<'tcx> {
107 tcx: TyCtxt<'tcx, 'tcx>,
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())
115 fn visit_item(&mut self, item: &'tcx hir::Item) {
116 convert_item(self.tcx, item.hir_id);
117 intravisit::walk_item(self, item);
120 fn visit_generics(&mut self, generics: &'tcx hir::Generics) {
121 for param in &generics.params {
123 hir::GenericParamKind::Lifetime { .. } => {}
124 hir::GenericParamKind::Type {
127 let def_id = self.tcx.hir().local_def_id_from_hir_id(param.hir_id);
128 self.tcx.type_of(def_id);
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);
137 intravisit::walk_generics(self, generics);
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);
146 intravisit::walk_expr(self, expr);
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);
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);
160 ///////////////////////////////////////////////////////////////////////////
161 // Utility types and common code for the above passes.
163 impl ItemCtxt<'tcx> {
164 pub fn new(tcx: TyCtxt<'tcx, 'tcx>, item_def_id: DefId) -> ItemCtxt<'tcx> {
165 ItemCtxt { tcx, item_def_id }
168 pub fn to_ty(&self, ast_ty: &hir::Ty) -> Ty<'tcx> {
169 AstConv::ast_ty_to_ty(self, ast_ty)
173 impl AstConv<'tcx, 'tcx> for ItemCtxt<'tcx> {
174 fn tcx(&self) -> TyCtxt<'tcx, 'tcx> {
178 fn get_type_parameter_bounds(&self, span: Span, def_id: DefId)
179 -> &'tcx ty::GenericPredicates<'tcx> {
182 .type_param_predicates((self.item_def_id, def_id))
187 _: Option<&ty::GenericParamDef>,
189 ) -> Option<ty::Region<'tcx>> {
193 fn ty_infer(&self, _: Option<&ty::GenericParamDef>, span: Span) -> Ty<'tcx> {
194 self.tcx().sess.struct_span_err_with_code(
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")
207 _: Option<&ty::GenericParamDef>,
209 ) -> &'tcx Const<'tcx> {
210 self.tcx().sess.struct_span_err_with_code(
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")
217 self.tcx().consts.err
220 fn projected_ty_from_poly_trait_ref(
224 poly_trait_ref: ty::PolyTraitRef<'tcx>,
226 if let Some(trait_ref) = poly_trait_ref.no_bound_vars() {
227 self.tcx().mk_projection(item_def_id, trait_ref.substs)
229 // no late-bound regions, we can just ignore the binder
234 "cannot extract an associated type from a higher-ranked trait bound \
241 fn normalize_ty(&self, _span: Span, ty: Ty<'tcx>) -> Ty<'tcx> {
242 // types in item signatures are not normalized, to avoid undue
247 fn set_tainted_by_errors(&self) {
248 // no obvious place to track this, so just let it go
251 fn record_ty(&self, _hir_id: hir::HirId, _ty: Ty<'tcx>, _span: Span) {
252 // no place to record types from signatures?
256 fn type_param_predicates<'tcx>(
257 tcx: TyCtxt<'tcx, 'tcx>,
258 (item_def_id, def_id): (DefId, DefId),
259 ) -> &'tcx ty::GenericPredicates<'tcx> {
262 // In the AST, bounds can derive from two places. Either
263 // written inline like `<T : Foo>` or in a where clause like
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());
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 {
277 tcx.generics_of(item_def_id).parent
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)
284 let mut extend = None;
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,
290 Node::ImplItem(item) => &item.generics,
292 Node::Item(item) => {
294 ItemKind::Fn(.., ref generics, _)
295 | ItemKind::Impl(_, _, _, ref generics, ..)
296 | ItemKind::Ty(_, ref generics)
297 | ItemKind::Existential(ExistTy {
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));
317 Node::ForeignItem(item) => match item.node {
318 ForeignItemKind::Fn(_, _, ref generics) => generics,
325 let icx = ItemCtxt::new(tcx, item_def_id);
326 let mut result = (*result).clone();
327 result.predicates.extend(extend.into_iter());
329 .extend(icx.type_parameter_bounds_in_generics(ast_generics, param_id, ty,
330 OnlySelfBounds(true)));
331 tcx.arena.alloc(result)
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(
341 ast_generics: &hir::Generics,
342 param_id: hir::HirId,
344 only_self_bounds: OnlySelfBounds,
345 ) -> Vec<(ty::Predicate<'tcx>, Span)> {
346 let from_ty_params = ast_generics
349 .filter_map(|param| match param.kind {
350 GenericParamKind::Type { .. } if param.hir_id == param_id => Some(¶m.bounds),
353 .flat_map(|bounds| bounds.iter())
354 .flat_map(|b| predicates_from_bound(self, ty, b));
356 let from_where_clauses = ast_generics
360 .filter_map(|wp| match *wp {
361 hir::WherePredicate::BoundPredicate(ref bp) => Some(bp),
365 let bt = if is_param(self.tcx, &bp.bounded_ty, param_id) {
367 } else if !only_self_bounds.0 {
368 Some(self.to_ty(&bp.bounded_ty))
372 bp.bounds.iter().filter_map(move |b| bt.map(|bt| (bt, b)))
374 .flat_map(|(bt, b)| predicates_from_bound(self, bt, b));
376 from_ty_params.chain(from_where_clauses).collect()
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 {
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)
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);
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);
412 tcx.predicates_of(def_id);
413 if let hir::ForeignItemKind::Fn(..) = item.node {
418 hir::ItemKind::Enum(ref enum_definition, _) => {
419 tcx.generics_of(def_id);
421 tcx.predicates_of(def_id);
422 convert_enum_variant_types(tcx, def_id, &enum_definition.variants);
424 hir::ItemKind::Impl(..) => {
425 tcx.generics_of(def_id);
427 tcx.impl_trait_ref(def_id);
428 tcx.predicates_of(def_id);
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);
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);
441 hir::ItemKind::Struct(ref struct_def, _) | hir::ItemKind::Union(ref struct_def, _) => {
442 tcx.generics_of(def_id);
444 tcx.predicates_of(def_id);
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);
450 tcx.predicates_of(def_id);
453 if let Some(ctor_hir_id) = struct_def.ctor_hir_id() {
454 convert_variant_ctor(tcx, ctor_hir_id);
458 // Desugared from `impl Trait`, so visited by the function's return type.
459 hir::ItemKind::Existential(hir::ExistTy {
460 impl_trait_fn: Some(_),
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);
471 tcx.predicates_of(def_id);
472 if let hir::ItemKind::Fn(..) = it.node {
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);
484 match trait_item.node {
485 hir::TraitItemKind::Const(..)
486 | hir::TraitItemKind::Type(_, Some(_))
487 | hir::TraitItemKind::Method(..) => {
489 if let hir::TraitItemKind::Method(..) = trait_item.node {
494 hir::TraitItemKind::Type(_, None) => {}
497 tcx.predicates_of(def_id);
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);
504 tcx.predicates_of(def_id);
505 if let hir::ImplItemKind::Method(..) = tcx.hir().expect_impl_item(impl_item_id).node {
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);
514 tcx.predicates_of(def_id);
517 fn convert_enum_variant_types<'tcx>(
518 tcx: TyCtxt<'tcx, 'tcx>,
520 variants: &[hir::Variant],
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>>;
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));
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) {
541 "enum discriminant overflowed"
544 format!("overflowed on value after {}", prev_discr.unwrap()),
546 "explicitly set `{} = {}` if that is desired outcome",
547 variant.node.ident, wrapped_discr
551 }.unwrap_or(wrapped_discr),
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);
558 tcx.predicates_of(def_id);
561 // Convert the ctor, if any. This also registers the variant as
563 if let Some(ctor_hir_id) = variant.node.data.ctor_hir_id() {
564 convert_variant_ctor(tcx, ctor_hir_id);
569 fn convert_variant<'tcx>(
570 tcx: TyCtxt<'tcx, 'tcx>,
571 variant_did: Option<DefId>,
572 ctor_did: Option<DefId>,
574 discr: ty::VariantDiscr,
575 def: &hir::VariantData,
576 adt_kind: ty::AdtKind,
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();
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 {
592 "field `{}` is already declared",
594 ).span_label(f.span, "field already declared")
595 .span_label(prev_span, format!("`{}` first declared here", f.ident))
598 seen_fields.insert(f.ident.modern(), f.span);
604 vis: ty::Visibility::from_hir(&f.vis, hir_id, tcx),
608 let recovered = match def {
609 hir::VariantData::Struct(_, r) => *r,
619 CtorKind::from_hir(def),
626 fn adt_def<'tcx>(tcx: TyCtxt<'tcx, 'tcx>, def_id: DefId) -> &'tcx ty::AdtDef {
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,
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
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));
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))
650 ty::VariantDiscr::Relative(distance_from_explicit)
652 distance_from_explicit += 1;
654 convert_variant(tcx, variant_did, ctor_did, v.node.ident, discr,
655 &v.node.data, AdtKind::Enum, def_id)
659 (AdtKind::Enum, variants)
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));
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,
671 (AdtKind::Struct, variants)
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));
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,
683 (AdtKind::Union, variants)
687 tcx.alloc_adt_def(def_id, kind, variants, repr)
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>,
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();
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),
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"),
711 let icx = ItemCtxt::new(tcx, trait_def_id);
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,
718 let superbounds1 = superbounds1.predicates(tcx, self_param_ty);
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));
729 // Combine the two lists to form the complete set of superbounds:
730 let superbounds: Vec<_> = superbounds1.into_iter().chain(superbounds2).collect();
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());
741 tcx.arena.alloc(ty::GenericPredicates {
743 predicates: superbounds,
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);
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"),
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(
761 "the `#[rustc_paren_sugar]` attribute is a temporary means of controlling \
762 which traits can use parenthetical notation",
766 "add `#![feature(unboxed_closures)]` to \
767 the crate attributes to use it"
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);
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>,
785 impl Visitor<'tcx> for LateBoundRegionsDetector<'tcx> {
786 fn nested_visit_map<'this>(&'this mut self) -> NestedVisitorMap<'this, 'tcx> {
787 NestedVisitorMap::None
790 fn visit_ty(&mut self, ty: &'tcx hir::Ty) {
791 if self.has_late_bound_regions.is_some() {
795 hir::TyKind::BareFn(..) => {
796 self.outer_index.shift_in(1);
797 intravisit::walk_ty(self, ty);
798 self.outer_index.shift_out(1);
800 _ => intravisit::walk_ty(self, ty),
804 fn visit_poly_trait_ref(
806 tr: &'tcx hir::PolyTraitRef,
807 m: hir::TraitBoundModifier,
809 if self.has_late_bound_regions.is_some() {
812 self.outer_index.shift_in(1);
813 intravisit::walk_poly_trait_ref(self, tr, m);
814 self.outer_index.shift_out(1);
817 fn visit_lifetime(&mut self, lt: &'tcx hir::Lifetime) {
818 if self.has_late_bound_regions.is_some() {
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(..))
830 self.has_late_bound_regions = Some(lt.span);
836 fn has_late_bound_regions<'tcx>(
837 tcx: TyCtxt<'tcx, 'tcx>,
838 generics: &'tcx hir::Generics,
839 decl: &'tcx hir::FnDecl,
841 let mut visitor = LateBoundRegionsDetector {
843 outer_index: ty::INNERMOST,
844 has_late_bound_regions: None,
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);
853 visitor.visit_fn_decl(decl);
854 visitor.has_late_bound_regions
858 Node::TraitItem(item) => match item.node {
859 hir::TraitItemKind::Method(ref sig, _) => {
860 has_late_bound_regions(tcx, &item.generics, &sig.decl)
864 Node::ImplItem(item) => match item.node {
865 hir::ImplItemKind::Method(ref sig, _) => {
866 has_late_bound_regions(tcx, &item.generics, &sig.decl)
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)
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)
886 fn generics_of<'tcx>(tcx: TyCtxt<'tcx, 'tcx>, def_id: DefId) -> &'tcx ty::Generics {
889 let hir_id = tcx.hir().as_local_hir_id(def_id).unwrap();
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))
898 Node::Expr(&hir::Expr {
899 node: hir::ExprKind::Closure(..),
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,
909 let mut opt_self = None;
910 let mut allow_defaults = false;
912 let no_generics = hir::Generics::empty();
913 let ast_generics = match node {
914 Node::TraitItem(item) => &item.generics,
916 Node::ImplItem(item) => &item.generics,
918 Node::Item(item) => {
920 ItemKind::Fn(.., ref generics, _) | ItemKind::Impl(_, _, _, ref generics, ..) => {
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;
933 ItemKind::Trait(_, _, ref generics, ..)
934 | ItemKind::TraitAlias(ref generics, ..) => {
935 // Add in the self type parameter.
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;
941 opt_self = Some(ty::GenericParamDef {
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 {
948 object_lifetime_default: rl::Set1::Empty,
953 allow_defaults = true;
961 Node::ForeignItem(item) => match item.node {
962 ForeignItemKind::Static(..) => &no_generics,
963 ForeignItemKind::Fn(_, _, ref generics) => generics,
964 ForeignItemKind::Type => &no_generics,
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()
981 let mut params: Vec<_> = opt_self.into_iter().collect();
983 let early_lifetimes = early_bound_lifetimes_from_generics(tcx, ast_generics);
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,
996 let object_lifetime_defaults = tcx.object_lifetime_defaults(hir_id);
998 // Now create the real type parameters.
999 let type_start = own_start - has_self as u32 + params.len() as u32;
1005 .filter_map(|param| {
1006 let kind = match param.kind {
1007 GenericParamKind::Type {
1012 if param.name.ident().name == kw::SelfUpper {
1015 "`Self` should not be the name of a regular parameter"
1019 if !allow_defaults && default.is_some() {
1020 if !tcx.features().default_type_parameter_fallback {
1022 lint::builtin::INVALID_TYPE_PARAM_DEFAULT,
1026 "defaults for type parameters are only allowed in \
1027 `struct`, `enum`, `type`, or `trait` definitions."
1033 ty::GenericParamDefKind::Type {
1034 has_default: default.is_some(),
1035 object_lifetime_default: object_lifetime_defaults
1037 .map_or(rl::Set1::Empty, |o| o[i]),
1041 GenericParamKind::Const { .. } => {
1042 if param.name.ident().name == kw::SelfUpper {
1045 "`Self` should not be the name of a regular parameter",
1049 ty::GenericParamDefKind::Const
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,
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),
1074 let dummy_args = if gen.is_some() {
1075 &["<yield_ty>", "<return_ty>", "<witness>"][..]
1077 &["<closure_kind>", "<closure_signature>"][..]
1084 .map(|(i, &arg)| ty::GenericParamDef {
1085 index: type_start + i as u32,
1086 name: InternedString::intern(arg),
1088 pure_wrt_drop: false,
1089 kind: ty::GenericParamDefKind::Type {
1091 object_lifetime_default: rl::Set1::Empty,
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>"),
1103 pure_wrt_drop: false,
1104 kind: ty::GenericParamDefKind::Type {
1106 object_lifetime_default: rl::Set1::Empty,
1114 let param_def_id_to_index = params
1116 .map(|param| (param.def_id, param.index))
1119 tcx.arena.alloc(ty::Generics {
1120 parent: parent_def_id,
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),
1129 fn report_assoc_ty_on_inherent_impl<'tcx>(tcx: TyCtxt<'tcx, 'tcx>, span: Span) {
1134 "associated types are not yet supported in inherent impls (see #8995)"
1138 fn type_of<'tcx>(tcx: TyCtxt<'tcx, 'tcx>, def_id: DefId) -> Ty<'tcx> {
1139 checked_type_of(tcx, def_id, true).unwrap()
1142 /// Same as [`type_of`] but returns [`Option`] instead of failing.
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>,
1150 ) -> Option<Ty<'tcx>> {
1153 let hir_id = match tcx.hir().as_local_hir_id(def_id) {
1154 Some(hir_id) => hir_id,
1159 bug!("invalid node");
1163 let icx = ItemCtxt::new(tcx, def_id);
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)
1171 TraitItemKind::Const(ref ty, _) | TraitItemKind::Type(_, Some(ref ty)) => icx.to_ty(ty),
1172 TraitItemKind::Type(_, None) => {
1176 span_bug!(item.span, "associated type missing default");
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)
1185 ImplItemKind::Const(ref ty, _) => icx.to_ty(ty),
1186 ImplItemKind::Existential(_) => {
1188 .impl_trait_ref(tcx.hir().get_parent_did_by_hir_id(hir_id))
1191 report_assoc_ty_on_inherent_impl(tcx, item.span);
1194 find_existential_constraints(tcx, def_id)
1196 ImplItemKind::Type(ref ty) => {
1198 .impl_trait_ref(tcx.hir().get_parent_did_by_hir_id(hir_id))
1201 report_assoc_ty_on_inherent_impl(tcx, item.span);
1208 Node::Item(item) => {
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)
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)
1223 ItemKind::Existential(hir::ExistTy {
1224 impl_trait_fn: None,
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),
1232 tcx.typeck_tables_of(owner)
1233 .concrete_existential_types
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(
1243 "owner {:?} has no existential type for {:?} in its tables",
1251 | ItemKind::TraitAlias(..)
1253 | ItemKind::ForeignMod(..)
1254 | ItemKind::GlobalAsm(..)
1255 | ItemKind::ExternCrate(..)
1256 | ItemKind::Use(..) => {
1262 "compute_type_of_item: unexpected item type: {:?}",
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)
1274 ForeignItemKind::Static(ref t, _) => icx.to_ty(t),
1275 ForeignItemKind::Type => tcx.mk_foreign(def_id),
1278 Node::Ctor(&ref def) | Node::Variant(&Spanned {
1279 node: hir::VariantKind { data: ref def, .. },
1282 VariantData::Unit(..) | VariantData::Struct(..) => {
1283 tcx.type_of(tcx.hir().get_parent_did_by_hir_id(hir_id))
1285 VariantData::Tuple(..) => {
1286 let substs = InternalSubsts::identity_for_item(tcx, def_id);
1287 tcx.mk_fn_def(def_id, substs)
1291 Node::Field(field) => icx.to_ty(&field.ty),
1293 Node::Expr(&hir::Expr {
1294 node: hir::ExprKind::Closure(.., gen),
1298 return Some(tcx.typeck_tables_of(def_id).node_type(hir_id));
1301 let substs = ty::ClosureSubsts {
1302 substs: InternalSubsts::identity_for_item(tcx, def_id),
1305 tcx.mk_closure(def_id, substs)
1308 Node::AnonConst(_) => {
1309 let parent_node = tcx.hir().get_by_hir_id(tcx.hir().get_parent_node_by_hir_id(hir_id));
1312 node: hir::TyKind::Array(_, ref constant),
1315 | Node::Ty(&hir::Ty {
1316 node: hir::TyKind::Typeof(ref constant),
1319 | Node::Expr(&hir::Expr {
1320 node: ExprKind::Repeat(_, ref constant),
1322 }) if constant.hir_id == hir_id =>
1327 Node::Variant(&Spanned {
1330 disr_expr: Some(ref e),
1334 }) if e.hir_id == hir_id =>
1336 tcx.adt_def(tcx.hir().get_parent_did_by_hir_id(hir_id))
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 {
1348 node: hir::TyKind::Path(QPath::Resolved(_, ref path)),
1351 | Node::Expr(&hir::Expr {
1352 node: ExprKind::Path(QPath::Resolved(_, ref path)),
1357 Node::Expr(&hir::Expr { node: ExprKind::Struct(ref path, ..), .. }) => {
1358 if let QPath::Resolved(_, ref path) = **path {
1364 Node::TraitRef(&hir::TraitRef { ref path, .. }) => Some(path),
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())
1374 .filter(|arg| arg.is_const())
1376 .filter(|(_, arg)| arg.id() == hir_id)
1377 .map(|(index, _)| index)
1384 bug!("no arg matching AnonConst in path")
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())
1395 Res::Def(_, def_id) => tcx.generics_of(def_id),
1396 Res::Err => return Some(tcx.types.err),
1397 _ if !fail => return None,
1399 tcx.sess.delay_span_bug(
1402 "unexpected const parent path def {:?}",
1406 return Some(tcx.types.err);
1410 generics.params.iter()
1412 if let ty::GenericParamDefKind::Const = param.kind {
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)
1427 tcx.sess.delay_span_bug(
1430 "unexpected const parent path {:?}",
1434 return Some(tcx.types.err);
1442 tcx.sess.delay_span_bug(
1445 "unexpected const parent in type_of_def_id(): {:?}", x
1453 Node::GenericParam(param) => match ¶m.kind {
1454 hir::GenericParamKind::Type { default: Some(ref ty), .. } |
1455 hir::GenericParamKind::Const { ref ty, .. } => {
1462 bug!("unexpected non-type Node::GenericParam: {:?}", x)
1470 bug!("unexpected sort of node in type_of_def_id(): {:?}", x);
1475 fn find_existential_constraints<'tcx>(tcx: TyCtxt<'tcx, 'tcx>, def_id: DefId) -> Ty<'tcx> {
1476 use rustc::hir::{ImplItem, Item, TraitItem};
1478 debug!("find_existential_constraints({:?})", def_id);
1480 struct ConstraintLocator<'tcx> {
1481 tcx: TyCtxt<'tcx, 'tcx>,
1483 // (first found type span, actual type, mapping from the existential type's generic
1484 // parameters to the concrete type's generic parameters)
1486 // The mapping is an index for each use site of a generic parameter in the concrete type
1488 // The indices index into the generic parameters on the existential type.
1489 found: Option<(Span, Ty<'tcx>, Vec<usize>)>,
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) {
1497 "find_existential_constraints: no constraint for `{:?}` at `{:?}`: no tables",
1505 .typeck_tables_of(def_id)
1506 .concrete_existential_types
1508 if let Some(ty::ResolvedOpaqueTy { concrete_type, substs }) = ty {
1510 "find_existential_constraints: found constraint for `{:?}` at `{:?}`: {:?}",
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
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
1528 self.tcx.sess.span_err(
1531 "defining existential type use restricts existential \
1532 type by using the generic parameter `{}` twice",
1539 self.tcx.sess.delay_span_bug(
1542 "non-defining exist ty use in defining scope: {:?}, {:?}",
1543 concrete_type, substs,
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)
1554 .filter_map(|t| match &t.sty {
1555 ty::Param(p) => Some(*index_map.get(p).unwrap()),
1558 let is_param = |ty: Ty<'_>| match ty.sty {
1559 ty::Param(_) => true,
1562 if !substs.types().all(is_param) {
1563 self.tcx.sess.span_err(
1565 "defining existential type use does not fully define existential type",
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,
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(
1582 "concrete type differs from previous defining existential type use",
1586 format!("expected `{}`, got `{}`", prev_ty, concrete_type),
1588 err.span_note(prev_span, "previous use here");
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(
1594 "concrete type's generic parameters differ from previous defining use",
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();
1604 write!(s, ", `{}`", substs[i]).unwrap();
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");
1617 self.found = Some((span, concrete_type, indices));
1621 "find_existential_constraints: no constraint for `{:?}` at `{:?}`",
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())
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 {
1638 intravisit::walk_item(self, it);
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 {
1646 intravisit::walk_impl_item(self, it);
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);
1652 intravisit::walk_trait_item(self, it);
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 {
1666 debug!("find_existential_constraints: scope={:?}", scope);
1668 if scope == hir::CRATE_HIR_ID {
1669 intravisit::walk_crate(&mut locator, tcx.hir().krate());
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),
1677 "{:?} is not a valid scope for an existential type item",
1683 match locator.found {
1684 Some((_, ty, _)) => ty,
1686 let span = tcx.def_span(def_id);
1687 tcx.sess.span_err(span, "could not find defining uses");
1693 fn fn_sig<'tcx>(tcx: TyCtxt<'tcx, 'tcx>, def_id: DefId) -> ty::PolyFnSig<'tcx> {
1695 use rustc::hir::Node::*;
1697 let hir_id = tcx.hir().as_local_hir_id(def_id).unwrap();
1699 let icx = ItemCtxt::new(tcx, def_id);
1701 match tcx.hir().get_by_hir_id(hir_id) {
1702 TraitItem(hir::TraitItem {
1703 node: TraitItemKind::Method(sig, _),
1706 | ImplItem(hir::ImplItem {
1707 node: ImplItemKind::Method(sig, _),
1709 }) => AstConv::ty_of_fn(&icx, sig.header.unsafety, sig.header.abi, &sig.decl),
1712 node: ItemKind::Fn(decl, header, _, _),
1714 }) => AstConv::ty_of_fn(&icx, header.unsafety, header.abi, decl),
1716 ForeignItem(&hir::ForeignItem {
1717 node: ForeignItemKind::Fn(ref fn_decl, _, _),
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)
1724 Ctor(data) | Variant(Spanned {
1725 node: hir::VariantKind { data, .. },
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()
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(
1736 hir::Unsafety::Normal,
1742 node: hir::ExprKind::Closure(..),
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`.
1752 // the signature of a closure, you should use the
1753 // `closure_sig` method on the `ClosureSubsts`:
1755 // closure_substs.closure_sig(def_id, tcx)
1757 // or, inside of an inference context, you can use
1759 // infcx.closure_sig(def_id, closure_substs)
1760 bug!("to get the signature of a closure, use `closure_sig()` not `fn_sig()`");
1764 bug!("unexpected sort of node in fn_sig(): {:?}", x);
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);
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)
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),
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> {
1804 .filter(move |param| match param.kind {
1805 GenericParamKind::Lifetime { .. } => {
1806 !tcx.is_late_bound(param.hir_id)
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>,
1818 ) -> &'tcx ty::GenericPredicates<'tcx> {
1819 debug!("predicates_defined_on({:?})", def_id);
1820 let mut result = tcx.explicit_predicates_of(def_id);
1822 "predicates_defined_on: explicit_predicates_of({:?}) = {:?}",
1826 let inferred_outlives = tcx.inferred_outlives_of(def_id);
1827 if !inferred_outlives.is_empty() {
1828 let span = tcx.def_span(def_id);
1830 "predicates_defined_on: inferred_outlives_of({:?}) = {:?}",
1834 let mut predicates = (*result).clone();
1835 predicates.predicates.extend(inferred_outlives.iter().map(|&p| (p, span)));
1836 result = tcx.arena.alloc(predicates);
1838 debug!("predicates_defined_on({:?}) = {:?}", def_id, result);
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>,
1848 ) -> &'tcx ty::GenericPredicates<'tcx> {
1849 let mut result = tcx.predicates_defined_on(def_id);
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.
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);
1869 debug!("predicates_of(def_id={:?}) = {:?}", def_id, result);
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>,
1878 ) -> &'tcx ty::GenericPredicates<'tcx> {
1880 use rustc_data_structures::fx::FxHashSet;
1882 debug!("explicit_predicates_of(def_id={:?})", def_id);
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)>,
1892 impl<'tcx> UniquePredicates<'tcx> {
1896 uniques: FxHashSet::default(),
1900 fn push(&mut self, value: (ty::Predicate<'tcx>, Span)) {
1901 if self.uniques.insert(value) {
1902 self.predicates.push(value);
1906 fn extend<I: IntoIterator<Item = (ty::Predicate<'tcx>, Span)>>(&mut self, iter: I) {
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),
1917 let node = tcx.hir().get_by_hir_id(hir_id);
1919 let mut is_trait = None;
1920 let mut is_default_impl_trait = None;
1922 let icx = ItemCtxt::new(tcx, def_id);
1923 let no_generics = hir::Generics::empty();
1924 let empty_trait_items = HirVec::new();
1926 let mut predicates = UniquePredicates::new();
1928 let ast_generics = match node {
1929 Node::TraitItem(item) => &item.generics,
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);
1936 // Collect the bounds, i.e., the `A + B + 'c` in `impl A + B + 'c`.
1937 let bounds = AstConv::compute_bounds(
1941 SizedByDefault::Yes,
1942 tcx.def_span(def_id),
1945 predicates.extend(bounds.predicates(tcx, opaque_ty));
1948 _ => &item.generics,
1951 Node::Item(item) => {
1953 ItemKind::Impl(_, _, defaultness, ref generics, ..) => {
1954 if defaultness.is_default() {
1955 is_default_impl_trait = tcx.impl_trait_ref(def_id);
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,
1965 ItemKind::Trait(_, _, ref generics, .., ref items) => {
1966 is_trait = Some((ty::TraitRef::identity(tcx, def_id), items));
1969 ItemKind::TraitAlias(ref generics, _) => {
1970 is_trait = Some((ty::TraitRef::identity(tcx, def_id), &empty_trait_items));
1973 ItemKind::Existential(ExistTy {
1979 let substs = InternalSubsts::identity_for_item(tcx, def_id);
1980 let opaque_ty = tcx.mk_opaque(def_id, substs);
1982 // Collect the bounds, i.e., the `A + B + 'c` in `impl A + B + 'c`.
1983 let bounds = AstConv::compute_bounds(
1987 SizedByDefault::Yes,
1988 tcx.def_span(def_id),
1991 let bounds_predicates = bounds.predicates(tcx, opaque_ty);
1992 if impl_trait_fn.is_some() {
1994 return tcx.arena.alloc(ty::GenericPredicates {
1996 predicates: bounds_predicates,
1999 // named existential types
2000 predicates.extend(bounds_predicates);
2009 Node::ForeignItem(item) => match item.node {
2010 ForeignItemKind::Static(..) => &no_generics,
2011 ForeignItemKind::Fn(_, _, ref generics) => generics,
2012 ForeignItemKind::Type => &no_generics,
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;
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());
2030 // In default impls, we can assume that the self type implements
2031 // the trait. So in:
2033 // default impl Foo for Bar { .. }
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)));
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),
2050 name: param.name.ident().as_interned_str(),
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, <, None);
2059 let outlives = ty::Binder::bind(ty::OutlivesPredicate(region, bound));
2060 predicates.push((outlives.to_predicate(), lt.span));
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);
2077 let sized = SizedByDefault::Yes;
2078 let bounds = AstConv::compute_bounds(&icx, param_ty, ¶m.bounds, sized, param.span);
2079 predicates.extend(bounds.predicates(tcx, param_ty));
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 {
2087 &hir::WherePredicate::BoundPredicate(ref bound_pred) => {
2088 let ty = icx.to_ty(&bound_pred.bounded_ty);
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.
2102 let span = bound_pred.bounded_ty.span;
2103 let predicate = ty::OutlivesPredicate(ty, tcx.mk_region(ty::ReEmpty));
2105 (ty::Predicate::TypeOutlives(ty::Binder::dummy(predicate)), span)
2110 for bound in bound_pred.bounds.iter() {
2112 &hir::GenericBound::Trait(ref poly_trait_ref, _) => {
2113 let mut bounds = Bounds::default();
2115 let (trait_ref, _) = AstConv::instantiate_poly_trait_ref(
2122 predicates.push((trait_ref.to_predicate(), poly_trait_ref.span));
2123 predicates.extend(bounds.predicates(tcx, ty));
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))
2135 &hir::WherePredicate::RegionPredicate(ref region_pred) => {
2136 let r1 = AstConv::ast_region_to_region(&icx, ®ion_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)
2144 let pred = ty::Binder::bind(ty::OutlivesPredicate(r1, r2));
2146 (ty::Predicate::RegionOutlives(pred), span)
2150 &hir::WherePredicate::EqPredicate(..) => {
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()
2166 tcx.mk_projection(tcx.hir().local_def_id_from_hir_id(trait_item.hir_id),
2167 self_trait_ref.substs);
2169 let bounds = AstConv::compute_bounds(
2170 &ItemCtxt::new(tcx, def_id),
2173 SizedByDefault::Yes,
2177 bounds.predicates(tcx, assoc_ty).into_iter()
2181 let mut predicates = predicates.predicates;
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`
2188 if let Node::Item(&Item {
2189 node: ItemKind::Impl(..),
2193 let self_ty = tcx.type_of(def_id);
2194 let trait_ref = tcx.impl_trait_ref(def_id);
2195 cgp::setup_constraining_predicates(
2199 &mut cgp::parameters_for_impl(self_ty, trait_ref),
2203 let result = tcx.arena.alloc(ty::GenericPredicates {
2204 parent: generics.parent,
2207 debug!("explicit_predicates_of(def_id={:?}) = {:?}", def_id, result);
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>,
2219 bound: &hir::GenericBound,
2220 ) -> Vec<(ty::Predicate<'tcx>, Span)> {
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))
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)]
2234 hir::GenericBound::Trait(_, hir::TraitBoundModifier::Maybe) => vec![],
2238 fn compute_sig_of_foreign_fn_decl<'tcx>(
2239 tcx: TyCtxt<'tcx, 'tcx>,
2243 ) -> ty::PolyFnSig<'tcx> {
2244 let unsafety = if abi == abi::Abi::RustIntrinsic {
2245 intrisic_operation_unsafety(&*tcx.item_name(def_id).as_str())
2247 hir::Unsafety::Unsafe
2249 let fty = AstConv::ty_of_fn(&ItemCtxt::new(tcx, def_id), unsafety, abi, decl);
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
2257 let check = |ast_ty: &hir::Ty, ty: Ty<'_>| {
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)
2268 .help("add #![feature(simd_ffi)] to the crate attributes to enable")
2272 for (input, ty) in decl.inputs.iter().zip(*fty.inputs().skip_binder()) {
2275 if let hir::Return(ref ty) = decl.output {
2276 check(&ty, *fty.output().skip_binder())
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,
2287 _ => bug!("is_foreign_item applied to non-local def-id {:?}", def_id),
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, _), ..
2296 Some(Node::ForeignItem( &hir::ForeignItem {
2297 node: hir::ForeignItemKind::Static(_, mutbl), ..
2300 _ => bug!("static_mutability applied to non-local def-id {:?}", def_id),
2304 fn from_target_feature(
2305 tcx: TyCtxt<'_, '_>,
2307 attr: &ast::Attribute,
2308 whitelist: &FxHashMap<String, Option<Symbol>>,
2309 target_features: &mut Vec<Symbol>,
2311 let list = match attr.meta_item_list() {
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)
2322 let rust_features = tcx.features();
2324 // Only `enable = ...` is accepted in the meta-item list.
2325 if !item.check_name(sym::enable) {
2326 bad_item(item.span());
2330 // Must be of the form `enable = "..."` (a string).
2331 let value = match item.value_str() {
2332 Some(value) => value,
2334 bad_item(item.span());
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) {
2346 "the feature named `{}` is not valid for this target",
2349 let mut err = tcx.sess.struct_span_err(item.span(), &msg);
2352 format!("`{}` is not valid for this target", feature),
2354 if feature.starts_with("+") {
2355 let valid = whitelist.contains_key(&feature[1..]);
2357 err.help("consider removing the leading `+` in the feature name");
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),
2385 if !allowed && id.is_local() {
2386 feature_gate::emit_feature_err(
2387 &tcx.sess.parse_sess,
2388 feature_gate.unwrap(),
2390 feature_gate::GateIssue::Language,
2391 &format!("the target feature `{}` is currently unstable", feature),
2394 Some(Symbol::intern(feature))
2399 fn linkage_by_name<'tcx>(tcx: TyCtxt<'tcx, 'tcx>, def_id: DefId, name: &str) -> Linkage {
2400 use rustc::mir::mono::Linkage::*;
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.
2408 // ghost, dllimport, dllexport and linkonce_odr_autohide are not supported
2409 // and don't have to be, LLVM treats them as no-ops.
2411 "appending" => Appending,
2412 "available_externally" => AvailableExternally,
2414 "extern_weak" => ExternalWeak,
2415 "external" => External,
2416 "internal" => Internal,
2417 "linkonce" => LinkOnceAny,
2418 "linkonce_odr" => LinkOnceODR,
2419 "private" => Private,
2421 "weak_odr" => WeakODR,
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")
2428 .fatal(&format!("invalid linkage specified: {}", name))
2434 fn codegen_fn_attrs<'tcx>(tcx: TyCtxt<'tcx, 'tcx>, id: DefId) -> CodegenFnAttrs {
2435 let attrs = tcx.get_attrs(id);
2437 let mut codegen_fn_attrs = CodegenFnAttrs::new();
2439 let whitelist = tcx.target_features_whitelist(LOCAL_CRATE);
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;
2453 // `#[ffi_returns_twice]` is only allowed `extern fn`s.
2458 "`#[ffi_returns_twice]` may only be used on foreign functions"
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.
2484 "`export_name` may not contain null characters"
2487 codegen_fn_attrs.export_name = Some(s);
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")
2497 from_target_feature(
2502 &mut codegen_fn_attrs.target_features,
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()));
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) {
2512 "illegal null byte in link_section \
2516 tcx.sess.span_err(attr.span, &msg);
2518 codegen_fn_attrs.link_section = Some(val);
2521 } else if attr.check_name(sym::link_name) {
2522 codegen_fn_attrs.link_name = attr.value_str();
2526 codegen_fn_attrs.inline = attrs.iter().fold(InlineAttr::None, |ia, attr| {
2527 if attr.path != sym::inline {
2530 match attr.meta().map(|i| i.node) {
2531 Some(MetaItemKind::Word) => {
2535 Some(MetaItemKind::List(ref items)) => {
2537 inline_span = Some(attr.span);
2538 if items.len() != 1 {
2540 tcx.sess.diagnostic(),
2543 "expected one argument"
2546 } else if list_contains_name(&items[..], sym::always) {
2548 } else if list_contains_name(&items[..], sym::never) {
2552 tcx.sess.diagnostic(),
2561 Some(MetaItemKind::NameValue(_)) => ia,
2566 codegen_fn_attrs.optimize = attrs.iter().fold(OptimizeAttr::None, |ia, attr| {
2567 if attr.path != sym::optimize {
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");
2576 Some(MetaItemKind::List(ref items)) => {
2578 inline_span = Some(attr.span);
2579 if items.len() != 1 {
2580 err(attr.span, "expected one argument");
2582 } else if list_contains_name(&items[..], sym::size) {
2584 } else if list_contains_name(&items[..], sym::speed) {
2587 err(items[0].span(), "invalid argument");
2591 Some(MetaItemKind::NameValue(_)) => ia,
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
2600 if codegen_fn_attrs.target_features.len() > 0 {
2601 if codegen_fn_attrs.inline == InlineAttr::Always {
2602 if let Some(span) = inline_span {
2605 "cannot use #[inline(always)] with \
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.
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;
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);
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;