Software `/` code `/` prosody

Comparison

prosodyctl @ 7923:81f3068fc30c

Merge 0.10->trunk

author	Kim Alvefur <zash@zash.se>
date	Wed, 01 Mar 2017 02:38:05 +0100
parent	7686:8d6f367bed8c
parent	7922:2fd20f372cb1
child	8015:ecb110f45c92

comparison

equal deleted inserted replaced

-:91db637be237
+:81f3068fc30c
 #!/usr/bin/env lua
 -- Prosody IM
 -- 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.
 --
 -- prosodyctl - command-line controller for Prosody XMPP server
 config = require "core.configmanager"
 local ENV_CONFIG;
 do
 	local filenames = {};
 	local filename;
 	if arg[1] == "--config" and arg[2] then
 		table.insert(filenames, arg[2]);
 		if CFG_CONFIGDIR then
 			table.insert(filenames, CFG_CONFIGDIR.."/"..arg[2]);
 if custom_plugin_paths then
 	local path_sep = package.config:sub(3,3);
 	-- path1;path2;path3;defaultpath...
 	CFG_PLUGINDIR = table.concat(custom_plugin_paths, path_sep)..path_sep..(CFG_PLUGINDIR or "plugins");
 end
 prosody.paths = { source = CFG_SOURCEDIR, config = CFG_CONFIGDIR,
 	          plugins = CFG_PLUGINDIR or "plugins", data = data_path };
 if prosody.installed then
 	-- Change working directory to data path.
 	require "lfs".chdir(data_path);
 		if not switched_user then
 			-- Boo!
 			print("Warning: Couldn't switch to Prosody user/group '"..tostring(desired_user).."'/'"..tostring(desired_group).."': "..tostring(err));
 		end
 	end
 	-- Set our umask to protect data files
 	pposix.umask(config.get("*", "umask") or "027");
 	pposix.setenv("HOME", data_path);
 	pposix.setenv("PROSODY_CONFIG", ENV_CONFIG);
 else
 	print("");
 	os.exit(1);
 end
 local error_messages = setmetatable({
 		["invalid-username"] = "The given username is invalid in a Jabber ID";
 		["invalid-hostname"] = "The given hostname is invalid";
 		["no-password"] = "No password was supplied";
 		["no-such-user"] = "The given user does not exist on the server";
 		["no-such-host"] = "The given hostname does not exist in the config";
 end
 for hostname, config in pairs(config.getconfig()) do
 	hosts[hostname] = make_host(hostname);
 end
 local modulemanager = require "core.modulemanager"
 local prosodyctl = require "util.prosodyctl"
 local socket = require "socket"
 -----------------------
 	if not user and host then
 		show_message [[Failed to understand JID, please supply the JID you want to create]]
 		show_usage [[adduser user@host]]
 		return 1;
 	end
 	if not host then
 		show_message [[Please specify a JID, including a host. e.g. alice@example.com]];
 		return 1;
 	end
 	if not hosts[host] then
 		show_warning("The host '%s' is not listed in the configuration file (or is not enabled).", host)
 		show_warning("The user will not be able to log in until this is changed.");
 		hosts[host] = make_host(host);
 	end
 	if prosodyctl.user_exists{ user = user, host = host } then
 		show_message [[That user already exists]];
 		return 1;
 	end
 	local password = read_password();
 	if not password then return 1; end
 	local ok, msg = prosodyctl.adduser { user = user, host = host, password = password };
 	if ok then return 0; end
 	show_message(msg)
 	return 1;
 end
 function commands.passwd(arg)
 	if not user and host then
 		show_message [[Failed to understand JID, please supply the JID you want to set the password for]]
 		show_usage [[passwd user@host]]
 		return 1;
 	end
 	if not host then
 		show_message [[Please specify a JID, including a host. e.g. alice@example.com]];
 		return 1;
 	end
 	if not hosts[host] then
 		show_warning("The host '%s' is not listed in the configuration file (or is not enabled).", host)
 		show_warning("The user will not be able to log in until this is changed.");
 		hosts[host] = make_host(host);
 	end
 	if not prosodyctl.user_exists { user = user, host = host } then
 		show_message [[That user does not exist, use prosodyctl adduser to create a new user]]
 		return 1;
 	end
 	local password = read_password();
 	if not password then return 1; end
 	local ok, msg = prosodyctl.passwd { user = user, host = host, password = password };
 	if ok then return 0; end
 	show_message(error_messages[msg])
 	return 1;
 end
 function commands.deluser(arg)
 	if not user and host then
 		show_message [[Failed to understand JID, please supply the JID to the user account you want to delete]]
 		show_usage [[deluser user@host]]
 		return 1;
 	end
 	if not host then
 		show_message [[Please specify a JID, including a host. e.g. alice@example.com]];
 		return 1;
 	end
 	if not hosts[host] then
 		show_warning("The host '%s' is not listed in the configuration file (or is not enabled).", host)
 		hosts[host] = make_host(host);
 	end
 	if not prosodyctl.user_exists { user = user, host = host } then
 		show_message [[That user does not exist on this server]]
 		return 1;
 	end
 	local ok, msg = prosodyctl.deluser { user = user, host = host };
 	if ok then return 0; end
 	show_message(error_messages[msg])
 	return 1;
 end
 function commands.start(arg)
 	local ok, ret = prosodyctl.isrunning();
 	if not ok then
 		show_message(error_messages[ret]);
 		return 1;
 	end
 	if ret then
 		local ok, ret = prosodyctl.getpid();
 		if not ok then
 			show_message("Couldn't get running Prosody's PID");
 			show_message(error_messages[ret]);
 			return 1;
 		end
 		show_message("Prosody is already running with PID %s", ret or "(unknown)");
 		return 1;
 	end
 	local ok, ret = prosodyctl.start();
 	if ok then
 		local daemonize = config.get("*", "daemonize");
 		if daemonize == nil then
 			daemonize = prosody.installed;
 		end
 		return 0;
 	end
 	show_message("Failed to start Prosody");
 	show_message(error_messages[ret])
 	return 1;
 end
 function commands.status(arg)
 	if arg[1] == "--help" then
 		show_usage([[status]], [[Reports the running status of Prosody]]);
 	local ok, ret = prosodyctl.isrunning();
 	if not ok then
 		show_message(error_messages[ret]);
 		return 1;
 	end
 	if ret then
 		local ok, ret = prosodyctl.getpid();
 		if not ok then
 			show_message("Couldn't get running Prosody's PID");
 			show_message(error_messages[ret]);
 	if not prosodyctl.isrunning() then
 		show_message("Prosody is not running");
 		return 1;
 	end
 	local ok, ret = prosodyctl.stop();
 	if ok then
 		local i=1;
 		while true do
 			local ok, running = prosodyctl.isrunning();
 function commands.restart(arg)
 	if arg[1] == "--help" then
 		show_usage([[restart]], [[Restart a running Prosody server]]);
 		return 1;
 	end
 	commands.stop(arg);
 	return commands.start(arg);
 end
 function commands.about(arg)
 	read_version();
 	if arg[1] == "--help" then
 		show_usage([[about]], [[Show information about this Prosody installation]]);
 		return 1;
 	end
 	local pwd = ".";
 	local lfs = require "lfs";
 	local array = require "util.array";
 	local keys = require "util.iterators".keys;
 	local hg = require"util.mercurial";
 	local relpath = config.resolve_relative_path;
 	print("Prosody "..(prosody.version or "(unknown version)"));
 	print("");
 	print("# Prosody directories");
 	print("Data directory:     "..relpath(pwd, data_path));
 	print("Config directory:   "..relpath(pwd, CFG_CONFIGDIR or "."));
 	if not prosodyctl.isrunning() then
 		show_message("Prosody is not running");
 		return 1;
 	end
 	local ok, ret = prosodyctl.reload();
 	if ok then
 		show_message("Prosody log files re-opened and config file reloaded. You may need to reload modules for some changes to take effect.");
 		return 0;
 	end
 	show_message(error_messages[ret]);
 	return 1;
 end
 -- ejabberdctl compatibility
+local unpack = table.unpack or unpack; -- luacheck: ignore 113
 function commands.register(arg)
 	local user, host, password = unpack(arg);
 	if (not (user and host)) or arg[1] == "--help" then
 		if user ~= "--help" then
 		if not password then
 			show_message [[Unable to register user with no password]];
 			return 1;
 		end
 	end
 	local ok, msg = prosodyctl.adduser { user = user, host = host, password = password };
 	if ok then return 0; end
 	show_message(error_messages[msg])
 	return 1;
 end
 function commands.unregister(arg)
 		show_usage("unregister USER HOST [PASSWORD]", "Permanently remove a user account from the server");
 		return 1;
 	end
 	local ok, msg = prosodyctl.deluser { user = user, host = host };
 	if ok then return 0; end
 	show_message(error_messages[msg])
 	return 1;
 end
 local openssl;
 			print("");
 			print("    You require encryption but mod_tls is not enabled.");
 			print("    Connections will fail.");
 			ok = false;
 		end
 		print("Done.\n");
 	end
 	if not what or what == "dns" then
 		local dns = require "net.dns";
 		local idna = require "util.encodings".idna;
 		local ip = require "util.ip";
 		local c2s_ports = set.new(config.get("*", "c2s_ports") or {5222});
 		local s2s_ports = set.new(config.get("*", "s2s_ports") or {5269});
 		local c2s_srv_required, s2s_srv_required;
 		if not c2s_ports:contains(5222) then
 			c2s_srv_required = true;
 		end
 		if not s2s_ports:contains(5269) then
 			s2s_srv_required = true;
 		end
 		local problem_hosts = set.new();
 		local external_addresses, internal_addresses = set.new(), set.new();
 		local fqdn = socket.dns.tohostname(socket.dns.gethostname());
 		if fqdn then
 			local res = dns.lookup(idna.to_ascii(fqdn), "A");
 			if res then
 				for _, record in ipairs(res) do
 				for _, record in ipairs(res) do
 					external_addresses:add(record.aaaa);
 				end
 			end
 		end
 		local local_addresses = require"util.net".local_addresses() or {};
 		for addr in it.values(local_addresses) do
 			if not ip.new_ip(addr).private then
 				external_addresses:add(addr);
 			else
 				internal_addresses:add(addr);
 			end
 		end
 		if external_addresses:empty() then
 			print("");
 			print("   Failed to determine the external addresses of this server. Checks may be inaccurate.");
 			c2s_srv_required, s2s_srv_required = true, true;
 		end
 		local v6_supported = not not socket.tcp6;
 		for jid, host_options in enabled_hosts() do
 			local all_targets_ok, some_targets_ok = true, false;
 			local node, host = jid_split(jid);
 			local is_component = not not host_options.component_module;
 			print("Checking DNS for "..(is_component and "component" or "host").." "..jid.."...");
 			if node then
 				print("Only the domain part ("..host..") is used in DNS.")
 			end
 				end
 			end
 			if target_hosts:empty() then
 				target_hosts:add(host);
 			end
 			if target_hosts:contains("localhost") then
 				print("    Target 'localhost' cannot be accessed from other servers");
 				target_hosts:remove("localhost");
 			end
 			local modules = set.new(it.to_array(it.values(host_options.modules_enabled or {})))
 			                + set.new(it.to_array(it.values(config.get("*", "modules_enabled") or {})))
 			                + set.new({ config.get(host, "component_module") });
 			if modules:contains("proxy65") then
 				end
 				if #prob > 0 then
 					print("    File transfer proxy "..proxy65_target.." has no "..table.concat(prob, "/").." record. Create one or set 'proxy65_address' to the correct host/IP.");
 				end
 			end
 			for host in target_hosts do
 				local host_ok_v4, host_ok_v6;
 				local res = dns.lookup(idna.to_ascii(host), "A");
 				if res then
 					for _, record in ipairs(res) do
 							print("    "..host.." AAAA record points to unknown address "..record.aaaa);
 							all_targets_ok = false;
 						end
 					end
 				end
 				local bad_protos = {}
 				if not host_ok_v4 then
 					table.insert(bad_protos, "IPv4");
 				end
 				if not host_ok_v6 then
 	local ret, err = modulemanager.load("*", module_name);
 	if not ret then
 		show_message("Failed to load module '"..module_name.."': "..err);
 		os.exit(1);
 	end
 	table.remove(arg, 1);
 	local module = modulemanager.get_module("*", module_name);
 	if not module then
 		show_message("Failed to load module '"..module_name.."': Unknown error");
 		os.exit(1);
 	end
 	if not modulemanager.module_has_method(module, "command") then
 		show_message("Fail: mod_"..module_name.." does not support any commands");
 		os.exit(1);
 	end
 	local ok, ret = modulemanager.call_module_method(module, "command", arg);
 	if ok then
 		if type(ret) == "number" then
 			os.exit(ret);
 		elseif type(ret) == "string" then
 			command{ "--help" };
 			print""
 			done[command_name] = true;
 		end
 	end
 	os.exit(0);
 end
 os.exit(commands[command]({ select(2, unpack(arg)) }));

prosody

81f3068fc30c

prosodyctl

Software / code / prosody

Comparison

prosodyctl @ 7923:81f3068fc30c

Software `/` code `/` prosody