diff options
Diffstat (limited to 'src/stun/stun_codec.erl')
| -rw-r--r-- | src/stun/stun_codec.erl | 282 |
1 files changed, 122 insertions, 160 deletions
diff --git a/src/stun/stun_codec.erl b/src/stun/stun_codec.erl index 2f4723dca..4d489e070 100644 --- a/src/stun/stun_codec.erl +++ b/src/stun/stun_codec.erl @@ -26,16 +26,11 @@ -module(stun_codec). %% API --export([decode/1, - encode/1, - version/1, - reason/1, +-export([decode/1, encode/1, version/1, reason/1, pp/1]). %% Tests --export([test_udp/2, - test_tcp/2, - test_tls/2, +-export([test_udp/2, test_tcp/2, test_tls/2, test_public/0]). -include("stun.hrl"). @@ -44,42 +39,34 @@ %% API %%==================================================================== decode(<<0:2, Type:14, Len:16, Magic:32, TrID:96, - Body:Len/binary, Tail/binary>>) -> + Body:Len/binary, Tail/binary>>) -> case catch decode(Type, Magic, TrID, Body) of - {'EXIT', _} -> - {error, unparsed}; - Res -> - {ok, Res, Tail} + {'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) -> +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 + request -> 0; + indication -> 1; + response -> 2; + error -> 3 end, - Type = ?STUN_TYPE(ClassCode, Method), + Type = (?STUN_TYPE(ClassCode, Method)), Attrs = enc_attrs(Msg), - Len = size(Attrs), - <<0:2, Type:14, Len:16, Magic:32, TrID:96, Attrs/binary>>. + Len = byte_size(Attrs), + <<0:2, Type:14, Len:16, Magic:32, TrID:96, + Attrs/binary>>. -pp(Term) -> - io_lib_pretty:print(Term, fun pp/2). +pp(Term) -> io_lib_pretty:print(Term, fun pp/2). -version(#stun{magic = ?STUN_MAGIC}) -> - new; -version(#stun{}) -> - old. +version(#stun{magic = ?STUN_MAGIC}) -> new; +version(#stun{}) -> old. reason(300) -> <<"Try Alternate">>; reason(400) -> <<"Bad Request">>; @@ -93,43 +80,41 @@ reason(_) -> <<"Undefined Error">>. %% Internal functions %%==================================================================== decode(Type, Magic, TrID, Body) -> - Method = ?STUN_METHOD(Type), + Method = (?STUN_METHOD(Type)), Class = case ?STUN_CLASS(Type) of - 0 -> request; - 1 -> indication; - 2 -> response; - 3 -> error + 0 -> request; + 1 -> indication; + 2 -> response; + 3 -> error end, - dec_attrs(Body, #stun{class = Class, - method = Method, - magic = Magic, - trid = TrID}). + 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) + if Type == (?STUN_ATTR_MESSAGE_INTEGRITY) -> NewMsg; + true -> dec_attrs(Tail, NewMsg) end; -dec_attrs(<<>>, Msg) -> - Msg. +dec_attrs(<<>>, Msg) -> Msg. enc_attrs(Msg) -> - list_to_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')]). + iolist_to_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, @@ -139,7 +124,8 @@ 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), + 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}; @@ -157,55 +143,52 @@ dec_attr(?STUN_ATTR_ALTERNATE_SERVER, Val, Msg) -> 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}} + 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>>) -> + when Attr =< 32767 -> + 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>>); + 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), + 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). + dec_unknown_attrs(Tail, [Attr | Acc]); +dec_unknown_attrs(<<>>, Acc) -> lists:reverse(Acc). -enc_attr(_Attr, undefined) -> - <<>>; +enc_attr(_Attr, undefined) -> <<>>; enc_attr(Attr, Val) -> - Len = size(Val), + Len = byte_size(Val), PaddLen = padd_len(Len), <<Attr:16, Len:16, Val/binary, 0:PaddLen>>. -enc_addr(_Type, undefined) -> - <<>>; +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_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}) -> +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, @@ -213,51 +196,46 @@ enc_xor_addr(Type, Magic, _TrID, {{A1, A2, A3, A4}, Port}) -> 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), + <<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(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([]) -> <<>>; enc_unknown_attrs(Attrs) -> enc_attr(?STUN_ATTR_UNKNOWN_ATTRIBUTES, - list_to_binary([<<Attr:16>> || Attr <- Attrs])). - %%==================================================================== %% Auxiliary functions %%==================================================================== -pp(Tag, N) -> - try - pp1(Tag, N) - catch _:_ -> - no - end. + iolist_to_binary([<<Attr:16>> || Attr <- Attrs])). + +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. +pp1(_, _) -> no. %% Workaround for stupid clients. -ifdef(NO_PADDING). -padd_len(_Len) -> - 0. + +padd_len(_Len) -> 0. + -else. + padd_len(Len) -> case Len rem 4 of - 0 -> 0; - N -> 8*(4-N) + 0 -> 0; + N -> 8 * (4 - N) end. + -endif. %%==================================================================== @@ -265,65 +243,49 @@ padd_len(Len) -> %%==================================================================== bind_msg() -> Msg = #stun{method = ?STUN_METHOD_BINDING, - class = request, - trid = random:uniform(1 bsl 96), + class = request, trid = random:uniform(1 bsl 96), 'SOFTWARE' = <<"test">>}, encode(Msg). -test_udp(Addr, Port) -> - test(Addr, Port, gen_udp). +test_udp(Addr, Port) -> test(Addr, Port, gen_udp). -test_tcp(Addr, Port) -> - test(Addr, Port, gen_tcp). +test_tcp(Addr, Port) -> test(Addr, Port, gen_tcp). -test_tls(Addr, Port) -> - test(Addr, Port, ssl). +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) + 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]) + {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]) + {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}, @@ -332,12 +294,12 @@ public_servers() -> 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()). + 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()). |