# HG changeset patch # User Matthew Wild # Date 1219614507 -3600 # Node ID dcc5ac721c20860922218a037a6150f5b55fe161 # Parent 7ad47ce20394591af103679add3a18bb49a38a08 - Remove some debugging code - Don't allow stanzas to be sent before successful auth (needs further review) - Make init_xmlhandlers local - Fix indentation in main.lua - Fix disconnect logic (it broke if client disconnected before auth, etc.) diff -r 7ad47ce20394 -r dcc5ac721c20 DEPENDS --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/DEPENDS Sun Aug 24 22:48:27 2008 +0100 @@ -0,0 +1,8 @@ +The easiest way to install dependencies is using the luarocks tool. + +Rocks: +luaexpat +luasocket + +Non-rocks: +LuaSec for SSL connections diff -r 7ad47ce20394 -r dcc5ac721c20 core/stanza_dispatch.lua --- a/core/stanza_dispatch.lua Sun Aug 24 18:08:04 2008 +0100 +++ b/core/stanza_dispatch.lua Sun Aug 24 22:48:27 2008 +0100 @@ -32,7 +32,6 @@ return true; else username, password, resource = t_concat(username), t_concat(password), t_concat(resource); - print(username, password, resource) local reply = st.reply(stanza); require "core.usermanager" if usermanager.validate_credentials(session.host, username, password) then @@ -96,7 +95,9 @@ send(format("", stanza.attr.id)); return; end - elseif stanza.name == "presence" then + end + if not session.username then log("warn", "Attempt to use an unauthed stream!"); return; end + if stanza.name == "presence" then if session.roster then local initial_presence = not session.last_presence; session.last_presence = stanza; @@ -112,7 +113,6 @@ broadcast.attr.to = contact_jid; send_to(contact_jid, broadcast); if initial_presence then - print("Initital presence"); local node, host = jid.split(contact_jid); if hosts[host] and hosts[host].type == "local" then local contact = hosts[host].sessions[node] @@ -137,7 +137,7 @@ -- Probe for our contacts' presence end end - else + elseif session.username then --end --if stanza.attr.to and ((not hosts[stanza.attr.to]) or hosts[stanza.attr.to].type ~= "local") then -- Need to route stanza @@ -147,4 +147,3 @@ end end - diff -r 7ad47ce20394 -r dcc5ac721c20 core/xmlhandlers.lua --- a/core/xmlhandlers.lua Sun Aug 24 18:08:04 2008 +0100 +++ b/core/xmlhandlers.lua Sun Aug 24 22:48:27 2008 +0100 @@ -42,7 +42,7 @@ session.streamid = m_random(1000000, 99999999); print(session, session.host, "Client opened stream"); send(""); - send(format("", session.streamid, session.host)); + send(format("", session.streamid, session.host)); --send(""); --send("PLAIN"); --send [[ ]] diff -r 7ad47ce20394 -r dcc5ac721c20 main.lua --- a/main.lua Sun Aug 24 18:08:04 2008 +0100 +++ b/main.lua Sun Aug 24 22:48:27 2008 +0100 @@ -10,7 +10,7 @@ end require "core.stanza_dispatch" -init_xmlhandlers = require "core.xmlhandlers" +local init_xmlhandlers = require "core.xmlhandlers" require "core.rostermanager" require "core.offlinemessage" require "core.usermanager" @@ -103,7 +103,7 @@ function handler(conn, data, err) local session = sessions[conn]; - + if not session then sessions[conn] = { conn = conn, notopen = true, priority = 0 }; session = sessions[conn]; @@ -114,6 +114,7 @@ do local conn_name = tostring(conn):match("%w+$"); log = function (type, area, message) mainlog(type, conn_name, message); end + --log = function () end end local print = function (...) log("info", "core", t_concatall({...}, "\t")); end session.log = log; @@ -125,24 +126,27 @@ local send = function (data) print("Sending...", tostring(data)); conn.write(tostring(data)); end; session.send, session.send_to = send, send_to; - print("Client connected"); + print("Client connected"); - session.stanza_dispatch = init_stanza_dispatcher(session); - session.xml_handlers = init_xmlhandlers(session); - session.parser = lxp.new(session.xml_handlers, ":"); + session.stanza_dispatch = init_stanza_dispatcher(session); + session.xml_handlers = init_xmlhandlers(session); + session.parser = lxp.new(session.xml_handlers, ":"); - function session.disconnect(err) - if session.last_presence.attr.type ~= "unavailable" then - local pres = st.presence{ type = "unavailable" }; - if err == "closed" then err = "connection closed"; end - pres:tag("status"):text("Disconnected: "..err); - session.stanza_dispatch(pres); - end + function session.disconnect(err) + if session.last_presence and session.last_presence.attr.type ~= "unavailable" then + local pres = st.presence{ type = "unavailable" }; + if err == "closed" then err = "connection closed"; end + pres:tag("status"):text("Disconnected: "..err); + session.stanza_dispatch(pres); + end + if session.username then hosts[session.host].sessions[session.username] = nil; - session = nil; - print("Disconnected: "..err); end + session = nil; + print("Disconnected: "..err); + collectgarbage("collect"); end + end if data then session.parser:parse(data); end