4 files changed, 227 insertions, 3 deletions

diff --git a/utils/nwztools/plattools/Makefile b/utils/nwztools/plattools/Makefile
index 8251afaf0a..f4e88cd778 100644
--- a/utils/nwztools/plattools/Makefile
+++ b/utils/nwztools/plattools/Makefile
@@ -7,7 +7,7 @@ INCLUDES=-I.
 LIB_FILES=nwz_lib.c nwz_lib_devlist.c
 TOOL_FILES=dest_tool.c test_adc.c test_adc.c test_bl.c test_display.c \
         test_keys.c test_power.c test_ts.c
-ALL_ELF=$(patsubst %.c,%.elf,$(TOOL_FILES)) all_tools.elf
+ALL_ELF=$(patsubst %.c,%.elf,$(TOOL_FILES)) all_tools.elf dualboot.elf
 
 all: $(ALL_ELF)
 
@@ -17,5 +17,8 @@ all: $(ALL_ELF)
 all_tools.elf: all_tools.c $(TOOL_FILES) $(LIB_FILES)
         $(CC) $(CFLAGS) -DNWZ_EMBED_TOOLS $(INCLUDES) -o $@ $^
 
+dualboot.elf: dualboot.c all_tools.c $(TOOL_FILES) $(LIB_FILES)
+        $(CC) $(CFLAGS) -DNWZ_DUALBOOT -DNWZ_EMBED_TOOLS $(INCLUDES) -o $@ $^
+
 clean:
         rm -rf $(ALL_ELF)
diff --git a/utils/nwztools/plattools/dualboot.c b/utils/nwztools/plattools/dualboot.c
new file mode 100644
index 0000000000..965fabed72
--- /dev/null
+++ b/utils/nwztools/plattools/dualboot.c
@@ -0,0 +1,181 @@
+/***************************************************************************
+ *             __________               __   ___.
+ *   Open      \______   \ ____   ____ |  | _\_ |__   _______  ___
+ *   Source     |       _//  _ \_/ ___\|  |/ /| __ \ /  _ \  \/  /
+ *   Jukebox    |    |   (  <_> )  \___|    < | \_\ (  <_> > <  <
+ *   Firmware   |____|_  /\____/ \___  >__|_ \|___  /\____/__/\_ \
+ *                     \/            \/     \/    \/            \/
+ * $Id$
+ *
+ * Copyright (C) 2011 by Amaury Pouly
+ *
+ * Based on Rockbox iriver bootloader by Linus Nielsen Feltzing
+ * and the ipodlinux bootloader by Daniel Palffy and Bernard Leach
+ * 
+ * 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 "nwz_lib.h"
+#include "nwz_plattools.h"
+#include <time.h>
+
+/* Important Note: this bootloader is carefully written so that in case of
+ * error, the OF is run. This seems like the safest option since the OF is
+ * always there and might do magic things. */
+
+bool boot_rockbox(void)
+{
+    /* get time */
+    struct timeval deadline;
+    if(gettimeofday(&deadline, NULL) != 0)
+    {
+        nwz_lcdmsg(false, 0, 2, "Cannot get time");
+        sleep(2);
+        return false;
+    }
+    /* open input device */
+    int input_fd = nwz_key_open();
+    if(input_fd < 0)
+    {
+        nwz_lcdmsg(false, 0, 2, "Cannot open input device");
+        sleep(2);
+        return false;
+    }
+    deadline.tv_sec += 5;
+    /* wait for user action */
+    bool boot_rb = false;
+    while(true)
+    {
+        /* get time */
+        struct timeval cur_time;
+        if(gettimeofday(&cur_time, NULL) != 0)
+        {
+            nwz_lcdmsg(false, 0, 4, "Cannot get time");
+            sleep(2);
+            break;
+        }
+        /* check timeout */
+        if(cur_time.tv_sec > deadline.tv_sec)
+            break;
+        if(cur_time.tv_sec == deadline.tv_sec && cur_time.tv_usec >= deadline.tv_usec)
+            break;
+        /* print message */
+        int sec_left = deadline.tv_sec - cur_time.tv_sec;
+        sec_left += (deadline.tv_usec - cur_time.tv_usec + 999999) / 1000000; /* round up */
+        nwz_lcdmsgf(false, 0, 2, "Booting OF in %d seconds  ", sec_left);
+        nwz_lcdmsg(false, 0, 3, "Press BACK to boot RB");
+        /* wait for a key (1s) */
+        int ret = nwz_key_wait_event(input_fd, 1000000);
+        if(ret != 1)
+            continue;
+        struct input_event evt;
+        if(nwz_key_read_event(input_fd, &evt) != 1)
+            continue;
+        if(nwz_key_event_get_keycode(&evt) == NWZ_KEY_BACK && !nwz_key_event_is_press(&evt))
+        {
+            boot_rb = true;
+            break;
+        }
+    }
+    nwz_key_close(input_fd);
+    return boot_rb;
+}
+
+static char *boot_rb_argv[] =
+{
+    "lcdmsg",
+    "-c",
+    "-l",
+    "0,3",
+    "Hello from RB",
+    NULL
+};
+
+static void wait_key(void)
+{
+    int input_fd = nwz_key_open();
+    /* display input state in a loop */
+    while(1)
+    {
+        /* wait for event (10ms) */
+        int ret = nwz_key_wait_event(input_fd, 10000);
+        if(ret != 1)
+            continue;
+        struct input_event evt;
+        if(nwz_key_read_event(input_fd, &evt) != 1)
+            continue;
+        if(nwz_key_event_get_keycode(&evt) == NWZ_KEY_BACK && !nwz_key_event_is_press(&evt))
+            break;
+    }
+    /* finish nicely */
+    nwz_key_close(input_fd);
+}
+
+int NWZ_TOOL_MAIN(all_tools)(int argc, char **argv);
+
+int main(int argc, char **argv)
+{
+#if 0
+    nwz_lcdmsg(true, 0, 0, "dualboot");
+    if(boot_rockbox())
+    {
+        /* boot rockox */
+        nwz_lcdmsg(true, 0, 3, "Booting rockbox...");
+        execvp("/usr/local/bin/lcdmsg", boot_rb_argv);
+        nwz_lcdmsg(false, 0, 4, "failed.");
+        sleep(5);
+    }
+    /* if for some reason, running rockbox failed, then try to run the OF */
+    /* boot OF */
+    nwz_lcdmsg(true, 0, 3, "Booting OF...");
+    execvp("/usr/local/bin/SpiderApp.of", argv);
+    nwz_lcdmsg(false, 0, 4, "failed.");
+    sleep(5);
+    /* if we reach this point, everything failed, so return an error so that
+     * sysmgrd knows something is wrong */
+    return 1;
+#elif 0
+    const char *args_mount[] = {"mount", NULL};
+    int status;
+    char *output = nwz_run_pipe("mount", args_mount, &status);
+    nwz_lcdmsgf(true, 0, 0, "%d\n%s", status, output);
+    free(output);
+    wait_key();
+    const char *args_ls[] = {"ls", "/var", NULL};
+    output = nwz_run_pipe("ls", args_ls, &status);
+    nwz_lcdmsgf(true, 0, 0, "%d\n%s", status, output);
+    free(output);
+    wait_key();
+    const char *args_glogctl[] = {"glogctl", "flush", NULL};
+    output = nwz_run_pipe("/usr/local/bin/glogctl", args_glogctl, &status);
+    nwz_lcdmsgf(true, 0, 0, "%d\n%s", status, output);
+    free(output);
+    wait_key();
+    system("cp /var/GEMINILOG* /contents/");
+    sync();
+    execvp("/usr/local/bin/SpiderApp.of", argv);
+    return 0;
+#else
+    /* make sure backlight is on */
+    int fb_fd = nwz_fb_open(true);
+    if(fb_fd >= 0)
+    {
+        struct nwz_fb_brightness bl;
+        nwz_fb_get_brightness(fb_fd, &bl);
+        bl.level = NWZ_FB_BL_MAX_LEVEL;
+        nwz_fb_set_brightness(fb_fd, &bl);
+        nwz_fb_close(fb_fd);
+    }
+    /* run all tools menu */
+    NWZ_TOOL_MAIN(all_tools)(argc, argv);
+    /* run OF */
+    execvp("/usr/local/bin/SpiderApp.of", argv);
+    return 0;
+#endif
+}
diff --git a/utils/nwztools/plattools/nwz_lib.c b/utils/nwztools/plattools/nwz_lib.c
index 4f49bec909..087db8834a 100644
--- a/utils/nwztools/plattools/nwz_lib.c
+++ b/utils/nwztools/plattools/nwz_lib.c
@@ -30,7 +30,7 @@ const char *nwz_get_model_name(unsigned long model_id)
     return NULL;
 }
 
-void nwz_run(const char *file, const char *args[], bool wait)
+int nwz_run(const char *file, const char *args[], bool wait)
 {
     pid_t child_pid = fork();
     if(child_pid != 0)
@@ -39,7 +39,10 @@ void nwz_run(const char *file, const char *args[], bool wait)
         {
             int status;
             waitpid(child_pid, &status, 0);
+            return status;
         }
+        else
+            return 0;
     }
     else
     {
@@ -48,6 +51,40 @@ void nwz_run(const char *file, const char *args[], bool wait)
     }
 }
 
+char *nwz_run_pipe(const char *file, const char *args[], int *status)
+{
+    int pipe_fds[2];
+    pipe(pipe_fds);
+    pid_t child_pid = fork();
+    if(child_pid == 0)
+    {
+        dup2(pipe_fds[1], 1); /* redirect stdout */
+        dup2(pipe_fds[1], 2); /* redirect stderr */
+        close(pipe_fds[0]); /* close reading */
+        close(pipe_fds[1]); /* close writing */
+        execvp(file, (char * const *)args);
+        _exit(1);
+    }
+    else
+    {
+        close(pipe_fds[1]); /* close writing */
+        char buffer[1024];
+        char *output = malloc(1);
+        ssize_t count;
+        size_t size = 0;
+        while((count = read(pipe_fds[0], buffer, sizeof(buffer))) > 0)
+        {
+            output = realloc(output, size + count + 1);
+            memcpy(output + size, buffer, count);
+            size += count;
+        }
+        close(pipe_fds[0]);
+        output[size] = 0;
+        waitpid(child_pid, status, 0);
+        return output;
+    }
+}
+
 void nwz_lcdmsg(bool clear, int x, int y, const char *msg)
 {
     const char *path_lcdmsg = "/usr/local/bin/lcdmsg";
diff --git a/utils/nwztools/plattools/nwz_lib.h b/utils/nwztools/plattools/nwz_lib.h
index 90d122003a..df0105a13d 100644
--- a/utils/nwztools/plattools/nwz_lib.h
+++ b/utils/nwztools/plattools/nwz_lib.h
@@ -30,6 +30,7 @@
 #include <linux/input.h>
 #include <fcntl.h>
 #include <string.h>
+#include <stdlib.h>
 
 #include "nwz_keys.h"
 #include "nwz_fb.h"
@@ -47,7 +48,9 @@ const char *nwz_get_model_name(unsigned long model_id);
 
 /* run a program and exit with nonzero status in case of error
  * argument list must be NULL terminated */
-void nwz_run(const char *file, const char *args[], bool wait);
+int nwz_run(const char *file, const char *args[], bool wait);
+/* run a program and return program output */
+char *nwz_run_pipe(const char *file, const char *args[], int *status);
 
 /* invoke /usr/bin/lcdmsg to display a message using the small font, optionally
  * clearing the screen before */