From c3bdb02e53706cafceccb109f8a5ace9a37356f4 Mon Sep 17 00:00:00 2001
From: Hans Svensson <hans.svensson@quviq.com>
Date: Wed, 16 May 2018 10:46:39 +0200
Subject: [PATCH] Don't trust send message to succeed

---
 src/enoise.erl | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/enoise.erl b/src/enoise.erl
index 275bda2..16b36b4 100644
--- a/src/enoise.erl
+++ b/src/enoise.erl
@@ -192,8 +192,12 @@ do_handshake(HState, ComState, Timeout) ->
             end;
         out ->
             {ok, HState1, Msg} = enoise_hs_state:write_message(HState, <<>>),
-            {ok, ComState1} = hs_send_msg(ComState, Msg),
-            do_handshake(HState1, ComState1, Timeout);
+            case hs_send_msg(ComState, Msg) of
+                {ok, ComState1} ->
+                    do_handshake(HState1, ComState1, Timeout);
+                Err = {error, _} ->
+                    Err
+            end;
         done ->
             {ok, Res} = enoise_hs_state:finalize(HState),
             {ok, Res, ComState}