aboutsummaryrefslogtreecommitdiffstats
path: root/main.lua
blob: ddab301b301f73a5ba84cb182367113e4149d4d6 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
require "luarocks.require"

server = require "net.server"
require "socket"
require "ssl"
require "lxp"

function log(type, area, message)
	print(type, area, message);
end

dofile "lxmppd.cfg"
 
sessions = {};
 
require "core.stanza_dispatch"
require "core.xmlhandlers"
require "core.rostermanager"
require "core.offlinemessage"
require "core.modulemanager"
require "core.usermanager"
require "core.sessionmanager"
require "core.stanza_router"
require "util.stanza"
require "util.jid"
 
-- Locals for faster access --
local t_insert = table.insert;
local t_concat = table.concat;
local t_concatall = function (t, sep) local tt = {}; for _, s in ipairs(t) do t_insert(tt, tostring(s)); end return t_concat(tt, sep); end
local m_random = math.random;
local format = string.format;
local st = stanza;
local init_xmlhandlers = xmlhandlers.init_xmlhandlers;
------------------------------



local hosts, users = hosts, users;

function connect_host(host)
	hosts[host] = { type = "remote", sendbuffer = {} };
end

local function route_stanza(stanza)
	if not stanza.attr.to then
		-- Has no 'to' attribute, handle internally
	end
	local node, host, resource = jid.split(stanza.attr.to);
	if host and hosts[host] and hosts[host].type == "local" then
			-- Is a local host, handle internally
			
	else
		-- Is not for us or a local user, route accordingly
	end
end

local function send_to(session, to, stanza)
	local node, host, resource = jid.split(to);
	if not hosts[host] then
		-- s2s
	elseif hosts[host].type == "local" then
		print("   ...is to a local user")
		local destuser = hosts[host].sessions[node];
		if destuser and destuser.sessions then
			if not destuser.sessions[resource] then
				local best_session;
				for resource, session in pairs(destuser.sessions) do
					if not best_session then best_session = session;
					elseif session.priority >= best_session.priority and session.priority >= 0 then
						best_session = session;
					end
				end
				if not best_session then
					offlinemessage.new(node, host, stanza);
				else
					print("resource '"..resource.."' was not online, have chosen to send to '"..best_session.username.."@"..best_session.host.."/"..best_session.resource.."'");
					resource = best_session.resource;
				end
			end
			if destuser.sessions[resource] == session then
				log("warn", "core", "Attempt to send stanza to self, dropping...");
			else
				print("...sending...", tostring(stanza));
				--destuser.sessions[resource].conn.write(tostring(data));
				print("   to conn ", destuser.sessions[resource].conn);
				destuser.sessions[resource].conn.write(tostring(stanza));
				print("...sent")
			end
		elseif stanza.name == "message" then
			print("   ...will be stored offline");
			offlinemessage.new(node, host, stanza);
		elseif stanza.name == "iq" then
			print("   ...is an iq");
			session.send(st.reply(stanza)
				:tag("error", { type = "cancel" })
					:tag("service-unavailable", { xmlns = "urn:ietf:params:xml:ns:xmpp-stanzas" }));
		end
		print("   ...done routing");
	end
end

function handler(conn, data, err)
	local session = sessions[conn];

	if not session then
		sessions[conn] = sessionmanager.new_session(conn);
		session = sessions[conn];

		-- Logging functions --

		local mainlog, log = log;
		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;

		print("Client connected");
		
		session.stanza_dispatch = function (stanza) return core_process_stanza(session, stanza); end
		session.xml_handlers = init_xmlhandlers(session);
		session.parser = lxp.new(session.xml_handlers, ":");
			
		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;
			end
			session = nil;
			print("Disconnected: "..err);
			collectgarbage("collect");
		end
	end
	if data then
		session.parser:parse(data);
	end
	
	--log("info", "core", "Client disconnected, connection closed");
end

function disconnect(conn, err)
	sessions[conn].disconnect(err);
	sessions[conn] = nil;
end

modulemanager.loadall();

setmetatable(_G, { __index = function (t, k) print("WARNING: ATTEMPT TO READ A NIL GLOBAL!!!", k); error("Attempt to read a non-existent global. Naughty boy.", 2); end, __newindex = function (t, k, v) print("ATTEMPT TO SET A GLOBAL!!!!", tostring(k).." = "..tostring(v)); error("Attempt to set a global. Naughty boy.", 2); end }) --]][][[]][];


local protected_handler = function (conn, data, err) local success, ret = pcall(handler, conn, data, err); if not success then print("ERROR on "..tostring(conn)..": "..ret); conn:close(); end end;
local protected_disconnect = function (conn, err) local success, ret = pcall(disconnect, conn, err); if not success then print("ERROR on "..tostring(conn).." disconnect: "..ret); conn:close(); end end;

server.add( { listener = protected_handler, disconnect = protected_disconnect }, 5222, "*", 1, nil ) -- server.add will send a status message
server.add( { listener = protected_handler, disconnect = protected_disconnect }, 5223, "*", 1, ssl_ctx ) -- server.add will send a status message

server.loop();