2 nixio - Linux I/O library for lua
4 Copyright 2009 Steven Barth <steven@midlink.org>
6 Licensed under the Apache License, Version 2.0 (the "License");
7 you may not use this file except in compliance with the License.
8 You may obtain a copy of the License at
10 http://www.apache.org/licenses/LICENSE-2.0
15 local table = require "table"
16 local nixio = require "nixio"
17 local getmetatable, assert, pairs = getmetatable, assert, pairs
21 local BUFFERSIZE = 8096
22 local socket = nixio.meta_socket
23 local tls_socket = nixio.meta_tls_socket
24 local file = nixio.meta_file
28 function meta.is_socket(self)
29 return (getmetatable(self) == socket)
32 function meta.is_tls_socket(self)
33 return (getmetatable(self) == tls_socket)
36 function meta.is_file(self)
37 return (getmetatable(self) == file)
40 function meta.readall(self, len)
41 local block, code, msg = self:read(len)
44 return "", code, msg, len
45 elseif #block == 0 then
46 return "", nil, nil, len
49 local data, total = {block}, #block
52 block, code, msg = self:read(len - total)
55 return data, code, msg, len - #data
56 elseif #block == 0 then
57 return data, nil, nil, len - #data
60 data[#data+1], total = block, total + #block
63 return (#data > 1 and table.concat(data) or data[1]), nil, nil, 0
65 meta.recvall = meta.readall
67 function meta.writeall(self, data)
68 local total, block = 0
69 local sent, code, msg = self:write(data)
72 return total, code, msg, data
76 block, total = data:sub(sent + 1), total + sent
77 sent, code, msg = self:write(block)
80 return total, code, msg, block
84 return total + sent, nil, nil, ""
86 meta.sendall = meta.writeall
88 function meta.linesource(self, limit)
89 limit = limit or BUFFERSIZE
92 return function(flush)
96 line = buffer:sub(bpos + 1)
103 _, endp, line = buffer:find("(.-)\r?\n", bpos + 1)
107 elseif #buffer < limit + bpos then
108 local newblock, code = self:read(limit + bpos - #buffer)
111 elseif #newblock == 0 then
114 buffer = buffer:sub(bpos + 1) .. newblock
123 function meta.blocksource(self, bs, limit)
124 bs = bs or BUFFERSIZE
130 elseif limit < toread then
135 local block, code, msg = self:read(toread)
139 elseif #block == 0 then
143 limit = limit - #block
151 function meta.sink(self, close)
152 return function(chunk, src_err)
153 if not chunk and not src_err and close then
155 elseif chunk and #chunk > 0 then
156 return self:writeall(chunk)
162 function tls_socket.close(self)
164 return self.socket:close()
167 for k, v in pairs(meta) do