rustc: make TyLayout::field(NonZero<*T>, 0) return &T.

1 files changed, 11 insertions, 5 deletions

diff --git a/src/librustc/ty/layout.rs b/src/librustc/ty/layout.rs
index a97574681a2..2919f25dc9d 100644
--- a/src/librustc/ty/layout.rs
+++ b/src/librustc/ty/layout.rs
@@ -2033,7 +2033,17 @@ impl<'a, 'tcx> TyLayout<'tcx> {
             ty::TyAdt(def, substs) => {
                 match self.variants {
                     Variants::Single { index } => {
-                        def.variants[index].fields[i].ty(tcx, substs)
+                        let mut field_ty = def.variants[index].fields[i].ty(tcx, substs);
+
+                        // Treat NonZero<*T> as containing &T.
+                        // This is especially useful for fat pointers.
+                        if Some(def.did) == tcx.lang_items().non_zero() {
+                            if let ty::TyRawPtr(mt) = field_ty.sty {
+                                field_ty = tcx.mk_ref(tcx.types.re_erased, mt);
+                            }
+                        }
+
+                        field_ty
                     }
 
                     // Discriminant field for enums (where applicable).
@@ -2109,10 +2119,6 @@ impl<'a, 'tcx> TyLayout<'tcx> {
                 let offset = self.fields.offset(0);
                 if let Abi::Scalar(value) = field.abi {
                     Ok(Some((offset, value)))
-                } else if let ty::TyRawPtr(_) = field.ty.sty {
-                    // If `NonZero` contains a non-scalar `*T`, it's
-                    // a fat pointer, which starts with a thin pointer.
-                    Ok(Some((offset, Pointer)))
                 } else {
                     Ok(None)
                 }