%% ``The contents of this file are subject to the Erlang Public License,
%% Version 1.1, (the "License"); you may not use this file except in
%% compliance with the License. You should have received a copy of the
%% Erlang Public License along with this software. If not, it can be
%% retrieved via the world wide web at http://www.erlang.org/.
%%
%% Software distributed under the License is distributed on an "AS IS"
%% basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See
%% the License for the specific language governing rights and limitations
%% under the License.
%%
%% The Initial Developer of the Original Code is Ericsson Utvecklings AB.
%% Portions created by Ericsson are Copyright 1999, Ericsson Utvecklings
%% AB. All Rights Reserved.''
%%
%% $Id: mod_auth_dets.erl,v 1.1 2008/12/17 09:53:35 mikpe Exp $
%%
-module(mod_auth_dets).
%% dets authentication storage
-export([get_user/2,
list_group_members/2,
add_user/2,
add_group_member/3,
list_users/1,
delete_user/2,
list_groups/1,
delete_group_member/3,
delete_group/2,
remove/1]).
-export([store_directory_data/2]).
-include("httpd.hrl").
-include("mod_auth.hrl").
store_directory_data(Directory, DirData) ->
?CDEBUG("store_directory_data -> ~n"
" Directory: ~p~n"
" DirData: ~p",
[Directory, DirData]),
PWFile = httpd_util:key1search(DirData, auth_user_file),
GroupFile = httpd_util:key1search(DirData, auth_group_file),
Addr = httpd_util:key1search(DirData, bind_address),
Port = httpd_util:key1search(DirData, port),
PWName = httpd_util:make_name("httpd_dets_pwdb",Addr,Port),
case dets:open_file(PWName,[{type,set},{file,PWFile},{repair,true}]) of
{ok, PWDB} ->
GDBName = httpd_util:make_name("httpd_dets_groupdb",Addr,Port),
case dets:open_file(GDBName,[{type,set},{file,GroupFile},{repair,true}]) of
{ok, GDB} ->
NDD1 = lists:keyreplace(auth_user_file, 1, DirData,
{auth_user_file, PWDB}),
NDD2 = lists:keyreplace(auth_group_file, 1, NDD1,
{auth_group_file, GDB}),
{ok, NDD2};
{error, Err}->
{error, {{file, GroupFile},Err}}
end;
{error, Err2} ->
{error, {{file, PWFile},Err2}}
end.
%%
%% Storage format of users in the dets table:
%% {{UserName, Addr, Port, Dir}, Password, UserData}
%%
add_user(DirData, UStruct) ->
{Addr, Port, Dir} = lookup_common(DirData),
PWDB = httpd_util:key1search(DirData, auth_user_file),
Record = {{UStruct#httpd_user.username, Addr, Port, Dir},
UStruct#httpd_user.password, UStruct#httpd_user.user_data},
case dets:lookup(PWDB, UStruct#httpd_user.username) of
[Record] ->
{error, user_already_in_db};
_ ->
dets:insert(PWDB, Record),
true
end.
get_user(DirData, UserName) ->
{Addr, Port, Dir} = lookup_common(DirData),
PWDB = httpd_util:key1search(DirData, auth_user_file),
User = {UserName, Addr, Port, Dir},
case dets:lookup(PWDB, User) of
[{User, Password, UserData}] ->
{ok, #httpd_user{username=UserName, password=Password, user_data=UserData}};
Other ->
{error, no_such_user}
end.
list_users(DirData) ->
?DEBUG("list_users -> ~n"
" DirData: ~p", [DirData]),
{Addr, Port, Dir} = lookup_common(DirData),
PWDB = httpd_util:key1search(DirData, auth_user_file),
case dets:traverse(PWDB, fun(X) -> {continue, X} end) of %% SOOOO Ugly !
Records when list(Records) ->
?DEBUG("list_users -> ~n"
" Records: ~p", [Records]),
{ok, [UserName || {{UserName, AnyAddr, AnyPort, AnyDir}, Password, _Data} <- Records,
AnyAddr == Addr, AnyPort == Port, AnyDir == Dir]};
O ->
?DEBUG("list_users -> ~n"
" O: ~p", [O]),
{ok, []}
end.
delete_user(DirData, UserName) ->
{Addr, Port, Dir} = lookup_common(DirData),
PWDB = httpd_util:key1search(DirData, auth_user_file),
User = {UserName, Addr, Port, Dir},
case dets:lookup(PWDB, User) of
[{User, SomePassword, UserData}] ->
dets:delete(PWDB, User),
lists:foreach(fun(Group) -> delete_group_member(DirData, Group, UserName) end,
list_groups(DirData)),
true;
_ ->
{error, no_such_user}
end.
%%
%% Storage of groups in the dets table:
%% {Group, UserList} where UserList is a list of strings.
%%
add_group_member(DirData, GroupName, UserName) ->
{Addr, Port, Dir} = lookup_common(DirData),
GDB = httpd_util:key1search(DirData, auth_group_file),
Group = {GroupName, Addr, Port, Dir},
case dets:lookup(GDB, Group) of
[{Group, Users}] ->
case lists:member(UserName, Users) of
true ->
true;
false ->
dets:insert(GDB, {Group, [UserName|Users]}),
true
end;
[] ->
dets:insert(GDB, {Group, [UserName]}),
true;
Other ->
{error, Other}
end.
list_group_members(DirData, GroupName) ->
{Addr, Port, Dir} = lookup_common(DirData),
GDB = httpd_util:key1search(DirData, auth_group_file),
Group = {GroupName, Addr, Port, Dir},
case dets:lookup(GDB, Group) of
[{Group, Users}] ->
{ok, Users};
Other ->
{error, no_such_group}
end.
list_groups(DirData) ->
{Addr, Port, Dir} = lookup_common(DirData),
GDB = httpd_util:key1search(DirData, auth_group_file),
case dets:match(GDB, {'$1', '_'}) of
[] ->
{ok, []};
List when list(List) ->
Groups = lists:flatten(List),
{ok, [GroupName || {GroupName, AnyAddr, AnyPort, AnyDir} <- Groups,
AnyAddr == Addr, AnyPort == Port, AnyDir == Dir]};
_ ->
{ok, []}
end.
delete_group_member(DirData, GroupName, UserName) ->
{Addr, Port, Dir} = lookup_common(DirData),
GDB = httpd_util:key1search(DirData, auth_group_file),
Group = {GroupName, Addr, Port, Dir},
case dets:lookup(GDB, GroupName) of
[{Group, Users}] ->
case lists:member(UserName, Users) of
true ->
dets:delete(GDB, Group),
dets:insert(GDB, {Group,
lists:delete(UserName, Users)}),
true;
false ->
{error, no_such_group_member}
end;
_ ->
{error, no_such_group}
end.
delete_group(DirData, GroupName) ->
{Addr, Port, Dir} = lookup_common(DirData),
GDB = httpd_util:key1search(DirData, auth_group_file),
Group = {GroupName, Addr, Port, Dir},
case dets:lookup(GDB, Group) of
[{Group, Users}] ->
dets:delete(GDB, Group),
true;
_ ->
{error, no_such_group}
end.
lookup_common(DirData) ->
Dir = httpd_util:key1search(DirData, path),
Port = httpd_util:key1search(DirData, port),
Addr = httpd_util:key1search(DirData, bind_address),
{Addr, Port, Dir}.
%% remove/1
%%
%% Closes dets tables used by this auth mod.
%%
remove(DirData) ->
PWDB = httpd_util:key1search(DirData, auth_user_file),
GDB = httpd_util:key1search(DirData, auth_group_file),
dets:close(GDB),
dets:close(PWDB),
ok.
