Software /
code /
prosody
File
plugins/mod_xmlrpc.lua @ 1151:8096941b6734
util.stanza: Omit unused clone parameter from error_reply()
author | Waqas Hussain <waqas20@gmail.com> |
---|---|
date | Fri, 15 May 2009 06:39:53 +0500 |
parent | 896:2c0b9e3c11c3 |
child | 1523:841d61be198f |
line wrap: on
line source
-- Prosody IM v0.4 -- Copyright (C) 2008-2009 Matthew Wild -- Copyright (C) 2008-2009 Waqas Hussain -- -- This project is MIT/X11 licensed. Please see the -- COPYING file in the source package for more information. -- module.host = "*" -- Global module local httpserver = require "net.httpserver"; local st = require "util.stanza"; local pcall = pcall; local unpack = unpack; local tostring = tostring; local is_admin = require "core.usermanager".is_admin; local jid_split = require "util.jid".split; local b64_decode = require "util.encodings".base64.decode; local get_method = require "core.objectmanager".get_object; local validate_credentials = require "core.usermanager".validate_credentials; local translate_request = require "util.xmlrpc".translate_request; local create_response = require "util.xmlrpc".create_response; local create_error_response = require "util.xmlrpc".create_error_response; local entity_map = setmetatable({ ["amp"] = "&"; ["gt"] = ">"; ["lt"] = "<"; ["apos"] = "'"; ["quot"] = "\""; }, {__index = function(_, s) if s:sub(1,1) == "#" then if s:sub(2,2) == "x" then return string.char(tonumber(s:sub(3), 16)); else return string.char(tonumber(s:sub(2))); end end end }); local function xml_unescape(str) return (str:gsub("&(.-);", entity_map)); end local function parse_xml(xml) local stanza = st.stanza("root"); local regexp = "<([^>]*)>([^<]*)"; for elem, text in xml:gmatch(regexp) do --print("[<"..elem..">|"..text.."]"); if elem:sub(1,1) == "!" or elem:sub(1,1) == "?" then -- neglect comments and processing-instructions elseif elem:sub(1,1) == "/" then -- end tag elem = elem:sub(2); stanza:up(); -- TODO check for start-end tag name match elseif elem:sub(-1,-1) == "/" then -- empty tag elem = elem:sub(1,-2); stanza:tag(elem):up(); else -- start tag stanza:tag(elem); end if #text ~= 0 then -- text stanza:text(xml_unescape(text)); end end return stanza.tags[1]; end local function handle_xmlrpc_request(method, args) method = get_method(method); if not method then return create_error_response(404, "method not found"); end args = args or {}; local success, result = pcall(method, unpack(args)); if success then success, result = pcall(create_response, result or "nil"); if success then return result; end return create_error_response(500, "Error in creating response: "..result); end return create_error_response(0, result or "nil"); end local function handle_xmpp_request(origin, stanza) local query = stanza.tags[1]; if query.name == "query" then if #query.tags == 1 then if is_admin(stanza.attr.from) then local success, method, args = pcall(translate_request, query.tags[1]); if success then local result = handle_xmlrpc_request(method, args); origin.send(st.reply(stanza):tag('query', {xmlns='jabber:iq:rpc'}):add_child(result)); else origin.send(st.error_reply(stanza, "modify", "bad-request", method)); end else origin.send(st.error_reply(stanza, "auth", "forbidden", "No content in XML-RPC request")); end else origin.send(st.error_reply(stanza, "modify", "bad-request", "No content in XML-RPC request")); end else origin.send(st.error_reply(stanza, "cancel", "service-unavailable")); end end module:add_iq_handler({"c2s", "s2sin"}, "jabber:iq:rpc", handle_xmpp_request); module:add_feature("jabber:iq:rpc"); -- TODO add <identity category='automation' type='rpc'/> to disco replies local default_headers = { ['Content-Type'] = 'text/xml' }; local unauthorized_response = { status = '401 UNAUTHORIZED', headers = {['Content-Type']='text/html', ['WWW-Authenticate']='Basic realm="WallyWorld"'}; body = "<html><body>Authentication required</body></html>"; }; local function handle_http_request(method, body, request) -- authenticate user local username, password = b64_decode(request['authorization'] or ''):gmatch('([^:]*):(.*)')(); -- TODO digest auth local node, host = jid_split(username); if not validate_credentials(host, node, password) and is_admin(username) then return unauthorized_response; end -- parse request local stanza = body and parse_xml(body); if (not stanza) or request.method ~= "POST" then return "<html><body>You really don't look like an XML-RPC client to me... what do you want?</body></html>"; end -- execute request local success, method, args = pcall(translate_request, stanza); if success then return { headers = default_headers; body = tostring(handle_xmlrpc_request(method, args)) }; end return "<html><body>Error parsing XML-RPC request: "..tostring(method).."</body></html>"; end httpserver.new{ port = 9000, base = "xmlrpc", handler = handle_http_request }