Chapril
/
xmpp.chapril.org-ejabberd
mirror of https://github.com/processone/ejabberd.git
24
1
Fork 0
Code Releases Activity

Check for 'max_user_sessions' option

This commit is contained in:
Evgeniy Khramtsov 2014-05-01 20:58:14 +04:00
parent 0117787317
commit 8e2bc8d19e
2 changed files with 13 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/ejabberd_sm.erl
View File

@ -56,6 +56,7 @@
get_session_pid/3,
get_user_info/3,
get_user_ip/3,
get_max_user_sessions/2,
is_existing_resource/3
]).

17
src/mod_sip_registrar.erl
View File

@ -198,7 +198,7 @@ unregister_session(US, SIPSocket, CallID, CSeq) ->
Msg = {delete, US, SIPSocket, CallID, CSeq},
call(Msg).
write_session(#sip_session{us = US,
write_session(#sip_session{us = {U, S} = US,
bindings = [#binding{socket = SIPSocket,
call_id = CallID,
expires = Expires,
@ -216,10 +216,15 @@ write_session(#sip_session{us = US,
mnesia:dirty_write(
#sip_session{us = US, bindings = NewBindings});
{error, notfound} ->
NewTRef = erlang:start_timer(Expires * 1000, self(), US),
NewBindings = [Binding#binding{tref = NewTRef}|Bindings],
mnesia:dirty_write(
#sip_session{us = US, bindings = NewBindings})
MaxSessions = ejabberd_sm:get_max_user_sessions(U, S),
if length(Bindings) < MaxSessions ->
NewTRef = erlang:start_timer(Expires * 1000, self(), US),
NewBindings = [Binding#binding{tref = NewTRef}|Bindings],
mnesia:dirty_write(
#sip_session{us = US, bindings = NewBindings});
true ->
{error, too_many_sessions}
end
end;
[] ->
NewTRef = erlang:start_timer(Expires * 1000, self(), US),
@ -307,5 +312,7 @@ make_status(cseq_out_of_order) ->
{500, <<"CSeq is Out of Order">>};
make_status(timeout) ->
{408, esip:reason(408)};
make_status(too_many_sessions) ->
{503, <<"Too Many Registered Sessions">>};
make_status(_) ->
{500, esip:reason(500)}.