processone / p1_utils / 6

Build # 6

Build Type

github

Committed by GitHub

Commit Message

<a href="https://github.com/processone/p1_utils/commit/<a class=hub.com/processone/p1_utils/commit/<a class="double-link" href="https://git"><a class=hub.com/processone/p1_utils/commit/<a class="double-link" href="https://git"><a class=hub.com/processone/p1_utils/commit/<a class="double-link" href="https://git"><a class=hub.com/processone/p1_utils/commit/320189a8ce04304114e1695ef1f86ed70b1af094">320189a8c">&lt;a href=&quot;https://github.com/processone/p1_utils/commit/</a><a class="double-link" href="https://github.com/processone/p1_utils/commit/&lt;a class=&quot;double-link&quot; href=&quot;https://git">&lt;a class=</a>hub.com/processone/p1_utils/commit/&lt;a class=&quot;double-link&quot; href=&quot;https://git">&lt;a class=</a>hub.com/processone/p1_utils/commit/&lt;a class=&quot;double-link&quot; href=&quot;https://git">&lt;a class=</a>hub.com/processone/p1_utils/commit/&lt;a class=&quot;double-link&quot; href=&quot;https://git">&lt;a class=</a>hub.com/processone/p1_utils/commit/320189a8ce04304114e1695ef1f86ed70b1af094">320189a8c</a><a href="https://github.com/processone/p1_utils/commit/320189a8ce04304114e1695ef1f86ed70b1af094">&quot;&gt;&amp;lt;a href=&amp;quot;https://github.com/processone/p1_utils/commit/&lt;/a&gt;&lt;a class=&quot;double-link&quot; href=&quot;https://github.com/processone/p1_utils/commit/&amp;lt;a class=&amp;quot;double-link&amp;quot; href=&amp;quot;https://git&quot;&gt;&amp;lt;a class=&lt;/a&gt;hub.com/processone/p1_utils/commit/&amp;lt;a class=&amp;quot;double-link&amp;quot; href=&amp;quot;https://git&quot;&gt;&amp;lt;a class=&lt;/a&gt;hub.com/processone/p1_utils/commit/&amp;lt;a class=&amp;quot;double-link&amp;quot; href=&amp;quot;https://git&quot;&gt;&amp;lt;a class=&lt;/a&gt;hub.com/processone/p1_utils/commit/&amp;lt;a class=&amp;quot;double-link&amp;quot; href=&amp;quot;https://git&quot;&gt;&amp;lt;a class=&lt;/a&gt;hub.com/processone/p1_utils/commit/320189a8ce04304114e1695ef1f86ed70b1af094&quot;&gt;320189a8c&lt;/a&gt;&lt;a href=&quot;https://github.com/processone/p1_utils/commit/320189a8ce04304114e1695ef1f86ed70b1af094&quot;&gt;&amp;quot;&amp;gt;&amp;amp;lt;a href=&amp;amp;quot;https://github.com/processone/p1_utils/commit/&amp;lt;/a&amp;gt;&amp;lt;a class=&amp;quot;double-link&amp;quot; href=&amp;quot;https://github.com/process... (continued)

Pull Request Pull Request #13: Add p1_proxy_protocol module

Run Details

48 of 48 new or added lines in 1 file covered. (100.0%)

210 of 1689 relevant lines covered (12.43%)

14.99 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

0.0

/src/p1_proxy_protocol.erl

%%%----------------------------------------------------------------------
%%% File    : p1_proxy_protocol.erl
%%% Author  : Paweł Chmielowski <pawel@process-one.net>
%%% Purpose :
%%% Created : 27 Nov 2018 by Paweł Chmielowski <pawel@process-one.net>
%%%
%%%
%%% Copyright (C) 2002-2021   ProcessOne
%%%
%%% This program is free software; you can redistribute it and/or
%%% modify it under the terms of the GNU General Public License as
%%% published by the Free Software Foundation; either version 2 of the
%%% License, or (at your option) any later version.
%%%
%%% This program is distributed in the hope that it will be useful,
%%% but WITHOUT ANY WARRANTY; without even the implied warranty of
%%% MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
%%% General Public License for more details.
%%%
%%% You should have received a copy of the GNU General Public License along
%%% with this program; if not, write to the Free Software Foundation, Inc.,
%%% 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
%%%
%%%----------------------------------------------------------------------
-module(p1_proxy_protocol).
-author("pawel@process-one.net").

%% API
-export([decode/3]).

-spec decode(gen_tcp | ssl, inet:socket(), integer())
      -> {{inet:ip_address(), inet:port_number()},
          {inet:ip_address(), inet:port_number()}}
         | {error, atom()}
         | {undefined, undefined}.
decode(SockMod, Socket, Timeout) ->
    V = SockMod:recv(Socket, 6, Timeout),
    case V of
        {ok, <<"PROXY ">>} ->
            decode_v1(SockMod, Socket, Timeout);
        {ok, <<16#0d, 16#0a, 16#0d, 16#0a, 16#00, 16#0d>>} ->
            decode_v2(SockMod, Socket, Timeout);
        _ ->
            {error, eproto}
    end.

decode_v1(SockMod, Socket, Timeout) ->
    case read_until_rn(SockMod, Socket, <<>>, false, Timeout) of
        {error, _} = Err ->
            Err;
        Val ->
            case binary:split(Val, <<" ">>, [global]) of
                [<<"TCP4">>, SAddr, DAddr, SPort, DPort] ->
                    try {inet_parse:ipv4strict_address(binary_to_list(SAddr)),
                         inet_parse:ipv4strict_address(binary_to_list(DAddr)),
                         binary_to_integer(SPort),
                         binary_to_integer(DPort)}
                    of
                        {{ok, DA}, {ok, SA}, DP, SP} ->
                            {{SA, SP}, {DA, DP}};
                        _ ->
                            {error, eproto}
                    catch
                        error:badarg ->
                            {error, eproto}
                    end;
                [<<"TCP6">>, SAddr, DAddr, SPort, DPort] ->
                    try {inet_parse:ipv6strict_address(binary_to_list(SAddr)),
                         inet_parse:ipv6strict_address(binary_to_list(DAddr)),
                         binary_to_integer(SPort),
                         binary_to_integer(DPort)}
                    of
                        {{ok, DA}, {ok, SA}, DP, SP} ->
                            {{SA, SP}, {DA, DP}};
                        _ ->
                            {error, eproto}
                    catch
                        error:badarg ->
                            {error, eproto}
                    end;
                [<<"UNKNOWN">> | _] ->
                    {undefined, undefined}
            end
    end.

decode_v2(SockMod, Socket, Timeout) ->
    case SockMod:recv(Socket, 10, Timeout) of
        {error, _} = Err ->
            Err;
        {ok, <<16#0a, 16#51, 16#55, 16#49, 16#54, 16#0a,
               2:4, Command:4, Transport:8, AddrLen:16/big-unsigned-integer>>} ->
            case SockMod:recv(Socket, AddrLen, Timeout) of
                {error, _} = Err ->
                    Err;
                {ok, Data} ->
                    case Command of
                        0 ->
                            case {inet:sockname(Socket), inet:peername(Socket)} of
                                {{ok, SA}, {ok, DA}} ->
                                    {SA, DA};
                                {{error, _} = E, _} ->
                                    E;
                                {_, {error, _} = E} ->
                                    E
                            end;
                        1 ->
                            case Transport of
                                % UNSPEC or UNIX
                                V when V == 0; V == 16#31; V == 16#32 ->
                                    {{unknown, unknown}, {unknown, unknown}};
                                % IPV4 over TCP or UDP
                                V when V == 16#11; V == 16#12 ->
                                    case Data of
                                        <<D1:8, D2:8, D3:8, D4:8,
                                          S1:8, S2:8, S3:8, S4:8,
                                          DP:16/big-unsigned-integer,
                                          SP:16/big-unsigned-integer,
                                          _/binary>> ->
                                            {{{S1, S2, S3, S4}, SP},
                                             {{D1, D2, D3, D4}, DP}};
                                        _ ->
                                            {error, eproto}
                                    end;
                                % IPV6 over TCP or UDP
                                V when V == 16#21; V == 16#22 ->
                                    case Data of
                                        <<D1:16/big-unsigned-integer,
                                          D2:16/big-unsigned-integer,
                                          D3:16/big-unsigned-integer,
                                          D4:16/big-unsigned-integer,
                                          D5:16/big-unsigned-integer,
                                          D6:16/big-unsigned-integer,
                                          D7:16/big-unsigned-integer,
                                          D8:16/big-unsigned-integer,
                                          S1:16/big-unsigned-integer,
                                          S2:16/big-unsigned-integer,
                                          S3:16/big-unsigned-integer,
                                          S4:16/big-unsigned-integer,
                                          S5:16/big-unsigned-integer,
                                          S6:16/big-unsigned-integer,
                                          S7:16/big-unsigned-integer,
                                          S8:16/big-unsigned-integer,
                                          DP:16/big-unsigned-integer,
                                          SP:16/big-unsigned-integer,
                                          _/binary>> ->
                                            {{{S1, S2, S3, S4, S5, S6, S7, S8}, SP},
                                             {{D1, D2, D3, D4, D5, D6, D7, D8}, DP}};
                                        _ ->
                                            {error, eproto}
                                    end
                            end;
                        _ ->
                            {error, eproto}
                    end
            end;
        <<16#0a, 16#51, 16#55, 16#49, 16#54, 16#0a, _/binary>> ->
            {error, eproto};
        _ ->
            {error, eproto}
    end.

read_until_rn(_SockMod, _Socket, Data, _, _) when size(Data) > 107 ->
    {error, eproto};
read_until_rn(SockMod, Socket, Data, true, Timeout) ->
    case SockMod:recv(Socket, 1, Timeout) of
        {ok, <<"\n">>} ->
            Data;
        {ok, <<"\r">>} ->
            read_until_rn(SockMod, Socket, <<Data/binary, "\r">>,
                          true, Timeout);
        {ok, Other} ->
            read_until_rn(SockMod, Socket, <<Data/binary, "\r", Other/binary>>,
                          false, Timeout);
        {error, _} = Err ->
            Err
    end;
read_until_rn(SockMod, Socket, Data, false, Timeout) ->
    case SockMod:recv(Socket, 2, Timeout) of
        {ok, <<"\r\n">>} ->
            Data;
        {ok, <<Byte:8, "\r">>} ->
            read_until_rn(SockMod, Socket, <<Data/binary, Byte:8>>,
                          true, Timeout);
        {ok, Other} ->
            read_until_rn(SockMod, Socket, <<Data/binary, Other/binary>>,
                          false, Timeout);
        {error, _} = Err ->
            Err
    end.

1	%%%----------------------------------------------------------------------
2	%%% File : p1_proxy_protocol.erl
3	%%% Author : Paweł Chmielowski <pawel@process-one.net>
4	%%% Purpose :
5	%%% Created : 27 Nov 2018 by Paweł Chmielowski <pawel@process-one.net>
6	%%%
7	%%%
8	%%% Copyright (C) 2002-2021 ProcessOne
9	%%%
10	%%% This program is free software; you can redistribute it and/or
11	%%% modify it under the terms of the GNU General Public License as
12	%%% published by the Free Software Foundation; either version 2 of the
13	%%% License, or (at your option) any later version.
14	%%%
15	%%% This program is distributed in the hope that it will be useful,
16	%%% but WITHOUT ANY WARRANTY; without even the implied warranty of
17	%%% MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
18	%%% General Public License for more details.
19	%%%
20	%%% You should have received a copy of the GNU General Public License along
21	%%% with this program; if not, write to the Free Software Foundation, Inc.,
22	%%% 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
23	%%%
24	%%%----------------------------------------------------------------------
25	-module(p1_proxy_protocol).
26	-author("pawel@process-one.net").
27
28	%% API
29	-export([decode/3]).
30
31	-spec decode(gen_tcp \| ssl, inet:socket(), integer())
32	-> {{inet:ip_address(), inet:port_number()},
33	{inet:ip_address(), inet:port_number()}}
34	\| {error, atom()}
35	\| {undefined, undefined}.
36	decode(SockMod, Socket, Timeout) ->
37	V = SockMod:recv(Socket, 6, Timeout),	×
38	case V of	×
39	{ok, <<"PROXY ">>} ->
40	decode_v1(SockMod, Socket, Timeout);	×
41	{ok, <<16#0d, 16#0a, 16#0d, 16#0a, 16#00, 16#0d>>} ->
42	decode_v2(SockMod, Socket, Timeout);	×
43	_ ->
44	{error, eproto}	×
45	end.
46
47	decode_v1(SockMod, Socket, Timeout) ->
48	case read_until_rn(SockMod, Socket, <<>>, false, Timeout) of	×
49	{error, _} = Err ->
50	Err;	×
51	Val ->
52	case binary:split(Val, <<" ">>, [global]) of	×
53	[<<"TCP4">>, SAddr, DAddr, SPort, DPort] ->
54	try {inet_parse:ipv4strict_address(binary_to_list(SAddr)),	×
55	inet_parse:ipv4strict_address(binary_to_list(DAddr)),
56	binary_to_integer(SPort),
57	binary_to_integer(DPort)}
58	of
59	{{ok, DA}, {ok, SA}, DP, SP} ->
60	{{SA, SP}, {DA, DP}};	×
61	_ ->
62	{error, eproto}	×
63	catch
64	error:badarg ->
65	{error, eproto}	×
66	end;
67	[<<"TCP6">>, SAddr, DAddr, SPort, DPort] ->
68	try {inet_parse:ipv6strict_address(binary_to_list(SAddr)),	×
69	inet_parse:ipv6strict_address(binary_to_list(DAddr)),
70	binary_to_integer(SPort),
71	binary_to_integer(DPort)}
72	of
73	{{ok, DA}, {ok, SA}, DP, SP} ->
74	{{SA, SP}, {DA, DP}};	×
75	_ ->
76	{error, eproto}	×
77	catch
78	error:badarg ->
79	{error, eproto}	×
80	end;
81	[<<"UNKNOWN">> \| _] ->
82	{undefined, undefined}	×
83	end
84	end.
85
86	decode_v2(SockMod, Socket, Timeout) ->
87	case SockMod:recv(Socket, 10, Timeout) of	×
88	{error, _} = Err ->
89	Err;	×
90	{ok, <<16#0a, 16#51, 16#55, 16#49, 16#54, 16#0a,
91	2:4, Command:4, Transport:8, AddrLen:16/big-unsigned-integer>>} ->
92	case SockMod:recv(Socket, AddrLen, Timeout) of	×
93	{error, _} = Err ->
94	Err;	×
95	{ok, Data} ->
96	case Command of	×
97	0 ->
98	case {inet:sockname(Socket), inet:peername(Socket)} of	×
99	{{ok, SA}, {ok, DA}} ->
100	{SA, DA};	×
101	{{error, _} = E, _} ->
102	E;	×
103	{_, {error, _} = E} ->
104	E	×
105	end;
106	1 ->
107	case Transport of	×
108	% UNSPEC or UNIX
109	V when V == 0; V == 16#31; V == 16#32 ->
110	{{unknown, unknown}, {unknown, unknown}};	×
111	% IPV4 over TCP or UDP
112	V when V == 16#11; V == 16#12 ->
113	case Data of	×
114	<<D1:8, D2:8, D3:8, D4:8,
115	S1:8, S2:8, S3:8, S4:8,
116	DP:16/big-unsigned-integer,
117	SP:16/big-unsigned-integer,
118	_/binary>> ->
119	{{{S1, S2, S3, S4}, SP},	×
120	{{D1, D2, D3, D4}, DP}};
121	_ ->
122	{error, eproto}	×
123	end;
124	% IPV6 over TCP or UDP
125	V when V == 16#21; V == 16#22 ->
126	case Data of	×
127	<<D1:16/big-unsigned-integer,
128	D2:16/big-unsigned-integer,
129	D3:16/big-unsigned-integer,
130	D4:16/big-unsigned-integer,
131	D5:16/big-unsigned-integer,
132	D6:16/big-unsigned-integer,
133	D7:16/big-unsigned-integer,
134	D8:16/big-unsigned-integer,
135	S1:16/big-unsigned-integer,
136	S2:16/big-unsigned-integer,
137	S3:16/big-unsigned-integer,
138	S4:16/big-unsigned-integer,
139	S5:16/big-unsigned-integer,
140	S6:16/big-unsigned-integer,
141	S7:16/big-unsigned-integer,
142	S8:16/big-unsigned-integer,
143	DP:16/big-unsigned-integer,
144	SP:16/big-unsigned-integer,
145	_/binary>> ->
146	{{{S1, S2, S3, S4, S5, S6, S7, S8}, SP},	×
147	{{D1, D2, D3, D4, D5, D6, D7, D8}, DP}};
148	_ ->
149	{error, eproto}	×
150	end
151	end;
152	_ ->
153	{error, eproto}	×
154	end
155	end;
156	<<16#0a, 16#51, 16#55, 16#49, 16#54, 16#0a, _/binary>> ->
157	{error, eproto};	×
158	_ ->
159	{error, eproto}	×
160	end.
161
162	read_until_rn(_SockMod, _Socket, Data, _, _) when size(Data) > 107 ->
163	{error, eproto};	×
164	read_until_rn(SockMod, Socket, Data, true, Timeout) ->
165	case SockMod:recv(Socket, 1, Timeout) of	×
166	{ok, <<"\n">>} ->
167	Data;	×
168	{ok, <<"\r">>} ->
169	read_until_rn(SockMod, Socket, <<Data/binary, "\r">>,	×
170	true, Timeout);
171	{ok, Other} ->
172	read_until_rn(SockMod, Socket, <<Data/binary, "\r", Other/binary>>,	×
173	false, Timeout);
174	{error, _} = Err ->
175	Err	×
176	end;
177	read_until_rn(SockMod, Socket, Data, false, Timeout) ->
178	case SockMod:recv(Socket, 2, Timeout) of	×
179	{ok, <<"\r\n">>} ->
180	Data;	×
181	{ok, <<Byte:8, "\r">>} ->
182	read_until_rn(SockMod, Socket, <<Data/binary, Byte:8>>,	×
183	true, Timeout);
184	{ok, Other} ->
185	read_until_rn(SockMod, Socket, <<Data/binary, Other/binary>>,	×
186	false, Timeout);
187	{error, _} = Err ->
188	Err	×
189	end.

processone / p1_utils / 6

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous