elli_tcp: cast 'accepted' AFTER TLS handshake succeeded

fix #84
elli-lib · Sep 26, 2020 · c919553 · c919553
1 parent 968afee
commit c919553
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/src/elli_tcp.erl b/src/elli_tcp.erl
@@ -37,26 +37,27 @@ accept({plain, Socket}, Server, Timeout) ->
 accept({ssl, Socket}, Server, Timeout) ->
     case ssl:transport_accept(Socket, Timeout) of
         {ok, S} ->
-            gen_server:cast(Server, accepted),
-            handshake(S, Timeout);
+            handshake(S, Server, Timeout);
         {error, Reason} ->
             {error, Reason}
     end.
 
 -ifdef(post20).
-handshake(S, Timeout) ->
+handshake(S, Server, Timeout) ->
     case ssl:handshake(S, Timeout) of
         {ok, S1} ->
+            gen_server:cast(Server, accepted),
             {ok, {ssl, S1}};
         {error, closed} ->
             {error, econnaborted};
         {error, Reason} ->
             {error, Reason}
     end.
 -else.
-handshake(S, Timeout) ->
+handshake(S, Server, Timeout) ->
     case ssl:ssl_accept(S, Timeout) of
         ok ->
+            gen_server:cast(Server, accepted),
             {ok, {ssl, S}};
         {error, closed} ->
             {error, econnaborted};