/* * (C) Copyright 2008 Semihalf * * Written by: Rafal Czubak <rcz@semihalf.com> * Bartlomiej Sieka <tur@semihalf.com> * * SPDX-License-Identifier: GPL-2.0+ */ #include <common.h> #if !(defined(CONFIG_FIT) && defined(CONFIG_OF_LIBFDT)) #error "CONFIG_FIT and CONFIG_OF_LIBFDT are required for auto-update feature" #endif #if defined(CONFIG_UPDATE_TFTP) && defined(CONFIG_SYS_NO_FLASH) #error "CONFIG_UPDATE_TFTP and CONFIG_SYS_NO_FLASH needed for legacy behaviour" #endif #include <command.h> #include <flash.h> #include <net.h> #include <net/tftp.h> #include <malloc.h> #include <dfu.h> #include <errno.h> /* env variable holding the location of the update file */ #define UPDATE_FILE_ENV "updatefile" /* set configuration defaults if needed */ #ifndef CONFIG_UPDATE_LOAD_ADDR #define CONFIG_UPDATE_LOAD_ADDR 0x100000 #endif #ifndef CONFIG_UPDATE_TFTP_MSEC_MAX #define CONFIG_UPDATE_TFTP_MSEC_MAX 100 #endif #ifndef CONFIG_UPDATE_TFTP_CNT_MAX #define CONFIG_UPDATE_TFTP_CNT_MAX 0 #endif extern ulong tftp_timeout_ms; extern int tftp_timeout_count_max; extern ulong load_addr; #ifndef CONFIG_SYS_NO_FLASH extern flash_info_t flash_info[]; static uchar *saved_prot_info; #endif static int update_load(char *filename, ulong msec_max, int cnt_max, ulong addr) { int size, rv; ulong saved_timeout_msecs; int saved_timeout_count; char *saved_netretry, *saved_bootfile; rv = 0; /* save used globals and env variable */ saved_timeout_msecs = tftp_timeout_ms; saved_timeout_count = tftp_timeout_count_max; saved_netretry = strdup(getenv("netretry")); saved_bootfile = strdup(net_boot_file_name); /* set timeouts for auto-update */ tftp_timeout_ms = msec_max; tftp_timeout_count_max = cnt_max; /* we don't want to retry the connection if errors occur */ setenv("netretry", "no"); /* download the update file */ load_addr = addr; copy_filename(net_boot_file_name, filename, sizeof(net_boot_file_name)); size = net_loop(TFTPGET); if (size < 0) rv = 1; else if (size > 0) flush_cache(addr, size); /* restore changed globals and env variable */ tftp_timeout_ms = saved_timeout_msecs; tftp_timeout_count_max = saved_timeout_count; setenv("netretry", saved_netretry); if (saved_netretry != NULL) free(saved_netretry); if (saved_bootfile != NULL) { copy_filename(net_boot_file_name, saved_bootfile, sizeof(net_boot_file_name)); free(saved_bootfile); } return rv; } #ifndef CONFIG_SYS_NO_FLASH static int update_flash_protect(int prot, ulong addr_first, ulong addr_last) { uchar *sp_info_ptr; ulong s; int i, bank, cnt; flash_info_t *info; sp_info_ptr = NULL; if (prot == 0) { saved_prot_info = calloc(CONFIG_SYS_MAX_FLASH_BANKS * CONFIG_SYS_MAX_FLASH_SECT, 1); if (!saved_prot_info) return 1; } for (bank = 0; bank < CONFIG_SYS_MAX_FLASH_BANKS; ++bank) { cnt = 0; info = &flash_info[bank]; /* Nothing to do if the bank doesn't exist */ if (info->sector_count == 0) return 0; /* Point to current bank protection information */ sp_info_ptr = saved_prot_info + (bank * CONFIG_SYS_MAX_FLASH_SECT); /* * Adjust addr_first or addr_last if we are on bank boundary. * Address space between banks must be continuous for other * flash functions (like flash_sect_erase or flash_write) to * succeed. Banks must also be numbered in correct order, * according to increasing addresses. */ if (addr_last > info->start[0] + info->size - 1) addr_last = info->start[0] + info->size - 1; if (addr_first < info->start[0]) addr_first = info->start[0]; for (i = 0; i < info->sector_count; i++) { /* Save current information about protected sectors */ if (prot == 0) { s = info->start[i]; if ((s >= addr_first) && (s <= addr_last)) sp_info_ptr[i] = info->protect[i]; } /* Protect/unprotect sectors */ if (sp_info_ptr[i] == 1) { #if defined(CONFIG_SYS_FLASH_PROTECTION) if (flash_real_protect(info, i, prot)) return 1; #else info->protect[i] = prot; #endif cnt++; } } if (cnt) { printf("%sProtected %d sectors\n", prot ? "": "Un-", cnt); } } if((prot == 1) && saved_prot_info) free(saved_prot_info); return 0; } #endif static int update_flash(ulong addr_source, ulong addr_first, ulong size) { #ifndef CONFIG_SYS_NO_FLASH ulong addr_last = addr_first + size - 1; /* round last address to the sector boundary */ if (flash_sect_roundb(&addr_last) > 0) return 1; if (addr_first >= addr_last) { printf("Error: end address exceeds addressing space\n"); return 1; } /* remove protection on processed sectors */ if (update_flash_protect(0, addr_first, addr_last) > 0) { printf("Error: could not unprotect flash sectors\n"); return 1; } printf("Erasing 0x%08lx - 0x%08lx", addr_first, addr_last); if (flash_sect_erase(addr_first, addr_last) > 0) { printf("Error: could not erase flash\n"); return 1; } printf("Copying to flash..."); if (flash_write((char *)addr_source, addr_first, size) > 0) { printf("Error: could not copy to flash\n"); return 1; } printf("done\n"); /* enable protection on processed sectors */ if (update_flash_protect(1, addr_first, addr_last) > 0) { printf("Error: could not protect flash sectors\n"); return 1; } #endif return 0; } static int update_fit_getparams(const void *fit, int noffset, ulong *addr, ulong *fladdr, ulong *size) { const void *data; if (fit_image_get_data(fit, noffset, &data, (size_t *)size)) return 1; if (fit_image_get_load(fit, noffset, (ulong *)fladdr)) return 1; *addr = (ulong)data; return 0; } int update_tftp(ulong addr, char *interface, char *devstring) { char *filename, *env_addr, *fit_image_name; ulong update_addr, update_fladdr, update_size; int images_noffset, ndepth, noffset; bool update_tftp_dfu; int ret = 0; void *fit; if (interface == NULL && devstring == NULL) { update_tftp_dfu = false; } else if (interface && devstring) { update_tftp_dfu = true; } else { error("Interface: %s and devstring: %s not supported!\n", interface, devstring); return -EINVAL; } /* use already present image */ if (addr) goto got_update_file; printf("Auto-update from TFTP: "); /* get the file name of the update file */ filename = getenv(UPDATE_FILE_ENV); if (filename == NULL) { printf("failed, env. variable '%s' not found\n", UPDATE_FILE_ENV); return 1; } printf("trying update file '%s'\n", filename); /* get load address of downloaded update file */ if ((env_addr = getenv("loadaddr")) != NULL) addr = simple_strtoul(env_addr, NULL, 16); else addr = CONFIG_UPDATE_LOAD_ADDR; if (update_load(filename, CONFIG_UPDATE_TFTP_MSEC_MAX, CONFIG_UPDATE_TFTP_CNT_MAX, addr)) { printf("Can't load update file, aborting auto-update\n"); return 1; } got_update_file: fit = (void *)addr; if (!fit_check_format((void *)fit)) { printf("Bad FIT format of the update file, aborting " "auto-update\n"); return 1; } /* process updates */ images_noffset = fdt_path_offset(fit, FIT_IMAGES_PATH); ndepth = 0; noffset = fdt_next_node(fit, images_noffset, &ndepth); while (noffset >= 0 && ndepth > 0) { if (ndepth != 1) goto next_node; fit_image_name = (char *)fit_get_name(fit, noffset, NULL); printf("Processing update '%s' :", fit_image_name); if (!fit_image_verify(fit, noffset)) { printf("Error: invalid update hash, aborting\n"); ret = 1; goto next_node; } printf("\n"); if (update_fit_getparams(fit, noffset, &update_addr, &update_fladdr, &update_size)) { printf("Error: can't get update parameteres, " "aborting\n"); ret = 1; goto next_node; } if (!update_tftp_dfu) { if (update_flash(update_addr, update_fladdr, update_size)) { printf("Error: can't flash update, aborting\n"); ret = 1; goto next_node; } } else if (fit_image_check_type(fit, noffset, IH_TYPE_FIRMWARE)) { ret = dfu_tftp_write(fit_image_name, update_addr, update_size, interface, devstring); if (ret) return ret; } next_node: noffset = fdt_next_node(fit, noffset, &ndepth); } return ret; }