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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
|
// only-aarch64
// only-linux
// needs-asm-support
// run-pass
#![feature(thread_local, asm_sym)]
use std::arch::asm;
extern "C" fn f1() -> i32 {
111
}
// The compiler will generate a shim to hide the caller location parameter.
#[track_caller]
fn f2() -> i32 {
222
}
macro_rules! call {
($func:path) => {
unsafe {
let result: i32;
asm!("bl {}", sym $func,
out("w0") result,
out("x20") _, out("x21") _, out("x22") _,
out("x23") _, out("x24") _, out("x25") _,
out("x26") _, out("x27") _, out("x28") _,
);
result
}
}
}
macro_rules! static_addr {
($s:expr) => {
unsafe {
let result: *const u32;
asm!(
// ADRP gives the address of a 4KB page from a PC-relative address
"adrp {out}, {sym}",
// We then add the remaining lower 12 bits
"add {out}, {out}, #:lo12:{sym}",
out = out(reg) result,
sym = sym $s);
result
}
}
}
macro_rules! static_tls_addr {
($s:expr) => {
unsafe {
let result: *const u32;
asm!(
// Load the thread pointer register
"mrs {out}, TPIDR_EL0",
// Add the top 12 bits of the symbol's offset
"add {out}, {out}, :tprel_hi12:{sym}",
// And the bottom 12 bits
"add {out}, {out}, :tprel_lo12_nc:{sym}",
out = out(reg) result,
sym = sym $s
);
result
}
}
}
static S1: u32 = 111;
#[thread_local]
static S2: u32 = 222;
fn main() {
assert_eq!(call!(f1), 111);
assert_eq!(call!(f2), 222);
assert_eq!(static_addr!(S1), &S1 as *const u32);
assert_eq!(static_tls_addr!(S2), &S2 as *const u32);
std::thread::spawn(|| {
assert_eq!(static_addr!(S1), &S1 as *const u32);
assert_eq!(static_tls_addr!(S2), &S2 as *const u32);
})
.join()
.unwrap();
}
|
