X-Git-Url: https://diplodocus.org/git/nmh/blobdiff_plain/35ef9d158396dbf013a862dd8f830db64909826b..23816efd3e8e294e8f1c1ce811ac3a1cd15d334e:/sbr/client.c

diff --git a/sbr/client.c b/sbr/client.c
index a64461bc..8d390a81 100644
--- a/sbr/client.c
+++ b/sbr/client.c
@@ -5,9 +5,11 @@
  * complete copyright information.
  */
 
-#include <h/mh.h>
-#include <h/mts.h>
-#include <h/utils.h>
+#include "h/mh.h"
+#include "client.h"
+#include "h/mts.h"
+#include "h/utils.h"
+#include <fcntl.h>
 #include <sys/socket.h>
 #include <netinet/in.h>
 #include <netdb.h>
@@ -70,7 +72,15 @@ client (char *server, char *service, char *response, int len_response,
 	}
 
 	if (connect(sd, ai->ai_addr, ai->ai_addrlen) == 0) {
+	    int flags;
+
 	    freeaddrinfo(res);
+
+	    flags = fcntl(sd, F_GETFD, 0);
+
+	    if (flags != -1)
+		fcntl(sd, F_SETFD, flags | FD_CLOEXEC);
+
 	    return sd;
 	}
 
@@ -92,10 +102,19 @@ client (char *server, char *service, char *response, int len_response,
 
     if (res->ai_next)
 	snprintf(response, len_response, "no servers available (use -snoop "
-		 "for details");
-    else
-	snprintf(response, len_response, "Connection to \"%s\" failed: %s",
-		 server, strerror(errno));
+		 "for details)");
+    else {
+	char port[NI_MAXSERV];
+
+	if (getnameinfo(res->ai_addr, res->ai_addrlen, NULL, 0, port,
+			sizeof port, NI_NUMERICSERV)) {
+	    strncpy(port, "unknown", sizeof port);
+	    port[sizeof(port) - 1] = '\0';
+	}
+
+	snprintf(response, len_response, "Connection to \"%s:%s\" failed: %s",
+		 server, port, strerror(errno));
+    }
 
     freeaddrinfo(res);