// SPDX-License-Identifier: GPL-2.0
#include <linux/init.h>
#include <linux/errno.h>
#include <linux/console.h>

#include <asm/sibyte/board.h>

#include <asm/fw/cfe/cfe_api.h>
#include <asm/fw/cfe/cfe_error.h>

extern int cfe_cons_handle;

static void cfe_console_write(struct console *cons, const char *str,
                       unsigned int count)
{
        int i, last, written;

        for (i=0, last=0; i<count; i++) {
                if (!str[i])
                        /* XXXKW can/should this ever happen? */
                        return;
                if (str[i] == '\n') {
                        do {
                                written = cfe_write(cfe_cons_handle, &str[last], i-last);
                                if (written < 0)
                                        ;
                                last += written;
                        } while (last < i);
                        while (cfe_write(cfe_cons_handle, "\r", 1) <= 0)
                                ;
                }
        }
        if (last != count) {
                do {
                        written = cfe_write(cfe_cons_handle, &str[last], count-last);
                        if (written < 0)
                                ;
                        last += written;
                } while (last < count);
        }

}

static int cfe_console_setup(struct console *cons, char *str)
{
        char consdev[32];
        /* XXXKW think about interaction with 'console=' cmdline arg */
        /* If none of the console options are configured, the build will break. */
        if (cfe_getenv("BOOT_CONSOLE", consdev, 32) >= 0) {
#ifdef CONFIG_SERIAL_SB1250_DUART
                if (!strcmp(consdev, "uart0")) {
                        setleds("u0cn");
                } else if (!strcmp(consdev, "uart1")) {
                        setleds("u1cn");
                } else
#endif
#ifdef CONFIG_VGA_CONSOLE
                       if (!strcmp(consdev, "pcconsole0")) {
                                setleds("pccn");
                } else
#endif
                        return -ENODEV;
        }
        return 0;
}

static struct console sb1250_cfe_cons = {
        .name           = "cfe",
        .write          = cfe_console_write,
        .setup          = cfe_console_setup,
        .flags          = CON_PRINTBUFFER,
        .index          = -1,
};

static int __init sb1250_cfe_console_init(void)
{
        register_console(&sb1250_cfe_cons);
        return 0;
}

console_initcall(sb1250_cfe_console_init);