aboutsummaryrefslogtreecommitdiff
path: root/mirai/asm/cpu/state.zig
blob: 4b20a11b4a4fe8f6954e6b5091c1d5028890233f (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
67
//! CPU State Operations

pub fn read_rsp() u64 {
    var rsp: u64 = undefined;
    asm volatile ("mov %%rsp, %[rsp]"
        : [rsp] "=r" (rsp),
    );
    return rsp;
}

pub fn read_rbp() u64 {
    var rbp: u64 = undefined;
    asm volatile ("mov %%rbp, %[rbp]"
        : [rbp] "=r" (rbp),
    );
    return rbp;
}

pub fn rdtsc() u64 {
    var low: u32 = undefined;
    var high: u32 = undefined;
    asm volatile ("rdtsc"
        : [low] "={eax}" (low),
          [high] "={edx}" (high),
    );
    return (@as(u64, high) << 32) | low;
}

pub fn read_ds() u16 {
    var ds: u16 = undefined;
    asm volatile ("mov %%ds, %[ds]"
        : [ds] "=r" (ds),
    );
    return ds;
}

pub fn read_es() u16 {
    var es: u16 = undefined;
    asm volatile ("mov %%es, %[es]"
        : [es] "=r" (es),
    );
    return es;
}

pub fn read_fs() u16 {
    var fs: u16 = undefined;
    asm volatile ("mov %%fs, %[fs]"
        : [fs] "=r" (fs),
    );
    return fs;
}

pub fn read_gs() u16 {
    var gs: u16 = undefined;
    asm volatile ("mov %%gs, %[gs]"
        : [gs] "=r" (gs),
    );
    return gs;
}

pub fn clear_task_switched() void {
    asm volatile (
        \\mov %%cr0, %%rax
        \\and $~8, %%rax
        \\mov %%rax, %%cr0
        ::: .{ .rax = true });
}