diff options
author | anzz1 <anzz1@live.com> | 2023-03-25 14:42:09 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-03-25 14:42:09 +0200 |
commit | e899bf54b291e8c84173a0e534a2c262f3f63229 (patch) | |
tree | bb63e9aec37ef4c713762de1efda6259bbdab81a | |
parent | fbd4d38c647f82b2598291ea9b8d0c09ac1ffb8c (diff) |
bounds checking for input prefix (#492)
-rw-r--r-- | utils.cpp | 6 |
1 files changed, 5 insertions, 1 deletions
@@ -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); |