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; }
/* * This function computes the start and end addresses for both * erase and protect commands. The range of the addresses on which * either of the commands is to operate can be given in two forms: * 1. <cmd> start end - operate on <'start', 'end') * 2. <cmd> start +length - operate on <'start', start + length) * If the second form is used and the end address doesn't fall on the * sector boundary, than it will be adjusted to the next sector boundary. * If it isn't in the flash, the function will fail (return -1). * Input: * arg1, arg2: address specification (i.e. both command arguments) * Output: * addr_first, addr_last: computed address range * Return: * 1: success * -1: failure (bad format, bad address). */ static int addr_spec(char *arg1, char *arg2, ulong *addr_first, ulong *addr_last) { char *ep; char len_used; /* indicates if the "start +length" form used */ *addr_first = simple_strtoul(arg1, &ep, 16); if (ep == arg1 || *ep != '\0') return -1; len_used = 0; if (arg2 && *arg2 == '+'){ len_used = 1; ++arg2; } *addr_last = simple_strtoul(arg2, &ep, 16); if (ep == arg2 || *ep != '\0') return -1; if (len_used){ /* * *addr_last has the length, compute correct *addr_last * XXX watch out for the integer overflow! Right now it is * checked for in both the callers. */ *addr_last = *addr_first + *addr_last - 1; /* * It may happen that *addr_last doesn't fall on the sector * boundary. We want to round such an address to the next * sector boundary, so that the commands don't fail later on. */ if (flash_sect_roundb(addr_last) > 0) return -1; } /* "start +length" from used */ return 1; }