beef/extensions/proxy/handlers/zombie/handler.rb

module BeEF
module Extension
module Proxy
module Handlers
module Zombie

  class Handler

    attr_reader :guard
    @response_body = nil
    H = BeEF::Core::Models::Http

    # This function will forward requests to the target and 
    # the browser will perform the request. Then the results
    # will be sent back.
    def forward_request(hooked_browser_id, req, res)

      # Append port to domain string if not 80 or 443
      if req.port != 80 or req.port != 443
        domain = req.host.to_s + ':' + req.port.to_s
      else
        domain = req.host.to_s
      end
      
      # Saves the new HTTP request to the db for processing by browser.
      # IDs are created and incremented automatically by DataMapper.
      http = H.new(
        :request => req,
        :method => req.request_method.to_s,
        :domain => domain,
        :path => req.path.to_s,
        :request_date => Time.now,
        :hooked_browser_id => hooked_browser_id
      )
      http.save

      # Starts a new thread scoped to this Handler instance, in order to minimize performance degradation
      # while waiting for the HTTP response to be stored in the db.
      print_info("[PROXY] Thread started in order to process request ##{http.id} to [#{req.path.to_s}] on domain [#{domain}]")
      @response_thread = Thread.new do

        while !H.first(:id => http.id).has_ran
          sleep 0.5
        end

        @response_body = H.first(:id => http.id).response_data

      end

      @response_thread.join
      print_info("[PROXY] Response for request ##{http.id} to [#{req.path.to_s}] on domain [#{domain}] correctly processed")

      res.body = @response_body
    end
  end
end
end
end
end
end