prosody-modules: mod_ircd/mod_ircd.in.lua comparison

comparison mod_ircd/mod_ircd.in.lua @ 488:4885ca74515c

mod_ircd: corrected an issue within the nick change logic, and improved it by implementing the USER command.

author	Marco Cirillo <maranda@lightwitch.org>
date	Fri, 02 Dec 2011 04:24:16 +0000
parents	8bdab5489653
children	067bbff6e5bd

comparison

equal deleted inserted replaced

-:8bdab5489653
+:4885ca74515c
 local sessions = {};
 local jids = {};
 local commands = {};
 local nicks = {};
+local usernames = {};
 local st = require "util.stanza";
 local conference_server = muc_server;
 function irc_listener.onincoming(conn, data)
 	local session = sessions[conn];
 	if not session then
 		session = { conn = conn, host = component_jid, reset_stream = function () end,
 			close = irc_close_session, log = logger.init("irc"..(conn.id or "1")),
-			rooms = {},
+			rooms = {}, roster = {}, has_un = false };
-			roster = {} };
 		sessions[conn] = session;
 		function session.data(data)
 			local parts = parse_line(data);
 			module:log("debug", require"util.serialization".serialize(parts));
 			local command = table.remove(parts, 1);
 			if not command then
 				return;
 			end
 			command = command:upper();
-			if not session.nick then
+			if not session.username and not session.nick then
 				if not (command == "USER" or command == "NICK") then
 					module:log("debug", "Client tried to send command %s before registering", command);
-					return session.send{from=muc_server, "451", command, "You have not registered"}
+					return session.send{from=muc_server, "451", command, "You have not completed the registration."}
 				end
 			end
 			if commands[command] then
 				local ret = commands[command](session, parts);
 				if ret then
 			nicks[session.nick] = nil;
 		end
 		if session.full_jid then
 			jids[session.full_jid] = nil;
 		end
+		if session.username then
+			usernames[session.username] = nil;
+		end
 	end
 	sessions[conn] = nil;
 end
 local function nick_inuse(nick)
 	if nicks[nick] then return true else return false end
 end
-local function change_nick_st(jid, room_jid, newnick)
+local function check_username(un)
-	return st.presence({ xmlns = xmlns_client, from = jid, to = room_jid, type = "unavailable" }):tag("status"):text("Changing nickname to "..newnick):up();
+	local count = 0;
-end
+	local result;
-function commands.NICK(session, args)
+	for name, given in pairs(usernames) do
-	local nick = args[1];
+		if un == given then count = count + 1; end
-	nick = nick:gsub("[^%w_]","");
+	end
-	local full_jid = jid.join(nick, component_jid, "ircd");
+	result = count + 1;
-	if session.nick and not nick_inuse(nick) then -- changing nick
-		local oldnick = session.nick;
+	if count > 0 then return tostring(un)..tostring(result); else return tostring(un); end
-		local old_full_jid = session.full_jid;
+end
-		local old_ar_last = jids[old_full_jid]["ar_last"];
+local function change_nick_st(fulljid, roomjid, tonick)
-		local old_nicks_changing = jids[old_full_jid]["nicks_changing"];
+	return st.presence({ from = fulljid, to = newjid, type = "unavailable" }):tag("status"):text("Changing nickname to: "..tonick):up();
+end
-		-- update and replace session data
+local function set_t_data(session, full_jid)
-		session.nick = nick;
+	session.full_jid = full_jid;
-		session.full_jid = full_jid;
-		jids[old_full_jid] = nil; nicks[oldnick] = nil;
-		nicks[nick] = session;
-		jids[full_jid] = session;
-		jids[full_jid]["ar_last"] = old_ar_last;
-		jids[full_jid]["nicks_changing"] = old_nicks_changing;
-		session.send{from=oldnick, "NICK", nick};
-		-- broadcast changes if required, todo: something better then forcing parting and rejoining
-		if session.rooms then
-			for id, room in pairs(session.rooms) do
-				session.nicks_changing[session.nick] = oldnick;
-				room:send(change_nick_st(old_full_jid, room.jid.."/"..oldnick, session.nick));
-				commands.JOIN(session, { id });
-			end
-			session.nicks_changing[session.nick] = nil;
-		end
-		return;
-	elseif nick_inuse(nick) then
-		session.send{from=muc_server, "433", nick, "The nickname "..nick.." is already in use"}; return;
-	end
 	jids[full_jid] = session;
 	jids[full_jid]["ar_last"] = {};
 	jids[full_jid]["nicks_changing"] = {};
-	nicks[nick] = session;
-	session.nick = nick;
+	if session.nick then nicks[session.nick] = session; end
-	session.full_jid = full_jid;
+end
-	session.type = "c2s";
+local function send_motd(session)
+	local nick = session.nick;
 	session.send{from = muc_server, "001", nick, "Welcome in the IRC to MUC XMPP Gateway, "..nick};
 	session.send{from = muc_server, "002", nick, "Your host is "..muc_server.." running Prosody "..prosody.version};
 	session.send{from = muc_server, "003", nick, "This server was created the "..os.date(nil, prosody.start_time)}
 	session.send{from = muc_server, "004", nick, table.concat({muc_server, "mod_ircd(alpha-0.8)", "i", "aoqv"}, " ")};
 	session.send{from = muc_server, "375", nick, "- "..muc_server.." Message of the day -"};
 	session.send{from = nick, "MODE", nick, "+i"}; -- why -> Invisible mode setting,
 						       --        enforce by default on most servers (since the source host doesn't show it's sensible to have it "set")
 end
-function commands.USER(session, params) -- To be done.
+function commands.NICK(session, args)
+	local nick = args[1];
+	nick = nick:gsub("[^%w_]","");
+	if session.nick and not nick_inuse(nick) then -- changing nick
+		local oldnick = session.nick;
+		-- update and replace session data
+		session.nick = nick;
+		nicks[oldnick] = nil;
+		nicks[nick] = session;
+		session.send{from=oldnick.."!"..nicks[nick].username, "NICK", nick};
+		-- broadcast changes if required
+		if session.rooms then
+			for id, room in pairs(session.rooms) do
+				session.nicks_changing[session.nick] = { oldnick, session.username };
+				local node = jid.split(room.jid);
+				local oldjid = jid.join(node, muc_server, session.nick);
+				local room_name = room.jid
+				room:send(change_nick_st(session.full_jid, jid.join(node, muc_server, oldnick), session.nick));
+				local room, err = c:join_room(room_name, session.nick, { source = session.full_jid } );
+				if not room then
+					session.send{from=nick.nick.."!"..session.username, "PART", id};
+					return ":"..muc_server.." ERR :Failed to change nick and rejoin: "..err
+				end
+			end
+		end
+		return;
+	elseif nick_inuse(nick) then
+		session.send{from=muc_server, "433", nick, "The nickname "..nick.." is already in use"}; return;
+	end
+	session.nick = nick;
+	session.type = "c2s";
+	nicks[nick] = session;
+	-- Some choppy clients send in NICK before USER, that needs to be handled
+	if session.username then
+		set_t_data(session, jid.join(session.username, component_jid, "ircd"));
+	end
+	if session.username and session.nick then -- send MOTD
+		send_motd(session);
+	end
+end
+function commands.USER(session, params)
+	local username = params[1];
+	if not session.has_un then
+		local un_checked = check_username(username);
+		usernames[un_checked] = username;
+		session.username = un_checked;
+		session.has_un = true;
+		if not session.full_jid then
+			set_t_data(session, jid.join(session.username, component_jid, "ircd"));
+		end
+	else
+		return session.send{from=muc_server, "462", "USER", "You may not re-register."}
+	end
+	if session.username and session.nick then -- send MOTD
+		send_motd(session);
+	end
 end
 local function mode_map(am, rm, nicks)
 	local rnick;
 	local c_modes;
 	room.session = session;
 if session.nicks_changing[session.nick] then -- my own nick is changing
 	commands.NAMES(session, channel);
 else
-	session.send{from=session.nick, "JOIN", channel};
+	session.send{from=session.nick.."!"..session.username, "JOIN", channel};
 	if room.subject then
 	        session.send{from=muc_server, 332, session.nick, channel, room.subject};
 	end
 	commands.NAMES(session, channel);
 end
 	local session = room.session or jids[room.opts.source];
 local channel = "#"..room.jid:match("^(.*)@");
 	room:hook("occupant-joined", function(nick)
 		if session.nicks_changing[nick.nick] then
-			session.send{from=session.nicks_changing[nick.nick], "NICK", nick.nick};
+			session.send{from=session.nicks_changing[nick.nick][1].."!"..(session.nicks_changing[nick.nick][2] or "xmpp"), "NICK", nick.nick};
 			session.nicks_changing[nick.nick] = nil;
 		else
-			session.send{from=nick.nick, "JOIN", channel};
+			session.send{from=nick.nick.."!"..(nicks[nick.nick] and nicks[nick.nick].username or "xmpp"), "JOIN", channel};
 		end
 	end);
 	room:hook("occupant-left", function(nick)
 		if jids[session.full_jid] then jids[session.full_jid].ar_last[nick.jid:match("^(.*)/")][nick.nick] = nil; end
 		local status_code =
 		nick.presence:get_child("x","http://jabber.org/protocol/muc#user") and
 		nick.presence:get_child("x","http://jabber.org/protocol/muc#user"):get_child("status") and
 		nick.presence:get_child("x","http://jabber.org/protocol/muc#user"):get_child("status").attr.code;
 		if status_code == "303" then
 			local newnick =
 			nick.presence:get_child("x","http://jabber.org/protocol/muc#user") and
-			nick.presence:get_child("x","http://jabber.org/protocol/muc#user"):get_child("status") and
+			nick.presence:get_child("x","http://jabber.org/protocol/muc#user"):get_child("item") and
-			nick.presence:get_child("x","http://jabber.org/protocol/muc#user"):get_child("status"):get_child("item") and
+			nick.presence:get_child("x","http://jabber.org/protocol/muc#user"):get_child("item").attr.nick;
-			nick.presence:get_child("x","http://jabber.org/protocol/muc#user"):get_child("status"):get_child("item").attr.nick;
-			session.nicks_changing[newnick] = nick.nick; return;
+			session.nicks_changing[newnick] = { nick.nick, (nicks[nick.nick] and nicks[nick.nick].username or "xmpp") }; return;
 		end
-		session.send{from=nick.nick, "PART", channel};
+		local self_change;
+		for _, data in pairs(session.nicks_changing) do
+			if data[1] == nick.nick then self_change = nick.nick; break; end
+		end
+		if self_change then return; end
+		session.send{from=nick.nick.."!"..(nicks[nick.nick] and nicks[nick.nick].username or "xmpp"), "PART", channel};
 	end);
 end);
 function commands.NAMES(session, channel)
 	local nicks = { };
 	if type(channel) == "table" then channel = channel[1] end
 	local room = session.rooms[channel];
 	local symbols_map = {
 		owner = "~",
 		administrator = "&",
 	for _, room in pairs(session.rooms) do
 		room:leave(args[1]);
 	end
 	jids[session.full_jid] = nil;
 	nicks[session.nick] = nil;
+	usernames[session.username] = nil;
 	sessions[session.conn] = nil;
 	session:close();
 end
 function commands.RAW(session, data) -- Empty command

Mercurial > prosody-modules

comparison mod_ircd/mod_ircd.in.lua @ 488:4885ca74515c