Skip to content

Commit

Permalink
Add ability to specify retries and timeout for failover RADIUS servers
Browse files Browse the repository at this point in the history
Previously we may specifiy the 'retries' and 'timeout' configuration
options for the RADIUS relay servers. This commit adds ability to
specify the same configuration options per RADIUS failover server.

Backward compatibility is preserved. The proxy clients already had
this options, they were not just used during the request but instead
the RADIUS client used the timeout and retries values from the primary
server configuration.
  • Loading branch information
0xAX committed Aug 30, 2024
1 parent 5fc23c6 commit 6f65858
Show file tree
Hide file tree
Showing 5 changed files with 17 additions and 28 deletions.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -305,7 +305,7 @@ All pools are configured via:
%%% ...
{servers_pool, [
{pool_name, [
{{127, 0, 0, 2}, 1812, <<"secret">>, [{retries, 3}]},
{{127, 0, 0, 2}, 1812, <<"secret">>, [{retries, 3}, {timeout, 3000}]},
{{127, 0, 0, 3}, 1812, <<"secret">>}
]}
]}
Expand Down
37 changes: 13 additions & 24 deletions src/eradius_client.erl
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@
-module(eradius_client).
-export([start_link/0, send_request/2, send_request/3, send_remote_request/3, send_remote_request/4]).
%% internal
-export([reconfigure/0, send_remote_request_loop/8, find_suitable_peer/1,
-export([reconfigure/0, send_remote_request_loop/8,
restore_upstream_server/1, store_radius_server_from_pool/3,
init_server_status_metrics/0]).

Expand Down Expand Up @@ -92,21 +92,14 @@ send_request({IP, Port, Secret}, Request, Options) when ?GOOD_CMD(Request) andal
[] ->
SendReqFn();
UpstreamServers ->
case find_suitable_peer([{IP, Port, Secret} | UpstreamServers]) of
case find_suitable_peer([{IP, Port, Secret, Options} | UpstreamServers]) of
[] ->
no_active_servers;
{{IP, Port, Secret}, _NewPool} ->
{{IP, Port, Secret, Options}, _NewPool} ->
SendReqFn();
{NewPeer, []} ->
% Special case, we don't have servers in the pool anymore, but we need
% to preserve `failover` option to mark current server as inactive if
% it will fail
NewOptions = lists:keyreplace(failover, 1, Options, {failover, undefined}),
send_request(NewPeer, Request, NewOptions);
{NewPeer, NewPool} ->
{{NewPeerAddr, NewPeerPort, NewPeerSecret, PeerOpts}, NewPool} ->
% current server is not in list of active servers, so use another one
NewOptions = lists:keyreplace(failover, 1, Options, {failover, NewPool}),
send_request(NewPeer, Request, NewOptions)
send_request({NewPeerAddr, NewPeerPort, NewPeerSecret}, Request, [{failover, NewPool} | PeerOpts])
end
end;
send_request({_IP, _Port, _Secret}, _Request, _Options) ->
Expand Down Expand Up @@ -223,10 +216,8 @@ handle_failed_request(Request, {ServerIP, Port} = _FailedServer, UpstreamServers
case find_suitable_peer(UpstreamServers) of
[] ->
Response;
{NewPeer, NewPool} ->
% leave only active upstream servers
NewOptions = lists:keyreplace(failover, 1, Options, {failover, NewPool}),
send_request(NewPeer, Request, NewOptions)
{{NewPeerAddr, NewPeerPort, NewPeerSecret, PeerOpts}, NewPool} ->
send_request({NewPeerAddr, NewPeerPort, NewPeerSecret}, Request, [{failover, NewPool} | PeerOpts])
end.

% @private
Expand Down Expand Up @@ -632,27 +623,25 @@ client_response_counter_account_match_spec_compile() ->
MatchSpecCompile
end.

find_suitable_peer(undefined) ->
[];
find_suitable_peer([]) ->
[];
find_suitable_peer([{Host, Port, Secret} | Pool]) when is_list(Host) ->
find_suitable_peer([{Host, Port, Secret, Opts} | Pool]) when is_list(Host) ->
try
IP = get_ip(Host),
find_suitable_peer([{IP, Port, Secret} | Pool])
find_suitable_peer([{IP, Port, Secret, Opts} | Pool])
catch _:_ ->
% can't resolve ip by some reasons, just ignore it
find_suitable_peer(Pool)
end;
find_suitable_peer([{IP, Port, Secret} | Pool]) ->
find_suitable_peer([{IP, Port, Secret, Opts} | Pool]) ->
case ets:lookup(?MODULE, {IP, Port}) of
[] ->
find_suitable_peer(Pool);
[{{IP, Port}, _Retries, _InitialRetries}] ->
{{IP, Port, Secret}, Pool}
{{IP, Port, Secret, Opts}, Pool}
end;
find_suitable_peer([{IP, Port, Secret, _Opts} | Pool]) ->
find_suitable_peer([{IP, Port, Secret} | Pool]).
find_suitable_peer([{IP, Port, Secret} | Pool]) ->
find_suitable_peer([{IP, Port, Secret, []} | Pool]).

get_ip(Host) ->
case inet:gethostbyname(Host) of
Expand Down
2 changes: 1 addition & 1 deletion src/eradius_proxy.erl
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@
%%
%% ```
%% {servers_pool, [{pool_name, [
%% {{127, 0, 0, 1}, 1815, <<"secret">>, [{retries, 3}]},
%% {{127, 0, 0, 1}, 1815, <<"secret">>, [{retries, 3}, {timeout, 5000}]},
%% {{127, 0, 0, 1}, 1816, <<"secret">>}]}]}
%% '''
%%
Expand Down
2 changes: 1 addition & 1 deletion test/eradius_client_SUITE.erl
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@
-include("test/eradius_test.hrl").

-define(BAD_SERVER_IP, {eradius_test_handler:localhost(ip), 1820, "secret"}).
-define(BAD_SERVER_INITIAL_RETRIES, 3).
-define(BAD_SERVER_INITIAL_RETRIES, 2).
-define(BAD_SERVER_TUPLE_INITIAL, {{eradius_test_handler:localhost(tuple), 1820},
?BAD_SERVER_INITIAL_RETRIES,
?BAD_SERVER_INITIAL_RETRIES}).
Expand Down
2 changes: 1 addition & 1 deletion test/eradius_test_handler.erl
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@ start() ->
application:set_env(eradius, unreachable_timeout, 2),
application:set_env(eradius, servers_pool, [{test_pool, [{localhost(tuple), 1812, "secret"},
% fake upstream server for fail-over
{localhost(string), 1820, "secret"}]}]),
{localhost(string), 1820, "secret", [{timeout, 1000}, {retries, 2}]}]}]),
application:ensure_all_started(eradius),
eradius:modules_ready([?MODULE]).

Expand Down

0 comments on commit 6f65858

Please sign in to comment.