diff options
Diffstat (limited to 'kernel/drivers/ps2')
| -rw-r--r-- | kernel/drivers/ps2/keyboard.c | 83 | ||||
| -rw-r--r-- | kernel/drivers/ps2/mouse.c | 196 | ||||
| -rw-r--r-- | kernel/drivers/ps2/ps2.c | 229 |
3 files changed, 508 insertions, 0 deletions
diff --git a/kernel/drivers/ps2/keyboard.c b/kernel/drivers/ps2/keyboard.c new file mode 100644 index 0000000..9a19b5c --- /dev/null +++ b/kernel/drivers/ps2/keyboard.c @@ -0,0 +1,83 @@ +// MIT License, Copyright (c) 2020 Marvin Borner + +#include <cpu.h> +#include <def.h> +#include <errno.h> +#include <fs.h> +#include <interrupts.h> +#include <mem.h> +#include <print.h> +#include <proc.h> +#include <ps2.h> +#include <stack.h> +#include <str.h> +#include <sys.h> + +PROTECTED static struct stack *queue = NULL; +PROTECTED static u32 dev_id = 0; + +static struct event_keyboard *event = NULL; +static int state = 0; +static int merged = 0; +static void keyboard_handler(struct regs *r) +{ + UNUSED(r); + u8 scancode = ps2_read_data(); + + // TODO: Support more than two-byte scancodes + if (scancode == 0xe0) { + merged = 0xe0; + state = 1; + return; + } else { + merged = scancode << 8 | merged; + } + + // TODO: "Merge" scancode to linux keycode? + /* printf("%x %x = %x\n", scancode, state ? 0xe0 : 0, merged); */ + + event = malloc(sizeof(*event)); + event->magic = KEYBOARD_MAGIC; + event->press = (scancode & 0x80) == 0; + event->scancode = event->press ? scancode : scancode & ~0x80; + stack_push_bot(queue, event); + + state = 0; + merged = 0; +} + +static res keyboard_read(void *buf, u32 offset, u32 count, struct vfs_dev *dev) +{ + UNUSED(dev); + if (stack_empty(queue)) + return -EINVAL; + + struct event_keyboard *e = stack_pop(queue); + memcpy_user(buf, (u8 *)e + offset, MIN(count, sizeof(*e))); + free(e); + return MIN(count, sizeof(*e)); +} + +static res keyboard_ready(void) +{ + return !stack_empty(queue); +} + +CLEAR void ps2_keyboard_reset(void) +{ + stack_clear(queue); +} + +CLEAR void ps2_keyboard_install(void) +{ + //keyboard_rate(); TODO: Fix keyboard rate? + irq_install_handler(1, keyboard_handler); + + queue = stack_new(); + struct vfs_dev *dev = zalloc(sizeof(*dev)); + dev->name = strdup("kbd"); + dev->type = DEV_CHAR; + dev->read = keyboard_read; + /* device_add(dev); */ + dev_id = dev->id; +} diff --git a/kernel/drivers/ps2/mouse.c b/kernel/drivers/ps2/mouse.c new file mode 100644 index 0000000..9b3fa2e --- /dev/null +++ b/kernel/drivers/ps2/mouse.c @@ -0,0 +1,196 @@ +// MIT License, Copyright (c) 2020 Marvin Borner + +#include <boot.h> +#include <cpu.h> +#include <errno.h> +#include <fs.h> +#include <interrupts.h> +#include <mem.h> +#include <print.h> +#include <proc.h> +#include <ps2.h> +#include <stack.h> +#include <str.h> +#include <sys.h> + +PROTECTED static struct stack *queue = NULL; +PROTECTED static u32 dev_id = 0; + +static struct event_mouse *event = NULL; +static char mouse_cycle = 0; +static char mouse_byte[3] = { 0 }; +static void mouse_handler(struct regs *r) +{ + UNUSED(r); + switch (mouse_cycle) { + case 0: + mouse_byte[0] = ps2_read_data(); + if (((mouse_byte[0] >> 3) & 1) == 1) + mouse_cycle++; + else + mouse_cycle = 0; + break; + case 1: + mouse_byte[1] = ps2_read_data(); + mouse_cycle++; + break; + case 2: + mouse_byte[2] = ps2_read_data(); + + event = malloc(sizeof(*event)); + event->magic = MOUSE_MAGIC; + event->diff_x = mouse_byte[1]; + event->diff_y = mouse_byte[2]; + event->but1 = mouse_byte[0] & 1; + event->but2 = (mouse_byte[0] >> 1) & 1; + event->but3 = (mouse_byte[0] >> 2) & 1; + stack_push_bot(queue, event); + mouse_cycle = 0; + break; + default: + break; + } +} + +#define MOUSE_WAIT_OUT 0 +#define MOUSE_WAIT_IN 1 +CLEAR static void mouse_serial_wait(u8 in) +{ + u32 time_out = 100000; + if (in) { + while (time_out--) + if (ps2_read_status().in_full) + return; + return; + } else { + while (time_out--) + if (ps2_read_status().out_full) + return; + return; + } +} + +CLEAR static void mouse_serial_write(u8 data) +{ + mouse_serial_wait(MOUSE_WAIT_IN); + ps2_write_command(0xd4); + mouse_serial_wait(MOUSE_WAIT_IN); + ps2_write_data(data); +} + +CLEAR static u8 mouse_serial_read(void) +{ + mouse_serial_wait(MOUSE_WAIT_OUT); + return ps2_read_data(); +} + +static res mouse_ready(void) +{ + return !stack_empty(queue); +} + +static res mouse_read(void *buf, u32 offset, u32 count, struct vfs_dev *dev) +{ + (void)dev; + if (stack_empty(queue)) + return -EINVAL; + + struct event_mouse *e = stack_pop(queue); + memcpy_user(buf, (u8 *)e + offset, MIN(count, sizeof(*e))); + free(e); + return MIN(count, sizeof(*e)); +} + +CLEAR void ps2_mouse_install(void) +{ + u8 status; + + // Enable auxiliary mouse device + mouse_serial_wait(MOUSE_WAIT_IN); + ps2_write_command(0xa8); + + // Enable interrupts + mouse_serial_wait(MOUSE_WAIT_IN); + ps2_write_command(0x20); + mouse_serial_wait(MOUSE_WAIT_OUT); + status = ps2_read_data() | 3; + mouse_serial_wait(MOUSE_WAIT_IN); + ps2_write_command(0x60); + mouse_serial_wait(MOUSE_WAIT_IN); + ps2_write_data(status); + + // Use default settings + mouse_serial_write(0xf6); + mouse_serial_read(); + + // Enable mousewheel + mouse_serial_write(0xf2); + mouse_serial_read(); + mouse_serial_read(); + mouse_serial_write(0xf3); + mouse_serial_read(); + mouse_serial_write(200); + mouse_serial_read(); + mouse_serial_write(0xf3); + mouse_serial_read(); + mouse_serial_write(100); + mouse_serial_read(); + mouse_serial_write(0xf3); + mouse_serial_read(); + mouse_serial_write(80); + mouse_serial_read(); + mouse_serial_write(0xf2); + mouse_serial_read(); + status = (u8)mouse_serial_read(); + if (status == 3) { + } + /* printf("Scrollwheel support!\n"); */ + + // Activate 4th and 5th mouse buttons + mouse_serial_write(0xf2); + mouse_serial_read(); + mouse_serial_read(); + mouse_serial_write(0xf3); + mouse_serial_read(); + mouse_serial_write(200); + mouse_serial_read(); + mouse_serial_write(0xf3); + mouse_serial_read(); + mouse_serial_write(200); + mouse_serial_read(); + mouse_serial_write(0xf3); + mouse_serial_read(); + mouse_serial_write(80); + mouse_serial_read(); + mouse_serial_write(0xf2); + mouse_serial_read(); + status = (u8)mouse_serial_read(); + if (status == 4) { + } + /* printf("4th and 5th mouse button support!\n"); */ + + /* TODO: Fix mouse laggyness + mouse_serial_write(0xE8); + mouse_serial_read(); + mouse_serial_write(0x03); + mouse_serial_read(); + mouse_serial_write(0xF3); + mouse_serial_read(); + mouse_serial_write(200); + mouse_serial_read(); */ + + // Enable mouse + mouse_serial_write(0xf4); + mouse_serial_read(); + + // Setup the mouse handler + irq_install_handler(12, mouse_handler); + + queue = stack_new(); + struct vfs_dev *dev = zalloc(sizeof(*dev)); + dev->name = strdup("mouse"); + dev->type = DEV_CHAR; + dev->read = mouse_read; + /* device_add(dev); */ + dev_id = dev->id; +} diff --git a/kernel/drivers/ps2/ps2.c b/kernel/drivers/ps2/ps2.c new file mode 100644 index 0000000..1c73855 --- /dev/null +++ b/kernel/drivers/ps2/ps2.c @@ -0,0 +1,229 @@ +// MIT License, Copyright (c) 2021 Marvin Borner + +#include <assert.h> +#include <cpu.h> +#include <def.h> +#include <print.h> +#include <ps2.h> + +#define PS2_TIMEOUT 100000 + +struct ps2_status ps2_read_status(void) +{ + u8 byte = inb(0x64); + return *(struct ps2_status *)&byte; +} + +static u8 ps2_wait_readable(void) +{ + u32 time_out = PS2_TIMEOUT; + while (time_out--) + if (ps2_read_status().in_full) + return 1; + /* print("PS/2 readable timeout\n"); */ + return 0; +} + +static u8 ps2_wait_writable(void) +{ + u32 time_out = PS2_TIMEOUT; + while (time_out--) + if (!ps2_read_status().out_full) + return 1; + /* print("PS/2 writable timeout\n"); */ + return 0; +} + +u8 ps2_read_data(void) +{ + if (ps2_wait_readable()) { + return inb(0x60); + } else { + return 0; + } +} + +u8 ps2_write_data(u8 byte) +{ + if (ps2_wait_writable()) { + outb(0x60, byte); + return 1; + } else { + return 0; + } +} + +u8 ps2_write_command(u8 byte) +{ + if (ps2_wait_writable()) { + outb(0x64, byte); + return 1; + } else { + return 0; + } +} + +CLEAR static struct ps2_config ps2_read_config(void) +{ + assert(ps2_write_command(0x20)); + u8 config = ps2_read_data(); + return *(struct ps2_config *)&config; +} + +CLEAR static u8 ps2_write_config(struct ps2_config config) +{ + // Select first byte + if (!ps2_write_command(0x60)) + return 0; + + return ps2_write_data(*(u8 *)&config); +} + +#define PS2_TYPE_STANDARD_MOUSE 0x0000 +#define PS2_TYPE_WHEEL_MOUSE 0x0003 +#define PS2_TYPE_BUTTON_MOUSE 0x0004 +#define PS2_TYPE_TRANSLATION_KEYBOARD1 0xab41 +#define PS2_TYPE_TRANSLATION_KEYBOARD2 0xabc1 +#define PS2_TYPE_STANDARD_KEYBOARD 0xab83 + +PROTECTED static struct { + u8 detected : 1; + struct { + u8 exists : 1; + u16 type; + } first; + struct { + u8 exists : 1; + u16 type; + } second; +} info = { 0 }; + +CLEAR static u8 ps2_write_device(u8 device, u8 data) +{ + u8 resp = PS2_RESEND; + for (u8 i = 0; resp == PS2_RESEND && i < 3; i++) { + if (device == 1) + ps2_write_command(0xd4); + ps2_write_data(data); + resp = ps2_read_data(); + } + + if (resp != PS2_ACK) + return 0; + + return 1; +} + +CLEAR static u8 ps2_device_keyboard(u16 type) +{ + return type == PS2_TYPE_TRANSLATION_KEYBOARD1 || type == PS2_TYPE_TRANSLATION_KEYBOARD2 || + type == PS2_TYPE_STANDARD_KEYBOARD; +} + +CLEAR static u8 ps2_device_mouse(u16 type) +{ + return type == PS2_TYPE_STANDARD_MOUSE || type == PS2_TYPE_WHEEL_MOUSE || + type == PS2_TYPE_BUTTON_MOUSE; +} + +CLEAR u8 ps2_keyboard_support(void) +{ + if (!info.detected) + return 0; + + // Find, reset and self-test + if ((info.first.exists && ps2_device_keyboard(info.first.type) && + ps2_write_device(0, 0xff)) || + (info.second.exists && ps2_device_keyboard(info.second.type) && + ps2_write_device(1, 0xff))) + return ps2_read_data() == 0xaa; + + return 0; +} + +CLEAR u8 ps2_mouse_support(void) +{ + if (!info.detected) + return 0; + + // Find, reset and self-test + if ((info.first.exists && ps2_device_mouse(info.first.type) && ps2_write_device(0, 0xff)) || + (info.second.exists && ps2_device_mouse(info.second.type) && ps2_write_device(1, 0xff))) + return ps2_read_data() == 0xaa; + + return 0; +} + +CLEAR void ps2_detect(void) +{ + // TODO: Read ACPI 8042 flag in FADT to verify PS/2 support + + // Disable PS/2 ports + ps2_write_command(0xad); + ps2_write_command(0xa7); + + // Get config and disable IRQs + struct ps2_config config = ps2_read_config(); + assert(!config.zero1 && !config.zero2); + config.first_int = 0; + config.second_int = 0; + ps2_write_config(config); + + // Test PS/2 controller + ps2_write_command(0xaa); + if (ps2_read_data() != 0x55) + return; + ps2_write_config(config); + + // Test first PS/2 port + ps2_write_command(0xab); + if (ps2_read_data() == 0x0) { + // Enable first port + ps2_write_command(0xae); + config.first_int = 1; + config.first_clock_disabled = 0; + info.first.exists = 1; + } + + // Test and enable second PS/2 port if available + if (config.second_clock_disabled) { + ps2_write_command(0xa9); + if (ps2_read_data() == 0x0) { + // Enable second port + ps2_write_command(0xa8); + config.second_int = 1; + config.second_clock_disabled = 0; + info.second.exists = 1; + } + } + + ps2_write_config(config); + + // Detect device type of first port + if (info.first.exists) { + if (!ps2_write_device(0, 0xf5)) + return; + + if (!ps2_write_device(0, 0xf2)) + return; + + u8 first = ps2_read_data(); + u8 second = ps2_read_data(); + info.first.type = (first << 8) | second; + } + + // Detect device type of second port + if (info.second.exists) { + if (!ps2_write_device(1, 0xf5)) + return; + + if (!ps2_write_device(1, 0xf2)) + return; + + u8 first = ps2_read_data(); + u8 second = ps2_read_data(); // This shall timeout if it's a mouse + info.second.type = (first << 8) | second; + } + + info.detected = 1; +} |