fix potential overflow

This commit is contained in:
Max Lv
2015-01-23 15:43:23 +08:00
parent 786360546a
commit 86b78dedd7
4 changed files with 15 additions and 7 deletions
+4 -2
View File
@@ -911,8 +911,10 @@ int main(int argc, char **argv)
}
break;
case 's':
remote_addr[remote_num].host = optarg;
remote_addr[remote_num++].port = NULL;
if (remote_num < MAX_REMOTE_NUM) {
remote_addr[remote_num].host = optarg;
remote_addr[remote_num++].port = NULL;
}
break;
case 'p':
remote_port = optarg;
+4 -2
View File
@@ -613,8 +613,10 @@ int main(int argc, char **argv)
while ((c = getopt(argc, argv, "f:s:p:l:k:t:m:c:b:a:")) != -1) {
switch (c) {
case 's':
remote_addr[remote_num].host = optarg;
remote_addr[remote_num++].port = NULL;
if (remote_num < MAX_REMOTE_NUM) {
remote_addr[remote_num].host = optarg;
remote_addr[remote_num++].port = NULL;
}
break;
case 'p':
remote_port = optarg;
+3 -1
View File
@@ -1043,7 +1043,9 @@ int main(int argc, char **argv)
}
break;
case 's':
server_host[server_num++] = optarg;
if (server_num < MAX_REMOTE_NUM) {
server_host[server_num++] = optarg;
}
break;
case 'p':
server_port = optarg;
+4 -2
View File
@@ -655,8 +655,10 @@ int main(int argc, char **argv)
while ((c = getopt(argc, argv, "f:s:p:l:k:t:m:i:c:b:L:a:uv")) != -1) {
switch (c) {
case 's':
remote_addr[remote_num].host = optarg;
remote_addr[remote_num++].port = NULL;
if (remote_num < MAX_REMOTE_NUM) {
remote_addr[remote_num].host = optarg;
remote_addr[remote_num++].port = NULL;
}
break;
case 'p':
remote_port = optarg;