25
1
mirror of https://github.com/processone/ejabberd.git synced 2024-10-31 15:21:38 +01:00
xmpp.chapril.org-ejabberd/src/ejabberd_stun.erl

172 lines
5.2 KiB
Erlang
Raw Normal View History

2014-05-08 14:08:07 +02:00
%%%-------------------------------------------------------------------
2017-01-03 15:58:52 +01:00
%%% File : ejabberd_stun.erl
%%% Author : Evgeny Khramtsov <ekhramtsov@process-one.net>
%%% Purpose : STUN RFC-5766
2014-05-08 14:08:07 +02:00
%%% Created : 8 May 2014 by Evgeny Khramtsov <ekhramtsov@process-one.net>
%%%
2017-01-03 15:58:52 +01:00
%%%
2020-01-28 13:34:02 +01:00
%%% ejabberd, Copyright (C) 2013-2020 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.
2017-11-10 17:51:22 +01:00
%%%
2014-05-08 14:08:07 +02:00
%%%-------------------------------------------------------------------
2017-11-10 17:51:22 +01:00
2014-05-08 14:08:07 +02:00
-module(ejabberd_stun).
2018-09-17 10:21:02 +02:00
-behaviour(ejabberd_listener).
2015-05-21 17:02:36 +02:00
-protocol({rfc, 5766}).
-protocol({xep, 176, '1.0'}).
2015-05-21 17:02:36 +02:00
-ifndef(STUN).
-include("logger.hrl").
-export([accept/1, start/3, start_link/3, listen_options/0]).
fail() ->
?CRITICAL_MSG("Listening module ~ts is not available: "
"ejabberd is not compiled with STUN/TURN support",
[?MODULE]),
erlang:error(stun_not_compiled).
2018-09-17 10:21:02 +02:00
accept(_) ->
fail().
2018-09-18 11:53:36 +02:00
listen_options() ->
fail().
start(_, _, _) ->
fail().
start_link(_, _, _) ->
fail().
-else.
-export([tcp_init/2, udp_init/2, udp_recv/5, start/3,
start_link/3, accept/1, listen_opt_type/1, listen_options/0]).
2014-05-08 14:08:07 +02:00
-include("logger.hrl").
%%%===================================================================
%%% API
%%%===================================================================
tcp_init(Socket, Opts) ->
ejabberd:start_app(stun),
2014-05-08 14:08:07 +02:00
stun:tcp_init(Socket, prepare_turn_opts(Opts)).
2019-06-14 11:33:26 +02:00
-dialyzer({nowarn_function, udp_init/2}).
2014-05-08 14:08:07 +02:00
udp_init(Socket, Opts) ->
ejabberd:start_app(stun),
2014-05-08 14:08:07 +02:00
stun:udp_init(Socket, prepare_turn_opts(Opts)).
udp_recv(Socket, Addr, Port, Packet, Opts) ->
stun:udp_recv(Socket, Addr, Port, Packet, Opts).
start(SockMod, Socket, Opts) ->
stun:start({SockMod, Socket}, Opts).
2014-05-08 14:08:07 +02:00
start_link(_SockMod, Socket, Opts) ->
stun:start_link(Socket, Opts).
2018-09-17 10:21:02 +02:00
accept(_Pid) ->
ok.
2014-05-08 14:08:07 +02:00
%%%===================================================================
%%% Internal functions
%%%===================================================================
prepare_turn_opts(Opts) ->
UseTurn = proplists:get_bool(use_turn, Opts),
prepare_turn_opts(Opts, UseTurn).
prepare_turn_opts(Opts, _UseTurn = false) ->
set_certfile(Opts);
2014-05-08 14:08:07 +02:00
prepare_turn_opts(Opts, _UseTurn = true) ->
2019-06-14 11:33:26 +02:00
NumberOfMyHosts = length(ejabberd_option:hosts()),
2014-05-08 14:08:07 +02:00
case proplists:get_value(turn_ip, Opts) of
undefined ->
2019-06-14 11:33:26 +02:00
?WARNING_MSG("Option 'turn_ip' is undefined, "
"most likely the TURN relay won't be working "
2014-05-08 14:08:07 +02:00
"properly", []);
_ ->
ok
end,
AuthFun = fun ejabberd_auth:get_password_s/2,
Shaper = proplists:get_value(shaper, Opts, none),
AuthType = proplists:get_value(auth_type, Opts, user),
Realm = case proplists:get_value(auth_realm, Opts) of
2014-05-08 14:08:07 +02:00
undefined when AuthType == user ->
if NumberOfMyHosts > 1 ->
2019-06-14 11:33:26 +02:00
?WARNING_MSG("You have several virtual "
2014-05-08 14:08:07 +02:00
"hosts configured, but option "
"'auth_realm' is undefined and "
"'auth_type' is set to 'user', "
2019-06-14 11:33:26 +02:00
"most likely the TURN relay won't "
"be working properly. Using ~ts as "
"a fallback", [ejabberd_config:get_myname()]);
2014-05-08 14:08:07 +02:00
true ->
ok
end,
[{auth_realm, ejabberd_config:get_myname()}];
2014-05-08 14:08:07 +02:00
_ ->
[]
end,
2018-07-05 08:31:55 +02:00
MaxRate = ejabberd_shaper:get_max_rate(Shaper),
Opts1 = Realm ++ [{auth_fun, AuthFun},{shaper, MaxRate} |
lists:keydelete(shaper, 1, Opts)],
set_certfile(Opts1).
set_certfile(Opts) ->
case lists:keymember(certfile, 1, Opts) of
true ->
Opts;
false ->
Realm = proplists:get_value(auth_realm, Opts, ejabberd_config:get_myname()),
case ejabberd_pkix:get_certfile(Realm) of
{ok, CertFile} ->
[{certfile, CertFile}|Opts];
error ->
2019-06-14 11:33:26 +02:00
Opts
end
end.
listen_opt_type(use_turn) ->
2019-06-14 11:33:26 +02:00
econf:bool();
listen_opt_type(ip) ->
econf:ipv4();
listen_opt_type(turn_ip) ->
2019-06-14 11:33:26 +02:00
econf:ipv4();
listen_opt_type(auth_type) ->
2019-06-14 11:33:26 +02:00
econf:enum([anonymous, user]);
listen_opt_type(auth_realm) ->
2019-06-14 11:33:26 +02:00
econf:binary();
listen_opt_type(turn_min_port) ->
2019-06-14 11:33:26 +02:00
econf:int(1025, 65535);
listen_opt_type(turn_max_port) ->
2019-06-14 11:33:26 +02:00
econf:int(1025, 65535);
listen_opt_type(turn_max_allocations) ->
2019-06-14 11:33:26 +02:00
econf:pos_int(infinity);
listen_opt_type(turn_max_permissions) ->
2019-06-14 11:33:26 +02:00
econf:pos_int(infinity);
listen_opt_type(server_name) ->
2019-06-14 11:33:26 +02:00
econf:binary();
listen_opt_type(certfile) ->
econf:pem().
2018-09-18 11:53:36 +02:00
listen_options() ->
[{shaper, none},
{use_turn, false},
{turn_ip, undefined},
2018-09-18 11:53:36 +02:00
{auth_type, user},
{auth_realm, undefined},
{tls, false},
{certfile, undefined},
{turn_min_port, 49152},
{turn_max_port, 65535},
{turn_max_allocations, 10},
{turn_max_permissions, 10},
{server_name, <<"ejabberd">>}].
-endif.