Mercurial > prosody-modules
diff mod_flash_policy/mod_flash_policy.lua @ 394:4219f69be1cf
Let session.send() actually send the config string
author | leonbogaert@gmail.com |
---|---|
date | Tue, 26 Jul 2011 13:18:38 +0000 |
parents | eebc19c224fb |
children | 77ca0947647b |
line wrap: on
line diff
--- a/mod_flash_policy/mod_flash_policy.lua Mon Jul 25 02:21:40 2011 +0500 +++ b/mod_flash_policy/mod_flash_policy.lua Tue Jul 26 13:18:38 2011 +0000 @@ -1,46 +1,41 @@ -local filters = require "util.filters"; -local config = {} -config.file = module:get_option_string("crossdomain_file", ""); -config.string = module:get_option_string("crossdomain_string", [[<?xml version="1.0"?><!DOCTYPE cross-domain-policy SYSTEM "/xml/dtds/cross-domain-policy.dtd"><cross-domain-policy><site-control permitted-cross-domain-policies="master-only"/><allow-access-from domain="*" /></cross-domain-policy>]]); -local string = '' -if not config.file ~= '' then - local f = assert(io.open(config.file)); - string = f:read("*all"); -else - string = config.string -end - -module:log("debug", "crossdomain string: "..string); - -module:set_global(); - -function filter_policy(data, session) - -- Since we only want to check the first block of data, remove the filter - filters.remove_filter(session, "bytes/in", filter_policy); - if data == "<policy-file-request/>\0" then - session.send([[<?xml version="1.0"?> - <!DOCTYPE cross-domain-policy SYSTEM "/xml/dtds/cross-domain-policy.dtd"> - <cross-domain-policy> - <site-control permitted-cross-domain-policies="master-only"/> - <allow-access-from domain="livechat.concepts.tim-online.nl" to-ports="5222" /> - </cross-domain-policy>]].."\0"); - return nil; -- Drop data to prevent it reaching the XMPP parser - else - return data; -- Pass data through, it wasn't a policy request - end - -end - -function filter_session(session) - if session.type == "c2s_unauthed" then - filters.add_filter(session, "bytes/in", filter_policy, -1); - end -end - -function module.load() - filters.add_filter_hook(filter_session); -end - -function module.unload() - filters.remove_filter_hook(filter_session); +local filters = require "util.filters"; +local config = {} +config.file = module:get_option_string("crossdomain_file", ""); +config.string = module:get_option_string("crossdomain_string", [[<?xml version="1.0"?><!DOCTYPE cross-domain-policy SYSTEM "/xml/dtds/cross-domain-policy.dt$ +local string = '' +if not config.file ~= '' then + local f = assert(io.open(config.file)); + string = f:read("*all"); +else + string = config.string +end + +module:log("debug", "crossdomain string: "..string); + +module:set_global(); + +function filter_policy(data, session) + -- Since we only want to check the first block of data, remove the filter + filters.remove_filter(session, "bytes/in", filter_policy); + if data == "<policy-file-request/>\0" then + session.send(string.."\0"); + return nil; -- Drop data to prevent it reaching the XMPP parser + else + return data; -- Pass data through, it wasn't a policy request + end + +end + +function filter_session(session) + if session.type == "c2s_unauthed" then + filters.add_filter(session, "bytes/in", filter_policy, -1); + end +end + +function module.load() + filters.add_filter_hook(filter_session); +end + +function module.unload() + filters.remove_filter_hook(filter_session); end \ No newline at end of file