add -spec for dialyzer

siu5 · Nov 10, 2014 · 6bab770 · 6bab770
1 parent 6b2e09e
commit 6bab770
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 0 deletions.
diff --git a/client/src/mpc_child.erl b/client/src/mpc_child.erl
@@ -176,6 +176,8 @@ code_change(_OldVsn, State, _Extra) ->
 %%% Internal functions
 %%%===================================================================
 
+-spec start_process(port(), nonempty_string()) -> {ok, port()} |
+                                                  {error, any()}.
 start_process(Socket, Key) ->
     {ok, RemoteAddr} = application:get_env(make_proxy_client, remote_addr),
     {ok, RemotePort} = application:get_env(make_proxy_client, remote_port),
@@ -195,6 +197,7 @@ start_process(Socket, Key) ->
     end.
 
 
+-spec find_target(port()) -> {ok, <<_:32, _:_*8>>}.
 find_target(Socket) ->
     {ok, <<5:8, Nmethods:8>>} = gen_tcp:recv(Socket, 2),
     {ok, _Methods} = gen_tcp:recv(Socket, Nmethods),

diff --git a/common/mp_crypto.erl b/common/mp_crypto.erl
@@ -5,6 +5,7 @@
 -define(DATALENGTH, 16).
 -define(IV, <<"90de3456asxdfrtg">>).
 
+-spec encrypt(nonempty_string(), binary()) -> <<_:_*8>>.
 encrypt(Key, Binary) ->
     BinaryLength = byte_size(Binary),
     Rem = (BinaryLength + 4) rem ?DATALENGTH,
@@ -15,6 +16,8 @@ encrypt(Key, Binary) ->
     crypto:block_encrypt(aes_cbc128, Key, ?IV, FinalBinary).
 
 
+-spec decrypt(nonempty_string(), <<_:_*8>>) -> {ok, binary()} |
+                                               {error, term()}.
 decrypt(Key, Binary) ->
     Data = crypto:block_decrypt(aes_cbc128, Key, ?IV, Binary),
     try

diff --git a/server/src/mp_child.erl b/server/src/mp_child.erl
@@ -180,6 +180,8 @@ code_change(_OldVsn, State, _Extra) ->
 %%% Internal functions
 %%%===================================================================
 
+-spec connect_to_remote(port(), nonempty_string()) -> {ok, {list()|tuple(), non_neg_integer()}} |
+                                                      {error, term()}.
 connect_to_remote(Socket, Key) ->
     {ok, EntrypedData} = gen_tcp:recv(Socket, 0),
 
@@ -193,6 +195,7 @@ connect_to_remote(Socket, Key) ->
     end.
 
 
+-spec parse_address(non_neg_integer(), binary()) -> {ok, {list()|tuple(), non_neg_integer()}}.
 parse_address(?IPV4, Data) ->
     <<Port:16, Destination/binary>> = Data,
     Address = list_to_tuple( binary_to_list(Destination) ),
@@ -210,6 +213,8 @@ parse_address(?DOMAIN, Data) ->
     {ok, {Address, Port}}.
 
 
+-spec connect_target(list() | tuple(), non_neg_integer()) -> {ok, port()} |
+                                                             {error, term()}.
 connect_target(Address, Port) ->
     connect_target(Address, Port, 2).