common/update.c

   1 /*
   2  * (C) Copyright 2008 Semihalf
   3  *
   4  * Written by: Rafal Czubak <rcz@semihalf.com>
   5  *             Bartlomiej Sieka <tur@semihalf.com>
   6  *
   7  * SPDX-License-Identifier:     GPL-2.0+
   8  */
   9
  10 #include <common.h>
  11
  12 #if !(defined(CONFIG_FIT) && defined(CONFIG_OF_LIBFDT))
  13 #error "CONFIG_FIT and CONFIG_OF_LIBFDT are required for auto-update feature"
  14 #endif
  15
  16 #if defined(CONFIG_SYS_NO_FLASH)
  17 #error "CONFIG_SYS_NO_FLASH defined, but FLASH is required for auto-update feature"
  18 #endif
  19
  20 #include <command.h>
  21 #include <flash.h>
  22 #include <net.h>
  23 #include <net/tftp.h>
  24 #include <malloc.h>
  25
  26 /* env variable holding the location of the update file */
  27 #define UPDATE_FILE_ENV         "updatefile"
  28
  29 /* set configuration defaults if needed */
  30 #ifndef CONFIG_UPDATE_LOAD_ADDR
  31 #define CONFIG_UPDATE_LOAD_ADDR 0x100000
  32 #endif
  33
  34 #ifndef CONFIG_UPDATE_TFTP_MSEC_MAX
  35 #define CONFIG_UPDATE_TFTP_MSEC_MAX     100
  36 #endif
  37
  38 #ifndef CONFIG_UPDATE_TFTP_CNT_MAX
  39 #define CONFIG_UPDATE_TFTP_CNT_MAX      0
  40 #endif
  41
  42 extern ulong TftpRRQTimeoutMSecs;
  43 extern int TftpRRQTimeoutCountMax;
  44 extern flash_info_t flash_info[];
  45 extern ulong load_addr;
  46
  47 static uchar *saved_prot_info;
  48
  49 static int update_load(char *filename, ulong msec_max, int cnt_max, ulong addr)
  50 {
  51         int size, rv;
  52         ulong saved_timeout_msecs;
  53         int saved_timeout_count;
  54         char *saved_netretry, *saved_bootfile;
  55
  56         rv = 0;
  57         /* save used globals and env variable */
  58         saved_timeout_msecs = TftpRRQTimeoutMSecs;
  59         saved_timeout_count = TftpRRQTimeoutCountMax;
  60         saved_netretry = strdup(getenv("netretry"));
  61         saved_bootfile = strdup(BootFile);
  62
  63         /* set timeouts for auto-update */
  64         TftpRRQTimeoutMSecs = msec_max;
  65         TftpRRQTimeoutCountMax = cnt_max;
  66
  67         /* we don't want to retry the connection if errors occur */
  68         setenv("netretry", "no");
  69
  70         /* download the update file */
  71         load_addr = addr;
  72         copy_filename(BootFile, filename, sizeof(BootFile));
  73         size = NetLoop(TFTPGET);
  74
  75         if (size < 0)
  76                 rv = 1;
  77         else if (size > 0)
  78                 flush_cache(addr, size);
  79
  80         /* restore changed globals and env variable */
  81         TftpRRQTimeoutMSecs = saved_timeout_msecs;
  82         TftpRRQTimeoutCountMax = saved_timeout_count;
  83
  84         setenv("netretry", saved_netretry);
  85         if (saved_netretry != NULL)
  86                 free(saved_netretry);
  87
  88         if (saved_bootfile != NULL) {
  89                 copy_filename(BootFile, saved_bootfile, sizeof(BootFile));
  90                 free(saved_bootfile);
  91         }
  92
  93         return rv;
  94 }
  95
  96 static int update_flash_protect(int prot, ulong addr_first, ulong addr_last)
  97 {
  98         uchar *sp_info_ptr;
  99         ulong s;
 100         int i, bank, cnt;
 101         flash_info_t *info;
 102
 103         sp_info_ptr = NULL;
 104
 105         if (prot == 0) {
 106                 saved_prot_info =
 107                         calloc(CONFIG_SYS_MAX_FLASH_BANKS * CONFIG_SYS_MAX_FLASH_SECT, 1);
 108                 if (!saved_prot_info)
 109                         return 1;
 110         }
 111
 112         for (bank = 0; bank < CONFIG_SYS_MAX_FLASH_BANKS; ++bank) {
 113                 cnt = 0;
 114                 info = &flash_info[bank];
 115
 116                 /* Nothing to do if the bank doesn't exist */
 117                 if (info->sector_count == 0)
 118                         return 0;
 119
 120                 /* Point to current bank protection information */
 121                 sp_info_ptr = saved_prot_info + (bank * CONFIG_SYS_MAX_FLASH_SECT);
 122
 123                 /*
 124                  * Adjust addr_first or addr_last if we are on bank boundary.
 125                  * Address space between banks must be continuous for other
 126                  * flash functions (like flash_sect_erase or flash_write) to
 127                  * succeed. Banks must also be numbered in correct order,
 128                  * according to increasing addresses.
 129                  */
 130                 if (addr_last > info->start[0] + info->size - 1)
 131                         addr_last = info->start[0] + info->size - 1;
 132                 if (addr_first < info->start[0])
 133                         addr_first = info->start[0];
 134
 135                 for (i = 0; i < info->sector_count; i++) {
 136                         /* Save current information about protected sectors */
 137                         if (prot == 0) {
 138                                 s = info->start[i];
 139                                 if ((s >= addr_first) && (s <= addr_last))
 140                                         sp_info_ptr[i] = info->protect[i];
 141
 142                         }
 143
 144                         /* Protect/unprotect sectors */
 145                         if (sp_info_ptr[i] == 1) {
 146 #if defined(CONFIG_SYS_FLASH_PROTECTION)
 147                                 if (flash_real_protect(info, i, prot))
 148                                         return 1;
 149 #else
 150                                 info->protect[i] = prot;
 151 #endif
 152                                 cnt++;
 153                         }
 154                 }
 155
 156                 if (cnt) {
 157                         printf("%sProtected %d sectors\n",
 158                                                 prot ? "": "Un-", cnt);
 159                 }
 160         }
 161
 162         if((prot == 1) && saved_prot_info)
 163                 free(saved_prot_info);
 164
 165         return 0;
 166 }
 167
 168 static int update_flash(ulong addr_source, ulong addr_first, ulong size)
 169 {
 170         ulong addr_last = addr_first + size - 1;
 171
 172         /* round last address to the sector boundary */
 173         if (flash_sect_roundb(&addr_last) > 0)
 174                 return 1;
 175
 176         if (addr_first >= addr_last) {
 177                 printf("Error: end address exceeds addressing space\n");
 178                 return 1;
 179         }
 180
 181         /* remove protection on processed sectors */
 182         if (update_flash_protect(0, addr_first, addr_last) > 0) {
 183                 printf("Error: could not unprotect flash sectors\n");
 184                 return 1;
 185         }
 186
 187         printf("Erasing 0x%08lx - 0x%08lx", addr_first, addr_last);
 188         if (flash_sect_erase(addr_first, addr_last) > 0) {
 189                 printf("Error: could not erase flash\n");
 190                 return 1;
 191         }
 192
 193         printf("Copying to flash...");
 194         if (flash_write((char *)addr_source, addr_first, size) > 0) {
 195                 printf("Error: could not copy to flash\n");
 196                 return 1;
 197         }
 198         printf("done\n");
 199
 200         /* enable protection on processed sectors */
 201         if (update_flash_protect(1, addr_first, addr_last) > 0) {
 202                 printf("Error: could not protect flash sectors\n");
 203                 return 1;
 204         }
 205
 206         return 0;
 207 }
 208
 209 static int update_fit_getparams(const void *fit, int noffset, ulong *addr,
 210                                                 ulong *fladdr, ulong *size)
 211 {
 212         const void *data;
 213
 214         if (fit_image_get_data(fit, noffset, &data, (size_t *)size))
 215                 return 1;
 216
 217         if (fit_image_get_load(fit, noffset, (ulong *)fladdr))
 218                 return 1;
 219
 220         *addr = (ulong)data;
 221
 222         return 0;
 223 }
 224
 225 int update_tftp(ulong addr)
 226 {
 227         char *filename, *env_addr;
 228         int images_noffset, ndepth, noffset;
 229         ulong update_addr, update_fladdr, update_size;
 230         void *fit;
 231         int ret = 0;
 232
 233         /* use already present image */
 234         if (addr)
 235                 goto got_update_file;
 236
 237         printf("Auto-update from TFTP: ");
 238
 239         /* get the file name of the update file */
 240         filename = getenv(UPDATE_FILE_ENV);
 241         if (filename == NULL) {
 242                 printf("failed, env. variable '%s' not found\n",
 243                                                         UPDATE_FILE_ENV);
 244                 return 1;
 245         }
 246
 247         printf("trying update file '%s'\n", filename);
 248
 249         /* get load address of downloaded update file */
 250         if ((env_addr = getenv("loadaddr")) != NULL)
 251                 addr = simple_strtoul(env_addr, NULL, 16);
 252         else
 253                 addr = CONFIG_UPDATE_LOAD_ADDR;
 254
 255
 256         if (update_load(filename, CONFIG_UPDATE_TFTP_MSEC_MAX,
 257                                         CONFIG_UPDATE_TFTP_CNT_MAX, addr)) {
 258                 printf("Can't load update file, aborting auto-update\n");
 259                 return 1;
 260         }
 261
 262 got_update_file:
 263         fit = (void *)addr;
 264
 265         if (!fit_check_format((void *)fit)) {
 266                 printf("Bad FIT format of the update file, aborting "
 267                                                         "auto-update\n");
 268                 return 1;
 269         }
 270
 271         /* process updates */
 272         images_noffset = fdt_path_offset(fit, FIT_IMAGES_PATH);
 273
 274         ndepth = 0;
 275         noffset = fdt_next_node(fit, images_noffset, &ndepth);
 276         while (noffset >= 0 && ndepth > 0) {
 277                 if (ndepth != 1)
 278                         goto next_node;
 279
 280                 printf("Processing update '%s' :",
 281                         fit_get_name(fit, noffset, NULL));
 282
 283                 if (!fit_image_verify(fit, noffset)) {
 284                         printf("Error: invalid update hash, aborting\n");
 285                         ret = 1;
 286                         goto next_node;
 287                 }
 288
 289                 printf("\n");
 290                 if (update_fit_getparams(fit, noffset, &update_addr,
 291                                         &update_fladdr, &update_size)) {
 292                         printf("Error: can't get update parameteres, "
 293                                                                 "aborting\n");
 294                         ret = 1;
 295                         goto next_node;
 296                 }
 297                 if (update_flash(update_addr, update_fladdr, update_size)) {
 298                         printf("Error: can't flash update, aborting\n");
 299                         ret = 1;
 300                         goto next_node;
 301                 }
 302 next_node:
 303                 noffset = fdt_next_node(fit, noffset, &ndepth);
 304         }
 305
 306         return ret;
 307 }