mirror of
https://github.com/processone/ejabberd.git
synced 2024-11-26 16:26:24 +01:00
restore missing stun directory
SVN Revision: 2458
This commit is contained in:
Evgeniy Khramtsov
parent
bf0dbeb8b3
commit
0a8bcf6530
38
src/stun/Makefile.in
Normal file
38
src/stun/Makefile.in
Normal file
@ -0,0 +1,38 @@
|
|||||||
|
# $Id: Makefile.in 1453 2008-07-16 16:58:42Z badlop $
|
||||||
|
|
||||||
|
CC = @CC@
|
||||||
|
CFLAGS = @CFLAGS@
|
||||||
|
CPPFLAGS = @CPPFLAGS@
|
||||||
|
LDFLAGS = @LDFLAGS@
|
||||||
|
LIBS = @LIBS@
|
||||||
|
|
||||||
|
ERLANG_CFLAGS = @ERLANG_CFLAGS@
|
||||||
|
ERLANG_LIBS = @ERLANG_LIBS@
|
||||||
|
|
||||||
|
EFLAGS += -I ..
|
||||||
|
EFLAGS += -pz ..
|
||||||
|
|
||||||
|
# make debug=true to compile Erlang module with debug informations.
|
||||||
|
ifdef debug
|
||||||
|
EFLAGS+=+debug_info
|
||||||
|
endif
|
||||||
|
|
||||||
|
OUTDIR = ..
|
||||||
|
SOURCES = $(wildcard *.erl)
|
||||||
|
BEAMS = $(addprefix $(OUTDIR)/,$(SOURCES:.erl=.beam))
|
||||||
|
|
||||||
|
|
||||||
|
all: $(BEAMS)
|
||||||
|
|
||||||
|
$(OUTDIR)/%.beam: %.erl
|
||||||
|
@ERLC@ -W $(EFLAGS) -o $(OUTDIR) $<
|
||||||
|
|
||||||
|
clean:
|
||||||
|
rm -f $(BEAMS)
|
||||||
|
|
||||||
|
distclean: clean
|
||||||
|
rm -f Makefile
|
||||||
|
|
||||||
|
TAGS:
|
||||||
|
etags *.erl
|
||||||
|
|
||||||
18
src/stun/Makefile.win32
Normal file
18
src/stun/Makefile.win32
Normal file
@ -0,0 +1,18 @@
|
|||||||
|
|
||||||
|
include ..\Makefile.inc
|
||||||
|
|
||||||
|
EFLAGS = -I .. -pz ..
|
||||||
|
|
||||||
|
OUTDIR = ..
|
||||||
|
BEAMS = ..\stun_codec.beam ..\ejabberd_stun.beam
|
||||||
|
|
||||||
|
ALL : $(BEAMS)
|
||||||
|
|
||||||
|
CLEAN :
|
||||||
|
-@erase $(BEAMS)
|
||||||
|
|
||||||
|
$(OUTDIR)\stun_codec.beam : stun_codec.erl
|
||||||
|
erlc -W $(EFLAGS) -o $(OUTDIR) stun_codec.erl
|
||||||
|
|
||||||
|
$(OUTDIR)\ejabberd_stun.beam : ejabberd_stun.erl
|
||||||
|
erlc -W $(EFLAGS) -o $(OUTDIR) ejabberd_stun.erl
|
||||||
260
src/stun/ejabberd_stun.erl
Normal file
260
src/stun/ejabberd_stun.erl
Normal file
@ -0,0 +1,260 @@
|
|||||||
|
%%%-------------------------------------------------------------------
|
||||||
|
%%% File : ejabberd_stun.erl
|
||||||
|
%%% Author : Evgeniy Khramtsov <ekhramtsov@process-one.net>
|
||||||
|
%%% Description : RFC5389 implementation.
|
||||||
|
%%% Currently only Binding usage is supported.
|
||||||
|
%%%
|
||||||
|
%%% Created : 8 Aug 2009 by Evgeniy Khramtsov <ekhramtsov@process-one.net>
|
||||||
|
%%%
|
||||||
|
%%%
|
||||||
|
%%% ejabberd, Copyright (C) 2002-2009 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., 59 Temple Place, Suite 330, Boston, MA
|
||||||
|
%%% 02111-1307 USA
|
||||||
|
%%%
|
||||||
|
%%%-------------------------------------------------------------------
|
||||||
|
-module(ejabberd_stun).
|
||||||
|
|
||||||
|
-behaviour(gen_fsm).
|
||||||
|
|
||||||
|
%% API
|
||||||
|
-export([start_link/2,
|
||||||
|
start/2,
|
||||||
|
socket_type/0,
|
||||||
|
udp_recv/5]).
|
||||||
|
|
||||||
|
%% gen_fsm callbacks
|
||||||
|
-export([init/1,
|
||||||
|
handle_event/3,
|
||||||
|
handle_sync_event/4,
|
||||||
|
handle_info/3,
|
||||||
|
terminate/3,
|
||||||
|
code_change/4]).
|
||||||
|
|
||||||
|
%% gen_fsm states
|
||||||
|
-export([wait_for_tls/2,
|
||||||
|
session_established/2]).
|
||||||
|
|
||||||
|
-include("ejabberd.hrl").
|
||||||
|
-include("stun.hrl").
|
||||||
|
|
||||||
|
-define(MAX_BUF_SIZE, 64*1024). %% 64kb
|
||||||
|
-define(TIMEOUT, 10000). %% 10 sec
|
||||||
|
|
||||||
|
-record(state, {sock,
|
||||||
|
sock_mod = gen_tcp,
|
||||||
|
certfile,
|
||||||
|
peer,
|
||||||
|
tref,
|
||||||
|
buf = <<>>}).
|
||||||
|
|
||||||
|
%%====================================================================
|
||||||
|
%% API
|
||||||
|
%%====================================================================
|
||||||
|
start({gen_tcp, Sock}, Opts) ->
|
||||||
|
supervisor:start_child(ejabberd_stun_sup, [Sock, Opts]).
|
||||||
|
|
||||||
|
start_link(Sock, Opts) ->
|
||||||
|
gen_fsm:start_link(?MODULE, [Sock, Opts], []).
|
||||||
|
|
||||||
|
socket_type() ->
|
||||||
|
raw.
|
||||||
|
|
||||||
|
udp_recv(Sock, Addr, Port, Data, _Opts) ->
|
||||||
|
case stun_codec:decode(Data) of
|
||||||
|
{ok, Msg, <<>>} ->
|
||||||
|
?DEBUG("got:~n~s", [stun_codec:pp(Msg)]),
|
||||||
|
case process(Addr, Port, Msg) of
|
||||||
|
RespMsg when is_record(RespMsg, stun) ->
|
||||||
|
?DEBUG("sent:~n~s", [stun_codec:pp(RespMsg)]),
|
||||||
|
Data1 = stun_codec:encode(RespMsg),
|
||||||
|
gen_udp:send(Sock, Addr, Port, Data1);
|
||||||
|
_ ->
|
||||||
|
ok
|
||||||
|
end;
|
||||||
|
_ ->
|
||||||
|
ok
|
||||||
|
end.
|
||||||
|
|
||||||
|
%%====================================================================
|
||||||
|
%% gen_fsm callbacks
|
||||||
|
%%====================================================================
|
||||||
|
init([Sock, Opts]) ->
|
||||||
|
case inet:peername(Sock) of
|
||||||
|
{ok, Addr} ->
|
||||||
|
inet:setopts(Sock, [{active, once}]),
|
||||||
|
TRef = erlang:start_timer(?TIMEOUT, self(), stop),
|
||||||
|
State = #state{sock = Sock, peer = Addr, tref = TRef},
|
||||||
|
case proplists:get_value(certfile, Opts) of
|
||||||
|
undefined ->
|
||||||
|
{ok, session_established, State};
|
||||||
|
CertFile ->
|
||||||
|
{ok, wait_for_tls, State#state{certfile = CertFile}}
|
||||||
|
end;
|
||||||
|
Err ->
|
||||||
|
Err
|
||||||
|
end.
|
||||||
|
|
||||||
|
wait_for_tls(Event, State) ->
|
||||||
|
?INFO_MSG("unexpected event in wait_for_tls: ~p", [Event]),
|
||||||
|
{next_state, wait_for_tls, State}.
|
||||||
|
|
||||||
|
session_established(Msg, State) when is_record(Msg, stun) ->
|
||||||
|
?DEBUG("got:~n~s", [stun_codec:pp(Msg)]),
|
||||||
|
{Addr, Port} = State#state.peer,
|
||||||
|
case process(Addr, Port, Msg) of
|
||||||
|
Resp when is_record(Resp, stun) ->
|
||||||
|
?DEBUG("sent:~n~s", [stun_codec:pp(Resp)]),
|
||||||
|
Data = stun_codec:encode(Resp),
|
||||||
|
(State#state.sock_mod):send(State#state.sock, Data);
|
||||||
|
_ ->
|
||||||
|
ok
|
||||||
|
end,
|
||||||
|
{next_state, session_established, State};
|
||||||
|
session_established(Event, State) ->
|
||||||
|
?INFO_MSG("unexpected event in session_established: ~p", [Event]),
|
||||||
|
{next_state, session_established, State}.
|
||||||
|
|
||||||
|
handle_event(_Event, StateName, State) ->
|
||||||
|
{next_state, StateName, State}.
|
||||||
|
|
||||||
|
handle_sync_event(_Event, _From, StateName, State) ->
|
||||||
|
{reply, {error, badarg}, StateName, State}.
|
||||||
|
|
||||||
|
handle_info({tcp, Sock, TLSData}, wait_for_tls, State) ->
|
||||||
|
Buf = <<(State#state.buf)/binary, TLSData/binary>>,
|
||||||
|
%% Check if the initial message is a TLS handshake
|
||||||
|
case Buf of
|
||||||
|
_ when size(Buf) < 3 ->
|
||||||
|
{next_state, wait_for_tls,
|
||||||
|
update_state(State#state{buf = Buf})};
|
||||||
|
<<_:16, 1, _/binary>> ->
|
||||||
|
TLSOpts = [{certfile, State#state.certfile}],
|
||||||
|
{ok, TLSSock} = tls:tcp_to_tls(Sock, TLSOpts),
|
||||||
|
NewState = State#state{sock = TLSSock,
|
||||||
|
buf = <<>>,
|
||||||
|
sock_mod = tls},
|
||||||
|
case tls:recv_data(TLSSock, Buf) of
|
||||||
|
{ok, Data} ->
|
||||||
|
process_data(session_established, NewState, Data);
|
||||||
|
_Err ->
|
||||||
|
{stop, normal, NewState}
|
||||||
|
end;
|
||||||
|
_ ->
|
||||||
|
process_data(session_established, State, TLSData)
|
||||||
|
end;
|
||||||
|
handle_info({tcp, _Sock, TLSData}, StateName,
|
||||||
|
#state{sock_mod = tls} = State) ->
|
||||||
|
case tls:recv_data(State#state.sock, TLSData) of
|
||||||
|
{ok, Data} ->
|
||||||
|
process_data(StateName, State, Data);
|
||||||
|
_Err ->
|
||||||
|
{stop, normal, State}
|
||||||
|
end;
|
||||||
|
handle_info({tcp, _Sock, Data}, StateName, State) ->
|
||||||
|
process_data(StateName, State, Data);
|
||||||
|
handle_info({tcp_closed, _Sock}, _StateName, State) ->
|
||||||
|
?DEBUG("connection reset by peer", []),
|
||||||
|
{stop, normal, State};
|
||||||
|
handle_info({tcp_error, _Sock, Reason}, _StateName, State) ->
|
||||||
|
?DEBUG("connection error: ~p", [Reason]),
|
||||||
|
{stop, normal, State};
|
||||||
|
handle_info({timeout, TRef, stop}, _StateName,
|
||||||
|
#state{tref = TRef} = State) ->
|
||||||
|
{stop, normal, State};
|
||||||
|
handle_info(Info, StateName, State) ->
|
||||||
|
?INFO_MSG("unexpected info: ~p", [Info]),
|
||||||
|
{next_state, StateName, State}.
|
||||||
|
|
||||||
|
terminate(_Reason, _StateName, State) ->
|
||||||
|
catch (State#state.sock_mod):close(State#state.sock),
|
||||||
|
ok.
|
||||||
|
|
||||||
|
code_change(_OldVsn, StateName, State, _Extra) ->
|
||||||
|
{ok, StateName, State}.
|
||||||
|
|
||||||
|
%%--------------------------------------------------------------------
|
||||||
|
%%% Internal functions
|
||||||
|
%%--------------------------------------------------------------------
|
||||||
|
process(Addr, Port, #stun{class = request, unsupported = []} = Msg) ->
|
||||||
|
Resp = prepare_response(Msg),
|
||||||
|
if Msg#stun.method == ?STUN_METHOD_BINDING ->
|
||||||
|
case stun_codec:version(Msg) of
|
||||||
|
old ->
|
||||||
|
Resp#stun{class = response,
|
||||||
|
'MAPPED-ADDRESS' = {Addr, Port}};
|
||||||
|
new ->
|
||||||
|
Resp#stun{class = response,
|
||||||
|
'MAPPED-ADDRESS' = {Addr, Port},
|
||||||
|
'XOR-MAPPED-ADDRESS' = {Addr, Port}}
|
||||||
|
end;
|
||||||
|
true ->
|
||||||
|
Resp#stun{class = error,
|
||||||
|
'ERROR-CODE' = {405, <<"Method Not Allowed">>}}
|
||||||
|
end;
|
||||||
|
process(_Addr, _Port, #stun{class = request} = Msg) ->
|
||||||
|
Resp = prepare_response(Msg),
|
||||||
|
Resp#stun{class = error,
|
||||||
|
'UNKNOWN-ATTRIBUTES' = Msg#stun.unsupported,
|
||||||
|
'ERROR-CODE' = {420, stun_codec:reason(420)}};
|
||||||
|
process(_Addr, _Port, _Msg) ->
|
||||||
|
pass.
|
||||||
|
|
||||||
|
prepare_response(Msg) ->
|
||||||
|
Version = list_to_binary("ejabberd " ++ ?VERSION),
|
||||||
|
#stun{method = Msg#stun.method,
|
||||||
|
magic = Msg#stun.magic,
|
||||||
|
trid = Msg#stun.trid,
|
||||||
|
'SOFTWARE' = Version}.
|
||||||
|
|
||||||
|
process_data(NextStateName, #state{buf = Buf} = State, Data) ->
|
||||||
|
NewBuf = <<Buf/binary, Data/binary>>,
|
||||||
|
case stun_codec:decode(NewBuf) of
|
||||||
|
{ok, Msg, Tail} ->
|
||||||
|
gen_fsm:send_event(self(), Msg),
|
||||||
|
process_data(NextStateName, State#state{buf = <<>>}, Tail);
|
||||||
|
empty ->
|
||||||
|
NewState = State#state{buf = <<>>},
|
||||||
|
{next_state, NextStateName, update_state(NewState)};
|
||||||
|
more when size(NewBuf) < ?MAX_BUF_SIZE ->
|
||||||
|
NewState = State#state{buf = NewBuf},
|
||||||
|
{next_state, NextStateName, update_state(NewState)};
|
||||||
|
_ ->
|
||||||
|
{stop, normal, State}
|
||||||
|
end.
|
||||||
|
|
||||||
|
update_state(#state{sock = Sock} = State) ->
|
||||||
|
case State#state.sock_mod of
|
||||||
|
gen_tcp ->
|
||||||
|
inet:setopts(Sock, [{active, once}]);
|
||||||
|
SockMod ->
|
||||||
|
SockMod:setopts(Sock, [{active, once}])
|
||||||
|
end,
|
||||||
|
cancel_timer(State#state.tref),
|
||||||
|
TRef = erlang:start_timer(?TIMEOUT, self(), stop),
|
||||||
|
State#state{tref = TRef}.
|
||||||
|
|
||||||
|
cancel_timer(TRef) ->
|
||||||
|
case erlang:cancel_timer(TRef) of
|
||||||
|
false ->
|
||||||
|
receive
|
||||||
|
{timeout, TRef, _} ->
|
||||||
|
ok
|
||||||
|
after 0 ->
|
||||||
|
ok
|
||||||
|
end;
|
||||||
|
_ ->
|
||||||
|
ok
|
||||||
|
end.
|
||||||
78
src/stun/stun.hrl
Normal file
78
src/stun/stun.hrl
Normal file
@ -0,0 +1,78 @@
|
|||||||
|
%%%-------------------------------------------------------------------
|
||||||
|
%%% File : stun.hrl
|
||||||
|
%%% Author : Evgeniy Khramtsov <ekhramtsov@process-one.net>
|
||||||
|
%%% Description : STUN values
|
||||||
|
%%% Created : 8 Aug 2009 by Evgeniy Khramtsov <ekhramtsov@process-one.net>
|
||||||
|
%%%
|
||||||
|
%%%
|
||||||
|
%%% ejabberd, Copyright (C) 2002-2009 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., 59 Temple Place, Suite 330, Boston, MA
|
||||||
|
%%% 02111-1307 USA
|
||||||
|
%%%
|
||||||
|
%%%-------------------------------------------------------------------
|
||||||
|
-define(STUN_MAGIC, 16#2112a442).
|
||||||
|
|
||||||
|
%% I know, this is terrible. Refer to 'STUN Message Structure' of
|
||||||
|
%% RFC5389 to understand this.
|
||||||
|
-define(STUN_METHOD(Type),
|
||||||
|
((Type band 16#3e00) bsr 2) bor
|
||||||
|
((Type band 16#e0) bsr 1) bor (Type band 16#f)).
|
||||||
|
-define(STUN_CLASS(Type),
|
||||||
|
((Type band 16#100) bsr 7) bor
|
||||||
|
((Type band 16#10) bsr 4)).
|
||||||
|
-define(STUN_TYPE(C, M),
|
||||||
|
(((M band 16#f80) bsl 2)
|
||||||
|
bor ((M band 16#70) bsl 1)
|
||||||
|
bor (M band 16#f) )
|
||||||
|
bor (((C band 16#2) bsl 7) bor ((C band 16#1) bsl 4))).
|
||||||
|
|
||||||
|
-define(is_required(A), (A =< 16#7fff)).
|
||||||
|
|
||||||
|
-define(STUN_METHOD_BINDING, 16#001).
|
||||||
|
|
||||||
|
%% Comprehension-required range (0x0000-0x7FFF)
|
||||||
|
-define(STUN_ATTR_MAPPED_ADDRESS, 16#0001).
|
||||||
|
-define(STUN_ATTR_USERNAME, 16#0006).
|
||||||
|
-define(STUN_ATTR_MESSAGE_INTEGRITY, 16#0008).
|
||||||
|
-define(STUN_ATTR_ERROR_CODE, 16#0009).
|
||||||
|
-define(STUN_ATTR_UNKNOWN_ATTRIBUTES, 16#000a).
|
||||||
|
-define(STUN_ATTR_REALM, 16#0014).
|
||||||
|
-define(STUN_ATTR_NONCE, 16#0015).
|
||||||
|
-define(STUN_ATTR_XOR_MAPPED_ADDRESS, 16#0020).
|
||||||
|
|
||||||
|
%% Comprehension-optional range (0x8000-0xFFFF)
|
||||||
|
-define(STUN_ATTR_SOFTWARE, 16#8022).
|
||||||
|
-define(STUN_ATTR_ALTERNATE_SERVER, 16#8023).
|
||||||
|
-define(STUN_ATTR_FINGERPRINT, 16#8028).
|
||||||
|
|
||||||
|
-record(stun, {class,
|
||||||
|
method,
|
||||||
|
magic = ?STUN_MAGIC,
|
||||||
|
trid,
|
||||||
|
unsupported = [],
|
||||||
|
'SOFTWARE',
|
||||||
|
'ALTERNATE-SERVER',
|
||||||
|
'MAPPED-ADDRESS',
|
||||||
|
'XOR-MAPPED-ADDRESS',
|
||||||
|
'USERNAME',
|
||||||
|
'REALM',
|
||||||
|
'NONCE',
|
||||||
|
'MESSAGE-INTEGRITY',
|
||||||
|
'ERROR-CODE',
|
||||||
|
'UNKNOWN-ATTRIBUTES' = []}).
|
||||||
|
|
||||||
|
%% Workarounds.
|
||||||
|
%%-define(NO_PADDING, true).
|
||||||
343
src/stun/stun_codec.erl
Normal file
343
src/stun/stun_codec.erl
Normal file
@ -0,0 +1,343 @@
|
|||||||
|
%%%-------------------------------------------------------------------
|
||||||
|
%%% File : stun_codec.erl
|
||||||
|
%%% Author : Evgeniy Khramtsov <ekhramtsov@process-one.net>
|
||||||
|
%%% Description : STUN codec
|
||||||
|
%%% Created : 7 Aug 2009 by Evgeniy Khramtsov <ekhramtsov@process-one.net>
|
||||||
|
%%%
|
||||||
|
%%%
|
||||||
|
%%% ejabberd, Copyright (C) 2002-2009 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., 59 Temple Place, Suite 330, Boston, MA
|
||||||
|
%%% 02111-1307 USA
|
||||||
|
%%%
|
||||||
|
%%%-------------------------------------------------------------------
|
||||||
|
-module(stun_codec).
|
||||||
|
|
||||||
|
%% API
|
||||||
|
-export([decode/1,
|
||||||
|
encode/1,
|
||||||
|
version/1,
|
||||||
|
reason/1,
|
||||||
|
pp/1]).
|
||||||
|
|
||||||
|
%% Tests
|
||||||
|
-export([test_udp/2,
|
||||||
|
test_tcp/2,
|
||||||
|
test_tls/2,
|
||||||
|
test_public/0]).
|
||||||
|
|
||||||
|
-include("stun.hrl").
|
||||||
|
|
||||||
|
%%====================================================================
|
||||||
|
%% API
|
||||||
|
%%====================================================================
|
||||||
|
decode(<<0:2, Type:14, Len:16, Magic:32, TrID:96,
|
||||||
|
Body:Len/binary, Tail/binary>>) ->
|
||||||
|
case catch decode(Type, Magic, TrID, Body) of
|
||||||
|
{'EXIT', _} ->
|
||||||
|
{error, unparsed};
|
||||||
|
Res ->
|
||||||
|
{ok, Res, Tail}
|
||||||
|
end;
|
||||||
|
decode(<<0:2, _/binary>>) ->
|
||||||
|
more;
|
||||||
|
decode(<<>>) ->
|
||||||
|
empty;
|
||||||
|
decode(_) ->
|
||||||
|
{error, unparsed}.
|
||||||
|
|
||||||
|
encode(#stun{class = Class,
|
||||||
|
method = Method,
|
||||||
|
magic = Magic,
|
||||||
|
trid = TrID} = Msg) ->
|
||||||
|
ClassCode = case Class of
|
||||||
|
request -> 0;
|
||||||
|
indication -> 1;
|
||||||
|
response -> 2;
|
||||||
|
error -> 3
|
||||||
|
end,
|
||||||
|
Type = ?STUN_TYPE(ClassCode, Method),
|
||||||
|
Attrs = enc_attrs(Msg),
|
||||||
|
Len = size(Attrs),
|
||||||
|
<<0:2, Type:14, Len:16, Magic:32, TrID:96, Attrs/binary>>.
|
||||||
|
|
||||||
|
pp(Term) ->
|
||||||
|
io_lib_pretty:print(Term, fun pp/2).
|
||||||
|
|
||||||
|
version(#stun{magic = ?STUN_MAGIC}) ->
|
||||||
|
new;
|
||||||
|
version(#stun{}) ->
|
||||||
|
old.
|
||||||
|
|
||||||
|
reason(300) -> <<"Try Alternate">>;
|
||||||
|
reason(400) -> <<"Bad Request">>;
|
||||||
|
reason(401) -> <<"Unauthorized">>;
|
||||||
|
reason(420) -> <<"Unknown Attribute">>;
|
||||||
|
reason(438) -> <<"Stale Nonce">>;
|
||||||
|
reason(500) -> <<"Server Error">>;
|
||||||
|
reason(_) -> <<"Undefined Error">>.
|
||||||
|
|
||||||
|
%%====================================================================
|
||||||
|
%% Internal functions
|
||||||
|
%%====================================================================
|
||||||
|
decode(Type, Magic, TrID, Body) ->
|
||||||
|
Method = ?STUN_METHOD(Type),
|
||||||
|
Class = case ?STUN_CLASS(Type) of
|
||||||
|
0 -> request;
|
||||||
|
1 -> indication;
|
||||||
|
2 -> response;
|
||||||
|
3 -> error
|
||||||
|
end,
|
||||||
|
dec_attrs(Body, #stun{class = Class,
|
||||||
|
method = Method,
|
||||||
|
magic = Magic,
|
||||||
|
trid = TrID}).
|
||||||
|
|
||||||
|
dec_attrs(<<Type:16, Len:16, Rest/binary>>, Msg) ->
|
||||||
|
PaddLen = padd_len(Len),
|
||||||
|
<<Val:Len/binary, _:PaddLen, Tail/binary>> = Rest,
|
||||||
|
NewMsg = dec_attr(Type, Val, Msg),
|
||||||
|
if Type == ?STUN_ATTR_MESSAGE_INTEGRITY ->
|
||||||
|
NewMsg;
|
||||||
|
true ->
|
||||||
|
dec_attrs(Tail, NewMsg)
|
||||||
|
end;
|
||||||
|
dec_attrs(<<>>, Msg) ->
|
||||||
|
Msg.
|
||||||
|
|
||||||
|
enc_attrs(Msg) ->
|
||||||
|
concat_binary(
|
||||||
|
[enc_attr(?STUN_ATTR_SOFTWARE, Msg#stun.'SOFTWARE'),
|
||||||
|
enc_addr(?STUN_ATTR_MAPPED_ADDRESS, Msg#stun.'MAPPED-ADDRESS'),
|
||||||
|
enc_xor_addr(?STUN_ATTR_XOR_MAPPED_ADDRESS,
|
||||||
|
Msg#stun.magic, Msg#stun.trid,
|
||||||
|
Msg#stun.'XOR-MAPPED-ADDRESS'),
|
||||||
|
enc_addr(?STUN_ATTR_ALTERNATE_SERVER, Msg#stun.'ALTERNATE-SERVER'),
|
||||||
|
enc_attr(?STUN_ATTR_USERNAME, Msg#stun.'USERNAME'),
|
||||||
|
enc_attr(?STUN_ATTR_REALM, Msg#stun.'REALM'),
|
||||||
|
enc_attr(?STUN_ATTR_NONCE, Msg#stun.'NONCE'),
|
||||||
|
enc_error_code(Msg#stun.'ERROR-CODE'),
|
||||||
|
enc_unknown_attrs(Msg#stun.'UNKNOWN-ATTRIBUTES')]).
|
||||||
|
|
||||||
|
dec_attr(?STUN_ATTR_MAPPED_ADDRESS, Val, Msg) ->
|
||||||
|
<<_, Family, Port:16, AddrBin/binary>> = Val,
|
||||||
|
Addr = dec_addr(Family, AddrBin),
|
||||||
|
Msg#stun{'MAPPED-ADDRESS' = {Addr, Port}};
|
||||||
|
dec_attr(?STUN_ATTR_XOR_MAPPED_ADDRESS, Val, Msg) ->
|
||||||
|
<<_, Family, XPort:16, XAddr/binary>> = Val,
|
||||||
|
Magic = Msg#stun.magic,
|
||||||
|
Port = XPort bxor (Magic bsr 16),
|
||||||
|
Addr = dec_xor_addr(Family, Magic, Msg#stun.trid, XAddr),
|
||||||
|
Msg#stun{'XOR-MAPPED-ADDRESS' = {Addr, Port}};
|
||||||
|
dec_attr(?STUN_ATTR_SOFTWARE, Val, Msg) ->
|
||||||
|
Msg#stun{'SOFTWARE' = Val};
|
||||||
|
dec_attr(?STUN_ATTR_USERNAME, Val, Msg) ->
|
||||||
|
Msg#stun{'USERNAME' = Val};
|
||||||
|
dec_attr(?STUN_ATTR_REALM, Val, Msg) ->
|
||||||
|
Msg#stun{'REALM' = Val};
|
||||||
|
dec_attr(?STUN_ATTR_NONCE, Val, Msg) ->
|
||||||
|
Msg#stun{'NONCE' = Val};
|
||||||
|
dec_attr(?STUN_ATTR_MESSAGE_INTEGRITY, Val, Msg) ->
|
||||||
|
Msg#stun{'MESSAGE-INTEGRITY' = Val};
|
||||||
|
dec_attr(?STUN_ATTR_ALTERNATE_SERVER, Val, Msg) ->
|
||||||
|
<<_, Family, Port:16, Address/binary>> = Val,
|
||||||
|
IP = dec_addr(Family, Address),
|
||||||
|
Msg#stun{'ALTERNATE-SERVER' = {IP, Port}};
|
||||||
|
dec_attr(?STUN_ATTR_ERROR_CODE, Val, Msg) ->
|
||||||
|
<<_:21, Class:3, Number:8, Reason/binary>> = Val,
|
||||||
|
if Class >=3, Class =< 6, Number >=0, Number =< 99 ->
|
||||||
|
Code = Class * 100 + Number,
|
||||||
|
Msg#stun{'ERROR-CODE' = {Code, Reason}}
|
||||||
|
end;
|
||||||
|
dec_attr(?STUN_ATTR_UNKNOWN_ATTRIBUTES, Val, Msg) ->
|
||||||
|
Attrs = dec_unknown_attrs(Val, []),
|
||||||
|
Msg#stun{'UNKNOWN-ATTRIBUTES' = Attrs};
|
||||||
|
dec_attr(Attr, _Val, #stun{unsupported = Attrs} = Msg)
|
||||||
|
when Attr =< 16#7fff ->
|
||||||
|
Msg#stun{unsupported = [Attr|Attrs]};
|
||||||
|
dec_attr(_Attr, _Val, Msg) ->
|
||||||
|
Msg.
|
||||||
|
|
||||||
|
dec_addr(1, <<A1, A2, A3, A4>>) ->
|
||||||
|
{A1, A2, A3, A4};
|
||||||
|
dec_addr(2, <<A1:16, A2:16, A3:16, A4:16,
|
||||||
|
A5:16, A6:16, A7:16, A8:16>>) ->
|
||||||
|
{A1, A2, A3, A4, A5, A6, A7, A8}.
|
||||||
|
|
||||||
|
dec_xor_addr(1, Magic, _TrID, <<XAddr:32>>) ->
|
||||||
|
Addr = XAddr bxor Magic,
|
||||||
|
dec_addr(1, <<Addr:32>>);
|
||||||
|
dec_xor_addr(2, Magic, TrID, <<XAddr:128>>) ->
|
||||||
|
Addr = XAddr bxor ((Magic bsl 96) bor TrID),
|
||||||
|
dec_addr(2, <<Addr:128>>).
|
||||||
|
|
||||||
|
dec_unknown_attrs(<<Attr:16, Tail/binary>>, Acc) ->
|
||||||
|
dec_unknown_attrs(Tail, [Attr|Acc]);
|
||||||
|
dec_unknown_attrs(<<>>, Acc) ->
|
||||||
|
lists:reverse(Acc).
|
||||||
|
|
||||||
|
enc_attr(_Attr, undefined) ->
|
||||||
|
<<>>;
|
||||||
|
enc_attr(Attr, Val) ->
|
||||||
|
Len = size(Val),
|
||||||
|
PaddLen = padd_len(Len),
|
||||||
|
<<Attr:16, Len:16, Val/binary, 0:PaddLen>>.
|
||||||
|
|
||||||
|
enc_addr(_Type, undefined) ->
|
||||||
|
<<>>;
|
||||||
|
enc_addr(Type, {{A1, A2, A3, A4}, Port}) ->
|
||||||
|
enc_attr(Type, <<0, 1, Port:16, A1, A2, A3, A4>>);
|
||||||
|
enc_addr(Type, {{A1, A2, A3, A4, A5, A6, A7, A8}, Port}) ->
|
||||||
|
enc_attr(Type, <<0, 2, Port:16, A1:16, A2:16, A3:16,
|
||||||
|
A4:16, A5:16, A6:16, A7:16, A8:16>>).
|
||||||
|
|
||||||
|
enc_xor_addr(_Type, _Magic, _TrID, undefined) ->
|
||||||
|
<<>>;
|
||||||
|
enc_xor_addr(Type, Magic, _TrID, {{A1, A2, A3, A4}, Port}) ->
|
||||||
|
XPort = Port bxor (Magic bsr 16),
|
||||||
|
<<Addr:32>> = <<A1, A2, A3, A4>>,
|
||||||
|
XAddr = Addr bxor Magic,
|
||||||
|
enc_attr(Type, <<0, 1, XPort:16, XAddr:32>>);
|
||||||
|
enc_xor_addr(Type, Magic, TrID,
|
||||||
|
{{A1, A2, A3, A4, A5, A6, A7, A8}, Port}) ->
|
||||||
|
XPort = Port bxor (Magic bsr 16),
|
||||||
|
<<Addr:128>> = <<A1:16, A2:16, A3:16, A4:16,
|
||||||
|
A5:16, A6:16, A7:16, A8:16>>,
|
||||||
|
XAddr = Addr bxor ((Magic bsl 96) bor TrID),
|
||||||
|
enc_attr(Type, <<0, 2, XPort:16, XAddr:128>>).
|
||||||
|
|
||||||
|
enc_error_code(undefined) ->
|
||||||
|
<<>>;
|
||||||
|
enc_error_code({Code, Reason}) ->
|
||||||
|
Class = Code div 100,
|
||||||
|
Number = Code rem 100,
|
||||||
|
enc_attr(?STUN_ATTR_ERROR_CODE,
|
||||||
|
<<0:21, Class:3, Number:8, Reason/binary>>).
|
||||||
|
|
||||||
|
enc_unknown_attrs([]) ->
|
||||||
|
<<>>;
|
||||||
|
enc_unknown_attrs(Attrs) ->
|
||||||
|
enc_attr(?STUN_ATTR_UNKNOWN_ATTRIBUTES,
|
||||||
|
concat_binary([<<Attr:16>> || Attr <- Attrs])).
|
||||||
|
|
||||||
|
%%====================================================================
|
||||||
|
%% Auxiliary functions
|
||||||
|
%%====================================================================
|
||||||
|
pp(Tag, N) ->
|
||||||
|
try
|
||||||
|
pp1(Tag, N)
|
||||||
|
catch _:_ ->
|
||||||
|
no
|
||||||
|
end.
|
||||||
|
|
||||||
|
pp1(stun, N) ->
|
||||||
|
N = record_info(size, stun) - 1,
|
||||||
|
record_info(fields, stun);
|
||||||
|
pp1(_, _) ->
|
||||||
|
no.
|
||||||
|
|
||||||
|
%% Workaround for stupid clients.
|
||||||
|
-ifdef(NO_PADDING).
|
||||||
|
padd_len(_Len) ->
|
||||||
|
0.
|
||||||
|
-else.
|
||||||
|
padd_len(Len) ->
|
||||||
|
case Len rem 4 of
|
||||||
|
0 -> 0;
|
||||||
|
N -> 8*(4-N)
|
||||||
|
end.
|
||||||
|
-endif.
|
||||||
|
|
||||||
|
%%====================================================================
|
||||||
|
%% Test functions
|
||||||
|
%%====================================================================
|
||||||
|
bind_msg() ->
|
||||||
|
Msg = #stun{method = ?STUN_METHOD_BINDING,
|
||||||
|
class = request,
|
||||||
|
trid = random:uniform(1 bsl 96),
|
||||||
|
'SOFTWARE' = <<"test">>},
|
||||||
|
encode(Msg).
|
||||||
|
|
||||||
|
test_udp(Addr, Port) ->
|
||||||
|
test(Addr, Port, gen_udp).
|
||||||
|
|
||||||
|
test_tcp(Addr, Port) ->
|
||||||
|
test(Addr, Port, gen_tcp).
|
||||||
|
|
||||||
|
test_tls(Addr, Port) ->
|
||||||
|
test(Addr, Port, ssl).
|
||||||
|
|
||||||
|
test(Addr, Port, Mod) ->
|
||||||
|
Res = case Mod of
|
||||||
|
gen_udp ->
|
||||||
|
Mod:open(0, [binary, {active, false}]);
|
||||||
|
_ ->
|
||||||
|
Mod:connect(Addr, Port,
|
||||||
|
[binary, {active, false}], 1000)
|
||||||
|
end,
|
||||||
|
case Res of
|
||||||
|
{ok, Sock} ->
|
||||||
|
if Mod == gen_udp ->
|
||||||
|
Mod:send(Sock, Addr, Port, bind_msg());
|
||||||
|
true ->
|
||||||
|
Mod:send(Sock, bind_msg())
|
||||||
|
end,
|
||||||
|
case Mod:recv(Sock, 0, 1000) of
|
||||||
|
{ok, {_, _, Data}} ->
|
||||||
|
try_dec(Data);
|
||||||
|
{ok, Data} ->
|
||||||
|
try_dec(Data);
|
||||||
|
Err ->
|
||||||
|
io:format("err: ~p~n", [Err])
|
||||||
|
end,
|
||||||
|
Mod:close(Sock);
|
||||||
|
Err ->
|
||||||
|
io:format("err: ~p~n", [Err])
|
||||||
|
end.
|
||||||
|
|
||||||
|
try_dec(Data) ->
|
||||||
|
case decode(Data) of
|
||||||
|
{ok, Msg, _} ->
|
||||||
|
io:format("got:~n~s~n", [pp(Msg)]);
|
||||||
|
Err ->
|
||||||
|
io:format("err: ~p~n", [Err])
|
||||||
|
end.
|
||||||
|
|
||||||
|
public_servers() ->
|
||||||
|
[{"stun.ekiga.net", 3478, 3478, 5349},
|
||||||
|
{"stun.fwdnet.net", 3478, 3478, 5349},
|
||||||
|
{"stun.ideasip.com", 3478, 3478, 5349},
|
||||||
|
{"stun01.sipphone.com", 3478, 3478, 5349},
|
||||||
|
{"stun.softjoys.com", 3478, 3478, 5349},
|
||||||
|
{"stun.voipbuster.com", 3478, 3478, 5349},
|
||||||
|
{"stun.voxgratia.org", 3478, 3478, 5349},
|
||||||
|
{"stun.xten.com", 3478, 3478, 5349},
|
||||||
|
{"stunserver.org", 3478, 3478, 5349},
|
||||||
|
{"stun.sipgate.net", 10000, 10000, 5349},
|
||||||
|
{"numb.viagenie.ca", 3478, 3478, 5349},
|
||||||
|
{"stun.ipshka.com", 3478, 3478, 5349},
|
||||||
|
{"localhost", 3478, 5349, 5349}].
|
||||||
|
|
||||||
|
test_public() ->
|
||||||
|
ssl:start(),
|
||||||
|
lists:foreach(
|
||||||
|
fun({Addr, UDPPort, TCPPort, TLSPort}) ->
|
||||||
|
io:format("trying ~s:~p on UDP... ", [Addr, UDPPort]),
|
||||||
|
test_udp(Addr, UDPPort),
|
||||||
|
io:format("trying ~s:~p on TCP... ", [Addr, TCPPort]),
|
||||||
|
test_tcp(Addr, TCPPort),
|
||||||
|
io:format("trying ~s:~p on TLS... ", [Addr, TLSPort]),
|
||||||
|
test_tls(Addr, TLSPort)
|
||||||
|
end, public_servers()).
|
||||||
Loading…
Reference in New Issue
Block a user