From: anzz1 <redacted>
Date: Sat, 25 Mar 2023 12:42:09 +0000 (+0200)
Subject: bounds checking for input prefix (#492)
X-Git-Tag: gguf-v0.4.0~1115
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=e899bf54b291e8c84173a0e534a2c262f3f63229;p=pkg%2Fggml%2Fsources%2Fllama.cpp

bounds checking for input prefix (#492)
---

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