2 -- Copyright (C) 2008-2009 Matthew Wild
3 -- Copyright (C) 2008-2009 Waqas Hussain
5 -- This project is MIT/X11 licensed. Please see the
6 -- COPYING file in the source package for more information.
10 module.host = "*" -- Global module
12 local httpserver = require "net.httpserver";
13 local st = require "util.stanza";
15 local unpack = unpack;
16 local tostring = tostring;
17 local is_admin = require "core.usermanager".is_admin;
18 local jid_split = require "util.jid".split;
19 local b64_decode = require "util.encodings".base64.decode;
20 local get_method = require "core.objectmanager".get_object;
21 local validate_credentials = require "core.usermanager".validate_credentials;
23 local translate_request = require "util.xmlrpc".translate_request;
24 local create_response = require "util.xmlrpc".create_response;
25 local create_error_response = require "util.xmlrpc".create_error_response;
27 local entity_map = setmetatable({
33 }, {__index = function(_, s)
34 if s:sub(1,1) == "#" then
35 if s:sub(2,2) == "x" then
36 return string.char(tonumber(s:sub(3), 16));
38 return string.char(tonumber(s:sub(2)));
43 local function xml_unescape(str)
44 return (str:gsub("&(.-);", entity_map));
46 local function parse_xml(xml)
47 local stanza = st.stanza("root");
48 local regexp = "<([^>]*)>([^<]*)";
49 for elem, text in xml:gmatch(regexp) do
50 --print("[<"..elem..">|"..text.."]");
51 if elem:sub(1,1) == "!" or elem:sub(1,1) == "?" then -- neglect comments and processing-instructions
52 elseif elem:sub(1,1) == "/" then -- end tag
54 stanza:up(); -- TODO check for start-end tag name match
55 elseif elem:sub(-1,-1) == "/" then -- empty tag
56 elem = elem:sub(1,-2);
57 stanza:tag(elem):up();
61 if #text ~= 0 then -- text
62 stanza:text(xml_unescape(text));
65 return stanza.tags[1];
68 local function handle_xmlrpc_request(method, args)
69 method = get_method(method);
70 if not method then return create_error_response(404, "method not found"); end
72 local success, result = pcall(method, unpack(args));
74 success, result = pcall(create_response, result or "nil");
78 return create_error_response(500, "Error in creating response: "..result);
80 return create_error_response(0, result or "nil");
83 local function handle_xmpp_request(origin, stanza)
84 local query = stanza.tags[1];
85 if query.name == "query" then
86 if #query.tags == 1 then
87 if is_admin(stanza.attr.from) then
88 local success, method, args = pcall(translate_request, query.tags[1]);
90 local result = handle_xmlrpc_request(method, args);
91 origin.send(st.reply(stanza):tag('query', {xmlns='jabber:iq:rpc'}):add_child(result));
93 origin.send(st.error_reply(stanza, "modify", "bad-request", method));
95 else origin.send(st.error_reply(stanza, "auth", "forbidden", "No content in XML-RPC request")); end
96 else origin.send(st.error_reply(stanza, "modify", "bad-request", "No content in XML-RPC request")); end
97 else origin.send(st.error_reply(stanza, "cancel", "service-unavailable")); end
99 module:add_iq_handler({"c2s", "s2sin"}, "jabber:iq:rpc", handle_xmpp_request);
100 module:add_feature("jabber:iq:rpc");
101 -- TODO add <identity category='automation' type='rpc'/> to disco replies
103 local default_headers = { ['Content-Type'] = 'text/xml' };
104 local unauthorized_response = { status = '401 UNAUTHORIZED', headers = {['Content-Type']='text/html', ['WWW-Authenticate']='Basic realm="WallyWorld"'}; body = "<html><body>Authentication required</body></html>"; };
105 local function handle_http_request(method, body, request)
107 local username, password = b64_decode(request['authorization'] or ''):gmatch('([^:]*):(.*)')(); -- TODO digest auth
108 local node, host = jid_split(username);
109 if not validate_credentials(host, node, password) and is_admin(username) then
110 return unauthorized_response;
113 local stanza = body and parse_xml(body);
114 if (not stanza) or request.method ~= "POST" then
115 return "<html><body>You really don't look like an XML-RPC client to me... what do you want?</body></html>";
118 local success, method, args = pcall(translate_request, stanza);
120 return { headers = default_headers; body = tostring(handle_xmlrpc_request(method, args)) };
122 return "<html><body>Error parsing XML-RPC request: "..tostring(method).."</body></html>";
124 httpserver.new{ port = 9000, base = "xmlrpc", handler = handle_http_request }