3 files changed, 14 insertions, 11 deletions

diff --git a/src/lib/libtls/tls.c b/src/lib/libtls/tls.c
index 4536bae183..c79191ee15 100644
--- a/src/lib/libtls/tls.c
+++ b/src/lib/libtls/tls.c
@@ -1,4 +1,4 @@
-/* $OpenBSD: tls.c,v 1.13 2015/08/22 14:20:53 jsing Exp $ */
+/* $OpenBSD: tls.c,v 1.14 2015/08/27 14:34:46 jsing Exp $ */
 /*
  * Copyright (c) 2014 Joel Sing <jsing@openbsd.org>
  *
@@ -233,6 +233,7 @@ tls_reset(struct tls *ctx)
         ctx->ssl_ctx = NULL;
 
         ctx->socket = -1;
+        ctx->state = 0;
 
         ctx->err = 0;
         free(ctx->errmsg);
diff --git a/src/lib/libtls/tls_client.c b/src/lib/libtls/tls_client.c
index 442ba4321e..241c506676 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.19 2015/08/22 14:51:34 jsing Exp $ */
+/* $OpenBSD: tls_client.c,v 1.20 2015/08/27 14:34:46 jsing Exp $ */
 /*
  * Copyright (c) 2014 Joel Sing <jsing@openbsd.org>
  *
@@ -168,14 +168,14 @@ tls_connect_fds(struct tls *ctx, int fd_read, int fd_write,
         X509 *cert = NULL;
         int ret, err;
 
-        if (ctx->flags & TLS_CONNECTING)
-                goto connecting;
-
         if ((ctx->flags & TLS_CLIENT) == 0) {
                 tls_set_error(ctx, "not a client context");
                 goto err;
         }
 
+        if (ctx->state & TLS_STATE_CONNECTING)
+                goto connecting;
+
         if (fd_read < 0 || fd_write < 0) {
                 tls_set_error(ctx, "invalid file descriptors");
                 return (-1);
@@ -248,16 +248,16 @@ tls_connect_fds(struct tls *ctx, int fd_read, int fd_write,
                 }
         }
 
- connecting:
+connecting:
         if ((ret = SSL_connect(ctx->ssl_conn)) != 1) {
                 err = tls_ssl_error(ctx, ctx->ssl_conn, ret, "connect");
                 if (err == TLS_READ_AGAIN || err == TLS_WRITE_AGAIN) {
-                        ctx->flags |= TLS_CONNECTING;
+                        ctx->state |= TLS_STATE_CONNECTING;
                         return (err);
                 }
                 goto err;
         }
-        ctx->flags &= ~TLS_CONNECTING;
+        ctx->state &= ~TLS_STATE_CONNECTING;
 
         if (ctx->config->verify_name) {
                 cert = SSL_get_peer_certificate(ctx->ssl_conn);
diff --git a/src/lib/libtls/tls_internal.h b/src/lib/libtls/tls_internal.h
index ba37e136e6..cf4a8e28ad 100644
--- a/src/lib/libtls/tls_internal.h
+++ b/src/lib/libtls/tls_internal.h
@@ -1,4 +1,4 @@
-/* $OpenBSD: tls_internal.h,v 1.12 2015/03/31 12:21:27 jsing Exp $ */
+/* $OpenBSD: tls_internal.h,v 1.13 2015/08/27 14:34:46 jsing Exp $ */
 /*
  * Copyright (c) 2014 Jeremie Courreges-Anglas <jca@openbsd.org>
  * Copyright (c) 2014 Joel Sing <jsing@openbsd.org>
@@ -51,11 +51,13 @@ struct tls_config {
 #define TLS_CLIENT              (1 << 0)
 #define TLS_SERVER              (1 << 1)
 #define TLS_SERVER_CONN         (1 << 2)
-#define TLS_CONNECTING          (1 << 3)
+
+#define TLS_STATE_CONNECTING    (1 << 0)
 
 struct tls {
         struct tls_config *config;
-        uint64_t flags;
+        uint32_t flags;
+        uint32_t state;
 
         int err;
         char *errmsg;