+#include <klibc/string.h>
+#include <lunaix/lxconsole.h>
+#include <lunaix/mm/pmm.h>
+#include <lunaix/mm/vmm.h>
+#include <lunaix/tty/console.h>
+#include <lunaix/tty/tty.h>
+
+static struct console lx_console;
+
+volatile int can_flush = 0;
+
+void
+lxconsole_init()
+{
+ memset(&lx_console, 0, sizeof(lx_console));
+ lx_console.buffer.data = VGA_BUFFER_VADDR + 0x1000;
+ lx_console.buffer.size = 8192;
+ mutex_init(&lx_console.buffer.lock);
+
+ // 分配控制台缓存
+ for (size_t i = 0; i < PG_ALIGN(lx_console.buffer.size); i += PG_SIZE) {
+ uintptr_t pa = pmm_alloc_page(KERNEL_PID, 0);
+ vmm_set_mapping(PD_REFERENCED,
+ (uintptr_t)lx_console.buffer.data + i,
+ pa,
+ PG_PREM_URW,
+ 0);
+ }
+
+ memset(lx_console.buffer.data, 0, lx_console.buffer.size);
+
+ lx_console.flush_timer = NULL;
+}
+
+void
+console_schedule_flush()
+{
+ // TODO make the flush on-demand rather than periodic
+}
+
+void
+console_view_up(struct fifo_buffer* buffer)
+{
+ // mutex_lock(&buffer->lock);
+ size_t p = buffer->rd_pos - 2;
+ while (p < buffer->rd_pos && p != buffer->wr_pos &&
+ ((char*)buffer->data)[p] != '\n') {
+ p--;
+ }
+ p++;
+
+ if (p < buffer->rd_pos) {
+ p = 0;
+ }
+
+ buffer->rd_pos = p;
+ // mutex_unlock(&buffer->lock);
+}
+
+void
+console_view_down(struct fifo_buffer* buffer)
+{
+ // mutex_lock(&buffer->lock);
+ size_t p = buffer->rd_pos;
+ while (p != buffer->wr_pos && ((char*)buffer->data)[p] != '\n') {
+ p = (p + 1) % buffer->size;
+ }
+
+ buffer->rd_pos = p + 1;
+ // mutex_unlock(&buffer->lock);
+}
+
+void
+__flush_cb(void* arg)
+{
+ if (mutex_on_hold(&lx_console.buffer.lock)) {
+ return;
+ }
+ if (!(lx_console.buffer.flags & FIFO_DIRTY)) {
+ return;
+ }
+
+ size_t pos = tty_flush_buffer(lx_console.buffer.data,
+ lx_console.buffer.rd_pos,
+ lx_console.buffer.wr_pos,
+ lx_console.buffer.size);
+ lx_console.flush_timer = NULL;
+ if (pos < lx_console.buffer.wr_pos) {
+ console_view_down(&lx_console.buffer);
+ } else {
+ // clear the dirty bit only if we have flush all the data
+ // that means: read pointer == write pointer
+ lx_console.buffer.flags &= ~FIFO_DIRTY;
+ }
+}
+
+void
+console_write(struct console* console, uint8_t* data, size_t size)
+{
+ mutex_lock(&console->buffer.lock);
+ uint8_t* buffer = console->buffer.data;
+ uintptr_t ptr = console->buffer.wr_pos;
+ uintptr_t rd_ptr = console->buffer.rd_pos;
+
+ for (size_t i = 0; i < size; i++) {
+ buffer[(ptr + i) % console->buffer.size] = data[i];
+ }
+
+ uintptr_t new_ptr = (ptr + size) % console->buffer.size;
+ console->buffer.wr_pos = new_ptr;
+ if (new_ptr < ptr + size && new_ptr > rd_ptr) {
+ console->buffer.rd_pos = new_ptr;
+ }
+ console->buffer.flags |= FIFO_DIRTY;
+ mutex_unlock(&console->buffer.lock);
+}
+
+void
+console_write_str(char* str)
+{
+ console_write(&lx_console, str, strlen(str));
+}
+
+void
+console_write_char(char str)
+{
+ console_write(&lx_console, &str, 1);
+}
+
+void
+console_start_flushing()
+{
+ struct lx_timer* timer =
+ timer_run_ms(20, __flush_cb, NULL, TIMER_MODE_PERIODIC);
+ lx_console.flush_timer = timer;
+}
\ No newline at end of file
git://scm.lunaixsky.com / lunaix-os.git / commitdiff