25
1
mirror of https://github.com/processone/ejabberd.git synced 2024-11-22 16:20:52 +01:00

Bugfix mod_muc odbc couldn't register nickname (EJAB-1364)

This commit is contained in:
Badlop 2010-12-23 01:35:46 +01:00
parent d3efcf8682
commit 93787bee73

View File

@ -39,7 +39,7 @@
%%% affiliation = owner | %%% affiliation = owner |
%%% reason = string() %%% reason = string()
%%% 3.0.0-alpha / mnesia / muc_registered %%% 3.0.0-alpha / mnesia / muc_registered
%%% us_host = {{Username::binary(), Host::binary()}, MucHost::binary()}} %%% user_host = {User::jid(), MucHost::binary()}}
%%% nick = binary() %%% nick = binary()
%%% %%%
%%% 3.0.0-alpha / odbc / muc_online_room %%% 3.0.0-alpha / odbc / muc_online_room
@ -57,7 +57,7 @@
%%% affiliation = text %%% affiliation = text
%%% reason = text %%% reason = text
%%% 3.0.0-alpha / mnesia / muc_registered %%% 3.0.0-alpha / mnesia / muc_registered
%%% us = text %%% user = text
%%% host = text %%% host = text
%%% nick = text %%% nick = text
%%% %%%
@ -96,7 +96,7 @@
-record(muc_room_opt, {name_host, opt, val}). -record(muc_room_opt, {name_host, opt, val}).
-record(muc_room_affiliation, {name_host, jid, affiliation, reason}). -record(muc_room_affiliation, {name_host, jid, affiliation, reason}).
-record(muc_online_room, {name_host, pid}). -record(muc_online_room, {name_host, pid}).
-record(muc_registered, {us_host, nick}). -record(muc_registered, {user_host, nick}).
-record(state, {host, -record(state, {host,
server_host, server_host,
@ -257,18 +257,17 @@ process_iq_disco_items(Host, From, To, #iq{lang = Lang} = IQ) when is_binary(Hos
can_use_nick(_Host, _JID, <<>>) -> can_use_nick(_Host, _JID, <<>>) ->
false; false;
can_use_nick(Host, JID, Nick) when is_binary(Host), is_binary(Nick) -> can_use_nick(Host, JID, Nick) when is_binary(Host), is_binary(Nick) ->
LUS = {exmpp_jid:prep_node(JID), exmpp_jid:prep_domain(JID)},
case catch gen_storage:dirty_select( case catch gen_storage:dirty_select(
Host, Host,
muc_registered, muc_registered,
[{'=', us_host, {'_', Host}}, [{'=', user_host, {'_', Host}},
{'=', nick, Nick}]) of {'=', nick, Nick}]) of
{'EXIT', _Reason} -> {'EXIT', _Reason} ->
true; true;
[] -> [] ->
true; true;
[#muc_registered{us_host = {U, _Host}}] -> [#muc_registered{user_host = {U, _Host}}] ->
U == LUS U == exmpp_jid:bare(JID)
end. end.
migrate(After) -> migrate(After) ->
@ -322,7 +321,7 @@ init([Host, Opts]) ->
[{disc_copies, [node()]}, [{disc_copies, [node()]},
{odbc_host, Host}, {odbc_host, Host},
{attributes, record_info(fields, muc_registered)}, {attributes, record_info(fields, muc_registered)},
{types, [{us_host, {text, text}}]}]), {types, [{user_host, {jid, text}}]}]),
gen_storage:create_table(Backend, MyHost, muc_online_room, gen_storage:create_table(Backend, MyHost, muc_online_room,
[{ram_copies, [node()]}, [{ram_copies, [node()]},
{odbc_host, Host}, {odbc_host, Host},
@ -854,11 +853,9 @@ flush() ->
children = [#xmlcdata{cdata = Val}]}]}). children = [#xmlcdata{cdata = Val}]}]}).
iq_get_register_info(Host, From, Lang) -> iq_get_register_info(Host, From, Lang) ->
LUser = exmpp_jid:prep_node(From), FromBare = exmpp_jid:bare(From),
LServer = exmpp_jid:prep_domain(From),
LUS = {LUser, LServer},
{Nick, Registered} = {Nick, Registered} =
case catch gen_storage:dirty_read(Host, muc_registered, {LUS, Host}) of case catch gen_storage:dirty_read(Host, muc_registered, {FromBare, Host}) of
{'EXIT', _Reason} -> {'EXIT', _Reason} ->
{"", []}; {"", []};
[] -> [] ->
@ -884,31 +881,29 @@ iq_get_register_info(Host, From, Lang) ->
iq_set_register_info(Host, From, Nick, Lang) when is_binary(Host), is_binary(Nick) -> iq_set_register_info(Host, From, Nick, Lang) when is_binary(Host), is_binary(Nick) ->
LUser = exmpp_jid:prep_node(From), FromBare = exmpp_jid:bare(From),
LServer = exmpp_jid:prep_domain(From),
LUS = {LUser, LServer},
F = fun() -> F = fun() ->
case Nick of case Nick of
<<>> -> <<>> ->
gen_storage:delete(Host, {muc_registered, {LUS, Host}}), gen_storage:delete(Host, {muc_registered, {FromBare, Host}}),
ok; ok;
_ -> _ ->
Allow = Allow =
case gen_storage:select( case gen_storage:select(
Host, Host,
muc_registered, muc_registered,
[{'=', us_host, {'_', Host}}, [{'=', user_host, {'_', Host}},
{'=', nick, Nick}]) of {'=', nick, Nick}]) of
[] -> [] ->
true; true;
[#muc_registered{us_host = {U, _Host}}] -> [#muc_registered{user_host = {U, _Host}}] ->
U == LUS U == FromBare
end, end,
if if
Allow -> Allow ->
gen_storage:write( gen_storage:write(
Host, Host,
#muc_registered{us_host = {LUS, Host}, #muc_registered{user_host = {FromBare, Host},
nick = Nick}), nick = Nick}),
ok; ok;
true -> true ->