lib/metasploit/aggregator/http/responder.rb in metasploit-aggregator-0.1.1 vs lib/metasploit/aggregator/http/responder.rb in metasploit-aggregator-0.1.2

@@ -1,5 +1,6 @@
+require "metasploit/aggregator/session_detail_service"
 require "metasploit/aggregator/http/request"
 
 module Metasploit
   module Aggregator
     module Http
@@ -17,10 +18,11 @@
           @uri = uri
           @queue = Queue.new
           @thread = Thread.new { process_requests }
           @time = Time.now
           @router = Router.instance
+          @session_service = SessionDetailService.instance
           @pending_requests = nil
         end
 
         def process_requests
 
@@ -37,17 +39,19 @@
                 send_parked_response(connection)
                 next
               end
 
               # response from get_forward will be a queue to push messages onto and a response queue to retrieve result from
+              @session_service.add_request(request_task, @uri)
               send << request_task
               @pending_request = connection
 
               log 'queued to console'
 
               # now get the response once available and send back using this connection
               begin
                 request_obj = recv.pop
+                @session_service.add_request(request_task, @uri)
                 @pending_request = nil
                 request_obj.headers.each do |line|
                   connection.write line
                 end
                 unless request_obj.body.nil?
\ No newline at end of file