diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/redis-benchmark.c | 8 | ||||
-rw-r--r-- | src/redis-cli.c | 8 |
2 files changed, 16 insertions, 0 deletions
diff --git a/src/redis-benchmark.c b/src/redis-benchmark.c index 2e0362b07..d727c4b63 100644 --- a/src/redis-benchmark.c +++ b/src/redis-benchmark.c @@ -1446,6 +1446,10 @@ int parseOptions(int argc, char **argv) { } else if (!strcmp(argv[i],"-p")) { if (lastarg) goto invalid; config.conn_info.hostport = atoi(argv[++i]); + if (config.conn_info.hostport < 0 || config.conn_info.hostport > 65535) { + fprintf(stderr, "Invalid server port.\n"); + exit(1); + } } else if (!strcmp(argv[i],"-s")) { if (lastarg) goto invalid; config.hostsocket = strdup(argv[++i]); @@ -1459,6 +1463,10 @@ int parseOptions(int argc, char **argv) { config.conn_info.user = sdsnew(argv[++i]); } else if (!strcmp(argv[i],"-u") && !lastarg) { parseRedisUri(argv[++i],"redis-benchmark",&config.conn_info,&config.tls); + if (config.conn_info.hostport < 0 || config.conn_info.hostport > 65535) { + fprintf(stderr, "Invalid server port.\n"); + exit(1); + } config.input_dbnumstr = sdsfromlonglong(config.conn_info.input_dbnum); } else if (!strcmp(argv[i],"-3")) { config.resp3 = 1; diff --git a/src/redis-cli.c b/src/redis-cli.c index 0de78c67e..a1a258f46 100644 --- a/src/redis-cli.c +++ b/src/redis-cli.c @@ -1961,6 +1961,10 @@ static int parseOptions(int argc, char **argv) { config.stdin_tag_name = argv[++i]; } else if (!strcmp(argv[i],"-p") && !lastarg) { config.conn_info.hostport = atoi(argv[++i]); + if (config.conn_info.hostport < 0 || config.conn_info.hostport > 65535) { + fprintf(stderr, "Invalid server port.\n"); + exit(1); + } } else if (!strcmp(argv[i],"-s") && !lastarg) { config.hostsocket = argv[++i]; } else if (!strcmp(argv[i],"-r") && !lastarg) { @@ -1982,6 +1986,10 @@ static int parseOptions(int argc, char **argv) { config.conn_info.user = sdsnew(argv[++i]); } else if (!strcmp(argv[i],"-u") && !lastarg) { parseRedisUri(argv[++i],"redis-cli",&config.conn_info,&config.tls); + if (config.conn_info.hostport < 0 || config.conn_info.hostport > 65535) { + fprintf(stderr, "Invalid server port.\n"); + exit(1); + } } else if (!strcmp(argv[i],"--raw")) { config.output = OUTPUT_RAW; } else if (!strcmp(argv[i],"--no-raw")) { |