]> git.kernelconcepts.de Git - karo-tx-uboot.git/blobdiff - drivers/serial/sandbox.c
sandbox: serial: Support a coloured console
[karo-tx-uboot.git] / drivers / serial / sandbox.c
index b73520ca940740351e69aea3e4f3ece04f94336f..cd2f91e28e8f96b0d651cac208cbbe1c4fb12303 100644 (file)
@@ -1,22 +1,7 @@
 /*
  * Copyright (c) 2011 The Chromium OS Authors.
- * See file CREDITS for list of people who contributed to this
- * project.
  *
- * This program is free software; you can redistribute it and/or
- * modify it under the terms of the GNU General Public License as
- * published by the Free Software Foundation; either version 2 of
- * the License, or (at your option) any later version.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- * GNU General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place, Suite 330, Boston,
- * MA 02111-1307 USA
+ * SPDX-License-Identifier:    GPL-2.0+
  */
 
 /*
  */
 
 #include <common.h>
+#include <dm.h>
+#include <fdtdec.h>
+#include <lcd.h>
 #include <os.h>
 #include <serial.h>
 #include <linux/compiler.h>
+#include <asm/state.h>
+
+DECLARE_GLOBAL_DATA_PTR;
 
 /*
  *
@@ -43,24 +34,69 @@ static char serial_buf[16];
 static unsigned int serial_buf_write;
 static unsigned int serial_buf_read;
 
-static int sandbox_serial_init(void)
+struct sandbox_serial_platdata {
+       int colour;     /* Text colour to use for output, -1 for none */
+};
+
+struct sandbox_serial_priv {
+       bool start_of_line;
+};
+
+/**
+ * output_ansi_colour() - Output an ANSI colour code
+ *
+ * @colour: Colour to output (0-7)
+ */
+static void output_ansi_colour(int colour)
 {
-       os_tty_raw(0);
-       return 0;
+       char ansi_code[] = "\x1b[1;3Xm";
+
+       ansi_code[5] = '0' + colour;
+       os_write(1, ansi_code, sizeof(ansi_code) - 1);
 }
 
-static void sandbox_serial_setbrg(void)
+static void output_ansi_reset(void)
 {
+       os_write(1, "\x1b[0m", 4);
 }
 
-static void sandbox_serial_putc(const char ch)
+static int sandbox_serial_probe(struct udevice *dev)
 {
-       os_write(1, &ch, 1);
+       struct sandbox_state *state = state_get_current();
+       struct sandbox_serial_priv *priv = dev_get_priv(dev);
+
+       if (state->term_raw != STATE_TERM_COOKED)
+               os_tty_raw(0, state->term_raw == STATE_TERM_RAW_WITH_SIGS);
+       priv->start_of_line = 0;
+
+       return 0;
 }
 
-static void sandbox_serial_puts(const char *str)
+static int sandbox_serial_remove(struct udevice *dev)
 {
-       os_write(1, str, strlen(str));
+       struct sandbox_serial_platdata *plat = dev->platdata;
+
+       if (plat->colour != -1)
+               output_ansi_reset();
+
+       return 0;
+}
+
+static int sandbox_serial_putc(struct udevice *dev, const char ch)
+{
+       struct sandbox_serial_priv *priv = dev_get_priv(dev);
+       struct sandbox_serial_platdata *plat = dev->platdata;
+
+       if (priv->start_of_line && plat->colour != -1) {
+               priv->start_of_line = false;
+               output_ansi_colour(plat->colour);
+       }
+
+       os_write(1, &ch, 1);
+       if (ch == '\n')
+               priv->start_of_line = true;
+
+       return 0;
 }
 
 static unsigned int increment_buffer_index(unsigned int index)
@@ -68,51 +104,96 @@ static unsigned int increment_buffer_index(unsigned int index)
        return (index + 1) % ARRAY_SIZE(serial_buf);
 }
 
-static int sandbox_serial_tstc(void)
+static int sandbox_serial_pending(struct udevice *dev, bool input)
 {
        const unsigned int next_index =
                increment_buffer_index(serial_buf_write);
        ssize_t count;
 
+       if (!input)
+               return 0;
+
        os_usleep(100);
+#ifdef CONFIG_LCD
+       lcd_sync();
+#endif
        if (next_index == serial_buf_read)
                return 1;       /* buffer full */
 
        count = os_read_no_block(0, &serial_buf[serial_buf_write], 1);
        if (count == 1)
                serial_buf_write = next_index;
+
        return serial_buf_write != serial_buf_read;
 }
 
-static int sandbox_serial_getc(void)
+static int sandbox_serial_getc(struct udevice *dev)
 {
        int result;
 
-       while (!sandbox_serial_tstc())
-               ;       /* buffer empty */
+       if (!sandbox_serial_pending(dev, true))
+               return -EAGAIN; /* buffer empty */
 
        result = serial_buf[serial_buf_read];
        serial_buf_read = increment_buffer_index(serial_buf_read);
        return result;
 }
 
-static struct serial_device sandbox_serial_drv = {
-       .name   = "sandbox_serial",
-       .start  = sandbox_serial_init,
-       .stop   = NULL,
-       .setbrg = sandbox_serial_setbrg,
-       .putc   = sandbox_serial_putc,
-       .puts   = sandbox_serial_puts,
-       .getc   = sandbox_serial_getc,
-       .tstc   = sandbox_serial_tstc,
+static const char * const ansi_colour[] = {
+       "black", "red", "green", "yellow", "blue", "megenta", "cyan",
+       "white",
 };
 
-void sandbox_serial_initialize(void)
+static int sandbox_serial_ofdata_to_platdata(struct udevice *dev)
 {
-       serial_register(&sandbox_serial_drv);
-}
+       struct sandbox_serial_platdata *plat = dev->platdata;
+       const char *colour;
+       int i;
+
+       plat->colour = -1;
+       colour = fdt_getprop(gd->fdt_blob, dev->of_offset,
+                            "sandbox,text-colour", NULL);
+       if (colour) {
+               for (i = 0; i < ARRAY_SIZE(ansi_colour); i++) {
+                       if (!strcmp(colour, ansi_colour[i])) {
+                               plat->colour = i;
+                               break;
+                       }
+               }
+       }
 
-__weak struct serial_device *default_serial_console(void)
-{
-       return &sandbox_serial_drv;
+       return 0;
 }
+
+static const struct dm_serial_ops sandbox_serial_ops = {
+       .putc = sandbox_serial_putc,
+       .pending = sandbox_serial_pending,
+       .getc = sandbox_serial_getc,
+};
+
+static const struct udevice_id sandbox_serial_ids[] = {
+       { .compatible = "sandbox,serial" },
+       { }
+};
+
+U_BOOT_DRIVER(serial_sandbox) = {
+       .name   = "serial_sandbox",
+       .id     = UCLASS_SERIAL,
+       .of_match = sandbox_serial_ids,
+       .ofdata_to_platdata = sandbox_serial_ofdata_to_platdata,
+       .platdata_auto_alloc_size = sizeof(struct sandbox_serial_platdata),
+       .priv_auto_alloc_size = sizeof(struct sandbox_serial_priv),
+       .probe = sandbox_serial_probe,
+       .remove = sandbox_serial_remove,
+       .ops    = &sandbox_serial_ops,
+       .flags = DM_FLAG_PRE_RELOC,
+};
+
+static const struct sandbox_serial_platdata platdata_non_fdt = {
+       .colour = -1,
+};
+
+U_BOOT_DEVICE(serial_sandbox_non_fdt) = {
+       .name = "serial_sandbox",
+       .platdata = &platdata_non_fdt,
+};