diff options
Diffstat (limited to 'etc/tftp.lua')
-rw-r--r-- | etc/tftp.lua | 154 |
1 files changed, 0 insertions, 154 deletions
diff --git a/etc/tftp.lua b/etc/tftp.lua deleted file mode 100644 index ed99cd1..0000000 --- a/etc/tftp.lua +++ /dev/null | |||
@@ -1,154 +0,0 @@ | |||
1 | ----------------------------------------------------------------------------- | ||
2 | -- TFTP support for the Lua language | ||
3 | -- LuaSocket toolkit. | ||
4 | -- Author: Diego Nehab | ||
5 | ----------------------------------------------------------------------------- | ||
6 | |||
7 | ----------------------------------------------------------------------------- | ||
8 | -- Load required files | ||
9 | ----------------------------------------------------------------------------- | ||
10 | local base = _G | ||
11 | local table = require("table") | ||
12 | local math = require("math") | ||
13 | local string = require("string") | ||
14 | local socket = require("socket") | ||
15 | local ltn12 = require("ltn12") | ||
16 | local url = require("socket.url") | ||
17 | module("socket.tftp") | ||
18 | |||
19 | ----------------------------------------------------------------------------- | ||
20 | -- Program constants | ||
21 | ----------------------------------------------------------------------------- | ||
22 | local char = string.char | ||
23 | local byte = string.byte | ||
24 | |||
25 | PORT = 69 | ||
26 | local OP_RRQ = 1 | ||
27 | local OP_WRQ = 2 | ||
28 | local OP_DATA = 3 | ||
29 | local OP_ACK = 4 | ||
30 | local OP_ERROR = 5 | ||
31 | local OP_INV = {"RRQ", "WRQ", "DATA", "ACK", "ERROR"} | ||
32 | |||
33 | ----------------------------------------------------------------------------- | ||
34 | -- Packet creation functions | ||
35 | ----------------------------------------------------------------------------- | ||
36 | local function RRQ(source, mode) | ||
37 | return char(0, OP_RRQ) .. source .. char(0) .. mode .. char(0) | ||
38 | end | ||
39 | |||
40 | local function WRQ(source, mode) | ||
41 | return char(0, OP_RRQ) .. source .. char(0) .. mode .. char(0) | ||
42 | end | ||
43 | |||
44 | local function ACK(block) | ||
45 | local low, high | ||
46 | low = math.mod(block, 256) | ||
47 | high = (block - low)/256 | ||
48 | return char(0, OP_ACK, high, low) | ||
49 | end | ||
50 | |||
51 | local function get_OP(dgram) | ||
52 | local op = byte(dgram, 1)*256 + byte(dgram, 2) | ||
53 | return op | ||
54 | end | ||
55 | |||
56 | ----------------------------------------------------------------------------- | ||
57 | -- Packet analysis functions | ||
58 | ----------------------------------------------------------------------------- | ||
59 | local function split_DATA(dgram) | ||
60 | local block = byte(dgram, 3)*256 + byte(dgram, 4) | ||
61 | local data = string.sub(dgram, 5) | ||
62 | return block, data | ||
63 | end | ||
64 | |||
65 | local function get_ERROR(dgram) | ||
66 | local code = byte(dgram, 3)*256 + byte(dgram, 4) | ||
67 | local msg | ||
68 | _,_, msg = string.find(dgram, "(.*)\000", 5) | ||
69 | return string.format("error code %d: %s", code, msg) | ||
70 | end | ||
71 | |||
72 | ----------------------------------------------------------------------------- | ||
73 | -- The real work | ||
74 | ----------------------------------------------------------------------------- | ||
75 | local function tget(gett) | ||
76 | local retries, dgram, sent, datahost, dataport, code | ||
77 | local last = 0 | ||
78 | socket.try(gett.host, "missing host") | ||
79 | local con = socket.try(socket.udp()) | ||
80 | local try = socket.newtry(function() con:close() end) | ||
81 | -- convert from name to ip if needed | ||
82 | gett.host = try(socket.dns.toip(gett.host)) | ||
83 | con:settimeout(1) | ||
84 | -- first packet gives data host/port to be used for data transfers | ||
85 | local path = string.gsub(gett.path or "", "^/", "") | ||
86 | path = url.unescape(path) | ||
87 | retries = 0 | ||
88 | repeat | ||
89 | sent = try(con:sendto(RRQ(path, "octet"), gett.host, gett.port)) | ||
90 | dgram, datahost, dataport = con:receivefrom() | ||
91 | retries = retries + 1 | ||
92 | until dgram or datahost ~= "timeout" or retries > 5 | ||
93 | try(dgram, datahost) | ||
94 | -- associate socket with data host/port | ||
95 | try(con:setpeername(datahost, dataport)) | ||
96 | -- default sink | ||
97 | local sink = gett.sink or ltn12.sink.null() | ||
98 | -- process all data packets | ||
99 | while 1 do | ||
100 | -- decode packet | ||
101 | code = get_OP(dgram) | ||
102 | try(code ~= OP_ERROR, get_ERROR(dgram)) | ||
103 | try(code == OP_DATA, "unhandled opcode " .. code) | ||
104 | -- get data packet parts | ||
105 | local block, data = split_DATA(dgram) | ||
106 | -- if not repeated, write | ||
107 | if block == last+1 then | ||
108 | try(sink(data)) | ||
109 | last = block | ||
110 | end | ||
111 | -- last packet brings less than 512 bytes of data | ||
112 | if string.len(data) < 512 then | ||
113 | try(con:send(ACK(block))) | ||
114 | try(con:close()) | ||
115 | try(sink(nil)) | ||
116 | return 1 | ||
117 | end | ||
118 | -- get the next packet | ||
119 | retries = 0 | ||
120 | repeat | ||
121 | sent = try(con:send(ACK(last))) | ||
122 | dgram, err = con:receive() | ||
123 | retries = retries + 1 | ||
124 | until dgram or err ~= "timeout" or retries > 5 | ||
125 | try(dgram, err) | ||
126 | end | ||
127 | end | ||
128 | |||
129 | local default = { | ||
130 | port = PORT, | ||
131 | path ="/", | ||
132 | scheme = "tftp" | ||
133 | } | ||
134 | |||
135 | local function parse(u) | ||
136 | local t = socket.try(url.parse(u, default)) | ||
137 | socket.try(t.scheme == "tftp", "invalid scheme '" .. t.scheme .. "'") | ||
138 | socket.try(t.host, "invalid host") | ||
139 | return t | ||
140 | end | ||
141 | |||
142 | local function sget(u) | ||
143 | local gett = parse(u) | ||
144 | local t = {} | ||
145 | gett.sink = ltn12.sink.table(t) | ||
146 | tget(gett) | ||
147 | return table.concat(t) | ||
148 | end | ||
149 | |||
150 | get = socket.protect(function(gett) | ||
151 | if base.type(gett) == "string" then return sget(gett) | ||
152 | else return tget(gett) end | ||
153 | end) | ||
154 | |||