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-x86_64
// only-linux
// needs-asm-support
// run-pass
#![feature(thread_local)]
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!("call {}", sym $func,
out("rax") result,
out("rcx") _, out("rdx") _, out("rdi") _, out("rsi") _,
out("r8") _, out("r9") _, out("r10") _, out("r11") _,
out("xmm0") _, out("xmm1") _, out("xmm2") _, out("xmm3") _,
out("xmm4") _, out("xmm5") _, out("xmm6") _, out("xmm7") _,
out("xmm8") _, out("xmm9") _, out("xmm10") _, out("xmm11") _,
out("xmm12") _, out("xmm13") _, out("xmm14") _, out("xmm15") _,
);
result
}
}
}
macro_rules! static_addr {
($s:expr) => {
unsafe {
let result: *const u32;
// LEA performs a RIP-relative address calculation and returns the address
asm!("lea {}, [rip + {}]", out(reg) result, sym $s);
result
}
}
}
macro_rules! static_tls_addr {
($s:expr) => {
unsafe {
let result: *const u32;
asm!(
"
# Load TLS base address
mov {out}, qword ptr fs:[0]
# Calculate the address of sym in the TLS block. The @tpoff
# relocation gives the offset of the symbol from the start
# of the TLS block.
lea {out}, [{out} + {sym}@tpoff]
",
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();
}
|
