aboutsummaryrefslogtreecommitdiff
path: root/mirai/invocations/kata/yield.zig
blob: 0da91afe9d5175bdb1562148735deb12ec84bfb6 (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
//! Yield invocation - Voluntarily give up CPU time to Sensei

const handler = @import("../handler.zig");
const kata_mod = @import("../../kata/kata.zig");
const result = @import("../../utils/types/result.zig");
const sensei = @import("../../kata/sensei/sensei.zig");

pub fn invoke(ctx: *handler.InvocationContext) void {
    const kata = sensei.get_current_kata() orelse return result.set_error(ctx);

    kata.state = kata_mod.State.Alive;

    // Save context before switching - schedule() may not return
    kata.context.rax = 0; // Return value for yield
    kata.context.rbx = ctx.rbx;
    kata.context.rcx = ctx.rcx;
    kata.context.rdx = ctx.rdx;
    kata.context.rsi = ctx.rsi;
    kata.context.rdi = ctx.rdi;
    kata.context.rbp = ctx.rbp;
    kata.context.rsp = ctx.rsp;
    kata.context.r8 = ctx.r8;
    kata.context.r9 = ctx.r9;
    kata.context.r10 = ctx.r10;
    kata.context.r11 = ctx.r11;
    kata.context.r12 = ctx.r12;
    kata.context.r13 = ctx.r13;
    kata.context.r14 = ctx.r14;
    kata.context.r15 = ctx.r15;
    kata.context.rip = ctx.rip;
    kata.context.rflags = ctx.rflags;
    kata.context.cs = ctx.cs;
    kata.context.ss = ctx.ss;

    if (!sensei.is_in_queue(kata)) {
        sensei.enqueue_kata(kata);
    }

    sensei.schedule();

    result.set_ok(ctx);
}