1 // Copyright 2012-2014 The Rust Project Developers. See the COPYRIGHT
2 // file at the top-level directory of this distribution and at
3 // http://rust-lang.org/COPYRIGHT.
5 // Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
6 // http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
7 // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
8 // option. This file may not be copied, modified, or distributed
9 // except according to those terms.
11 use rustc::hir::def_id::DefId;
12 use rustc::middle::lang_items::DropInPlaceFnLangItem;
14 use rustc::ty::adjustment::CustomCoerceUnsized;
15 use rustc::ty::{self, Ty, TyCtxt};
17 pub use rustc::ty::Instance;
18 pub use self::item::{MonoItem, MonoItemExt};
24 #[inline(never)] // give this a place in the profiler
25 pub fn assert_symbols_are_distinct<'a, 'tcx, I>(tcx: TyCtxt<'a, 'tcx, 'tcx>, mono_items: I)
26 where I: Iterator<Item=&'a MonoItem<'tcx>>
28 let mut symbols: Vec<_> = mono_items.map(|mono_item| {
29 (mono_item, mono_item.symbol_name(tcx))
32 (&mut symbols[..]).sort_by(|&(_, ref sym1), &(_, ref sym2)|{
36 for pair in (&symbols[..]).windows(2) {
37 let sym1 = &pair[0].1;
38 let sym2 = &pair[1].1;
41 let mono_item1 = pair[0].0;
42 let mono_item2 = pair[1].0;
44 let span1 = mono_item1.local_span(tcx);
45 let span2 = mono_item2.local_span(tcx);
47 // Deterministically select one of the spans for error reporting
48 let span = match (span1, span2) {
49 (Some(span1), Some(span2)) => {
50 Some(if span1.lo().0 > span2.lo().0 {
57 (None, Some(span)) => Some(span),
61 let error_message = format!("symbol `{}` is already defined", sym1);
63 if let Some(span) = span {
64 tcx.sess.span_fatal(span, &error_message)
66 tcx.sess.fatal(&error_message)
72 fn fn_once_adapter_instance<'a, 'tcx>(
73 tcx: TyCtxt<'a, 'tcx, 'tcx>,
75 substs: ty::ClosureSubsts<'tcx>,
77 debug!("fn_once_adapter_shim({:?}, {:?})",
80 let fn_once = tcx.lang_items().fn_once_trait().unwrap();
81 let call_once = tcx.associated_items(fn_once)
82 .find(|it| it.kind == ty::AssociatedKind::Method)
84 let def = ty::InstanceDef::ClosureOnceShim { call_once };
86 let self_ty = tcx.mk_closure(closure_did, substs);
88 let sig = substs.closure_sig(closure_did, tcx);
89 let sig = tcx.normalize_erasing_late_bound_regions(ty::ParamEnv::reveal_all(), &sig);
90 assert_eq!(sig.inputs().len(), 1);
91 let substs = tcx.mk_substs_trait(self_ty, &[sig.inputs()[0].into()]);
93 debug!("fn_once_adapter_shim: self_ty={:?} sig={:?}", self_ty, sig);
94 Instance { def, substs }
97 fn needs_fn_once_adapter_shim(actual_closure_kind: ty::ClosureKind,
98 trait_closure_kind: ty::ClosureKind)
101 match (actual_closure_kind, trait_closure_kind) {
102 (ty::ClosureKind::Fn, ty::ClosureKind::Fn) |
103 (ty::ClosureKind::FnMut, ty::ClosureKind::FnMut) |
104 (ty::ClosureKind::FnOnce, ty::ClosureKind::FnOnce) => {
105 // No adapter needed.
108 (ty::ClosureKind::Fn, ty::ClosureKind::FnMut) => {
109 // The closure fn `llfn` is a `fn(&self, ...)`. We want a
110 // `fn(&mut self, ...)`. In fact, at codegen time, these are
111 // basically the same thing, so we can just return llfn.
114 (ty::ClosureKind::Fn, ty::ClosureKind::FnOnce) |
115 (ty::ClosureKind::FnMut, ty::ClosureKind::FnOnce) => {
116 // The closure fn `llfn` is a `fn(&self, ...)` or `fn(&mut
117 // self, ...)`. We want a `fn(self, ...)`. We can produce
118 // this by doing something like:
120 // fn call_once(self, ...) { call_mut(&self, ...) }
121 // fn call_once(mut self, ...) { call_mut(&mut self, ...) }
123 // These are both the same at codegen time.
130 pub fn resolve_closure<'a, 'tcx> (
131 tcx: TyCtxt<'a, 'tcx, 'tcx>,
133 substs: ty::ClosureSubsts<'tcx>,
134 requested_kind: ty::ClosureKind)
137 let actual_kind = substs.closure_kind(def_id, tcx);
139 match needs_fn_once_adapter_shim(actual_kind, requested_kind) {
140 Ok(true) => fn_once_adapter_instance(tcx, def_id, substs),
141 _ => Instance::new(def_id, substs.substs)
145 pub fn resolve_drop_in_place<'a, 'tcx>(
146 tcx: TyCtxt<'a, 'tcx, 'tcx>,
148 -> ty::Instance<'tcx>
150 let def_id = tcx.require_lang_item(DropInPlaceFnLangItem);
151 let substs = tcx.intern_substs(&[ty.into()]);
152 Instance::resolve(tcx, ty::ParamEnv::reveal_all(), def_id, substs).unwrap()
155 pub fn custom_coerce_unsize_info<'a, 'tcx>(tcx: TyCtxt<'a, 'tcx, 'tcx>,
158 -> CustomCoerceUnsized {
159 let def_id = tcx.lang_items().coerce_unsized_trait().unwrap();
161 let trait_ref = ty::Binder::bind(ty::TraitRef {
163 substs: tcx.mk_substs_trait(source_ty, &[target_ty.into()])
166 match tcx.codegen_fulfill_obligation( (ty::ParamEnv::reveal_all(), trait_ref)) {
167 traits::VtableImpl(traits::VtableImplData { impl_def_id, .. }) => {
168 tcx.coerce_unsized_info(impl_def_id).custom_kind.unwrap()
171 bug!("invalid CoerceUnsized vtable: {:?}", vtable);