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