dome/daemon.lua at master · ring0networks/dome · GitHub

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
--
-- SPDX-FileCopyrightText: (c) 2024-2025 Ring Zero Desenvolvimento de Software LTDA
-- SPDX-License-Identifier: GPL-2.0-only
--

local linux      = require("linux")
local runner     = require("lunatik.runner")
local thread     = require("thread")
local inet       = require("socket.inet")
local mailbox    = require("mailbox")
local data       = require("data")
local config     = require("dome/config")
local reply      = require("dome/reply")

local shouldstop = thread.shouldstop

local inbox = mailbox.inbox(config.mailbox_max)

local telegraf = inet.udp()
function telegraf:push(message)
	self:sendto(message, inet.localhost, 8094)
end

local function handle()
	local message = inbox:receive()
	local header, body = message:match("^(%a*)|(.+)")
	if header == "notify" then
		telegraf:push(config.notify_header .. body)
	elseif header == "reply" then
		local what, frame = body:match('(%a*)|(.*)')
		local packet = data.new(#frame)
		packet:setstring(0, frame)
		reply[what](packet)
	else
		error("invalid message: %s", message)
	end
end

local function daemon()
	print("[ring-0/dome] started")
	while (not shouldstop()) do
		local ok, err = pcall(handle)
		if not ok then
			linux.schedule(100)
		end
	end
	print("[ring-0/dome] stopped")
end

runner.run("dome/filter", false):resume(inbox.queue, inbox.event)

return daemon