Two performance fixes for HTTP Servers. The first reduces latency from 0.25s to near-zero for back-to-back connections by using the Queue class as a notify system. The second switches the slow socket.get for the much faster socket.read

git-svn-id: file:///home/svn/framework3/trunk@12615 4d416f70-5f16-0410-b530-b9f4589650da

lib/rex/io/stream_server.rb

@@ -1,3 +1,5 @@
+require 'thread'
+
 module Rex
 module IO
 
@@ -61,6 +63,7 @@ module StreamServer
 	#
 	def start
 		self.clients = []
+		self.client_waiter = ::Queue.new
 
 		self.listener_thread = Rex::ThreadFactory.spawn("StreamServerListener", false) {
 			monitor_listener
@@ -128,6 +131,7 @@ module StreamServer
 
 	attr_accessor :clients # :nodoc:
 	attr_accessor :listener_thread, :clients_thread # :nodoc:
+	attr_accessor :client_waiter
 
 protected
 
@@ -152,6 +156,9 @@ protected
 				# Initialize the connection processing
 				on_client_connect(cli)
 				
+				# Notify the client monitor
+				self.client_waiter.push(cli)
+
 			# Skip exceptions caused by accept() [ SSL ]
 			rescue ::EOFError, ::Errno::ECONNRESET, ::Errno::ENOTCONN, ::Errno::ECONNABORTED
 			rescue ::Interrupt
@@ -170,12 +177,14 @@ protected
 	#
 	def monitor_clients
 		begin
+		
+			# Wait for a notify if our client list is empty
 			if (clients.length == 0)
-				Rex::ThreadSafe::sleep(0.25)
+				self.client_waiter.pop
 				next
 			end
 
-			sd = Rex::ThreadSafe.select(clients)
+			sd = Rex::ThreadSafe.select(clients, nil, nil, nil)
 
 			sd[0].each { |cfd|
 				begin

lib/rex/proto/http/server.rb

@@ -278,7 +278,7 @@ protected
 			#
 			# XXX: Handle ParseCode::Partial
 			#
-			data = cli.get
+			data = cli.read(65535)
 
 			raise ::EOFError if not data
 			raise ::EOFError if data.empty?