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