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
166
167
168
169
170
171
172
173
174
175
176
|
require "luarocks.require"
server = require "net.server"
require "socket"
require "ssl"
require "lxp"
function log(type, area, message)
print(type, area, message);
end
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;
------------------------------
sessions = {};
hosts = {
["localhost"] = {
type = "local";
connected = true;
sessions = {};
};
["getjabber.ath.cx"] = {
type = "local";
connected = true;
sessions = {};
};
}
local hosts, users = hosts, users;
local ssl_ctx = { mode = "server", protocol = "sslv23", key = "/home/matthew/ssl_cert/server.key",
certificate = "/home/matthew/ssl_cert/server.crt", capath = "/etc/ssl/certs", verify = "none", }
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);
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();
|
