prosody-hg: net/server_event.lua comparison

comparison net/server_event.lua @ 2535:75bb31b6c53f

net.server_event: Trailing whitespace

author	Matthew Wild <mwild1@gmail.com>
date	Fri, 29 Jan 2010 16:49:43 +0000
parents	1f46a9d9b16a
children	e7bec4072c61

comparison

equal deleted inserted replaced

-:1f46a9d9b16a
+:75bb31b6c53f
 	MAX_SEND_LENGTH       = 1024 * 1024 * 1024 * 1024,  -- max bytes size of write buffer (for writing on sockets)
 	ACCEPT_DELAY          = 10,  -- seconds to wait until the next attemp of a full server to accept
 	READ_TIMEOUT          = 60 * 30,  -- timeout in seconds for read data from socket
 	WRITE_TIMEOUT         = 30,  -- timeout in seconds for write data on socket
 	CONNECT_TIMEOUT       = 10,  -- timeout in seconds for connection attemps
 	CLEAR_DELAY           = 5,  -- seconds to wait for clearing interface list (and calling ondisconnect listeners)
 	DEBUG                 = true,  -- show debug messages
 }
 local function use(x) return rawget(_G, x); end
 local print = use "print"
 local bitor = ( function( ) -- thx Rici Lake
 	local hasbit = function( x, p )
 		return x % ( p + p ) >= p
 	end
 	return function( x, y )
 		local p = 1
 		local z = 0
 		local limit = x > y and x or y
 		while p <= limit do
 			if hasbit( x, p ) or hasbit( y, p ) then
 				z = z + p
 			end
 			p = p + p
 		end
 				interface:_position( position )  -- set new position in array
 			else  -- free last position
 				array[ len ] = nil
 			end
 			len = len - 1
 			return len
 		else
 			return array
 		end
 	end
 end )( )
 	end
 	function interface_mt:_start_session(onconnect) -- new session, for example after startssl
 		if self.type == "client" then
 			local callback = function( )
 				self:_lock( false,  false, false )
 				--vdebug( "start listening on client socket with id:", self.id )
 				self.eventread = addevent( base, self.conn, EV_READ, self.readcallback, cfg.READ_TIMEOUT )  -- register callback
 				self:onconnect()
 				self.eventsession = nil
 				return -1
 			end
 								self:_start_session( onsomething )
 								debug( "ssl handshake done" )
 								self.eventhandshake = nil
 								return -1
 							end
 							debug( "error during ssl handshake:", err )
 							if err == "wantwrite" then
 								event = EV_WRITE
 							elseif err == "wantread" then
 								event = EV_READ
 							else
 								self.fatalerror = err
 							end
 						end
 						if self.fatalerror then
 							if "onconnect" == arg then
 								self.ondisconnect = nil  -- dont call this when client isnt really connected
 							end
 						event = coroutine_yield( event, cfg.HANDSHAKE_TIMEOUT )  -- yield this monster...
 					end
 				end
 			)
 			debug "starting handshake..."
 			self:_lock( false, true, true )  -- unlock read/write events, but keep interface locked
 			self.eventhandshake = addevent( base, self.conn, EV_READWRITE, handshakecallback, cfg.HANDSHAKE_TIMEOUT )
 			return true
 	end
 	function interface_mt:_destroy()  -- close this interface + events and call last listener
 			debug( "closing client with id:", self.id )
 		local total = len + self.writebufferlen
 		if total > cfg.MAX_SEND_LENGTH then  -- check buffer length
 			local err = "send buffer exceeded"
 			debug( "error:", err )  -- to much, check your app
 			return nil, err
 		end
 		self.writebuffer = self.writebuffer .. data -- new buffer
 		self.writebufferlen = total
 		if not self.eventwrite then  -- register new write event
 			--vdebug( "register new write event" )
 			self.eventwrite = addevent( base, self.conn, EV_WRITE, self.writecallback, cfg.WRITE_TIMEOUT )
 		elseif self._usingssl then  -- startssl was already called
 			err = "ssl already active"
 		end
 		if err then
 			debug( "error:", err )
 			return nil, err
 		end
 		self._usingssl = true
 		self.startsslcallback = function( )  -- we have to start the handshake outside of a read/write event
 			self.startsslcallback = nil
 			self:_start_ssl();
 	end
 	function interface_mt:ondisconnect()
 	end
 	function interface_mt:ontimeout()
 	end
 end
 -- End of client interface methods
 local handleclient;
 do
 				interface.eventwrite = false
 				return -1
 			end
 			if EV_TIMEOUT == event then  -- took too long to write some data to socket -> disconnect
 				interface.fatalerror = "timeout during writing"
 				debug( "writing failed:", interface.fatalerror )
 				interface:_close()
 				interface.eventwrite = false
 				return -1
 			else  -- can write :)
 				if interface._usingssl then  -- handle luasec
 					interface.eventwrite = nil
 					return -1
 				elseif byte then  -- want write again
 					--vdebug( "writebuffer is not empty:", err )
 					interface.writebuffer = string_sub( interface.writebuffer, byte + 1, interface.writebufferlen )  -- new buffer
 					interface.writebufferlen = interface.writebufferlen - byte
 					if "wantread" == err then  -- happens only with luasec
 						local callback = function( )
 							interface:_close()
 							interface.eventwritetimeout = nil
 							return evreturn, evtimeout
 						interface.eventwritetimeout = addevent( base, nil, EV_TIMEOUT, callback, cfg.WRITE_TIMEOUT )  -- reg a new timeout event
 						debug( "wantread during write attemp, reg it in readcallback but dont know what really happens next..." )
 						-- hopefully this works with luasec; its simply not possible to use 2 different write events on a socket in luaevent
 						return -1
 					end
 					return EV_WRITE, cfg.WRITE_TIMEOUT
 				else  -- connection was closed during writing or fatal error
 					interface.fatalerror = err or "fatal error"
 					debug( "connection failed in write event:", interface.fatalerror )
 					interface:_close()
 					interface.eventwrite = nil
 					return -1
 				end
 			end
 				interface.eventread = nil
 				return -1
 			end
 			if EV_TIMEOUT == event then  -- took too long to get some data from client -> disconnect
 				interface.fatalerror = "timeout during receiving"
 				debug( "connection failed:", interface.fatalerror )
 				interface:_close()
 				interface.eventread = nil
 				return -1
 			else -- can read
 				if interface._usingssl then  -- handle luasec
 						interface.eventreadtimeout:close( )
 						interface.eventreadtimeout = nil
 					end
 				end
 				local buffer, err, part = interface.conn:receive( pattern )  -- receive buffer with "pattern"
 				--vdebug( "read data:", tostring(buffer), "error:", tostring(err), "part:", tostring(part) )
 				buffer = buffer or part or ""
 				local len = string_len( buffer )
 				if len > cfg.MAX_READ_LENGTH then  -- check buffer length
 					interface.fatalerror = "receive buffer exceeded"
 					debug( "fatal error:", interface.fatalerror )
 						end
 						interface.eventreadtimeout = addevent( base, nil, EV_TIMEOUT,
 							function( )
 								interface:_close()
 							end, cfg.READ_TIMEOUT
 						)
 						debug( "wantwrite during read attemp, reg it in writecallback but dont know what really happens next..." )
 						-- to be honest i dont know what happens next, if it is allowed to first read, the write etc...
 					else  -- connection was closed or fatal error
 						interface.fatalerror = err
 						debug( "connection failed in read event:", interface.fatalerror )
 						interface:_close()
 						interface.eventread = nil
 						return -1
 					end
 				end
 			sslctx, err = ssl.newcontext( sslcfg )
 			if err then
 				debug( "error while creating new ssl context for server socket:", err )
 				return nil, err
 			end
 		end
 		local interface = handleserver( server, addr, port, pattern, listener, sslctx, startssl )  -- new server handler
 		debug( "new server created with id:", tostring(interface))
 		return interface
 	end
 end )( )
 	end
 	function addclient( addr, serverport, listener, pattern, localaddr, localport, sslcfg, startssl )
 		local client, err = socket.tcp()  -- creating new socket
 		if not client then
 			debug( "cannot create socket:", err )
 			return nil, err
 		end
 		client:settimeout( 0 )  -- set nonblocking
 		if localaddr then
 			local res, err = client:bind( localaddr, localport, -1 )
 			end
 		end
 		local sslctx
 		if sslcfg then  -- handle ssl/new context
 			if not ssl then
 				debug "need luasec, but not available"
 				return nil, "luasec not found"
 			end
 			sslctx, err = ssl.newcontext( sslcfg )
 			if err then
 				debug( "cannot create new ssl context:", err )

Mercurial > prosody-hg

comparison net/server_event.lua @ 2535:75bb31b6c53f