Software `/` code `/` prosody

Annotate

net/server_event.lua @ 7567:495de404a8ae

ejabberdsql2prosody: rename variable 'host' to prevent shadowing upvalue [luacheck] Functions roster(), roster_pending(), roster_group(), private_storage() and offline_msg() have argument named "host", which used to shadow upvalue of this variable before this change. Instead of renaming this argument, let's rename the variable to match what the script says in usage: Usage: ejabberdsql2prosody.lua filename.txt hostname

author	Anton Shestakov <av6@dwimlabs.net>
date	Fri, 12 Aug 2016 13:44:47 +0800
parent	7420:3fc5560557a5
child	7422:41be97f8158f
child	7688:88ad2d5bcf80

rev	line source
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	1 --[[
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	2
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	3
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	4 server.lua based on lua/libevent by blastbeat
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	5
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	6 notes:
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	7 -- when using luaevent, never register 2 or more EV_READ at one socket, same for EV_WRITE
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	8 -- you cant even register a new EV_READ/EV_WRITE callback inside another one
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	9 -- to do some of the above, use timeout events or something what will called from outside
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	10 -- dont let garbagecollect eventcallbacks, as long they are running
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	11 -- when using luasec, there are 4 cases of timeout errors: wantread or wantwrite during reading or writing
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	12
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	13 --]]
6852 626d8152b1ad server_event: Add luacheck annotations Kim Alvefur <zash@zash.se> parents: 6851 diff changeset	14 -- luacheck: ignore 212/self 431/err 211/ret
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	15
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	16 local SCRIPT_NAME = "server_event.lua"
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	17 local SCRIPT_VERSION = "0.05"
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	18 local SCRIPT_AUTHOR = "blastbeat"
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	19 local LAST_MODIFIED = "2009/11/20"
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	20
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	21 local cfg = {
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	22 MAX_CONNECTIONS = 100000, -- max per server connections (use "ulimit -n" on *nix)
3237 721a83a3beb9 net.server_event: attemp to fix more spelling fail. Waqas Hussain <waqas20@gmail.com> parents: 3149 diff changeset	23 MAX_HANDSHAKE_ATTEMPTS= 1000, -- attempts to finish ssl handshake
721a83a3beb9 net.server_event: attemp to fix more spelling fail. Waqas Hussain <waqas20@gmail.com> parents: 3149 diff changeset	24 HANDSHAKE_TIMEOUT = 60, -- timeout in seconds per handshake attempt
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	25 MAX_READ_LENGTH = 1024 * 1024 * 1024 * 1024, -- max bytes allowed to read from sockets
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	26 MAX_SEND_LENGTH = 1024 * 1024 * 1024 * 1024, -- max bytes size of write buffer (for writing on sockets)
5531 483f795f6f99 net.server: Allow the TCP backlog parameter to be set in the config Kim Alvefur <zash@zash.se>* parents: 5339 diff changeset	27 ACCEPT_QUEUE = 128, -- might influence the length of the pending sockets queue
2952 5069534f7f36 net.server_event: Fixed spelling fail. Waqas Hussain <waqas20@gmail.com> parents: 2941 diff changeset	28 ACCEPT_DELAY = 10, -- seconds to wait until the next attempt of a full server to accept
2939 dc73cbc69bd5 net.server_event: Bump timeout values up somewhat Matthew Wild <mwild1@gmail.com> parents: 2731 diff changeset	29 READ_TIMEOUT = 60 * 60 * 6, -- timeout in seconds for read data from socket
dc73cbc69bd5 net.server_event: Bump timeout values up somewhat Matthew Wild <mwild1@gmail.com> parents: 2731 diff changeset	30 WRITE_TIMEOUT = 180, -- timeout in seconds for write data on socket
2952 5069534f7f36 net.server_event: Fixed spelling fail. Waqas Hussain <waqas20@gmail.com> parents: 2941 diff changeset	31 CONNECT_TIMEOUT = 20, -- timeout in seconds for connection attempts
2535 75bb31b6c53f net.server_event: Trailing whitespace Matthew Wild <mwild1@gmail.com> parents: 2534 diff changeset	32 CLEAR_DELAY = 5, -- seconds to wait for clearing interface list (and calling ondisconnect listeners)
7420 3fc5560557a5 net.server_event: Schedule another read callback if there is still data left in buffer after reading (fixes #583 for real) Kim Alvefur <zash@zash.se> parents: 7419 diff changeset	33 READ_RETRY_DELAY = 1e-06, -- if, after reading, there is still data in buffer, wait this long and continue reading
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	34 DEBUG = true, -- show debug messages
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	35 }
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	36
6850 41de00647ad3 server_event: Move local references to various functions to top of file Kim Alvefur <zash@zash.se> parents: 6849 diff changeset	37 local pairs = pairs
41de00647ad3 server_event: Move local references to various functions to top of file Kim Alvefur <zash@zash.se> parents: 6849 diff changeset	38 local select = select
41de00647ad3 server_event: Move local references to various functions to top of file Kim Alvefur <zash@zash.se> parents: 6849 diff changeset	39 local require = require
41de00647ad3 server_event: Move local references to various functions to top of file Kim Alvefur <zash@zash.se> parents: 6849 diff changeset	40 local tostring = tostring
41de00647ad3 server_event: Move local references to various functions to top of file Kim Alvefur <zash@zash.se> parents: 6849 diff changeset	41 local setmetatable = setmetatable
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	42
5286 0cbe57fbe2cd net.server_event: Optimize socket write buffers (don't do buf=buf..newdata). Waqas Hussain <waqas20@gmail.com> parents: 5277 diff changeset	43 local t_insert = table.insert
0cbe57fbe2cd net.server_event: Optimize socket write buffers (don't do buf=buf..newdata). Waqas Hussain <waqas20@gmail.com> parents: 5277 diff changeset	44 local t_concat = table.concat
6850 41de00647ad3 server_event: Move local references to various functions to top of file Kim Alvefur <zash@zash.se> parents: 6849 diff changeset	45 local s_sub = string.sub
5286 0cbe57fbe2cd net.server_event: Optimize socket write buffers (don't do buf=buf..newdata). Waqas Hussain <waqas20@gmail.com> parents: 5277 diff changeset	46
6850 41de00647ad3 server_event: Move local references to various functions to top of file Kim Alvefur <zash@zash.se> parents: 6849 diff changeset	47 local coroutine_wrap = coroutine.wrap
41de00647ad3 server_event: Move local references to various functions to top of file Kim Alvefur <zash@zash.se> parents: 6849 diff changeset	48 local coroutine_yield = coroutine.yield
5286 0cbe57fbe2cd net.server_event: Optimize socket write buffers (don't do buf=buf..newdata). Waqas Hussain <waqas20@gmail.com> parents: 5277 diff changeset	49
6782 ec172dbe9d14 net.server_{select,event}: Don't rely on LuaSocket and LuaSec being present in the globals table Kim Alvefur <zash@zash.se> parents: 6382 diff changeset	50 local has_luasec, ssl = pcall ( require , "ssl" )
6850 41de00647ad3 server_event: Move local references to various functions to top of file Kim Alvefur <zash@zash.se> parents: 6849 diff changeset	51 local socket = require "socket"
6853 a2cc477eb19b server_event: Rename luaevent module variable to avoid name clash [luacheck] Kim Alvefur <zash@zash.se> parents: 6852 diff changeset	52 local levent = require "luaevent.core"
6850 41de00647ad3 server_event: Move local references to various functions to top of file Kim Alvefur <zash@zash.se> parents: 6849 diff changeset	53
41de00647ad3 server_event: Move local references to various functions to top of file Kim Alvefur <zash@zash.se> parents: 6849 diff changeset	54 local socket_gettime = socket.gettime
6819 ffb2b5e31456 net.server_{select,event}: addclient: Use getaddrinfo to detect IP address type if no socket type argument given. (Argument must be given for non-TCP) Kim Alvefur <zash@zash.se> parents: 6818 diff changeset	55 local getaddrinfo = socket.dns.getaddrinfo
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	56
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	57 local log = require ("util.logger").init("socket")
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	58
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	59 local function debug(...)
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	60 return log("debug", ("%s "):rep(select('#', ...)), ...)
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	61 end
6855 40236602ab5b server_event: Comment out the unused vdebug function [luacheck] Kim Alvefur <zash@zash.se> parents: 6854 diff changeset	62 -- local vdebug = debug;
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	63
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	64 local bitor = ( function( ) -- thx Rici Lake
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	65 local hasbit = function( x, p )
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	66 return x % ( p + p ) >= p
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	67 end
2535 75bb31b6c53f net.server_event: Trailing whitespace Matthew Wild <mwild1@gmail.com> parents: 2534 diff changeset	68 return function( x, y )
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	69 local p = 1
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	70 local z = 0
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	71 local limit = x > y and x or y
2535 75bb31b6c53f net.server_event: Trailing whitespace Matthew Wild <mwild1@gmail.com> parents: 2534 diff changeset	72 while p <= limit do
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	73 if hasbit( x, p ) or hasbit( y, p ) then
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	74 z = z + p
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	75 end
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	76 p = p + p
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	77 end
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	78 return z
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	79 end
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	80 end )( )
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	81
6853 a2cc477eb19b server_event: Rename luaevent module variable to avoid name clash [luacheck] Kim Alvefur <zash@zash.se> parents: 6852 diff changeset	82 local base = levent.new( )
6850 41de00647ad3 server_event: Move local references to various functions to top of file Kim Alvefur <zash@zash.se> parents: 6849 diff changeset	83 local addevent = base.addevent
6853 a2cc477eb19b server_event: Rename luaevent module variable to avoid name clash [luacheck] Kim Alvefur <zash@zash.se> parents: 6852 diff changeset	84 local EV_READ = levent.EV_READ
a2cc477eb19b server_event: Rename luaevent module variable to avoid name clash [luacheck] Kim Alvefur <zash@zash.se> parents: 6852 diff changeset	85 local EV_WRITE = levent.EV_WRITE
a2cc477eb19b server_event: Rename luaevent module variable to avoid name clash [luacheck] Kim Alvefur <zash@zash.se> parents: 6852 diff changeset	86 local EV_TIMEOUT = levent.EV_TIMEOUT
a2cc477eb19b server_event: Rename luaevent module variable to avoid name clash [luacheck] Kim Alvefur <zash@zash.se> parents: 6852 diff changeset	87 local EV_SIGNAL = levent.EV_SIGNAL
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	88
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	89 local EV_READWRITE = bitor( EV_READ, EV_WRITE )
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	90
6849 0455b9686e16 server_event: Replace implementation of ordered set with a simple table set Kim Alvefur <zash@zash.se> parents: 6819 diff changeset	91 local interfacelist = { }
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	92
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	93 -- Client interface methods
6851 1f1bed8ebc41 server_event: Remove needless scoping and indentation Kim Alvefur <zash@zash.se> parents: 6850 diff changeset	94 local interface_mt = {}; interface_mt.__index = interface_mt;
5776 bd0ff8ae98a8 Remove all trailing whitespace Florian Zeitz <florob@babelmonkeys.de> parents: 5756 diff changeset	95
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	96 -- Private methods
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	97 function interface_mt:_close()
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	98 return self:_destroy();
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	99 end
5776 bd0ff8ae98a8 Remove all trailing whitespace Florian Zeitz <florob@babelmonkeys.de> parents: 5756 diff changeset	100
7339 b5565715fce8 server_event: addclient: wrapclient already calls startconnection for us [backported from trunk] daurnimator <quae@daurnimator.com> parents: 7337 diff changeset	101 function interface_mt:_start_connection(plainssl) -- called from wrapclient
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	102 local callback = function( event )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	103 if EV_TIMEOUT == event then -- timeout during connection
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	104 self.fatalerror = "connection timeout"
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	105 self:ontimeout() -- call timeout listener
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	106 self:_close()
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	107 debug( "new connection failed. id:", self.id, "error:", self.fatalerror )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	108 else
6858 d8f8c0b2fda8 server_event: Normalize indentation Kim Alvefur <zash@zash.se> parents: 6856 diff changeset	109 if plainssl and has_luasec then -- start ssl session
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	110 self:starttls(self._sslctx, true)
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	111 else -- normal connection
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	112 self:_start_session(true)
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	113 end
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	114 debug( "new connection established. id:", self.id )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	115 end
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	116 self.eventconnect = nil
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	117 return -1
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	118 end
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	119 self.eventconnect = addevent( base, self.conn, EV_WRITE, callback, cfg.CONNECT_TIMEOUT )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	120 return true
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	121 end
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	122 function interface_mt:_start_session(call_onconnect) -- new session, for example after startssl
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	123 if self.type == "client" then
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	124 local callback = function( )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	125 self:_lock( false, false, false )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	126 --vdebug( "start listening on client socket with id:", self.id )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	127 self.eventread = addevent( base, self.conn, EV_READ, self.readcallback, cfg.READ_TIMEOUT ); -- register callback
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	128 if call_onconnect then
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	129 self:onconnect()
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	130 end
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	131 self.eventsession = nil
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	132 return -1
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	133 end
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	134 self.eventsession = addevent( base, nil, EV_TIMEOUT, callback, 0 )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	135 else
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	136 self:_lock( false )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	137 --vdebug( "start listening on server socket with id:", self.id )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	138 self.eventread = addevent( base, self.conn, EV_READ, self.readcallback ) -- register callback
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	139 end
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	140 return true
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	141 end
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	142 function interface_mt:_start_ssl(call_onconnect) -- old socket will be destroyed, therefore we have to close read/write events first
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	143 --vdebug( "starting ssl session with client id:", self.id )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	144 local _
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	145 _ = self.eventread and self.eventread:close( ) -- close events; this must be called outside of the event callbacks!
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	146 _ = self.eventwrite and self.eventwrite:close( )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	147 self.eventread, self.eventwrite = nil, nil
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	148 local err
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	149 self.conn, err = ssl.wrap( self.conn, self._sslctx )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	150 if err then
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	151 self.fatalerror = err
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	152 self.conn = nil -- cannot be used anymore
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	153 if call_onconnect then
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	154 self.ondisconnect = nil -- dont call this when client isnt really connected
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	155 end
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	156 self:_close()
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	157 debug( "fatal error while ssl wrapping:", err )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	158 return false
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	159 end
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	160 self.conn:settimeout( 0 ) -- set non blocking
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	161 local handshakecallback = coroutine_wrap(function( event )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	162 local _, err
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	163 local attempt = 0
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	164 local maxattempt = cfg.MAX_HANDSHAKE_ATTEMPTS
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	165 while attempt < maxattempt do -- no endless loop
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	166 attempt = attempt + 1
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	167 debug( "ssl handshake of client with id:"..tostring(self)..", attempt:"..attempt )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	168 if attempt > maxattempt then
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	169 self.fatalerror = "max handshake attempts exceeded"
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	170 elseif EV_TIMEOUT == event then
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	171 self.fatalerror = "timeout during handshake"
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	172 else
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	173 _, err = self.conn:dohandshake( )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	174 if not err then
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	175 self:_lock( false, false, false ) -- unlock the interface; sending, closing etc allowed
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	176 self.send = self.conn.send -- caching table lookups with new client object
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	177 self.receive = self.conn.receive
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	178 if not call_onconnect then -- trigger listener
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	179 self:onstatus("ssl-handshake-complete");
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	180 end
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	181 self:_start_session( call_onconnect )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	182 debug( "ssl handshake done" )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	183 self.eventhandshake = nil
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	184 return -1
3396 23cf369ed1c3 net.server_event: Fix to not call onconnect a second time after the SSL handshake for starttls connections (thanks Flo) Matthew Wild <mwild1@gmail.com> parents: 3387 diff changeset	185 end
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	186 if err == "wantwrite" then
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	187 event = EV_WRITE
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	188 elseif err == "wantread" then
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	189 event = EV_READ
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	190 else
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	191 debug( "ssl handshake error:", err )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	192 self.fatalerror = err
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	193 end
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	194 end
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	195 if self.fatalerror then
3396 23cf369ed1c3 net.server_event: Fix to not call onconnect a second time after the SSL handshake for starttls connections (thanks Flo) Matthew Wild <mwild1@gmail.com> parents: 3387 diff changeset	196 if call_onconnect then
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	197 self.ondisconnect = nil -- dont call this when client isnt really connected
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	198 end
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	199 self:_close()
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	200 debug( "handshake failed because:", self.fatalerror )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	201 self.eventhandshake = nil
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	202 return -1
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	203 end
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	204 event = coroutine_yield( event, cfg.HANDSHAKE_TIMEOUT ) -- yield this monster...
3000 43a4cb707fe4 net.server_select, net.server_event: Support for :pause() and :resume() on connections Matthew Wild <mwild1@gmail.com> parents: 2988 diff changeset	205 end
43a4cb707fe4 net.server_select, net.server_event: Support for :pause() and :resume() on connections Matthew Wild <mwild1@gmail.com> parents: 2988 diff changeset	206 end
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	207 )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	208 debug "starting handshake..."
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	209 self:_lock( false, true, true ) -- unlock read/write events, but keep interface locked
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	210 self.eventhandshake = addevent( base, self.conn, EV_READWRITE, handshakecallback, cfg.HANDSHAKE_TIMEOUT )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	211 return true
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	212 end
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	213 function interface_mt:_destroy() -- close this interface + events and call last listener
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	214 debug( "closing client with id:", self.id, self.fatalerror )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	215 self:_lock( true, true, true ) -- first of all, lock the interface to avoid further actions
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	216 local _
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	217 _ = self.eventread and self.eventread:close( )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	218 if self.type == "client" then
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	219 _ = self.eventwrite and self.eventwrite:close( )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	220 _ = self.eventhandshake and self.eventhandshake:close( )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	221 _ = self.eventstarthandshake and self.eventstarthandshake:close( )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	222 _ = self.eventconnect and self.eventconnect:close( )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	223 _ = self.eventsession and self.eventsession:close( )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	224 _ = self.eventwritetimeout and self.eventwritetimeout:close( )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	225 _ = self.eventreadtimeout and self.eventreadtimeout:close( )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	226 _ = self.ondisconnect and self:ondisconnect( self.fatalerror ~= "client to close" and self.fatalerror) -- call ondisconnect listener (wont be the case if handshake failed on connect)
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	227 _ = self.conn and self.conn:close( ) -- close connection
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	228 _ = self._server and self._server:counter(-1);
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	229 self.eventread, self.eventwrite = nil, nil
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	230 self.eventstarthandshake, self.eventhandshake, self.eventclose = nil, nil, nil
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	231 self.readcallback, self.writecallback = nil, nil
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	232 else
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	233 self.conn:close( )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	234 self.eventread, self.eventclose = nil, nil
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	235 self.interface, self.readcallback = nil, nil
3000 43a4cb707fe4 net.server_select, net.server_event: Support for :pause() and :resume() on connections Matthew Wild <mwild1@gmail.com> parents: 2988 diff changeset	236 end
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	237 interfacelist[ self ] = nil
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	238 return true
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	239 end
3000 43a4cb707fe4 net.server_select, net.server_event: Support for :pause() and :resume() on connections Matthew Wild <mwild1@gmail.com> parents: 2988 diff changeset	240
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	241 function interface_mt:_lock(nointerface, noreading, nowriting) -- lock or unlock this interface or events
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	242 self.nointerface, self.noreading, self.nowriting = nointerface, noreading, nowriting
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	243 return nointerface, noreading, nowriting
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	244 end
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	245
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	246 --TODO: Deprecate
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	247 function interface_mt:lock_read(switch)
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	248 if switch then
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	249 return self:pause();
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	250 else
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	251 return self:resume();
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	252 end
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	253 end
3000 43a4cb707fe4 net.server_select, net.server_event: Support for :pause() and :resume() on connections Matthew Wild <mwild1@gmail.com> parents: 2988 diff changeset	254
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	255 function interface_mt:pause()
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	256 return self:_lock(self.nointerface, true, self.nowriting);
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	257 end
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	258
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	259 function interface_mt:resume()
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	260 self:_lock(self.nointerface, false, self.nowriting);
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	261 if self.readcallback and not self.eventread then
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	262 self.eventread = addevent( base, self.conn, EV_READ, self.readcallback, cfg.READ_TIMEOUT ); -- register callback
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	263 return true;
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	264 end
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	265 end
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	266
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	267 function interface_mt:counter(c)
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	268 if c then
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	269 self._connections = self._connections + c
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	270 end
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	271 return self._connections
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	272 end
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	273
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	274 -- Public methods
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	275 function interface_mt:write(data)
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	276 if self.nowriting then return nil, "locked" end
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	277 --vdebug( "try to send data to client, id/data:", self.id, data )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	278 data = tostring( data )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	279 local len = #data
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	280 local total = len + self.writebufferlen
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	281 if total > cfg.MAX_SEND_LENGTH then -- check buffer length
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	282 local err = "send buffer exceeded"
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	283 debug( "error:", err ) -- to much, check your app
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	284 return nil, err
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	285 end
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	286 t_insert(self.writebuffer, data) -- new buffer
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	287 self.writebufferlen = total
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	288 if not self.eventwrite then -- register new write event
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	289 --vdebug( "register new write event" )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	290 self.eventwrite = addevent( base, self.conn, EV_WRITE, self.writecallback, cfg.WRITE_TIMEOUT )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	291 end
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	292 return true
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	293 end
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	294 function interface_mt:close()
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	295 if self.nointerface then return nil, "locked"; end
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	296 debug( "try to close client connection with id:", self.id )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	297 if self.type == "client" then
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	298 self.fatalerror = "client to close"
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	299 if self.eventwrite then -- wait for incomplete write request
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	300 self:_lock( true, true, false )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	301 debug "closing delayed until writebuffer is empty"
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	302 return nil, "writebuffer not empty, waiting"
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	303 else -- close now
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	304 self:_lock( true, true, true )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	305 self:_close()
4676 1ce2ff9ec313 net.server_event: Do not automatically close clients when closing a server port Matthew Wild <mwild1@gmail.com> parents: 4468 diff changeset	306 return true
2535 75bb31b6c53f net.server_event: Trailing whitespace Matthew Wild <mwild1@gmail.com> parents: 2534 diff changeset	307 end
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	308 else
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	309 debug( "try to close server with id:", tostring(self.id))
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	310 self.fatalerror = "server to close"
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	311 self:_lock( true )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	312 self:_close( 0 )
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	313 return true
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	314 end
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	315 end
7224 07a4c807a94a Merge 0.9->0.10 Kim Alvefur <zash@zash.se> parents: 7040 7223 diff changeset	316
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	317 function interface_mt:socket()
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	318 return self.conn
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	319 end
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	320
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	321 function interface_mt:server()
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	322 return self._server or self;
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	323 end
7224 07a4c807a94a Merge 0.9->0.10 Kim Alvefur <zash@zash.se> parents: 7040 7223 diff changeset	324
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	325 function interface_mt:port()
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	326 return self._port
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	327 end
7224 07a4c807a94a Merge 0.9->0.10 Kim Alvefur <zash@zash.se> parents: 7040 7223 diff changeset	328
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	329 function interface_mt:serverport()
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	330 return self._serverport
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	331 end
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	332
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	333 function interface_mt:ip()
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	334 return self._ip
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	335 end
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	336
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	337 function interface_mt:ssl()
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	338 return self._usingssl
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	339 end
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	340 interface_mt.clientport = interface_mt.port -- COMPAT server_select
7224 07a4c807a94a Merge 0.9->0.10 Kim Alvefur <zash@zash.se> parents: 7040 7223 diff changeset	341
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	342 function interface_mt:type()
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	343 return self._type or "client"
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	344 end
7224 07a4c807a94a Merge 0.9->0.10 Kim Alvefur <zash@zash.se> parents: 7040 7223 diff changeset	345
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	346 function interface_mt:connections()
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	347 return self._connections
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	348 end
7224 07a4c807a94a Merge 0.9->0.10 Kim Alvefur <zash@zash.se> parents: 7040 7223 diff changeset	349
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	350 function interface_mt:address()
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	351 return self.addr
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	352 end
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	353
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	354 function interface_mt:set_sslctx(sslctx)
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	355 self._sslctx = sslctx;
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	356 if sslctx then
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	357 self.starttls = nil; -- use starttls() of interface_mt
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	358 else
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	359 self.starttls = false; -- prevent starttls()
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	360 end
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	361 end
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	362
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	363 function interface_mt:set_mode(pattern)
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	364 if pattern then
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	365 self._pattern = pattern;
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	366 end
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	367 return self._pattern;
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	368 end
7224 07a4c807a94a Merge 0.9->0.10 Kim Alvefur <zash@zash.se> parents: 7040 7223 diff changeset	369
07a4c807a94a Merge 0.9->0.10 Kim Alvefur <zash@zash.se> parents: 7040 7223 diff changeset	370 function interface_mt:set_send(new_send) -- luacheck: ignore 212
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	371 -- No-op, we always use the underlying connection's send
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	372 end
3001 fa00c6f563b4 net.server_select, net.server_event: Rename :pattern() method to :set_mode() for select backend, and add :set_mode() to event backend Matthew Wild <mwild1@gmail.com> parents: 3000 diff changeset	373
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	374 function interface_mt:starttls(sslctx, call_onconnect)
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	375 debug( "try to start ssl at client id:", self.id )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	376 local err
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	377 self._sslctx = sslctx;
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	378 if self._usingssl then -- startssl was already called
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	379 err = "ssl already active"
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	380 end
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	381 if err then
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	382 debug( "error:", err )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	383 return nil, err
3001 fa00c6f563b4 net.server_select, net.server_event: Rename :pattern() method to :set_mode() for select backend, and add :set_mode() to event backend Matthew Wild <mwild1@gmail.com> parents: 3000 diff changeset	384 end
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	385 self._usingssl = true
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	386 self.startsslcallback = function( ) -- we have to start the handshake outside of a read/write event
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	387 self.startsslcallback = nil
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	388 self:_start_ssl(call_onconnect);
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	389 self.eventstarthandshake = nil
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	390 return -1
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	391 end
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	392 if not self.eventwrite then
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	393 self:_lock( true, true, true ) -- lock the interface, to not disturb the handshake
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	394 self.eventstarthandshake = addevent( base, nil, EV_TIMEOUT, self.startsslcallback, 0 ) -- add event to start handshake
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	395 else
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	396 -- wait until writebuffer is empty
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	397 self:_lock( true, true, false )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	398 debug "ssl session delayed until writebuffer is empty..."
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	399 end
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	400 self.starttls = false;
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	401 return true
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	402 end
5625 e67891ad18d6 net.server_event: Add support for listener.onreadtimeout(conn), which can return true to prevent the connection from being closed when a read timeout occurs Matthew Wild <mwild1@gmail.com> parents: 5531 diff changeset	403
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	404 function interface_mt:setoption(option, value)
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	405 if self.conn.setoption then
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	406 return self.conn:setoption(option, value);
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	407 end
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	408 return false, "setoption not implemented";
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	409 end
6851 1f1bed8ebc41 server_event: Remove needless scoping and indentation Kim Alvefur <zash@zash.se> parents: 6850 diff changeset	410
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	411 function interface_mt:setlistener(listener)
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	412 self:ondetach(); -- Notify listener that it is no longer responsible for this connection
7337 8ec935687bae server_event: Split long list of assignments into multiple lines for readability Kim Alvefur <zash@zash.se> parents: 7336 diff changeset	413 self.onconnect = listener.onconnect;
8ec935687bae server_event: Split long list of assignments into multiple lines for readability Kim Alvefur <zash@zash.se> parents: 7336 diff changeset	414 self.ondisconnect = listener.ondisconnect;
8ec935687bae server_event: Split long list of assignments into multiple lines for readability Kim Alvefur <zash@zash.se> parents: 7336 diff changeset	415 self.onincoming = listener.onincoming;
8ec935687bae server_event: Split long list of assignments into multiple lines for readability Kim Alvefur <zash@zash.se> parents: 7336 diff changeset	416 self.ontimeout = listener.ontimeout;
8ec935687bae server_event: Split long list of assignments into multiple lines for readability Kim Alvefur <zash@zash.se> parents: 7336 diff changeset	417 self.onreadtimeout = listener.onreadtimeout;
8ec935687bae server_event: Split long list of assignments into multiple lines for readability Kim Alvefur <zash@zash.se> parents: 7336 diff changeset	418 self.onstatus = listener.onstatus;
8ec935687bae server_event: Split long list of assignments into multiple lines for readability Kim Alvefur <zash@zash.se> parents: 7336 diff changeset	419 self.ondetach = listener.ondetach;
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	420 end
6851 1f1bed8ebc41 server_event: Remove needless scoping and indentation Kim Alvefur <zash@zash.se> parents: 6850 diff changeset	421
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	422 -- Stub handlers
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	423 function interface_mt:onconnect()
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	424 end
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	425 function interface_mt:onincoming()
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	426 end
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	427 function interface_mt:ondisconnect()
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	428 end
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	429 function interface_mt:ontimeout()
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	430 end
6851 1f1bed8ebc41 server_event: Remove needless scoping and indentation Kim Alvefur <zash@zash.se> parents: 6850 diff changeset	431 function interface_mt:onreadtimeout()
1f1bed8ebc41 server_event: Remove needless scoping and indentation Kim Alvefur <zash@zash.se> parents: 6850 diff changeset	432 self.fatalerror = "timeout during receiving"
1f1bed8ebc41 server_event: Remove needless scoping and indentation Kim Alvefur <zash@zash.se> parents: 6850 diff changeset	433 debug( "connection failed:", self.fatalerror )
1f1bed8ebc41 server_event: Remove needless scoping and indentation Kim Alvefur <zash@zash.se> parents: 6850 diff changeset	434 self:_close()
1f1bed8ebc41 server_event: Remove needless scoping and indentation Kim Alvefur <zash@zash.se> parents: 6850 diff changeset	435 self.eventread = nil
1f1bed8ebc41 server_event: Remove needless scoping and indentation Kim Alvefur <zash@zash.se> parents: 6850 diff changeset	436 end
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	437 function interface_mt:ondrain()
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	438 end
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	439 function interface_mt:ondetach()
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	440 end
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	441 function interface_mt:onstatus()
2535 75bb31b6c53f net.server_event: Trailing whitespace Matthew Wild <mwild1@gmail.com> parents: 2534 diff changeset	442 end
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	443
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	444 -- End of client interface methods
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	445
6851 1f1bed8ebc41 server_event: Remove needless scoping and indentation Kim Alvefur <zash@zash.se> parents: 6850 diff changeset	446 local function handleclient( client, ip, port, server, pattern, listener, sslctx ) -- creates an client interface
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	447 --vdebug("creating client interfacce...")
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	448 local interface = {
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	449 type = "client";
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	450 conn = client;
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	451 currenttime = socket_gettime( ); -- safe the origin
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	452 writebuffer = {}; -- writebuffer
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	453 writebufferlen = 0; -- length of writebuffer
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	454 send = client.send; -- caching table lookups
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	455 receive = client.receive;
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	456 onconnect = listener.onconnect; -- will be called when client disconnects
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	457 ondisconnect = listener.ondisconnect; -- will be called when client disconnects
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	458 onincoming = listener.onincoming; -- will be called when client sends data
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	459 ontimeout = listener.ontimeout; -- called when fatal socket timeout occurs
6851 1f1bed8ebc41 server_event: Remove needless scoping and indentation Kim Alvefur <zash@zash.se> parents: 6850 diff changeset	460 onreadtimeout = listener.onreadtimeout; -- called when socket inactivity timeout occurs
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	461 ondrain = listener.ondrain; -- called when writebuffer is empty
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	462 ondetach = listener.ondetach; -- called when disassociating this listener from this connection
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	463 onstatus = listener.onstatus; -- called for status changes (e.g. of SSL/TLS)
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	464 eventread = false, eventwrite = false, eventclose = false,
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	465 eventhandshake = false, eventstarthandshake = false; -- event handler
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	466 eventconnect = false, eventsession = false; -- more event handler...
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	467 eventwritetimeout = false; -- even more event handler...
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	468 eventreadtimeout = false;
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	469 fatalerror = false; -- error message
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	470 writecallback = false; -- will be called on write events
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	471 readcallback = false; -- will be called on read events
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	472 nointerface = true; -- lock/unlock parameter of this interface
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	473 noreading = false, nowriting = false; -- locks of the read/writecallback
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	474 startsslcallback = false; -- starting handshake callback
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	475 position = false; -- position of client in interfacelist
5776 bd0ff8ae98a8 Remove all trailing whitespace Florian Zeitz <florob@babelmonkeys.de> parents: 5756 diff changeset	476
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	477 -- Properties
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	478 _ip = ip, _port = port, _server = server, _pattern = pattern,
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	479 _serverport = (server and server:port() or nil),
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	480 _sslctx = sslctx; -- parameters
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	481 _usingssl = false; -- client is using ssl;
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	482 }
6851 1f1bed8ebc41 server_event: Remove needless scoping and indentation Kim Alvefur <zash@zash.se> parents: 6850 diff changeset	483 if not has_luasec then interface.starttls = false; end
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	484 interface.id = tostring(interface):match("%x+$");
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	485 interface.writecallback = function( event ) -- called on write events
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	486 --vdebug( "new client write event, id/ip/port:", interface, ip, port )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	487 if interface.nowriting or ( interface.fatalerror and ( "client to close" ~= interface.fatalerror ) ) then -- leave this event
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	488 --vdebug( "leaving this event because:", interface.nowriting or interface.fatalerror )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	489 interface.eventwrite = false
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	490 return -1
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	491 end
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	492 if EV_TIMEOUT == event then -- took too long to write some data to socket -> disconnect
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	493 interface.fatalerror = "timeout during writing"
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	494 debug( "writing failed:", interface.fatalerror )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	495 interface:_close()
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	496 interface.eventwrite = false
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	497 return -1
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	498 else -- can write :)
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	499 if interface._usingssl then -- handle luasec
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	500 if interface.eventreadtimeout then -- we have to read first
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	501 local ret = interface.readcallback( ) -- call readcallback
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	502 --vdebug( "tried to read in writecallback, result:", ret )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	503 end
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	504 if interface.eventwritetimeout then -- luasec only
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	505 interface.eventwritetimeout:close( ) -- first we have to close timeout event which where regged after a wantread error
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	506 interface.eventwritetimeout = false
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	507 end
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	508 end
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	509 interface.writebuffer = { t_concat(interface.writebuffer) }
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	510 local succ, err, byte = interface.conn:send( interface.writebuffer[1], 1, interface.writebufferlen )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	511 --vdebug( "write data:", interface.writebuffer, "error:", err, "part:", byte )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	512 if succ then -- writing succesful
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	513 interface.writebuffer[1] = nil
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	514 interface.writebufferlen = 0
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	515 interface:ondrain();
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	516 if interface.fatalerror then
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	517 debug "closing client after writing"
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	518 interface:_close() -- close interface if needed
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	519 elseif interface.startsslcallback then -- start ssl connection if needed
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	520 debug "starting ssl handshake after writing"
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	521 interface.eventstarthandshake = addevent( base, nil, EV_TIMEOUT, interface.startsslcallback, 0 )
7399 066a4dfdac18 Merge 0.9->0.10 Kim Alvefur <zash@zash.se> parents: 7394 7398 diff changeset	522 elseif interface.writebufferlen ~= 0 then
7394 a700d54c620a Merge 0.9->0.10 Kim Alvefur <zash@zash.se> parents: 7385 7393 diff changeset	523 -- data possibly written from ondrain
a700d54c620a Merge 0.9->0.10 Kim Alvefur <zash@zash.se> parents: 7385 7393 diff changeset	524 return EV_WRITE, cfg.WRITE_TIMEOUT
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	525 elseif interface.eventreadtimeout then
7385 17929cdacec2 Merge 0.9->0.10 Kim Alvefur <zash@zash.se> parents: 7339 7384 diff changeset	526 return EV_WRITE, cfg.WRITE_TIMEOUT
17929cdacec2 Merge 0.9->0.10 Kim Alvefur <zash@zash.se> parents: 7339 7384 diff changeset	527 end
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	528 interface.eventwrite = nil
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	529 return -1
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	530 elseif byte and (err == "timeout" or err == "wantwrite") then -- want write again
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	531 --vdebug( "writebuffer is not empty:", err )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	532 interface.writebuffer[1] = s_sub( interface.writebuffer[1], byte + 1, interface.writebufferlen ) -- new buffer
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	533 interface.writebufferlen = interface.writebufferlen - byte
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	534 if "wantread" == err then -- happens only with luasec
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	535 local callback = function( )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	536 interface:_close()
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	537 interface.eventwritetimeout = nil
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	538 return -1;
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	539 end
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	540 interface.eventwritetimeout = addevent( base, nil, EV_TIMEOUT, callback, cfg.WRITE_TIMEOUT ) -- reg a new timeout event
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	541 debug( "wantread during write attempt, reg it in readcallback but dont know what really happens next..." )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	542 -- hopefully this works with luasec; its simply not possible to use 2 different write events on a socket in luaevent
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	543 return -1
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	544 end
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	545 return EV_WRITE, cfg.WRITE_TIMEOUT
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	546 else -- connection was closed during writing or fatal error
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	547 interface.fatalerror = err or "fatal error"
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	548 debug( "connection failed in write event:", interface.fatalerror )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	549 interface:_close()
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	550 interface.eventwrite = nil
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	551 return -1
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	552 end
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	553 end
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	554 end
7224 07a4c807a94a Merge 0.9->0.10 Kim Alvefur <zash@zash.se> parents: 7040 7223 diff changeset	555
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	556 interface.readcallback = function( event ) -- called on read events
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	557 --vdebug( "new client read event, id/ip/port:", tostring(interface.id), tostring(ip), tostring(port) )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	558 if interface.noreading or interface.fatalerror then -- leave this event
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	559 --vdebug( "leaving this event because:", tostring(interface.noreading or interface.fatalerror) )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	560 interface.eventread = nil
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	561 return -1
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	562 end
7420 3fc5560557a5 net.server_event: Schedule another read callback if there is still data left in buffer after reading (fixes #583 for real) Kim Alvefur <zash@zash.se> parents: 7419 diff changeset	563 if EV_TIMEOUT == event and not interface.conn:dirty() and interface:onreadtimeout() ~= true then
6851 1f1bed8ebc41 server_event: Remove needless scoping and indentation Kim Alvefur <zash@zash.se> parents: 6850 diff changeset	564 return -1 -- took too long to get some data from client -> disconnect
1f1bed8ebc41 server_event: Remove needless scoping and indentation Kim Alvefur <zash@zash.se> parents: 6850 diff changeset	565 end
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	566 if interface._usingssl then -- handle luasec
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	567 if interface.eventwritetimeout then -- ok, in the past writecallback was regged
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	568 local ret = interface.writecallback( ) -- call it
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	569 --vdebug( "tried to write in readcallback, result:", tostring(ret) )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	570 end
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	571 if interface.eventreadtimeout then
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	572 interface.eventreadtimeout:close( )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	573 interface.eventreadtimeout = nil
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	574 end
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	575 end
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	576 local buffer, err, part = interface.conn:receive( interface._pattern ) -- receive buffer with "pattern"
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	577 --vdebug( "read data:", tostring(buffer), "error:", tostring(err), "part:", tostring(part) )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	578 buffer = buffer or part
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	579 if buffer and #buffer > cfg.MAX_READ_LENGTH then -- check buffer length
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	580 interface.fatalerror = "receive buffer exceeded"
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	581 debug( "fatal error:", interface.fatalerror )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	582 interface:_close()
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	583 interface.eventread = nil
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	584 return -1
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	585 end
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	586 if err and ( err ~= "timeout" and err ~= "wantread" ) then
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	587 if "wantwrite" == err then -- need to read on write event
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	588 if not interface.eventwrite then -- register new write event if needed
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	589 interface.eventwrite = addevent( base, interface.conn, EV_WRITE, interface.writecallback, cfg.WRITE_TIMEOUT )
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	590 end
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	591 interface.eventreadtimeout = addevent( base, nil, EV_TIMEOUT,
7419 325c03f5481b net.server_event: Collect entire one line function on one line Kim Alvefur <zash@zash.se> parents: 7399 diff changeset	592 function( ) interface:_close() end, cfg.READ_TIMEOUT)
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	593 debug( "wantwrite during read attempt, reg it in writecallback but dont know what really happens next..." )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	594 -- to be honest i dont know what happens next, if it is allowed to first read, the write etc...
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	595 else -- connection was closed or fatal error
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	596 interface.fatalerror = err
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	597 debug( "connection failed in read event:", interface.fatalerror )
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	598 interface:_close()
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	599 interface.eventread = nil
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	600 return -1
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	601 end
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	602 else
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	603 interface.onincoming( interface, buffer, err ) -- send new data to listener
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	604 end
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	605 if interface.noreading then
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	606 interface.eventread = nil;
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	607 return -1;
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	608 end
7420 3fc5560557a5 net.server_event: Schedule another read callback if there is still data left in buffer after reading (fixes #583 for real) Kim Alvefur <zash@zash.se> parents: 7419 diff changeset	609 if interface.conn:dirty() then -- still data left in buffer
3fc5560557a5 net.server_event: Schedule another read callback if there is still data left in buffer after reading (fixes #583 for real) Kim Alvefur <zash@zash.se> parents: 7419 diff changeset	610 return EV_TIMEOUT, cfg.READ_RETRY_DELAY;
3fc5560557a5 net.server_event: Schedule another read callback if there is still data left in buffer after reading (fixes #583 for real) Kim Alvefur <zash@zash.se> parents: 7419 diff changeset	611 end
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	612 return EV_READ, cfg.READ_TIMEOUT
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	613 end
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	614
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	615 client:settimeout( 0 ) -- set non blocking
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	616 setmetatable(interface, interface_mt)
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	617 interfacelist[ interface ] = true -- add to interfacelist
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	618 return interface
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	619 end
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	620
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	621 local function handleserver( server, addr, port, pattern, listener, sslctx ) -- creates an server interface
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	622 debug "creating server interface..."
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	623 local interface = {
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	624 _connections = 0;
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	625
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	626 type = "server";
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	627 conn = server;
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	628 onconnect = listener.onconnect; -- will be called when new client connected
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	629 eventread = false; -- read event handler
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	630 eventclose = false; -- close event handler
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	631 readcallback = false; -- read event callback
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	632 fatalerror = false; -- error message
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	633 nointerface = true; -- lock/unlock parameter
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	634
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	635 _ip = addr, _port = port, _pattern = pattern,
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	636 _sslctx = sslctx;
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	637 }
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	638 interface.id = tostring(interface):match("%x+$");
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	639 interface.readcallback = function( event ) -- server handler, called on incoming connections
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	640 --vdebug( "server can accept, id/addr/port:", interface, addr, port )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	641 if interface.fatalerror then
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	642 --vdebug( "leaving this event because:", self.fatalerror )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	643 interface.eventread = nil
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	644 return -1
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	645 end
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	646 local delay = cfg.ACCEPT_DELAY
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	647 if EV_TIMEOUT == event then
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	648 if interface._connections >= cfg.MAX_CONNECTIONS then -- check connection count
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	649 debug( "to many connections, seconds to wait for next accept:", delay )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	650 return EV_TIMEOUT, delay -- timeout...
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	651 else
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	652 return EV_READ -- accept again
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	653 end
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	654 end
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	655 --vdebug("max connection check ok, accepting...")
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	656 local client, err = server:accept() -- try to accept; TODO: check err
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	657 while client do
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	658 if interface._connections >= cfg.MAX_CONNECTIONS then
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	659 client:close( ) -- refuse connection
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	660 debug( "maximal connections reached, refuse client connection; accept delay:", delay )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	661 return EV_TIMEOUT, delay -- delay for next accept attempt
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	662 end
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	663 local client_ip, client_port = client:getpeername( )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	664 interface._connections = interface._connections + 1 -- increase connection count
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	665 local clientinterface = handleclient( client, client_ip, client_port, interface, pattern, listener, sslctx )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	666 --vdebug( "client id:", clientinterface, "startssl:", startssl )
7224 07a4c807a94a Merge 0.9->0.10 Kim Alvefur <zash@zash.se> parents: 7040 7223 diff changeset	667 if has_luasec and sslctx then
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	668 clientinterface:starttls(sslctx, true)
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	669 else
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	670 clientinterface:_start_session( true )
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	671 end
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	672 debug( "accepted incoming client connection from:", client_ip or "<unknown IP>", client_port or "<unknown port>", "to", port or "<unknown port>");
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	673
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	674 client, err = server:accept() -- try to accept again
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	675 end
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	676 return EV_READ
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	677 end
6851 1f1bed8ebc41 server_event: Remove needless scoping and indentation Kim Alvefur <zash@zash.se> parents: 6850 diff changeset	678
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	679 server:settimeout( 0 )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	680 setmetatable(interface, interface_mt)
7224 07a4c807a94a Merge 0.9->0.10 Kim Alvefur <zash@zash.se> parents: 7040 7223 diff changeset	681 interfacelist[ interface ] = true
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	682 interface:_start_session()
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	683 return interface
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	684 end
6851 1f1bed8ebc41 server_event: Remove needless scoping and indentation Kim Alvefur <zash@zash.se> parents: 6850 diff changeset	685
1f1bed8ebc41 server_event: Remove needless scoping and indentation Kim Alvefur <zash@zash.se> parents: 6850 diff changeset	686 local function addserver( addr, port, listener, pattern, sslctx, startssl ) -- TODO: check arguments
1f1bed8ebc41 server_event: Remove needless scoping and indentation Kim Alvefur <zash@zash.se> parents: 6850 diff changeset	687 --vdebug( "creating new tcp server with following parameters:", addr or "nil", port or "nil", sslctx or "nil", startssl or "nil")
1f1bed8ebc41 server_event: Remove needless scoping and indentation Kim Alvefur <zash@zash.se> parents: 6850 diff changeset	688 if sslctx and not has_luasec then
1f1bed8ebc41 server_event: Remove needless scoping and indentation Kim Alvefur <zash@zash.se> parents: 6850 diff changeset	689 debug "fatal error: luasec not found"
1f1bed8ebc41 server_event: Remove needless scoping and indentation Kim Alvefur <zash@zash.se> parents: 6850 diff changeset	690 return nil, "luasec not found"
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	691 end
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	692 local server, err = socket.bind( addr, port, cfg.ACCEPT_QUEUE ) -- create server socket
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	693 if not server then
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	694 debug( "creating server socket on "..addr.." port "..port.." failed:", err )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	695 return nil, err
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	696 end
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	697 local interface = handleserver( server, addr, port, pattern, listener, sslctx, startssl ) -- new server handler
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	698 debug( "new server created with id:", tostring(interface))
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	699 return interface
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	700 end
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	701
6851 1f1bed8ebc41 server_event: Remove needless scoping and indentation Kim Alvefur <zash@zash.se> parents: 6850 diff changeset	702 local function wrapclient( client, ip, port, listeners, pattern, sslctx )
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	703 local interface = handleclient( client, ip, port, nil, pattern, listeners, sslctx )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	704 interface:_start_connection(sslctx)
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	705 return interface, client
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	706 --function handleclient( client, ip, port, server, pattern, listener, _, sslctx ) -- creates an client interface
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	707 end
6851 1f1bed8ebc41 server_event: Remove needless scoping and indentation Kim Alvefur <zash@zash.se> parents: 6850 diff changeset	708
1f1bed8ebc41 server_event: Remove needless scoping and indentation Kim Alvefur <zash@zash.se> parents: 6850 diff changeset	709 local function addclient( addr, serverport, listener, pattern, sslctx, typ )
1f1bed8ebc41 server_event: Remove needless scoping and indentation Kim Alvefur <zash@zash.se> parents: 6850 diff changeset	710 if sslctx and not has_luasec then
1f1bed8ebc41 server_event: Remove needless scoping and indentation Kim Alvefur <zash@zash.se> parents: 6850 diff changeset	711 debug "need luasec, but not available"
1f1bed8ebc41 server_event: Remove needless scoping and indentation Kim Alvefur <zash@zash.se> parents: 6850 diff changeset	712 return nil, "luasec not found"
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	713 end
6851 1f1bed8ebc41 server_event: Remove needless scoping and indentation Kim Alvefur <zash@zash.se> parents: 6850 diff changeset	714 if not typ then
1f1bed8ebc41 server_event: Remove needless scoping and indentation Kim Alvefur <zash@zash.se> parents: 6850 diff changeset	715 local addrinfo, err = getaddrinfo(addr)
1f1bed8ebc41 server_event: Remove needless scoping and indentation Kim Alvefur <zash@zash.se> parents: 6850 diff changeset	716 if not addrinfo then return nil, err end
1f1bed8ebc41 server_event: Remove needless scoping and indentation Kim Alvefur <zash@zash.se> parents: 6850 diff changeset	717 if addrinfo[1] and addrinfo[1].family == "inet6" then
1f1bed8ebc41 server_event: Remove needless scoping and indentation Kim Alvefur <zash@zash.se> parents: 6850 diff changeset	718 typ = "tcp6"
1f1bed8ebc41 server_event: Remove needless scoping and indentation Kim Alvefur <zash@zash.se> parents: 6850 diff changeset	719 else
1f1bed8ebc41 server_event: Remove needless scoping and indentation Kim Alvefur <zash@zash.se> parents: 6850 diff changeset	720 typ = "tcp"
1f1bed8ebc41 server_event: Remove needless scoping and indentation Kim Alvefur <zash@zash.se> parents: 6850 diff changeset	721 end
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	722 end
6851 1f1bed8ebc41 server_event: Remove needless scoping and indentation Kim Alvefur <zash@zash.se> parents: 6850 diff changeset	723 local create = socket[typ]
1f1bed8ebc41 server_event: Remove needless scoping and indentation Kim Alvefur <zash@zash.se> parents: 6850 diff changeset	724 if type( create ) ~= "function" then
1f1bed8ebc41 server_event: Remove needless scoping and indentation Kim Alvefur <zash@zash.se> parents: 6850 diff changeset	725 return nil, "invalid socket type"
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	726 end
6851 1f1bed8ebc41 server_event: Remove needless scoping and indentation Kim Alvefur <zash@zash.se> parents: 6850 diff changeset	727 local client, err = create() -- creating new socket
1f1bed8ebc41 server_event: Remove needless scoping and indentation Kim Alvefur <zash@zash.se> parents: 6850 diff changeset	728 if not client then
1f1bed8ebc41 server_event: Remove needless scoping and indentation Kim Alvefur <zash@zash.se> parents: 6850 diff changeset	729 debug( "cannot create socket:", err )
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	730 return nil, err
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	731 end
6851 1f1bed8ebc41 server_event: Remove needless scoping and indentation Kim Alvefur <zash@zash.se> parents: 6850 diff changeset	732 client:settimeout( 0 ) -- set nonblocking
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	733 local res, err = client:connect( addr, serverport ) -- connect
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	734 if res or ( err == "timeout" ) then
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	735 local ip, port = client:getsockname( )
6851 1f1bed8ebc41 server_event: Remove needless scoping and indentation Kim Alvefur <zash@zash.se> parents: 6850 diff changeset	736 local interface = wrapclient( client, ip, serverport, listener, pattern, sslctx )
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	737 debug( "new connection id:", interface.id )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	738 return interface, err
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	739 else
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	740 debug( "new connection failed:", err )
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	741 return nil, err
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	742 end
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	743 end
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	744
6851 1f1bed8ebc41 server_event: Remove needless scoping and indentation Kim Alvefur <zash@zash.se> parents: 6850 diff changeset	745 local function loop( ) -- starts the event loop
2333 ed824115becf net.server_event: Return "quitting" on loop exit to be compatible with net.server_select, and give Prosody a way to know why the loop broke Matthew Wild <mwild1@gmail.com> parents: 2318 diff changeset	746 base:loop( )
ed824115becf net.server_event: Return "quitting" on loop exit to be compatible with net.server_select, and give Prosody a way to know why the loop broke Matthew Wild <mwild1@gmail.com> parents: 2318 diff changeset	747 return "quitting";
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	748 end
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	749
6850 41de00647ad3 server_event: Move local references to various functions to top of file Kim Alvefur <zash@zash.se> parents: 6849 diff changeset	750 local function newevent( ... )
41de00647ad3 server_event: Move local references to various functions to top of file Kim Alvefur <zash@zash.se> parents: 6849 diff changeset	751 return addevent( base, ... )
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	752 end
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	753
6851 1f1bed8ebc41 server_event: Remove needless scoping and indentation Kim Alvefur <zash@zash.se> parents: 6850 diff changeset	754 local function closeallservers ( arg )
6849 0455b9686e16 server_event: Replace implementation of ordered set with a simple table set Kim Alvefur <zash@zash.se> parents: 6819 diff changeset	755 for item in pairs( interfacelist ) do
2334 4cedfeeaaed4 net.server_event: Fix closeallservers() to work correctly after the giant refactorizification Matthew Wild <mwild1@gmail.com> parents: 2333 diff changeset	756 if item.type == "server" then
4cedfeeaaed4 net.server_event: Fix closeallservers() to work correctly after the giant refactorizification Matthew Wild <mwild1@gmail.com> parents: 2333 diff changeset	757 item:close( arg )
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	758 end
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	759 end
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	760 end
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	761
2335 060fb67cefd5 net.server_event: Add setquitting() function to be compatible with net.server_select Matthew Wild <mwild1@gmail.com> parents: 2334 diff changeset	762 local function setquitting(yes)
060fb67cefd5 net.server_event: Add setquitting() function to be compatible with net.server_select Matthew Wild <mwild1@gmail.com> parents: 2334 diff changeset	763 if yes then
7336 0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	764 -- Quit now
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	765 closeallservers();
0d9ac472e58c server_event: Fix indentation (again) Kim Alvefur <zash@zash.se> parents: 7224 diff changeset	766 base:loopexit();
2335 060fb67cefd5 net.server_event: Add setquitting() function to be compatible with net.server_select Matthew Wild <mwild1@gmail.com> parents: 2334 diff changeset	767 end
060fb67cefd5 net.server_event: Add setquitting() function to be compatible with net.server_select Matthew Wild <mwild1@gmail.com> parents: 2334 diff changeset	768 end
060fb67cefd5 net.server_event: Add setquitting() function to be compatible with net.server_select Matthew Wild <mwild1@gmail.com> parents: 2334 diff changeset	769
4399 1b4161970842 net.server_event, net.server_select: Fixed some global pollution. Waqas Hussain <waqas20@gmail.com> parents: 4377 diff changeset	770 local function get_backend()
2340 b286d60cb4fc net.server_event: Add get_backend() method Matthew Wild <mwild1@gmail.com> parents: 2336 diff changeset	771 return base:method();
b286d60cb4fc net.server_event: Add get_backend() method Matthew Wild <mwild1@gmail.com> parents: 2336 diff changeset	772 end
b286d60cb4fc net.server_event: Add get_backend() method Matthew Wild <mwild1@gmail.com> parents: 2336 diff changeset	773
2434 4403d4473783 net.server_event: Add server.hook_signal(signal_num, handler) to wrap libevent's signal-catching interface Matthew Wild <mwild1@gmail.com> parents: 2375 diff changeset	774 -- We need to hold onto the events to stop them
4403d4473783 net.server_event: Add server.hook_signal(signal_num, handler) to wrap libevent's signal-catching interface Matthew Wild <mwild1@gmail.com> parents: 2375 diff changeset	775 -- being garbage-collected
4403d4473783 net.server_event: Add server.hook_signal(signal_num, handler) to wrap libevent's signal-catching interface Matthew Wild <mwild1@gmail.com> parents: 2375 diff changeset	776 local signal_events = {}; -- [signal_num] -> event object
4399 1b4161970842 net.server_event, net.server_select: Fixed some global pollution. Waqas Hussain <waqas20@gmail.com> parents: 4377 diff changeset	777 local function hook_signal(signal_num, handler)
6856 489f4ae291bf server_event: Remove unused argument [luacheck] Kim Alvefur <zash@zash.se> parents: 6855 diff changeset	778 local function _handler()
2434 4403d4473783 net.server_event: Add server.hook_signal(signal_num, handler) to wrap libevent's signal-catching interface Matthew Wild <mwild1@gmail.com> parents: 2375 diff changeset	779 local ret = handler();
4403d4473783 net.server_event: Add server.hook_signal(signal_num, handler) to wrap libevent's signal-catching interface Matthew Wild <mwild1@gmail.com> parents: 2375 diff changeset	780 if ret ~= false then -- Continue handling this signal?
4403d4473783 net.server_event: Add server.hook_signal(signal_num, handler) to wrap libevent's signal-catching interface Matthew Wild <mwild1@gmail.com> parents: 2375 diff changeset	781 return EV_SIGNAL; -- Yes
4403d4473783 net.server_event: Add server.hook_signal(signal_num, handler) to wrap libevent's signal-catching interface Matthew Wild <mwild1@gmail.com> parents: 2375 diff changeset	782 end
4403d4473783 net.server_event: Add server.hook_signal(signal_num, handler) to wrap libevent's signal-catching interface Matthew Wild <mwild1@gmail.com> parents: 2375 diff changeset	783 return -1; -- Close this event
4403d4473783 net.server_event: Add server.hook_signal(signal_num, handler) to wrap libevent's signal-catching interface Matthew Wild <mwild1@gmail.com> parents: 2375 diff changeset	784 end
4403d4473783 net.server_event: Add server.hook_signal(signal_num, handler) to wrap libevent's signal-catching interface Matthew Wild <mwild1@gmail.com> parents: 2375 diff changeset	785 signal_events[signal_num] = base:addevent(signal_num, EV_SIGNAL, _handler);
4403d4473783 net.server_event: Add server.hook_signal(signal_num, handler) to wrap libevent's signal-catching interface Matthew Wild <mwild1@gmail.com> parents: 2375 diff changeset	786 return signal_events[signal_num];
4403d4473783 net.server_event: Add server.hook_signal(signal_num, handler) to wrap libevent's signal-catching interface Matthew Wild <mwild1@gmail.com> parents: 2375 diff changeset	787 end
4403d4473783 net.server_event: Add server.hook_signal(signal_num, handler) to wrap libevent's signal-catching interface Matthew Wild <mwild1@gmail.com> parents: 2375 diff changeset	788
3003 48871a549b4c net.server_select, net.server_event: Add server.link() to link 2 connections with an intermediate buffer of the specified size Matthew Wild <mwild1@gmail.com> parents: 3002 diff changeset	789 local function link(sender, receiver, buffersize)
48871a549b4c net.server_select, net.server_event: Add server.link() to link 2 connections with an intermediate buffer of the specified size Matthew Wild <mwild1@gmail.com> parents: 3002 diff changeset	790 local sender_locked;
5776 bd0ff8ae98a8 Remove all trailing whitespace Florian Zeitz <florob@babelmonkeys.de> parents: 5756 diff changeset	791
3003 48871a549b4c net.server_select, net.server_event: Add server.link() to link 2 connections with an intermediate buffer of the specified size Matthew Wild <mwild1@gmail.com> parents: 3002 diff changeset	792 function receiver:ondrain()
48871a549b4c net.server_select, net.server_event: Add server.link() to link 2 connections with an intermediate buffer of the specified size Matthew Wild <mwild1@gmail.com> parents: 3002 diff changeset	793 if sender_locked then
48871a549b4c net.server_select, net.server_event: Add server.link() to link 2 connections with an intermediate buffer of the specified size Matthew Wild <mwild1@gmail.com> parents: 3002 diff changeset	794 sender:resume();
48871a549b4c net.server_select, net.server_event: Add server.link() to link 2 connections with an intermediate buffer of the specified size Matthew Wild <mwild1@gmail.com> parents: 3002 diff changeset	795 sender_locked = nil;
48871a549b4c net.server_select, net.server_event: Add server.link() to link 2 connections with an intermediate buffer of the specified size Matthew Wild <mwild1@gmail.com> parents: 3002 diff changeset	796 end
48871a549b4c net.server_select, net.server_event: Add server.link() to link 2 connections with an intermediate buffer of the specified size Matthew Wild <mwild1@gmail.com> parents: 3002 diff changeset	797 end
5776 bd0ff8ae98a8 Remove all trailing whitespace Florian Zeitz <florob@babelmonkeys.de> parents: 5756 diff changeset	798
3003 48871a549b4c net.server_select, net.server_event: Add server.link() to link 2 connections with an intermediate buffer of the specified size Matthew Wild <mwild1@gmail.com> parents: 3002 diff changeset	799 function sender:onincoming(data)
48871a549b4c net.server_select, net.server_event: Add server.link() to link 2 connections with an intermediate buffer of the specified size Matthew Wild <mwild1@gmail.com> parents: 3002 diff changeset	800 receiver:write(data);
48871a549b4c net.server_select, net.server_event: Add server.link() to link 2 connections with an intermediate buffer of the specified size Matthew Wild <mwild1@gmail.com> parents: 3002 diff changeset	801 if receiver.writebufferlen >= buffersize then
48871a549b4c net.server_select, net.server_event: Add server.link() to link 2 connections with an intermediate buffer of the specified size Matthew Wild <mwild1@gmail.com> parents: 3002 diff changeset	802 sender_locked = true;
48871a549b4c net.server_select, net.server_event: Add server.link() to link 2 connections with an intermediate buffer of the specified size Matthew Wild <mwild1@gmail.com> parents: 3002 diff changeset	803 sender:pause();
48871a549b4c net.server_select, net.server_event: Add server.link() to link 2 connections with an intermediate buffer of the specified size Matthew Wild <mwild1@gmail.com> parents: 3002 diff changeset	804 end
48871a549b4c net.server_select, net.server_event: Add server.link() to link 2 connections with an intermediate buffer of the specified size Matthew Wild <mwild1@gmail.com> parents: 3002 diff changeset	805 end
6314 8029b8a5f37a net.server_select/event: Switch sender mode to a when reading, to make sure we get all available data Matthew Wild <mwild1@gmail.com>* parents: 6047 diff changeset	806 sender:set_mode("*a");
3003 48871a549b4c net.server_select, net.server_event: Add server.link() to link 2 connections with an intermediate buffer of the specified size Matthew Wild <mwild1@gmail.com> parents: 3002 diff changeset	807 end
48871a549b4c net.server_select, net.server_event: Add server.link() to link 2 connections with an intermediate buffer of the specified size Matthew Wild <mwild1@gmail.com> parents: 3002 diff changeset	808
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	809 return {
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	810 cfg = cfg,
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	811 base = base,
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	812 loop = loop,
3003 48871a549b4c net.server_select, net.server_event: Add server.link() to link 2 connections with an intermediate buffer of the specified size Matthew Wild <mwild1@gmail.com> parents: 3002 diff changeset	813 link = link,
6853 a2cc477eb19b server_event: Rename luaevent module variable to avoid name clash [luacheck] Kim Alvefur <zash@zash.se> parents: 6852 diff changeset	814 event = levent,
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	815 event_base = base,
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	816 addevent = newevent,
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	817 addserver = addserver,
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	818 addclient = addclient,
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	819 wrapclient = wrapclient,
2336 fb1d9d2444df Rename closeallservers to closeall in the public API (same as net.server_select) and expose setquitting Matthew Wild <mwild1@gmail.com> parents: 2335 diff changeset	820 setquitting = setquitting,
fb1d9d2444df Rename closeallservers to closeall in the public API (same as net.server_select) and expose setquitting Matthew Wild <mwild1@gmail.com> parents: 2335 diff changeset	821 closeall = closeallservers,
2340 b286d60cb4fc net.server_event: Add get_backend() method Matthew Wild <mwild1@gmail.com> parents: 2336 diff changeset	822 get_backend = get_backend,
2434 4403d4473783 net.server_event: Add server.hook_signal(signal_num, handler) to wrap libevent's signal-catching interface Matthew Wild <mwild1@gmail.com> parents: 2375 diff changeset	823 hook_signal = hook_signal,
2318 a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	824
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	825 __NAME = SCRIPT_NAME,
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	826 __DATE = LAST_MODIFIED,
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	827 __AUTHOR = SCRIPT_AUTHOR,
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	828 __VERSION = SCRIPT_VERSION,
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	829
a831de056de3 net.server_select, net.server_event: Convert to unix line endings from Windows Matthew Wild <mwild1@gmail.com> parents: 2303 diff changeset	830 }

prosody

495de404a8ae

net/server_event.lua

Software / code / prosody

Annotate

net/server_event.lua @ 7567:495de404a8ae

Software `/` code `/` prosody