[people/ms/u-boot.git] / common / spl / spl_ymodem.c

/*
 * (C) Copyright 2000-2004
 * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
 *
 * (C) Copyright 2011
 * Texas Instruments, <www.ti.com>
 *
 * Matt Porter <mporter@ti.com>
 *
 * SPDX-License-Identifier:	GPL-2.0+
 */
#include <common.h>
#include <spl.h>
#include <xyzModem.h>
#include <asm/u-boot.h>
#include <asm/utils.h>
#include <libfdt.h>

#define BUF_SIZE 1024

/*
 * Information required to load image using ymodem.
 *
 * @image_read: Now of bytes read from the image.
 * @buf: pointer to the previous read block.
 */
struct ymodem_fit_info {
	int image_read;
	char *buf;
};

static int getcymodem(void) {
	if (tstc())
		return (getc());
	return -1;
}

static ulong ymodem_read_fit(struct spl_load_info *load, ulong offset,
			     ulong size, void *addr)
{
	int res, err;
	struct ymodem_fit_info *info = load->priv;
	char *buf = info->buf;

	while (info->image_read < offset) {
		res = xyzModem_stream_read(buf, BUF_SIZE, &err);
		if (res <= 0)
			return res;
		info->image_read += res;
	}

	if (info->image_read > offset) {
		res = info->image_read - offset;
		memcpy(addr, &buf[BUF_SIZE - res], res);
		addr = addr + res;
	}

	while (info->image_read < offset + size) {
		res = xyzModem_stream_read(buf, BUF_SIZE, &err);
		if (res <= 0)
			return res;

		memcpy(addr, buf, res);
		info->image_read += res;
		addr += res;
	}

	return size;
}

static int spl_ymodem_load_image(struct spl_image_info *spl_image,
				 struct spl_boot_device *bootdev)
{
	int size = 0;
	int err;
	int res;
	int ret;
	connection_info_t info;
	char buf[BUF_SIZE];
	ulong addr = 0;

	info.mode = xyzModem_ymodem;
	ret = xyzModem_stream_open(&info, &err);
	if (ret) {
		printf("spl: ymodem err - %s\n", xyzModem_error(err));
		return ret;
	}

	res = xyzModem_stream_read(buf, BUF_SIZE, &err);
	if (res <= 0)
		goto end_stream;

	if (IS_ENABLED(CONFIG_SPL_LOAD_FIT) &&
	    image_get_magic((struct image_header *)buf) == FDT_MAGIC) {
		struct spl_load_info load;
		struct ymodem_fit_info info;

		debug("Found FIT\n");
		load.dev = NULL;
		load.priv = (void *)&info;
		load.filename = NULL;
		load.bl_len = 1;
		info.buf = buf;
		info.image_read = BUF_SIZE;
		load.read = ymodem_read_fit;
		ret = spl_load_simple_fit(spl_image, &load, 0, (void *)buf);
		size = info.image_read;

		while ((res = xyzModem_stream_read(buf, BUF_SIZE, &err)) > 0)
			size += res;
	} else {
		ret = spl_parse_image_header(spl_image,
					     (struct image_header *)buf);
		if (ret)
			return ret;
		addr = spl_image->load_addr;
		memcpy((void *)addr, buf, res);
		size += res;
		addr += res;

		while ((res = xyzModem_stream_read(buf, BUF_SIZE, &err)) > 0) {
			memcpy((void *)addr, buf, res);
			size += res;
			addr += res;
		}
	}

end_stream:
	xyzModem_stream_close(&err);
	xyzModem_stream_terminate(false, &getcymodem);

	printf("Loaded %d bytes\n", size);
	return 0;
}
SPL_LOAD_IMAGE_METHOD("UART", 0, BOOT_DEVICE_UART, spl_ymodem_load_image);
Commit	Line	Data
24de357a MP	1	/*
	2	* (C) Copyright 2000-2004
	3	* Wolfgang Denk, DENX Software Engineering, wd@denx.de.
	4	*
	5	* (C) Copyright 2011
	6	* Texas Instruments, <www.ti.com>
	7	*
	8	* Matt Porter <mporter@ti.com>
	9	*
1a459660	10	* SPDX-License-Identifier: GPL-2.0+
24de357a MP	11	*/
24de357a MP	12	#include <common.h>
47f7bcae	13	#include <spl.h>
24de357a MP	14	#include <xyzModem.h>
	15	#include <asm/u-boot.h>
	16	#include <asm/utils.h>
fa715193	17	#include <libfdt.h>
24de357a MP	18
	19	#define BUF_SIZE 1024
	20
fa715193 LV	21	/*
	22	* Information required to load image using ymodem.
	23	*
	24	* @image_read: Now of bytes read from the image.
	25	* @buf: pointer to the previous read block.
	26	*/
	27	struct ymodem_fit_info {
	28	int image_read;
	29	char *buf;
	30	};
	31
24de357a MP	32	static int getcymodem(void) {
	33	if (tstc())
	34	return (getc());
	35	return -1;
	36	}
	37
fa715193 LV	38	static ulong ymodem_read_fit(struct spl_load_info *load, ulong offset,
	39	ulong size, void *addr)
	40	{
	41	int res, err;
	42	struct ymodem_fit_info *info = load->priv;
	43	char *buf = info->buf;
	44
	45	while (info->image_read < offset) {
	46	res = xyzModem_stream_read(buf, BUF_SIZE, &err);
	47	if (res <= 0)
	48	return res;
	49	info->image_read += res;
	50	}
	51
	52	if (info->image_read > offset) {
	53	res = info->image_read - offset;
	54	memcpy(addr, &buf[BUF_SIZE - res], res);
	55	addr = addr + res;
	56	}
	57
	58	while (info->image_read < offset + size) {
	59	res = xyzModem_stream_read(buf, BUF_SIZE, &err);
	60	if (res <= 0)
	61	return res;
	62
	63	memcpy(addr, buf, res);
	64	info->image_read += res;
	65	addr += res;
	66	}
	67
	68	return size;
	69	}
	70
2a2ee2ac SG	71	static int spl_ymodem_load_image(struct spl_image_info *spl_image,
2a2ee2ac SG	72	struct spl_boot_device *bootdev)
24de357a MP	73	{
	74	int size = 0;
	75	int err;
	76	int res;
	77	int ret;
	78	connection_info_t info;
	79	char buf[BUF_SIZE];
24de357a MP	80	ulong addr = 0;
	81
	82	info.mode = xyzModem_ymodem;
	83	ret = xyzModem_stream_open(&info, &err);
fa715193 LV	84	if (ret) {
	85	printf("spl: ymodem err - %s\n", xyzModem_error(err));
	86	return ret;
	87	}
	88
	89	res = xyzModem_stream_read(buf, BUF_SIZE, &err);
	90	if (res <= 0)
	91	goto end_stream;
	92
	93	if (IS_ENABLED(CONFIG_SPL_LOAD_FIT) &&
	94	image_get_magic((struct image_header *)buf) == FDT_MAGIC) {
	95	struct spl_load_info load;
	96	struct ymodem_fit_info info;
	97
	98	debug("Found FIT\n");
	99	load.dev = NULL;
	100	load.priv = (void *)&info;
	101	load.filename = NULL;
	102	load.bl_len = 1;
	103	info.buf = buf;
	104	info.image_read = BUF_SIZE;
	105	load.read = ymodem_read_fit;
f4d7d859	106	ret = spl_load_simple_fit(spl_image, &load, 0, (void *)buf);
fa715193	107	size = info.image_read;
24de357a	108
fa715193 LV	109	while ((res = xyzModem_stream_read(buf, BUF_SIZE, &err)) > 0)
	110	size += res;
	111	} else {
2a2ee2ac	112	ret = spl_parse_image_header(spl_image,
71316c1d	113	(struct image_header *)buf);
fa715193 LV	114	if (ret)
fa715193 LV	115	return ret;
2a2ee2ac	116	addr = spl_image->load_addr;
fa715193 LV	117	memcpy((void *)addr, buf, res);
	118	size += res;
	119	addr += res;
	120
	121	while ((res = xyzModem_stream_read(buf, BUF_SIZE, &err)) > 0) {
	122	memcpy((void *)addr, buf, res);
24de357a MP	123	size += res;
24de357a MP	124	addr += res;
24de357a	125	}
24de357a MP	126	}
24de357a MP	127
fa715193	128	end_stream:
24de357a MP	129	xyzModem_stream_close(&err);
	130	xyzModem_stream_terminate(false, &getcymodem);
	131
	132	printf("Loaded %d bytes\n", size);
36afd451	133	return 0;
24de357a	134	}
ebc4ef61	135	SPL_LOAD_IMAGE_METHOD("UART", 0, BOOT_DEVICE_UART, spl_ymodem_load_image);