A safe shutdown function has been added for the VGA, mouse, and keyboard drivers.

Author: OpenSoftware-World

Drivers/Keyboard/keyboard.c

@@ -5,6 +5,9 @@
 #define KBD_STAT 0x64
 
 static bool_t caps_lock = false;
+static bool_t kbd_enabled_driver = false;
+
+extern void sleep(uint32_t seconds);
 
 static const char kmap[128] __attribute__((section(".rodata"))) = {
     0,  27, '1','2','3','4','5','6','7','8','9','0','-','=', '\b',
@@ -19,6 +22,7 @@ static uint8_t buf_head = 0;
 static uint8_t buf_tail = 0;
 
 static char to_upper(char c) {
+    if (!kbd_enabled_driver) return;
     if (caps_lock && c >= 'a' && c <= 'z') {
         return c - 32;
     }
@@ -29,13 +33,16 @@ void kbd_init() {
     buf_head = 0;
     buf_tail = 0;
     caps_lock = false;
+    kbd_enabled_driver = true;
 }
 
 uint8_t kb_check() {
+    if (!kbd_enabled_driver) return;
     return buf_head != buf_tail;
 }
 
 char get_char() {
+    if (!kbd_enabled_driver) return;
     uint8_t scancode;
 
     while (1) {
@@ -64,7 +71,16 @@ char get_char() {
 }
 
 void kbd_hndlr(uint8_t scancode) {
+    if (!kbd_enabled_driver) return;
     if (scancode == 0x3A) {
         caps_lock = !caps_lock;
     }
+}
+
+void kbd_close() {
+    buf_head = 0;
+    buf_tail = 0;
+    sleep(1);
+    caps_lock = false;
+    kbd_enabled_driver = false;
 }

Drivers/Keyboard/keyboard.h

@@ -9,11 +9,12 @@ void kbd_init();
 char get_char();
 uint8_t kb_check();
 void kbd_hndlr(uint8_t scancode);
+void kbd_close();
 
 #define KBD_DRIVER_NAME "OpenKernel Keyboard Driver"
-#define KBD_DRIVER_VER "0.2"
+#define KBD_DRIVER_VER "0.3"
 #define KBD_DRIVER_DESC "A simple Keyboard driver for OpenKernel"
 #define KBD_DRIVER_AUTHOR "OpenSoftware-World"
-#define KBD_DRIVER_KRNL_VER "1.0"
+#define KBD_DRIVER_KRNL_VER "3.0"
 
 #endif

Drivers/Mouse/mouse.c

@@ -10,9 +10,11 @@
 #define MOUSE_LEFT_LIMIT 0
 #define MOUSE_RIGHT_LIMIT (VGA_WIDTH - 1)
 
-uint16_t* vga_buffer = (uint16_t*)0xB8000;
+extern uint16_t *vgabuffer = (uint16_t *)0xB8000;
 extern uint16_t screen_buffer[VGA_WIDTH * VGA_HEIGHT];
 
+static bool_t mouse_enabled_driver = false;
+
 static uint8_t mouse_cycle = 0;
 static int8_t mouse_byte[3];
 
@@ -48,9 +50,11 @@ void mouse_init() {
     mouse_y = 12;
     prev_mouse_x = mouse_x;
     prev_mouse_y = mouse_y;
+    mouse_enabled_driver = true;
 }
 
 void mouse_poll() {
+    if (!mouse_enabled_driver) return;
     uint8_t status = inb(0x64);
 
     if (!(status & 1)) return;
@@ -99,18 +103,21 @@ void mouse_poll() {
 }
 
 void mouse_draw() {
+    if (!mouse_enabled_driver) return;
     uint16_t index = mouse_y * VGA_WIDTH + mouse_x;
 
-    vga_buffer[index] = (0x1F << 8) | 0xDB;
+    vgabuffer[index] = (0x1F << 8) | 0xDB;
 }
 
 void mouse_restore() {
+    if (!mouse_enabled_driver) return;
     uint16_t index = prev_mouse_y * VGA_WIDTH + prev_mouse_x;
 
-    vga_buffer[index] = screen_buffer[index] ? screen_buffer[index] : (' ' | (0x1F << 8));
+    vgabuffer[index] = screen_buffer[index] ? screen_buffer[index] : (' ' | (0x1F << 8));
 }
 
 void mouse_update() {
+    if (!mouse_enabled_driver) return;
     mouse_poll();
 
     if (mouse_x != prev_mouse_x || mouse_y != prev_mouse_y) {
@@ -123,8 +130,13 @@ void mouse_update() {
     }
 }
 
-void screen_init() {
-    for (int i = 0; i < VGA_WIDTH * VGA_HEIGHT; i++) {
-        screen_buffer[i] = vga_buffer[i];
-    }
+void mouse_close() {
+    sleep(1);
+    mouse_cycle = 0;
+    mouse_x = 0;
+    mouse_y = 0;
+    sleep(1);
+    prev_mouse_x = mouse_x;
+    prev_mouse_y = mouse_y;
+    mouse_enabled_driver = false;
 }

Drivers/Mouse/mouse.h

@@ -7,7 +7,6 @@ extern int mouse_x;
 extern int mouse_y;
 extern int prev_mouse_x;
 extern int prev_mouse_y;
-extern uint16_t* vga_buffer;
 extern uint16_t prev_cell;
 
 void mouse_init();
@@ -17,12 +16,12 @@ void mouse_clear();
 void mouse_put_char(int x, int y, char c, uint8_t color);
 void mouse_restore();
 void mouse_update();
-void screen_init();
+void mouse_close();
 
 #define MOUSE_DRIVER_NAME "OpenKernel Mouse Driver"
-#define MOUSE_DRIVER_VER "0.1"
+#define MOUSE_DRIVER_VER "0.2"
 #define MOUSE_DRIVER_DESC "A simple mouse driver for OpenKernel"
 #define MOUSE_DRIVER_AUTHOR "OpenSoftware-World"
-#define MOUSE_DRIVER_KRNL_VER "2.0"
+#define MOUSE_DRIVER_KRNL_VER "3.0"
 
 #endif

Drivers/Vga/vga.c

@@ -3,12 +3,15 @@
 
 #define SCREEN_SIZE (VWIDTH * VHEIGHT)
 
-static uint16_t *vgabuffer = (uint16_t *)VBUFFER;
+uint16_t *vgabuffer = (uint16_t *)VBUFFER;
+static bool_t vga_enabled_driver = false;
 uint16_t screen_buffer[SCREEN_SIZE];
 static uint8_t txt_color = VGA_COLOR(VGA_COLOR_BLACK, VGA_COLOR_LIGHT_GREY);
 static uint8_t cx = 0; // cursor x
 static uint8_t cy = 0; // cursor y
 
+extern void sleep(uint32_t seconds);
+
 void vga_clear_screen(const char *mode) {
     if (mode[0] == 'F') {
         for (uint16_t y = 0; y < VHEIGHT; y++) {
@@ -57,6 +60,8 @@ void vga_set_color_scheme(uint8_t bg_color, uint8_t text_color) {
 }
 
 void ptchar(char c) {
+    if (!vga_enabled_driver) return;
+
     if (c == '\n') {
         cx = 0;
         cy++;
@@ -83,6 +88,8 @@ void ptchar(char c) {
 }
 
 void vga_print_scr(const char *str) {
+    if (!vga_enabled_driver) return;
+
     while (*str) {
         ptchar(*str++);
     }
@@ -186,6 +193,7 @@ void vga_newline() {
 
 void vga_init() {
     vga_clear_screen("F");
+    vga_enabled_driver = true;
 }
 
 void vga_bckspc() {
@@ -204,4 +212,22 @@ void vga_bckspc() {
     screen_buffer[cy * VWIDTH + cx] = cell;
 
     vga_set_cursor(cx, cy);
+}
+
+void vga_close() {
+    sleep(1);
+    vga_clear_screen("F");
+    cx = 0;
+    cy = 0;
+    sleep(1);
+    vgabuffer = 0;
+    screen_buffer = 0;
+    sleep(1);
+    vga_enabled_driver = false;
+}
+
+void screen_init() {
+    for (int i = 0; i < VWIDTH * VHEIGHT; i++) {
+        screen_buffer[i] = vgabuffer[i];
+    }
 }

Drivers/Vga/vga.h

@@ -8,6 +8,7 @@
 #define VBUFFER 0xB8000
 
 void vga_init();
+void vga_close();
 void vga_clear_screen(const char *mode);
 void vga_set_bg_color(uint8_t color);
 void vga_set_text_color(uint8_t color);
@@ -22,6 +23,7 @@ void vga_print_hex(uint32_t n);
 void vga_print_dec(int num);
 void vga_print_bin(unsigned int num);
 void vga_cursor_mode(const char *mode, uint8_t start, uint8_t end);
+void screen_init();
 
 #define VGA_COLOR_BLACK 0x0
 #define VGA_COLOR_BLUE 0x1
@@ -49,9 +51,9 @@ Example: VGA_COLOR_BLUE 0x1 -> 1: Blue background color
 #define VGA_ENTRY(ch, color) ((uint16_t)ch | (color << 8))
 
 #define VGA_DRIVER_NAME "OpenKernel VGA Driver"
-#define VGA_DRIVER_VER "0.4"
+#define VGA_DRIVER_VER "0.5"
 #define VGA_DRIVER_DESC "A simple VGA text mode driver for OpenKernel"
 #define VGA_DRIVER_AUTHOR "OpenSoftware-World"
-#define VGA_DRIVER_KRNL_VER "2.0"
+#define VGA_DRIVER_KRNL_VER "3.0"
 
 #endif