Skip to content

Commit 5c15ad7

Browse files
tmandrytmandry
committed
NiceRegionError: Use written return type for async fn
1 parent 6487845 commit 5c15ad7

File tree

13 files changed

+181
-165
lines changed

13 files changed

+181
-165
lines changed

‎compiler/rustc_hir/src/hir.rs‎

Lines changed: 14 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2726,6 +2726,10 @@ pub struct FnHeader {
27262726
}
27272727

27282728
impl FnHeader {
2729+
pub fn is_async(&self) -> bool {
2730+
matches!(&self.asyncness, IsAsync::Async)
2731+
}
2732+
27292733
pub fn is_const(&self) -> bool {
27302734
matches!(&self.constness, Constness::Const)
27312735
}
@@ -3169,7 +3173,7 @@ impl<'hir> Node<'hir> {
31693173
}
31703174
}
31713175

3172-
pub fn fn_decl(&self) -> Option<&FnDecl<'hir>> {
3176+
pub fn fn_decl(&self) -> Option<&'hir FnDecl<'hir>> {
31733177
match self {
31743178
Node::TraitItem(TraitItem { kind: TraitItemKind::Fn(fn_sig, _), .. })
31753179
| Node::ImplItem(ImplItem { kind: ImplItemKind::Fn(fn_sig, _), .. })
@@ -3181,6 +3185,15 @@ impl<'hir> Node<'hir> {
31813185
}
31823186
}
31833187

3188+
pub fn fn_sig(&self) -> Option<&'hir FnSig<'hir>> {
3189+
match self {
3190+
Node::TraitItem(TraitItem { kind: TraitItemKind::Fn(fn_sig, _), .. })
3191+
| Node::ImplItem(ImplItem { kind: ImplItemKind::Fn(fn_sig, _), .. })
3192+
| Node::Item(Item { kind: ItemKind::Fn(fn_sig, _, _), .. }) => Some(fn_sig),
3193+
_ => None,
3194+
}
3195+
}
3196+
31843197
pub fn body_id(&self) -> Option<BodyId> {
31853198
match self {
31863199
Node::TraitItem(TraitItem {

‎compiler/rustc_infer/src/infer/error_reporting/mod.rs‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -65,9 +65,9 @@ use rustc_hir::def_id::DefId;
6565
use rustc_hir::lang_items::LangItem;
6666
use rustc_hir::{Item, ItemKind, Node};
6767
use rustc_middle::dep_graph::DepContext;
68-
use rustc_middle::ty::error::TypeError;
6968
use rustc_middle::ty::{
7069
self,
70+
error::TypeError,
7171
subst::{GenericArgKind, Subst, SubstsRef},
7272
Binder, Region, Ty, TyCtxt, TypeFoldable,
7373
};

‎compiler/rustc_infer/src/infer/error_reporting/nice_region_error/different_lifetimes.rs‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -171,6 +171,7 @@ impl<'a, 'tcx> NiceRegionError<'a, 'tcx> {
171171

172172
self.suggest_adding_lifetime_params(sub, ty_sup, ty_sub, &mut err);
173173

174+
// TODO: This is only helpful if the lifetime more visible in the impl Future type than in the signature.
174175
if let Some(t) = future_return_type {
175176
let snip = self
176177
.tcx()

‎compiler/rustc_infer/src/infer/error_reporting/nice_region_error/find_anon_type.rs‎

Lines changed: 6 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,5 @@
11
use rustc_hir as hir;
22
use rustc_hir::intravisit::{self, Visitor};
3-
use rustc_hir::Node;
43
use rustc_middle::hir::map::Map;
54
use rustc_middle::hir::nested_filter;
65
use rustc_middle::middle::resolve_lifetime as rl;
@@ -25,25 +24,19 @@ pub(crate) fn find_anon_type<'tcx>(
2524
tcx: TyCtxt<'tcx>,
2625
region: Region<'tcx>,
2726
br: &ty::BoundRegionKind,
28-
) -> Option<(&'tcx hir::Ty<'tcx>, &'tcx hir::FnDecl<'tcx>)> {
27+
) -> Option<(&'tcx hir::Ty<'tcx>, &'tcx hir::FnSig<'tcx>)> {
2928
if let Some(anon_reg) = tcx.is_suitable_region(region) {
3029
let hir_id = tcx.hir().local_def_id_to_hir_id(anon_reg.def_id);
31-
let fndecl = match tcx.hir().get(hir_id) {
32-
Node::Item(&hir::Item { kind: hir::ItemKind::Fn(ref m, ..), .. })
33-
| Node::TraitItem(&hir::TraitItem {
34-
kind: hir::TraitItemKind::Fn(ref m, ..), ..
35-
})
36-
| Node::ImplItem(&hir::ImplItem { kind: hir::ImplItemKind::Fn(ref m, ..), .. }) => {
37-
&m.decl
38-
}
39-
_ => return None,
30+
let Some(fn_sig) = tcx.hir().get(hir_id).fn_sig() else {
31+
return None
4032
};
4133

42-
fndecl
34+
fn_sig
35+
.decl
4336
.inputs
4437
.iter()
4538
.find_map(|arg| find_component_for_bound_region(tcx, arg, br))
46-
.map(|ty| (ty, &**fndecl))
39+
.map(|ty| (ty, fn_sig))
4740
} else {
4841
None
4942
}

‎compiler/rustc_infer/src/infer/error_reporting/nice_region_error/util.rs‎

Lines changed: 26 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
use crate::infer::error_reporting::nice_region_error::NiceRegionError;
55
use rustc_hir as hir;
66
use rustc_hir::def_id::LocalDefId;
7-
use rustc_middle::ty::{self, DefIdTree, Region, Ty};
7+
use rustc_middle::ty::{self, Binder, DefIdTree, Region, Ty, TypeFoldable};
88
use rustc_span::Span;
99

1010
/// Information about the anonymous region we are searching for.
@@ -149,26 +149,41 @@ impl<'a, 'tcx> NiceRegionError<'a, 'tcx> {
149149
}
150150

151151
// Here, we check for the case where the anonymous region
152-
// is in the return type.
152+
// is in the return type as written by the user.
153153
// FIXME(#42703) - Need to handle certain cases here.
154154
pub(super) fn is_return_type_anon(
155155
&self,
156156
scope_def_id: LocalDefId,
157157
br: ty::BoundRegionKind,
158-
decl: &hir::FnDecl<'_>,
158+
hir_sig: &hir::FnSig<'_>,
159159
) -> Option<Span> {
160-
let ret_ty = self.tcx().type_of(scope_def_id);
161-
if let ty::FnDef(_, _) = ret_ty.kind() {
162-
let sig = ret_ty.fn_sig(self.tcx());
163-
let late_bound_regions =
164-
self.tcx().collect_referenced_late_bound_regions(&sig.output());
165-
if late_bound_regions.iter().any(|r| *r == br) {
166-
return Some(decl.output.span());
167-
}
160+
let fn_ty = self.tcx().type_of(scope_def_id);
161+
if let ty::FnDef(_, _) = fn_ty.kind() {
162+
let ret_ty = fn_ty.fn_sig(self.tcx()).output();
163+
let span = hir_sig.decl.output.span();
164+
let future_output = if hir_sig.header.is_async() {
165+
ret_ty.map_bound(|ty| self.infcx.get_impl_future_output_ty(ty)).transpose()
166+
} else {
167+
None
168+
};
169+
return match future_output {
170+
Some(output) if self.includes_region(output, br) => Some(span),
171+
None if self.includes_region(ret_ty, br) => Some(span),
172+
_ => None,
173+
};
168174
}
169175
None
170176
}
171177

178+
fn includes_region(
179+
&self,
180+
ty: Binder<'tcx, impl TypeFoldable<'tcx>>,
181+
region: ty::BoundRegionKind,
182+
) -> bool {
183+
let late_bound_regions = self.tcx().collect_referenced_late_bound_regions(&ty);
184+
late_bound_regions.iter().any(|r| *r == region)
185+
}
186+
172187
// Here we check for the case where anonymous region
173188
// corresponds to self and if yes, we display E0312.
174189
// FIXME(#42700) - Need to format self properly to

‎src/test/ui/async-await/issue-76547.stderr‎

Lines changed: 4 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -2,23 +2,17 @@ error[E0623]: lifetime mismatch
22
--> $DIR/issue-76547.rs:20:13
33
|
44
LL | async fn fut(bufs: &mut [&mut [u8]]) {
5-
| --------- -
6-
| | |
7-
| | this `async fn` implicitly returns an `impl Future<Output = ()>`
8-
| this parameter and the returned future are declared with different lifetimes...
5+
| ---------------- these two types are declared with different lifetimes...
96
LL | ListFut(bufs).await
10-
| ^^^^ ...but data from `bufs` is held across an await point here
7+
| ^^^^ ...but data from `bufs` flows into `bufs` here
118

129
error[E0623]: lifetime mismatch
1310
--> $DIR/issue-76547.rs:34:14
1411
|
1512
LL | async fn fut2(bufs: &mut [&mut [u8]]) -> i32 {
16-
| --------- ---
17-
| | |
18-
| | this `async fn` implicitly returns an `impl Future<Output = i32>`
19-
| this parameter and the returned future are declared with different lifetimes...
13+
| ---------------- these two types are declared with different lifetimes...
2014
LL | ListFut2(bufs).await
21-
| ^^^^ ...but data from `bufs` is held across an await point here
15+
| ^^^^ ...but data from `bufs` flows into `bufs` here
2216

2317
error: aborting due to 2 previous errors
2418

‎src/test/ui/async-await/issues/issue-63388-1.stderr‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@ error[E0623]: lifetime mismatch
22
--> $DIR/issue-63388-1.rs:14:9
33
|
44
LL | &'a self, foo: &dyn Foo
5-
| -------- this parameter and the returned future are declared with different lifetimes...
5+
| -------- this parameter and the returned future are declared with different lifetimes...
66
LL | ) -> &dyn Foo
77
| --------
88
| |

‎src/test/ui/self/arbitrary_self_types_pin_lifetime_mismatch-async.stderr‎

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -2,28 +2,28 @@ error[E0623]: lifetime mismatch
22
--> $DIR/arbitrary_self_types_pin_lifetime_mismatch-async.rs:8:52
33
|
44
LL | async fn a(self: Pin<&Foo>, f: &Foo) -> &Foo { f }
5-
| ---- ---- ^ ...but data from `f` is held across an await point here
6-
| | |
7-
| | this `async fn` implicitly returns an `impl Future<Output = &Foo>`
8-
| this parameter and the returned future are declared with different lifetimes...
5+
| ---- ---- ^ ...but data from `f` is held across an await point here
6+
| | |
7+
| | this `async fn` implicitly returns an `impl Future<Output = &Foo>`
8+
| this parameter and the returned future are declared with different lifetimes...
99

1010
error[E0623]: lifetime mismatch
1111
--> $DIR/arbitrary_self_types_pin_lifetime_mismatch-async.rs:11:82
1212
|
1313
LL | async fn c(self: Pin<&Self>, f: &Foo, g: &Foo) -> (Pin<&Foo>, &Foo) { (self, f) }
14-
| ----- ----------------- ^ ...but data from `f` is held across an await point here
15-
| | |
16-
| | this `async fn` implicitly returns an `impl Future<Output = (Pin<&Foo>, &Foo)>`
17-
| this parameter and the returned future are declared with different lifetimes...
14+
| ---- ----------------- ^ ...but data from `f` is held across an await point here
15+
| | |
16+
| | this `async fn` implicitly returns an `impl Future<Output = (Pin<&Foo>, &Foo)>`
17+
| this parameter and the returned future are declared with different lifetimes...
1818

1919
error[E0623]: lifetime mismatch
2020
--> $DIR/arbitrary_self_types_pin_lifetime_mismatch-async.rs:17:64
2121
|
2222
LL | async fn bar<'a>(self: Alias<&Self>, arg: &'a ()) -> &() { arg }
23-
| ----- --- ^^^ ...but data from `arg` is held across an await point here
24-
| | |
25-
| | this `async fn` implicitly returns an `impl Future<Output = &()>`
26-
| this parameter and the returned future are declared with different lifetimes...
23+
| ------ --- ^^^ ...but data from `arg` is held across an await point here
24+
| | |
25+
| | this `async fn` implicitly returns an `impl Future<Output = &()>`
26+
| this parameter and the returned future are declared with different lifetimes...
2727

2828
error: aborting due to 3 previous errors
2929

‎src/test/ui/self/elision/lt-ref-self-async.stderr‎

Lines changed: 24 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -2,65 +2,65 @@ error[E0623]: lifetime mismatch
22
--> $DIR/lt-ref-self-async.rs:13:9
33
|
44
LL | async fn ref_self(&self, f: &u32) -> &u32 {
5-
| ----- ----
6-
| | |
7-
| | this `async fn` implicitly returns an `impl Future<Output = &u32>`
8-
| this parameter and the returned future are declared with different lifetimes...
5+
| ---- ----
6+
| | |
7+
| | this `async fn` implicitly returns an `impl Future<Output = &u32>`
8+
| this parameter and the returned future are declared with different lifetimes...
99
LL | f
1010
| ^ ...but data from `f` is held across an await point here
1111

1212
error[E0623]: lifetime mismatch
1313
--> $DIR/lt-ref-self-async.rs:19:9
1414
|
1515
LL | async fn ref_Self(self: &Self, f: &u32) -> &u32 {
16-
| ----- ----
17-
| | |
18-
| | this `async fn` implicitly returns an `impl Future<Output = &u32>`
19-
| this parameter and the returned future are declared with different lifetimes...
16+
| ---- ----
17+
| | |
18+
| | this `async fn` implicitly returns an `impl Future<Output = &u32>`
19+
| this parameter and the returned future are declared with different lifetimes...
2020
LL | f
2121
| ^ ...but data from `f` is held across an await point here
2222

2323
error[E0623]: lifetime mismatch
2424
--> $DIR/lt-ref-self-async.rs:23:9
2525
|
2626
LL | async fn box_ref_Self(self: Box<&Self>, f: &u32) -> &u32 {
27-
| ----- ----
28-
| | |
29-
| | this `async fn` implicitly returns an `impl Future<Output = &u32>`
30-
| this parameter and the returned future are declared with different lifetimes...
27+
| ---- ----
28+
| | |
29+
| | this `async fn` implicitly returns an `impl Future<Output = &u32>`
30+
| this parameter and the returned future are declared with different lifetimes...
3131
LL | f
3232
| ^ ...but data from `f` is held across an await point here
3333

3434
error[E0623]: lifetime mismatch
3535
--> $DIR/lt-ref-self-async.rs:27:9
3636
|
3737
LL | async fn pin_ref_Self(self: Pin<&Self>, f: &u32) -> &u32 {
38-
| ----- ----
39-
| | |
40-
| | this `async fn` implicitly returns an `impl Future<Output = &u32>`
41-
| this parameter and the returned future are declared with different lifetimes...
38+
| ---- ----
39+
| | |
40+
| | this `async fn` implicitly returns an `impl Future<Output = &u32>`
41+
| this parameter and the returned future are declared with different lifetimes...
4242
LL | f
4343
| ^ ...but data from `f` is held across an await point here
4444

4545
error[E0623]: lifetime mismatch
4646
--> $DIR/lt-ref-self-async.rs:31:9
4747
|
4848
LL | async fn box_box_ref_Self(self: Box<Box<&Self>>, f: &u32) -> &u32 {
49-
| ----- ----
50-
| | |
51-
| | this `async fn` implicitly returns an `impl Future<Output = &u32>`
52-
| this parameter and the returned future are declared with different lifetimes...
49+
| ---- ----
50+
| | |
51+
| | this `async fn` implicitly returns an `impl Future<Output = &u32>`
52+
| this parameter and the returned future are declared with different lifetimes...
5353
LL | f
5454
| ^ ...but data from `f` is held across an await point here
5555

5656
error[E0623]: lifetime mismatch
5757
--> $DIR/lt-ref-self-async.rs:35:9
5858
|
5959
LL | async fn box_pin_Self(self: Box<Pin<&Self>>, f: &u32) -> &u32 {
60-
| ----- ----
61-
| | |
62-
| | this `async fn` implicitly returns an `impl Future<Output = &u32>`
63-
| this parameter and the returned future are declared with different lifetimes...
60+
| ---- ----
61+
| | |
62+
| | this `async fn` implicitly returns an `impl Future<Output = &u32>`
63+
| this parameter and the returned future are declared with different lifetimes...
6464
LL | f
6565
| ^ ...but data from `f` is held across an await point here
6666

0 commit comments

Comments
 (0)