io.asm

....
global load_idt

load_idt:
    push rbp
    mov rbp, rsp

    lidt [rdi]

    mov rsp, rbp
    pop rbp

    ret



global gdt_flush
gdt_flush:
    lgdt [rdi]
    mov ax, 0x30
    mov ds, ax
    mov es, ax
    mov fs, ax
    mov gs, ax
    mov ss, ax
    pop rdi
    mov rax, 0x28
    push rax
    push rdi
    retfq