From 1e6ee6703fc401a46fce8f1a400a68734c0a356d Mon Sep 17 00:00:00 2001
From: Hans Svensson <hans.svensson@quviq.com>
Date: Tue, 29 Jan 2019 09:08:17 +0100
Subject: [PATCH] More improved typespecs

---
 src/enoise_connection.erl | 2 +-
 src/enoise_hs_state.erl   | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/enoise_connection.erl b/src/enoise_connection.erl
index 215a937..53c12dd 100644
--- a/src/enoise_connection.erl
+++ b/src/enoise_connection.erl
@@ -57,7 +57,7 @@ start_link(TcpSock, Rx, Tx, Owner, {Active0, Buf}) ->
 send(Noise, Data) ->
     gen_server:call(Noise, {send, Data}).
 
--spec set_active(Noise :: pid(), Active :: boolean()) -> ok | {error, term()}.
+-spec set_active(Noise :: pid(), Active :: true | once) -> ok | {error, term()}.
 set_active(Noise, Active) ->
     gen_server:call(Noise, {active, self(), Active}).
 
diff --git a/src/enoise_hs_state.erl b/src/enoise_hs_state.erl
index dff6f2d..301a936 100644
--- a/src/enoise_hs_state.erl
+++ b/src/enoise_hs_state.erl
@@ -90,6 +90,7 @@ read_message(HS = #noise_hs{ msgs = [{in, Msg} | Msgs] }, Message) ->
         Err = {error, _} -> Err
     end.
 
+-spec remote_keys(HS :: state()) -> keypair().
 remote_keys(#noise_hs{ rs = RS }) ->
     RS.