bjoernager/rust
bjoernager/rust
1
Fork 0
Code Issues Pull requests Activity

Instantiate binders in supertrait_vtable_slot

Browse source
This commit is contained in:
Michael Goulet 2024-09-29 17:51:38 -04:00
parent c3ce4e66a5
commit af3f212453
2 changed files with 38 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

40
compiler/rustc_trait_selection/src/traits/vtable.rs
View file

@ -2,6 +2,8 @@ use std::fmt::Debug;
use std::ops::ControlFlow;
use rustc_hir::def_id::DefId;
use rustc_infer::infer::{BoundRegionConversionTime, TyCtxtInferExt};
use rustc_infer::traits::ObligationCause;
use rustc_infer::traits::util::PredicateSet;
use rustc_middle::bug;
use rustc_middle::query::Providers;
@ -13,7 +15,7 @@ use smallvec::{SmallVec, smallvec};
use tracing::debug;
use crate::errors::DumpVTableEntries;
use crate::traits::{impossible_predicates, is_vtable_safe_method};
use crate::traits::{ObligationCtxt, impossible_predicates, is_vtable_safe_method};
#[derive(Clone, Debug)]
pub enum VtblSegment<'tcx> {
@ -383,17 +385,37 @@ pub(crate) fn supertrait_vtable_slot<'tcx>(
let ty::Dynamic(target, _, _) = *target.kind() else {
bug!();
};
let target_principal = tcx
.normalize_erasing_regions(ty::ParamEnv::reveal_all(), target.principal()?)
.with_self_ty(tcx, tcx.types.trait_object_dummy_self);
let target_principal = target.principal()?.with_self_ty(tcx, tcx.types.trait_object_dummy_self);
// Given that we have a target principal, it is a bug for there not to be a source principal.
let ty::Dynamic(source, _, _) = *source.kind() else {
bug!();
};
let source_principal = tcx
.normalize_erasing_regions(ty::ParamEnv::reveal_all(), source.principal().unwrap())
.with_self_ty(tcx, tcx.types.trait_object_dummy_self);
let source_principal =
source.principal().unwrap().with_self_ty(tcx, tcx.types.trait_object_dummy_self);
let infcx = tcx.infer_ctxt().build();
let param_env = ty::ParamEnv::reveal_all();
let trait_refs_are_compatible =
|source: ty::PolyTraitRef<'tcx>, target: ty::PolyTraitRef<'tcx>| {
infcx.probe(|_| {
let ocx = ObligationCtxt::new(&infcx);
let source = ocx.normalize(&ObligationCause::dummy(), param_env, source);
let target = ocx.normalize(&ObligationCause::dummy(), param_env, target);
infcx.enter_forall(target, |target| {
let source = infcx.instantiate_binder_with_fresh_vars(
DUMMY_SP,
BoundRegionConversionTime::HigherRankedType,
source,
);
let Ok(()) = ocx.eq(&ObligationCause::dummy(), param_env, target, source)
else {
return false;
};
ocx.select_all_or_error().is_empty()
})
})
};
let vtable_segment_callback = {
let mut vptr_offset = 0;
@ -404,9 +426,7 @@ pub(crate) fn supertrait_vtable_slot<'tcx>(
}
VtblSegment::TraitOwnEntries { trait_ref, emit_vptr } => {
vptr_offset += tcx.own_existential_vtable_entries(trait_ref.def_id()).len();
if tcx.normalize_erasing_regions(ty::ParamEnv::reveal_all(), trait_ref)
== target_principal
{
if trait_refs_are_compatible(trait_ref, target_principal) {
if emit_vptr {
return ControlFlow::Break(Some(vptr_offset));
} else {