drivers: create a subdir just for driver code

Create a subdirectory branching from the project root. This directory
will contain nothing but driver and device code.

Signed-off-by: Danny Holman <dholman@gymli.org>

2 files changed, 251 insertions, 0 deletions

diff --git a/drivers/input/keyboard.c b/drivers/input/keyboard.c
new file mode 100644
index 0000000..f2a8eba
--- /dev/null
+++ b/drivers/input/keyboard.c
@@ -0,0 +1,142 @@
+#include <kernel/input/keyboard.h>
+#include <kernel/tty/tty_vga.h>
+#include <kernel/tty.h>
+#include <kernel/kthread.h>
+#include <kernel/pic.h>
+#include <libk/string.h>
+
+static uint8_t keymap_modifiers[256] = {
+        [0x1D] KB_CTL,
+        [0x2A] KB_SHIFT,
+        [0x36] KB_SHIFT,
+        [0x38] KB_ALT,
+        [0x9D] KB_CTL,
+        [0xB8] KB_ALT
+};
+
+static uint8_t keymap_toggles[256] = {
+        [0x3A] KB_CPSLK,
+        [0x45] KB_NUMLK,
+        [0x46] KB_SCLLK
+};
+
+static uint8_t keymap_normal[256] = {
+        0, 0x1B, '1', '2', '3', '4', '5', '6',
+        '7', '8', '9', '0', '-', '=', '\b', '\t',
+        'q', 'w', 'e', 'r', 't', 'y', 'u', 'i',
+        'o', 'p', '[', ']', '\n', 0, 'a', 's',
+        'd', 'f', 'g', 'h', 'j', 'k', 'l', ';',
+        '\'', '`', 0, '\\', 'z', 'x', 'c', 'v',
+        'b', 'n', 'm', ',', '.', '/', 0, '*',
+        0, ' ', 0, 0, 0, 0, 0, 0,
+        0, 0, 0, 0, 0, 0, 0, '7',
+        '8', '9', '-', '4', '5', '6', '+', '1',
+        '2', '3', '0', '.', 0, 0, 0, 0,
+        [0x9C] '\n',
+        [0xB5] '/',
+        [0xC8] KB_UP,
+        [0xD0] KB_DOWN,
+        [0xC9] KB_PGUP,
+        [0xD1] KB_PGDN,
+        [0xCB] KB_LEFT,
+        [0xCD] KB_RGHT,
+        [0x97] KB_HOME,
+        [0xCF] KB_END,
+        [0xD2] KB_INS,
+        [0xD3] KB_DEL
+};
+
+static uint8_t keymap_shifted[256] = {
+        0, 0x33, '!', '@', '#', '$', '%', '^',
+        '&', '*', '(', ')', '_', '+', '\b', '\t',
+        'Q', 'W', 'E', 'R', 'T', 'Y', 'U', 'I',
+        'O', 'P', '{', '}', '\n', 0, 'A', 'S',
+        'D', 'F', 'G', 'H', 'J', 'K', 'L', ':',
+        '"', '~', 0, '|', 'Z', 'X', 'C', 'V',
+        'B', 'N', 'M', '<', '>', '?', 0, '*',
+        0, ' ', 0, 0, 0, 0, 0, 0,
+        0, 0, 0, 0, 0, 0, 0, '7',
+        '8', '9', '-', '4', '5', '6', '+', '1',
+        '2', '3', '0', '.', 0, 0, 0, 0,
+        [0x9C] '\n',
+        [0xB5] '/',
+        [0xC8] KB_UP,
+        [0xD0] KB_DOWN,
+        [0xC9] KB_PGUP,
+        [0xD1] KB_PGDN,
+        [0xCB] KB_LEFT,
+        [0xCD] KB_RGHT,
+        [0x97] KB_HOME,
+        [0xCF] KB_END,
+        [0xD2] KB_INS,
+        [0xD3] KB_DEL
+};
+
+static uint8_t keymap_control[256] = {
+        0, 0, 0, 0, 0, 0, 0, 0,
+        0, 0, 0, 0, 0, 0, 0, 0,
+        C('Q'), C('W'), C('E'), C('R'), C('T'), C('Y'), C('U'), C('I'),
+        C('O'), C('P'), 0, 0, '\r', 0, C('A'), C('S'),
+        C('D'), C('F'), C('G'), C('H'), C('J'), C('K'), C('L'), 0,
+        0, 0, 0, C('\\'), C('Z'), C('X'), C('C'), C('V'),
+        C('B'), C('N'), C('M'), 0, 0, C('/'), 0, 0,
+        [0x9C] '\r',
+        [0xB5] C('/'),
+        [0xC8] KB_UP,
+        [0xD0] KB_DOWN,
+        [0xC9] KB_PGUP,
+        [0xD1] KB_PGDN,
+        [0xCB] KB_LEFT,
+        [0xCD] KB_RGHT,
+        [0x97] KB_HOME,
+        [0xCF] KB_END,
+        [0xD2] KB_INS,
+        [0xD3] KB_DEL
+};
+
+static uint8_t* keymaps[4] = {
+        keymap_normal,
+        keymap_shifted,
+        keymap_control,
+        keymap_control,
+};
+
+char keyboard_getchar(void) {
+        static int shift = 0;
+        uint8_t st = inb(KB_STAT);
+        if ((st & KB_DIB) == 0)
+                return -1;
+
+        uint8_t data = inb(KB_DATA);
+
+        if (data == 0xE0) {
+                shift |= KB_E0ESC;
+                return -1;
+        } else if (data & 0x80) {
+                if (!(shift & KB_E0ESC))
+                        data &= 0x7F;
+                shift &= ~(keymap_modifiers[data] | KB_E0ESC);
+                return -1;
+        } else if (shift & KB_E0ESC) {
+                data |= 0x80;
+                shift &= ~KB_E0ESC;
+        }
+
+        shift |= keymap_modifiers[data];
+        shift ^= keymap_toggles[data];
+        char c = keymaps[shift & (KB_CTL | KB_SHIFT)][data];
+        if (shift & KB_CPSLK) {
+                if ('a' <= c && c <= 'z')
+                        c += 'A' - 'a';
+                else if ('A' <= c && c <= 'Z')
+                        c += 'a' - 'A';
+        }
+        return c;
+}
+
+void keyboard_handler(struct isr_frame *frame) {
+        char c = keyboard_getchar();
+        if (c != -1)
+                tty_getchar(c);
+        return;
+}
diff --git a/drivers/video/framebuffer.c b/drivers/video/framebuffer.c
new file mode 100644
index 0000000..3df3df4
--- /dev/null
+++ b/drivers/video/framebuffer.c
@@ -0,0 +1,109 @@
+#include <kernel/video/framebuffer.h>
+#include <kernel/video/vga.h>
+#include <kernel/asm.h>
+#include <kernel/kthread.h>
+#include <kernel/paging.h>
+#include <kernel/pic.h>
+#include <libk/string.h>
+#include <stddef.h>
+#include <stdint.h>
+
+static size_t fb_row;
+static size_t fb_column;
+static uint8_t fb_color;
+static uint16_t *framebuffer;
+
+void _enable_cursor(uint8_t cursor_start, uint8_t cursor_end) {
+        outb(0x3D4, 0x0A);
+        outb(0x3D5, (inb(0x3D5) & 0xC0) | cursor_start);
+
+        outb(0x3D4, 0x0B);
+        outb(0x3D5, (inb(0x3D5) & 0xE0) | cursor_end);
+}
+
+void _update_cursor(size_t x, size_t y) {
+        uint16_t pos = y * VGA_WIDTH + x;
+        
+        outb(0x3D4, 0x0F);
+        outb(0x3D5, (uint8_t)(pos & 0xFF));
+        outb(0x3D4, 0x0E);
+        outb(0x3D5, (uint8_t)((pos >> 8) & 0xFF));
+}
+
+void _fb_putentryat(unsigned char c, uint8_t color, size_t x, size_t y) {
+        const size_t index = y * VGA_WIDTH + x;
+        framebuffer[index] = vga_entry(c, color);
+}
+
+void _fb_scroll(void) {
+        for (size_t i = 0; i < VGA_HEIGHT-1; i++) {
+                for (size_t j = 0; j < VGA_WIDTH; j++)
+                        framebuffer[i * VGA_WIDTH + j] = VGA_MEMORY[(i+1) * VGA_WIDTH + j];
+        }
+}
+
+void fb_init(void) {
+        fb_row = 0;
+        fb_column = 0;
+        fb_color = vga_entry_color(VGA_COLOR_LIGHT_GREY, VGA_COLOR_BLACK);
+        framebuffer = VGA_MEMORY;
+        for (size_t y = 0; y < VGA_HEIGHT; y++) {
+                for (size_t x = 0; x < VGA_WIDTH; x++)
+                        _fb_putentryat(' ', fb_color, x, y);
+        }
+        _enable_cursor(0, 0);
+        _update_cursor(0, 0);
+}
+
+void fb_setcolor(uint8_t color) {
+        fb_color = color;
+}
+
+void fb_setpos(int x, int y) {
+        fb_row = y;
+        fb_column = x;
+        _update_cursor(fb_column, fb_row+1);
+}
+
+void fb_offsetpos(int dx, int dy) {
+        fb_row += dy;
+        fb_column += dx;
+        _update_cursor(fb_column, fb_row+1);
+}
+
+void fb_putchar(char c) {
+        unsigned char uc = c;
+        switch (uc) {
+                case '\r':
+                        fb_column = 0;
+                        break;
+                case '\n':
+                        fb_column = 0;
+                        if (++fb_row == VGA_HEIGHT) {
+                                fb_row--;
+                                _fb_scroll();
+                        }
+                        break;
+                case '\t':
+                        fb_column += 4;
+                        if (++fb_column == VGA_WIDTH) {
+                                fb_column = 4;
+                                if (++fb_row == VGA_HEIGHT) {
+                                        fb_row--;
+                                        _fb_scroll();
+                                }
+                        }
+                        break;
+                default:
+                        _fb_putentryat(uc, fb_color, fb_column, fb_row);
+                        if (++fb_column == VGA_WIDTH) {
+                                fb_column = 0;
+                                if (++fb_row == VGA_HEIGHT) {
+                                        fb_row--;
+                                        _fb_scroll();
+                                }
+                        }
+                        break;
+        }
+        _update_cursor(fb_column, fb_row+1);
+}