h1x0/h300: bring the two bootloaders more in sync

The most major change here is the porting of the failsafe boot
menu and eeprom settings support from the h1x0 bootloader to the
h300 bootloader. This has been successfully tested already and
indeed works about the same as it does on the h1x0 bootloader.

The other major change is the addition of new code to both
bootloaders that will retry the flash boot function after
exitting disk mode. It still falls back to booting from disk
if this either fails or is not configured to boot from flash.

There were also various other modifications to bring the two
closer in sync so there are fewer differences.

Change-Id: I17a5724e03225b57e9d0071387294aa6cd025178

1 files changed, 248 insertions, 3 deletions

diff --git a/bootloader/iriver_h300.c b/bootloader/iriver_h300.c
index ee344d4165..ff3c8de2af 100644
--- a/bootloader/iriver_h300.c
+++ b/bootloader/iriver_h300.c
@@ -43,11 +43,12 @@
 #include "power.h"
 #include "powermgmt.h"
 #include "file.h"
+#include "eeprom_settings.h"
+#include "rbunicode.h"
 #include "pcf50606.h"
 #include "common.h"
 #include "rb-loader.h"
 #include "loader_strerror.h"
-#include "rbunicode.h"
 #include "isp1362.h"
 #include "version.h"
 
@@ -62,6 +63,8 @@
 extern int line;
 extern int remote_line;
 
+static bool recovery_mode = false;
+
 /* Reset the cookie for the crt0 crash check */
 inline void __reset_cookie(void)
 {
@@ -90,9 +93,57 @@ void start_firmware(void)
     asm(" jmp (%a0)");
 }
 
+void start_flashed_romimage(void)
+{
+    uint8_t *src = (uint8_t *)FLASH_ROMIMAGE_ENTRY;
+    uint32_t *reset_vector;
+
+    if (!detect_flashed_romimage())
+        return ;
+
+    reset_vector = (uint32_t *)(&src[sizeof(struct flash_header)+sizeof(uint32_t)]);
+
+    asm(" move.w #0x2700,%sr");
+    __reset_cookie();
+
+    asm(" move.l %0,%%d0" :: "i"(DRAM_START));
+    asm(" movec.l %d0,%vbr");
+    asm(" move.l %0,%%sp" :: "m"(reset_vector[0]));
+    asm(" move.l %0,%%a0" :: "m"(reset_vector[1]));
+    asm(" jmp (%a0)");
+
+    /* Failure */
+    power_off();
+}
+
+void start_flashed_ramimage(void)
+{
+    struct flash_header hdr;
+    uint8_t *buf = (uint8_t *)DRAM_START;
+    uint8_t *src = (uint8_t *)FLASH_RAMIMAGE_ENTRY;
+
+    if (!detect_flashed_ramimage())
+         return;
+
+    /* Load firmware from flash */
+    cpu_boost(true);
+    memcpy(&hdr, src, sizeof(struct flash_header));
+    src += sizeof(struct flash_header);
+    memcpy(buf, src, hdr.length);
+    cpu_boost(false);
+
+    start_firmware();
+
+    /* Failure */
+    power_off();
+}
+
 void shutdown(void)
 {
     printf("Shutting down...");
+    /* Reset the rockbox crash check. */
+    firmware_settings.bl_version = 0;
+    eeprom_settings_store();
 
     /* We need to gracefully spin down the disk to prevent clicks. */
     if (ide_powered())
@@ -106,6 +157,7 @@ void shutdown(void)
 
     sleep(HZ*2);
 
+    /* Backlight OFF */
     backlight_hw_off();
     remote_backlight_hw_off();
 
@@ -131,6 +183,176 @@ void check_battery(void)
     }
 }
 
+void initialize_eeprom(void)
+{
+    if (detect_original_firmware())
+        return ;
+
+    if (!eeprom_settings_init())
+    {
+        recovery_mode = true;
+        return ;
+    }
+
+    /* If bootloader version has not been reset, disk might
+     * not be intact. */
+    if (firmware_settings.bl_version || !firmware_settings.disk_clean)
+    {
+        firmware_settings.disk_clean = false;
+        recovery_mode = true;
+    }
+
+    firmware_settings.bl_version = EEPROM_SETTINGS_BL_MINVER;
+    eeprom_settings_store();
+}
+
+void try_flashboot(void)
+{
+    if (!firmware_settings.initialized)
+        return ;
+
+   switch (firmware_settings.bootmethod)
+    {
+        case BOOT_DISK:
+            return;
+
+        case BOOT_ROM:
+            start_flashed_romimage();
+            recovery_mode = true;
+            break;
+
+        case BOOT_RAM:
+            start_flashed_ramimage();
+            recovery_mode = true;
+            break;
+
+        default:
+            recovery_mode = true;
+            return;
+    }
+}
+
+void failsafe_menu(void)
+{
+    static const char *options[] =
+    {
+        "Boot from disk",
+        "Boot RAM image",
+        "Boot ROM image",
+        "Shutdown"
+    };
+    const int FAILSAFE_OPTIONS = sizeof(options) / sizeof(*options);
+    const long TIMEOUT = 15 * HZ;
+    long start_tick = current_tick;
+    int option = 3;
+    int button;
+    int defopt = -1;
+    char buf[32];
+    int i;
+
+    reset_screen();
+    printf("Bootloader %s", rbversion);
+    check_battery();
+    printf("=========================");
+    line += FAILSAFE_OPTIONS;
+    printf("");
+    printf("  [NAVI] to confirm.");
+    printf("  [REC] to set as default.");
+    printf("");
+
+    if (firmware_settings.initialized)
+    {
+        defopt = firmware_settings.bootmethod;
+        if (defopt < 0 || defopt >= FAILSAFE_OPTIONS)
+            defopt = option;
+    }
+
+    while (current_tick - start_tick < TIMEOUT)
+    {
+        /* Draw the menu. */
+        line = 3;
+        for (i = 0; i < FAILSAFE_OPTIONS; i++)
+        {
+            char *def = "[DEF]";
+            char *arrow = "->";
+
+            if (i != defopt)
+                def = "";
+            if (i != option)
+                arrow = "  ";
+
+            printf("%s %s %s", arrow, options[i], def);
+        }
+
+        snprintf(buf, sizeof(buf), "Time left: %lds",
+                 (TIMEOUT - (current_tick - start_tick)) / HZ);
+        lcd_puts(0, 10, buf);
+        lcd_update();
+        button = button_get_w_tmo(HZ);
+
+        if (button == BUTTON_NONE || button & SYS_EVENT)
+            continue ;
+
+        start_tick = current_tick;
+
+        /* Ignore the ON/PLAY -button because it can cause trouble
+           with the RTC alarm mod. */
+        switch (button & ~(BUTTON_ON))
+        {
+            case BUTTON_UP:
+            case BUTTON_RC_REW:
+                if (option > 0)
+                    option--;
+                break ;
+
+            case BUTTON_DOWN:
+            case BUTTON_RC_FF:
+                if (option < FAILSAFE_OPTIONS-1)
+                    option++;
+                break ;
+
+            case BUTTON_SELECT:
+            case BUTTON_RC_ON:
+                goto execute;
+
+            case BUTTON_REC:
+            case BUTTON_RC_REC:
+                if (firmware_settings.initialized)
+                {
+                    firmware_settings.bootmethod = option;
+                    eeprom_settings_store();
+                    defopt = option;
+                }
+                break ;
+        }
+    }
+
+    execute:
+
+    lcd_puts(0, 10, "Executing command...");
+    lcd_update();
+    sleep(HZ);
+    reset_screen();
+
+    switch (option)
+    {
+        case BOOT_DISK:
+            return ;
+
+        case BOOT_RAM:
+            start_flashed_ramimage();
+            printf("Image not found");
+            break;
+
+        case BOOT_ROM:
+            start_flashed_romimage();
+            printf("Image not found");
+            break;
+    }
+
+    shutdown();
+}
+
 /* From the pcf50606 driver */
 extern unsigned char pcf50606_intregs[3];
 
@@ -187,6 +409,7 @@ void main(void)
     backlight_hw_init();
     backlight_hw_off();
 
+    /* Remote backlight ON */
     remote_backlight_hw_on();
 
     system_init();
@@ -197,11 +420,20 @@ void main(void)
     coldfire_set_pllcr_audio_bits(DEFAULT_PLLCR_AUDIO_BITS);
     enable_irq();
 
+    initialize_eeprom();
+
     isp1362_init();
 
     adc_init();
     button_init();
 
+    /* Power on the hard drive early, to speed up the loading. */
+    if (!hold_status && !recovery_mode)
+        ide_power_enable(true);
+
+    if (!hold_status && (usb_detect() != USB_INSERTED) && !recovery_mode)
+        try_flashboot();
+
     lcd_init();
     lcd_remote_init();
     font_init();
@@ -229,14 +461,16 @@ void main(void)
     {
         hold_status = true;
     }
-    if (hold_status && !rtc_alarm && (usb_detect() != USB_INSERTED) &&
-        !charger_inserted())
+    if ((hold_status || recovery_mode) && !rtc_alarm &&
+        (usb_detect() != USB_INSERTED) && !charger_inserted())
     {
         if (detect_original_firmware())
         {
             printf("Hold switch on");
             shutdown();
         }
+
+        failsafe_menu();
     }
 
     /* Holding REC while starting runs the original firmware */
@@ -321,6 +555,12 @@ void main(void)
         lcd_remote_puts(0, 3, msg);
         lcd_remote_update();
 
+        if (firmware_settings.initialized)
+        {
+            firmware_settings.disk_clean = false;
+            eeprom_settings_store();
+        }
+
         ide_power_enable(true);
         storage_enable(false);
         sleep(HZ/20);
@@ -351,6 +591,11 @@ void main(void)
         usb_charge = false;
     }
 
+    /* boot from flash if that is the default */
+    try_flashboot();
+    if (recovery_mode)
+        printf("Falling back to boot from disk");
+
     rc = storage_init();
     if(rc)
     {