Implement a memory for the controller (accesible via mem.) and cleanup code

Based on PilzAdam's code.
This commit is contained in:
Jeija 2013-01-13 11:05:04 +01:00
parent bd749ec4d4
commit e297a02ec2

View File

@ -26,6 +26,11 @@ rules.b = {x = 0, y = 0, z = 1}
rules.c = {x = 1, y = 0, z = 0} rules.c = {x = 1, y = 0, z = 0}
rules.d = {x = 0, y = 0, z = -1} rules.d = {x = 0, y = 0, z = -1}
------------------
-- Action stuff --
------------------
-- These helpers are required to set the portstates of the luacontroller
local get_real_portstates = function(pos) -- determine if ports are powered (by itself or from outside) local get_real_portstates = function(pos) -- determine if ports are powered (by itself or from outside)
ports = { ports = {
a = mesecon:is_power_on(mesecon:addPosRule(pos, rules.a)) a = mesecon:is_power_on(mesecon:addPosRule(pos, rules.a))
@ -80,7 +85,10 @@ local action = function (pos, ports)
action_setports (pos, ports, vports) action_setports (pos, ports, vports)
end end
-- Overheat Stuff --------------------
-- Overheat stuff --
--------------------
local heat = function (meta) -- warm up local heat = function (meta) -- warm up
h = meta:get_int("heat") h = meta:get_int("heat")
if h ~= nil then if h ~= nil then
@ -106,14 +114,14 @@ local overheat = function (meta) -- determine if too hot
end end
local overheat_off = function(pos) local overheat_off = function(pos)
rules = mesecon:get_rules("mesecons_microcontroller:microcontroller1111") mesecon:receptor_off(pos, mesecon.rules.flat)
mesecon:receptor_off(pos, rules)
end end
local update = function (pos) -------------------
local meta = minetest.env:get_meta(pos) -- Parsing stuff --
code = meta:get_string("code") -------------------
local code_prohibited = function(code)
-- Clean code -- Clean code
local prohibited = {"while", "for", "repeat", "until"} local prohibited = {"while", "for", "repeat", "until"}
for _, p in ipairs(prohibited) do for _, p in ipairs(prohibited) do
@ -121,27 +129,36 @@ local update = function (pos)
return "Prohibited command: "..p return "Prohibited command: "..p
end end
end end
end
local safeprint = function(param)
print(dump(param))
end
local create_environment = function(pos, mem)
-- Gather variables for the environment -- Gather variables for the environment
local vports = minetest.registered_nodes[minetest.env:get_node(pos).name].virtual_portstates local vports = minetest.registered_nodes[minetest.env:get_node(pos).name].virtual_portstates
vports = {a = vports.a, b = vports.b, c = vports.c, d = vports.d} vports = {a = vports.a, b = vports.b, c = vports.c, d = vports.d}
local rports = get_real_portstates(pos) local rports = get_real_portstates(pos)
local env = { print = print, return { print = safeprint,
selfpos = pos,
dump = dump,
pin = merge_portstates(vports, rports), pin = merge_portstates(vports, rports),
port = vports} port = vports,
mem = mem}
end
local create_sandbox = function (code, env)
-- Create Sandbox -- Create Sandbox
if code:byte(1) == 27 then if code:byte(1) == 27 then
return "You Hacker You! Don't use binary code!" return _, "You Hacker You! Don't use binary code!"
end end
local f, msg = loadstring(code) f, msg = loadstring(code)
if not f then return msg end if not f then return _, msg end
setfenv(f, env) setfenv(f, env)
success, msg = pcall(f) return f
end
local do_overheat = function (pos, meta)
-- Overheat protection -- Overheat protection
heat(meta) heat(meta)
minetest.after(0.5, cool, meta) minetest.after(0.5, cool, meta)
@ -151,14 +168,41 @@ local update = function (pos)
minetest.env:add_item(pos, BASENAME.."0000") minetest.env:add_item(pos, BASENAME.."0000")
return return
end end
end
local load_memory = function(meta)
return minetest.deserialize(meta:get_string("lc_memory")) or {}
end
local save_memory = function(meta, mem)
meta:set_string("lc_memory", minetest.serialize(mem))
end
----------------------
-- Parsing function --
----------------------
local update = function (pos)
local meta = minetest.env:get_meta(pos)
local mem = load_memory(meta)
local code = meta:get_string("code")
local prohibited = code_prohibited(code)
if prohibited then return prohibited end
local env = create_environment(pos, mem)
local chunk, msg = create_sandbox (code, env)
if not chunk then return msg end
local success, msg = pcall(f)
if not success then return msg end
do_overheat(pos, meta)
save_memory(meta, mem)
-- Actually set the ports -- Actually set the ports
if not success then
return msg
else
action(pos, env.port) action(pos, env.port)
end end
end
local reset_meta = function(pos, code, errmsg) local reset_meta = function(pos, code, errmsg)
local meta = minetest.env:get_meta(pos) local meta = minetest.env:get_meta(pos)
@ -186,7 +230,10 @@ end
-- | | | | |\ | | |_| | | | | |_ |_| -- | | | | |\ | | |_| | | | | |_ |_|
-- | |______ |__| | \| | | \ |__| |_ |_ |_ |\ -- | |______ |__| | \| | | \ |__| |_ |_ |_ |\
-- --
-- Actually register all the stuff:
-----------------------
-- Node Registration --
-----------------------
for a = 0, 1 do for a = 0, 1 do
for b = 0, 1 do for b = 0, 1 do
@ -293,6 +340,10 @@ end
end end
end end
------------------------
-- Craft Registration --
------------------------
minetest.register_craft({ minetest.register_craft({
output = BASENAME.."0000 2", output = BASENAME.."0000 2",
recipe = { recipe = {
@ -301,3 +352,4 @@ minetest.register_craft({
{'group:mesecon_conductor_craftable', 'group:mesecon_conductor_craftable', ''}, {'group:mesecon_conductor_craftable', 'group:mesecon_conductor_craftable', ''},
} }
}) })