diff options
Diffstat (limited to 'rbutil/mks5lboot/dualboot/dualboot.c')
-rw-r--r-- | rbutil/mks5lboot/dualboot/dualboot.c | 287 |
1 files changed, 287 insertions, 0 deletions
diff --git a/rbutil/mks5lboot/dualboot/dualboot.c b/rbutil/mks5lboot/dualboot/dualboot.c new file mode 100644 index 0000000000..b8167ec124 --- /dev/null +++ b/rbutil/mks5lboot/dualboot/dualboot.c @@ -0,0 +1,287 @@ +/*************************************************************************** + * __________ __ ___. + * Open \______ \ ____ ____ | | _\_ |__ _______ ___ + * Source | _// _ \_/ ___\| |/ /| __ \ / _ \ \/ / + * Jukebox | | ( <_> ) \___| < | \_\ ( <_> > < < + * Firmware |____|_ /\____/ \___ >__|_ \|___ /\____/__/\_ \ + * \/ \/ \/ \/ \/ + * $Id$ + * + * Copyright (C) 2015 by Cástor Muñoz + * + * This program is free software; you can redistribute it and/or + * modify it under the terms of the GNU General Public License + * as published by the Free Software Foundation; either version 2 + * of the License, or (at your option) any later version. + * + * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY + * KIND, either express or implied. + * + ****************************************************************************/ +#include <stdint.h> +#include <string.h> + +#include "config.h" +#include "system.h" +#include "button.h" + +#include "s5l8702.h" +#include "clocking-s5l8702.h" +#include "spi-s5l8702.h" +#include "nor-target.h" +#include "piezo.h" + +/* How it works: + * + * - dualboot-installer: installs or updates a RB bootloader, the bootloader + * to install/update is already included into dualboot-installer.dfu file, + * once it is executed by the iPod device: + * + * 1) locates an original NORBOOT (ONB): first it looks at offset=32KB, if + * a NORBOOT is found but it is not an ONB then it is supposed it is a + * RB bootloader (that should be updated), then the ONB is loaded from + * offset=32KB+old_BLSIZE). + * 2) write ONB at 32KB+new_BLSIZE, if it fails then: + * 2a) try to restore ONB to its 'pristine' place (offset=32KB), if it + * also fails then the NOR got corrupted (ONB probably destroyed) + * and iTunes should be used to restore the iPod. + * 3) write new (included) RB bootloader at offset=32KB, it it fails then + * goto 2a) + * + * - dualboot-uninstaller: uninstall RB bootloader from NOR, leaving it at + * it's previous (pristine) state. + * + * See bootloader/ipod6g.c for notes on how the RB bootloader works. + * + * + * Pristine NOR Rockboxed NOR + * 1MB ______________ + * | | + * | flsh DIR | + * 1MB-0x200 |______________| + * | | + * | File 1 | + * |..............| + * | | + * . . + * . . + * . . + * | | + * |..............| + * | | . . + * | File N | . . + * |______________| |______________| + * | | | | + * | | | | + * | | | Unused | + * | | | | + * | Unused | 160KB+BLSZ |______________| + * | | | | + * | | | Original | + * | | | NOR boot | + * 160KB |______________| | (decrypted) | + * | | | | + * | | 32KB+BLSZ |______________| + * | Original | | | + * | NOR boot | | Decrypted | + * | (encrypted) | | Rockbox | + * | | | Bootloader | + * 32KB |______________| 32KB |______________| + * | | | | + * | | . . + * | | . . + * |______________| + * | | + * | SysCfg | + * 0 |______________| + * + */ + +#define OF_LOADADDR IRAM1_ORIG + +/* tone sequences: period (uS), duration (ms), silence (ms) */ +static uint16_t alive[] = { 500,100,0, 0 }; +static uint16_t happy[] = { 1000,100,0, 500,150,0, 0 }; +static uint16_t fatal[] = { 3000,500,500, 3000,500,500, 3000,500,0, 0 }; +#define sad2 (&fatal[3]) +#define sad (&fatal[6]) + +/* iPod Classic: decrypted hashes for known OFs */ +static unsigned char of_sha[][SIGN_SZ] = { + "\x66\x66\x76\xDC\x1D\x32\xB2\x46\xA6\xC9\x7D\x5A\x61\xD3\x49\x4C", /* v1.1.2 */ + "\x1E\xF0\xD9\xDE\xC2\x7E\xEC\x02\x7C\x15\x76\xBB\x5C\x4F\x2D\x95", /* v2.0.1 */ + "\x06\x85\xDF\x28\xE4\xD7\xF4\x82\xC0\x73\xB0\x53\x26\xFC\xB0\xFE", /* v2.0.4 */ + "\x60\x80\x7D\x33\xA8\xDE\xF8\x49\xBB\xBE\x01\x45\xFF\x62\x40\x19" /* v2.0.5 */ +}; +#define N_OF (int)(sizeof(of_sha)/SIGN_SZ) + +/* we can assume that unknown FW is a RB bootloader */ +#define FW_RB N_OF + +static int identify_fw(struct Im3Info *hinfo) +{ + unsigned char hash[SIGN_SZ]; + int of; + + /* decrypt hash to identify OF */ + memcpy(hash, hinfo->u.enc12.data_sign, SIGN_SZ); + hwkeyaes(HWKEYAES_DECRYPT, HWKEYAES_UKEY, hash, SIGN_SZ); + + for (of = 0; of < N_OF; of++) + if (memcmp(hash, of_sha[of], SIGN_SZ) == 0) + break; + + return of; +} + +#ifdef DUALBOOT_UNINSTALL +/* Uninstall RB bootloader */ +void main(void) +{ + struct Im3Info *hinfo; + void *fw_addr; + uint16_t *status; + unsigned bl_nor_sz; + + usec_timer_init(); + piezo_seq(alive); + spi_clkdiv(SPI_PORT, 4); /* SPI clock = 27/5 MHz. */ + + hinfo = (struct Im3Info*)OF_LOADADDR; + fw_addr = (void*)hinfo + IM3HDR_SZ; + + if (im3_read(NORBOOT_OFF, hinfo, NULL) != 0) { + status = sad; + goto bye; /* no FW found */ + } + + if (identify_fw(hinfo) != FW_RB) { + status = happy; + goto bye; /* RB bootloader not installed, nothing to do */ + } + + /* if found FW is a RB bootloader, OF should start just behind it */ + bl_nor_sz = im3_nor_sz(hinfo); + if ((im3_read(NORBOOT_OFF + bl_nor_sz, hinfo, fw_addr) != 0) + || (identify_fw(hinfo) == FW_RB)) { + status = sad; + goto bye; /* OF not found */ + } + + /* decrypted OF correctly loaded, encrypt it before restoration */ + im3_crypt(HWKEYAES_ENCRYPT, hinfo, fw_addr); + + /* restore OF to it's original place */ + if (!im3_write(NORBOOT_OFF, hinfo)) { + status = fatal; + goto bye; /* corrupted NOR, use iTunes to restore */ + } + + /* erase freed NOR blocks */ + bootflash_init(SPI_PORT); + bootflash_erase_blocks(SPI_PORT, + (NORBOOT_OFF + im3_nor_sz(hinfo)) >> 12, bl_nor_sz >> 12); + bootflash_close(SPI_PORT); + + status = happy; + +bye: + /* minimum time between the initial and the final beeps */ + while (USEC_TIMER < 2000000); + piezo_seq(status); + WDTCON = 0x100000; /* WDT reset */ + while (1); +} + +#else +/* Install RB bootloader */ +struct Im3Info bl_hinfo __attribute__((section(".im3info.data"))) = +{ + .ident = IM3_IDENT, + .version = IM3_VERSION, + .enc_type = 2, +}; + +static uint32_t get_uint32le(unsigned char *p) +{ + return p[0] | (p[1] << 8) | (p[2] << 16) | (p[3] << 24); +} + +void main(void) +{ + uint16_t *status = happy; + int single_boot; + struct Im3Info *hinfo; + void *fw_addr; + unsigned bl_nor_sz; + + usec_timer_init(); + piezo_seq(alive); + spi_clkdiv(SPI_PORT, 4); /* SPI clock = 27/5 MHz. */ + + /* check for single boot installation, is is configured when + mks5lboot.exe builds the .dfu image */ + single_boot = bl_hinfo.info_sign[0]; + + /* sign RB bootloader (data and header), but don't encrypt it, + use current decrypted image for faster load */ + im3_sign(HWKEYAES_UKEY, (void*)&bl_hinfo + IM3HDR_SZ, + get_uint32le(bl_hinfo.data_sz), bl_hinfo.u.enc12.data_sign); + im3_sign(HWKEYAES_UKEY, &bl_hinfo, IM3INFOSIGN_SZ, bl_hinfo.info_sign); + + if (single_boot) { + if (!im3_write(NORBOOT_OFF, &bl_hinfo)) + status = sad; + goto bye; + } + + hinfo = (struct Im3Info*)OF_LOADADDR; + fw_addr = (void*)hinfo + IM3HDR_SZ; + + if (im3_read(NORBOOT_OFF, hinfo, fw_addr) != 0) { + status = sad; + goto bye; /* no FW found */ + } + + if (identify_fw(hinfo) == FW_RB) { + /* FW found, but not OF, assume it is a RB bootloader, + already decrypted OF should be located just behind */ + int nor_offset = NORBOOT_OFF + im3_nor_sz(hinfo); + if ((im3_read(nor_offset, hinfo, fw_addr) != 0) + || (identify_fw(hinfo) == FW_RB)) { + status = sad; + goto bye; /* OF not found, use iTunes to restore */ + } + } + + bl_nor_sz = im3_nor_sz(&bl_hinfo); + /* safety check - verify we are not going to overwrite useful data */ + if (flsh_get_unused() < bl_nor_sz) { + status = sad2; + goto bye; /* no space if flash, use iTunes to restore */ + } + + /* write decrypted OF and RB bootloader, if any of these fails we + will try to retore OF to its original place */ + if (!im3_write(NORBOOT_OFF + bl_nor_sz, hinfo) + || !im3_write(NORBOOT_OFF, &bl_hinfo)) { + im3_crypt(HWKEYAES_ENCRYPT, hinfo, fw_addr); + if (!im3_write(NORBOOT_OFF, hinfo)) { + /* corrupted NOR, use iTunes to restore */ + status = fatal; + } + else { + /* RB bootloader not succesfully intalled, but device + was restored and should be working as before */ + status = sad; + } + } + +bye: + /* minimum time between the initial and the final beeps */ + while (USEC_TIMER < 2000000); + piezo_seq(status); + WDTCON = 0x100000; /* WDT reset */ + while (1); +} +#endif /* DUALBOOT_UNINSTALL */ |