- fixes parallel builds (make -j)

[oweals/busybox.git] / libbb / xconnect.c

diff --git a/libbb/xconnect.c b/libbb/xconnect.c
index b3619fd0ee43cb312bcfbc38b0823346da319bf6..ec99c58829db8c980a665898037fa1c2e56475b9 100644 (file)
--- a/libbb/xconnect.c
+++ b/libbb/xconnect.c
@@ -2,7 +2,7 @@
 /*
  * Utility routines.
  *
- * Connect to host at port using address resolusion from getaddrinfo
+ * Connect to host at port using address resolution from getaddrinfo
  *
  */
 
@@ -23,21 +23,28 @@
  * If "port" is a name it is looked up in /etc/services, if it isnt found return
  * default_port
  */
-unsigned short bb_lookup_port(const char *port, unsigned short default_port)
+unsigned short bb_lookup_port(const char *port, const char *protocol, unsigned short default_port)
 {
        unsigned short port_nr = htons(default_port);
        if (port) {
-       char *endptr;
-               long port_long = strtol(port, &endptr, 10);
+               char *endptr;
+               int old_errno;
+               long port_long;
 
+               /* Since this is a lib function, we're not allowed to reset errno to 0.
+                * Doing so could break an app that is deferring checking of errno. */
+               old_errno = errno;
+               errno = 0;
+               port_long = strtol(port, &endptr, 10);
                if (errno != 0 || *endptr!='\0' || endptr==port || port_long < 0 || port_long > 65535) {
-                       struct servent *tserv = getservbyname(port, "tcp");
+                       struct servent *tserv = getservbyname(port, protocol);
                        if (tserv) {
-                       port_nr = tserv->s_port;
-               }
-       } else {
+                               port_nr = tserv->s_port;
+                       }
+               } else {
                        port_nr = htons(port_long);
                }
+               errno = old_errno;
        }
        return port_nr;
 }
@@ -55,9 +62,10 @@ void bb_lookup_host(struct sockaddr_in *s_in, const char *host)
 int xconnect(struct sockaddr_in *s_addr)
 {
        int s = socket(AF_INET, SOCK_STREAM, 0);
-       if (connect(s, (struct sockaddr_in *)s_addr, sizeof(struct sockaddr_in)) < 0)
+       if (connect(s, (struct sockaddr *)s_addr, sizeof(struct sockaddr_in)) < 0)
        {
-               bb_perror_msg_and_die("Unable to connect to remote host (%s)", 
+               if (ENABLE_FEATURE_CLEAN_UP) close(s);
+               bb_perror_msg_and_die("Unable to connect to remote host (%s)",
                                inet_ntoa(s_addr->sin_addr));
        }
        return s;