github/metasploit-framework
1
Fork 0
mirror of https://github.com/rapid7/metasploit-framework synced 2024-11-12 11:52:01 +01:00
Code Releases Activity

Dispatch incoming requests as threads

Browse Source 
git-svn-id: file:///home/svn/framework3/trunk@10579 4d416f70-5f16-0410-b530-b9f4589650da
This commit is contained in:
HD Moore 2010-10-06 23:20:13 +00:00
parent cb453a97ab
commit 352b4cf1c8
1 changed files with 16 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
lib/msf/core/rpc/service.rb
View File

@ -8,8 +8,12 @@ module RPC
class Service < ::XMLRPC::BasicServer class Service < ::XMLRPC::BasicServer
attr_accessor :service, :state, :on_input, :on_output, :on_error attr_accessor :service, :state, :on_input, :on_output, :on_error
attr_accessor :dispatcher_timeout
def initialize(srvhost, srvport, ssl=false, cert=nil, ckey=nil) def initialize(srvhost, srvport, ssl=false, cert=nil, ckey=nil)
self.dispatcher_timeout = 0
self.service = Rex::Socket::TcpServer.create( self.service = Rex::Socket::TcpServer.create(
'LocalHost' => srvhost, 'LocalHost' => srvhost,
'LocalPort' => srvport, 'LocalPort' => srvport,
@ -20,7 +24,18 @@ class Service < ::XMLRPC::BasicServer
on_client_connect(client) on_client_connect(client)
} }
self.service.on_client_data_proc = Proc.new { |client| self.service.on_client_data_proc = Proc.new { |client|
on_client_data(client) Thread.new(client) do |client_copy|
begin
Timeout.timeout(self.dispatcher_timeout) do
on_client_data(client)
end
rescue ::EOFError => e
raise e
rescue ::Exception => e
wlog("XMLRPC Server Error: #{client.inspect} #{e.class} #{e} #{e.backtrace}")
raise e
end
end
} }
self.service.on_client_close_proc = Proc.new { |client| self.service.on_client_close_proc = Proc.new { |client|
on_client_close(client) on_client_close(client)