Use \xff\0 as terminator

This prevents people depending on nul-termination.

Author: bjorn3

compiler/rustc_const_eval/src/interpret/intrinsics.rs

@@ -28,12 +28,12 @@ pub(crate) fn alloc_type_name<'tcx>(
     tcx: TyCtxt<'tcx>,
     ty: Ty<'tcx>,
 ) -> (ConstAllocation<'tcx>, u64) {
-    let mut path = crate::util::type_name(tcx, ty);
+    let mut path = crate::util::type_name(tcx, ty).into_bytes();
     let path_len = path.len().try_into().unwrap();
-    if !path.contains('\0') {
-        path.push('\0');
+    if !path.contains(&0) {
+        path.extend(b"\xff\0");
     };
-    let alloc = Allocation::from_bytes_byte_aligned_immutable(path.into_bytes(), ());
+    let alloc = Allocation::from_bytes_byte_aligned_immutable(path, ());
     (tcx.mk_const_alloc(alloc), path_len)
 }
 

compiler/rustc_const_eval/src/interpret/place.rs

@@ -1021,7 +1021,7 @@ where
     ) -> InterpResult<'tcx, MPlaceTy<'tcx, M::Provenance>> {
         let ptr = if !s.contains('\0') {
             let mut bytes = s.as_bytes().to_owned();
-            bytes.push(0);
+            bytes.extend(b"\xff\0");
             self.allocate_bytes_dedup(&bytes)?
         } else {
             self.allocate_bytes_dedup(s.as_bytes())?

compiler/rustc_mir_build/src/builder/expr/as_constant.rs

@@ -122,9 +122,9 @@ fn lit_to_mir_constant<'tcx>(tcx: TyCtxt<'tcx>, lit_input: LitToConstInput<'tcx>
         (ast::LitKind::Str(s, _), ty::Ref(_, inner_ty, _)) if inner_ty.is_str() => {
             let s = s.as_str();
             let allocation = if !s.contains('\0') {
-                let mut s = s.to_owned();
-                s.push('\0');
-                Allocation::from_bytes_byte_aligned_immutable(s.as_bytes(), ())
+                let mut s = s.as_bytes().to_owned();
+                s.extend(b"\xff\0");
+                Allocation::from_bytes_byte_aligned_immutable(s, ())
             } else {
                 Allocation::from_bytes_byte_aligned_immutable(s.as_bytes(), ())
             };

tests/codegen/remap_path_prefix/main.rs

@@ -12,7 +12,7 @@ mod aux_mod;
 include!("aux_mod.rs");
 
 // Here we check that the expansion of the file!() macro is mapped.
-// CHECK: @alloc_4079a2e7607f89f86df6b8a72ba0dd06 = private unnamed_addr constant [35 x i8] c"/the/src/remap_path_prefix/main.rs\00"
+// CHECK: @alloc_643660180b5bd639a42b5b1707ce11a5 = private unnamed_addr constant [36 x i8] c"/the/src/remap_path_prefix/main.rs\FF\00"
 pub static FILE_PATH: &'static str = file!();
 
 fn main() {

tests/mir-opt/const_allocation.main.GVN.after.32bit.mir

@@ -33,27 +33,27 @@ ALLOC2 (size: 16, align: 4) {
     ╾ALLOC4<imm>╼ 03 00 00 00 ╾ALLOC5<imm>╼ 03 00 00 00 │ ╾──╼....╾──╼....
 }
 
-ALLOC4 (size: 4, align: 1) {
-    66 6f 6f 00                                     │ foo.
+ALLOC4 (size: 5, align: 1) {
+    66 6f 6f ff 00                                  │ foo..
 }
 
-ALLOC5 (size: 4, align: 1) {
-    62 61 72 00                                     │ bar.
+ALLOC5 (size: 5, align: 1) {
+    62 61 72 ff 00                                  │ bar..
 }
 
 ALLOC3 (size: 24, align: 4) {
     0x00 │ ╾ALLOC6<imm>╼ 03 00 00 00 ╾ALLOC7<imm>╼ 03 00 00 00 │ ╾──╼....╾──╼....
     0x10 │ ╾ALLOC8<imm>╼ 04 00 00 00                         │ ╾──╼....
 }
 
-ALLOC6 (size: 4, align: 1) {
-    6d 65 68 00                                     │ meh.
+ALLOC6 (size: 5, align: 1) {
+    6d 65 68 ff 00                                  │ meh..
 }
 
-ALLOC7 (size: 4, align: 1) {
-    6d 6f 70 00                                     │ mop.
+ALLOC7 (size: 5, align: 1) {
+    6d 6f 70 ff 00                                  │ mop..
 }
 
-ALLOC8 (size: 5, align: 1) {
-    6d c3 b6 70 00                                  │ m..p.
+ALLOC8 (size: 6, align: 1) {
+    6d c3 b6 70 ff 00                               │ m..p..
 }

tests/mir-opt/const_allocation.main.GVN.after.64bit.mir

@@ -36,12 +36,12 @@ ALLOC2 (size: 32, align: 8) {
     0x10 │ ╾ALLOC5<imm>╼ 03 00 00 00 00 00 00 00 │ ╾──────╼........
 }
 
-ALLOC4 (size: 4, align: 1) {
-    66 6f 6f 00                                     │ foo.
+ALLOC4 (size: 5, align: 1) {
+    66 6f 6f ff 00                                  │ foo..
 }
 
-ALLOC5 (size: 4, align: 1) {
-    62 61 72 00                                     │ bar.
+ALLOC5 (size: 5, align: 1) {
+    62 61 72 ff 00                                  │ bar..
 }
 
 ALLOC3 (size: 48, align: 8) {
@@ -50,14 +50,14 @@ ALLOC3 (size: 48, align: 8) {
     0x20 │ ╾ALLOC8<imm>╼ 04 00 00 00 00 00 00 00 │ ╾──────╼........
 }
 
-ALLOC6 (size: 4, align: 1) {
-    6d 65 68 00                                     │ meh.
+ALLOC6 (size: 5, align: 1) {
+    6d 65 68 ff 00                                  │ meh..
 }
 
-ALLOC7 (size: 4, align: 1) {
-    6d 6f 70 00                                     │ mop.
+ALLOC7 (size: 5, align: 1) {
+    6d 6f 70 ff 00                                  │ mop..
 }
 
-ALLOC8 (size: 5, align: 1) {
-    6d c3 b6 70 00                                  │ m..p.
+ALLOC8 (size: 6, align: 1) {
+    6d c3 b6 70 ff 00                               │ m..p..
 }

tests/ui-fulldeps/stable-mir/check_allocation.rs

@@ -78,12 +78,13 @@ fn check_bar(item: CrateItem) {
 
     let alloc_id_0 = alloc.provenance.ptrs[0].1.0;
     let GlobalAlloc::Memory(allocation) = GlobalAlloc::from(alloc_id_0) else { unreachable!() };
-    assert_eq!(allocation.bytes.len(), 4);
+    assert_eq!(allocation.bytes.len(), 5);
     assert_eq!(allocation.bytes[0].unwrap(), Char::CapitalB.to_u8());
     assert_eq!(allocation.bytes[1].unwrap(), Char::SmallA.to_u8());
     assert_eq!(allocation.bytes[2].unwrap(), Char::SmallR.to_u8());
-    assert_eq!(allocation.bytes[3].unwrap(), Char::Null.to_u8());
-    assert_eq!(std::str::from_utf8(&allocation.raw_bytes().unwrap()), Ok("Bar\0"));
+    assert_eq!(allocation.bytes[3].unwrap(), 0xff);
+    assert_eq!(allocation.bytes[4].unwrap(), Char::Null.to_u8());
+    assert_eq!(allocation.raw_bytes().unwrap(), b"Bar\xff\0");
 }
 
 /// Check the allocation data for static `C_STR`.

tests/ui-fulldeps/stable-mir/check_transform.rs

@@ -17,14 +17,15 @@ extern crate rustc_driver;
 extern crate rustc_interface;
 extern crate stable_mir;
 
+use std::convert::TryFrom;
+use std::io::Write;
+use std::ops::ControlFlow;
+
 use stable_mir::mir::alloc::GlobalAlloc;
 use stable_mir::mir::mono::Instance;
 use stable_mir::mir::{Body, ConstOperand, Operand, Rvalue, StatementKind, TerminatorKind};
 use stable_mir::ty::{ConstantKind, MirConst};
 use stable_mir::{CrateDef, CrateItems, ItemKind};
-use std::convert::TryFrom;
-use std::io::Write;
-use std::ops::ControlFlow;
 
 const CRATE_NAME: &str = "input";
 
@@ -37,17 +38,17 @@ fn test_transform() -> ControlFlow<()> {
     let target_fn = *get_item(&items, (ItemKind::Fn, "dummy")).unwrap();
     let instance = Instance::try_from(target_fn).unwrap();
     let body = instance.body().unwrap();
-    check_msg(&body, "oops\0");
+    check_msg(&body, b"oops\xff\0");
 
     let new_msg = "new panic message";
     let new_body = change_panic_msg(body, new_msg);
-    check_msg(&new_body, new_msg);
+    check_msg(&new_body, new_msg.as_bytes());
 
     ControlFlow::Continue(())
 }
 
 /// Check that the body panic message matches the given message.
-fn check_msg(body: &Body, expected: &str) {
+fn check_msg(body: &Body, expected: &[u8]) {
     let msg = body
         .blocks
         .iter()
@@ -80,7 +81,7 @@ fn check_msg(body: &Body, expected: &str) {
                     unreachable!()
                 };
                 let bytes = val.raw_bytes().unwrap();
-                Some(std::str::from_utf8(&bytes).unwrap().to_string())
+                Some(bytes.to_owned())
             }
             _ => None,
         })