#include "interrupts.h"
#include "types.h"
#include "asm.h"
#include "memory.h"
#include "video.h"

#define SIZEIDT		256	/* nombre de descripteurs */

#define BASEIDT		0x00000000	/* addr de la IDT */

 /* registre idt */
static struct idtr idtreg;

/* table de IDT */
static idtdes idt[256];

/******************************************************************************/

/* Initialise le controleur d'interruption 8259A */

void initpic(void)
{
	/* MASTER */
	/* Initialisation de ICW1 */
	outb(PIC1_CMD, ICW1_INIT + ICW1_ICW4);
	nop();
	/* Initialisation de ICW2 - vecteur de depart = 32 */
	outb(PIC1_DATA, 0x20);
	nop();
	/* Initialisation de ICW3 */
	outb(PIC1_DATA, 0x04);
	nop();
	/* Initialisation de ICW4 */
	outb(PIC1_DATA, ICW4_8086);
	nop();
	/* masquage des interruptions */
	outb(PIC1_DATA, 0xFF);
	nop();
	/* SLAVE */
	/* Initialisation de ICW1 */
	outb(PIC2_CMD, ICW1_INIT + ICW1_ICW4);
	nop();
	/* Initialisation de ICW2 - vecteur de depart = 96 */
	outb(PIC2_DATA, 0x60);
	nop();
	/* Initialisation de ICW3 */
	outb(PIC2_DATA, 0x02);
	nop();
	/* Initialisation de ICW4 */
	outb(PIC2_DATA, ICW4_8086);
	nop();
	/* masquage des interruptions */
	outb(PIC2_DATA, 0xFF);
	nop();
	/* Demasquage des irqs sauf clavier
	   outb(PIC1_DATA,0xFD);
	   nop();
	 */
}

/******************************************************************************/

/* Active une IRQ */

void enableirq(u8 irq)
{
	u16 port;
	cli();
	port = (((irq & 0x08) << 4) + PIC1_DATA);
	outb(port, inb(port) & ~(1 << (irq & 7)));
	sti();
}

/******************************************************************************/

/* Désactive une IRQ */

void disableirq(u8 irq)
{
	u16 port;
	cli();
	port = (((irq & 0x08) << 4) + PIC1_DATA);
	outb(port, inb(port) | (1 << (irq & 7)));
	sti();
}

/******************************************************************************/

/* Créé un descripteur pour l'IDT */

void makeidtdes(u32 offset, u16 select, u16 type, idtdes * desc)
{
	desc->offset0_15 = (offset & 0xffff);
	desc->select = select;
	desc->type = type;
	desc->offset16_31 = (offset & 0xffff0000) >> 16;
	return;
}

/******************************************************************************/

/* Change une entrée dans l'IDT */

void setidt(u32 offset, u16 select, u16 type, u16 index)
{
	cli();
	idtdes *desc;
	desc = idtreg.base;
	desc[index].offset0_15 = (offset & 0xffff);
	desc[index].select = select;
	desc[index].type = type;
	desc[index].offset16_31 = (offset & 0xffff0000) >> 16;
	sti();
}

/******************************************************************************/

/* Met une entrée dans l'IDT */

void putidt(u32 offset, u16 select, u16 type, u16 index)
{
	idtdes temp;
	makeidtdes(offset, select, type, &temp);
	idt[index] = temp;
}

/******************************************************************************/

/* Affiche une erreur CPU et fige l'ordinateur */

void cpuerror(const u8 * src)
{
	print("\033[31m***** ERREUR CPU ****\r\n -");
	print(src);
	dump_regs();
	while (true) {
		nop();
	}
}

/******************************************************************************/

/* Déclenché lors de l'appel d'une interruption */

void interruption()
{
	cli();
	pushf();
	pushad();
	print("Appel d'une interruption\r\n");
	popad();
	popf();
	sti();
	iret();
}

/******************************************************************************/

/* Les expections */

void exception0()
{
	print("divide error\r\n");
}

void exception1()
{
	cpuerror("debug exception\r\n");
}

void exception2()
{
	cpuerror("non-maskable hardware interrupt\r\n");
}

void exception3()
{
	cpuerror("INT3 instruction\r\n");
}

void exception4()
{
	cpuerror("INTO instruction detected overflow\r\n");
}

void exception5()
{
	print("BOUND instruction detected overrange\r\n");
}

void exception6()
{
	cpuerror("invalid instruction opcode\r\n");
}

void exception7()
{
	cpuerror("no coprocessor\r\n");
}

void exception8()
{
	cpuerror("double fault\r\n");
}

void exception9()
{
	cpuerror("coprocessor segment overrun\r\n");
}

void exception10()
{
	cpuerror("invalid task state segment (TSS)\r\n");
}

void exception11()
{
	cpuerror("segment not present\r\n");
}

void exception12()
{
	cpuerror("stack fault");
}

void exception13()
{
	cpuerror("general protection fault (GPF)\r\n");
}

void exception14()
{
	cpuerror("page fault\r\n");
}

void exception15()
{
	cpuerror("(reserved)\r\n");
}

void exception16()
{
	cpuerror("coprocessor error\r\n");
}

void exception17()
{
	cpuerror("alignment check\r\n");
}

void exception18()
{
	cpuerror("machine check");
}

/******************************************************************************/

/* Les IRQ par défaut */

void irq0()
{
	cli();
	pushf();
	pushad();
	print("irq 0");
	irqendmaster();
	popad();
	popf();
	sti();
	asm("addl  $0x0C, %esp;");
	iret();
}

void irq1()
{
	cli();
	pushf();
	pushad();
	print("irq 1");
	while ((inb(0x64) & 1) == 0) ;
	inb(0x60);
	irqendmaster();
	popad();
	popf();
	sti();
	asm("addl  $0x01C, %esp;");
	iret();
}

void irq2()
{
	cli();
	pushf();
	pushad();
	print("irq 2");
	irqendmaster();
	popad();
	popf();
	sti();
	asm("addl  $0x0C, %esp;");
	iret();
}

void irq3()
{
	cli();
	pushf();
	pushad();
	print("irq 3");
	irqendmaster();
	popad();
	popf();
	sti();
	asm("addl  $0x0C, %esp;");
	iret();
}

void irq4()
{
	cli();
	pushf();
	pushad();
	print("irq 4");
	irqendmaster();
	popad();
	popf();
	sti();
	asm("addl  $0x0C, %esp;");
	iret();
}

void irq5()
{
	cli();
	pushf();
	pushad();
	print("irq 5");
	irqendmaster();
	popad();
	popf();
	sti();
	asm("addl  $0x0C, %esp;");
	iret();
}

void irq6()
{
	cli();
	pushf();
	pushad();
	print("irq 6");
	irqendmaster();
	popad();
	popf();
	sti();
	asm("addl  $0x0C, %esp;");
	iret();
}

void irq7()
{
	cli();
	pushf();
	pushad();
	print("irq 7");
	irqendmaster();
	popad();
	popf();
	sti();
	asm("addl  $0x0C, %esp;");
	iret();
}

void irq8()
{
	cli();
	pushf();
	pushad();
	print("irq 8");
	irqendslave();
	irqendmaster();
	popad();
	popf();
	sti();
	asm("addl  $0x0C, %esp;");
	iret();
}

void irq9()
{
	cli();
	pushf();
	pushad();
	print("irq 9");
	irqendslave();
	irqendmaster();
	popad();
	popf();
	sti();
	asm("addl  $0x0C, %esp;");
	iret();
}

void irq10()
{
	cli();
	pushf();
	pushad();
	print("irq 10");
	irqendslave();
	irqendmaster();
	popad();
	popf();
	sti();
	asm("addl  $0x0C, %esp;");
	iret();
}

void irq11()
{
	cli();
	pushf();
	pushad();
	print("irq 11");
	irqendslave();
	irqendmaster();
	popad();
	popf();
	sti();
	asm("addl  $0x0C, %esp;");
	iret();
}

void irq12()
{
	cli();
	pushf();
	pushad();
	print("irq 12");
	while ((inb(0x64) & 1) == 0) ;
	inb(0x60);
	irqendslave();
	irqendmaster();
	popad();
	popf();
	sti();
	asm("addl  $0x1C, %esp;");
	iret();
}

void irq13()
{
	cli();
	pushf();
	pushad();
	print("irq 13");
	irqendslave();
	irqendmaster();
	popad();
	popf();
	sti();
	asm("addl  $0x0C, %esp;");
	iret();
}

void irq14()
{
	cli();
	pushf();
	pushad();
	print("irq 14");
	irqendslave();
	irqendmaster();
	popad();
	popf();
	sti();
	asm("addl  $0x0C, %esp;");
	iret();
}

void irq15()
{
	cli();
	print("irq 15");
	irqendslave();
	irqendmaster();
	popad();
	popf();
	sti();
	asm("addl  $0x0C, %esp;");
	iret();
}

/******************************************************************************/

/* Initialise une IDT */

void initidt(void)
{
	u16 i;
	putidt((u32) exception0, 0x20, INTGATE, 0);
	putidt((u32) exception1, 0x20, INTGATE, 1);
	putidt((u32) exception2, 0x20, INTGATE, 2);
	putidt((u32) exception3, 0x20, INTGATE, 3);
	putidt((u32) exception4, 0x20, INTGATE, 4);
	putidt((u32) exception5, 0x20, INTGATE, 5);
	putidt((u32) exception6, 0x20, INTGATE, 6);
	putidt((u32) exception7, 0x20, INTGATE, 7);
	putidt((u32) exception8, 0x20, INTGATE, 8);
	putidt((u32) exception9, 0x20, INTGATE, 9);
	putidt((u32) exception10, 0x20, INTGATE, 10);
	putidt((u32) exception11, 0x20, INTGATE, 11);
	putidt((u32) exception12, 0x20, INTGATE, 12);
	putidt((u32) exception13, 0x20, INTGATE, 13);
	putidt((u32) exception14, 0x20, INTGATE, 14);
	putidt((u32) exception15, 0x20, INTGATE, 15);
	putidt((u32) exception16, 0x20, INTGATE, 16);
	putidt((u32) exception17, 0x20, INTGATE, 17);
	putidt((u32) exception18, 0x20, INTGATE, 18);
	for (i = 19; i < 32; i++) {
		putidt((u32) interruption, 0x20, TRAPGATE, i);
	}
	putidt((u32) irq0, 0x20, INTGATE, 32);
	putidt((u32) irq1, 0x20, INTGATE, 33);
	putidt((u32) irq2, 0x20, INTGATE, 34);
	putidt((u32) irq3, 0x20, INTGATE, 35);
	putidt((u32) irq4, 0x20, INTGATE, 36);
	putidt((u32) irq5, 0x20, INTGATE, 37);
	putidt((u32) irq6, 0x20, INTGATE, 38);
	putidt((u32) irq7, 0x20, INTGATE, 39);
	for (i = 40; i < 96; i++) {
		putidt((u32) interruption, 0x20, TRAPGATE, i);
	}
	putidt((u32) irq8, 0x20, INTGATE, 96);
	putidt((u32) irq9, 0x20, INTGATE, 97);
	putidt((u32) irq10, 0x20, INTGATE, 98);
	putidt((u32) irq11, 0x20, INTGATE, 99);
	putidt((u32) irq12, 0x20, INTGATE, 100);
	putidt((u32) irq13, 0x20, INTGATE, 101);
	putidt((u32) irq14, 0x20, INTGATE, 102);
	putidt((u32) irq15, 0x20, INTGATE, 103);
	for (i = 104; i < SIZEIDT; i++) {
		putidt((u32) interruption, 0x20, TRAPGATE, i);
	}
	/* initialise le registre idt */
	idtreg.limite = SIZEIDT * 8;
	idtreg.base = BASEIDT;
	/* recopie de la IDT a son adresse */
	memcpy(&idt, (u8 *) idtreg.base, idtreg.limite, 1);
	/* chargement du registre IDTR */
	lidt(&idtreg);
}

/******************************************************************************/

/* 8253/8254 PIT (Programmable Interval Timer) Timer ajustable */

void inittimer()
{
    outb(TIMER_MODE, RATE_GENERATOR);
    outb(TIMER0, (u8) (TIMER_FREQ/HZ) );
    outb(TIMER0, (u8) ((TIMER_FREQ/HZ) >> 8));
}