diff mod_rawdebug/mod_rawdebug.lua @ 1444:56c394b9e60d

mod_rawdebug: Adds a telnet command for enabling logging of entire stanzas that are sent and received
author Kim Alvefur <zash@zash.se>
date Mon, 23 Jun 2014 20:22:19 +0200
parents
children ae1d7665cde9
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/mod_rawdebug/mod_rawdebug.lua	Mon Jun 23 20:22:19 2014 +0200
@@ -0,0 +1,55 @@
+
+local tostring = tostring;
+local filters = require "util.filters";
+
+local def_env = module:shared("/*/admin_telnet/env");
+local rawdebug_enabled = module:shared("sessions");
+local full_sessions = prosody.full_sessions;
+local log = module._log;
+
+local rawdebug = {};
+def_env.rawdebug = rawdebug;
+
+local function new_logger(log, prefix)
+	local msg = prefix .. ": %s";
+	return function (data)
+		log("debug", msg, tostring(data))
+		return data;
+	end
+end
+
+function rawdebug:enable(sessionid)
+	local session = full_sessions[sessionid];
+	if not session then
+		return nil, "No such session";
+	end
+	local f = {
+		["stanzas/in"]  = new_logger(session.log or log, "RECV");
+		["stanzas/out"] = new_logger(session.log or log, "SEND");
+	};
+	for type, callback in pairs(f) do
+		filters.add_filter(session, type, callback)
+	end
+	rawdebug_enabled[session] = f;
+end
+
+function rawdebug:disable(sessionid)
+	local session = full_sessions[sessionid];
+	if not session then
+		return nil, "No such session";
+	end
+	local f = rawdebug_enabled[session];
+	for type, callback in pairs(f) do
+		filters.remove_filter(session, type, callback)
+	end
+end
+
+function module.unload()
+	def_env.rawdebug = nil;
+	for session, f in pairs(rawdebug_enabled) do
+		for type, callback in pairs(f) do
+			filters.remove_filter(session, type, callback)
+		end
+	end
+end
+