proxy_protocol.erl 646 B

12345678910111213141516171819202122232425
  1. -module(proxy_protocol).
  2. -behaviour(ranch_protocol).
  3. -export([start_link/4]).
  4. -export([init/3]).
  5. start_link(Ref, _Socket, 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. {ok, ProxyInfo} = Transport:recv_proxy_header(Socket, 1000),
  11. Pid = ct_helper:get_remote_pid_tcp(Socket),
  12. Pid ! {?MODULE, ProxyInfo},
  13. loop(Socket, Transport).
  14. loop(Socket, Transport) ->
  15. case Transport:recv(Socket, 0, 5000) of
  16. {ok, Data} ->
  17. _ = Transport:send(Socket, Data),
  18. loop(Socket, Transport);
  19. _ ->
  20. ok = Transport:close(Socket)
  21. end.