luasocket/etc/tftp.lua

152 lines
4.5 KiB
Lua
Raw Normal View History

2003-05-25 03:54:13 +02:00
-----------------------------------------------------------------------------
-- TFTP support for the Lua language
-- LuaSocket toolkit.
2003-05-25 03:54:13 +02:00
-- Author: Diego Nehab
-- RCS ID: $Id$
-----------------------------------------------------------------------------
-----------------------------------------------------------------------------
-- Load required files
-----------------------------------------------------------------------------
local socket = require("socket")
local ltn12 = require("ltn12")
local url = require("url")
-----------------------------------------------------------------------------
-- Setup namespace
-----------------------------------------------------------------------------
_LOADED["tftp"] = getfenv(1)
2003-05-25 03:54:13 +02:00
-----------------------------------------------------------------------------
-- Program constants
-----------------------------------------------------------------------------
local char = string.char
local byte = string.byte
PORT = 69
local OP_RRQ = 1
local OP_WRQ = 2
local OP_DATA = 3
local OP_ACK = 4
local OP_ERROR = 5
local OP_INV = {"RRQ", "WRQ", "DATA", "ACK", "ERROR"}
2003-05-25 03:54:13 +02:00
-----------------------------------------------------------------------------
-- Packet creation functions
-----------------------------------------------------------------------------
local function RRQ(source, mode)
return char(0, OP_RRQ) .. source .. char(0) .. mode .. char(0)
2003-05-25 03:54:13 +02:00
end
local function WRQ(source, mode)
return char(0, OP_RRQ) .. source .. char(0) .. mode .. char(0)
2003-05-25 03:54:13 +02:00
end
local function ACK(block)
2003-05-25 03:54:13 +02:00
local low, high
low = math.mod(block, 256)
high = (block - low)/256
return char(0, OP_ACK, high, low)
2003-05-25 03:54:13 +02:00
end
local function get_OP(dgram)
2003-05-25 03:54:13 +02:00
local op = byte(dgram, 1)*256 + byte(dgram, 2)
return op
end
-----------------------------------------------------------------------------
-- Packet analysis functions
-----------------------------------------------------------------------------
local function split_DATA(dgram)
2003-05-25 03:54:13 +02:00
local block = byte(dgram, 3)*256 + byte(dgram, 4)
local data = string.sub(dgram, 5)
return block, data
end
local function get_ERROR(dgram)
2003-05-25 03:54:13 +02:00
local code = byte(dgram, 3)*256 + byte(dgram, 4)
local msg
_,_, msg = string.find(dgram, "(.*)\000", 5)
return string.format("error code %d: %s", code, msg)
end
-----------------------------------------------------------------------------
2004-06-16 06:28:21 +02:00
-- The real work
2003-05-25 03:54:13 +02:00
-----------------------------------------------------------------------------
local function tget(gett)
2003-05-25 03:54:13 +02:00
local retries, dgram, sent, datahost, dataport, code
local last = 0
local con = socket.try(socket.udp())
2003-05-25 03:54:13 +02:00
-- convert from name to ip if needed
gett.host = socket.try(socket.dns.toip(gett.host))
con:settimeout(1)
2003-05-25 03:54:13 +02:00
-- first packet gives data host/port to be used for data transfers
retries = 0
repeat
sent = socket.try(con:sendto(RRQ(gett.path, "octet"),
gett.host, gett.port))
dgram, datahost, dataport = con:receivefrom()
2003-05-25 03:54:13 +02:00
retries = retries + 1
until dgram or datahost ~= "timeout" or retries > 5
socket.try(dgram, datahost)
2003-05-25 03:54:13 +02:00
-- associate socket with data host/port
socket.try(con:setpeername(datahost, dataport))
-- default sink
local sink = gett.sink or ltn12.sink.null()
2003-05-25 03:54:13 +02:00
-- process all data packets
while 1 do
-- decode packet
code = get_OP(dgram)
socket.try(code ~= OP_ERROR, get_ERROR(dgram))
socket.try(code == OP_DATA, "unhandled opcode " .. code)
2003-05-25 03:54:13 +02:00
-- get data packet parts
local block, data = split_DATA(dgram)
2003-05-25 03:54:13 +02:00
-- if not repeated, write
if block == last+1 then
socket.try(sink(data))
2003-05-25 03:54:13 +02:00
last = block
end
-- last packet brings less than 512 bytes of data
if string.len(data) < 512 then
socket.try(con:send(ACK(block)))
socket.try(con:close())
socket.try(sink(nil))
return 1
2003-05-25 03:54:13 +02:00
end
-- get the next packet
retries = 0
repeat
sent = socket.try(con:send(ACK(last)))
dgram, err = con:receive()
2003-05-25 03:54:13 +02:00
retries = retries + 1
until dgram or err ~= "timeout" or retries > 5
socket.try(dgram, err)
2003-05-25 03:54:13 +02:00
end
end
local default = {
port = PORT,
path ="/",
scheme = "tftp"
}
local function parse(u)
local t = socket.try(url.parse(u, default))
socket.try(t.scheme == "tftp", "invalid scheme '" .. t.scheme .. "'")
socket.try(t.host, "invalid host")
return t
end
local function sget(u)
local gett = parse(u)
local t = {}
gett.sink = ltn12.sink.table(t)
tget(gett)
return table.concat(t)
end
get = socket.protect(function(gett)
if type(gett) == "string" then return sget(gett)
else return tget(gett) end
end)