Mercurial > prosody-modules
view mod_srvinjection/mod_srvinjection.lua @ 989:7c04c5856daa
mod_register_json: major code overhaul into a token based registration & verification system.
author | Marco Cirillo <maranda@lightwitch.org> |
---|---|
date | Mon, 29 Apr 2013 22:53:39 +0200 |
parents | b3d130e4b3ae |
children | 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