prosody-modules: mod_dnsupdate/mod_dnsupdate.lua annotate

annotate mod_dnsupdate/mod_dnsupdate.lua @ 5648:c217f4edfc4f

misc/mtail: Start of an mtail config Stashing it here in case anyone wants to continue working on it. Currently it's only counting log messages by level. Due to the permissions set by systemd on Prosody logs, mtail never managed to start correctly until permissions were manually relaxed.

author	Kim Alvefur <zash@zash.se>
date	Sun, 17 Sep 2023 13:36:30 +0200
parents	54fa2116bbf3
children	429be658c0bb

rev	line source
4762 ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	1 module:set_global();
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	2
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	3 local config = require "core.configmanager";
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	4 local argparse = require "util.argparse";
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	5 local dns = require"net.adns".resolver();
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	6 local async = require "util.async";
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	7 local set = require "util.set";
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	8 local nameprep = require"util.encodings".stringprep.nameprep;
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	9 local idna_to_ascii = require"util.encodings".idna.to_ascii;
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	10
4779 6395d4732bc6 mod_dnsupdate: Add support for components Kim Alvefur <zash@zash.se> parents: 4765 diff changeset	11 local virtualhost_services = { "xmpp-client"; "xmpps-client"; "xmpp-server"; "xmpps-server" }
6395d4732bc6 mod_dnsupdate: Add support for components Kim Alvefur <zash@zash.se> parents: 4765 diff changeset	12 local component_services = { "xmpp-server"; "xmpps-server" }
4762 ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	13
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	14 local function validate_dnsname_option(options, option_name, default)
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	15 local host = options[option_name];
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	16 if host == nil then return default end
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	17 local normalized = nameprep(host);
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	18 if not normalized then
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	19 module:log("error", "--%s %q fails normalization");
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	20 return;
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	21 end
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	22 local alabel = idna_to_ascii(normalized);
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	23 if not alabel then
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	24 module:log("error", "--%s %q fails IDNA");
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	25 return;
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	26 end
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	27 return alabel;
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	28 end
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	29
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	30 function module.command(arg)
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	31 local opts = argparse.parse(arg, {
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	32 short_params = { d = "domain"; p = "primary"; t = "target"; l = "ttl"; h = "help"; ["?"] = "help" };
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	33 value_params = { domain = true; primary = true; target = true; ttl = true };
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	34 });
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	35
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	36 if not arg[1] or arg[2] or not opts or opts.help or not opts.domain then
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	37 local out = opts.help and io.stdout or io.stderr;
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	38 out:write("prosodyctl mod_dnsupdate [options] virtualhost\n");
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	39 out:write("\t-d --domain\tbase domain name required\n");
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	40 out:write("\t-p --primary\tprimary DNS name server\n");
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	41 out:write("\t-t --target\ttarget hostname for SRV\n");
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	42 out:write("\t-l --ttl\tTTL to use\n");
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	43 out:write("\t--each\tremove and replace individual SRV records\n");
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	44 out:write("\t--reset\tremove and replace all SRV records\n");
4896 54fa2116bbf3 mod_dnsupdate: Add a --remove option to delete all SRV records Kim Alvefur <zash@zash.se> parents: 4882 diff changeset	45 out:write("\t--remove\tremove all SRV records\n");
4762 ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	46 return opts and opts.help and 0 or 1;
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	47 end
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	48
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	49 local vhost = nameprep(arg[1]); -- TODO loop over arg[]?
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	50 if not vhost then
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	51 module:log("error", "Host %q fails normalization", arg[1]);
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	52 return 1;
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	53 end
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	54 local ihost = idna_to_ascii(vhost);
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	55 if not ihost then
4882 28f6322fad50 mod_dnsupdate: Fix typo Kim Alvefur <zash@zash.se> parents: 4779 diff changeset	56 module:log("error", "Host %q fails IDNA", vhost);
4762 ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	57 return 1;
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	58 end
4779 6395d4732bc6 mod_dnsupdate: Add support for components Kim Alvefur <zash@zash.se> parents: 4765 diff changeset	59 local is_component = config.get(vhost, "component_module");
6395d4732bc6 mod_dnsupdate: Add support for components Kim Alvefur <zash@zash.se> parents: 4765 diff changeset	60 if not is_component and not config.get(vhost, "defined") then
4762 ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	61 module:log("error", "Host %q is not defined in the config", vhost);
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	62 return 1;
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	63 end
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	64
4779 6395d4732bc6 mod_dnsupdate: Add support for components Kim Alvefur <zash@zash.se> parents: 4765 diff changeset	65 local services = virtualhost_services;
6395d4732bc6 mod_dnsupdate: Add support for components Kim Alvefur <zash@zash.se> parents: 4765 diff changeset	66 if is_component then services = component_services; end
6395d4732bc6 mod_dnsupdate: Add support for components Kim Alvefur <zash@zash.se> parents: 4765 diff changeset	67
4762 ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	68 local domain = validate_dnsname_option(opts, "domain");
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	69 if not domain then
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	70 module:log("error", "--domain is required");
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	71 return 1;
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	72 end
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	73 local primary = validate_dnsname_option(opts, "primary")
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	74 or async.wait_for(dns:lookup_promise(domain, "SOA"):next(function(ret) return ret[1].soa.mname; end));
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	75 if not primary then
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	76 module:log("error", "Could not discover primary name server, specify it with --primary");
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	77 return 1;
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	78 end
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	79 local target = validate_dnsname_option(opts, "target", module:context(vhost):get_option_string("xmpp_host", ihost));
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	80 -- TODO validate that target has A/AAAA
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	81
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	82 local configured_ports = {
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	83 ["xmpp-client"] = module:get_option_array("c2s_ports", { 5222 });
4763 91077c928c57 mod_dnsupdate: Fix to use correct port settings Kim Alvefur <zash@zash.se> parents: 4762 diff changeset	84 ["xmpp-server"] = module:get_option_array("s2s_ports", { 5269 });
4762 ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	85 ["xmpps-client"] = module:get_option_array("c2s_direct_tls_ports", {});
4763 91077c928c57 mod_dnsupdate: Fix to use correct port settings Kim Alvefur <zash@zash.se> parents: 4762 diff changeset	86 ["xmpps-server"] = module:get_option_array("s2s_direct_tls_ports", {});
4762 ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	87 };
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	88
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	89 if opts.multiplex then
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	90 for opt, ports in pairs(configured_ports) do
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	91 ports:append(module:get_option_array(opt:sub(1, 5) == "xmpps" and "ssl_ports" or "ports", {}));
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	92 end
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	93 end
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	94
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	95 local existing_srv = {};
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	96 for _, service in ipairs(services) do
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	97 existing_srv[service] = dns:lookup_promise(("_%s._tcp.%s"):format(service, ihost), "SRV");
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	98 end
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	99
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	100 print("zone", domain);
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	101 print("server", primary);
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	102 print("ttl " .. tostring(opts.ttl or 60 * 60));
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	103
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	104 for _, service in ipairs(services) do
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	105 local ports = set.new(configured_ports[service]);
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	106 local records = (async.wait_for(existing_srv[service]));
4896 54fa2116bbf3 mod_dnsupdate: Add a --remove option to delete all SRV records Kim Alvefur <zash@zash.se> parents: 4882 diff changeset	107 if opts.remove or opts.reset then
4764 a754f7e380b2 mod_dnsupdate: Rewrite port config vs DNS comparison code Kim Alvefur <zash@zash.se> parents: 4763 diff changeset	108 print(("del _%s._tcp.%s IN SRV"):format(service, ihost));
a754f7e380b2 mod_dnsupdate: Rewrite port config vs DNS comparison code Kim Alvefur <zash@zash.se> parents: 4763 diff changeset	109 else
a754f7e380b2 mod_dnsupdate: Rewrite port config vs DNS comparison code Kim Alvefur <zash@zash.se> parents: 4763 diff changeset	110 for _, rr in ipairs(records) do
a754f7e380b2 mod_dnsupdate: Rewrite port config vs DNS comparison code Kim Alvefur <zash@zash.se> parents: 4763 diff changeset	111 if ports:contains(rr.srv.port) and target == nameprep(rr.srv.target):gsub("%.$", "") then
a754f7e380b2 mod_dnsupdate: Rewrite port config vs DNS comparison code Kim Alvefur <zash@zash.se> parents: 4763 diff changeset	112 ports:remove(rr.srv.port)
4765 3632836f35b0 mod_dnsupdate: Restore the --each option Kim Alvefur <zash@zash.se> parents: 4764 diff changeset	113 elseif not opts.each then
3632836f35b0 mod_dnsupdate: Restore the --each option Kim Alvefur <zash@zash.se> parents: 4764 diff changeset	114 print(("del _%s._tcp.%s IN SRV"):format(service, ihost));
3632836f35b0 mod_dnsupdate: Restore the --each option Kim Alvefur <zash@zash.se> parents: 4764 diff changeset	115 break
4764 a754f7e380b2 mod_dnsupdate: Rewrite port config vs DNS comparison code Kim Alvefur <zash@zash.se> parents: 4763 diff changeset	116 else
a754f7e380b2 mod_dnsupdate: Rewrite port config vs DNS comparison code Kim Alvefur <zash@zash.se> parents: 4763 diff changeset	117 print(("del _%s._tcp.%s IN SRV %s"):format(service, ihost, rr));
4762 ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	118 end
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	119 end
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	120 end
4896 54fa2116bbf3 mod_dnsupdate: Add a --remove option to delete all SRV records Kim Alvefur <zash@zash.se> parents: 4882 diff changeset	121 if not opts.remove then
54fa2116bbf3 mod_dnsupdate: Add a --remove option to delete all SRV records Kim Alvefur <zash@zash.se> parents: 4882 diff changeset	122 for port in ports do print(("add _%s._tcp.%s IN SRV 1 1 %d %s"):format(service, ihost, port, target)); end
54fa2116bbf3 mod_dnsupdate: Add a --remove option to delete all SRV records Kim Alvefur <zash@zash.se> parents: 4882 diff changeset	123 end
4762 ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	124 end
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	125
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	126 print("show");
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	127 print("send");
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	128 print("answer");
ba312cd7907f mod_dnsupdate: Use with nsupdate to update DNS records from config Kim Alvefur <zash@zash.se> parents: diff changeset	129 end

Mercurial > prosody-modules

annotate mod_dnsupdate/mod_dnsupdate.lua @ 5648:c217f4edfc4f