X-Git-Url: https://git.rohieb.name/hackover2013-badge-firmware.git/blobdiff_plain/c672ffeb5b298b4b8580e24916da2c80b2518ac2..5830bb11cb11ca00fa53b04413d0587bac89b70f:/badge/jumpnrun/starter.c?ds=sidebyside

diff --git a/badge/jumpnrun/starter.c b/badge/jumpnrun/starter.c
index 4deb635..93e769c 100644
--- a/badge/jumpnrun/starter.c
+++ b/badge/jumpnrun/starter.c
@@ -48,9 +48,9 @@ static uint8_t jumpnrun_pick_level_from_fd(char *buf, size_t *first_visible, siz
     return JUMPNRUN_ERROR;
   }
 
-  char menu_buf[levelcount][MENU_BUFLEN];
-  char const *menu_index[levelcount];
-  char const *fnames[levelcount];
+  char menu_buf[levelcount + 1][MENU_BUFLEN];
+  char const *menu_index[levelcount + 1];
+  char const *fnames[levelcount + 1];
   unsigned i;
 
   for(i = 0; i < levelcount && f_gets(menu_buf[i], MENU_BUFLEN, fd); ++i) {
@@ -64,8 +64,15 @@ static uint8_t jumpnrun_pick_level_from_fd(char *buf, size_t *first_visible, siz
     fnames[i] = p;
   }
 
-  *selected = badge_menu(menu_index, i, first_visible, *selected);
+  strcpy(menu_buf[i], "exit");
+  menu_index[i] = menu_buf[i];
+  size_t choice = badge_menu(menu_index, i + 1, first_visible, *selected);
 
+  if(choice == levelcount) {
+    return 1; // exit
+  }
+
+  *selected = choice;
   strcpy(buf, fnames[*selected]);
   return 0;
 }
@@ -92,7 +99,7 @@ void jumpnrun_play(void) {
 
   while(0 == jumpnrun_pick_level(buf, &first_visible, &selected, progress)) {
     if(JUMPNRUN_WON == jumpnrun_play_level(buf) && selected == progress) {
-      ++progress;
+      selected = ++progress;
       jumpnrun_save_progress(progress);
     }
   }