2 -- server.lua by blastbeat of the luadch project
3 -- Re-used here under the MIT/X Consortium License
5 -- Modifications (C) 2008-2010 Matthew Wild, Waqas Hussain
8 -- // wrapping luadch stuff // --
10 local use = function( what )
14 local log, table_concat = require ("util.logger").init("socket"), table.concat;
15 local out_put = function (...) return log("debug", table_concat{...}); end
16 local out_error = function (...) return log("warn", table_concat{...}); end
18 ----------------------------------// DECLARATION //--
22 local STAT_UNIT = 1 -- byte
24 --// lua functions //--
26 local type = use "type"
27 local pairs = use "pairs"
28 local ipairs = use "ipairs"
29 local tonumber = use "tonumber"
30 local tostring = use "tostring"
34 local table = use "table"
35 local string = use "string"
36 local coroutine = use "coroutine"
38 --// lua lib methods //--
40 local math_min = math.min
41 local math_huge = math.huge
42 local table_concat = table.concat
43 local table_insert = table.insert
44 local string_sub = string.sub
45 local coroutine_wrap = coroutine.wrap
46 local coroutine_yield = coroutine.yield
50 local has_luasec, luasec = pcall ( require , "ssl" )
51 local luasocket = use "socket" or require "socket"
52 local luasocket_gettime = luasocket.gettime
53 local getaddrinfo = luasocket.dns.getaddrinfo
55 --// extern lib methods //--
57 local ssl_wrap = ( has_luasec and luasec.wrap )
58 local socket_bind = luasocket.bind
59 local socket_select = luasocket.select
92 --// simple data types //--
119 local _maxsslhandshake
121 ----------------------------------// DEFINITION //--
123 _server = { } -- key = port, value = table; list of listening servers
124 _readlist = { } -- array with sockets to read from
125 _sendlist = { } -- arrary with sockets to write to
126 _timerlist = { } -- array of timer functions
127 _socketlist = { } -- key = socket, value = wrapped socket (handlers)
128 _readtimes = { } -- key = handler, value = timestamp of last data reading
129 _writetimes = { } -- key = handler, value = timestamp of last data writing/sending
130 _closelist = { } -- handlers to close
131 _fullservers = { } -- servers in a paused state while there are too many clients
133 _readlistlen = 0 -- length of readlist
134 _sendlistlen = 0 -- length of sendlist
135 _timerlistlen = 0 -- lenght of timerlist
137 _sendtraffic = 0 -- some stats
140 _selecttimeout = 1 -- timeout of socket.select
141 _tcpbacklog = 128 -- some kind of hint to the OS
142 _accepretry = 10 -- seconds to wait until the next attempt of a full server to accept
144 _maxsendlen = 51000 * 1024 -- max len of send buffer
145 _maxreadlen = 25000 * 1024 -- max len of read buffer
147 _checkinterval = 30 -- interval in secs to check idle clients
148 _sendtimeout = 60000 -- allowed send idle time in secs
149 _readtimeout = 6 * 60 * 60 -- allowed read idle time in secs
151 local is_windows = package.config:sub(1,1) == "\\" -- check the directory separator, to detemine whether this is Windows
152 _maxfd = (is_windows and math.huge) or luasocket._SETSIZE or 1024 -- max fd number, limit to 1024 by default to prevent glibc buffer overflow, but not on Windows
153 _maxselectlen = luasocket._SETSIZE or 1024 -- But this still applies on Windows
155 _maxsslhandshake = 30 -- max handshake round-trips
157 ----------------------------------// PRIVATE //--
159 wrapserver = function( listeners, socket, ip, serverport, pattern, sslctx ) -- this function wraps a server -- FIXME Make sure FD < _maxfd
161 if socket:getfd() >= _maxfd then
162 out_error("server.lua: Disallowed FD number: "..socket:getfd())
164 return nil, "fd-too-large"
167 local connections = 0
169 local dispatch, disconnect = listeners.onconnect, listeners.ondisconnect
171 local accept = socket.accept
173 --// public methods of the object //--
177 handler.shutdown = function( ) end
179 handler.ssl = function( )
182 handler.sslctx = function( )
185 handler.remove = function( )
186 connections = connections - 1
191 handler.close = function()
193 _sendlistlen = removesocket( _sendlist, socket, _sendlistlen )
194 _readlistlen = removesocket( _readlist, socket, _readlistlen )
195 _server[ip..":"..serverport] = nil;
196 _socketlist[ socket ] = nil
200 out_put "server.lua: closed server handler and removed sockets from list"
202 handler.pause = function( hard )
203 if not handler.paused then
204 _readlistlen = removesocket( _readlist, socket, _readlistlen )
206 _socketlist[ socket ] = nil
210 handler.paused = true;
211 out_put("server.lua: server [", ip, "]:", serverport, " paused")
214 handler.resume = function( )
215 if handler.paused then
217 socket = socket_bind( ip, serverport, _tcpbacklog );
218 socket:settimeout( 0 )
220 _readlistlen = addsocket(_readlist, socket, _readlistlen)
221 _socketlist[ socket ] = handler
222 _fullservers[ handler ] = nil
223 handler.paused = false;
224 out_put("server.lua: server [", ip, "]:", serverport, " resumed")
227 handler.ip = function( )
230 handler.serverport = function( )
233 handler.socket = function( )
236 handler.readbuffer = function( )
237 if _readlistlen >= _maxselectlen or _sendlistlen >= _maxselectlen then
239 _fullservers[ handler ] = _currenttime
240 out_put( "server.lua: refused new client connection: server full" )
243 local client, err = accept( socket ) -- try to accept
245 local ip, clientport = client:getpeername( )
246 local handler, client, err = wrapconnection( handler, listeners, client, ip, serverport, clientport, pattern, sslctx ) -- wrap new client socket
247 if err then -- error while wrapping ssl socket
250 connections = connections + 1
251 out_put( "server.lua: accepted new client connection from ", tostring(ip), ":", tostring(clientport), " to ", tostring(serverport))
252 if dispatch and not sslctx then -- SSL connections will notify onconnect when handshake completes
253 return dispatch( handler );
256 elseif err then -- maybe timeout or something else
257 out_put( "server.lua: error with new client connection: ", tostring(err) )
259 _fullservers[ handler ] = _currenttime
266 wrapconnection = function( server, listeners, socket, ip, serverport, clientport, pattern, sslctx ) -- this function wraps a client to a handler object
268 if socket:getfd() >= _maxfd then
269 out_error("server.lua: Disallowed FD number: "..socket:getfd()) -- PROTIP: Switch to libevent
270 socket:close( ) -- Should we send some kind of error here?
272 _fullservers[ server ] = _currenttime
275 return nil, nil, "fd-too-large"
277 socket:settimeout( 0 )
279 --// local import of socket methods //--
285 --// private closures of the object //--
289 local dispatch = listeners.onincoming
290 local status = listeners.onstatus
291 local disconnect = listeners.ondisconnect
292 local drain = listeners.ondrain
293 local onreadtimeout = listeners.onreadtimeout;
294 local detach = listeners.ondetach
296 local bufferqueue = { } -- buffer array
297 local bufferqueuelen = 0 -- end of buffer array
307 local sendtraffic, readtraffic = 0, 0
309 local maxsendlen = _maxsendlen
310 local maxreadlen = _maxreadlen
312 --// public methods of the object //--
314 local handler = bufferqueue -- saves a table ^_^
316 handler.dispatch = function( )
319 handler.disconnect = function( )
322 handler.onreadtimeout = onreadtimeout;
324 handler.setlistener = function( self, listeners )
326 detach(self) -- Notify listener that it is no longer responsible for this connection
328 dispatch = listeners.onincoming
329 disconnect = listeners.ondisconnect
330 status = listeners.onstatus
331 drain = listeners.ondrain
332 handler.onreadtimeout = listeners.onreadtimeout
333 detach = listeners.ondetach
335 handler.getstats = function( )
336 return readtraffic, sendtraffic
338 handler.ssl = function( )
341 handler.sslctx = function ( )
344 handler.send = function( _, data, i, j )
345 return send( socket, data, i, j )
347 handler.receive = function( pattern, prefix )
348 return receive( socket, pattern, prefix )
350 handler.shutdown = function( pattern )
351 return shutdown( socket, pattern )
353 handler.setoption = function (self, option, value)
354 if socket.setoption then
355 return socket:setoption(option, value);
357 return false, "setoption not implemented";
359 handler.force_close = function ( self, err )
360 if bufferqueuelen ~= 0 then
361 out_put("server.lua: discarding unwritten data for ", tostring(ip), ":", tostring(clientport))
364 return self:close(err);
366 handler.close = function( self, err )
367 if not handler then return true; end
368 _readlistlen = removesocket( _readlist, socket, _readlistlen )
369 _readtimes[ handler ] = nil
370 if bufferqueuelen ~= 0 then
371 handler.sendbuffer() -- Try now to send any outstanding data
372 if bufferqueuelen ~= 0 then -- Still not empty, so we'll try again later
374 handler.write = nil -- ... but no further writing allowed
381 _ = shutdown and shutdown( socket )
383 _sendlistlen = removesocket( _sendlist, socket, _sendlistlen )
384 _socketlist[ socket ] = nil
387 out_put "server.lua: socket already closed"
390 _writetimes[ handler ] = nil
391 _closelist[ handler ] = nil
392 local _handler = handler;
395 disconnect(_handler, err or false);
402 out_put "server.lua: closed client handler and removed socket from list"
405 handler.server = function ( )
408 handler.ip = function( )
411 handler.serverport = function( )
414 handler.clientport = function( )
417 handler.port = handler.clientport -- COMPAT server_event
418 local write = function( self, data )
419 bufferlen = bufferlen + #data
420 if bufferlen > maxsendlen then
421 _closelist[ handler ] = "send buffer exceeded" -- cannot close the client at the moment, have to wait to the end of the cycle
422 handler.write = idfalse -- dont write anymore
424 elseif socket and not _sendlist[ socket ] then
425 _sendlistlen = addsocket(_sendlist, socket, _sendlistlen)
427 bufferqueuelen = bufferqueuelen + 1
428 bufferqueue[ bufferqueuelen ] = data
430 _writetimes[ handler ] = _writetimes[ handler ] or _currenttime
434 handler.write = write
435 handler.bufferqueue = function( self )
438 handler.socket = function( self )
441 handler.set_mode = function( self, new )
442 pattern = new or pattern
445 handler.set_send = function ( self, newsend )
446 send = newsend or send
449 handler.bufferlen = function( self, readlen, sendlen )
450 maxsendlen = sendlen or maxsendlen
451 maxreadlen = readlen or maxreadlen
452 return bufferlen, maxreadlen, maxsendlen
455 handler.lock_read = function (self, switch)
456 if switch == true then
457 local tmp = _readlistlen
458 _readlistlen = removesocket( _readlist, socket, _readlistlen )
459 _readtimes[ handler ] = nil
460 if _readlistlen ~= tmp then
463 elseif switch == false then
466 _readlistlen = addsocket(_readlist, socket, _readlistlen)
467 _readtimes[ handler ] = _currenttime
472 handler.pause = function (self)
473 return self:lock_read(true);
475 handler.resume = function (self)
476 return self:lock_read(false);
478 handler.lock = function( self, switch )
479 handler.lock_read (switch)
480 if switch == true then
481 handler.write = idfalse
482 local tmp = _sendlistlen
483 _sendlistlen = removesocket( _sendlist, socket, _sendlistlen )
484 _writetimes[ handler ] = nil
485 if _sendlistlen ~= tmp then
488 elseif switch == false then
489 handler.write = write
495 return noread, nosend
497 local _readbuffer = function( ) -- this function reads data
498 local buffer, err, part = receive( socket, pattern ) -- receive buffer with "pattern"
499 if not err or (err == "wantread" or err == "timeout") then -- received something
500 local buffer = buffer or part or ""
502 if len > maxreadlen then
503 handler:close( "receive buffer exceeded" )
506 local count = len * STAT_UNIT
507 readtraffic = readtraffic + count
508 _readtraffic = _readtraffic + count
509 _readtimes[ handler ] = _currenttime
510 --out_put( "server.lua: read data '", buffer:gsub("[^%w%p ]", "."), "', error: ", err )
511 return dispatch( handler, buffer, err )
512 else -- connections was closed or fatal error
513 out_put( "server.lua: client ", tostring(ip), ":", tostring(clientport), " read error: ", tostring(err) )
514 _ = handler and handler:force_close( err )
518 local _sendbuffer = function( ) -- this function sends data
519 local succ, err, byte, buffer, count;
521 buffer = table_concat( bufferqueue, "", 1, bufferqueuelen )
522 succ, err, byte = send( socket, buffer, 1, bufferlen )
523 count = ( succ or byte or 0 ) * STAT_UNIT
524 sendtraffic = sendtraffic + count
525 _sendtraffic = _sendtraffic + count
526 for i = bufferqueuelen,1,-1 do
527 bufferqueue[ i ] = nil
529 --out_put( "server.lua: sended '", buffer, "', bytes: ", tostring(succ), ", error: ", tostring(err), ", part: ", tostring(byte), ", to: ", tostring(ip), ":", tostring(clientport) )
531 succ, err, count = false, "unexpected close", 0;
533 if succ then -- sending succesful
536 _sendlistlen = removesocket( _sendlist, socket, _sendlistlen ) -- delete socket from writelist
537 _writetimes[ handler ] = nil
541 _ = needtls and handler:starttls(nil)
542 _ = toclose and handler:force_close( )
544 elseif byte and ( err == "timeout" or err == "wantwrite" ) then -- want write
545 buffer = string_sub( buffer, byte + 1, bufferlen ) -- new buffer
546 bufferqueue[ 1 ] = buffer -- insert new buffer in queue
548 bufferlen = bufferlen - byte
549 _writetimes[ handler ] = _currenttime
551 else -- connection was closed during sending or fatal error
552 out_put( "server.lua: client ", tostring(ip), ":", tostring(clientport), " write error: ", tostring(err) )
553 _ = handler and handler:force_close( err )
560 function handler.set_sslctx(self, new_sslctx)
563 handshake = coroutine_wrap( function( client ) -- create handshake coroutine
565 for i = 1, _maxsslhandshake do
566 _sendlistlen = ( wrote and removesocket( _sendlist, client, _sendlistlen ) ) or _sendlistlen
567 _readlistlen = ( read and removesocket( _readlist, client, _readlistlen ) ) or _readlistlen
568 read, wrote = nil, nil
569 _, err = client:dohandshake( )
571 out_put( "server.lua: ssl handshake done" )
572 handler.readbuffer = _readbuffer -- when handshake is done, replace the handshake function with regular functions
573 handler.sendbuffer = _sendbuffer
574 _ = status and status( handler, "ssl-handshake-complete" )
575 if self.autostart_ssl and listeners.onconnect then
576 listeners.onconnect(self);
577 if bufferqueuelen ~= 0 then
578 _sendlistlen = addsocket(_sendlist, client, _sendlistlen)
581 _readlistlen = addsocket(_readlist, client, _readlistlen)
584 if err == "wantwrite" then
585 _sendlistlen = addsocket(_sendlist, client, _sendlistlen)
587 elseif err == "wantread" then
588 _readlistlen = addsocket(_readlist, client, _readlistlen)
594 coroutine_yield( ) -- handshake not finished
597 err = "ssl handshake error: " .. ( err or "handshake too long" );
598 out_put( "server.lua: ", err );
599 _ = handler and handler:force_close(err)
600 return false, err -- handshake failed
605 handler.starttls = function( self, _sslctx)
607 handler:set_sslctx(_sslctx);
609 if bufferqueuelen > 0 then
610 out_put "server.lua: we need to do tls, but delaying until send buffer empty"
614 out_put( "server.lua: attempting to start tls on " .. tostring( socket ) )
615 local oldsocket, err = socket
616 socket, err = ssl_wrap( socket, sslctx ) -- wrap socket
618 out_put( "server.lua: error while starting tls on client: ", tostring(err or "unknown error") )
619 return nil, err -- fatal error
622 socket:settimeout( 0 )
624 -- add the new socket to our system
626 receive = socket.receive
628 _socketlist[ socket ] = handler
629 _readlistlen = addsocket(_readlist, socket, _readlistlen)
631 -- remove traces of the old socket
632 _readlistlen = removesocket( _readlist, oldsocket, _readlistlen )
633 _sendlistlen = removesocket( _sendlist, oldsocket, _sendlistlen )
634 _socketlist[ oldsocket ] = nil
636 handler.starttls = nil
639 -- Secure now (if handshake fails connection will close)
642 handler.readbuffer = handshake
643 handler.sendbuffer = handshake
644 return handshake( socket ) -- do handshake
648 handler.readbuffer = _readbuffer
649 handler.sendbuffer = _sendbuffer
651 receive = socket.receive
652 shutdown = ( ssl and id ) or socket.shutdown
654 _socketlist[ socket ] = handler
655 _readlistlen = addsocket(_readlist, socket, _readlistlen)
657 if sslctx and has_luasec then
658 out_put "server.lua: auto-starting ssl negotiation..."
659 handler.autostart_ssl = true;
660 local ok, err = handler:starttls(sslctx);
666 return handler, socket
672 idfalse = function( )
676 addsocket = function( list, socket, len )
677 if not list[ socket ] then
685 removesocket = function( list, socket, len ) -- this function removes sockets from a list ( copied from copas )
686 local pos = list[ socket ]
689 local last = list[ len ]
691 if last ~= socket then
700 closesocket = function( socket )
701 _sendlistlen = removesocket( _sendlist, socket, _sendlistlen )
702 _readlistlen = removesocket( _readlist, socket, _readlistlen )
703 _socketlist[ socket ] = nil
708 local function link(sender, receiver, buffersize)
710 local _sendbuffer = receiver.sendbuffer;
711 function receiver.sendbuffer()
713 if sender_locked and receiver.bufferlen() < buffersize then
714 sender:lock_read(false); -- Unlock now
719 local _readbuffer = sender.readbuffer;
720 function sender.readbuffer()
722 if not sender_locked and receiver.bufferlen() >= buffersize then
723 sender_locked = true;
724 sender:lock_read(true);
727 sender:set_mode("*a");
730 ----------------------------------// PUBLIC //--
732 addserver = function( addr, port, listeners, pattern, sslctx ) -- this function provides a way for other scripts to reg a server
735 if type( listeners ) ~= "table" then
736 err = "invalid listener table"
737 elseif type ( addr ) ~= "string" then
738 err = "invalid address"
739 elseif type( port ) ~= "number" or not ( port >= 0 and port <= 65535 ) then
741 elseif _server[ addr..":"..port ] then
742 err = "listeners on '[" .. addr .. "]:" .. port .. "' already exist"
743 elseif sslctx and not has_luasec then
744 err = "luasec not found"
747 out_error( "server.lua, [", addr, "]:", port, ": ", err )
750 local server, err = socket_bind( addr, port, _tcpbacklog )
752 out_error( "server.lua, [", addr, "]:", port, ": ", err )
755 local handler, err = wrapserver( listeners, server, addr, port, pattern, sslctx ) -- wrap new server socket
760 server:settimeout( 0 )
761 _readlistlen = addsocket(_readlist, server, _readlistlen)
762 _server[ addr..":"..port ] = handler
763 _socketlist[ server ] = handler
764 out_put( "server.lua: new "..(sslctx and "ssl " or "").."server listener on '[", addr, "]:", port, "'" )
768 getserver = function ( addr, port )
769 return _server[ addr..":"..port ];
772 removeserver = function( addr, port )
773 local handler = _server[ addr..":"..port ]
775 return nil, "no server found on '[" .. addr .. "]:" .. tostring( port ) .. "'"
778 _server[ addr..":"..port ] = nil
782 closeall = function( )
783 for _, handler in pairs( _socketlist ) do
785 _socketlist[ _ ] = nil
798 getsettings = function( )
800 select_timeout = _selecttimeout;
801 tcp_backlog = _tcpbacklog;
802 max_send_buffer_size = _maxsendlen;
803 max_receive_buffer_size = _maxreadlen;
804 select_idle_check_interval = _checkinterval;
805 send_timeout = _sendtimeout;
806 read_timeout = _readtimeout;
807 max_connections = _maxselectlen;
808 max_ssl_handshake_roundtrips = _maxsslhandshake;
809 highest_allowed_fd = _maxfd;
810 accept_retry_interval = _accepretry;
814 changesettings = function( new )
815 if type( new ) ~= "table" then
816 return nil, "invalid settings table"
818 _selecttimeout = tonumber( new.select_timeout ) or _selecttimeout
819 _maxsendlen = tonumber( new.max_send_buffer_size ) or _maxsendlen
820 _maxreadlen = tonumber( new.max_receive_buffer_size ) or _maxreadlen
821 _checkinterval = tonumber( new.select_idle_check_interval ) or _checkinterval
822 _tcpbacklog = tonumber( new.tcp_backlog ) or _tcpbacklog
823 _sendtimeout = tonumber( new.send_timeout ) or _sendtimeout
824 _readtimeout = tonumber( new.read_timeout ) or _readtimeout
825 _accepretry = tonumber( new.accept_retry_interval ) or _accepretry
826 _maxselectlen = new.max_connections or _maxselectlen
827 _maxsslhandshake = new.max_ssl_handshake_roundtrips or _maxsslhandshake
828 _maxfd = new.highest_allowed_fd or _maxfd
832 addtimer = function( listener )
833 if type( listener ) ~= "function" then
834 return nil, "invalid listener function"
836 _timerlistlen = _timerlistlen + 1
837 _timerlist[ _timerlistlen ] = listener
845 function add_task(delay, callback)
846 local current_time = luasocket_gettime();
847 delay = delay + current_time;
848 if delay >= current_time then
849 table_insert(new_data, {delay, callback});
851 local r = callback(current_time);
852 if r and type(r) == "number" then
853 return add_task(r, callback);
858 addtimer(function(current_time)
859 if #new_data > 0 then
860 for _, d in pairs(new_data) do
861 table_insert(data, d);
866 local next_time = math_huge;
867 for i, d in pairs(data) do
868 local t, callback = d[1], d[2];
869 if t <= current_time then
871 local r = callback(current_time);
872 if type(r) == "number" then
873 add_task(r, callback);
874 next_time = math_min(next_time, r);
877 next_time = math_min(next_time, t - current_time);
885 return _readtraffic, _sendtraffic, _readlistlen, _sendlistlen, _timerlistlen
890 local function setquitting(quit)
891 quitting = not not quit;
894 loop = function(once) -- this is the main loop of the program
895 if quitting then return "quitting"; end
896 if once then quitting = "once"; end
897 _currenttime = luasocket_gettime( )
900 local next_timer_time = math_huge;
901 for i = 1, _timerlistlen do
902 local t = _timerlist[ i ]( _currenttime ) -- fire timers
903 if t then next_timer_time = math_min(next_timer_time, t); end
906 local read, write, err = socket_select( _readlist, _sendlist, math_min(_selecttimeout, next_timer_time) )
907 for i, socket in ipairs( write ) do -- send data waiting in writequeues
908 local handler = _socketlist[ socket ]
910 handler.sendbuffer( )
912 closesocket( socket )
913 out_put "server.lua: found no handler and closed socket (writelist)" -- this should not happen
916 for i, socket in ipairs( read ) do -- receive data
917 local handler = _socketlist[ socket ]
919 handler.readbuffer( )
921 closesocket( socket )
922 out_put "server.lua: found no handler and closed socket (readlist)" -- this can happen
925 for handler, err in pairs( _closelist ) do
926 handler.disconnect( )( handler, err )
927 handler:force_close() -- forced disconnect
928 _closelist[ handler ] = nil;
930 _currenttime = luasocket_gettime( )
932 -- Check for socket timeouts
933 if _currenttime - _starttime > _checkinterval then
934 _starttime = _currenttime
935 for handler, timestamp in pairs( _writetimes ) do
936 if _currenttime - timestamp > _sendtimeout then
937 handler.disconnect( )( handler, "send timeout" )
938 handler:force_close() -- forced disconnect
941 for handler, timestamp in pairs( _readtimes ) do
942 if _currenttime - timestamp > _readtimeout then
943 if not(handler.onreadtimeout) or handler:onreadtimeout() ~= true then
944 handler.disconnect( )( handler, "read timeout" )
945 handler:close( ) -- forced disconnect?
947 _readtimes[ handler ] = _currenttime -- reset timer
953 for server, paused_time in pairs( _fullservers ) do
954 if _currenttime - paused_time > _accepretry then
955 _fullservers[ server ] = nil;
960 if once and quitting == "once" then quitting = nil; return; end
965 local function step()
969 local function get_backend()
973 --// EXPERIMENTAL //--
975 local wrapclient = function( socket, ip, serverport, listeners, pattern, sslctx )
976 local handler, socket, err = wrapconnection( nil, listeners, socket, ip, serverport, "clientport", pattern, sslctx )
977 if not handler then return nil, err end
978 _socketlist[ socket ] = handler
980 _sendlistlen = addsocket(_sendlist, socket, _sendlistlen)
981 if listeners.onconnect then
982 -- When socket is writeable, call onconnect
983 local _sendbuffer = handler.sendbuffer;
984 handler.sendbuffer = function ()
985 handler.sendbuffer = _sendbuffer;
986 listeners.onconnect(handler);
987 return _sendbuffer(); -- Send any queued outgoing data
991 return handler, socket
994 local addclient = function( address, port, listeners, pattern, sslctx, typ )
996 if type( listeners ) ~= "table" then
997 err = "invalid listener table"
998 elseif type ( address ) ~= "string" then
999 err = "invalid address"
1000 elseif type( port ) ~= "number" or not ( port >= 0 and port <= 65535 ) then
1001 err = "invalid port"
1002 elseif sslctx and not has_luasec then
1003 err = "luasec not found"
1005 if getaddrinfo and not typ then
1006 local addrinfo, err = getaddrinfo(address)
1007 if not addrinfo then return nil, err end
1008 if addrinfo[1] and addrinfo[1].family == "inet6" then
1012 local create = luasocket[typ or "tcp"]
1013 if type( create ) ~= "function" then
1014 err = "invalid socket type"
1018 out_error( "server.lua, addclient: ", err )
1022 local client, err = create( )
1026 client:settimeout( 0 )
1027 local ok, err = client:connect( address, port )
1028 if ok or err == "timeout" or err == "Operation already in progress" then
1029 return wrapclient( client, address, port, listeners, pattern, sslctx )
1035 ----------------------------------// BEGIN //--
1037 use "setmetatable" ( _socketlist, { __mode = "k" } )
1038 use "setmetatable" ( _readtimes, { __mode = "k" } )
1039 use "setmetatable" ( _writetimes, { __mode = "k" } )
1041 _starttime = luasocket_gettime( )
1043 local function setlogger(new_logger)
1044 local old_logger = log;
1051 ----------------------------------// PUBLIC INTERFACE //--
1054 _addtimer = addtimer,
1055 add_task = add_task;
1057 addclient = addclient,
1058 wrapclient = wrapclient,
1064 closeall = closeall,
1065 addserver = addserver,
1066 getserver = getserver,
1067 setlogger = setlogger,
1068 getsettings = getsettings,
1069 setquitting = setquitting,
1070 removeserver = removeserver,
1071 get_backend = get_backend,
1072 changesettings = changesettings,