plugins/mod_debug.lua

   1 -- Prosody IM v0.4
   2 -- Copyright (C) 2008-2009 Matthew Wild
   3 -- Copyright (C) 2008-2009 Waqas Hussain
   4 --
   5 -- This project is MIT/X11 licensed. Please see the
   6 -- COPYING file in the source package for more information.
   7 --
   8
   9 module.host = "*";
  10
  11 local connlisteners_register = require "net.connlisteners".register;
  12
  13 local console_listener = { default_port = 5582; default_mode = "*l"; };
  14
  15 local sha256, missingglobal = require "util.hashes".sha256;
  16
  17 local commands = {};
  18 local debug_env = {};
  19 local debug_env_mt = { __index = function (t, k) return rawget(_G, k) or missingglobal(k); end, __newindex = function (t, k, v) rawset(_G, k, v); end };
  20
  21 local t_insert, t_concat = table.insert, table.concat;
  22 local t_concatall = function (t, sep) local tt = {}; for k, s in pairs(t) do tt[k] = tostring(s); end return t_concat(tt, sep); end
  23
  24
  25 setmetatable(debug_env, debug_env_mt);
  26
  27 console = {};
  28
  29 function console:new_session(conn)
  30         local w = function(s) conn.write(s:gsub("\n", "\r\n")); end;
  31         local session = { conn = conn;
  32                         send = function (t) w(tostring(t)); end;
  33                         print = function (t) w("| "..tostring(t).."\n"); end;
  34                         disconnect = function () conn.close(); end;
  35                         };
  36
  37         return session;
  38 end
  39
  40 local sessions = {};
  41
  42 function console_listener.listener(conn, data)
  43         local session = sessions[conn];
  44
  45         if not session then
  46                 -- Handle new connection
  47                 session = console:new_session(conn);
  48                 sessions[conn] = session;
  49                 printbanner(session);
  50         end
  51         if data then
  52                 -- Handle data
  53                 (function(session, data)
  54                         if data:match("[!.]$") then
  55                                 local command = data:lower();
  56                                 command = data:match("^%w+") or data:match("%p");
  57                                 if commands[command] then
  58                                         commands[command](session, data);
  59                                         return;
  60                                 end
  61                         end
  62
  63                         local chunk, err = loadstring("return "..data);
  64                         if not chunk then
  65                                 chunk, err = loadstring(data);
  66                                 if not chunk then
  67                                         err = err:gsub("^%[string .-%]:%d+: ", "");
  68                                         err = err:gsub("^:%d+: ", "");
  69                                         err = err:gsub("'<eof>'", "the end of the line");
  70                                         session.print("Sorry, I couldn't understand that... "..err);
  71                                         return;
  72                                 end
  73                         end
  74
  75                         debug_env.print = session.print;
  76
  77                         setfenv(chunk, debug_env);
  78
  79                         local ret = { pcall(chunk) };
  80
  81                         if not ret[1] then
  82                                 session.print("Fatal error while running command, it did not complete");
  83                                 session.print("Error: "..ret[2]);
  84                                 return;
  85                         end
  86
  87                         table.remove(ret, 1);
  88
  89                         local retstr = t_concatall(ret, ", ");
  90                         if retstr ~= "" then
  91                                 session.print("Result: "..retstr);
  92                         else
  93                                 session.print("No result, or nil");
  94                                 return;
  95                         end
  96                 end)(session, data);
  97         end
  98         session.send(string.char(0));
  99 end
 100
 101 function console_listener.disconnect(conn, err)
 102
 103 end
 104
 105 connlisteners_register('console', console_listener);
 106
 107 -- Console commands --
 108 -- These are simple commands, not valid standalone in Lua
 109
 110 function commands.bye(session)
 111         session.print("See you! :)");
 112         session.disconnect();
 113 end
 114
 115 commands["!"] = function (session, data)
 116         if data:match("^!!") then
 117                 session.print("!> "..session.env._);
 118                 return console_listener.listener(session.conn, session.env._);
 119         end
 120         local old, new = data:match("^!(.-[^\\])!(.-)!$");
 121         if old and new then
 122                 local ok, res = pcall(string.gsub, session.env._, old, new);
 123                 if not ok then
 124                         session.print(res)
 125                         return;
 126                 end
 127                 session.print("!> "..res);
 128                 return console_listener.listener(session.conn, res);
 129         end
 130         session.print("Sorry, not sure what you want");
 131 end
 132
 133 function printbanner(session)
 134 session.print [[
 135                    ____                \   /     _
 136                     |  _ \ _ __ ___  ___  _-_   __| |_   _
 137                     | |_) | '__/ _ \/ __|/ _ \ / _` | | | |
 138                     |  __/| | | (_) \__ \ |_| | (_| | |_| |
 139                     |_|   |_|  \___/|___/\___/ \__,_|\__, |
 140                     A study in simplicity            |___/
 141
 142 ]]
 143 session.print("Welcome to the Prosody debug console. For a list of commands, type: help");
 144 session.print("You may find more help on using this console in our online documentation at ");
 145 session.print("http://prosody.im/doc/debugconsole\n");
 146 end
 147
 148 local t_insert = table.insert;
 149 local byte, char = string.byte, string.char;
 150 local gmatch, gsub = string.gmatch, string.gsub;
 151
 152 local function vdecode(ciphertext, key)
 153         local keyarr = {};
 154         for l in gmatch(key, ".") do t_insert(keyarr, byte(l) - 32) end
 155         local pos, keylen = 0, #keyarr;
 156         return (gsub(ciphertext, ".",   function (letter)
 157                                                         if byte(letter) < 32 then return ""; end
 158                                                         pos = (pos%keylen)+1;
 159                                                         return char(((byte(letter) - 32 - keyarr[pos]) % 94) + 32);
 160                                                 end));
 161 end
 162
 163 local subst = {
 164         ["fc3a2603a0795a7d1b192704a3af95fa661e1c5bc63b393ebf75904fa53d3683"] =
 165                 [=[<M|V2n]c30, )Y|X1H" '7 %W3KI1zf6-(vY1(&[cf$[x-(s]=];
 166         ["40a0da62932391196c18baa1c297e97b14b27bf64689dbe7f8b3b9cfad6cfbee"] =
 167                 [=[]0W!RG6-**2t'%vzz^=8MWh&c<CA30xl;>c38]=];
 168         ["1ba18bc69e1584170a4ca5d676903141a79c629236e91afa2e14b3e6b0f75a19"] =
 169                 [=[dSU%3nc1*\1y)$8-0Ku[H5K&(-"x3cU^a-*cz{.$!w`9'KQV2Tv)WtN{]=];
 170         ["a4d8bdafa6ae55d75fc971d193eef41f89499a79dbd24f44999d06025fb7a4f9"] =
 171                 [=[+yNDbYHMP+a`&,d}&]S}7'Nz.3VUM4Ko8Z$42D2EdXNs$S)4!*-dq$|2
 172                 0WY+a)]+S%X.ndDVG6FVyzp7vVI9x}R14$\YfvvQ("4-$J!/dMT2uZ{+( )
 173                 Z%D0e&UI-L#M.o]=];
 174         ["7a2ea4b076b8df73131059ac54434337084fd86d05814b37b7beb510d74b2728"] =
 175                 [=[pR)eG%R7-6H}YM++v3'x .aJv)*x(3x wD4ZKy$R+53"+bw(R>Xe|>]=];
 176         }
 177
 178 function missingglobal(name)
 179         if sha256 then
 180                 local hash = sha256(name..name:reverse(), true);
 181
 182                 if subst[hash] then
 183                         return vdecode(subst[hash], hash);
 184                 end
 185         end
 186 end