From f9c800008044d529bbffbb24e87b3ba49cdcd719 Mon Sep 17 00:00:00 2001 From: XeonSquared Date: Tue, 26 Nov 2019 21:34:40 +1100 Subject: [PATCH] added an mtrpc library --- lib/rpc.lua | 79 +++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 79 insertions(+) create mode 100644 lib/rpc.lua diff --git a/lib/rpc.lua b/lib/rpc.lua new file mode 100644 index 0000000..fd9286f --- /dev/null +++ b/lib/rpc.lua @@ -0,0 +1,79 @@ +local serial = require "serialization" +local minitel = require "minitel" +local event = require "event" +local rpc = {} +_G.rpcf = {} +rpc.port = 111 + +local function rpcexec(_, from, port, data) + if port == rpc.port then + local rpcrq = serial.unserialize(data) + local rpcn, rpcid = table.remove(rpcrq,1), table.remove(rpcrq,1) + if rpcf[rpcn] then + local rt = {pcall(rpcf[rpcn],table.unpack(rpcrq))} + if rt[1] == true then + table.remove(rt,1) + end + minitel.send(from,port,serial.serialize({rpcid,table.unpack(rt)})) + else + end + end +end +function rpcf.list() + local rt = {} + for k,v in pairs(rpcf) do + rt[#rt+1] = k + end + return rt +end + +function rpc.call(hostname,fn,...) + if hostname == "localhost" then + return rpcf[fn](...) + end + local rv = minitel.genPacketID() + minitel.rsend(hostname,rpc.port,serial.serialize({fn,rv,...}),true) + local st = computer.uptime() + local rt = {} + repeat + local _, from, port, data = event.pull(30, "net_msg", hostname, rpc.port) + rt = serial.unserialize(data) or {} + until rt[1] == rv or computer.uptime() > st + 30 + if table.remove(rt,1) == rv then + return table.unpack(rt) + end + return false +end +function rpc.proxy(hostname,filter) + filter=(filter or "").."(.+)" + local fnames = rpc.call(hostname,"list") + if not fnames then return false end + local rt = {} + for k,v in pairs(fnames) do + fv = v:match(filter) + if fv then + rt[fv] = function(...) + return rpc.call(hostname,v,...) + end + end + end + return rt +end +function rpc.register(name,fn) + local rpcrunning = false + for k,v in pairs(os.tasks()) do + if os.taskInfo(v).name == "rpc daemon" then + rpcrunning = true + end + end + if not rpcrunning then + os.spawn(function() + while true do + rpcexec(event.pull("net_msg")) + end + end,"rpc daemon") + end + rpcf[name] = fn +end + +return rpc