105 lines
2.5 KiB
Lua
105 lines
2.5 KiB
Lua
do -- so local works
|
|
-- task format:
|
|
-- {
|
|
-- ["n"] = "name",
|
|
-- ["c"] = coroutine of task,
|
|
-- ["e"] = { table, of, environment, variables },
|
|
-- ["p"] = parent pid,
|
|
-- ["u"] = user ID,
|
|
-- ["ep"]= event queue pointer
|
|
-- }
|
|
local tT,nP,rg,eq,p = {},1,_G,{},1 -- taskTable,nextPid,real _G,event queue
|
|
_G.cT,sbt,C,T,io = 0,{},coroutine,table,io -- currentTask,sandboxTable
|
|
function _G.spawn(n,f,e)
|
|
tT[nP] = {}
|
|
sbt[nP] = {}
|
|
setmetatable(sbt[nP],{__index=_G})
|
|
_ENV = sbt[nP]
|
|
tT[nP].c = coroutine.create(f)
|
|
_ENV = rg
|
|
tT[nP].n = n
|
|
tT[nP].p = cT or -1
|
|
if tT[cT] then
|
|
tT[nP].u,tT[nP].ep,tT[nP].e = tT[cT].u,tT[cT].ep,e or tT[cT].e or {}
|
|
else
|
|
tT[nP].u,tT[nP].ep,tT[nP].e = "superuser",1,{}
|
|
end
|
|
nP = nP + 1
|
|
end
|
|
function _G.sched()
|
|
_G.sched = nil
|
|
while #tT > 0 do
|
|
eq[#eq+1]={computer.pullSignal(p)} -- add the latest event to the eq
|
|
if #eq > 16 then
|
|
table.remove(eq,1) -- remove the earliest if the eq is full
|
|
for pid,proc in pairs(tT) do
|
|
if proc.ep > 1 then
|
|
proc.ep = proc.ep - 1 -- decrement pointers for tasks to keep them on the same ones
|
|
end
|
|
end
|
|
end
|
|
for pid,proc in pairs(tT) do
|
|
if coroutine.status(proc.c) == "dead" then
|
|
tT[pid] = nil
|
|
else
|
|
cT=pid
|
|
for i = 1,tT[pid].ep, #eq do
|
|
sbt[pid].ev = eq[#eq] -- make an ev for MultICE compat
|
|
coroutine.resume(proc.c)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
_G.event = {}
|
|
function event.get() -- get the next event in the queue, or nil
|
|
if eq[tT[cT].ep] then
|
|
tT[cT].ep = tT[cT].ep + 1
|
|
if tT[cT].ep > 17 then
|
|
tT[cT].ep = 17
|
|
end
|
|
return eq[tT[cT].ep - 1]
|
|
end
|
|
end
|
|
function event.pull(t,time) -- return or wait for the next event, optionally with the first param matching t
|
|
time = time or math.huge
|
|
time = time + computer.uptime()
|
|
while computer.uptime() < time do
|
|
local e = event.get()
|
|
if e then
|
|
if t then
|
|
if e[1] == t then
|
|
return table.unpack(e)
|
|
end
|
|
else
|
|
return table.unpack(e)
|
|
end
|
|
end
|
|
coroutine.yield()
|
|
end
|
|
end
|
|
event.push = computer.pushSignal
|
|
function os.getenv(k) -- gets an envionment variable from tT[cT].e
|
|
if tT[cT] then
|
|
return tT[cT].e[k]
|
|
end
|
|
end
|
|
function os.setenv(k,v) -- sets "
|
|
if tT[cT] then
|
|
tT[cT].e[k] = v
|
|
end
|
|
end
|
|
function os.tasks() -- returns a table of tasks running on the system
|
|
local t = {}
|
|
for k,v in pairs(tT) do
|
|
t[k] = v.n
|
|
end
|
|
return t
|
|
end
|
|
function os.taskinfo(pid) -- returns info on a task
|
|
if tT[pid] then
|
|
return tT[pid].n,tT[pid].p,tT[pid].u
|
|
end
|
|
end
|
|
end
|