prosody
6f840763fc73
net/server_epoll.lua

Find changesets by keywords by author, files, the commit message, revision number or hash, or revset expression.

Software / code / prosody

Comparison

net/server_epoll.lua @ 13490:6f840763fc73

net.server_epoll: Add support for systemd socket activation Allows creating listening sockets and accepting client connections before Prosody starts. This is unlike normal Prosody dynamic resource management, where ports may added and removed at any time, and the ports defined by the config. Weird things happen if these are closed (e.g. due to reload) so here we prevent closing and ensure sockets are reused when opened again.
author Kim Alvefur <zash@zash.se>
date Tue, 14 May 2024 17:07:47 +0200
parent 13450:f36e1e1b84fe
child 13552:f0b2c026e542
comparison
equal deleted inserted replaced
13489:ae65f199f408 13490:6f840763fc73
33 33
34 local poller = require "prosody.util.poll" 34 local poller = require "prosody.util.poll"
35 local EEXIST = poller.EEXIST; 35 local EEXIST = poller.EEXIST;
36 local ENOENT = poller.ENOENT; 36 local ENOENT = poller.ENOENT;
37 37
38 -- systemd socket activation
39 local SD_LISTEN_FDS_START = 3;
40 local SD_LISTEN_FDS = tonumber(os.getenv("LISTEN_FDS")) or 0;
41
42 local inherited_sockets = setmetatable({}, {
43 __index = function(t, k)
44 local serv_mt = debug.getregistry()["tcp{server}"];
45 for i = 1, SD_LISTEN_FDS do
46 local serv = socket.tcp();
47 if serv:getfd() ~= _SOCKETINVALID then
48 -- If LuaSocket allocated a FD for then we can't really close it and it would leak.
49 log("error", "LuaSocket not compatible with socket activation. Upgrade LuaSocket or disable socket activation.");
50 setmetatable(t, nil);
51 break
52 end
53 serv:setfd(SD_LISTEN_FDS_START + i - 1);
54 debug.setmetatable(serv, serv_mt);
55 serv:settimeout(0);
56 local ip, port = serv:getsockname();
57 t[ip .. ":" .. port] = serv;
58 if ip == "0.0.0.0" then
59 -- LuaSocket treats '*' as an alias for '0.0.0.0'
60 t["*:" .. port] = serv;
61 end
62 end
63
64 -- Disable lazy-loading mechanism once performed
65 setmetatable(t, nil);
66 return t[k];
67 end;
68 });
69
38 local poll = assert(poller.new()); 70 local poll = assert(poller.new());
39 71
40 local _ENV = nil; 72 local _ENV = nil;
41 -- luacheck: std none 73 -- luacheck: std none
42 74
942 server:add(true, false); 974 server:add(true, false);
943 return server; 975 return server;
944 end 976 end
945 977
946 local function listen(addr, port, listeners, config) 978 local function listen(addr, port, listeners, config)
979 local inherited = inherited_sockets[addr .. ":" .. port];
980 if inherited then
981 local conn = wrapserver(inherited, addr, port, listeners, config);
982 -- sockets created by systemd must not be :close() since we may not have
983 -- privileges to create them
984 conn.destroy = interface.del;
985 return conn;
986 end
947 local conn, err = socket.bind(addr, port, cfg.tcp_backlog); 987 local conn, err = socket.bind(addr, port, cfg.tcp_backlog);
948 if not conn then return conn, err; end 988 if not conn then return conn, err; end
949 conn:settimeout(0); 989 conn:settimeout(0);
950 return wrapserver(conn, addr, port, listeners, config); 990 return wrapserver(conn, addr, port, listeners, config);
951 end 991 end