Merge 0.9->trunk
[prosody.git] / plugins / mod_admin_telnet.lua
index b698d20034231a264d00de408c366ae4e27beb4c..0af571eb0aef00300b245c1cd64b6b9dc3ce185d 100644 (file)
@@ -8,10 +8,16 @@
 
 module:set_global();
 
+local hostmanager = require "core.hostmanager";
+local modulemanager = require "core.modulemanager";
+local s2smanager = require "core.s2smanager";
+local portmanager = require "core.portmanager";
+
 local _G = _G;
 
 local prosody = _G.prosody;
 local hosts = prosody.hosts;
+local incoming_s2s = prosody.incoming_s2s;
 
 local console_listener = { default_port = 5582; default_mode = "*a"; interface = "127.0.0.1" };
 
@@ -21,17 +27,20 @@ local jid = require "util.jid";
 local jid_bare, jid_split = jid.bare, jid.split;
 local set, array = require "util.set", require "util.array";
 local cert_verify_identity = require "util.x509".verify_identity;
+local envload = require "util.envload".envload;
+local envloadfile = require "util.envload".envloadfile;
 
 local commands = module:shared("commands")
 local def_env = module:shared("env");
 local default_env_mt = { __index = def_env };
+local core_post_stanza = prosody.core_post_stanza;
 
 local function redirect_output(_G, session)
        local env = setmetatable({ print = session.print }, { __index = function (t, k) return rawget(_G, k); end });
        env.dofile = function(name)
-               local f, err = loadfile(name);
+               local f, err = envloadfile(name, env);
                if not f then return f, err; end
-               return setfenv(f, env)();
+               return f();
        end;
        return env;
 end
@@ -63,6 +72,64 @@ function console:new_session(conn)
        return session;
 end
 
+function console:process_line(session, line)
+       local useglobalenv;
+       
+       if line:match("^>") then
+               line = line:gsub("^>", "");
+               useglobalenv = true;
+       elseif line == "\004" then
+               commands["bye"](session, line);
+               return;
+       else
+               local command = line:match("^%w+") or line:match("%p");
+               if commands[command] then
+                       commands[command](session, line);
+                       return;
+               end
+       end
+       
+       session.env._ = line;
+       
+       local chunkname = "=console";
+       local env = (useglobalenv and redirect_output(_G, session)) or session.env or nil
+       local chunk, err = envload("return "..line, chunkname, env);
+       if not chunk then
+               chunk, err = envload(line, chunkname, env);
+               if not chunk then
+                       err = err:gsub("^%[string .-%]:%d+: ", "");
+                       err = err:gsub("^:%d+: ", "");
+                       err = err:gsub("'<eof>'", "the end of the line");
+                       session.print("Sorry, I couldn't understand that... "..err);
+                       return;
+               end
+       end
+       
+       local ranok, taskok, message = pcall(chunk);
+       
+       if not (ranok or message or useglobalenv) and commands[line:lower()] then
+               commands[line:lower()](session, line);
+               return;
+       end
+       
+       if not ranok then
+               session.print("Fatal error while running command, it did not complete");
+               session.print("Error: "..taskok);
+               return;
+       end
+       
+       if not message then
+               session.print("Result: "..tostring(taskok));
+               return;
+       elseif (not taskok) and message then
+               session.print("Command completed with a problem");
+               session.print("Message: "..tostring(message));
+               return;
+       end
+       
+       session.print("OK: "..tostring(message));
+end
+
 local sessions = {};
 
 function console_listener.onconnect(conn)
@@ -82,66 +149,7 @@ function console_listener.onincoming(conn, data)
        end
 
        for line in data:gmatch("[^\n]*[\n\004]") do
-               -- Handle data (loop allows us to break to add \0 after response)
-               repeat
-                       local useglobalenv;
-
-                       if line:match("^>") then
-                               line = line:gsub("^>", "");
-                               useglobalenv = true;
-                       elseif line == "\004" then
-                               commands["bye"](session, line);
-                               break;
-                       else
-                               local command = line:match("^%w+") or line:match("%p");
-                               if commands[command] then
-                                       commands[command](session, line);
-                                       break;
-                               end
-                       end
-
-                       session.env._ = line;
-                       
-                       local chunkname = "=console";
-                       local chunk, err = loadstring("return "..line, chunkname);
-                       if not chunk then
-                               chunk, err = loadstring(line, chunkname);
-                               if not chunk then
-                                       err = err:gsub("^%[string .-%]:%d+: ", "");
-                                       err = err:gsub("^:%d+: ", "");
-                                       err = err:gsub("'<eof>'", "the end of the line");
-                                       session.print("Sorry, I couldn't understand that... "..err);
-                                       break;
-                               end
-                       end
-                       
-                       setfenv(chunk, (useglobalenv and redirect_output(_G, session)) or session.env or nil);
-                       
-                       local ranok, taskok, message = pcall(chunk);
-                       
-                       if not (ranok or message or useglobalenv) and commands[line:lower()] then
-                               commands[line:lower()](session, line);
-                               break;
-                       end
-                       
-                       if not ranok then
-                               session.print("Fatal error while running command, it did not complete");
-                               session.print("Error: "..taskok);
-                               break;
-                       end
-                       
-                       if not message then
-                               session.print("Result: "..tostring(taskok));
-                               break;
-                       elseif (not taskok) and message then
-                               session.print("Command completed with a problem");
-                               session.print("Message: "..tostring(message));
-                               break;
-                       end
-                       
-                       session.print("OK: "..tostring(message));
-               until true
-               
+               console:process_line(session, line);
                session.send(string.char(0));
        end
        session.partial_data = data:match("[^\n]+$");
@@ -219,7 +227,8 @@ function commands.help(session, data)
        elseif section == "user" then
                print [[user:create(jid, password) - Create the specified user account]]
                print [[user:password(jid, password) - Set the password for the specified user account]]
-               print [[user:delete(jid, password) - Permanently remove the specified user account]]
+               print [[user:delete(jid) - Permanently remove the specified user account]]
+               print [[user:list(hostname, pattern) - List users on the specified host, optionally filtering with a pattern]]
        elseif section == "server" then
                print [[server:version() - Show the server's version number]]
                print [[server:uptime() - Show how long the server has been running]]
@@ -512,7 +521,7 @@ function def_env.c2s:show_secure(match_jid)
 end
 
 function def_env.c2s:close(match_jid)
-       local print, count = self.session.print, 0;
+       local count = 0;
        show_c2s(function (jid, session)
                if jid == match_jid or jid_bare(jid) == match_jid then
                        count = count + 1;
@@ -641,6 +650,7 @@ local function print_errors(print, errors)
 end
 
 function def_env.s2s:showcert(domain)
+       local ser = require "util.serialization".serialize;
        local print = self.session.print;
        local domain_sessions = set.new(array.collect(keys(incoming_s2s)))
                /function(session) return session.from_host == domain and session or nil; end;
@@ -867,7 +877,7 @@ function def_env.port:close(close_port, close_interface)
                                        self.session.print("Closing ["..interface.."]:"..close_port.."...");
                                        local ok, err = portmanager.close(interface, close_port)
                                        if not ok then
-                                               self.session.print("Failed to close "..interface.." "..port..": "..err);
+                                               self.session.print("Failed to close "..interface.." "..close_port..": "..err);
                                        else
                                                n_closed = n_closed + 1;
                                        end
@@ -906,6 +916,9 @@ local um = require"core.usermanager";
 def_env.user = {};
 function def_env.user:create(jid, password)
        local username, host = jid_split(jid);
+       if um.user_exists(username, host) then
+               return nil, "User exists";
+       end
        local ok, err = um.create_user(username, password, host);
        if ok then
                return true, "User created";
@@ -916,6 +929,9 @@ end
 
 function def_env.user:delete(jid)
        local username, host = jid_split(jid);
+       if not um.user_exists(username, host) then
+               return nil, "No such user";
+       end
        local ok, err = um.delete_user(username, host);
        if ok then
                return true, "User deleted";
@@ -924,22 +940,56 @@ function def_env.user:delete(jid)
        end
 end
 
-function def_env.user:passwd(jid, password)
+function def_env.user:password(jid, password)
        local username, host = jid_split(jid);
+       if not um.user_exists(username, host) then
+               return nil, "No such user";
+       end
        local ok, err = um.set_password(username, password, host);
        if ok then
-               return true, "User created";
+               return true, "User password changed";
        else
                return nil, "Could not change password for user: "..err;
        end
 end
 
+function def_env.user:list(host, pat)
+       if not host then
+               return nil, "No host given";
+       elseif not hosts[host] then
+               return nil, "No such host";
+       end
+       local print = self.session.print;
+       local count = 0;
+       for user in um.users(host) do
+               if not pat or user:match(pat) then
+                       print(user.."@"..host);
+               end
+               count = count + 1;
+       end
+       return true, count .. " users total";
+end
+
+def_env.xmpp = {};
+
+local st = require "util.stanza";
+function def_env.xmpp:ping(localhost, remotehost)
+       if hosts[localhost] then
+               core_post_stanza(hosts[localhost],
+                       st.iq{ from=localhost, to=remotehost, type="get", id="ping" }
+                               :tag("ping", {xmlns="urn:xmpp:ping"}));
+               return true, "Sent ping";
+       else
+               return nil, "No such host";
+       end
+end
+
 -------------
 
 function printbanner(session)
-       local option = config.get("*", "core", "console_banner");
-if option == nil or option == "full" or option == "graphic" then
-session.print [[
+       local option = module:get_option("console_banner");
+       if option == nil or option == "full" or option == "graphic" then
+               session.print [[
                    ____                \   /     _       
                     |  _ \ _ __ ___  ___  _-_   __| |_   _ 
                     | |_) | '__/ _ \/ __|/ _ \ / _` | | | |
@@ -948,23 +998,22 @@ session.print [[
                     A study in simplicity            |___/ 
 
 ]]
-end
-if option == nil or option == "short" or option == "full" then
-session.print("Welcome to the Prosody administration console. For a list of commands, type: help");
-session.print("You may find more help on using this console in our online documentation at ");
-session.print("http://prosody.im/doc/console\n");
-end
-if option and option ~= "short" and option ~= "full" and option ~= "graphic" then
-       if type(option) == "string" then
-               session.print(option)
-       elseif type(option) == "function" then
-               setfenv(option, redirect_output(_G, session));
-               pcall(option, session);
        end
-end
+       if option == nil or option == "short" or option == "full" then
+       session.print("Welcome to the Prosody administration console. For a list of commands, type: help");
+       session.print("You may find more help on using this console in our online documentation at ");
+       session.print("http://prosody.im/doc/console\n");
+       end
+       if option and option ~= "short" and option ~= "full" and option ~= "graphic" then
+               if type(option) == "string" then
+                       session.print(option)
+               elseif type(option) == "function" then
+                       module:log("warn", "Using functions as value for the console_banner option is no longer supported");
+               end
+       end
 end
 
-module:add_item("net-provider", {
+module:provides("net", {
        name = "console";
        listener = console_listener;
        default_port = 5582;