mirror of
https://github.com/rust-lang/rust.git
synced 2026-04-26 13:01:27 +03:00
All nested statics in a single interning run have the same parent
So we do not need to disambiguate considering parents
This commit is contained in:
@@ -16,10 +16,7 @@
|
||||
use hir::def::DefKind;
|
||||
use rustc_ast::Mutability;
|
||||
use rustc_data_structures::fx::{FxHashSet, FxIndexMap};
|
||||
use rustc_hir::def_id::LocalDefIdMap;
|
||||
use rustc_hir::definitions::{
|
||||
DefPathData, PerParentDisambiguatorState, PerParentDisambiguatorsMap,
|
||||
};
|
||||
use rustc_hir::definitions::{DefPathData, PerParentDisambiguatorState};
|
||||
use rustc_hir::{self as hir};
|
||||
use rustc_middle::middle::codegen_fn_attrs::CodegenFnAttrs;
|
||||
use rustc_middle::mir::interpret::{
|
||||
@@ -108,7 +105,7 @@ fn intern_shallow<'tcx, M: CompileTimeMachine<'tcx>>(
|
||||
ecx: &mut InterpCx<'tcx, M>,
|
||||
alloc_id: AllocId,
|
||||
mutability: Mutability,
|
||||
disambiguators: Option<&mut LocalDefIdMap<PerParentDisambiguatorState>>,
|
||||
disambiguator: Option<&mut PerParentDisambiguatorState>,
|
||||
) -> Result<impl Iterator<Item = CtfeProvenance> + 'tcx, InternError> {
|
||||
trace!("intern_shallow {:?}", alloc_id);
|
||||
// remove allocation
|
||||
@@ -132,7 +129,7 @@ fn intern_shallow<'tcx, M: CompileTimeMachine<'tcx>>(
|
||||
static_id,
|
||||
alloc_id,
|
||||
alloc,
|
||||
disambiguators.expect("disambiguators needed"),
|
||||
disambiguator.expect("disambiguator needed"),
|
||||
);
|
||||
} else {
|
||||
ecx.tcx.set_alloc_id_memory(alloc_id, alloc);
|
||||
@@ -147,19 +144,18 @@ fn intern_as_new_static<'tcx>(
|
||||
static_id: LocalDefId,
|
||||
alloc_id: AllocId,
|
||||
alloc: ConstAllocation<'tcx>,
|
||||
disambiguators: &mut LocalDefIdMap<PerParentDisambiguatorState>,
|
||||
disambiguator: &mut PerParentDisambiguatorState,
|
||||
) {
|
||||
// `intern_const_alloc_recursive` is called once per static and it contains the `DisambiguatorState`.
|
||||
// `intern_const_alloc_recursive` is called once per static and it contains the `PerParentDisambiguatorState`.
|
||||
// The `<static_id>::{{nested}}` path is thus unique to `intern_const_alloc_recursive` and the
|
||||
// `DisambiguatorState` ensures the generated path is unique for this call as we generate
|
||||
// `PerParentDisambiguatorState` ensures the generated path is unique for this call as we generate
|
||||
// `<static_id>::{{nested#n}}` where `n` is the `n`th `intern_as_new_static` call.
|
||||
let feed = tcx.create_def(
|
||||
static_id,
|
||||
None,
|
||||
DefKind::Static { safety: hir::Safety::Safe, mutability: alloc.0.mutability, nested: true },
|
||||
Some(DefPathData::NestedStatic),
|
||||
//FIXME(oli-obk): cleanup (https://github.com/rust-lang/rust/pull/155547#discussion_r3110792640)
|
||||
disambiguators.get_or_create(static_id),
|
||||
disambiguator,
|
||||
);
|
||||
tcx.set_nested_alloc_id_static(alloc_id, feed.def_id());
|
||||
|
||||
@@ -209,7 +205,9 @@ pub fn intern_const_alloc_recursive<'tcx, M: CompileTimeMachine<'tcx>>(
|
||||
intern_kind: InternKind,
|
||||
ret: &MPlaceTy<'tcx>,
|
||||
) -> Result<(), InternError> {
|
||||
let mut disambiguators = Default::default();
|
||||
let mut disambiguator =
|
||||
ecx.machine.static_def_id().map(|id| PerParentDisambiguatorState::new(id));
|
||||
let mut disambiguator = disambiguator.as_mut();
|
||||
|
||||
// We are interning recursively, and for mutability we are distinguishing the "root" allocation
|
||||
// that we are starting in, and all other allocations that we are encountering recursively.
|
||||
@@ -248,13 +246,15 @@ pub fn intern_const_alloc_recursive<'tcx, M: CompileTimeMachine<'tcx>>(
|
||||
// This gives us the initial set of nested allocations, which will then all be processed
|
||||
// recursively in the loop below.
|
||||
let mut todo: Vec<_> = if is_static {
|
||||
assert!(disambiguator.is_some());
|
||||
// Do not steal the root allocation, we need it later to create the return value of `eval_static_initializer`.
|
||||
// But still change its mutability to match the requested one.
|
||||
let (kind, alloc) = ecx.memory.alloc_map.get_mut(&base_alloc_id).unwrap();
|
||||
prepare_alloc(*ecx.tcx, *kind, alloc, base_mutability)?;
|
||||
alloc.provenance().ptrs().iter().map(|&(_, prov)| prov).collect()
|
||||
} else {
|
||||
intern_shallow(ecx, base_alloc_id, base_mutability, Some(&mut disambiguators))?.collect()
|
||||
assert!(disambiguator.is_none());
|
||||
intern_shallow(ecx, base_alloc_id, base_mutability, None)?.collect()
|
||||
};
|
||||
// We need to distinguish "has just been interned" from "was already in `tcx`",
|
||||
// so we track this in a separate set.
|
||||
@@ -336,7 +336,7 @@ pub fn intern_const_alloc_recursive<'tcx, M: CompileTimeMachine<'tcx>>(
|
||||
// okay with losing some potential for immutability here. This can anyway only affect
|
||||
// `static mut`.
|
||||
just_interned.insert(alloc_id);
|
||||
let next = intern_shallow(ecx, alloc_id, inner_mutability, Some(&mut disambiguators))?;
|
||||
let next = intern_shallow(ecx, alloc_id, inner_mutability, disambiguator.as_deref_mut())?;
|
||||
todo.extend(next);
|
||||
}
|
||||
if found_bad_mutable_ptr {
|
||||
|
||||
Reference in New Issue
Block a user