Let netcat support the use of service names instead of port numbers.

based on a diff from Andras Farkas <deepbluemistake@gmail.com>
ok deraadt@

1 files changed, 30 insertions, 14 deletions

diff --git a/src/usr.bin/nc/netcat.c b/src/usr.bin/nc/netcat.c
index 95f276bc89..cf8a87d883 100644
--- a/src/usr.bin/nc/netcat.c
+++ b/src/usr.bin/nc/netcat.c
@@ -1,4 +1,4 @@
-/* $OpenBSD: netcat.c,v 1.152 2016/05/28 20:14:58 beck Exp $ */
+/* $OpenBSD: netcat.c,v 1.153 2016/06/02 04:26:32 beck Exp $ */
 /*
  * Copyright (c) 2001 Eric Jackson <ericj@monkey.org>
  * Copyright (c) 2015 Bob Beck.  All rights reserved.
@@ -1283,6 +1283,27 @@ atelnet(int nfd, unsigned char *buf, unsigned int size)
         }
 }
 
+
+int
+strtoport(char *portstr, int udp)
+{
+        struct servent *entry;
+        const char *errstr;
+        char *proto;
+        int port = -1;
+
+        proto = udp ? "udp" : "tcp";
+
+        port = strtonum(portstr, 1, PORT_MAX, &errstr);
+        if (errstr == NULL)
+                return port;
+        if (errno != EINVAL)
+                errx(1, "port number %s: %s", errstr, portstr);
+        if ((entry = getservbyname(portstr, proto)) == NULL)
+                errx(1, "service \"%s\" unknown", portstr);
+        return ntohs(entry->s_port);
+}
+
 /*
  * build_ports()
  * Build an array of ports in portlist[], listing each port
@@ -1291,7 +1312,6 @@ atelnet(int nfd, unsigned char *buf, unsigned int size)
 void
 build_ports(char *p)
 {
-        const char *errstr;
         char *n;
         int hi, lo, cp;
         int x = 0;
@@ -1301,13 +1321,8 @@ build_ports(char *p)
                 n++;
 
                 /* Make sure the ports are in order: lowest->highest. */
-                hi = strtonum(n, 1, PORT_MAX, &errstr);
-                if (errstr)
-                        errx(1, "port number %s: %s", errstr, n);
-                lo = strtonum(p, 1, PORT_MAX, &errstr);
-                if (errstr)
-                        errx(1, "port number %s: %s", errstr, p);
-
+                hi = strtoport(n, uflag);
+                lo = strtoport(p, uflag);
                 if (lo > hi) {
                         cp = hi;
                         hi = lo;
@@ -1333,11 +1348,12 @@ build_ports(char *p)
                         }
                 }
         } else {
-                hi = strtonum(p, 1, PORT_MAX, &errstr);
-                if (errstr)
-                        errx(1, "port number %s: %s", errstr, p);
-                portlist[0] = strdup(p);
-                if (portlist[0] == NULL)
+                char *tmp;
+
+                hi = strtoport(p, uflag);
+                if (asprintf(&tmp, "%d", hi) != -1)
+                        portlist[0] = tmp;
+                else
                         err(1, NULL);
         }
 }