StorageLive: refresh storage (instead of UB) when local is already live

This commit is contained in:
Ralf Jung 2024-06-08 12:04:38 +02:00
parent d8fde50745
commit 9b05e154f3
3 changed files with 10 additions and 11 deletions

View file

@ -73,8 +73,6 @@ const_eval_division_by_zero =
dividing by zero dividing by zero
const_eval_division_overflow = const_eval_division_overflow =
overflow in signed division (dividing MIN by -1) overflow in signed division (dividing MIN by -1)
const_eval_double_storage_live =
StorageLive on a local that was already live
const_eval_dyn_call_not_a_method = const_eval_dyn_call_not_a_method =
`dyn` call trying to call something that is not a method `dyn` call trying to call something that is not a method

View file

@ -1103,11 +1103,9 @@ impl<'tcx, M: Machine<'tcx>> InterpCx<'tcx, M> {
Operand::Immediate(Immediate::Uninit) Operand::Immediate(Immediate::Uninit)
}); });
// StorageLive expects the local to be dead, and marks it live. // If the local is already live, deallocate its old memory.
let old = mem::replace(&mut self.frame_mut().locals[local].value, local_val); let old = mem::replace(&mut self.frame_mut().locals[local].value, local_val);
if !matches!(old, LocalValue::Dead) { self.deallocate_local(old)?;
throw_ub_custom!(fluent::const_eval_double_storage_live);
}
Ok(()) Ok(())
} }
@ -1121,7 +1119,7 @@ impl<'tcx, M: Machine<'tcx>> InterpCx<'tcx, M> {
assert!(local != mir::RETURN_PLACE, "Cannot make return place dead"); assert!(local != mir::RETURN_PLACE, "Cannot make return place dead");
trace!("{:?} is now dead", local); trace!("{:?} is now dead", local);
// It is entirely okay for this local to be already dead (at least that's how we currently generate MIR) // If the local is already dead, this is a NOP.
let old = mem::replace(&mut self.frame_mut().locals[local].value, LocalValue::Dead); let old = mem::replace(&mut self.frame_mut().locals[local].value, LocalValue::Dead);
self.deallocate_local(old)?; self.deallocate_local(old)?;
Ok(()) Ok(())

View file

@ -361,16 +361,19 @@ pub enum StatementKind<'tcx> {
/// At any point during the execution of a function, each local is either allocated or /// At any point during the execution of a function, each local is either allocated or
/// unallocated. Except as noted below, all locals except function parameters are initially /// unallocated. Except as noted below, all locals except function parameters are initially
/// unallocated. `StorageLive` statements cause memory to be allocated for the local while /// unallocated. `StorageLive` statements cause memory to be allocated for the local while
/// `StorageDead` statements cause the memory to be freed. Using a local in any way (not only /// `StorageDead` statements cause the memory to be freed. In other words,
/// reading/writing from it) while it is unallocated is UB. /// `StorageLive`/`StorageDead` act like the heap operations `allocate`/`deallocate`, but for
/// stack-allocated local variables. Using a local in any way (not only reading/writing from it)
/// while it is unallocated is UB.
/// ///
/// Some locals have no `StorageLive` or `StorageDead` statements within the entire MIR body. /// Some locals have no `StorageLive` or `StorageDead` statements within the entire MIR body.
/// These locals are implicitly allocated for the full duration of the function. There is a /// These locals are implicitly allocated for the full duration of the function. There is a
/// convenience method at `rustc_mir_dataflow::storage::always_storage_live_locals` for /// convenience method at `rustc_mir_dataflow::storage::always_storage_live_locals` for
/// computing these locals. /// computing these locals.
/// ///
/// If the local is already allocated, calling `StorageLive` again is UB. However, for an /// If the local is already allocated, calling `StorageLive` again will implicitly free the
/// unallocated local an additional `StorageDead` all is simply a nop. /// local and then allocate fresh uninitilized memory. If a local is already deallocated,
/// calling `StorageDead` again is a NOP.
StorageLive(Local), StorageLive(Local),
/// See `StorageLive` above. /// See `StorageLive` above.