diff options
author | Loïc Hoguin <[email protected]> | 2013-12-07 13:53:00 +0100 |
---|---|---|
committer | Loïc Hoguin <[email protected]> | 2013-12-07 13:53:00 +0100 |
commit | b0c37e853ef0600ca5b86216e427d767b6bc6126 (patch) | |
tree | f25ccf1af5d97df4665aa7bc3477592f7f59e3f2 /examples/tcp_reverse/src/reverse_protocol.erl | |
parent | 7194df4568e66c1f2fee86816ace3308ec9eb302 (diff) | |
parent | 4e5e04f66869d2a34b8d32232e1bbb0e813befba (diff) | |
download | ranch-b0c37e853ef0600ca5b86216e427d767b6bc6126.tar.gz ranch-b0c37e853ef0600ca5b86216e427d767b6bc6126.tar.bz2 ranch-b0c37e853ef0600ca5b86216e427d767b6bc6126.zip |
Merge branch 'tcp_echo_example_updates' of git://github.com/joshrotenberg/ranch
Diffstat (limited to 'examples/tcp_reverse/src/reverse_protocol.erl')
-rw-r--r-- | examples/tcp_reverse/src/reverse_protocol.erl | 61 |
1 files changed, 61 insertions, 0 deletions
diff --git a/examples/tcp_reverse/src/reverse_protocol.erl b/examples/tcp_reverse/src/reverse_protocol.erl new file mode 100644 index 0000000..ea28d4e --- /dev/null +++ b/examples/tcp_reverse/src/reverse_protocol.erl @@ -0,0 +1,61 @@ +%% Feel free to use, reuse and abuse the code in this file. + +-module(reverse_protocol). + +-behaviour(gen_server). +-behaviour(ranch_protocol). + +-export([start_link/4]). + +-export([init/1, init/4, handle_call/3, handle_cast/2, handle_info/2, + terminate/2, code_change/3]). + +-export([reverse_binary/1]). + +-define(TIMEOUT, 5000). + +-record(state, {socket, transport}). + +reverse_binary(B) when is_binary(B) -> + [list_to_binary( + lists:reverse( + binary_to_list(binary:part(B, {0, byte_size(B)-2})))), + "\r\n"]. + +start_link(Ref, Socket, Transport, Opts) -> + proc_lib:start_link(?MODULE, init, [Ref, Socket, Transport, Opts]). + +init(Args) -> + {ok, Args}. + +init(Ref, Socket, Transport, _Opts = []) -> + ok = proc_lib:init_ack({ok, self()}), + ok = ranch:accept_ack(Ref), + ok = Transport:setopts(Socket, [{active, once}]), + gen_server:enter_loop(?MODULE,[], {state, Socket, Transport},?TIMEOUT). + +handle_info({tcp, Socket, Data}, #state{transport = Transport} = State) -> + inet:setopts(State#state.socket, [{active, once}]), + Transport:send(Socket, reverse_binary(Data)), + {noreply, State, ?TIMEOUT}; +handle_info({tcp_closed, _Socket}, State) -> + {stop, normal, State}; +handle_info({tcp_error, _, Reason}, State) -> + {stop, Reason, State}; +handle_info(timeout, State) -> + {stop, normal, State}; +handle_info(_Info, State) -> + {stop, normal, State}. + +handle_call(_Request, _From, State) -> + {reply, ok, State}. + +handle_cast(_Msg, State) -> + {noreply, State}. + +terminate(_Reason, _State) -> + ok. + +code_change(_OldVsn, State, _Extra) -> + {ok, State}. + |