ssl_upgrade_protocol.erl 821 B

12345678910111213141516171819202122232425262728
  1. -module(ssl_upgrade_protocol).
  2. -behaviour(ranch_protocol).
  3. -export([start_link/3]).
  4. -export([init/3]).
  5. start_link(Ref, Transport, Opts) ->
  6. Pid = spawn_link(?MODULE, init, [Ref, Transport, Opts]),
  7. {ok, Pid}.
  8. init(Ref, Transport, _Opts = []) ->
  9. {ok, Socket} = ranch:handshake(Ref),
  10. loop(Socket, Transport).
  11. loop(Socket, Transport) ->
  12. case Transport:recv(Socket, 0, 5000) of
  13. {ok, <<"UPGRADE">>} when Transport =:= ranch_tcp ->
  14. ok = Transport:send(Socket, <<"READY">>),
  15. Opts = ct_helper:get_certs_from_ets(),
  16. {ok, NewSocket} = ranch_ssl:handshake(Socket,
  17. [{fail_if_no_peer_cert, false}, {verify, verify_none}|Opts], 1000),
  18. loop(NewSocket, ranch_ssl);
  19. {ok, <<"ECHO ", More/binary>>} ->
  20. ok = Transport:send(Socket, More),
  21. loop(Socket, Transport);
  22. _ ->
  23. ok = Transport:close(Socket)
  24. end.