#include "interrupt.h"
#include "stdint.h"
#include "global.h"
#include "../lib/kernel/print.h"
#include "../lib/kernel/io.h"

#define IDT_DESC_CNT 0x30 // 目前支持的中断数
#define PIC_M_CTRL 0x20   // 主片的控制端口是 0x20
#define PIC_M_DATA 0x21   // 主片的数据端口是 0x21
#define PIC_S_CTRL 0xA0   // 从片的控制端口是 0xA0
#define PIC_S_DATA 0xA1   // 从片的数据端口是 0xA1

#define EFLAGS_IF 0x200 // eflags寄存器中的if位为1
#define GET_EFLAGS(EFLAG_VAR) asm volatile("pushfl; popl %0" : "=g"(EFLAG_VAR))

/* 中断门描述符结构体 */
// Example type_attributes values that people are likely to use (assuming DPL is 0):
// 32-bit Interrupt Gate: 0x8E (p=1, dpl=0b00, type=0b1110 => type_attributes=0b1000_1110=0x8E)
// 32-bit Trap Gate: 0x8F (p=1, dpl=0b00, type=0b1111 => type_attributes=1000_1111b=0x8F)
// Task Gate: 0x85 (p=1, dpl=0b00, type=0b0101 => type_attributes=0b1000_0101=0x85)
typedef struct
{
    uint16_t offset_low;  // 16位 偏移量的低 16 位 (0..15)
    uint16_t selector;    // 选择器 acode segment selector in GDT or LDT
    uint8_t dcount;       // 置 0
    uint8_t attribute;    // 述符类型 (bit 7:0)  gate type, dpl, and p fields
    uint16_t offset_high; // 16 位 偏移量的高 16 位 (16..31)
} __attribute__((packed)) gate_desc;

char *intr_name[IDT_DESC_CNT]; // 用于保存异常的名字
/****************************** 定义中断数组 **********************
 * 在 kernel.S 中定义的 intrXXentry 只是中断处理程序的入口，
 * 最终调用的 idt_table 中的处理程序*/
intr_handler idt_table[IDT_DESC_CNT]; // 中断处理函数数组
/***************************************************************/

static gate_desc idt[IDT_DESC_CNT];                 // 中断描述符表,本质上就是个中断门描述符数组
extern intr_handler intr_entry_table[IDT_DESC_CNT]; // 声明引用定义在 kernel.S 中的中断处理函数入口数组

/*创建中断门描述符*/
static void make_idt_desc(gate_desc *p_gdesc, uint8_t attr, intr_handler function)
{
    p_gdesc->offset_low = (uint32_t)((uintptr_t)function & 0x0000FFFF);
    p_gdesc->selector = SELECTOR_K_CODE;
    p_gdesc->dcount = 0;
    p_gdesc->attribute = attr;
    p_gdesc->offset_high = (uint32_t)(((uintptr_t)function & 0xFFFF0000) >> 16);
}
/* 初始化中断描述符表*/
static void itd_desc_init(void)
{
    int i;
    for (i = 0; i < IDT_DESC_CNT; i++)
    {
        make_idt_desc(&idt[i], IDT_DESC_ATTR_DPL0, intr_entry_table[i]);
    }
    put_str("   idt_desc_init done\n");
}

/* 初始化可编程中断控制器 8259A */
static void pic_init(void)
{
    /* 初始化主片*/
    outb(PIC_M_CTRL, 0x11); // ICW1: 边沿触发，级联8259， 需要 ICW4
    outb(PIC_M_DATA, 0x20); // ICW2: 起始中断向量号 0x20, 也就是 IRQ[0～7] 为 0x20~0x27
    outb(PIC_M_DATA, 0x04); // ICW3: IR2 接从片
    outb(PIC_M_DATA, 0x01); // ICW4: 8086 模式， 正常 EOI

    /* 初始化从片 */
    outb(PIC_S_CTRL, 0x11); // ICW1: 边沿触发，级联8259， 需要 ICW4
    outb(PIC_S_DATA, 0x28); // ICW2: 起始中断向量号 0x28, 也就是 IRQ[8～15] 以 0x28~0x2F
    outb(PIC_S_DATA, 0x02); // ICW3: 设置从片连接到主片的 IR2 引脚
    outb(PIC_S_DATA, 0x01); // ICW4: 8086 模式， 正常 EOI

    /* 打开主片上 IR0， 也就是目前只接受时钟产生的中断*/
    // outb(PIC_M_DATA, 0xfe);
    // outb(PIC_M_DATA, 0xfd); // 测试键盘，只打开键盘中断，其他全部关闭
    outb(PIC_M_DATA, 0xfc); // 测试键盘,时钟，开键盘,时钟中断，其他全部关闭 1111 1100
    outb(PIC_S_DATA, 0xff);

    put_str("   pic_init done\n");
}

// 通用中断处理函数, 一般用于处理异常
static void general_intr_handler(uint8_t vec_nr)
{
    if (vec_nr == 0x27 || vec_nr == 0x2f)
    {
        // IRQ7 和 IRQ15 会产生伪中断，无需处理
        // 0x2f 是从片8259A上的最后一个 IRQ 引脚，保留项
        return;
    }
    set_cursor(0); // 将光标置为屏幕左上角，从此处打印异常信息，方便阅读
    int cursor_pos = 0;
    while (cursor_pos < 320)
    { // 清空 4 行位置的字符
        put_char(' ');
        cursor_pos++;
    }
    set_cursor(0); // 重置光标位置
    put_str("!!!!!!! exception message begin !!!!!!!!!!!!!\n");
    set_cursor(88); // 从第 2 行第 8 个字符开始打印, 80 字符宽
    put_str("intr name: ");
    put_str(intr_name[vec_nr]);
    if (vec_nr == 14) // 若为 Pagefault, 将缺失的地址打印出来并悬停
    {
        int page_fault_vaddr = 0;
        asm("movl %%cr2, %0" : "=r"(page_fault_vaddr)); // cr2是存放page_fault的线性地址
        put_str("\nPage fault addr is: 0X");            // 输出缺失的地址
        put_int(page_fault_vaddr);
    }
    put_str("\n!!!!!!! exception message end !!!!!!!!!!!!!\n");
    // 能进入中断处理程序就表示已经处在关中断情况下, 不会出现调度进程的情况。
    // 所以此处的 while 不会再被中断打断，此处的死循环是为了保证当前任务继续运行
    while (1)
        ;
}

static void exception_init(void)
{
    int i;
    for (i = 0; i < IDT_DESC_CNT; i++)
    {
        /*idt_table 数组中的函数是进入中断后根据中断向量号调用的，见 kernel/kernel.S 的 call [itd_table + %1*4]*/
        idt_table[i] = general_intr_handler; // 默认为 general_intr_handler
        intr_name[i] = "unknown";
    }
    intr_name[0] = "#DE Divide Error";
    intr_name[1] = "#DB Debug Exception";
    intr_name[2] = "#NMI Interrupt";
    intr_name[3] = "#BP Breakpoint Exception";
    intr_name[4] = "#OF Overflow Exception";
    intr_name[5] = "#BR BOUND Range Exceeded Exception";
    intr_name[6] = "#UD Invalid Opcode Exception";
    intr_name[7] = "#NM Device Not Available Exception";
    intr_name[8] = "#DF Double Fault Exception";
    intr_name[9] = "#Coprocessor Segment Overrun";
    intr_name[10] = "#TS Invalid TSS Exception";
    intr_name[11] = "#NP Segment Not Present";
    intr_name[12] = "#SS Stack Fault Exception";
    intr_name[13] = "#GP General Protection Exception";
    intr_name[14] = "#PF Page-Fault Exception";
    // intr_name[15] 第 15 项是 intel 保留项，未使用
    intr_name[16] = "#MF x87 FPU Floating-Point Error";
    intr_name[17] = "#AC Alignment Check Exception";
    intr_name[18] = "#MC Machine-Check Exception";
    intr_name[19] = "#XF SIMD Floating-Point Exception";
    put_str("   exception_init done\n");
}

/* 中断初始化主函数 */
void itd_init(void)
{
    put_str("init_idt start\n");
    itd_desc_init();  // 初始化中断描述符表
    exception_init(); // 完成一般中断处理函数注册及异常名称注册
    pic_init();       // 初始化可编程中断控制器8259A

    /* 加载 idt */
    uint64_t idt_operand = ((sizeof(idt) - 1) | ((uint64_t)((uintptr_t)idt << 16)));
    asm volatile("lidt %0" ::"m"(idt_operand));
    put_str("itd_init done\n");
}

enum intr_status intr_get_status(void)
{
    uint32_t eflags = 0;
    GET_EFLAGS(eflags);
    return (EFLAGS_IF & eflags) ? INTR_ON : INTR_OFF;
}
enum intr_status intr_set_status(enum intr_status status)
{
    return status & INTR_ON ? intr_enable() : intr_disable();
}
enum intr_status intr_enable(void)
{
    enum intr_status old_status;
    if (INTR_ON == intr_get_status())
    {
        old_status = INTR_ON;
        return old_status;
    }
    else
    {
        old_status = INTR_OFF;
        asm volatile("sti"); // 开中断, sti 指令将IF位置1
        return old_status;
    }
}
enum intr_status intr_disable(void)
{
    enum intr_status old_status;
    if (INTR_ON == intr_get_status())
    {
        old_status = INTR_ON;
        asm volatile("cli" : : : "memory"); // 关中断, cli 指令将IF位置0
        return old_status;
    }
    else
    {
        old_status = INTR_OFF;
        return old_status;
    }
}

void register_handler(uint8_t vector_no, intr_handler function)
{ // idt_table 数组中的函数是进入中断后根据中断向量号调用的，
    // 见 kernel/kernel.S 的 call [itd_table + %1*4]
    idt_table[vector_no] = function;
}