example/mini_core_hello_world.rs

   1 // Adapted from https://github.com/sunfishcode/mir2cranelift/blob/master/rust-examples/nocore-hello-world.rs
   2
   3 #![feature(
   4     no_core, unboxed_closures, start, lang_items, box_syntax, never_type, linkage,
   5     extern_types, thread_local
   6 )]
   7 #![no_core]
   8 #![allow(dead_code, non_camel_case_types)]
   9
  10 extern crate mini_core;
  11
  12 use mini_core::*;
  13 use mini_core::libc::*;
  14
  15 unsafe extern "C" fn my_puts(s: *const i8) {
  16     puts(s);
  17 }
  18
  19 #[lang = "termination"]
  20 trait Termination {
  21     fn report(self) -> i32;
  22 }
  23
  24 impl Termination for () {
  25     fn report(self) -> i32 {
  26         unsafe {
  27             NUM = 6 * 7 + 1 + (1u8 == 1u8) as u8; // 44
  28             *NUM_REF as i32
  29         }
  30     }
  31 }
  32
  33 trait SomeTrait {
  34     fn object_safe(&self);
  35 }
  36
  37 impl SomeTrait for &'static str {
  38     fn object_safe(&self) {
  39         unsafe {
  40             puts(*self as *const str as *const i8);
  41         }
  42     }
  43 }
  44
  45 struct NoisyDrop {
  46     text: &'static str,
  47     inner: NoisyDropInner,
  48 }
  49
  50 struct NoisyDropInner;
  51
  52 impl Drop for NoisyDrop {
  53     fn drop(&mut self) {
  54         unsafe {
  55             puts(self.text as *const str as *const i8);
  56         }
  57     }
  58 }
  59
  60 impl Drop for NoisyDropInner {
  61     fn drop(&mut self) {
  62         unsafe {
  63             puts("Inner got dropped!\0" as *const str as *const i8);
  64         }
  65     }
  66 }
  67
  68 impl SomeTrait for NoisyDrop {
  69     fn object_safe(&self) {}
  70 }
  71
  72 enum Ordering {
  73     Less = -1,
  74     Equal = 0,
  75     Greater = 1,
  76 }
  77
  78 #[lang = "start"]
  79 fn start<T: Termination + 'static>(
  80     main: fn() -> T,
  81     argc: isize,
  82     argv: *const *const u8,
  83 ) -> isize {
  84     if argc == 3 {
  85         unsafe { puts(*argv as *const i8); }
  86         unsafe { puts(*((argv as usize + intrinsics::size_of::<*const u8>()) as *const *const i8)); }
  87         unsafe { puts(*((argv as usize + 2 * intrinsics::size_of::<*const u8>()) as *const *const i8)); }
  88     }
  89
  90     main().report();
  91     0
  92 }
  93
  94 static mut NUM: u8 = 6 * 7;
  95 static NUM_REF: &'static u8 = unsafe { &NUM };
  96
  97 macro_rules! assert {
  98     ($e:expr) => {
  99         if !$e {
 100             panic(stringify!(! $e));
 101         }
 102     };
 103 }
 104
 105 macro_rules! assert_eq {
 106     ($l:expr, $r: expr) => {
 107         if $l != $r {
 108             panic(stringify!($l != $r));
 109         }
 110     }
 111 }
 112
 113 struct Unique<T: ?Sized> {
 114     pointer: *const T,
 115     _marker: PhantomData<T>,
 116 }
 117
 118 impl<T: ?Sized, U: ?Sized> CoerceUnsized<Unique<U>> for Unique<T> where T: Unsize<U> {}
 119
 120 unsafe fn zeroed<T>() -> T {
 121     let mut uninit = MaybeUninit { uninit: () };
 122     intrinsics::write_bytes(&mut uninit.value.value as *mut T, 0, 1);
 123     uninit.value.value
 124 }
 125
 126 fn take_f32(_f: f32) {}
 127 fn take_unique(_u: Unique<()>) {}
 128
 129 fn return_u128_pair() -> (u128, u128) {
 130     (0, 0)
 131 }
 132
 133 fn call_return_u128_pair() {
 134     return_u128_pair();
 135 }
 136
 137 fn main() {
 138     take_unique(Unique {
 139         pointer: 0 as *const (),
 140         _marker: PhantomData,
 141     });
 142     take_f32(0.1);
 143
 144     call_return_u128_pair();
 145
 146     let slice = &[0, 1] as &[i32];
 147     let slice_ptr = slice as *const [i32] as *const i32;
 148
 149     assert_eq!(slice_ptr as usize % 4, 0);
 150
 151     //return;
 152
 153     unsafe {
 154         printf("Hello %s\n\0" as *const str as *const i8, "printf\0" as *const str as *const i8);
 155
 156         let hello: &[u8] = b"Hello\0" as &[u8; 6];
 157         let ptr: *const i8 = hello as *const [u8] as *const i8;
 158         puts(ptr);
 159
 160         let world: Box<&str> = box "World!\0";
 161         puts(*world as *const str as *const i8);
 162         world as Box<dyn SomeTrait>;
 163
 164         assert_eq!(intrinsics::bitreverse(0b10101000u8), 0b00010101u8);
 165
 166         assert_eq!(intrinsics::bswap(0xabu8), 0xabu8);
 167         assert_eq!(intrinsics::bswap(0xddccu16), 0xccddu16);
 168         assert_eq!(intrinsics::bswap(0xffee_ddccu32), 0xccdd_eeffu32);
 169         assert_eq!(intrinsics::bswap(0x1234_5678_ffee_ddccu64), 0xccdd_eeff_7856_3412u64);
 170
 171         assert_eq!(intrinsics::size_of_val(hello) as u8, 6);
 172
 173         let chars = &['C', 'h', 'a', 'r', 's'];
 174         let chars = chars as &[char];
 175         assert_eq!(intrinsics::size_of_val(chars) as u8, 4 * 5);
 176
 177         let a: &dyn SomeTrait = &"abc\0";
 178         a.object_safe();
 179
 180         assert_eq!(intrinsics::size_of_val(a) as u8, 16);
 181         assert_eq!(intrinsics::size_of_val(&0u32) as u8, 4);
 182
 183         assert_eq!(intrinsics::min_align_of::<u16>() as u8, 2);
 184         assert_eq!(intrinsics::min_align_of_val(&a) as u8, intrinsics::min_align_of::<&str>() as u8);
 185
 186         assert!(!intrinsics::needs_drop::<u8>());
 187         assert!(intrinsics::needs_drop::<NoisyDrop>());
 188
 189         Unique {
 190             pointer: 0 as *const &str,
 191             _marker: PhantomData,
 192         } as Unique<dyn SomeTrait>;
 193
 194         struct MyDst<T: ?Sized>(T);
 195
 196         intrinsics::size_of_val(&MyDst([0u8; 4]) as &MyDst<[u8]>);
 197
 198         struct Foo {
 199             x: u8,
 200             y: !,
 201         }
 202
 203         unsafe fn uninitialized<T>() -> T {
 204             MaybeUninit { uninit: () }.value.value
 205         }
 206
 207         zeroed::<(u8, u8)>();
 208         #[allow(unreachable_code)]
 209         {
 210             if false {
 211                 zeroed::<!>();
 212                 zeroed::<Foo>();
 213                 uninitialized::<Foo>();
 214             }
 215         }
 216     }
 217
 218     let _ = box NoisyDrop {
 219         text: "Boxed outer got dropped!\0",
 220         inner: NoisyDropInner,
 221     } as Box<dyn SomeTrait>;
 222
 223     const FUNC_REF: Option<fn()> = Some(main);
 224     match FUNC_REF {
 225         Some(_) => {},
 226         None => assert!(false),
 227     }
 228
 229     match Ordering::Less {
 230         Ordering::Less => {},
 231         _ => assert!(false),
 232     }
 233
 234     [NoisyDropInner, NoisyDropInner];
 235
 236     let x = &[0u32, 42u32] as &[u32];
 237     match x {
 238         [] => assert_eq!(0u32, 1),
 239         [_, ref y @ ..] => assert_eq!(&x[1] as *const u32 as usize, &y[0] as *const u32 as usize),
 240     }
 241
 242     assert_eq!(((|()| 42u8) as fn(()) -> u8)(()), 42);
 243
 244     extern {
 245         #[linkage = "extern_weak"]
 246         static ABC: *const u8;
 247     }
 248
 249     {
 250         extern {
 251             #[linkage = "extern_weak"]
 252             static ABC: *const u8;
 253         }
 254     }
 255
 256     unsafe { assert_eq!(ABC as usize, 0); }
 257
 258     &mut (|| Some(0 as *const ())) as &mut dyn FnMut() -> Option<*const ()>;
 259
 260     let f = 1000.0;
 261     assert_eq!(f as u8, 255);
 262     let f2 = -1000.0;
 263     assert_eq!(f2 as i8, -128);
 264     assert_eq!(f2 as u8, 0);
 265
 266     static ANOTHER_STATIC: &u8 = &A_STATIC;
 267     assert_eq!(*ANOTHER_STATIC, 42);
 268
 269     check_niche_behavior();
 270
 271     extern "C" {
 272         type ExternType;
 273     }
 274
 275     struct ExternTypeWrapper {
 276         _a: ExternType,
 277     }
 278
 279     let nullptr = 0 as *const ();
 280     let extern_nullptr = nullptr as *const ExternTypeWrapper;
 281     extern_nullptr as *const ();
 282     let slice_ptr = &[] as *const [u8];
 283     slice_ptr as *const u8;
 284
 285     #[cfg(not(jit))]
 286     test_tls();
 287
 288     #[cfg(all(not(jit), target_os = "linux"))]
 289     unsafe {
 290         global_asm_test();
 291     }
 292 }
 293
 294 #[cfg(all(not(jit), target_os = "linux"))]
 295 extern "C" {
 296     fn global_asm_test();
 297 }
 298
 299 #[cfg(all(not(jit), target_os = "linux"))]
 300 global_asm! {
 301     "
 302     .global global_asm_test
 303     global_asm_test:
 304     // comment that would normally be removed by LLVM
 305     ret
 306     "
 307 }
 308
 309 #[repr(C)]
 310 enum c_void {
 311     _1,
 312     _2,
 313 }
 314
 315 type c_int = i32;
 316 type c_ulong = u64;
 317
 318 type pthread_t = c_ulong;
 319
 320 #[repr(C)]
 321 struct pthread_attr_t {
 322     __size: [u64; 7],
 323 }
 324
 325 #[link(name = "pthread")]
 326 extern "C" {
 327     fn pthread_attr_init(attr: *mut pthread_attr_t) -> c_int;
 328
 329     fn pthread_create(
 330         native: *mut pthread_t,
 331         attr: *const pthread_attr_t,
 332         f: extern "C" fn(_: *mut c_void) -> *mut c_void,
 333         value: *mut c_void
 334     ) -> c_int;
 335
 336     fn pthread_join(
 337         native: pthread_t,
 338         value: *mut *mut c_void
 339     ) -> c_int;
 340 }
 341
 342 #[thread_local]
 343 #[cfg(not(jit))]
 344 static mut TLS: u8 = 42;
 345
 346 #[cfg(not(jit))]
 347 extern "C" fn mutate_tls(_: *mut c_void) -> *mut c_void {
 348     unsafe { TLS = 0; }
 349     0 as *mut c_void
 350 }
 351
 352 #[cfg(not(jit))]
 353 fn test_tls() {
 354     unsafe {
 355         let mut attr: pthread_attr_t = zeroed();
 356         let mut thread: pthread_t = 0;
 357
 358         assert_eq!(TLS, 42);
 359
 360         if pthread_attr_init(&mut attr) != 0 {
 361             assert!(false);
 362         }
 363
 364         if pthread_create(&mut thread, &attr, mutate_tls, 0 as *mut c_void) != 0 {
 365             assert!(false);
 366         }
 367
 368         let mut res = 0 as *mut c_void;
 369         pthread_join(thread, &mut res);
 370
 371         // TLS of main thread must not have been changed by the other thread.
 372         assert_eq!(TLS, 42);
 373
 374         puts("TLS works!\n\0" as *const str as *const i8);
 375     }
 376 }
 377
 378 // Copied ui/issues/issue-61696.rs
 379
 380 pub enum Infallible {}
 381
 382 // The check that the `bool` field of `V1` is encoding a "niche variant"
 383 // (i.e. not `V1`, so `V3` or `V4`) used to be mathematically incorrect,
 384 // causing valid `V1` values to be interpreted as other variants.
 385 pub enum E1 {
 386     V1 { f: bool },
 387     V2 { f: Infallible },
 388     V3,
 389     V4,
 390 }
 391
 392 // Computing the discriminant used to be done using the niche type (here `u8`,
 393 // from the `bool` field of `V1`), overflowing for variants with large enough
 394 // indices (`V3` and `V4`), causing them to be interpreted as other variants.
 395 pub enum E2<X> {
 396     V1 { f: bool },
 397
 398     /*_00*/ _01(X), _02(X), _03(X), _04(X), _05(X), _06(X), _07(X),
 399     _08(X), _09(X), _0A(X), _0B(X), _0C(X), _0D(X), _0E(X), _0F(X),
 400     _10(X), _11(X), _12(X), _13(X), _14(X), _15(X), _16(X), _17(X),
 401     _18(X), _19(X), _1A(X), _1B(X), _1C(X), _1D(X), _1E(X), _1F(X),
 402     _20(X), _21(X), _22(X), _23(X), _24(X), _25(X), _26(X), _27(X),
 403     _28(X), _29(X), _2A(X), _2B(X), _2C(X), _2D(X), _2E(X), _2F(X),
 404     _30(X), _31(X), _32(X), _33(X), _34(X), _35(X), _36(X), _37(X),
 405     _38(X), _39(X), _3A(X), _3B(X), _3C(X), _3D(X), _3E(X), _3F(X),
 406     _40(X), _41(X), _42(X), _43(X), _44(X), _45(X), _46(X), _47(X),
 407     _48(X), _49(X), _4A(X), _4B(X), _4C(X), _4D(X), _4E(X), _4F(X),
 408     _50(X), _51(X), _52(X), _53(X), _54(X), _55(X), _56(X), _57(X),
 409     _58(X), _59(X), _5A(X), _5B(X), _5C(X), _5D(X), _5E(X), _5F(X),
 410     _60(X), _61(X), _62(X), _63(X), _64(X), _65(X), _66(X), _67(X),
 411     _68(X), _69(X), _6A(X), _6B(X), _6C(X), _6D(X), _6E(X), _6F(X),
 412     _70(X), _71(X), _72(X), _73(X), _74(X), _75(X), _76(X), _77(X),
 413     _78(X), _79(X), _7A(X), _7B(X), _7C(X), _7D(X), _7E(X), _7F(X),
 414     _80(X), _81(X), _82(X), _83(X), _84(X), _85(X), _86(X), _87(X),
 415     _88(X), _89(X), _8A(X), _8B(X), _8C(X), _8D(X), _8E(X), _8F(X),
 416     _90(X), _91(X), _92(X), _93(X), _94(X), _95(X), _96(X), _97(X),
 417     _98(X), _99(X), _9A(X), _9B(X), _9C(X), _9D(X), _9E(X), _9F(X),
 418     _A0(X), _A1(X), _A2(X), _A3(X), _A4(X), _A5(X), _A6(X), _A7(X),
 419     _A8(X), _A9(X), _AA(X), _AB(X), _AC(X), _AD(X), _AE(X), _AF(X),
 420     _B0(X), _B1(X), _B2(X), _B3(X), _B4(X), _B5(X), _B6(X), _B7(X),
 421     _B8(X), _B9(X), _BA(X), _BB(X), _BC(X), _BD(X), _BE(X), _BF(X),
 422     _C0(X), _C1(X), _C2(X), _C3(X), _C4(X), _C5(X), _C6(X), _C7(X),
 423     _C8(X), _C9(X), _CA(X), _CB(X), _CC(X), _CD(X), _CE(X), _CF(X),
 424     _D0(X), _D1(X), _D2(X), _D3(X), _D4(X), _D5(X), _D6(X), _D7(X),
 425     _D8(X), _D9(X), _DA(X), _DB(X), _DC(X), _DD(X), _DE(X), _DF(X),
 426     _E0(X), _E1(X), _E2(X), _E3(X), _E4(X), _E5(X), _E6(X), _E7(X),
 427     _E8(X), _E9(X), _EA(X), _EB(X), _EC(X), _ED(X), _EE(X), _EF(X),
 428     _F0(X), _F1(X), _F2(X), _F3(X), _F4(X), _F5(X), _F6(X), _F7(X),
 429     _F8(X), _F9(X), _FA(X), _FB(X), _FC(X), _FD(X), _FE(X), _FF(X),
 430
 431     V3,
 432     V4,
 433 }
 434
 435 fn check_niche_behavior () {
 436     if let E1::V2 { .. } = (E1::V1 { f: true }) {
 437         intrinsics::abort();
 438     }
 439
 440     if let E2::V1 { .. } = E2::V3::<Infallible> {
 441         intrinsics::abort();
 442     }
 443 }