mirror of
https://github.com/processone/ejabberd.git
synced 2024-12-08 16:52:59 +01:00
b154f75523
* src/ejabberd_auth.erl: LDAP attribute that holds user ID now configurable * src/ejabberd.cfg.example: Updated SVN Revision: 178
360 lines
8.7 KiB
Erlang
360 lines
8.7 KiB
Erlang
%%%----------------------------------------------------------------------
|
|
%%% File : ejabberd_auth.erl
|
|
%%% Author : Alexey Shchepin <alexey@sevcom.net>
|
|
%%% Purpose :
|
|
%%% Created : 23 Nov 2002 by Alexey Shchepin <alexey@sevcom.net>
|
|
%%% Id : $Id$
|
|
%%%----------------------------------------------------------------------
|
|
|
|
-module(ejabberd_auth).
|
|
-author('alexey@sevcom.net').
|
|
-vsn('$Revision$ ').
|
|
|
|
-behaviour(gen_server).
|
|
|
|
%% External exports
|
|
-export([start/0, start_link/0,
|
|
set_password/2,
|
|
check_password/2,
|
|
check_password/4,
|
|
try_register/2,
|
|
dirty_get_registered_users/0,
|
|
get_password/1,
|
|
get_password_s/1,
|
|
is_user_exists/1,
|
|
remove_user/1,
|
|
remove_user/2,
|
|
plain_password_required/0,
|
|
check_password_ldap/2, % TODO: remove
|
|
is_user_exists_ldap/1 % TODO: remove
|
|
]).
|
|
|
|
%% gen_server callbacks
|
|
-export([init/1,
|
|
handle_call/3,
|
|
handle_cast/2,
|
|
code_change/3,
|
|
handle_info/2,
|
|
terminate/2]).
|
|
|
|
-include("eldap/eldap.hrl").
|
|
|
|
-record(state, {}).
|
|
|
|
-record(passwd, {user, password}).
|
|
|
|
%%%----------------------------------------------------------------------
|
|
%%% API
|
|
%%%----------------------------------------------------------------------
|
|
start() ->
|
|
gen_server:start({local, ejabberd_auth}, ejabberd_auth, [], []).
|
|
start_link() ->
|
|
gen_server:start_link({local, ejabberd_auth}, ejabberd_auth, [], []).
|
|
|
|
%%%----------------------------------------------------------------------
|
|
%%% Callback functions from gen_server
|
|
%%%----------------------------------------------------------------------
|
|
|
|
%%----------------------------------------------------------------------
|
|
%% Func: init/1
|
|
%% Returns: {ok, State} |
|
|
%% {ok, State, Timeout} |
|
|
%% ignore |
|
|
%% {stop, Reason}
|
|
%%----------------------------------------------------------------------
|
|
init([]) ->
|
|
mnesia:create_table(passwd,[{disc_copies, [node()]},
|
|
{attributes, record_info(fields, passwd)}]),
|
|
case auth_method() of
|
|
internal ->
|
|
ok;
|
|
ldap ->
|
|
LDAPServers = ejabberd_config:get_local_option(ldap_servers),
|
|
eldap:start_link("ejabberd", LDAPServers, 389, "", "")
|
|
end,
|
|
{ok, #state{}}.
|
|
|
|
%%----------------------------------------------------------------------
|
|
%% Func: handle_call/3
|
|
%% Returns: {reply, Reply, State} |
|
|
%% {reply, Reply, State, Timeout} |
|
|
%% {noreply, State} |
|
|
%% {noreply, State, Timeout} |
|
|
%% {stop, Reason, Reply, State} | (terminate/2 is called)
|
|
%% {stop, Reason, State} (terminate/2 is called)
|
|
%%----------------------------------------------------------------------
|
|
handle_call(_Request, _From, State) ->
|
|
Reply = ok,
|
|
{reply, Reply, State}.
|
|
|
|
%%----------------------------------------------------------------------
|
|
%% Func: handle_cast/2
|
|
%% Returns: {noreply, State} |
|
|
%% {noreply, State, Timeout} |
|
|
%% {stop, Reason, State} (terminate/2 is called)
|
|
%%----------------------------------------------------------------------
|
|
handle_cast(_Msg, State) ->
|
|
{noreply, State}.
|
|
|
|
|
|
code_change(_OldVsn, State, _Extra) ->
|
|
{ok, State}.
|
|
|
|
%%----------------------------------------------------------------------
|
|
%% Func: handle_info/2
|
|
%% Returns: {noreply, State} |
|
|
%% {noreply, State, Timeout} |
|
|
%% {stop, Reason, State} (terminate/2 is called)
|
|
%%----------------------------------------------------------------------
|
|
handle_info(_Info, State) ->
|
|
{noreply, State}.
|
|
|
|
%%----------------------------------------------------------------------
|
|
%% Func: terminate/2
|
|
%% Purpose: Shutdown the server
|
|
%% Returns: any (ignored by gen_server)
|
|
%%----------------------------------------------------------------------
|
|
terminate(_Reason, _State) ->
|
|
ok.
|
|
|
|
%%%----------------------------------------------------------------------
|
|
%%% Internal functions
|
|
%%%----------------------------------------------------------------------
|
|
|
|
auth_method() ->
|
|
case ejabberd_config:get_local_option(auth_method) of
|
|
ldap ->
|
|
ldap;
|
|
_ ->
|
|
internal
|
|
end.
|
|
|
|
plain_password_required() ->
|
|
case auth_method() of
|
|
internal ->
|
|
false;
|
|
ldap ->
|
|
true
|
|
end.
|
|
|
|
|
|
check_password(User, Password) ->
|
|
case auth_method() of
|
|
internal ->
|
|
check_password_internal(User, Password);
|
|
ldap ->
|
|
check_password_ldap(User, Password)
|
|
end.
|
|
|
|
check_password_internal(User, Password) ->
|
|
LUser = jlib:nodeprep(User),
|
|
case catch mnesia:dirty_read({passwd, LUser}) of
|
|
[#passwd{password = Password}] ->
|
|
true;
|
|
_ ->
|
|
false
|
|
end.
|
|
|
|
check_password(User, Password, StreamID, Digest) ->
|
|
case auth_method() of
|
|
internal ->
|
|
check_password_internal(User, Password, StreamID, Digest);
|
|
ldap ->
|
|
check_password_ldap(User, Password, StreamID, Digest)
|
|
end.
|
|
|
|
check_password_internal(User, Password, StreamID, Digest) ->
|
|
LUser = jlib:nodeprep(User),
|
|
case catch mnesia:dirty_read({passwd, LUser}) of
|
|
[#passwd{password = Passwd}] ->
|
|
DigRes = if
|
|
Digest /= "" ->
|
|
Digest == sha:sha(StreamID ++ Passwd);
|
|
true ->
|
|
false
|
|
end,
|
|
if DigRes ->
|
|
true;
|
|
true ->
|
|
Passwd == Password
|
|
end;
|
|
_ ->
|
|
false
|
|
end.
|
|
|
|
|
|
set_password(User, Password) ->
|
|
case jlib:nodeprep(User) of
|
|
error -> {error, invalid_jid};
|
|
LUser ->
|
|
F = fun() ->
|
|
mnesia:write(#passwd{user = LUser,
|
|
password = Password})
|
|
end,
|
|
mnesia:transaction(F)
|
|
end.
|
|
|
|
|
|
try_register(User, Password) ->
|
|
case auth_method() of
|
|
internal ->
|
|
try_register_internal(User, Password);
|
|
ldap ->
|
|
{error, not_allowed}
|
|
end.
|
|
|
|
try_register_internal(User, Password) ->
|
|
case jlib:nodeprep(User) of
|
|
error -> {error, invalid_jid};
|
|
LUser ->
|
|
F = fun() ->
|
|
case mnesia:read({passwd, LUser}) of
|
|
[] ->
|
|
mnesia:write(#passwd{user = LUser,
|
|
password = Password}),
|
|
ok;
|
|
[_E] ->
|
|
exists
|
|
end
|
|
end,
|
|
mnesia:transaction(F)
|
|
end.
|
|
|
|
dirty_get_registered_users() ->
|
|
mnesia:dirty_all_keys(passwd).
|
|
|
|
get_password(User) ->
|
|
LUser = jlib:nodeprep(User),
|
|
case catch mnesia:dirty_read(passwd, LUser) of
|
|
[#passwd{password = Password}] ->
|
|
Password;
|
|
_ ->
|
|
false
|
|
end.
|
|
|
|
get_password_s(User) ->
|
|
LUser = jlib:nodeprep(User),
|
|
case catch mnesia:dirty_read(passwd, LUser) of
|
|
[#passwd{password = Password}] ->
|
|
Password;
|
|
_ ->
|
|
[]
|
|
end.
|
|
|
|
is_user_exists(User) ->
|
|
case auth_method() of
|
|
internal ->
|
|
is_user_exists_internal(User);
|
|
ldap ->
|
|
is_user_exists_ldap(User)
|
|
end.
|
|
|
|
is_user_exists_internal(User) ->
|
|
LUser = jlib:nodeprep(User),
|
|
case catch mnesia:dirty_read({passwd, LUser}) of
|
|
[] ->
|
|
false;
|
|
[_] ->
|
|
true;
|
|
_ ->
|
|
false
|
|
end.
|
|
|
|
remove_user(User) ->
|
|
case auth_method() of
|
|
internal ->
|
|
remove_user_internal(User);
|
|
ldap ->
|
|
{error, not_allowed}
|
|
end.
|
|
|
|
remove_user_internal(User) ->
|
|
LUser = jlib:nodeprep(User),
|
|
F = fun() ->
|
|
mnesia:delete({passwd, LUser})
|
|
end,
|
|
mnesia:transaction(F),
|
|
catch mod_roster:remove_user(User),
|
|
catch mod_offline:remove_user(User),
|
|
catch mod_last:remove_user(User),
|
|
catch mod_vcard:remove_user(User),
|
|
catch mod_private:remove_user(User).
|
|
|
|
remove_user(User, Password) ->
|
|
case auth_method() of
|
|
internal ->
|
|
remove_user_internal(User, Password);
|
|
ldap ->
|
|
not_allowed
|
|
end.
|
|
|
|
remove_user_internal(User, Password) ->
|
|
LUser = jlib:nodeprep(User),
|
|
F = fun() ->
|
|
case mnesia:read({passwd, LUser}) of
|
|
[#passwd{password = Password}] ->
|
|
mnesia:delete({passwd, LUser}),
|
|
ok;
|
|
[_] ->
|
|
not_allowed;
|
|
_ ->
|
|
not_exists
|
|
end
|
|
end,
|
|
case mnesia:transaction(F) of
|
|
{atomic, ok} ->
|
|
catch mod_roster:remove_user(User),
|
|
catch mod_offline:remove_user(User),
|
|
catch mod_last:remove_user(User),
|
|
catch mod_vcard:remove_user(User),
|
|
catch mod_private:remove_user(User),
|
|
ok;
|
|
{atomic, Res} ->
|
|
Res;
|
|
_ ->
|
|
bad_request
|
|
end.
|
|
|
|
|
|
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
|
|
|
|
check_password_ldap(User, Password, StreamID, Digest) ->
|
|
check_password_ldap(User, Password).
|
|
|
|
check_password_ldap(User, Password) ->
|
|
case find_user_dn(User) of
|
|
false ->
|
|
false;
|
|
DN ->
|
|
case eldap:bind("ejabberd", DN, Password) of
|
|
ok ->
|
|
true;
|
|
_ ->
|
|
false
|
|
end
|
|
end.
|
|
|
|
is_user_exists_ldap(User) ->
|
|
case find_user_dn(User) of
|
|
false ->
|
|
false;
|
|
_DN ->
|
|
true
|
|
end.
|
|
|
|
find_user_dn(User) ->
|
|
Attr = ejabberd_config:get_local_option(ldap_uidattr),
|
|
Filter = eldap:equalityMatch(Attr, User),
|
|
Base = ejabberd_config:get_local_option(ldap_base),
|
|
case eldap:search("ejabberd", [{base, Base},
|
|
{filter, Filter},
|
|
{attributes, []}]) of
|
|
#eldap_search_result{entries = [E | _]} ->
|
|
E#eldap_entry.object_name;
|
|
_ ->
|
|
false
|
|
end.
|
|
|
|
|
|
|