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