spl: Convert spl_spi_load_image() to use linker list
[oweals/u-boot.git] / common / spl / spl_ymodem.c
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  *
10  * SPDX-License-Identifier:     GPL-2.0+
11  */
12 #include <common.h>
13 #include <spl.h>
14 #include <xyzModem.h>
15 #include <asm/u-boot.h>
16 #include <asm/utils.h>
17 #include <libfdt.h>
18
19 #define BUF_SIZE 1024
20
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
32 static int getcymodem(void) {
33         if (tstc())
34                 return (getc());
35         return -1;
36 }
37
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
71 static int spl_ymodem_load_image(struct spl_boot_device *bootdev)
72 {
73         int size = 0;
74         int err;
75         int res;
76         int ret;
77         connection_info_t info;
78         char buf[BUF_SIZE];
79         ulong addr = 0;
80
81         info.mode = xyzModem_ymodem;
82         ret = xyzModem_stream_open(&info, &err);
83         if (ret) {
84                 printf("spl: ymodem err - %s\n", xyzModem_error(err));
85                 return ret;
86         }
87
88         res = xyzModem_stream_read(buf, BUF_SIZE, &err);
89         if (res <= 0)
90                 goto end_stream;
91
92         if (IS_ENABLED(CONFIG_SPL_LOAD_FIT) &&
93             image_get_magic((struct image_header *)buf) == FDT_MAGIC) {
94                 struct spl_load_info load;
95                 struct ymodem_fit_info info;
96
97                 debug("Found FIT\n");
98                 load.dev = NULL;
99                 load.priv = (void *)&info;
100                 load.filename = NULL;
101                 load.bl_len = 1;
102                 info.buf = buf;
103                 info.image_read = BUF_SIZE;
104                 load.read = ymodem_read_fit;
105                 ret =  spl_load_simple_fit(&load, 0, (void *)buf);
106                 size = info.image_read;
107
108                 while ((res = xyzModem_stream_read(buf, BUF_SIZE, &err)) > 0)
109                         size += res;
110         } else {
111                 spl_parse_image_header(&spl_image, (struct image_header *)buf);
112                 ret = spl_parse_image_header(&spl_image,
113                                              (struct image_header *)buf);
114                 if (ret)
115                         return ret;
116                 addr = spl_image.load_addr;
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);
123                         size += res;
124                         addr += res;
125                 }
126         }
127
128 end_stream:
129         xyzModem_stream_close(&err);
130         xyzModem_stream_terminate(false, &getcymodem);
131
132         printf("Loaded %d bytes\n", size);
133         return 0;
134 }
135 SPL_LOAD_IMAGE_METHOD(0, BOOT_DEVICE_UART, spl_ymodem_load_image);