- markus@cvs.openbsd.org 2001/04/11 10:59:01

     [ssh.c]
     use strtol() for ports, thanks jakob@

1 files changed, 12 insertions, 4 deletions

diff --git a/ssh.c b/ssh.c
index 75094a106..89ced97e2 100644
--- a/ssh.c
+++ b/ssh.c
@@ -39,7 +39,7 @@
  */
 
 #include "includes.h"
-RCSID("$OpenBSD: ssh.c,v 1.108 2001/04/07 08:55:18 markus Exp $");
+RCSID("$OpenBSD: ssh.c,v 1.109 2001/04/11 10:59:01 markus Exp $");
 
 #include <openssl/evp.h>
 #include <openssl/err.h>
@@ -243,7 +243,7 @@ main(int ac, char **av)
 {
         int i, opt, optind, exit_status, ok;
         u_short fwd_port, fwd_host_port;
-        char *optarg, *cp, buf[256];
+        char *optarg, *cp, *endofnumber, buf[256];
         struct stat st;
         struct passwd *pw;
         int dummy;
@@ -456,7 +456,11 @@ main(int ac, char **av)
                         }
                         break;
                 case 'p':
-                        options.port = atoi(optarg);
+                        options.port = strtol(optarg, &endofnumber, 0);
+                        if (optarg == endofnumber) {
+                                fprintf(stderr, "Bad port '%s'\n", optarg);
+                                exit(1);
+                        }
                         break;
                 case 'l':
                         options.user = optarg;
@@ -485,7 +489,11 @@ main(int ac, char **av)
                         break;
 
                 case 'D':
-                        fwd_port = atoi(optarg);
+                        fwd_port = strtol(optarg, &endofnumber, 0);
+                        if (optarg == endofnumber) {
+                                fprintf(stderr, "Bad port '%s'\n", optarg);
+                                exit(1);
+                        }
                         add_local_forward(&options, fwd_port, "socks4", 0);
                         break;