Software /
code /
prosody-modules
File
mod_srvinjection/mod_srvinjection.lua @ 724:b94010de43f6
mod_host_guard: referenced configmanager from prosody's _G instead of requiring it.
author | Marco Cirillo <maranda@lightwitch.org> |
---|---|
date | Mon, 25 Jun 2012 22:32:54 +0000 |
parent | 662:b3d130e4b3ae |
child | 1248:69f7840923f5 |
line wrap: on
line source
module:set_global(); local adns = require "net.adns"; local map = module:get_option("srvinjection") or {}; for host, mapping in pairs(map) do if type(mapping) == "table" and type(mapping[1]) == "string" and (type(mapping[2]) == "number") then local connecthost, connectport = mapping[1], mapping[2] or 5269; map[host] = {{ srv = { target = connecthost.."."; port = connectport; priority = 1; weight = 0; }; }}; else module:log("warn", "Ignoring invalid SRV injection for host '%s'", host); map[host] = nil; end end local original_lookup = adns.lookup; function adns.lookup(handler, qname, qtype, qclass) if qtype == "SRV" then local host = qname:match("^_xmpp%-server%._tcp%.(.*)%.$"); local mapping = map[host] or map["*"]; if mapping then handler(mapping); return; end elseif qtype == "A" and (qname == "localhost." or qname == "127.0.0.1.") then handler({{ a = "127.0.0.1" }}); return; end return original_lookup(handler, qname, qtype, qclass); end function module.unload() adns.lookup = original_lookup; end