summaryrefslogtreecommitdiffstats
path: root/tests/mir-opt/building/custom/references.rs
blob: f87f6664c7a5c55b86fa488c7f6092de604b84f3 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
#![feature(custom_mir, core_intrinsics)]

extern crate core;
use core::intrinsics::mir::*;
use core::ptr::{addr_of, addr_of_mut};

// EMIT_MIR references.mut_ref.built.after.mir
#[custom_mir(dialect = "runtime", phase = "optimized")]
pub fn mut_ref(x: &mut i32) -> &mut i32 {
    mir!(
        let t: *mut i32;

        {
            t = addr_of_mut!(*x);
            RET = &mut *t;
            Retag(RET);
            Return()
        }
    )
}

// EMIT_MIR references.immut_ref.built.after.mir
#[custom_mir(dialect = "runtime", phase = "optimized")]
pub fn immut_ref(x: &i32) -> &i32 {
    mir!(
        let t: *const i32;

        {
            t = addr_of!(*x);
            RET = & *t;
            Retag(RET);
            Return()
        }
    )
}

// EMIT_MIR references.raw_pointer.built.after.mir
#[custom_mir(dialect = "built")]
pub fn raw_pointer(x: *const i32) -> *const i32 {
    // Regression test for a bug in which unsafetyck was not correctly turned off for
    // `dialect = "built"`
    mir!({
        RET = addr_of!(*x);
        Return()
    })
}

// EMIT_MIR references.raw_pointer_offset.built.after.mir
#[custom_mir(dialect = "built")]
pub fn raw_pointer_offset(x: *const i32) -> *const i32 {
    mir!({
        RET = Offset(x, 1_isize);
        Return()
    })
}

fn main() {
    let mut x = 5;
    let arr = [1, 2];
    assert_eq!(*mut_ref(&mut x), 5);
    assert_eq!(*immut_ref(&x), 5);
    unsafe {
        assert_eq!(*raw_pointer(addr_of!(x)), 5);
        assert_eq!(*raw_pointer_offset(addr_of!(arr[0])), 2);
    }
}