util.stanza: Iterate on childtags instead of all childs.

[prosody.git] / util / prosodyctl.lua

diff --git a/util/prosodyctl.lua b/util/prosodyctl.lua
index e44c78d7d14f69c2b6feaf1c0aa3f717f1655cff..d7ca1a3063727abd832c6293bd2471a51fe1724d 100644 (file)
--- a/util/prosodyctl.lua
+++ b/util/prosodyctl.lua
@@ -1,6 +1,6 @@
 -- Prosody IM
--- Copyright (C) 2008-2009 Matthew Wild
--- Copyright (C) 2008-2009 Waqas Hussain
+-- Copyright (C) 2008-2010 Matthew Wild
+-- Copyright (C) 2008-2010 Waqas Hussain
 -- 
 -- This project is MIT/X11 licensed. Please see the
 -- COPYING file in the source package for more information.
@@ -12,6 +12,8 @@ local encodings = require "util.encodings";
 local stringprep = encodings.stringprep;
 local usermanager = require "core.usermanager";
 local signal = require "util.signal";
+local set = require "util.set";
+local lfs = require "lfs";
 
 local nodeprep, nameprep = stringprep.nodeprep, stringprep.nameprep;
 
@@ -20,6 +22,8 @@ local tostring, tonumber = tostring, tonumber;
 
 local CFG_SOURCEDIR = _G.CFG_SOURCEDIR;
 
+local prosody = prosody;
+
 module "prosodyctl"
 
 function adduser(params)
@@ -29,6 +33,11 @@ function adduser(params)
        elseif not host then
                return false, "invalid-hostname";
        end
+
+       local provider = prosody.hosts[host].users;
+       if not(provider) or provider.name == "null" then
+               usermanager.initialize_host(host);
+       end
        
        local ok = usermanager.create_user(user, password, host);
        if not ok then
@@ -38,7 +47,13 @@ function adduser(params)
 end
 
 function user_exists(params)
-       return usermanager.user_exists(params.user, params.host);
+       local user, host, password = nodeprep(params.user), nameprep(params.host), params.password;
+       local provider = prosody.hosts[host].users;
+       if not(provider) or provider.name == "null" then
+               usermanager.initialize_host(host);
+       end
+       
+       return usermanager.user_exists(user, host);
 end
 
 function passwd(params)
@@ -64,11 +79,22 @@ function getpid()
                return false, "no-pidfile";
        end
        
-       local file, err = io.open(pidfile);
+       local modules_enabled = set.new(config.get("*", "core", "modules_enabled"));
+       if not modules_enabled:contains("posix") then
+               return false, "no-posix";
+       end
+       
+       local file, err = io.open(pidfile, "r+");
        if not file then
                return false, "pidfile-read-failed", err;
        end
        
+       local locked, err = lfs.lock(file, "w");
+       if locked then
+               file:close();
+               return false, "pidfile-not-locked";
+       end
+       
        local pid = tonumber(file:read("*a"));
        file:close();
        
@@ -82,7 +108,7 @@ end
 function isrunning()
        local ok, pid, err = _M.getpid();
        if not ok then
-               if pid == "pidfile-read-failed" then
+               if pid == "pidfile-read-failed" or pid == "pidfile-not-locked" then
                        -- Report as not running, since we can't open the pidfile
                        -- (it probably doesn't exist)
                        return true, false;