Browse Source

fix potential overflow

pull/237/head
Max Lv 10 years ago
parent
commit
86b78dedd7
4 changed files with 15 additions and 7 deletions
  1. 6
      src/local.c
  2. 6
      src/redir.c
  3. 4
      src/server.c
  4. 6
      src/tunnel.c

6
src/local.c

@ -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;

6
src/redir.c

@ -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;

4
src/server.c

@ -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;

6
src/tunnel.c

@ -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;

Loading…
Cancel
Save