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

Save command status when using websockets - Fix #1200

Browse Source
This commit is contained in:
Brendan Coles
2016-01-27 16:50:51 +00:00
parent 96716e248a
commit 19563d696f
1 changed files with 11 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
core/main/network_stack/websocket/websocket.rb
View File

@@ -244,10 +244,19 @@ module BeEF
(print_error "command_id is invalid"; return) if not BeEF::Filters.is_valid_command_id?(data["cid"])
(print_error "command name is empty"; return) if data["handler"].empty?
(print_error "command results are empty"; return) if command_results.empty?
(print_error "command status is invalid"; return) unless data["status"] =~ /\A0|1|2|undefined\z/
if data["status"] == "undefined"
status = 0
else
status = data["status"].to_i
end
handler = data["handler"]
if handler.match(/command/)
BeEF::Core::Models::Command.save_result(hooked_browser, data["cid"],
@@config.get("beef.module.#{handler.gsub("/command/", "").gsub(".js", "")}.name"), command_results)
BeEF::Core::Models::Command.save_result(hooked_browser,
data["cid"],
@@config.get("beef.module.#{handler.gsub("/command/", "").gsub(".js", "")}.name"),
command_results,
status)
else #processing results from extensions, call the right handler
data["beefhook"] = hooked_browser
data["results"] = JSON.parse(Base64.decode64(data["result"]))