2 nixio - Linux I/O library for lua
4 Copyright 2008 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 = getmetatable, assert
21 local BUFFERSIZE = 8096
22 local socket = nixio.socket_meta
23 local tls_socket = nixio.tls_socket_meta
25 function socket.is_socket(self)
26 return (getmetatable(self) == socket)
28 tls_socket.is_socket = socket.is_socket
30 function socket.is_tls_socket(self)
31 return (getmetatable(self) == tls_socket)
33 tls_socket.is_tls_socket = socket.is_tls_socket
35 function socket.recvall(self, len)
36 local block, code, msg = self:recv(len)
39 return "", code, msg, len
40 elseif #block == 0 then
41 return "", nil, nil, len
44 local data, total = {block}, #block
47 block, code, msg = self:recv(len - total)
50 return data, code, msg, len - #data
51 elseif #block == 0 then
52 return data, nil, nil, len - #data
55 data[#data+1], total = block, total + #block
58 return (#data > 1 and table.concat(data) or data[1]), nil, nil, 0
60 tls_socket.recvall = socket.recvall
62 function socket.sendall(self, data)
63 local total, block = 0
64 local sent, code, msg = self:send(data)
67 return total, code, msg, data
71 block, total = data:sub(sent + 1), total + sent
72 sent, code, msg = self:send(block)
75 return total, code, msg, block
79 return total + sent, nil, nil, ""
81 tls_socket.sendall = socket.sendall
83 function socket.linesource(self, limit)
84 limit = limit or BUFFERSIZE
87 return function(flush)
91 line = buffer:sub(bpos + 1)
98 _, endp, line = buffer:find("(.-)\r?\n", bpos + 1)
102 elseif #buffer < limit + bpos then
103 local newblock, code = self:recv(limit + bpos - #buffer)
106 elseif #newblock == 0 then
109 buffer = buffer:sub(bpos + 1) .. newblock
117 tls_socket.linesource = socket.linesource
119 function socket.blocksource(self, bs, limit)
120 bs = bs or BUFFERSIZE
126 elseif limit < toread then
131 local block, code, msg = self:recv(toread)
135 elseif #block == 0 then
139 limit = limit - #block
146 tls_socket.blocksource = socket.blocksource
148 function tls_socket.close(self)
150 return self.socket:close()