Software `/` code `/` prosody

Comparison

net/server_event.lua @ 10411:db2a06b9ff98

Merge 0.11->trunk

author	Kim Alvefur <zash@zash.se>
date	Sat, 16 Nov 2019 16:52:31 +0100
parent	10407:bd8fbee04a2f
child	10547:c77471c180ea

comparison

equal deleted inserted replaced

-:659b577f280c
+:db2a06b9ff98
 		end
 		self:_close()
 		debug( "fatal error while ssl wrapping:", err )
 		return false
 	end
+	if self.conn.sni then
+		if self.servername then
+			self.conn:sni(self.servername);
+		elseif self._server and type(self._server.hosts) == "table" and next(self._server.hosts) ~= nil then
+			self.conn:sni(self._server.hosts, true);
+		end
+	end
 	self.conn:settimeout( 0 )  -- set non blocking
 	local handshakecallback = coroutine_wrap(function( event )
 		local _, err
 		local attempt = 0
 		local maxattempt = cfg.MAX_HANDSHAKE_ATTEMPTS
 	return nointerface, noreading, nowriting
 end
 --TODO: Deprecate
 function interface_mt:lock_read(switch)
+	log("warn", ":lock_read is deprecated, use :pasue() and :resume()");
 	if switch then
 		return self:pause();
 	else
 		return self:resume();
 	end
 		self.eventread = addevent( base, self.conn, EV_READ, self.readcallback, cfg.READ_TIMEOUT );  -- register callback
 		return true;
 	end
 end
+function interface_mt:pause_writes()
+	return self:_lock(self.nointerface, self.noreading, true);
+end
+function interface_mt:resume_writes()
+	self:_lock(self.nointerface, self.noreading, false);
+	if self.writecallback and not self.eventwrite then
+		self.eventwrite = addevent( base, self.conn, EV_WRITE, self.writecallback, cfg.WRITE_TIMEOUT );  -- register callback
+		return true;
+	end
+end
 function interface_mt:counter(c)
 	if c then
 		self._connections = self._connections + c
 	end
 	return self._connections
 end
 -- Public methods
 function interface_mt:write(data)
-	if self.nowriting then return nil, "locked" end
+	if self.nointerface then return nil, "locked"; end
 	--vdebug( "try to send data to client, id/data:", self.id, data )
 	data = tostring( data )
 	local len = #data
 	local total = len + self.writebufferlen
 	if total > cfg.MAX_SEND_LENGTH then  -- check buffer length
 		debug( "error:", err )  -- to much, check your app
 		return nil, err
 	end
 	t_insert(self.writebuffer, data) -- new buffer
 	self.writebufferlen = total
-	if not self.eventwrite then  -- register new write event
+	if not self.eventwrite and not self.nowriting  then  -- register new write event
 		--vdebug( "register new write event" )
 		self.eventwrite = addevent( base, self.conn, EV_WRITE, self.writecallback, cfg.WRITE_TIMEOUT )
 	end
 	return true
 end
 function interface_mt:ondisconnect()
 end
 function interface_mt:ontimeout()
 end
 function interface_mt:onreadtimeout()
-	self.fatalerror = "timeout during receiving"
-	debug( "connection failed:", self.fatalerror )
-	self:_close()
-	self.eventread = nil
 end
 function interface_mt:ondrain()
 end
 function interface_mt:ondetach()
 end
 function interface_mt:onstatus()
 end
 -- End of client interface methods
-local function handleclient( client, ip, port, server, pattern, listener, sslctx )  -- creates an client interface
+local function handleclient( client, ip, port, server, pattern, listener, sslctx, extra )  -- creates an client interface
 	--vdebug("creating client interfacce...")
 	local interface = {
 		type = "client";
 		conn = client;
 		currenttime = socket_gettime( );  -- safe the origin
 		-- Properties
 		_ip = ip, _port = port, _server = server, _pattern = pattern,
 		_serverport = (server and server:port() or nil),
 		_sslctx = sslctx; -- parameters
 		_usingssl = false;  -- client is using ssl;
+		extra = extra;
+		servername = extra and extra.servername;
 	}
 	if not has_luasec then interface.starttls = false; end
 	interface.id = tostring(interface):match("%x+$");
 	interface.writecallback = function( event )  -- called on write events
 		--vdebug( "new client write event, id/ip/port:", interface, ip, port )
 	setmetatable(interface, interface_mt)
 	interfacelist[ interface ] = true  -- add to interfacelist
 	return interface
 end
-local function handleserver( server, addr, port, pattern, listener, sslctx )  -- creates an server interface
+local function handleserver( server, addr, port, pattern, listener, sslctx, startssl )  -- creates a server interface
 	debug "creating server interface..."
 	local interface = {
 		_connections = 0;
 		type = "server";
 		fatalerror = false; -- error message
 		nointerface = true;  -- lock/unlock parameter
 		_ip = addr, _port = port, _pattern = pattern,
 		_sslctx = sslctx;
+		hosts = {};
 	}
 	interface.id = tostring(interface):match("%x+$");
 	interface.readcallback = function( event )  -- server handler, called on incoming connections
 		--vdebug( "server can accept, id/addr/port:", interface, addr, port )
 		if interface.fatalerror then
 			end
 			local client_ip, client_port = client:getpeername( )
 			interface._connections = interface._connections + 1  -- increase connection count
 			local clientinterface = handleclient( client, client_ip, client_port, interface, pattern, listener, sslctx )
 			--vdebug( "client id:", clientinterface, "startssl:", startssl )
-			if has_luasec and sslctx then
+			if has_luasec and startssl then
 				clientinterface:starttls(sslctx, true)
 			else
 				clientinterface:_start_session( true )
 			end
 			debug( "accepted incoming client connection from:", client_ip or "<unknown IP>", client_port or "<unknown port>", "to", port or "<unknown port>");
 	interfacelist[ interface ] = true
 	interface:_start_session()
 	return interface
 end
-local function addserver( addr, port, listener, pattern, sslctx, startssl )  -- TODO: check arguments
+local function listen(addr, port, listener, config)
-	--vdebug( "creating new tcp server with following parameters:", addr or "nil", port or "nil", sslctx or "nil", startssl or "nil")
+	config = config or {}
-	if sslctx and not has_luasec then
+	if config.sslctx and not has_luasec then
 		debug "fatal error: luasec not found"
 		return nil, "luasec not found"
 	end
 	local server, err = socket.bind( addr, port, cfg.ACCEPT_QUEUE )  -- create server socket
 	if not server then
 		debug( "creating server socket on "..addr.." port "..port.." failed:", err )
 		return nil, err
 	end
-	local interface = handleserver( server, addr, port, pattern, listener, sslctx, startssl )  -- new server handler
+	local interface = handleserver( server, addr, port, config.read_size, listener, config.tls_ctx, config.tls_direct)  -- new server handler
 	debug( "new server created with id:", tostring(interface))
 	return interface
 end
-local function wrapclient( client, ip, port, listeners, pattern, sslctx )
+local function addserver( addr, port, listener, pattern, sslctx )  -- TODO: check arguments
-	local interface = handleclient( client, ip, port, nil, pattern, listeners, sslctx )
+	--vdebug( "creating new tcp server with following parameters:", addr or "nil", port or "nil", sslctx or "nil", startssl or "nil")
+	return listen( addr, port, listener, {
+		read_size = pattern,
+		tls_ctx = sslctx,
+		tls_direct = not not sslctx,
+	});
+end
+local function wrapclient( client, ip, port, listeners, pattern, sslctx, extra )
+	local interface = handleclient( client, ip, port, nil, pattern, listeners, sslctx, extra )
 	interface:_start_connection(sslctx)
 	return interface, client
 	--function handleclient( client, ip, port, server, pattern, listener, _, sslctx )  -- creates an client interface
 end
-local function addclient( addr, serverport, listener, pattern, sslctx, typ )
+local function addclient( addr, serverport, listener, pattern, sslctx, typ, extra )
 	if sslctx and not has_luasec then
 		debug "need luasec, but not available"
 		return nil, "luasec not found"
 	end
 	if not typ then
 	end
 	client:settimeout( 0 )  -- set nonblocking
 	local res, err = client:setpeername( addr, serverport )  -- connect
 	if res or ( err == "timeout" ) then
 		local ip, port = client:getsockname( )
-		local interface = wrapclient( client, ip, serverport, listener, pattern, sslctx )
+		local interface = wrapclient( client, ip, serverport, listener, pattern, sslctx, extra )
 		debug( "new connection id:", interface.id )
 		return interface, err
 	else
 		debug( "new connection failed:", err )
 		return nil, err
 	link = link,
 	event = levent,
 	event_base = base,
 	addevent = newevent,
 	addserver = addserver,
+	listen = listen,
 	addclient = addclient,
 	wrapclient = wrapclient,
 	setquitting = setquitting,
 	closeall = closeallservers,
 	get_backend = get_backend,

prosody

db2a06b9ff98

net/server_event.lua

Software / code / prosody

Comparison

net/server_event.lua @ 10411:db2a06b9ff98

Software `/` code `/` prosody