Skip to content
This repository was archived by the owner on May 28, 2025. It is now read-only.

Commit 347ca50

Browse files
RalfJungRalfJung
committed
mentioned items: also handle vtables
1 parent ee4b758 commit 347ca50

File tree

6 files changed

+107
-45
lines changed

6 files changed

+107
-45
lines changed

‎compiler/rustc_middle/src/mir/mod.rs‎

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -319,7 +319,11 @@ impl<'tcx> CoroutineInfo<'tcx> {
319319
pub enum MentionedItem<'tcx> {
320320
Fn(DefId, GenericArgsRef<'tcx>),
321321
Drop(Ty<'tcx>),
322-
// FIXME: add Vtable { source_ty: Ty<'tcx>, target_ty: Ty<'tcx> },
322+
/// Unsizing casts might require vtables, so we have to record them.
323+
UnsizeCast {
324+
source_ty: Ty<'tcx>,
325+
target_ty: Ty<'tcx>,
326+
},
323327
// FIXME: do we have to add closures?
324328
}
325329

‎compiler/rustc_mir_transform/src/mentioned_items.rs‎

Lines changed: 25 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
use rustc_middle::mir::visit::Visitor;
22
use rustc_middle::mir::{self, ConstOperand, Location, MentionedItem, MirPass};
3-
use rustc_middle::ty::{self, TyCtxt};
3+
use rustc_middle::ty::{self, adjustment::PointerCoercion, TyCtxt};
44
use rustc_session::Session;
55
use rustc_span::source_map::Spanned;
66

@@ -54,4 +54,28 @@ impl<'tcx> Visitor<'tcx> for MentionedItemsVisitor<'_, 'tcx> {
5454
_ => {}
5555
}
5656
}
57+
58+
fn visit_rvalue(&mut self, rvalue: &mir::Rvalue<'tcx>, location: Location) {
59+
self.super_rvalue(rvalue, location);
60+
match *rvalue {
61+
// We need to detect unsizing casts that required vtables.
62+
mir::Rvalue::Cast(
63+
mir::CastKind::PointerCoercion(PointerCoercion::Unsize),
64+
ref operand,
65+
target_ty,
66+
)
67+
| mir::Rvalue::Cast(mir::CastKind::DynStar, ref operand, target_ty) => {
68+
let span = self.body.source_info(location).span;
69+
self.mentioned_items.push(Spanned {
70+
node: MentionedItem::UnsizeCast {
71+
source_ty: operand.ty(self.body, self.tcx),
72+
target_ty,
73+
},
74+
span,
75+
});
76+
}
77+
// Function pointer casts are already handled by `visit_constant` above.
78+
_ => {}
79+
}
80+
}
5781
}

‎compiler/rustc_monomorphize/src/collector.rs‎

Lines changed: 45 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -736,7 +736,7 @@ impl<'a, 'tcx> MirUsedCollector<'a, 'tcx> {
736736
where
737737
T: TypeFoldable<TyCtxt<'tcx>>,
738738
{
739-
debug!("monomorphize: self.instance={:?}", self.instance);
739+
trace!("monomorphize: self.instance={:?}", self.instance);
740740
self.instance.instantiate_mir_and_normalize_erasing_regions(
741741
self.tcx,
742742
ty::ParamEnv::reveal_all(),
@@ -1342,35 +1342,37 @@ fn create_mono_items_for_vtable_methods<'tcx>(
13421342
) {
13431343
assert!(!trait_ty.has_escaping_bound_vars() && !impl_ty.has_escaping_bound_vars());
13441344

1345-
if let ty::Dynamic(trait_ty, ..) = trait_ty.kind() {
1346-
if let Some(principal) = trait_ty.principal() {
1347-
let poly_trait_ref = principal.with_self_ty(tcx, impl_ty);
1348-
assert!(!poly_trait_ref.has_escaping_bound_vars());
1349-
1350-
// Walk all methods of the trait, including those of its supertraits
1351-
let entries = tcx.vtable_entries(poly_trait_ref);
1352-
let methods = entries
1353-
.iter()
1354-
.filter_map(|entry| match entry {
1355-
VtblEntry::MetadataDropInPlace
1356-
| VtblEntry::MetadataSize
1357-
| VtblEntry::MetadataAlign
1358-
| VtblEntry::Vacant => None,
1359-
VtblEntry::TraitVPtr(_) => {
1360-
// all super trait items already covered, so skip them.
1361-
None
1362-
}
1363-
VtblEntry::Method(instance) => {
1364-
Some(*instance).filter(|instance| should_codegen_locally(tcx, instance))
1365-
}
1366-
})
1367-
.map(|item| create_fn_mono_item(tcx, item, source));
1368-
output.extend(methods);
1369-
}
1370-
1371-
// Also add the destructor.
1372-
visit_drop_use(tcx, impl_ty, false, source, output);
1345+
let ty::Dynamic(trait_ty, ..) = trait_ty.kind() else {
1346+
bug!("create_mono_items_for_vtable_methods: {trait_ty:?} not a trait type");
1347+
};
1348+
if let Some(principal) = trait_ty.principal() {
1349+
let poly_trait_ref = principal.with_self_ty(tcx, impl_ty);
1350+
assert!(!poly_trait_ref.has_escaping_bound_vars());
1351+
1352+
// Walk all methods of the trait, including those of its supertraits
1353+
let entries = tcx.vtable_entries(poly_trait_ref);
1354+
debug!(?entries);
1355+
let methods = entries
1356+
.iter()
1357+
.filter_map(|entry| match entry {
1358+
VtblEntry::MetadataDropInPlace
1359+
| VtblEntry::MetadataSize
1360+
| VtblEntry::MetadataAlign
1361+
| VtblEntry::Vacant => None,
1362+
VtblEntry::TraitVPtr(_) => {
1363+
// all super trait items already covered, so skip them.
1364+
None
1365+
}
1366+
VtblEntry::Method(instance) => {
1367+
Some(*instance).filter(|instance| should_codegen_locally(tcx, instance))
1368+
}
1369+
})
1370+
.map(|item| create_fn_mono_item(tcx, item, source));
1371+
output.extend(methods);
13731372
}
1373+
1374+
// Also add the destructor.
1375+
visit_drop_use(tcx, impl_ty, false, source, output);
13741376
}
13751377

13761378
//=-----------------------------------------------------------------------------
@@ -1693,7 +1695,8 @@ fn collect_items_of_instance<'tcx>(
16931695
}
16941696
}
16951697

1696-
/// `item` must be already monomorphized
1698+
/// `item` must be already monomorphized.
1699+
#[instrument(skip(tcx, span, output), level = "debug")]
16971700
fn visit_mentioned_item<'tcx>(
16981701
tcx: TyCtxt<'tcx>,
16991702
item: &MentionedItem<'tcx>,
@@ -1712,6 +1715,18 @@ fn visit_mentioned_item<'tcx>(
17121715
MentionedItem::Drop(ty) => {
17131716
visit_drop_use(tcx, ty, /*is_direct_call*/ true, span, output);
17141717
}
1718+
MentionedItem::UnsizeCast { source_ty, target_ty } => {
1719+
let (source_ty, target_ty) =
1720+
find_vtable_types_for_unsizing(tcx.at(span), source_ty, target_ty);
1721+
// This could also be a different Unsize instruction, like
1722+
// from a fixed sized array to a slice. But we are only
1723+
// interested in things that produce a vtable.
1724+
if (target_ty.is_trait() && !source_ty.is_trait())
1725+
|| (target_ty.is_dyn_star() && !source_ty.is_dyn_star())
1726+
{
1727+
create_mono_items_for_vtable_methods(tcx, target_ty, source_ty, span, output);
1728+
}
1729+
}
17151730
}
17161731
}
17171732

‎tests/ui/consts/required-consts/collect-in-dead-vtable.noopt.stderr‎

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,21 +1,21 @@
11
error[E0080]: evaluation of `Fail::<i32>::C` failed
2-
--> $DIR/collect-in-dead-vtable.rs:11:19
2+
--> $DIR/collect-in-dead-vtable.rs:8:19
33
|
44
LL | const C: () = panic!();
5-
| ^^^^^^^^ the evaluated program panicked at 'explicit panic', $DIR/collect-in-dead-vtable.rs:11:19
5+
| ^^^^^^^^ the evaluated program panicked at 'explicit panic', $DIR/collect-in-dead-vtable.rs:8:19
66
|
77
= note: this error originates in the macro `$crate::panic::panic_2015` which comes from the expansion of the macro `panic` (in Nightly builds, run with -Z macro-backtrace for more info)
88

99
note: erroneous constant encountered
10-
--> $DIR/collect-in-dead-vtable.rs:25:21
10+
--> $DIR/collect-in-dead-vtable.rs:21:21
1111
|
1212
LL | let _ = Fail::<T>::C;
1313
| ^^^^^^^^^^^^
1414

1515
note: the above error was encountered while instantiating `fn <std::vec::Vec<i32> as MyTrait>::not_called`
16-
--> $DIR/collect-in-dead-vtable.rs:34:40
16+
--> $DIR/collect-in-dead-vtable.rs:30:40
1717
|
18-
LL | let gen_vtable: &dyn MyTrait = &v; // vtable "appears" here
18+
LL | let gen_vtable: &dyn MyTrait = &v; // vtable is "mentioned" here
1919
| ^^
2020

2121
error: aborting due to 1 previous error

‎tests/ui/consts/required-consts/collect-in-dead-vtable.opt.stderr‎

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,23 @@
1+
error[E0080]: evaluation of `Fail::<i32>::C` failed
2+
--> $DIR/collect-in-dead-vtable.rs:8:19
3+
|
4+
LL | const C: () = panic!();
5+
| ^^^^^^^^ the evaluated program panicked at 'explicit panic', $DIR/collect-in-dead-vtable.rs:8:19
6+
|
7+
= note: this error originates in the macro `$crate::panic::panic_2015` which comes from the expansion of the macro `panic` (in Nightly builds, run with -Z macro-backtrace for more info)
8+
9+
note: erroneous constant encountered
10+
--> $DIR/collect-in-dead-vtable.rs:21:21
11+
|
12+
LL | let _ = Fail::<T>::C;
13+
| ^^^^^^^^^^^^
14+
15+
note: the above error was encountered while instantiating `fn <std::vec::Vec<i32> as MyTrait>::not_called`
16+
--> $DIR/collect-in-dead-vtable.rs:30:40
17+
|
18+
LL | let gen_vtable: &dyn MyTrait = &v; // vtable is "mentioned" here
19+
| ^^
20+
21+
error: aborting due to 1 previous error
22+
23+
For more information about this error, try `rustc --explain E0080`.

‎tests/ui/consts/required-consts/collect-in-dead-vtable.rs‎

Lines changed: 4 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,14 +1,11 @@
11
//@revisions: noopt opt
2-
//@[noopt] build-fail
2+
//@ build-fail
33
//@[opt] compile-flags: -O
4-
//FIXME: `opt` revision currently does not stop with an error due to
5-
//<https://github.com/rust-lang/rust/issues/107503>.
6-
//@[opt] build-pass
74
//! This fails without optimizations, so it should also fail with optimizations.
85
96
struct Fail<T>(T);
107
impl<T> Fail<T> {
11-
const C: () = panic!(); //[noopt]~ERROR evaluation of `Fail::<i32>::C` failed
8+
const C: () = panic!(); //~ERROR evaluation of `Fail::<i32>::C` failed
129
}
1310

1411
trait MyTrait {
@@ -17,8 +14,7 @@ trait MyTrait {
1714

1815
// This function is not actually called, but it is mentioned in a vtable in a function that is
1916
// called. Make sure we still find this error.
20-
// This relies on mono-item collection checking `required_consts` in functions that are referenced
21-
// in vtables that syntactically appear in collected functions (even inside dead code).
17+
// This ensures that we are properly considering vtables when gathering "mentioned" items.
2218
impl<T> MyTrait for Vec<T> {
2319
fn not_called(&self) {
2420
if false {
@@ -31,7 +27,7 @@ impl<T> MyTrait for Vec<T> {
3127
fn called<T>() {
3228
if false {
3329
let v: Vec<T> = Vec::new();
34-
let gen_vtable: &dyn MyTrait = &v; // vtable "appears" here
30+
let gen_vtable: &dyn MyTrait = &v; // vtable is "mentioned" here
3531
}
3632
}
3733

0 commit comments

Comments
 (0)