Validate server port argument

2010-02-04 17:17:19 +00:00
parent 1a5749424b
commit f47b84b57f
4 changed files with 50 additions and 5 deletions
--- a/tests/server/sws.c
+++ b/tests/server/sws.c
@@ -1082,7 +1082,16 @@ int main(int argc, char *argv[])
    else if(!strcmp("--port", argv[arg])) {
      arg++;
      if(argc>arg) {
-        port = (unsigned short)atoi(argv[arg]);
+        char *endptr;
+        long lnum = -1;
+        lnum = strtol(argv[arg], &endptr, 10);
+        if((endptr != argv[arg] + strlen(argv[arg])) ||
+           (lnum < 1025L) || (lnum > 65535L)) {
+          fprintf(stderr, "sws: invalid --port argument (%s)\n",
+                  argv[arg]);
+          return 0;
+        }
+        port = (unsigned short)(lnum & 0xFFFFL);
        arg++;
      }
    }