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);