[people/ms/u-boot.git] / drivers / serial / serial-uclass.c

/*
 * Copyright (c) 2014 The Chromium OS Authors.
 *
 * SPDX-License-Identifier:	GPL-2.0+
 */

#include <common.h>
#include <dm.h>
#include <environment.h>
#include <errno.h>
#include <os.h>
#include <serial.h>
#include <stdio_dev.h>
#include <watchdog.h>
#include <dm/lists.h>
#include <dm/device-internal.h>
#include <dm/of_access.h>

DECLARE_GLOBAL_DATA_PTR;

/*
 * Table with supported baudrates (defined in config_xyz.h)
 */
static const unsigned long baudrate_table[] = CONFIG_SYS_BAUDRATE_TABLE;

#if !CONFIG_VAL(SYS_MALLOC_F_LEN)
#error "Serial is required before relocation - define CONFIG_$(SPL_)SYS_MALLOC_F_LEN to make this work"
#endif

static int serial_check_stdout(const void *blob, struct udevice **devp)
{
	int node;

	/* Check for a chosen console */
	node = fdtdec_get_chosen_node(blob, "stdout-path");
	if (node < 0) {
		const char *str, *p, *name;

		/*
		 * Deal with things like
		 *	stdout-path = "serial0:115200n8";
		 *
		 * We need to look up the alias and then follow it to the
		 * correct node.
		 */
		str = fdtdec_get_chosen_prop(blob, "stdout-path");
		if (str) {
			p = strchr(str, ':');
			name = fdt_get_alias_namelen(blob, str,
					p ? p - str : strlen(str));
			if (name)
				node = fdt_path_offset(blob, name);
		}
	}
	if (node < 0)
		node = fdt_path_offset(blob, "console");
	if (!uclass_get_device_by_of_offset(UCLASS_SERIAL, node, devp))
		return 0;

	/*
	 * If the console is not marked to be bound before relocation, bind it
	 * anyway.
	 */
	if (node > 0 && !lists_bind_fdt(gd->dm_root, offset_to_ofnode(node),
					devp)) {
		if (!device_probe(*devp))
			return 0;
	}

	return -ENODEV;
}

static void serial_find_console_or_panic(void)
{
	const void *blob = gd->fdt_blob;
	struct udevice *dev;
#ifdef CONFIG_SERIAL_SEARCH_ALL
	int ret;
#endif

	if (CONFIG_IS_ENABLED(OF_PLATDATA)) {
		uclass_first_device(UCLASS_SERIAL, &dev);
		if (dev) {
			gd->cur_serial_dev = dev;
			return;
		}
	} else if (CONFIG_IS_ENABLED(OF_CONTROL) && blob) {
		/* Live tree has support for stdout */
		if (of_live_active()) {
			struct device_node *np = of_get_stdout();

			if (np && !uclass_get_device_by_ofnode(UCLASS_SERIAL,
					np_to_ofnode(np), &dev)) {
				gd->cur_serial_dev = dev;
				return;
			}
		} else {
			if (!serial_check_stdout(blob, &dev)) {
				gd->cur_serial_dev = dev;
				return;
			}
		}
	}
	if (!SPL_BUILD || !CONFIG_IS_ENABLED(OF_CONTROL) || !blob) {
		/*
		 * Try to use CONFIG_CONS_INDEX if available (it is numbered
		 * from 1!).
		 *
		 * Failing that, get the device with sequence number 0, or in
		 * extremis just the first working serial device we can find.
		 * But we insist on having a console (even if it is silent).
		 */
#ifdef CONFIG_CONS_INDEX
#define INDEX (CONFIG_CONS_INDEX - 1)
#else
#define INDEX 0
#endif

#ifdef CONFIG_SERIAL_SEARCH_ALL
		if (!uclass_get_device_by_seq(UCLASS_SERIAL, INDEX, &dev) ||
		    !uclass_get_device(UCLASS_SERIAL, INDEX, &dev)) {
			if (dev->flags & DM_FLAG_ACTIVATED) {
				gd->cur_serial_dev = dev;
				return;
			}
		}

		/* Search for any working device */
		for (ret = uclass_first_device_check(UCLASS_SERIAL, &dev);
		     dev;
		     ret = uclass_next_device_check(&dev)) {
			if (!ret) {
				/* Device did succeed probing */
				gd->cur_serial_dev = dev;
				return;
			}
		}
#else
		if (!uclass_get_device_by_seq(UCLASS_SERIAL, INDEX, &dev) ||
		    !uclass_get_device(UCLASS_SERIAL, INDEX, &dev) ||
		    (!uclass_first_device(UCLASS_SERIAL, &dev) && dev)) {
			gd->cur_serial_dev = dev;
			return;
		}
#endif

#undef INDEX
	}

#ifdef CONFIG_REQUIRE_SERIAL_CONSOLE
	panic_str("No serial driver found");
#endif
}

/* Called prior to relocation */
int serial_init(void)
{
	serial_find_console_or_panic();
	gd->flags |= GD_FLG_SERIAL_READY;

	return 0;
}

/* Called after relocation */
void serial_initialize(void)
{
	serial_init();
}

static void _serial_putc(struct udevice *dev, char ch)
{
	struct dm_serial_ops *ops = serial_get_ops(dev);
	int err;

	if (ch == '\n')
		_serial_putc(dev, '\r');

	do {
		err = ops->putc(dev, ch);
	} while (err == -EAGAIN);
}

static void _serial_puts(struct udevice *dev, const char *str)
{
	while (*str)
		_serial_putc(dev, *str++);
}

static int __serial_getc(struct udevice *dev)
{
	struct dm_serial_ops *ops = serial_get_ops(dev);
	int err;

	do {
		err = ops->getc(dev);
		if (err == -EAGAIN)
			WATCHDOG_RESET();
	} while (err == -EAGAIN);

	return err >= 0 ? err : 0;
}

static int __serial_tstc(struct udevice *dev)
{
	struct dm_serial_ops *ops = serial_get_ops(dev);

	if (ops->pending)
		return ops->pending(dev, true);

	return 1;
}

#if CONFIG_IS_ENABLED(SERIAL_RX_BUFFER)
static int _serial_tstc(struct udevice *dev)
{
	struct serial_dev_priv *upriv = dev_get_uclass_priv(dev);

	/* Read all available chars into the RX buffer */
	while (__serial_tstc(dev)) {
		upriv->buf[upriv->wr_ptr++] = __serial_getc(dev);
		upriv->wr_ptr %= CONFIG_SERIAL_RX_BUFFER_SIZE;
	}

	return upriv->rd_ptr != upriv->wr_ptr ? 1 : 0;
}

static int _serial_getc(struct udevice *dev)
{
	struct serial_dev_priv *upriv = dev_get_uclass_priv(dev);
	char val;

	val = upriv->buf[upriv->rd_ptr++];
	upriv->rd_ptr %= CONFIG_SERIAL_RX_BUFFER_SIZE;

	return val;
}

#else /* CONFIG_IS_ENABLED(SERIAL_RX_BUFFER) */

static int _serial_getc(struct udevice *dev)
{
	return __serial_getc(dev);
}

static int _serial_tstc(struct udevice *dev)
{
	return __serial_tstc(dev);
}
#endif /* CONFIG_IS_ENABLED(SERIAL_RX_BUFFER) */

void serial_putc(char ch)
{
	if (gd->cur_serial_dev)
		_serial_putc(gd->cur_serial_dev, ch);
}

void serial_puts(const char *str)
{
	if (gd->cur_serial_dev)
		_serial_puts(gd->cur_serial_dev, str);
}

int serial_getc(void)
{
	if (!gd->cur_serial_dev)
		return 0;

	return _serial_getc(gd->cur_serial_dev);
}

int serial_tstc(void)
{
	if (!gd->cur_serial_dev)
		return 0;

	return _serial_tstc(gd->cur_serial_dev);
}

void serial_setbrg(void)
{
	struct dm_serial_ops *ops;

	if (!gd->cur_serial_dev)
		return;

	ops = serial_get_ops(gd->cur_serial_dev);
	if (ops->setbrg)
		ops->setbrg(gd->cur_serial_dev, gd->baudrate);
}

void serial_stdio_init(void)
{
}

#if defined(CONFIG_DM_STDIO)

#if CONFIG_IS_ENABLED(SERIAL_PRESENT)
static void serial_stub_putc(struct stdio_dev *sdev, const char ch)
{
	_serial_putc(sdev->priv, ch);
}
#endif

static void serial_stub_puts(struct stdio_dev *sdev, const char *str)
{
	_serial_puts(sdev->priv, str);
}

static int serial_stub_getc(struct stdio_dev *sdev)
{
	return _serial_getc(sdev->priv);
}

static int serial_stub_tstc(struct stdio_dev *sdev)
{
	return _serial_tstc(sdev->priv);
}
#endif

/**
 * on_baudrate() - Update the actual baudrate when the env var changes
 *
 * This will check for a valid baudrate and only apply it if valid.
 */
static int on_baudrate(const char *name, const char *value, enum env_op op,
	int flags)
{
	int i;
	int baudrate;

	switch (op) {
	case env_op_create:
	case env_op_overwrite:
		/*
		 * Switch to new baudrate if new baudrate is supported
		 */
		baudrate = simple_strtoul(value, NULL, 10);

		/* Not actually changing */
		if (gd->baudrate == baudrate)
			return 0;

		for (i = 0; i < ARRAY_SIZE(baudrate_table); ++i) {
			if (baudrate == baudrate_table[i])
				break;
		}
		if (i == ARRAY_SIZE(baudrate_table)) {
			if ((flags & H_FORCE) == 0)
				printf("## Baudrate %d bps not supported\n",
				       baudrate);
			return 1;
		}
		if ((flags & H_INTERACTIVE) != 0) {
			printf("## Switch baudrate to %d bps and press ENTER ...\n",
			       baudrate);
			udelay(50000);
		}

		gd->baudrate = baudrate;

		serial_setbrg();

		udelay(50000);

		if ((flags & H_INTERACTIVE) != 0)
			while (1) {
				if (getc() == '\r')
					break;
			}

		return 0;
	case env_op_delete:
		printf("## Baudrate may not be deleted\n");
		return 1;
	default:
		return 0;
	}
}
U_BOOT_ENV_CALLBACK(baudrate, on_baudrate);

#if CONFIG_IS_ENABLED(SERIAL_PRESENT)
static int serial_post_probe(struct udevice *dev)
{
	struct dm_serial_ops *ops = serial_get_ops(dev);
#ifdef CONFIG_DM_STDIO
	struct serial_dev_priv *upriv = dev_get_uclass_priv(dev);
	struct stdio_dev sdev;
#endif
	int ret;

#if defined(CONFIG_NEEDS_MANUAL_RELOC)
	if (ops->setbrg)
		ops->setbrg += gd->reloc_off;
	if (ops->getc)
		ops->getc += gd->reloc_off;
	if (ops->putc)
		ops->putc += gd->reloc_off;
	if (ops->pending)
		ops->pending += gd->reloc_off;
	if (ops->clear)
		ops->clear += gd->reloc_off;
#if CONFIG_POST & CONFIG_SYS_POST_UART
	if (ops->loop)
		ops->loop += gd->reloc_off
#endif
#endif
	/* Set the baud rate */
	if (ops->setbrg) {
		ret = ops->setbrg(dev, gd->baudrate);
		if (ret)
			return ret;
	}

#ifdef CONFIG_DM_STDIO
	if (!(gd->flags & GD_FLG_RELOC))
		return 0;
	memset(&sdev, '\0', sizeof(sdev));

	strncpy(sdev.name, dev->name, sizeof(sdev.name));
	sdev.flags = DEV_FLAGS_OUTPUT | DEV_FLAGS_INPUT | DEV_FLAGS_DM;
	sdev.priv = dev;
	sdev.putc = serial_stub_putc;
	sdev.puts = serial_stub_puts;
	sdev.getc = serial_stub_getc;
	sdev.tstc = serial_stub_tstc;

#if CONFIG_IS_ENABLED(SERIAL_RX_BUFFER)
	/* Allocate the RX buffer */
	upriv->buf = malloc(CONFIG_SERIAL_RX_BUFFER_SIZE);
#endif

	stdio_register_dev(&sdev, &upriv->sdev);
#endif
	return 0;
}

static int serial_pre_remove(struct udevice *dev)
{
#if CONFIG_IS_ENABLED(SYS_STDIO_DEREGISTER)
	struct serial_dev_priv *upriv = dev_get_uclass_priv(dev);

	if (stdio_deregister_dev(upriv->sdev, true))
		return -EPERM;
#endif

	return 0;
}

UCLASS_DRIVER(serial) = {
	.id		= UCLASS_SERIAL,
	.name		= "serial",
	.flags		= DM_UC_FLAG_SEQ_ALIAS,
	.post_probe	= serial_post_probe,
	.pre_remove	= serial_pre_remove,
	.per_device_auto_alloc_size = sizeof(struct serial_dev_priv),
};
#endif
Commit	Line	Data
57d92753 SG	1	/*
	2	* Copyright (c) 2014 The Chromium OS Authors.
	3	*
	4	* SPDX-License-Identifier: GPL-2.0+
	5	*/
	6
	7	#include <common.h>
	8	#include <dm.h>
ad1b81c8	9	#include <environment.h>
57d92753	10	#include <errno.h>
57d92753 SG	11	#include <os.h>
	12	#include <serial.h>
	13	#include <stdio_dev.h>
c487fd47	14	#include <watchdog.h>
57d92753 SG	15	#include <dm/lists.h>
57d92753 SG	16	#include <dm/device-internal.h>
f93472a0	17	#include <dm/of_access.h>
57d92753 SG	18
	19	DECLARE_GLOBAL_DATA_PTR;
	20
ad1b81c8 SG	21	/*
	22	* Table with supported baudrates (defined in config_xyz.h)
	23	*/
	24	static const unsigned long baudrate_table[] = CONFIG_SYS_BAUDRATE_TABLE;
	25
f1896c45 AY	26	#if !CONFIG_VAL(SYS_MALLOC_F_LEN)
f1896c45 AY	27	#error "Serial is required before relocation - define CONFIG_$(SPL_)SYS_MALLOC_F_LEN to make this work"
57d92753 SG	28	#endif
57d92753 SG	29
d0960853 SG	30	static int serial_check_stdout(const void blob, struct udevice *devp)
	31	{
	32	int node;
	33
	34	/* Check for a chosen console */
	35	node = fdtdec_get_chosen_node(blob, "stdout-path");
	36	if (node < 0) {
	37	const char str, p, *name;
	38
	39	/*
	40	* Deal with things like
	41	* stdout-path = "serial0:115200n8";
	42	*
	43	* We need to look up the alias and then follow it to the
	44	* correct node.
	45	*/
	46	str = fdtdec_get_chosen_prop(blob, "stdout-path");
	47	if (str) {
	48	p = strchr(str, ':');
	49	name = fdt_get_alias_namelen(blob, str,
	50	p ? p - str : strlen(str));
	51	if (name)
	52	node = fdt_path_offset(blob, name);
	53	}
	54	}
	55	if (node < 0)
	56	node = fdt_path_offset(blob, "console");
	57	if (!uclass_get_device_by_of_offset(UCLASS_SERIAL, node, devp))
	58	return 0;
	59
	60	/*
	61	* If the console is not marked to be bound before relocation, bind it
	62	* anyway.
	63	*/
	64	if (node > 0 && !lists_bind_fdt(gd->dm_root, offset_to_ofnode(node),
	65	devp)) {
	66	if (!device_probe(*devp))
	67	return 0;
	68	}
	69
	70	return -ENODEV;
	71	}
	72
57d92753 SG	73	static void serial_find_console_or_panic(void)
57d92753 SG	74	{
6bdc593e	75	const void *blob = gd->fdt_blob;
469a579d	76	struct udevice *dev;
ae5326a6	77	#ifdef CONFIG_SERIAL_SEARCH_ALL
608b0c4a	78	int ret;
ae5326a6	79	#endif
57d92753	80
b484b0da SG	81	if (CONFIG_IS_ENABLED(OF_PLATDATA)) {
	82	uclass_first_device(UCLASS_SERIAL, &dev);
	83	if (dev) {
	84	gd->cur_serial_dev = dev;
	85	return;
	86	}
	87	} else if (CONFIG_IS_ENABLED(OF_CONTROL) && blob) {
f93472a0 SG	88	/* Live tree has support for stdout */
	89	if (of_live_active()) {
	90	struct device_node *np = of_get_stdout();
	91
	92	if (np && !uclass_get_device_by_ofnode(UCLASS_SERIAL,
	93	np_to_ofnode(np), &dev)) {
	94	gd->cur_serial_dev = dev;
	95	return;
	96	}
	97	} else {
	98	if (!serial_check_stdout(blob, &dev)) {
	99	gd->cur_serial_dev = dev;
	100	return;
	101	}
469a579d	102	}
af282245	103	}
6bdc593e	104	if (!SPL_BUILD \|\| !CONFIG_IS_ENABLED(OF_CONTROL) \|\| !blob) {
e9fc0583	105	/*
eb623b98 SR	106	* Try to use CONFIG_CONS_INDEX if available (it is numbered
	107	* from 1!).
	108	*
	109	* Failing that, get the device with sequence number 0, or in
608b0c4a AG	110	* extremis just the first working serial device we can find.
608b0c4a AG	111	* But we insist on having a console (even if it is silent).
eb623b98	112	*/
91155c65 SG	113	#ifdef CONFIG_CONS_INDEX
	114	#define INDEX (CONFIG_CONS_INDEX - 1)
	115	#else
	116	#define INDEX 0
	117	#endif
ae5326a6 AG	118
ae5326a6 AG	119	#ifdef CONFIG_SERIAL_SEARCH_ALL
e9fc0583	120	if (!uclass_get_device_by_seq(UCLASS_SERIAL, INDEX, &dev) \|\|
608b0c4a AG	121	!uclass_get_device(UCLASS_SERIAL, INDEX, &dev)) {
	122	if (dev->flags & DM_FLAG_ACTIVATED) {
	123	gd->cur_serial_dev = dev;
	124	return;
	125	}
	126	}
	127
	128	/* Search for any working device */
	129	for (ret = uclass_first_device_check(UCLASS_SERIAL, &dev);
	130	dev;
	131	ret = uclass_next_device_check(&dev)) {
	132	if (!ret) {
	133	/* Device did succeed probing */
	134	gd->cur_serial_dev = dev;
	135	return;
	136	}
e9fc0583	137	}
ae5326a6 AG	138	#else
	139	if (!uclass_get_device_by_seq(UCLASS_SERIAL, INDEX, &dev) \|\|
	140	!uclass_get_device(UCLASS_SERIAL, INDEX, &dev) \|\|
	141	(!uclass_first_device(UCLASS_SERIAL, &dev) && dev)) {
	142	gd->cur_serial_dev = dev;
	143	return;
	144	}
	145	#endif
	146
91155c65	147	#undef INDEX
e9fc0583 SG	148	}
e9fc0583 SG	149
8c458588	150	#ifdef CONFIG_REQUIRE_SERIAL_CONSOLE
e9fc0583	151	panic_str("No serial driver found");
8c458588	152	#endif
57d92753 SG	153	}
	154
	155	/* Called prior to relocation */
	156	int serial_init(void)
	157	{
	158	serial_find_console_or_panic();
	159	gd->flags \|= GD_FLG_SERIAL_READY;
	160
	161	return 0;
	162	}
	163
	164	/* Called after relocation */
	165	void serial_initialize(void)
	166	{
e4d6ab0c	167	serial_init();
57d92753 SG	168	}
57d92753 SG	169
eea5a4cc	170	static void _serial_putc(struct udevice *dev, char ch)
57d92753	171	{
bac64467	172	struct dm_serial_ops *ops = serial_get_ops(dev);
57d92753 SG	173	int err;
57d92753 SG	174
c5917b4b AW	175	if (ch == '\n')
	176	_serial_putc(dev, '\r');
	177
57d92753	178	do {
bac64467	179	err = ops->putc(dev, ch);
57d92753	180	} while (err == -EAGAIN);
57d92753 SG	181	}
57d92753 SG	182
eea5a4cc	183	static void _serial_puts(struct udevice dev, const char str)
b8893327	184	{
eea5a4cc MY	185	while (*str)
eea5a4cc MY	186	_serial_putc(dev, *str++);
b8893327 SG	187	}
b8893327 SG	188
3ca7a06a	189	static int __serial_getc(struct udevice *dev)
57d92753	190	{
eea5a4cc MY	191	struct dm_serial_ops *ops = serial_get_ops(dev);
eea5a4cc MY	192	int err;
57d92753	193
eea5a4cc MY	194	do {
	195	err = ops->getc(dev);
	196	if (err == -EAGAIN)
	197	WATCHDOG_RESET();
	198	} while (err == -EAGAIN);
57d92753	199
eea5a4cc	200	return err >= 0 ? err : 0;
57d92753 SG	201	}
57d92753 SG	202
3ca7a06a	203	static int __serial_tstc(struct udevice *dev)
57d92753	204	{
eea5a4cc	205	struct dm_serial_ops *ops = serial_get_ops(dev);
57d92753 SG	206
57d92753 SG	207	if (ops->pending)
eea5a4cc	208	return ops->pending(dev, true);
57d92753 SG	209
	210	return 1;
	211	}
	212
3ca7a06a SR	213	#if CONFIG_IS_ENABLED(SERIAL_RX_BUFFER)
	214	static int _serial_tstc(struct udevice *dev)
	215	{
	216	struct serial_dev_priv *upriv = dev_get_uclass_priv(dev);
	217
	218	/* Read all available chars into the RX buffer */
	219	while (__serial_tstc(dev)) {
	220	upriv->buf[upriv->wr_ptr++] = __serial_getc(dev);
	221	upriv->wr_ptr %= CONFIG_SERIAL_RX_BUFFER_SIZE;
	222	}
	223
	224	return upriv->rd_ptr != upriv->wr_ptr ? 1 : 0;
	225	}
	226
	227	static int _serial_getc(struct udevice *dev)
	228	{
	229	struct serial_dev_priv *upriv = dev_get_uclass_priv(dev);
	230	char val;
	231
	232	val = upriv->buf[upriv->rd_ptr++];
	233	upriv->rd_ptr %= CONFIG_SERIAL_RX_BUFFER_SIZE;
	234
	235	return val;
	236	}
	237
	238	#else /* CONFIG_IS_ENABLED(SERIAL_RX_BUFFER) */
	239
	240	static int _serial_getc(struct udevice *dev)
	241	{
	242	return __serial_getc(dev);
	243	}
	244
	245	static int _serial_tstc(struct udevice *dev)
	246	{
	247	return __serial_tstc(dev);
	248	}
	249	#endif /* CONFIG_IS_ENABLED(SERIAL_RX_BUFFER) */
	250
eea5a4cc	251	void serial_putc(char ch)
57d92753	252	{
8c458588 HG	253	if (gd->cur_serial_dev)
8c458588 HG	254	_serial_putc(gd->cur_serial_dev, ch);
eea5a4cc	255	}
57d92753	256
eea5a4cc MY	257	void serial_puts(const char *str)
eea5a4cc MY	258	{
8c458588 HG	259	if (gd->cur_serial_dev)
8c458588 HG	260	_serial_puts(gd->cur_serial_dev, str);
57d92753 SG	261	}
57d92753 SG	262
b8893327 SG	263	int serial_getc(void)
b8893327 SG	264	{
8c458588 HG	265	if (!gd->cur_serial_dev)
	266	return 0;
	267
469a579d	268	return _serial_getc(gd->cur_serial_dev);
eea5a4cc MY	269	}
	270
	271	int serial_tstc(void)
	272	{
8c458588 HG	273	if (!gd->cur_serial_dev)
	274	return 0;
	275
469a579d	276	return _serial_tstc(gd->cur_serial_dev);
eea5a4cc MY	277	}
	278
	279	void serial_setbrg(void)
	280	{
8c458588 HG	281	struct dm_serial_ops *ops;
	282
	283	if (!gd->cur_serial_dev)
	284	return;
eea5a4cc	285
8c458588	286	ops = serial_get_ops(gd->cur_serial_dev);
eea5a4cc	287	if (ops->setbrg)
469a579d	288	ops->setbrg(gd->cur_serial_dev, gd->baudrate);
b8893327 SG	289	}
b8893327 SG	290
57d92753 SG	291	void serial_stdio_init(void)
	292	{
	293	}
	294
49ddcf3e MY	295	#if defined(CONFIG_DM_STDIO)
	296
	297	#if CONFIG_IS_ENABLED(SERIAL_PRESENT)
b8893327	298	static void serial_stub_putc(struct stdio_dev *sdev, const char ch)
57d92753	299	{
eea5a4cc	300	_serial_putc(sdev->priv, ch);
57d92753	301	}
236f2bd3	302	#endif
57d92753	303
49ddcf3e	304	static void serial_stub_puts(struct stdio_dev sdev, const char str)
57d92753	305	{
eea5a4cc	306	_serial_puts(sdev->priv, str);
57d92753 SG	307	}
57d92753 SG	308
49ddcf3e	309	static int serial_stub_getc(struct stdio_dev *sdev)
57d92753	310	{
eea5a4cc	311	return _serial_getc(sdev->priv);
57d92753 SG	312	}
57d92753 SG	313
49ddcf3e	314	static int serial_stub_tstc(struct stdio_dev *sdev)
57d92753	315	{
eea5a4cc	316	return _serial_tstc(sdev->priv);
57d92753	317	}
49ddcf3e	318	#endif
57d92753	319
ad1b81c8 SG	320	/**
	321	* on_baudrate() - Update the actual baudrate when the env var changes
	322	*
	323	* This will check for a valid baudrate and only apply it if valid.
	324	*/
	325	static int on_baudrate(const char name, const char value, enum env_op op,
	326	int flags)
	327	{
	328	int i;
	329	int baudrate;
	330
	331	switch (op) {
	332	case env_op_create:
	333	case env_op_overwrite:
	334	/*
	335	* Switch to new baudrate if new baudrate is supported
	336	*/
	337	baudrate = simple_strtoul(value, NULL, 10);
	338
	339	/* Not actually changing */
	340	if (gd->baudrate == baudrate)
	341	return 0;
	342
	343	for (i = 0; i < ARRAY_SIZE(baudrate_table); ++i) {
	344	if (baudrate == baudrate_table[i])
	345	break;
	346	}
	347	if (i == ARRAY_SIZE(baudrate_table)) {
	348	if ((flags & H_FORCE) == 0)
	349	printf("## Baudrate %d bps not supported\n",
	350	baudrate);
	351	return 1;
	352	}
	353	if ((flags & H_INTERACTIVE) != 0) {
	354	printf("## Switch baudrate to %d bps and press ENTER ...\n",
	355	baudrate);
	356	udelay(50000);
	357	}
	358
	359	gd->baudrate = baudrate;
	360
	361	serial_setbrg();
	362
	363	udelay(50000);
	364
	365	if ((flags & H_INTERACTIVE) != 0)
	366	while (1) {
	367	if (getc() == '\r')
	368	break;
	369	}
	370
	371	return 0;
	372	case env_op_delete:
	373	printf("## Baudrate may not be deleted\n");
	374	return 1;
	375	default:
	376	return 0;
	377	}
	378	}
	379	U_BOOT_ENV_CALLBACK(baudrate, on_baudrate);
	380
92c55b68	381	#if CONFIG_IS_ENABLED(SERIAL_PRESENT)
57d92753 SG	382	static int serial_post_probe(struct udevice *dev)
57d92753 SG	383	{
57d92753	384	struct dm_serial_ops *ops = serial_get_ops(dev);
236f2bd3	385	#ifdef CONFIG_DM_STDIO
e564f054	386	struct serial_dev_priv *upriv = dev_get_uclass_priv(dev);
236f2bd3 SG	387	struct stdio_dev sdev;
236f2bd3 SG	388	#endif
57d92753 SG	389	int ret;
57d92753 SG	390
484fdf5b MS	391	#if defined(CONFIG_NEEDS_MANUAL_RELOC)
	392	if (ops->setbrg)
	393	ops->setbrg += gd->reloc_off;
	394	if (ops->getc)
	395	ops->getc += gd->reloc_off;
	396	if (ops->putc)
	397	ops->putc += gd->reloc_off;
	398	if (ops->pending)
	399	ops->pending += gd->reloc_off;
	400	if (ops->clear)
	401	ops->clear += gd->reloc_off;
	402	#if CONFIG_POST & CONFIG_SYS_POST_UART
	403	if (ops->loop)
	404	ops->loop += gd->reloc_off
	405	#endif
	406	#endif
57d92753 SG	407	/* Set the baud rate */
	408	if (ops->setbrg) {
	409	ret = ops->setbrg(dev, gd->baudrate);
	410	if (ret)
	411	return ret;
	412	}
	413
236f2bd3	414	#ifdef CONFIG_DM_STDIO
57d92753 SG	415	if (!(gd->flags & GD_FLG_RELOC))
57d92753 SG	416	return 0;
57d92753 SG	417	memset(&sdev, '\0', sizeof(sdev));
	418
	419	strncpy(sdev.name, dev->name, sizeof(sdev.name));
7b3c4c3a	420	sdev.flags = DEV_FLAGS_OUTPUT \| DEV_FLAGS_INPUT \| DEV_FLAGS_DM;
57d92753 SG	421	sdev.priv = dev;
	422	sdev.putc = serial_stub_putc;
	423	sdev.puts = serial_stub_puts;
	424	sdev.getc = serial_stub_getc;
	425	sdev.tstc = serial_stub_tstc;
3ca7a06a SR	426
	427	#if CONFIG_IS_ENABLED(SERIAL_RX_BUFFER)
	428	/* Allocate the RX buffer */
	429	upriv->buf = malloc(CONFIG_SERIAL_RX_BUFFER_SIZE);
	430	#endif
	431
57d92753	432	stdio_register_dev(&sdev, &upriv->sdev);
236f2bd3	433	#endif
57d92753 SG	434	return 0;
	435	}
	436
	437	static int serial_pre_remove(struct udevice *dev)
	438	{
869588de	439	#if CONFIG_IS_ENABLED(SYS_STDIO_DEREGISTER)
e564f054	440	struct serial_dev_priv *upriv = dev_get_uclass_priv(dev);
57d92753	441
e98856fc	442	if (stdio_deregister_dev(upriv->sdev, true))
57d92753 SG	443	return -EPERM;
	444	#endif
	445
	446	return 0;
	447	}
	448
	449	UCLASS_DRIVER(serial) = {
	450	.id = UCLASS_SERIAL,
	451	.name = "serial",
9cc36a2b	452	.flags = DM_UC_FLAG_SEQ_ALIAS,
57d92753 SG	453	.post_probe = serial_post_probe,
	454	.pre_remove = serial_pre_remove,
	455	.per_device_auto_alloc_size = sizeof(struct serial_dev_priv),
	456	};
92c55b68	457	#endif