94c9c96250
git-svn-id: https://beef.googlecode.com/svn/trunk@919 b87d56ec-f9c0-11de-8c8a-61c5e9addfc9
66 lines
1.8 KiB
Ruby
66 lines
1.8 KiB
Ruby
module BeEF
|
|
module Extension
|
|
module Proxy
|
|
module Handlers
|
|
module Zombie
|
|
|
|
module Handler
|
|
|
|
# Variable representing the Http DB model.
|
|
H = BeEF::Core::Models::Http
|
|
# This function will forward requests to the zombie and
|
|
# the browser will perform the request. Then the results
|
|
# will be sent back to use
|
|
def forward_request(hooked_browser_id, req, res)
|
|
|
|
# Generate a id for the req in the http table and check it doesnt already exist
|
|
http_id = rand(10000)
|
|
http_db = H.first(:id => http_id) || nil
|
|
|
|
while !http_db.nil?
|
|
http_id = rand(10000)
|
|
http_db = H.first(:id => http_id) || nil
|
|
end
|
|
|
|
# some debug info
|
|
print_debug "[PROXY] Forwarding request #" + http_id.to_s + " from zombie [" + hooked_browser_id.to_s + "]" + " to host [" + req.host.to_s + "]"
|
|
|
|
# Saves the new HTTP request to the db for processing by HB
|
|
http = H.new(
|
|
:id => http_id,
|
|
:request => req,
|
|
:method => req.request_method.to_s,
|
|
:domain => req.host.to_s,
|
|
:path => req.path.to_s,
|
|
:date => Time.now,
|
|
:hooked_browser_id => hooked_browser_id
|
|
)
|
|
http.save
|
|
|
|
# Polls the DB for the response and then sets it when present
|
|
|
|
http_db = H.first(:id => http_id)
|
|
|
|
while !http_db.has_ran
|
|
#sleep 1 # adding a sleep here is a bottleneck. Even querying in this way is not a good way.
|
|
# By the way removing the sleep instead the proxy response time is 8/10 seconds instead of almost 20 seconds.
|
|
# This code should be reimplemented with Threading.
|
|
http_db = H.first(:id => http_id)
|
|
end
|
|
#
|
|
res.body = http_db.response
|
|
|
|
res
|
|
|
|
end
|
|
|
|
module_function :forward_request
|
|
|
|
end
|
|
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|