Fixed argument checking on some argument driven actions.

1 files changed, 6 insertions, 2 deletions

diff --git a/main.c b/main.c
index 8817d6d..8e6cfbc 100644
--- a/main.c
+++ b/main.c
@@ -1,6 +1,8 @@
 // Compilation command:
 // - release: gcc main.c -Wall -Werror -pedantic -O2 -m64 -lncursesw -o ttt
 // - debug  : gcc main.c -Wall -Werror -pedantic -g3 -m64 -lncursesw -o ttt -D DEBUG
+// Usage hints:
+// - To changes app data path change the environment variable HOME (USERPROFILE for windows users).
 
 
 #include <assert.h>
@@ -996,8 +998,9 @@ int main(int argc, char *argv[]) {
 			action = "--icsv";
 			do_action = strncmp(argv[idx], action, strlen(action)+1) == 0;
 			if (do_action) {
-				if (argc < idx+1) {
+				if (idx+1 >= argc) {
 					fprintf(stdout, "Missing CSV file path to import.\n");
+					return EXIT_FAILURE;
 				}
 				load_database(&database, db_file_path);
 				import_from_csv(&database, argv[idx+1]);
@@ -1009,8 +1012,9 @@ int main(int argc, char *argv[]) {
 			action = "--ecsv";
 			do_action = strncmp(argv[idx], action, strlen(action)+1) == 0;
 			if (do_action) {
-				if (argc < idx+1) {
+				if (idx+1 >= argc) {
 					fprintf(stdout, "Missing CSV file path to export.\n");
+					return EXIT_FAILURE;
 				}
 				load_database(&database, db_file_path);
 				export_to_csv(&database, argv[idx+1]);