Add an fsync (Ted T'so would be pleased)

[mnemon] / mnemon.c

diff --git a/mnemon.c b/mnemon.c
index 4c5acd26287ce381e13a25efa86c81eb800536da..5ce736433eed2cff4c931dde86a39c490f8f15f0 100644 (file)
--- a/mnemon.c
+++ b/mnemon.c
@@ -26,6 +26,7 @@
 
 #include <sys/types.h>
 #include <sys/time.h>
+#include <unistd.h>
 #include <dirent.h>
 #include <errno.h>
 #include <string.h>
@@ -730,6 +731,7 @@ mnemon_save (mnemon_t *mnemon)
 
        category_print (category, file);
 
+       fsync (fileno (file));
        fclose (file);
 
        err = rename (lock_filename, filename);
@@ -1151,7 +1153,8 @@ mnemon_do_challenges (mnemon_t *mnemon)
 
        while (1) {
            if (category->time_limit > 0.0) {
-               readline ("The next one is timed. Press enter when ready:");
+               response = readline ("The next one is timed. Press enter when ready:");
+               free (response);
            }
                
            printf ("%s\n", item->challenge);
@@ -1166,10 +1169,12 @@ mnemon_do_challenges (mnemon_t *mnemon)
                return;
            }
 
-           if (response[0] == '/')
+           if (response[0] == '/') {
                mnemon_handle_command (mnemon, response + 1);
-           else
+               free (response);
+           } else {
                break;
+           }
        }
 
        mnemon_handle_response (mnemon, bin, item_index,
@@ -1177,17 +1182,17 @@ mnemon_do_challenges (mnemon_t *mnemon)
                                (end.tv_sec + end.tv_usec / 1e6) -
                                (start.tv_sec + start.tv_usec / 1e6),
                                category->time_limit);
+       free (response);
     } while (mnemon->to_introduce ||
             mnemon->unlearned ||
             mnemon->to_master > 0);
-
-    printf ("Great job.\n");
 }
 
 int
 main (int argc, char *argv[])
 {
     mnemon_t mnemon;
+    char *response;
 
     srand (time (NULL));
 
@@ -1201,5 +1206,15 @@ main (int argc, char *argv[])
 
     mnemon_fini (&mnemon);
 
+    mnemon_init (&mnemon);
+    mnemon_load (&mnemon);
+
+    printf ("Great job.\nHere are your current results:\n");
+    mnemon_print_histogram (&mnemon, NULL, 0);
+    response = readline ("Press enter to quit.\n");
+    free (response);
+
+    mnemon_fini (&mnemon);
+
     return 0;
 }