/* uart_console.c - UART-driven console */

/*
 * Copyright (c) 2011-2012, 2014-2015 Wind River Systems, Inc.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 * 1) Redistributions of source code must retain the above copyright notice,
 * this list of conditions and the following disclaimer.
 *
 * 2) Redistributions in binary form must reproduce the above copyright notice,
 * this list of conditions and the following disclaimer in the documentation
 * and/or other materials provided with the distribution.
 *
 * 3) Neither the name of Wind River Systems nor the names of its contributors
 * may be used to endorse or promote products derived from this software without
 * specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 */

/*
  DESCRIPTION

  Serial console driver.
  Hooks into the printk and fputc (for printf) modules. Poll driven.
 */

#include <nanokernel.h>
#include <arch/cpu.h>

#include <stdio.h>
#include <stdint.h>
#include <errno.h>

#include <device.h>
#include <init.h>

#include <board.h>
#include <drivers/uart.h>
#include <console/uart_console.h>
#include <toolchain.h>
#include <sections.h>

#if 0 /* NOTUSED */
/**
 *
 * @brief Get a character from UART
 *
 * @return the character or EOF if nothing present
 */

static int consoleIn(void)
{
#ifdef UART_CONSOLE_DEV
	unsigned char c;
	if (uart_poll_in(UART_CONSOLE_DEV, &c) < 0)
		return EOF;
	else
		return (int)c;
#else
	return 0;
#endif
}
#endif

#if defined(CONFIG_PRINTK) || defined(CONFIG_STDOUT_CONSOLE)
/**
 *
 * @brief Output one character to UART
 *
 * Outputs both line feed and carriage return in the case of a '\n'.
 *
 * @return The character passed as input.
 */

static int consoleOut(int c /* character to output */
	)
{
#ifdef UART_CONSOLE_DEV
	uart_poll_out(UART_CONSOLE_DEV, (unsigned char)c);
	if ('\n' == c) {
		uart_poll_out(UART_CONSOLE_DEV, (unsigned char)'\r');
	}
	return c;
#else
	return 0;
#endif
}
#endif

#if defined(CONFIG_STDOUT_CONSOLE)
extern void __stdout_hook_install(int (*hook)(int));
#else
#define __stdout_hook_install(x)		\
	do {/* nothing */			\
	} while ((0))
#endif

#if defined(CONFIG_PRINTK)
extern void __printk_hook_install(int (*fn)(int));
#else
#define __printk_hook_install(x)		\
	do {/* nothing */			\
	} while ((0))
#endif

#if defined(CONFIG_CONSOLE_HANDLER)
static size_t pos = 0;

static struct nano_fifo *avail_queue;
static struct nano_fifo *lines_queue;

static int read_uart(struct device *uart, uint8_t *buf, unsigned int size)
{
	int rx;

	rx = uart_fifo_read(uart, buf, size);
	if (rx < 0) {
		/* Overrun issue. Stop the UART */
		uart_irq_rx_disable(uart);

		return -EIO;
	}

	return rx;
}

void uart_console_isr(void *unused)
{
	ARG_UNUSED(unused);

	while (uart_irq_update(UART_CONSOLE_DEV)
	       && uart_irq_is_pending(UART_CONSOLE_DEV)) {
		/* Character(s) have been received */
		if (uart_irq_rx_ready(UART_CONSOLE_DEV)) {
			static struct uart_console_input *cmd;
			uint8_t byte;
			int rx;


			if (!cmd) {
				cmd = nano_isr_fifo_get(avail_queue);
				if (!cmd)
					return;
			}

			rx = read_uart(UART_CONSOLE_DEV, &byte, 1);
			if (rx < 0) {
				nano_isr_fifo_put(avail_queue, cmd);
				return;
			}

			/* Echo back to console */
			uart_poll_out(UART_CONSOLE_DEV, byte);

			if (byte == '\r' || byte == '\n' ||
			    pos == sizeof(cmd->line) - 1) {
				cmd->line[pos] = '\0';
				uart_poll_out(UART_CONSOLE_DEV, '\n');
				pos = 0;

				nano_isr_fifo_put(lines_queue, cmd);
				cmd = NULL;
			} else {
				cmd->line[pos++] = byte;
			}

		}
	}
}

IRQ_CONNECT_STATIC(console, CONFIG_UART_CONSOLE_IRQ,
		   CONFIG_UART_CONSOLE_INT_PRI, uart_console_isr, 0);

static void console_input_init(void)
{
	uint8_t c;

	uart_irq_rx_disable(UART_CONSOLE_DEV);
	uart_irq_tx_disable(UART_CONSOLE_DEV);
	IRQ_CONFIG(console, uart_irq_get(UART_CONSOLE_DEV));
	irq_enable(uart_irq_get(UART_CONSOLE_DEV));

	/* Drain the fifo */
	while (uart_irq_rx_ready(UART_CONSOLE_DEV)) {
		uart_fifo_read(UART_CONSOLE_DEV, &c, 1);
	}

	uart_irq_rx_enable(UART_CONSOLE_DEV);
}

void uart_register_input(struct nano_fifo *avail, struct nano_fifo *lines)
{
	avail_queue = avail;
	lines_queue = lines;

	console_input_init();
}
#else
#define console_input_init(x)			\
	do {/* nothing */			\
	} while ((0))
#define uart_register_input(x)			\
	do {/* nothing */			\
	} while ((0))
#endif

/**
 *
 * @brief Install printk/stdout hook for UART console output
 *
 * @return N/A
 */

void uart_console_hook_install(void)
{
	__stdout_hook_install(consoleOut);
	__printk_hook_install(consoleOut);
}

/**
 *
 * @brief Initialize one UART as the console/debug port
 *
 * @return DEV_OK if successful, otherwise failed.
 */
static int uart_console_init(struct device *arg)
{
	ARG_UNUSED(arg);

	uart_console_hook_install();

	return DEV_OK;
}
DECLARE_DEVICE_INIT_CONFIG(uart_console, "", uart_console_init, NULL);
pure_init(uart_console, NULL);