1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
|
--[[
nixio - Linux I/O library for lua
Copyright 2009 Steven Barth <steven@midlink.org>
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
$Id$
]]--
local table = require "table"
local nixio = require "nixio"
local getmetatable, assert, pairs, type = getmetatable, assert, pairs, type
local tostring = tostring
module "nixio.util"
local BUFFERSIZE = nixio.const.buffersize
local ZIOBLKSIZE = 65536
local socket = nixio.meta_socket
local tls_socket = nixio.meta_tls_socket
local file = nixio.meta_file
local uname = nixio.uname()
local ZBUG = uname.sysname == "Linux" and uname.release:sub(1, 3) == "2.4"
function consume(iter, append)
local tbl = append or {}
if iter then
for obj in iter do
tbl[#tbl+1] = obj
end
end
return tbl
end
local meta = {}
function meta.is_socket(self)
return (getmetatable(self) == socket)
end
function meta.is_tls_socket(self)
return (getmetatable(self) == tls_socket)
end
function meta.is_file(self)
return (getmetatable(self) == file)
end
function meta.readall(self, len)
local block, code, msg = self:read(len or BUFFERSIZE)
if not block then
return nil, code, msg, ""
elseif #block == 0 then
return "", nil, nil, ""
end
local data, total = {block}, #block
while not len or len > total do
block, code, msg = self:read(len and (len - total) or BUFFERSIZE)
if not block then
return nil, code, msg, table.concat(data)
elseif #block == 0 then
break
end
data[#data+1], total = block, total + #block
end
local data = #data > 1 and table.concat(data) or data[1]
return data, nil, nil, data
end
meta.recvall = meta.readall
function meta.writeall(self, data)
data = tostring(data)
local sent, code, msg = self:write(data)
if not sent then
return nil, code, msg, 0
end
local total = sent
while total < #data do
sent, code, msg = self:write(data, total)
if not sent then
return nil, code, msg, total
end
total = total + sent
end
return total, nil, nil, total
end
meta.sendall = meta.writeall
function meta.linesource(self, limit)
limit = limit or BUFFERSIZE
local buffer = ""
local bpos = 0
return function(flush)
local line, endp, _
if flush then
line = buffer:sub(bpos + 1)
buffer = type(flush) == "string" and flush or ""
bpos = 0
return line
end
while not line do
_, endp, line = buffer:find("(.-)\r?\n", bpos + 1)
if line then
bpos = endp
return line
elseif #buffer < limit + bpos then
local newblock, code, msg = self:read(limit + bpos - #buffer)
if not newblock then
return nil, code, msg
elseif #newblock == 0 then
return nil
end
buffer = buffer:sub(bpos + 1) .. newblock
bpos = 0
else
return nil, 0
end
end
end
end
function meta.blocksource(self, bs, limit)
bs = bs or BUFFERSIZE
return function()
local toread = bs
if limit then
if limit < 1 then
return nil
elseif limit < toread then
toread = limit
end
end
local block, code, msg = self:read(toread)
if not block then
return nil, code, msg
elseif #block == 0 then
return nil
else
if limit then
limit = limit - #block
end
return block
end
end
end
function meta.sink(self, close)
return function(chunk, src_err)
if not chunk and not src_err and close then
if self.shutdown then
self:shutdown()
end
self:close()
elseif chunk and #chunk > 0 then
return self:writeall(chunk)
end
return true
end
end
function meta.copy(self, fdout, size)
local source = self:blocksource(nil, size)
local sink = fdout:sink()
local sent, chunk, code, msg = 0
repeat
chunk, code, msg = source()
sink(chunk, code, msg)
sent = chunk and (sent + #chunk) or sent
until not chunk
return not code and sent or nil, code, msg, sent
end
function meta.copyz(self, fd, size)
local sent, lsent, code, msg = 0
local splicable
if not ZBUG and self:is_file() then
local ftype = self:stat("type")
if nixio.sendfile and fd:is_socket() and ftype == "reg" then
repeat
lsent, code, msg = nixio.sendfile(fd, self, size or ZIOBLKSIZE)
if lsent then
sent = sent + lsent
size = size and (size - lsent)
end
until (not lsent or lsent == 0 or (size and size == 0))
if lsent or (not lsent and sent == 0 and
code ~= nixio.const.ENOSYS and code ~= nixio.const.EINVAL) then
return lsent and sent, code, msg, sent
end
elseif nixio.splice and not fd:is_tls_socket() and ftype == "fifo" then
splicable = true
end
end
if nixio.splice and fd:is_file() and not splicable then
splicable = not self:is_tls_socket() and fd:stat("type") == "fifo"
end
if splicable then
repeat
lsent, code, msg = nixio.splice(self, fd, size or ZIOBLKSIZE)
if lsent then
sent = sent + lsent
size = size and (size - lsent)
end
until (not lsent or lsent == 0 or (size and size == 0))
if lsent or (not lsent and sent == 0 and
code ~= nixio.const.ENOSYS and code ~= nixio.const.EINVAL) then
return lsent and sent, code, msg, sent
end
end
return self:copy(fd, size)
end
if tls_socket then
function tls_socket.close(self)
return self.socket:close()
end
function tls_socket.getsockname(self)
return self.socket:getsockname()
end
function tls_socket.getpeername(self)
return self.socket:getpeername()
end
function tls_socket.getsockopt(self, ...)
return self.socket:getsockopt(...)
end
tls_socket.getopt = tls_socket.getsockopt
function tls_socket.setsockopt(self, ...)
return self.socket:setsockopt(...)
end
tls_socket.setopt = tls_socket.setsockopt
end
for k, v in pairs(meta) do
file[k] = v
socket[k] = v
if tls_socket then
tls_socket[k] = v
end
end
|