Chapril/xmpp.chapril.org-ejabberd
25
1
Fork 0
mirror of https://github.com/processone/ejabberd.git synced 2024-11-30 16:36:29 +01:00
Code Releases Activity

Move websocket options

Browse Source
This commit is contained in:
Evgeny Khramtsov 2019-04-23 16:21:06 +03:00
parent cc892ddc01
commit 5faae61bef
2 changed files with 22 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/ejabberd_http_ws.erl
View File

@ -24,7 +24,6 @@
%%%---------------------------------------------------------------------- %%%----------------------------------------------------------------------
-module(ejabberd_http_ws). -module(ejabberd_http_ws).
-author('ecestari@process-one.net'). -author('ecestari@process-one.net').
-behaviour(ejabberd_config).
-behaviour(xmpp_socket). -behaviour(xmpp_socket).
-behaviour(p1_fsm). -behaviour(p1_fsm).
@ -33,7 +32,7 @@
terminate/3, send_xml/2, setopts/2, sockname/1, terminate/3, send_xml/2, setopts/2, sockname/1,
peername/1, controlling_process/2, get_owner/1, peername/1, controlling_process/2, get_owner/1,
reset_stream/1, close/1, change_shaper/2, reset_stream/1, close/1, change_shaper/2,
socket_handoff/3, get_transport/1, opt_type/1]). socket_handoff/3, get_transport/1]).
-include("logger.hrl"). -include("logger.hrl").
@ -367,12 +366,3 @@ parsed_items(List) ->
after 0 -> after 0 ->
lists:reverse(List) lists:reverse(List)
end. end.
opt_type(websocket_ping_interval) ->
fun (I) when is_integer(I), I >= 0 -> I end;
opt_type(websocket_timeout) ->
fun (I) when is_integer(I), I > 0 -> I end;
opt_type(websocket_origin) ->
fun (O) -> O end;
opt_type(_) ->
[websocket_ping_interval, websocket_timeout, websocket_origin].

24
src/ejabberd_websocket.erl
View File

@ -37,12 +37,12 @@
%%%---------------------------------------------------------------------- %%%----------------------------------------------------------------------
-module(ejabberd_websocket). -module(ejabberd_websocket).
-behaviour(ejabberd_config).
-protocol({rfc, 6455}). -protocol({rfc, 6455}).
-author('ecestari@process-one.net'). -author('ecestari@process-one.net').
-export([check/2, socket_handoff/5]). -export([check/2, socket_handoff/5, opt_type/1]).
-include("logger.hrl"). -include("logger.hrl").
@ -409,4 +409,22 @@ websocket_close(Socket, WsHandleLoopPid, SocketMode, _CloseCode) ->
SocketMode:close(Socket). SocketMode:close(Socket).
get_origin() -> get_origin() ->
ejabberd_config:get_option({websocket_origin, ejabberd_config:get_myname()}, ignore). ejabberd_config:get_option(websocket_origin, ignore).
opt_type(websocket_ping_interval) ->
fun (I) when is_integer(I), I >= 0 -> I end;
opt_type(websocket_timeout) ->
fun (I) when is_integer(I), I > 0 -> I end;
opt_type(websocket_origin) ->
%% Accept only values conforming to RFC6454 section 7.1
fun (<<"null">>) -> <<"null">>;
(null) -> <<"null">>;
(Origin) ->
URIs = [_|_] = lists:flatmap(
fun(<<>>) -> [];
(URI) -> [misc:try_url(URI)]
end, re:split(Origin, "\\s")),
str:join(URIs, <<" ">>)
end;
opt_type(_) ->
[websocket_ping_interval, websocket_timeout, websocket_origin].