Only take ownership of a socket if we allocated it within libtls. If we are

passed a socket then the caller is responsible for closing it.

ok bcook@

2 files changed, 5 insertions, 11 deletions

diff --git a/src/lib/libtls/tls_client.c b/src/lib/libtls/tls_client.c
index 55fb49ff78..81b5510431 100644
--- a/src/lib/libtls/tls_client.c
+++ b/src/lib/libtls/tls_client.c
@@ -1,4 +1,4 @@
-/* $OpenBSD: tls_client.c,v 1.22 2015/09/08 15:29:34 jsing Exp $ */
+/* $OpenBSD: tls_client.c,v 1.23 2015/09/09 14:32:06 jsing Exp $ */
 /*
  * Copyright (c) 2014 Joel Sing <jsing@openbsd.org>
  *
@@ -145,6 +145,8 @@ tls_connect_servername(struct tls *ctx, const char *host, const char *port,
                 goto err;
         }
 
+        ctx->socket = s;
+
         rv = 0;
 
 err:
@@ -157,8 +159,6 @@ err:
 int
 tls_connect_socket(struct tls *ctx, int s, const char *servername)
 {
-        ctx->socket = s;
-
         return tls_connect_fds(ctx, s, s, servername);
 }
 
diff --git a/src/lib/libtls/tls_server.c b/src/lib/libtls/tls_server.c
index bb29c7ce42..190682e630 100644
--- a/src/lib/libtls/tls_server.c
+++ b/src/lib/libtls/tls_server.c
@@ -1,4 +1,4 @@
-/* $OpenBSD: tls_server.c,v 1.10 2015/08/27 15:26:50 jsing Exp $ */
+/* $OpenBSD: tls_server.c,v 1.11 2015/09/09 14:32:06 jsing Exp $ */
 /*
  * Copyright (c) 2014 Joel Sing <jsing@openbsd.org>
  *
@@ -148,11 +148,5 @@ err:
 int
 tls_accept_socket(struct tls *ctx, struct tls **cctx, int socket)
 {
-        int rv;
-
-        rv = tls_accept_fds(ctx, cctx, socket, socket);
-        if (*cctx != NULL)
-                (*cctx)->socket = socket;
-
-        return (rv);
+        return (tls_accept_fds(ctx, cctx, socket, socket));
 }