فهرست منبع

bounds checking for input prefix (#492)

anzz1 2 سال پیش
والد
کامیت
e899bf54b2
1فایلهای تغییر یافته به همراه5 افزوده شده و 1 حذف شده
  1. 5 1
      utils.cpp

+ 5 - 1
utils.cpp

@@ -156,7 +156,11 @@ bool gpt_params_parse(int argc, char ** argv, gpt_params & params) {
         } else if (arg == "--random-prompt") {
         } else if (arg == "--random-prompt") {
             params.random_prompt = true;
             params.random_prompt = true;
         } else if (arg == "--in-prefix") {
         } else if (arg == "--in-prefix") {
-            params.input_prefix = argv[++i];
+            if (++i >= argc) {
+                invalid_param = true;
+                break;
+            }
+            params.input_prefix = argv[i];
         } else {
         } else {
             fprintf(stderr, "error: unknown argument: %s\n", arg.c_str());
             fprintf(stderr, "error: unknown argument: %s\n", arg.c_str());
             gpt_print_usage(argc, argv, params);
             gpt_print_usage(argc, argv, params);