| David Tolnay | c5a52f9 | 2020-09-14 00:43:29 -0400 | [diff] [blame] | 1 | use alloc::string::String; |
| David Tolnay | 56377fc | 2021-04-10 12:18:31 -0700 | [diff] [blame^] | 2 | use core::mem::{self, MaybeUninit}; |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 3 | |
| David Tolnay | 56377fc | 2021-04-10 12:18:31 -0700 | [diff] [blame^] | 4 | // ABI compatible with C++ rust::String (not necessarily alloc::string::String). |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 5 | #[repr(C)] |
| 6 | pub struct RustString { |
| David Tolnay | 56377fc | 2021-04-10 12:18:31 -0700 | [diff] [blame^] | 7 | repr: [MaybeUninit<usize>; mem::size_of::<String>() / mem::size_of::<usize>()], |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 8 | } |
| 9 | |
| 10 | impl RustString { |
| 11 | pub fn from(s: String) -> Self { |
| David Tolnay | 56377fc | 2021-04-10 12:18:31 -0700 | [diff] [blame^] | 12 | unsafe { mem::transmute::<String, RustString>(s) } |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 13 | } |
| 14 | |
| 15 | pub fn from_ref(s: &String) -> &Self { |
| David Tolnay | db96ed9 | 2020-03-18 17:20:39 -0700 | [diff] [blame] | 16 | unsafe { &*(s as *const String as *const RustString) } |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 17 | } |
| 18 | |
| David Tolnay | f1c7f32 | 2020-08-27 00:46:01 -0700 | [diff] [blame] | 19 | pub fn from_mut(s: &mut String) -> &mut Self { |
| 20 | unsafe { &mut *(s as *mut String as *mut RustString) } |
| 21 | } |
| 22 | |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 23 | pub fn into_string(self) -> String { |
| David Tolnay | 56377fc | 2021-04-10 12:18:31 -0700 | [diff] [blame^] | 24 | unsafe { mem::transmute::<RustString, String>(self) } |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 25 | } |
| 26 | |
| 27 | pub fn as_string(&self) -> &String { |
| David Tolnay | 56377fc | 2021-04-10 12:18:31 -0700 | [diff] [blame^] | 28 | unsafe { &*(self as *const RustString as *const String) } |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 29 | } |
| David Tolnay | 40226ab | 2020-03-03 00:05:35 -0800 | [diff] [blame] | 30 | |
| 31 | pub fn as_mut_string(&mut self) -> &mut String { |
| David Tolnay | 56377fc | 2021-04-10 12:18:31 -0700 | [diff] [blame^] | 32 | unsafe { &mut *(self as *mut RustString as *mut String) } |
| David Tolnay | 40226ab | 2020-03-03 00:05:35 -0800 | [diff] [blame] | 33 | } |
| David Tolnay | 7db7369 | 2019-10-20 14:51:12 -0400 | [diff] [blame] | 34 | } |
| 35 | |
| David Tolnay | 56377fc | 2021-04-10 12:18:31 -0700 | [diff] [blame^] | 36 | const_assert_eq!(mem::size_of::<[usize; 3]>(), mem::size_of::<RustString>()); |
| 37 | const_assert_eq!(mem::size_of::<String>(), mem::size_of::<RustString>()); |
| 38 | const_assert_eq!(mem::align_of::<String>(), mem::align_of::<RustString>()); |