Introduce ranch_transport:peercert/1 callback

ninenines · Jan 12, 2022 · c79afa5 · c79afa5
1 parent 0906ab7
commit c79afa5
Show file tree

Hide file tree

Showing 4 changed files with 18 additions and 0 deletions.
diff --git a/src/ranch_ssl.erl b/src/ranch_ssl.erl
@@ -43,6 +43,7 @@
 -export([controlling_process/2]).
 -export([peername/1]).
 -export([sockname/1]).
+-export([peercert/1]).
 -export([shutdown/2]).
 -export([close/1]).
 -export([cleanup/1]).
@@ -296,6 +297,11 @@ peername(Socket) ->
 sockname(Socket) ->
 	ssl:sockname(Socket).
 
+-spec peercert(ssl:sslsocket())
+	-> {ok, public_key:der_encoded()} | {error, any()}.
+peercert(Socket) ->
+	ssl:peercert(Socket).
+
 -spec shutdown(ssl:sslsocket(), read | write | read_write)
 	-> ok | {error, atom()}.
 shutdown(Socket, How) ->

diff --git a/src/ranch_tcp.erl b/src/ranch_tcp.erl
@@ -42,6 +42,7 @@
 -export([controlling_process/2]).
 -export([peername/1]).
 -export([sockname/1]).
+-export([peercert/1]).
 -export([shutdown/2]).
 -export([close/1]).
 -export([cleanup/1]).
@@ -265,6 +266,10 @@ peername(Socket) ->
 sockname(Socket) ->
 	inet:sockname(Socket).
 
+-spec peercert(inet:socket()) -> no_return().
+peercert(_Socket) ->
+	error(not_supported).
+
 -spec shutdown(inet:socket(), read | write | read_write)
 	-> ok | {error, atom()}.
 shutdown(Socket, How) ->

diff --git a/src/ranch_transport.erl b/src/ranch_transport.erl
@@ -64,6 +64,7 @@
 	-> {ok, {inet:ip_address(), inet:port_number()} | {local, binary()}} | {error, atom()}.
 -callback sockname(socket())
 	-> {ok, {inet:ip_address(), inet:port_number()} | {local, binary()}} | {error, atom()}.
+-callback peercert(socket()) -> {ok, public_key:der_encoded()} | {error, any()}.
 -callback shutdown(socket(), read | write | read_write)
 	-> ok | {error, atom()}.
 -callback close(socket()) -> ok.

diff --git a/test/ranch_erlang_transport.erl b/test/ranch_erlang_transport.erl
@@ -40,6 +40,7 @@
 -export([controlling_process/2]).
 -export([peername/1]).
 -export([sockname/1]).
+-export([peercert/1]).
 -export([shutdown/2]).
 -export([close/1]).
 -export([cleanup/1]).
@@ -160,6 +161,11 @@ peername(_Socket) ->
 sockname(_Socket) ->
 	{ok, {{127, 0, 0, 1}, 12710}}.
 
+-spec peercert(reference())
+	-> no_return().
+peercert(_Socket) ->
+	error(not_supported).
+
 -spec shutdown(reference(), read | write | read_write)
 	-> ok | {error, atom()}.
 shutdown(_Socket, _How) ->