1 files changed, 496 insertions, 79 deletions

diff --git a/sys/dev/cons/cons.c b/sys/dev/cons/cons.c
index 4b85240..9921ff8 100644
--- a/sys/dev/cons/cons.c
+++ b/sys/dev/cons/cons.c
@@ -31,35 +31,60 @@
 #include <sys/types.h>
 #include <sys/ascii.h>
 #include <sys/device.h>
+#include <sys/errno.h>
+#include <sys/panic.h>
 #include <dev/video/fbdev.h>
 #include <dev/cons/font.h>
 #include <dev/cons/cons.h>
 #include <fs/devfs.h>
+#include <fs/ctlfs.h>
 #include <vm/dynalloc.h>
 #include <string.h>
 
+#define HIDE_CURSOR(SCR) \
+    cons_draw_cursor((SCR), (SCR)->bg)
+
+#define SHOW_CURSOR(SCR) \
+    cons_draw_cursor((SCR), rgb_invert((SCR)->bg))
+
+/* Console background from kconf */
+#if defined(__CONSOLE_BG)
+#define CONSOLE_BG __CONSOLE_BG
+#else
+#define CONSOLE_BG 0x000000
+#endif /* __CONSOLE_BG */
+
+/* Console foreground from kconf */
+#if defined(__CONSOLE_FG)
+#define CONSOLE_FG __CONSOLE_FG
+#else
+#define CONSOLE_FG 0x00AA00
+#endif  /* __CONSOLE_FG */
+
 struct cons_screen g_root_scr = {0};
 static struct cdevsw cons_cdevsw;
+static struct ctlops cons_feat_ctl;
 
-/*
- * Create a chracter descriptor for drawing
- * characters.
- *
- * @c: Character.
- * @fg: Foreground.
- * @bg: Background.
- */
-static inline struct cons_char
-cons_make_char(char c, uint32_t fg, uint32_t bg)
+static void cons_draw_cursor(struct cons_screen *scr, uint32_t color);
+static int cons_handle_special(struct cons_screen *scr, char c);
+
+static uint32_t
+rgb_invert(uint32_t rgb)
 {
-    struct cons_char ch;
+    uint8_t r, g, b;
+    uint32_t ret;
 
-    ch.fg = fg;
-    ch.bg = bg;
-    ch.c = c;
-    return ch;
+    r = (rgb >> 16) & 0xFF;
+    g = (rgb >> 8) & 0xFF;
+    b = rgb & 0xFF;
+
+    ret = (255 - r) << 16;
+    ret |= (255 - g) << 8;
+    ret |= 255 - b;
+    return ret;
 }
 
+
 /*
  * Render a character onto the screen.
  *
@@ -69,10 +94,10 @@ cons_make_char(char c, uint32_t fg, uint32_t bg)
  * @y: Y position of char.
  */
 static void
-cons_draw_char(struct cons_screen *scr, struct cons_char ch,
-                 uint32_t x, uint32_t y)
+cons_draw_char(struct cons_screen *scr, struct cons_char ch)
 {
     size_t idx;
+    uint32_t x, y;
     const uint8_t *glyph;
 
     if (scr->fb_mem == NULL) {
@@ -80,41 +105,56 @@ cons_draw_char(struct cons_screen *scr, struct cons_char ch,
     }
 
     glyph = &CONS_FONT[(int)ch.c*16];
+    x = ch.x;
+    y = ch.y;
+
     for (uint32_t cy = 0; cy < FONT_HEIGHT; ++cy) {
+        idx = fbdev_get_index(&scr->fbdev, x + (FONT_WIDTH - 1), y + cy);
         for (uint32_t cx = 0; cx < FONT_WIDTH; ++cx) {
-            idx = fbdev_get_index(&scr->fbdev, x + (FONT_WIDTH - 1) - cx, y + cy);
-            scr->fb_mem[idx] = ISSET(glyph[cy], BIT(cx)) ? ch.fg : ch.bg;
+            scr->fb_mem[idx--] = ISSET(glyph[cy], BIT(cx)) ? ch.fg : ch.bg;
         }
     }
 }
 
-static void
-cons_draw_cursor(struct cons_screen *scr, uint32_t color)
+/*
+ * Internal helper - flush console row
+ *
+ * @row: Row to flush.
+ */
+static int
+cons_flush_row(struct cons_screen *scr, uint32_t row)
 {
-    size_t idx;
+    struct cons_buf *bp;
+    struct cons_char cc;
 
-    for (uint32_t cy = 0; cy < FONT_HEIGHT; ++cy) {
-        for (uint32_t cx = 0; cx < FONT_WIDTH; ++cx) {
-            idx = fbdev_get_index(&scr->fbdev, (scr->curs_col * FONT_WIDTH) + cx, (scr->curs_row * FONT_HEIGHT) + cy);
-            scr->fb_mem[idx] = color;
+    bp = scr->ob[row];
+    if (ISSET(bp->flags, CONS_BUF_CLEAN)) {
+        return -EIO;
+    }
+
+    for (int j = 0; j < bp->len; ++j) {
+        if (cons_obuf_pop(bp, &cc) != 0) {
+            continue;
         }
+
+        cons_draw_char(scr, cc);
+        bp->flags |= CONS_BUF_CLEAN;
     }
+
+    return 0;
 }
 
 /*
- * Clear the screen.
+ * Internal helper - flush console
  *
- * @scr: Screen to clear.
- * @bg: Color to clear it to.
  */
-static void
-cons_clear_scr(struct cons_screen *scr, uint32_t bg)
+static int
+cons_flush(struct cons_screen *scr)
 {
-    struct fbdev fbdev = scr->fbdev;
-
-    for (size_t i = 0; i < fbdev.height * fbdev.pitch; ++i) {
-        scr->fb_mem[i] = bg;
+    for (int i = 0; i < scr->nrows; ++i) {
+        cons_flush_row(scr, i);
     }
+    return 0;
 }
 
 /*
@@ -128,88 +168,439 @@ cons_clear_scr(struct cons_screen *scr, uint32_t bg)
 static int
 cons_handle_special(struct cons_screen *scr, char c)
 {
+    struct cons_buf *bp;
+
+    if (scr->ch_col >= scr->ncols - 20) {
+        scr->ch_col = 0;
+        cons_handle_special(scr, '\n');
+    }
+
     switch (c) {
+    case ASCII_HT:
+        HIDE_CURSOR(scr);
+        scr->curs_col += 4;
+        scr->ch_col += 4;
+        if (scr->ch_col >= scr->ncols - 1) {
+           cons_handle_special(scr, '\n');
+        }
+        SHOW_CURSOR(scr);
+        return 0;
+    case ASCII_NUL:
+        return 0;
+    case ASCII_BS:
+        bp = scr->ob[scr->ch_row];
+        if (bp->head > bp->tail) {
+            --bp->head;
+        }
+
+        HIDE_CURSOR(scr);
+        if (scr->ch_col > 0 && scr->curs_col > 0) {
+            --scr->ch_col;
+            --scr->curs_col;
+        }
+        SHOW_CURSOR(scr);
+        return 0;
     case ASCII_LF:
+        /* Are we past screen width? */
+        if (scr->ch_row >= scr->nrows - 1) {
+            cons_clear_scr(scr, scr->bg);
+            return 0;
+        }
+
+        HIDE_CURSOR(scr);
+
         /* Make a newline */
-        scr->ch_col = 0;
+        cons_flush(scr);
         ++scr->ch_row;
+        scr->ch_col = 0;
+        cons_flush(scr);
 
-        cons_draw_cursor(scr, scr->bg);
+        /* Update cursor */
+        scr->curs_row += 1;
         scr->curs_col = 0;
-        scr->curs_row++;
-        cons_draw_cursor(scr, scr->last_chr.fg);
+        SHOW_CURSOR(scr);
+        return 0;
+    case ASCII_FF:
+        /*
+         * Fuck what they say, this is now the
+         * "flush" byte ::)
+         */
+        cons_flush(scr);
         return 0;
     }
 
     return -1;
 }
 
+static void
+cons_draw_cursor(struct cons_screen *scr, uint32_t color)
+{
+    struct console_feat *featp;
+    size_t idx;
+
+    featp = &scr->feat;
+    if (!featp->show_curs) {
+        color = scr->bg;
+    }
+
+    /* Past screen width? */
+    if (scr->curs_col >= scr->ncols) {
+        scr->curs_col = 0;
+        scr->curs_row++;
+    }
+
+    for (uint32_t cy = 0; cy < FONT_HEIGHT; ++cy) {
+        idx = fbdev_get_index(&scr->fbdev, scr->curs_col * FONT_WIDTH, (scr->curs_row * FONT_HEIGHT) + cy);
+        for (uint32_t cx = 0; cx < FONT_WIDTH; ++cx) {
+            scr->fb_mem[idx++] = color;
+        }
+    }
+}
+
 /*
- * Character device function.
+ * Clear the screen.
+ *
+ * @scr: Screen to clear.
+ * @bg: Color to clear it to.
  */
-static int
-dev_write(dev_t dev, struct sio_txn *sio, int flags)
+void
+cons_clear_scr(struct cons_screen *scr, uint32_t bg)
 {
-    char *p;
+    struct fbdev fbdev = scr->fbdev;
 
-    p = sio->buf;
-    spinlock_acquire(&g_root_scr.lock);
+    cons_flush(scr);
+    HIDE_CURSOR(scr);
 
-    for (size_t i = 0; i < sio->len; ++i) {
-        cons_putch(&g_root_scr, p[i]);
+    scr->ch_col = 0;
+    scr->ch_row = 0;
+    scr->curs_col = 0;
+    scr->curs_row = 0;
+
+    for (size_t i = 0; i < fbdev.height * fbdev.pitch; ++i) {
+        scr->fb_mem[i] = bg;
     }
 
-    spinlock_release(&g_root_scr.lock);
-    return sio->len;
+    SHOW_CURSOR(scr);
+
 }
 
 /*
- * Put a character on the screen.
+ * Quickly put a character on the screen.
+ * XXX: Does not acquire the screen's lock or show/hide the cursor.
  *
  * @scr: Screen.
  * @c: Character to draw.
  */
-int
-cons_putch(struct cons_screen *scr, char c)
+static void
+cons_fast_putch(struct cons_screen *scr, char c)
 {
-    struct cons_char cons_chr;
+    struct cons_char cc;
+    struct cons_buf *bp;
+    int ansi;
+
+    ansi = ansi_feed(&scr->ansi_s, c);
+    if (ansi > 0) {
+        c = ASCII_NUL;
+    } else if (ansi < 0) {
+        c = ASCII_NUL;
+    }
 
-    if (scr->ch_col > scr->ncols) {
-        /* Make a newline as we past the max col */
-        scr->ch_col = 0;
-        ++scr->ch_row;
+    /* Handle specials */
+    if (cons_handle_special(scr, c) == 0) {
+        return;
     }
 
-    if (scr->curs_row >= scr->nrows) {
-        /* Went over the screen size */
-        /* TODO: Scroll instead of just clearing the screen */
+    /* Create a new character */
+    cc.c = c;
+    cc.fg = scr->fg;
+    cc.bg = scr->bg;
+    cc.x = scr->ch_col * FONT_WIDTH;
+    cc.y = scr->ch_row * FONT_HEIGHT;
+
+    /* Push our new character */
+    bp = scr->ob[scr->ch_row];
+    bp->flags &= ~CONS_BUF_CLEAN;
+    cons_obuf_push(bp, cc);
+    ++scr->ch_col;
+
+    /* Check screen bounds */
+    if (cc.x >= (scr->ncols * FONT_WIDTH) - 1) {
         scr->ch_col = 0;
-        scr->ch_row = 0;
-        cons_clear_scr(scr, scr->bg);
+        ++scr->ch_row;
+    }
 
+    ++scr->curs_col;
+    if (scr->curs_col > scr->ncols - 1) {
         scr->curs_col = 0;
-        scr->curs_row = 0;
-        cons_draw_cursor(scr, scr->last_chr.fg);
+        if (scr->curs_row < scr->nrows)
+            ++scr->curs_row;
+    }
+}
+
+/*
+ * Character device function.
+ */
+static int
+dev_write(dev_t dev, struct sio_txn *sio, int flags)
+{
+    cons_attach();
+    cons_putstr(&g_root_scr, sio->buf, sio->len);
+    cons_flush(&g_root_scr);
+    return sio->len;
+}
+
+/*
+ * Character device function.
+ */
+static int
+dev_read(dev_t dev, struct sio_txn *sio, int flags)
+{
+    struct cons_input input;
+    uint8_t *p;
+    int retval;
+    size_t n;
+
+    p = (uint8_t *)sio->buf;
+    n = sio->len;
+
+    /* Must be a power of two */
+    if ((n & 1) != 0) {
+        return -EFAULT;
+    }
+
+    retval = cons_ibuf_pop(&g_root_scr, &input);
+    if (retval < 0) {
+        return -EAGAIN;
+    }
+
+    cons_attach();
+    spinlock_acquire(&g_root_scr.lock);
+    for (;;) {
+        /* Buffer too small */
+        if (n == 0) {
+            break;
+        }
+
+        *p++ = input.chr;
+        *p++ = input.scancode;
+        n -= 2;
+
+        /* Try to get the next byte */
+        retval = cons_ibuf_pop(&g_root_scr, &input);
+        if (retval < 0) {
+            break;
+        }
+    }
+    spinlock_release(&g_root_scr.lock);
+    return sio->len;
+}
+
+static int
+cons_init_bufs(struct cons_screen *scr)
+{
+    struct cons_buf *bp;
+    size_t ob_len;
+
+    scr->ib = cons_new_buf(CONS_BUF_INPUT, scr->ncols);
+    if (g_root_scr.ib == NULL) {
+        panic("out of memory\n");
+    }
+
+    ob_len = sizeof(*scr->ob) * scr->nrows;
+    scr->ob = dynalloc(ob_len);
+
+    /* Allocate all output buffers per line */
+    for (size_t i = 0; i < scr->nrows; ++i) {
+        bp = cons_new_buf(CONS_BUF_OUTPUT, scr->ncols);
+        if (bp == NULL) {
+            panic("out of memory\n");
+        }
+        bp->flags |= CONS_BUF_CLEAN;
+        scr->ob[i] = bp;
     }
 
+    return 0;
+}
+
+static int
+ctl_feat_read(struct ctlfs_dev *cdp, struct sio_txn *sio)
+{
+    struct cons_screen *scr = &g_root_scr;
+    struct console_feat *featp;
+
+    if (sio->buf == NULL || sio->len == 0) {
+        return -EINVAL;
+    }
+
+    featp = &scr->feat;
+    if (sio->len > sizeof(*featp)) {
+        sio->len = sizeof(*featp);
+    }
+
+    memcpy(sio->buf, featp, sio->len);
+    return sio->len;
+}
+
+static int
+ctl_feat_write(struct ctlfs_dev *cdp, struct sio_txn *sio)
+{
+    struct cons_screen *scr = &g_root_scr;
+    struct console_feat *featp, oldfeat;
+
+    featp = &scr->feat;
+    oldfeat = *featp;
+    if (sio->len > sizeof(*featp)) {
+        sio->len = sizeof(*featp);
+    }
+
+    memcpy(featp, sio->buf, sio->len);
+
     /*
-     * If this is a special char that we can handle
-     * then handle it and return.
+     * If we are suddenly trying to reset the cursor
+     * status, redraw it.
      */
-    if (cons_handle_special(scr, c) == 0) {
+    if (featp->show_curs != oldfeat.show_curs) {
+        if (featp->show_curs == 0) {
+            HIDE_CURSOR(scr);
+        } else {
+            SHOW_CURSOR(scr);
+        }
+    }
+    return sio->len;
+}
+
+/*
+ * Detach the currently running process from the
+ * console.
+ */
+int
+cons_detach(void)
+{
+    struct cons_screen *scr;
+
+    scr = &g_root_scr;
+    if (scr->atproc == NULL) {
+        return 0;
+    }
+    if (scr->atproc_lock == NULL) {
+        return 0;
+    }
+
+    scr = &g_root_scr;
+    scr->atproc = NULL;
+    mutex_release(scr->atproc_lock);
+    return 0;
+}
+
+/*
+ * Attach the current process to the
+ * console.
+ */
+int
+cons_attach(void)
+{
+    struct cons_screen *scr;
+    struct proc *td, *atproc;
+
+    td = this_td();
+    if (td == NULL) {
+        return -1;
+    }
+
+    scr = &g_root_scr;
+    if (scr->atproc_lock == NULL) {
         return 0;
     }
 
-    cons_chr = cons_make_char(c, scr->fg, scr->bg);
-    scr->last_chr = cons_chr;
+    scr = &g_root_scr;
+    atproc = scr->atproc;
 
-    /* Draw cursor and character */
-    scr->curs_col++;
-    cons_draw_cursor(scr, scr->last_chr.fg);
-    cons_draw_char(scr, cons_chr, scr->ch_col * FONT_WIDTH,
-        scr->ch_row * FONT_HEIGHT);
+    if (atproc != NULL) {
+        if (atproc->pid == td->pid) {
+            return 0;
+        }
 
-    ++scr->ch_col;
+        /*
+         * Do not release this here as we want
+         * any other process that tries to attach
+         * to wait.
+         */
+        mutex_acquire(scr->atproc_lock, 0);
+    }
+
+    scr->atproc = td;
+    return 0;
+}
+
+/*
+ * Reset console color.
+ */
+void
+cons_reset_color(struct cons_screen *scr)
+{
+    g_root_scr.fg = CONSOLE_FG;
+    g_root_scr.bg = CONSOLE_BG;
+}
+
+void
+cons_update_color(struct cons_screen *scr, uint32_t fg, uint32_t bg)
+{
+    scr->fg = fg;
+    scr->bg = bg;
+}
+
+void
+cons_reset_cursor(struct cons_screen *scr)
+{
+    HIDE_CURSOR(scr);
+    scr->ch_col = 0;
+    scr->ch_row = 0;
+    scr->curs_col = 0;
+    scr->curs_row = 0;
+    SHOW_CURSOR(scr);
+}
+
+/*
+ * Put a character on the screen.
+ *
+ * @scr: Screen.
+ * @c: Character to draw.
+ */
+int
+cons_putch(struct cons_screen *scr, char c)
+{
+    spinlock_acquire(&scr->lock);
+    HIDE_CURSOR(scr);
+
+    cons_fast_putch(scr, c);
+
+    SHOW_CURSOR(scr);
+    spinlock_release(&scr->lock);
+    return 0;
+}
+
+/*
+ * Put a string on the screen.
+ *
+ * @scr: Screen.
+ * @s: String to draw.
+ * @l: Length of s.
+ */
+int
+cons_putstr(struct cons_screen *scr, const char *s, size_t len)
+{
+    const char *p = s;
+
+    spinlock_acquire(&scr->lock);
+    HIDE_CURSOR(scr);
+
+    while (len--) {
+        cons_fast_putch(scr, *p);
+        ++p;
+    }
+
+    SHOW_CURSOR(scr);
+    spinlock_release(&scr->lock);
     return 0;
 }
 
@@ -217,14 +608,24 @@ void
 cons_init(void)
 {
     struct fbdev fbdev = fbdev_get();
-
-    g_root_scr.fg = 0x00AA00;
-    g_root_scr.bg = 0x000000;
+    struct console_feat *featp;
+
+    featp = &g_root_scr.feat;
+    featp->ansi_esc = 1;
+    featp->show_curs = 1;
+    g_root_scr.ch_col = 0;
+    g_root_scr.ch_row = 0;
+    g_root_scr.fg = CONSOLE_FG;
+    g_root_scr.bg = CONSOLE_BG;
     g_root_scr.fb_mem = fbdev.mem;
     g_root_scr.nrows = fbdev.height / FONT_HEIGHT;
     g_root_scr.ncols = fbdev.width / FONT_WIDTH;
     g_root_scr.fbdev = fbdev;
+    g_root_scr.atproc = NULL;
+    g_root_scr.atproc_lock = NULL;
     memset(&g_root_scr.lock, 0, sizeof(g_root_scr.lock));
+    cons_init_bufs(&g_root_scr);
+    SHOW_CURSOR(&g_root_scr);
 }
 
 /*
@@ -234,6 +635,7 @@ void
 cons_expose(void)
 {
     static int once = 0;
+    struct ctlfs_dev ctl;
     char devname[] = "console";
     devmajor_t major;
     dev_t dev;
@@ -243,15 +645,30 @@ cons_expose(void)
         return;
     }
 
+    /* Init the attached proc mutex lock */
+    g_root_scr.atproc_lock = mutex_new("console0");
+
     /* Register the device here */
     major = dev_alloc_major();
     dev = dev_alloc(major);
     dev_register(major, dev, &cons_cdevsw);
     devfs_create_entry(devname, major, dev, 0444);
+
+    /* Register the control file */
+    ctl.mode = 0666;
+    ctlfs_create_node(devname, &ctl);
+    ctl.devname = devname;
+    ctl.ops = &cons_feat_ctl;
+    ctlfs_create_entry("feat", &ctl);
     once ^= 1;
 }
 
 static struct cdevsw cons_cdevsw = {
-    .read = noread,
+    .read = dev_read,
     .write = dev_write
 };
+
+static struct ctlops cons_feat_ctl = {
+    .read = ctl_feat_read,
+    .write = ctl_feat_write
+};