Packages
amqp_client
3.7.21
4.2.1
4.2.0
retired
4.1.6
4.1.5
retired
4.1.5-rc.2
retired
4.1.5-rc.1
retired
4.1.5-1
4.0.3
4.0.3-rc.1
3.13.0-rc.2
3.13.0-rc.1
3.12.14
3.12.13
3.12.12
3.12.11
3.12.10
3.12.9
3.12.8
3.12.7
3.12.6
3.12.5
3.12.4
3.12.3
3.12.2
3.12.1
3.12.0
3.12.0-rc.4
3.11.28
3.11.27
3.11.26
3.11.25
3.11.24
3.11.23
3.11.22
3.11.21
3.11.20
3.11.19
3.11.18
3.11.17
3.11.13
3.11.12
3.11.11
3.11.10
3.11.9
3.11.8
3.11.7
3.11.6
3.11.5
3.11.4
3.11.3
3.11.2
3.11.1
3.11.0
3.11.0-rc.2
3.11.0-rc.1
3.11.0-1
3.10.20
3.10.19
3.10.18
3.10.17
3.10.16
3.10.15
3.10.14
3.10.13
3.10.12
3.10.11
3.10.10
3.10.9
3.10.8
3.10.7
3.10.6
3.10.5
3.10.4
3.10.3
3.10.2
3.10.1
3.10.0
3.10.0-rc.6
3.10.0-rc.5
3.9.29
3.9.28
3.9.27
3.9.26
3.9.25
3.9.24
3.9.23
3.9.22
3.9.21
3.9.20
3.9.19
3.9.18
3.9.17
3.9.16
3.9.15
3.9.11
3.9.8
3.9.5
3.9.4
3.9.1
3.8.35
3.8.34
3.8.33
3.8.32
3.8.31
3.8.30
3.8.25
3.8.21
3.8.19
3.8.14
3.8.11
3.8.10
3.8.10-rc.1
3.8.9
3.8.7
3.8.6
3.8.6-rc.2
3.8.6-rc.1
3.8.5
3.8.5-rc.2
3.8.5-rc.1
3.8.4
3.8.4-rc.3
3.8.4-rc.1
3.8.3
3.8.3-rc.2
3.8.3-rc.1
3.8.2
3.8.2-rc.1
3.8.1
3.8.1-rc.1
3.8.0
3.8.0-rc.3
3.8.0-rc.2
3.8.0-rc.1
3.7.28
3.7.27
3.7.27-rc.2
3.7.27-rc.1
3.7.26
3.7.25
3.7.25-rc.1
3.7.24
3.7.24-rc.2
3.7.24-rc.1
3.7.23
3.7.23-rc.1
3.7.22
3.7.22-rc.2
3.7.22-rc.1
3.7.21
3.7.20
3.7.20-rc.2
3.7.20-rc.1
3.7.19
3.7.18
3.7.18-rc.1
3.7.17
3.7.17-rc.3
3.7.17-rc.2
3.7.15
3.7.14
3.7.14-rc.2
3.7.14-rc.1
3.7.13
3.7.13-rc.2
3.7.13-rc.1
3.7.12
3.7.12-rc.2
3.7.12-rc.1
3.7.11
3.7.11-rc.2
3.7.11-rc.1
3.7.10-rc.4
3.7.10-rc.3
3.7.10-rc.2
3.7.10-rc.1
3.7.9
3.7.9-rc.3
3.7.9-rc.2
3.7.8
3.7.8-rc.4
3.7.8-rc.3
3.7.8-rc.2
3.7.8-rc.1
3.7.7
3.7.7-rc.2
3.7.7-rc.1
3.7.6
3.7.6-rc.2
3.7.6-rc.1
3.7.5
3.7.5-rc.1
3.7.4
3.7.4-rc.4
3.7.4-rc.3
3.7.4-rc.2
3.7.4-rc.1
3.7.3
3.7.3-rc.2
3.7.3-rc.1
3.7.2
3.7.1
3.7.0-rc.2
3.6.16
3.6.16-rc.1
3.6.15
3.6.15-rc.1
3.6.14
3.6.13
3.6.12
3.6.11
3.6.10
3.6.9
3.6.8
3.6.7
3.6.7-pre.1
3.5.6
3.5.0
3.4.0
3.3.5
3.0.2
0.0.0-rc.1
RabbitMQ AMQP Client
Current section
Files
Jump to
Current section
Files
src/amqp_main_reader.erl
%% The contents of this file are subject to the Mozilla Public License
%% Version 1.1 (the "License"); you may not use this file except in
%% compliance with the License. You may obtain a copy of the License at
%% http://www.mozilla.org/MPL/
%%
%% Software distributed under the License is distributed on an "AS IS"
%% basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See the
%% License for the specific language governing rights and limitations
%% under the License.
%%
%% The Original Code is RabbitMQ.
%%
%% The Initial Developer of the Original Code is GoPivotal, Inc.
%% Copyright (c) 2007-2017 Pivotal Software, Inc. All rights reserved.
%%
%% @private
-module(amqp_main_reader).
-include("amqp_client_internal.hrl").
-behaviour(gen_server).
-export([start_link/5, post_init/1]).
-export([init/1, terminate/2, code_change/3, handle_call/3, handle_cast/2,
handle_info/2]).
-record(state, {sock,
timer,
connection,
channels_manager,
astate,
message = none %% none | {Type, Channel, Length}
}).
%%---------------------------------------------------------------------------
%% Interface
%%---------------------------------------------------------------------------
start_link(Sock, Connection, ChMgr, AState, ConnName) ->
gen_server:start_link(
?MODULE, [Sock, Connection, ConnName, ChMgr, AState], []).
post_init(Reader) ->
gen_server:call(Reader, post_init).
%%---------------------------------------------------------------------------
%% gen_server callbacks
%%---------------------------------------------------------------------------
init([Sock, Connection, ConnName, ChMgr, AState]) ->
?store_proc_name(ConnName),
State = #state{sock = Sock,
connection = Connection,
channels_manager = ChMgr,
astate = AState,
message = none},
{ok, State}.
terminate(_Reason, _State) ->
ok.
code_change(_OldVsn, State, _Extra) ->
{ok, State}.
%% We need to use a call because we are not controlling the socket yet.
handle_call(post_init, _From, State = #state{sock = Sock}) ->
case rabbit_net:setopts(Sock, [{active, once}]) of
ok -> {reply, ok, set_timeout(State)};
{error, Reason} -> handle_error(Reason, State)
end;
handle_call(Call, From, State) ->
{stop, {unexpected_call, Call, From}, State}.
handle_cast(Cast, State) ->
{stop, {unexpected_cast, Cast}, State}.
handle_info({Tag, Sock, Data}, State = #state{sock = Sock})
when Tag =:= tcp; Tag =:= ssl ->
%% Latency hiding: Request next packet first, then process data
case rabbit_net:setopts(Sock, [{active, once}]) of
ok -> handle_data(Data, set_timeout(State));
{error, Reason} -> handle_error(Reason, State)
end;
handle_info({Tag, Sock}, State = #state{sock = Sock})
when Tag =:= tcp_closed; Tag =:= ssl_closed ->
handle_error(closed, State);
handle_info({Tag, Sock, Reason}, State = #state{sock = Sock})
when Tag =:= tcp_error; Tag =:= ssl_error ->
handle_error(Reason, State);
handle_info({timeout, _TimerRef, idle_timeout}, State) ->
handle_error(timeout, State).
handle_data(<<Type:8, Channel:16, Length:32, Payload:Length/binary, ?FRAME_END,
More/binary>>,
#state{message = none} = State) when
Type =:= ?FRAME_METHOD; Type =:= ?FRAME_HEADER;
Type =:= ?FRAME_BODY; Type =:= ?FRAME_HEARTBEAT ->
%% Optimisation for the direct match
handle_data(
More, process_frame(Type, Channel, Payload, State#state{message = none}));
handle_data(<<Type:8, Channel:16, Length:32, Data/binary>>,
#state{message = none} = State) when
Type =:= ?FRAME_METHOD; Type =:= ?FRAME_HEADER;
Type =:= ?FRAME_BODY; Type =:= ?FRAME_HEARTBEAT ->
{noreply, State#state{message = {Type, Channel, Length, Data}}};
handle_data(<<"AMQP", A, B, C>>, #state{sock = Sock, message = none} = State) ->
{ok, <<D>>} = rabbit_net:sync_recv(Sock, 1),
handle_error({refused, {A, B, C, D}}, State);
handle_data(<<Malformed:7/binary, _Rest/binary>>,
#state{message = none} = State) ->
handle_error({malformed_header, Malformed}, State);
handle_data(<<Data/binary>>, #state{message = none} = State) ->
{noreply, State#state{message = {expecting_header, Data}}};
handle_data(Data, #state{message = {Type, Channel, L, OldData}} = State) ->
case <<OldData/binary, Data/binary>> of
<<Payload:L/binary, ?FRAME_END, More/binary>> ->
handle_data(More,
process_frame(Type, Channel, Payload,
State#state{message = none}));
NotEnough ->
%% Read in more data from the socket
{noreply, State#state{message = {Type, Channel, L, NotEnough}}}
end;
handle_data(Data,
#state{message = {expecting_header, Old}} = State) ->
handle_data(<<Old/binary, Data/binary>>, State#state{message = none});
handle_data(<<>>, State) ->
{noreply, State}.
%%---------------------------------------------------------------------------
%% Internal plumbing
%%---------------------------------------------------------------------------
set_timeout(State0) ->
State = cancel_timeout(State0),
TimerRef = case amqp_util:call_timeout() of
infinity -> undefined;
Timeout -> erlang:start_timer(Timeout, self(), idle_timeout)
end,
State#state{timer=TimerRef}.
cancel_timeout(State=#state{timer=TimerRef}) ->
ok = case TimerRef of
undefined -> ok;
_ -> erlang:cancel_timer(TimerRef, [{async, true}, {info, false}])
end,
State#state{timer=undefined}.
process_frame(Type, ChNumber, Payload,
State = #state{connection = Connection,
channels_manager = ChMgr,
astate = AState}) ->
case rabbit_command_assembler:analyze_frame(Type, Payload, ?PROTOCOL) of
heartbeat when ChNumber /= 0 ->
amqp_gen_connection:server_misbehaved(
Connection,
#amqp_error{name = command_invalid,
explanation = "heartbeat on non-zero channel"}),
State;
%% Match heartbeats but don't do anything with them
heartbeat ->
State;
AnalyzedFrame when ChNumber /= 0 ->
amqp_channels_manager:pass_frame(ChMgr, ChNumber, AnalyzedFrame),
State;
AnalyzedFrame ->
State#state{astate = amqp_channels_manager:process_channel_frame(
AnalyzedFrame, 0, Connection, AState)}
end.
handle_error(closed, State = #state{connection = Conn}) ->
Conn ! socket_closed,
{noreply, State};
handle_error({refused, Version}, State = #state{connection = Conn}) ->
Conn ! {refused, Version},
{noreply, State};
handle_error({malformed_header, Version}, State = #state{connection = Conn}) ->
Conn ! {malformed_header, Version},
{noreply, State};
handle_error(Reason, State = #state{connection = Conn}) ->
Conn ! {socket_error, Reason},
{stop, {socket_error, Reason}, State}.