2018-05-06 21:58:06 +00:00
|
|
|
// SPDX-License-Identifier: GPL-2.0+
|
2012-01-31 12:03:57 +00:00
|
|
|
/*
|
|
|
|
* (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>
|
|
|
|
*/
|
|
|
|
#include <common.h>
|
2019-08-01 15:46:36 +00:00
|
|
|
#include <gzip.h>
|
2020-05-10 17:40:01 +00:00
|
|
|
#include <image.h>
|
2020-05-10 17:40:05 +00:00
|
|
|
#include <log.h>
|
2012-08-13 19:03:19 +00:00
|
|
|
#include <spl.h>
|
2012-01-31 12:03:57 +00:00
|
|
|
#include <xyzModem.h>
|
|
|
|
#include <asm/u-boot.h>
|
2018-03-04 16:20:11 +00:00
|
|
|
#include <linux/libfdt.h>
|
2012-01-31 12:03:57 +00:00
|
|
|
|
|
|
|
#define BUF_SIZE 1024
|
|
|
|
|
2016-05-24 05:04:44 +00:00
|
|
|
/*
|
|
|
|
* 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;
|
|
|
|
};
|
|
|
|
|
2012-01-31 12:03:57 +00:00
|
|
|
static int getcymodem(void) {
|
|
|
|
if (tstc())
|
2020-10-07 16:11:48 +00:00
|
|
|
return (getchar());
|
2012-01-31 12:03:57 +00:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
2016-05-24 05:04:44 +00:00
|
|
|
static ulong ymodem_read_fit(struct spl_load_info *load, ulong offset,
|
|
|
|
ulong size, void *addr)
|
|
|
|
{
|
2019-11-14 13:03:30 +00:00
|
|
|
int res, err, buf_offset;
|
2016-05-24 05:04:44 +00:00
|
|
|
struct ymodem_fit_info *info = load->priv;
|
|
|
|
char *buf = info->buf;
|
2022-01-31 04:19:19 +00:00
|
|
|
ulong copy_size = size;
|
2016-05-24 05:04:44 +00:00
|
|
|
|
|
|
|
while (info->image_read < offset) {
|
|
|
|
res = xyzModem_stream_read(buf, BUF_SIZE, &err);
|
|
|
|
if (res <= 0)
|
2019-08-15 20:55:26 +00:00
|
|
|
break;
|
|
|
|
|
2016-05-24 05:04:44 +00:00
|
|
|
info->image_read += res;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (info->image_read > offset) {
|
|
|
|
res = info->image_read - offset;
|
2019-11-14 13:03:30 +00:00
|
|
|
if (info->image_read % BUF_SIZE)
|
|
|
|
buf_offset = (info->image_read % BUF_SIZE);
|
|
|
|
else
|
|
|
|
buf_offset = BUF_SIZE;
|
2022-01-31 04:19:19 +00:00
|
|
|
|
|
|
|
if (res > copy_size) {
|
|
|
|
memcpy(addr, &buf[buf_offset - res], copy_size);
|
|
|
|
goto done;
|
|
|
|
}
|
2019-11-14 13:03:30 +00:00
|
|
|
memcpy(addr, &buf[buf_offset - res], res);
|
2016-05-24 05:04:44 +00:00
|
|
|
addr = addr + res;
|
2022-01-31 04:19:19 +00:00
|
|
|
copy_size -= res;
|
2016-05-24 05:04:44 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
while (info->image_read < offset + size) {
|
|
|
|
res = xyzModem_stream_read(buf, BUF_SIZE, &err);
|
|
|
|
if (res <= 0)
|
2019-08-15 20:55:26 +00:00
|
|
|
break;
|
2016-05-24 05:04:44 +00:00
|
|
|
|
|
|
|
info->image_read += res;
|
2022-01-31 04:19:19 +00:00
|
|
|
if (res > copy_size) {
|
|
|
|
memcpy(addr, buf, copy_size);
|
|
|
|
goto done;
|
|
|
|
}
|
|
|
|
memcpy(addr, buf, res);
|
2016-05-24 05:04:44 +00:00
|
|
|
addr += res;
|
2022-01-31 04:19:19 +00:00
|
|
|
copy_size -= res;
|
2016-05-24 05:04:44 +00:00
|
|
|
}
|
|
|
|
|
2022-01-31 04:19:19 +00:00
|
|
|
done:
|
2016-05-24 05:04:44 +00:00
|
|
|
return size;
|
|
|
|
}
|
|
|
|
|
2019-08-15 20:55:27 +00:00
|
|
|
int spl_ymodem_load_image(struct spl_image_info *spl_image,
|
|
|
|
struct spl_boot_device *bootdev)
|
2012-01-31 12:03:57 +00:00
|
|
|
{
|
2019-01-07 20:23:22 +00:00
|
|
|
ulong size = 0;
|
2012-01-31 12:03:57 +00:00
|
|
|
int err;
|
|
|
|
int res;
|
|
|
|
int ret;
|
|
|
|
connection_info_t info;
|
|
|
|
char buf[BUF_SIZE];
|
2019-03-12 03:00:09 +00:00
|
|
|
struct image_header *ih = NULL;
|
2012-01-31 12:03:57 +00:00
|
|
|
ulong addr = 0;
|
|
|
|
|
|
|
|
info.mode = xyzModem_ymodem;
|
|
|
|
ret = xyzModem_stream_open(&info, &err);
|
2016-05-24 05:04:44 +00:00
|
|
|
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;
|
|
|
|
|
2019-03-06 21:04:31 +00:00
|
|
|
if (IS_ENABLED(CONFIG_SPL_LOAD_FIT_FULL) &&
|
|
|
|
image_get_magic((struct image_header *)buf) == FDT_MAGIC) {
|
|
|
|
addr = CONFIG_SYS_LOAD_ADDR;
|
|
|
|
ih = (struct image_header *)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;
|
|
|
|
}
|
|
|
|
|
2022-01-14 13:31:38 +00:00
|
|
|
ret = spl_parse_image_header(spl_image, bootdev, ih);
|
2019-03-06 21:04:31 +00:00
|
|
|
if (ret)
|
|
|
|
return ret;
|
|
|
|
} else if (IS_ENABLED(CONFIG_SPL_LOAD_FIT) &&
|
2016-05-24 05:04:44 +00:00
|
|
|
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;
|
2016-09-25 00:20:16 +00:00
|
|
|
ret = spl_load_simple_fit(spl_image, &load, 0, (void *)buf);
|
2016-05-24 05:04:44 +00:00
|
|
|
size = info.image_read;
|
2012-01-31 12:03:57 +00:00
|
|
|
|
2016-05-24 05:04:44 +00:00
|
|
|
while ((res = xyzModem_stream_read(buf, BUF_SIZE, &err)) > 0)
|
|
|
|
size += res;
|
|
|
|
} else {
|
2019-01-07 20:23:22 +00:00
|
|
|
ih = (struct image_header *)buf;
|
2022-01-14 13:31:38 +00:00
|
|
|
ret = spl_parse_image_header(spl_image, bootdev, ih);
|
2016-05-24 05:04:44 +00:00
|
|
|
if (ret)
|
2019-03-12 03:02:39 +00:00
|
|
|
goto end_stream;
|
2019-01-07 20:23:22 +00:00
|
|
|
#ifdef CONFIG_SPL_GZIP
|
|
|
|
if (ih->ih_comp == IH_COMP_GZIP)
|
|
|
|
addr = CONFIG_SYS_LOAD_ADDR;
|
|
|
|
else
|
|
|
|
#endif
|
|
|
|
addr = spl_image->load_addr;
|
2016-05-24 05:04:44 +00:00
|
|
|
memcpy((void *)addr, buf, res);
|
2019-01-07 20:23:22 +00:00
|
|
|
ih = (struct image_header *)addr;
|
2016-05-24 05:04:44 +00:00
|
|
|
size += res;
|
|
|
|
addr += res;
|
|
|
|
|
|
|
|
while ((res = xyzModem_stream_read(buf, BUF_SIZE, &err)) > 0) {
|
|
|
|
memcpy((void *)addr, buf, res);
|
2012-01-31 12:03:57 +00:00
|
|
|
size += res;
|
|
|
|
addr += res;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2016-05-24 05:04:44 +00:00
|
|
|
end_stream:
|
2012-01-31 12:03:57 +00:00
|
|
|
xyzModem_stream_close(&err);
|
|
|
|
xyzModem_stream_terminate(false, &getcymodem);
|
|
|
|
|
2019-01-07 20:23:22 +00:00
|
|
|
printf("Loaded %lu bytes\n", size);
|
2019-03-12 03:02:39 +00:00
|
|
|
|
2019-03-12 03:00:09 +00:00
|
|
|
#ifdef CONFIG_SPL_GZIP
|
|
|
|
if (!(IS_ENABLED(CONFIG_SPL_LOAD_FIT) &&
|
|
|
|
image_get_magic((struct image_header *)buf) == FDT_MAGIC) &&
|
|
|
|
(ih->ih_comp == IH_COMP_GZIP)) {
|
|
|
|
if (gunzip((void *)(spl_image->load_addr + sizeof(*ih)),
|
|
|
|
CONFIG_SYS_BOOTM_LEN,
|
|
|
|
(void *)(CONFIG_SYS_LOAD_ADDR + sizeof(*ih)),
|
|
|
|
&size)) {
|
|
|
|
puts("Uncompressing error\n");
|
|
|
|
return -EIO;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2019-03-12 03:02:39 +00:00
|
|
|
return ret;
|
2012-01-31 12:03:57 +00:00
|
|
|
}
|
2016-11-30 22:30:50 +00:00
|
|
|
SPL_LOAD_IMAGE_METHOD("UART", 0, BOOT_DEVICE_UART, spl_ymodem_load_image);
|