OC-PsychOS2/lib/netutil.lua

72 lines
2.2 KiB
Lua
Raw Normal View History

2020-03-18 01:11:53 +11:00
local computer = require "computer"
local minitel = require "minitel"
local event = require "event"
2020-03-26 17:37:14 +11:00
local ufs = require "unionfs"
2020-03-18 01:11:53 +11:00
local rpc = require "rpc"
local netutil = {}
2020-03-18 14:06:40 +11:00
function netutil.importfs(host,rpath,lpath) -- import filesystem *rpath* from *host* and attach it to *lpath*
2020-03-18 01:11:53 +11:00
local px = rpc.proxy(host,rpath.."_")
function px.getLabel()
return host..":"..rpath
end
2020-03-26 17:37:14 +11:00
px.address = host..":"..rpath
2020-03-18 14:06:40 +11:00
return fs.mount(lpath,px)
2020-03-18 01:11:53 +11:00
end
2020-03-18 14:06:40 +11:00
function netutil.exportfs(path) -- export the directory *path* over RPC
2020-03-18 01:11:53 +11:00
local path = "/"..table.concat(fs.segments(path),"/")
local px = ufs.create(path)
for k,v in pairs(px) do
2020-03-26 17:37:14 +11:00
rpc.register(path.."_"..k,v)
2020-03-18 01:11:53 +11:00
print(path.."_"..k)
end
return true
2020-03-18 01:11:53 +11:00
end
2020-03-18 14:06:40 +11:00
function netutil.ping(addr,times,timeout, silent) -- Request acknowledgment from *addr*, waiting *timeout* seconds each try, and try *times* times. If *silent* is true, don't print status. Returns true if there was at least one successful ping, the number of successes, the number of failures, and the average round trip time.
2020-03-18 01:11:53 +11:00
local times, timeout = times or 5, timeout or 30
local success, fail, time, avg = 0, 0, 0, 0
2020-03-18 01:11:53 +11:00
for i = 1, times do
local ipt = computer.uptime()
local pid = minitel.genPacketID()
computer.pushSignal("net_send",1,addr,0,"ping",pid)
local t,a = event.pull(timeout,"net_ack")
if t == "net_ack" and a == pid then
if not silent then print("Ping reply: "..tostring(computer.uptime()-ipt).." seconds.") end
success = success + 1
time = time + computer.uptime()-ipt
avg = time / success
2020-03-18 01:11:53 +11:00
else
if not silent then print("Timed out.") end
fail = fail + 1
2020-03-18 01:11:53 +11:00
end
end
if not silent then print(string.format("%d packets transmitted, %d received, %0.0f%% packet loss, %0.1fs",times,success,fail/times*100,time)) end
return success > 0, success, fail, avg
2020-03-18 01:11:53 +11:00
end
function netutil.nc(host,port)
port = port or 22
local socket = minitel.open(host,port)
if not socket then return false end
local b = ""
os.spawn(function()
repeat
local b = socket:read("*a")
if b and b:len() > 0 then
io.write(b)
end
coroutine.yield()
until socket.state ~= "open"
end)
repeat
local b = io.read()
if b and b:len() > 0 then
socket:write(b.."\n")
end
until socket.state ~= "open"
end
2020-03-18 01:11:53 +11:00
return netutil