efi_loader: remove superfluous check in efi_setup_loaded_image()
[oweals/u-boot.git] / common / spl / spl_ymodem.c
1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3  * (C) Copyright 2000-2004
4  * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
5  *
6  * (C) Copyright 2011
7  * Texas Instruments, <www.ti.com>
8  *
9  * Matt Porter <mporter@ti.com>
10  */
11 #include <common.h>
12 #include <spl.h>
13 #include <xyzModem.h>
14 #include <asm/u-boot.h>
15 #include <linux/libfdt.h>
16
17 #define BUF_SIZE 1024
18
19 /*
20  * Information required to load image using ymodem.
21  *
22  * @image_read: Now of bytes read from the image.
23  * @buf: pointer to the previous read block.
24  */
25 struct ymodem_fit_info {
26         int image_read;
27         char *buf;
28 };
29
30 static int getcymodem(void) {
31         if (tstc())
32                 return (getc());
33         return -1;
34 }
35
36 static ulong ymodem_read_fit(struct spl_load_info *load, ulong offset,
37                              ulong size, void *addr)
38 {
39         int res, err;
40         struct ymodem_fit_info *info = load->priv;
41         char *buf = info->buf;
42
43         while (info->image_read < offset) {
44                 res = xyzModem_stream_read(buf, BUF_SIZE, &err);
45                 if (res <= 0)
46                         return res;
47                 info->image_read += res;
48         }
49
50         if (info->image_read > offset) {
51                 res = info->image_read - offset;
52                 memcpy(addr, &buf[BUF_SIZE - res], res);
53                 addr = addr + res;
54         }
55
56         while (info->image_read < offset + size) {
57                 res = xyzModem_stream_read(buf, BUF_SIZE, &err);
58                 if (res <= 0)
59                         return res;
60
61                 memcpy(addr, buf, res);
62                 info->image_read += res;
63                 addr += res;
64         }
65
66         return size;
67 }
68
69 static int spl_ymodem_load_image(struct spl_image_info *spl_image,
70                                  struct spl_boot_device *bootdev)
71 {
72         ulong size = 0;
73         int err;
74         int res;
75         int ret;
76         connection_info_t info;
77         char buf[BUF_SIZE];
78         struct image_header *ih;
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(spl_image, &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                 ih = (struct image_header *)buf;
112                 ret = spl_parse_image_header(spl_image, ih);
113                 if (ret)
114                         return ret;
115 #ifdef CONFIG_SPL_GZIP
116                 if (ih->ih_comp == IH_COMP_GZIP)
117                         addr = CONFIG_SYS_LOAD_ADDR;
118                 else
119 #endif
120                         addr = spl_image->load_addr;
121                 memcpy((void *)addr, buf, res);
122                 ih = (struct image_header *)addr;
123                 size += res;
124                 addr += res;
125
126                 while ((res = xyzModem_stream_read(buf, BUF_SIZE, &err)) > 0) {
127                         memcpy((void *)addr, buf, res);
128                         size += res;
129                         addr += res;
130                 }
131
132 #ifdef CONFIG_SPL_GZIP
133                 if (ih->ih_comp == IH_COMP_GZIP) {
134                         if (gunzip((void *)(spl_image->load_addr + sizeof(*ih)),
135                                    CONFIG_SYS_BOOTM_LEN,
136                                    (void *)(CONFIG_SYS_LOAD_ADDR + sizeof(*ih)),
137                                    &size)) {
138                                 puts("Uncompressing error\n");
139                                 return -EIO;
140                         }
141                 }
142 #endif
143         }
144
145 end_stream:
146         xyzModem_stream_close(&err);
147         xyzModem_stream_terminate(false, &getcymodem);
148
149         printf("Loaded %lu bytes\n", size);
150         return 0;
151 }
152 SPL_LOAD_IMAGE_METHOD("UART", 0, BOOT_DEVICE_UART, spl_ymodem_load_image);