net.server_event: Check the buffer *length*, not the buffer itself (Fixes 100% cpu...
[prosody.git] / core / usermanager.lua
1 -- Prosody IM
2 -- Copyright (C) 2008-2010 Matthew Wild
3 -- Copyright (C) 2008-2010 Waqas Hussain
4 --
5 -- This project is MIT/X11 licensed. Please see the
6 -- COPYING file in the source package for more information.
7 --
8
9 local modulemanager = require "core.modulemanager";
10 local log = require "util.logger".init("usermanager");
11 local type = type;
12 local ipairs = ipairs;
13 local pairs = pairs;
14 local jid_bare = require "util.jid".bare;
15 local jid_prep = require "util.jid".prep;
16 local config = require "core.configmanager";
17 local hosts = hosts;
18 local sasl_new = require "util.sasl".new;
19 local storagemanager = require "core.storagemanager";
20
21 local prosody = _G.prosody;
22
23 local setmetatable = setmetatable;
24
25 local default_provider = "internal_plain";
26
27 module "usermanager"
28
29 function new_null_provider()
30         local function dummy() return nil, "method not implemented"; end;
31         local function dummy_get_sasl_handler() return sasl_new(nil, {}); end
32         return setmetatable({name = "null", get_sasl_handler = dummy_get_sasl_handler}, {
33                 __index = function(self, method) return dummy; end
34         });
35 end
36
37 local provider_mt = { __index = new_null_provider() };
38
39 function initialize_host(host)
40         local host_session = hosts[host];
41         if host_session.type ~= "local" then return; end
42         
43         host_session.events.add_handler("item-added/auth-provider", function (event)
44                 local provider = event.item;
45                 local auth_provider = config.get(host, "authentication") or default_provider;
46                 if config.get(host, "anonymous_login") then
47                         log("error", "Deprecated config option 'anonymous_login'. Use authentication = 'anonymous' instead.");
48                         auth_provider = "anonymous";
49                 end -- COMPAT 0.7
50                 if provider.name == auth_provider then
51                         host_session.users = setmetatable(provider, provider_mt);
52                 end
53                 if host_session.users ~= nil and host_session.users.name ~= nil then
54                         log("debug", "host '%s' now set to use user provider '%s'", host, host_session.users.name);
55                 end
56         end);
57         host_session.events.add_handler("item-removed/auth-provider", function (event)
58                 local provider = event.item;
59                 if host_session.users == provider then
60                         host_session.users = new_null_provider();
61                 end
62         end);
63         host_session.users = new_null_provider(); -- Start with the default usermanager provider
64         local auth_provider = config.get(host, "authentication") or default_provider;
65         if config.get(host, "anonymous_login") then auth_provider = "anonymous"; end -- COMPAT 0.7
66         if auth_provider ~= "null" then
67                 modulemanager.load(host, "auth_"..auth_provider);
68         end
69 end;
70 prosody.events.add_handler("host-activated", initialize_host, 100);
71
72 function test_password(username, host, password)
73         return hosts[host].users.test_password(username, password);
74 end
75
76 function get_password(username, host)
77         return hosts[host].users.get_password(username);
78 end
79
80 function set_password(username, password, host)
81         return hosts[host].users.set_password(username, password);
82 end
83
84 function user_exists(username, host)
85         return hosts[host].users.user_exists(username);
86 end
87
88 function create_user(username, password, host)
89         return hosts[host].users.create_user(username, password);
90 end
91
92 function delete_user(username, host)
93         local ok, err = hosts[host].users.delete_user(username);
94         if not ok then return nil, err; end
95         prosody.events.fire_event("user-deleted", { username = username, host = host });
96         return storagemanager.purge(username, host);
97 end
98
99 function users(host)
100         return hosts[host].users.users();
101 end
102
103 function get_sasl_handler(host, session)
104         return hosts[host].users.get_sasl_handler(session);
105 end
106
107 function get_provider(host)
108         return hosts[host].users;
109 end
110
111 function is_admin(jid, host)
112         if host and not hosts[host] then return false; end
113         if type(jid) ~= "string" then return false; end
114
115         local is_admin;
116         jid = jid_bare(jid);
117         host = host or "*";
118         
119         local host_admins = config.get(host, "admins");
120         local global_admins = config.get("*", "admins");
121         
122         if host_admins and host_admins ~= global_admins then
123                 if type(host_admins) == "table" then
124                         for _,admin in ipairs(host_admins) do
125                                 if jid_prep(admin) == jid then
126                                         is_admin = true;
127                                         break;
128                                 end
129                         end
130                 elseif host_admins then
131                         log("error", "Option 'admins' for host '%s' is not a list", host);
132                 end
133         end
134         
135         if not is_admin and global_admins then
136                 if type(global_admins) == "table" then
137                         for _,admin in ipairs(global_admins) do
138                                 if jid_prep(admin) == jid then
139                                         is_admin = true;
140                                         break;
141                                 end
142                         end
143                 elseif global_admins then
144                         log("error", "Global option 'admins' is not a list");
145                 end
146         end
147         
148         -- Still not an admin, check with auth provider
149         if not is_admin and host ~= "*" and hosts[host].users and hosts[host].users.is_admin then
150                 is_admin = hosts[host].users.is_admin(jid);
151         end
152         return is_admin or false;
153 end
154
155 return _M;