Skip to content

Commit 4755e2f

Browse files
borsbors
committed
Auto merge of #56741 - RalfJung:retag-to-raw, r=oli-obk
treat ref-to-raw cast like a reborrow: do a special kind of retag r? @oli-obk Cc @nikomatsakis
2 parents 817dda7 + c1160a8 commit 4755e2f

File tree

22 files changed

+76
-128
lines changed

22 files changed

+76
-128
lines changed

‎src/librustc/ich/impls_mir.rs‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -200,13 +200,13 @@ impl_stable_hash_for!(impl<'gcx> for enum mir::StatementKind<'gcx> [ mir::Statem
200200
SetDiscriminant { place, variant_index },
201201
StorageLive(place),
202202
StorageDead(place),
203-
EscapeToRaw(place),
204-
Retag { fn_entry, two_phase, place },
203+
Retag(retag_kind, place),
205204
AscribeUserType(place, variance, c_ty),
206205
Nop,
207206
InlineAsm { asm, outputs, inputs },
208207
});
209208

209+
impl_stable_hash_for!(enum mir::RetagKind { FnEntry, TwoPhase, Raw, Default });
210210
impl_stable_hash_for!(enum mir::FakeReadCause { ForMatchGuard, ForMatchedPlace, ForLet });
211211

212212
impl<'a, 'gcx> HashStable<StableHashingContext<'a>> for mir::Place<'gcx> {

‎src/librustc/mir/mod.rs‎

Lines changed: 24 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -1788,23 +1788,7 @@ pub enum StatementKind<'tcx> {
17881788
/// by miri and only generated when "-Z mir-emit-retag" is passed.
17891789
/// See <https://internals.rust-lang.org/t/stacked-borrows-an-aliasing-model-for-rust/8153/>
17901790
/// for more details.
1791-
Retag {
1792-
/// `fn_entry` indicates whether this is the initial retag that happens in the
1793-
/// function prolog.
1794-
fn_entry: bool,
1795-
/// `two_phase` indicates whether this is just the reservation action of
1796-
/// a two-phase borrow.
1797-
two_phase: bool,
1798-
/// The place to retag
1799-
place: Place<'tcx>,
1800-
},
1801-
1802-
/// Escape the given reference to a raw pointer, so that it can be accessed
1803-
/// without precise provenance tracking. These statements are currently only interpreted
1804-
/// by miri and only generated when "-Z mir-emit-retag" is passed.
1805-
/// See <https://internals.rust-lang.org/t/stacked-borrows-an-aliasing-model-for-rust/8153/>
1806-
/// for more details.
1807-
EscapeToRaw(Operand<'tcx>),
1791+
Retag(RetagKind, Place<'tcx>),
18081792

18091793
/// Encodes a user's type ascription. These need to be preserved
18101794
/// intact so that NLL can respect them. For example:
@@ -1824,6 +1808,19 @@ pub enum StatementKind<'tcx> {
18241808
Nop,
18251809
}
18261810

1811+
/// `RetagKind` describes what kind of retag is to be performed.
1812+
#[derive(Copy, Clone, RustcEncodable, RustcDecodable, Debug, PartialEq, Eq)]
1813+
pub enum RetagKind {
1814+
/// The initial retag when entering a function
1815+
FnEntry,
1816+
/// Retag preparing for a two-phase borrow
1817+
TwoPhase,
1818+
/// Retagging raw pointers
1819+
Raw,
1820+
/// A "normal" retag
1821+
Default,
1822+
}
1823+
18271824
/// The `FakeReadCause` describes the type of pattern why a `FakeRead` statement exists.
18281825
#[derive(Copy, Clone, RustcEncodable, RustcDecodable, Debug)]
18291826
pub enum FakeReadCause {
@@ -1859,13 +1856,16 @@ impl<'tcx> Debug for Statement<'tcx> {
18591856
match self.kind {
18601857
Assign(ref place, ref rv) => write!(fmt, "{:?} = {:?}", place, rv),
18611858
FakeRead(ref cause, ref place) => write!(fmt, "FakeRead({:?}, {:?})", cause, place),
1862-
Retag { fn_entry, two_phase, ref place } =>
1863-
write!(fmt, "Retag({}{}{:?})",
1864-
if fn_entry { "[fn entry] " } else { "" },
1865-
if two_phase { "[2phase] " } else { "" },
1859+
Retag(ref kind, ref place) =>
1860+
write!(fmt, "Retag({}{:?})",
1861+
match kind {
1862+
RetagKind::FnEntry => "[fn entry] ",
1863+
RetagKind::TwoPhase => "[2phase] ",
1864+
RetagKind::Raw => "[raw] ",
1865+
RetagKind::Default => "",
1866+
},
18661867
place,
18671868
),
1868-
EscapeToRaw(ref place) => write!(fmt, "EscapeToRaw({:?})", place),
18691869
StorageLive(ref place) => write!(fmt, "StorageLive({:?})", place),
18701870
StorageDead(ref place) => write!(fmt, "StorageDead({:?})", place),
18711871
SetDiscriminant {
@@ -2979,6 +2979,7 @@ CloneTypeFoldableAndLiftImpls! {
29792979
SourceInfo,
29802980
UpvarDecl,
29812981
FakeReadCause,
2982+
RetagKind,
29822983
SourceScope,
29832984
SourceScopeData,
29842985
SourceScopeLocalData,
@@ -3046,8 +3047,7 @@ EnumTypeFoldableImpl! {
30463047
(StatementKind::StorageLive)(a),
30473048
(StatementKind::StorageDead)(a),
30483049
(StatementKind::InlineAsm) { asm, outputs, inputs },
3049-
(StatementKind::Retag) { fn_entry, two_phase, place },
3050-
(StatementKind::EscapeToRaw)(place),
3050+
(StatementKind::Retag)(kind, place),
30513051
(StatementKind::AscribeUserType)(a, v, b),
30523052
(StatementKind::Nop),
30533053
}

‎src/librustc/mir/visit.rs‎

Lines changed: 6 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -153,11 +153,10 @@ macro_rules! make_mir_visitor {
153153
}
154154

155155
fn visit_retag(&mut self,
156-
fn_entry: & $($mutability)* bool,
157-
two_phase: & $($mutability)* bool,
156+
kind: & $($mutability)* RetagKind,
158157
place: & $($mutability)* Place<'tcx>,
159158
location: Location) {
160-
self.super_retag(fn_entry, two_phase, place, location);
159+
self.super_retag(kind, place, location);
161160
}
162161

163162
fn visit_place(&mut self,
@@ -385,9 +384,6 @@ macro_rules! make_mir_visitor {
385384
location
386385
);
387386
}
388-
StatementKind::EscapeToRaw(ref $($mutability)* op) => {
389-
self.visit_operand(op, location);
390-
}
391387
StatementKind::StorageLive(ref $($mutability)* local) => {
392388
self.visit_local(
393389
local,
@@ -417,10 +413,9 @@ macro_rules! make_mir_visitor {
417413
self.visit_operand(input, location);
418414
}
419415
}
420-
StatementKind::Retag { ref $($mutability)* fn_entry,
421-
ref $($mutability)* two_phase,
422-
ref $($mutability)* place } => {
423-
self.visit_retag(fn_entry, two_phase, place, location);
416+
StatementKind::Retag ( ref $($mutability)* kind,
417+
ref $($mutability)* place ) => {
418+
self.visit_retag(kind, place, location);
424419
}
425420
StatementKind::AscribeUserType(
426421
ref $($mutability)* place,
@@ -725,8 +720,7 @@ macro_rules! make_mir_visitor {
725720
}
726721

727722
fn super_retag(&mut self,
728-
_fn_entry: & $($mutability)* bool,
729-
_two_phase: & $($mutability)* bool,
723+
_kind: & $($mutability)* RetagKind,
730724
place: & $($mutability)* Place<'tcx>,
731725
location: Location) {
732726
self.visit_place(

‎src/librustc_codegen_ssa/mir/statement.rs‎

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -106,7 +106,6 @@ impl<'a, 'tcx: 'a, Bx: BuilderMethods<'a, 'tcx>> FunctionCx<'a, 'tcx, Bx> {
106106
}
107107
mir::StatementKind::FakeRead(..) |
108108
mir::StatementKind::Retag { .. } |
109-
mir::StatementKind::EscapeToRaw { .. } |
110109
mir::StatementKind::AscribeUserType(..) |
111110
mir::StatementKind::Nop => bx,
112111
}

‎src/librustc_mir/borrow_check/mod.rs‎

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -590,7 +590,6 @@ impl<'cx, 'gcx, 'tcx> DataflowResultsConsumer<'cx, 'tcx> for MirBorrowckCtxt<'cx
590590
StatementKind::Nop
591591
| StatementKind::AscribeUserType(..)
592592
| StatementKind::Retag { .. }
593-
| StatementKind::EscapeToRaw { .. }
594593
| StatementKind::StorageLive(..) => {
595594
// `Nop`, `AscribeUserType`, `Retag`, and `StorageLive` are irrelevant
596595
// to borrow check.

‎src/librustc_mir/borrow_check/nll/invalidation.rs‎

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -135,7 +135,6 @@ impl<'cx, 'tcx, 'gcx> Visitor<'tcx> for InvalidationGenerator<'cx, 'tcx, 'gcx> {
135135
StatementKind::Nop |
136136
StatementKind::AscribeUserType(..) |
137137
StatementKind::Retag { .. } |
138-
StatementKind::EscapeToRaw { .. } |
139138
StatementKind::StorageLive(..) => {
140139
// `Nop`, `AscribeUserType`, `Retag`, and `StorageLive` are irrelevant
141140
// to borrow check.

‎src/librustc_mir/borrow_check/nll/type_check/mod.rs‎

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1316,7 +1316,6 @@ impl<'a, 'gcx, 'tcx> TypeChecker<'a, 'gcx, 'tcx> {
13161316
| StatementKind::StorageDead(..)
13171317
| StatementKind::InlineAsm { .. }
13181318
| StatementKind::Retag { .. }
1319-
| StatementKind::EscapeToRaw { .. }
13201319
| StatementKind::Nop => {}
13211320
}
13221321
}

‎src/librustc_mir/dataflow/impls/borrows.rs‎

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -324,7 +324,6 @@ impl<'a, 'gcx, 'tcx> BitDenotation<'tcx> for Borrows<'a, 'gcx, 'tcx> {
324324
mir::StatementKind::SetDiscriminant { .. } |
325325
mir::StatementKind::StorageLive(..) |
326326
mir::StatementKind::Retag { .. } |
327-
mir::StatementKind::EscapeToRaw { .. } |
328327
mir::StatementKind::AscribeUserType(..) |
329328
mir::StatementKind::Nop => {}
330329

‎src/librustc_mir/dataflow/move_paths/builder.rs‎

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -302,7 +302,6 @@ impl<'b, 'a, 'gcx, 'tcx> Gatherer<'b, 'a, 'gcx, 'tcx> {
302302
"SetDiscriminant should not exist during borrowck");
303303
}
304304
StatementKind::Retag { .. } |
305-
StatementKind::EscapeToRaw { .. } |
306305
StatementKind::AscribeUserType(..) |
307306
StatementKind::Nop => {}
308307
}

‎src/librustc_mir/interpret/cast.rs‎

Lines changed: 7 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -44,34 +44,28 @@ impl<'a, 'mir, 'tcx, M: Machine<'a, 'mir, 'tcx>> EvalContext<'a, 'mir, 'tcx, M>
4444
}
4545

4646
Misc => {
47-
let src_layout = src.layout;
4847
let src = self.read_immediate(src)?;
4948

50-
// There are no casts to references
51-
assert!(!dest.layout.ty.is_region_ptr());
52-
// Hence we make all casts erase the tag
53-
let src = src.erase_tag().with_default_tag();
54-
55-
if self.type_is_fat_ptr(src_layout.ty) {
56-
match (src, self.type_is_fat_ptr(dest.layout.ty)) {
49+
if self.type_is_fat_ptr(src.layout.ty) {
50+
match (*src, self.type_is_fat_ptr(dest.layout.ty)) {
5751
// pointers to extern types
5852
(Immediate::Scalar(_),_) |
5953
// slices and trait objects to other slices/trait objects
6054
(Immediate::ScalarPair(..), true) => {
6155
// No change to immediate
62-
self.write_immediate(src, dest)?;
56+
self.write_immediate(*src, dest)?;
6357
}
6458
// slices and trait objects to thin pointers (dropping the metadata)
6559
(Immediate::ScalarPair(data, _), false) => {
6660
self.write_scalar(data, dest)?;
6761
}
6862
}
6963
} else {
70-
match src_layout.variants {
64+
match src.layout.variants {
7165
layout::Variants::Single { index } => {
72-
if let Some(def) = src_layout.ty.ty_adt_def() {
66+
if let Some(def) = src.layout.ty.ty_adt_def() {
7367
// Cast from a univariant enum
74-
assert!(src_layout.is_zst());
68+
assert!(src.layout.is_zst());
7569
let discr_val = def
7670
.discriminant_for_variant(*self.tcx, index)
7771
.val;
@@ -84,7 +78,7 @@ impl<'a, 'mir, 'tcx, M: Machine<'a, 'mir, 'tcx>> EvalContext<'a, 'mir, 'tcx, M>
8478
layout::Variants::NicheFilling { .. } => {},
8579
}
8680

87-
let dest_val = self.cast_scalar(src.to_scalar()?, src_layout, dest.layout)?;
81+
let dest_val = self.cast_scalar(src.to_scalar()?, src.layout, dest.layout)?;
8882
self.write_scalar(dest_val, dest)?;
8983
}
9084
}

0 commit comments

Comments
 (0)