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
|
#include "dnsdist-lua-hooks.hh"
#include "dnsdist-lua.hh"
#include "lock.hh"
#include "tcpiohandler.hh"
namespace dnsdist::lua::hooks
{
using ExitCallback = std::function<void()>;
using MaintenanceCallback = std::function<void()>;
using TicketsKeyAddedHook = std::function<void(const std::string&, size_t)>;
static LockGuarded<std::vector<ExitCallback>> s_exitCallbacks;
static LockGuarded<std::vector<MaintenanceCallback>> s_maintenanceHooks;
void runMaintenanceHooks(const LuaContext& context)
{
(void)context;
for (const auto& callback : *(s_maintenanceHooks.lock())) {
callback();
}
}
static void addMaintenanceCallback(const LuaContext& context, MaintenanceCallback callback)
{
(void)context;
s_maintenanceHooks.lock()->push_back(std::move(callback));
}
void clearMaintenanceHooks()
{
s_maintenanceHooks.lock()->clear();
}
void runExitCallbacks(const LuaContext& context)
{
(void)context;
for (const auto& callback : *(s_exitCallbacks.lock())) {
callback();
}
}
static void addExitCallback(const LuaContext& context, ExitCallback callback)
{
(void)context;
s_exitCallbacks.lock()->push_back(std::move(callback));
}
void clearExitCallbacks()
{
s_exitCallbacks.lock()->clear();
}
static void setTicketsKeyAddedHook(const LuaContext& context, const TicketsKeyAddedHook& hook)
{
(void)context;
TLSCtx::setTicketsKeyAddedHook([hook](const std::string& key) {
try {
auto lua = g_lua.lock();
hook(key, key.size());
}
catch (const std::exception& exp) {
warnlog("Error calling the Lua hook after new tickets key has been added: %s", exp.what());
}
});
}
void setupLuaHooks(LuaContext& luaCtx)
{
luaCtx.writeFunction("addMaintenanceCallback", [&luaCtx](const MaintenanceCallback& callback) {
setLuaSideEffect();
addMaintenanceCallback(luaCtx, callback);
});
luaCtx.writeFunction("addExitCallback", [&luaCtx](const ExitCallback& callback) {
setLuaSideEffect();
addExitCallback(luaCtx, callback);
});
luaCtx.writeFunction("setTicketsKeyAddedHook", [&luaCtx](const TicketsKeyAddedHook& hook) {
setLuaSideEffect();
setTicketsKeyAddedHook(luaCtx, hook);
});
}
}
|
