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
|
function send(text)
local text = tostring(text)
local len = #output
if len < 1 then
output = { text }
else
output[len] = output[len] .. text
end
end
function sendln(text)
send(text)
table.insert(output, "")
end
local sandbox_env = {
print = send,
println = sendln,
eval = nil,
args = args,
input = input,
user = user,
channel = channel,
request = download,
string = string,
math = math,
table = table,
pairs = pairs,
ipairs = ipairs,
next = next,
select = select,
unpack = unpack,
tostring = tostring,
tonumber = tonumber,
type = type,
assert = assert,
error = error,
pcall = pcall,
xpcall = xpcall,
_VERSION = _VERSION
}
sandbox_env.os = {
clock = os.clock,
time = os.time,
difftime = os.difftime
}
sandbox_env.string.rep = nil
sandbox_env.string.dump = nil
sandbox_env.math.randomseed = nil
-- Temporary evaluation function
function eval(code)
local c, e = load(code, nil, nil, sandbox_env)
if c then
return c()
else
error(e)
end
end
sandbox_env.eval = eval
-- Check if the factoid timed out
function checktime(event, line)
if os.time() - time >= timeout then
error("Timed out after " .. timeout .. " seconds", 0)
else
-- Limit the cpu usage of factoids
sleep(1)
end
end
local f, e = load(factoid, nil, nil, sandbox_env)
-- Add timeout hook
time = os.time()
timeout = 30
debug.sethook(checktime, "l")
if f then
f()
else
error(e)
end
|
