From 256ef0f2e8ed0d292e2ab9c87966bc382c45e899 Mon Sep 17 00:00:00 2001 From: antisnatchor Date: Fri, 11 May 2012 11:12:12 +0100 Subject: [PATCH] Fix issue 673 (command module friendly name in WebSocket fork) --- core/main/network_stack/websocket/websocket.rb | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/core/main/network_stack/websocket/websocket.rb b/core/main/network_stack/websocket/websocket.rb index a446129ce..e66487426 100644 --- a/core/main/network_stack/websocket/websocket.rb +++ b/core/main/network_stack/websocket/websocket.rb @@ -95,9 +95,7 @@ module BeEF BeEF::Core::Handlers::Commands #call the handler for websocket cmd response #@param [Hash] data contains the answer of a command - #@todo ve this stuff in an Handler and resolve the Module friendly name def execute (data) - command_results=Hash.new command_results["data"]=Base64.decode64(data["result"]) command_results["data"].force_encoding('UTF-8') @@ -105,8 +103,8 @@ 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? - BeEF::Core::Models::Command.save_result(data["bh"], data["cid"], data["handler"], command_results) - + BeEF::Core::Models::Command.save_result(data["bh"], data["cid"], + @@config.get("beef.module.#{data["handler"].gsub("/command/","").gsub(".js","")}.name"), command_results) end