caos_pit.c

#include "console.h"
#include "cpu/isr.h"
#include "drivers/keyboard.h"
#include "drivers/port.h"
#include "drivers/uart.h"
#include "drivers/vga.h"

enum {
    PIT_CRYSTAL_HZ = 1193182,

    CLOCK_PRECISION_HZ = 100,

    PIT_PROGRAM_REG = PIT_CRYSTAL_HZ / CLOCK_PRECISION_HZ,

    PIT_SELECT_CHANNEL0 = 0x0,
    PIT_SELECT_CHANNEL1 = 0x1,
    PIT_SELECT_CHANNEL2 = 0x2,
    PIT_SELECT_CHANNEL_RB = 0x3,
    PIT_ACCESS_MODE_LATCH_COUNT_VALUE_COMMAND = 0x0,
    PIT_ACCESS_MODE_LOBYTE_ONLY = 0x1,
    PIT_ACCESS_MODE_HIBYTE_ONLY = 0x2,
    PIT_ACCESS_MODE_LOHIBYTE = 0x3,
    PIT_MODE_INTERRUPT_ON_TERMINAL_COUNT = 0x0,
    PIT_MODE_HW_ONESHOT = 0x1,
    PIT_MODE_RATE_GENERATOR = 0x2,
    PIT_MODES_SQUARE_WAVE_GENERATOR = 0x3,
    PIT_MODES_SW_TRIGGERRED_STROBE = 0x4,
    PIT_MODES_HW_TRIGGERRED_STROBE = 0x5,
};

static void timer_interrupt_handler(registers_t *r);

struct pit_command_t {
    unsigned char bcd : 1;
    unsigned char operating_mode : 3;
    unsigned char access_mode : 2;
    unsigned char select_channel : 2;
} __attribute__((packed));

static void init_pit() {
    cli();
    struct pit_command_t cmd = {
        .select_channel = PIT_SELECT_CHANNEL0,
        .access_mode = PIT_ACCESS_MODE_LOHIBYTE,
        .operating_mode = PIT_MODES_SQUARE_WAVE_GENERATOR,
        .bcd = 0,
    };
    port_byte_out(0x43, *(unsigned char *)(&cmd));
    port_byte_out(0x40, PIT_PROGRAM_REG & 0xff);
    port_byte_out(0x40, (PIT_PROGRAM_REG & 0xff00) >> 8);

    register_interrupt_handler(IRQ0, timer_interrupt_handler);
}

static int sleep_counter = 0;

static void timer_interrupt_handler(registers_t *r) {
    sleep_counter--;
}

static void msleep(int ms) {
    sleep_counter = ms / 10;
    while (sleep_counter > 0) {
        asm("hlt");
    }
}

void _start() {
    uartinit();
    init_keyboard();
    init_pit();
    load_idt();
    sti();

    vga_clear_screen();
    printk("\nYABLOKO\n> ");

    printk("\nNow ready to periodically sleeping for 1 sec...\n> ");
    while (1) {
        printk(".");
        msleep(1000);
    }

    while (1) {
        asm("hlt");
    }
}