Add demos handler.rb #1044
This commit is contained in:
59
extensions/demos/handler.rb
Normal file
59
extensions/demos/handler.rb
Normal file
@@ -0,0 +1,59 @@
|
|||||||
|
#
|
||||||
|
# Copyright (c) 2006-2014 Wade Alcorn - wade@bindshell.net
|
||||||
|
# Browser Exploitation Framework (BeEF) - http://beefproject.com
|
||||||
|
# See the file 'doc/COPYING' for copying permission
|
||||||
|
#
|
||||||
|
module BeEF
|
||||||
|
module Extension
|
||||||
|
module Demos
|
||||||
|
|
||||||
|
class Handler
|
||||||
|
|
||||||
|
def initialize(file_path)
|
||||||
|
if File.exists?(file_path)
|
||||||
|
@file_path = file_path
|
||||||
|
else
|
||||||
|
print_error "[Demos] File does not exist: #{file_path}"
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
def call(env)
|
||||||
|
@body = ''
|
||||||
|
@request = Rack::Request.new(env)
|
||||||
|
@params = @request.query_string
|
||||||
|
@response = Rack::Response.new(body=[], 200, header={})
|
||||||
|
config = BeEF::Core::Configuration.instance
|
||||||
|
eruby = Erubis::FastEruby.new(File.read(@file_path))
|
||||||
|
@body << eruby.evaluate({
|
||||||
|
'hook_uri' => config.get("beef.http.hook_file")
|
||||||
|
})
|
||||||
|
|
||||||
|
@response = Rack::Response.new(
|
||||||
|
body = [@body],
|
||||||
|
status = 200,
|
||||||
|
header = {
|
||||||
|
'Pragma' => 'no-cache',
|
||||||
|
'Cache-Control' => 'no-cache',
|
||||||
|
'Expires' => '0',
|
||||||
|
'Content-Type' => 'text/html'
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
end
|
||||||
|
|
||||||
|
private
|
||||||
|
|
||||||
|
# @note String representing the absolute path to the .html file
|
||||||
|
@file_path
|
||||||
|
|
||||||
|
# @note Object representing the HTTP request
|
||||||
|
@request
|
||||||
|
|
||||||
|
# @note Object representing the HTTP response
|
||||||
|
@response
|
||||||
|
|
||||||
|
end
|
||||||
|
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
Reference in New Issue
Block a user