From 64cee24c65af4126405243a6c20c8288e3a932aa Mon Sep 17 00:00:00 2001 From: antisnatchor Date: Sat, 7 Apr 2012 12:46:57 +0100 Subject: [PATCH] Added WebSocket config options for enable, secure and port --- beef | 4 +++- config.yaml | 4 ++++ core/main/network_stack/websocket/websocket.rb | 6 ++++-- 3 files changed, 11 insertions(+), 3 deletions(-) diff --git a/beef b/beef index c37e7b50e..622e03a27 100755 --- a/beef +++ b/beef @@ -113,7 +113,9 @@ BeEF::Core::Console::Banners.print_network_interfaces_routes print_info "RESTful API key: #{BeEF::Core::Crypto::api_token}" #@note Starts the WebSocket server -BeEF::Core::Websocket::Websocket.new +if config.get("beef.http.websocket.enable") + BeEF::Core::Websocket::Websocket.new +end diff --git a/config.yaml b/config.yaml index cc3de7e80..e94e8ab2e 100644 --- a/config.yaml +++ b/config.yaml @@ -37,6 +37,10 @@ beef: hook_file: "/hook.js" hook_session_name: "BEEFHOOK" session_cookie_name: "BEEFSESSION" + websocket: + enable: true + secure: false # use WebSocketSecure + port: 6666 database: # For information on using other databases please read the diff --git a/core/main/network_stack/websocket/websocket.rb b/core/main/network_stack/websocket/websocket.rb index 159234655..a290f8b66 100644 --- a/core/main/network_stack/websocket/websocket.rb +++ b/core/main/network_stack/websocket/websocket.rb @@ -21,10 +21,12 @@ module BeEF def initialize print_info("Starting WebSockets") + config = BeEF::Core::Configuration.instance - #todo antisnatchor: add config file options for IP/port/enable websocket + #todo antisnatchor: start websocket secure if beef.http.websocket.secure == true + server = WebSocketServer.new :accepted_domains => "0.0.0.0", + :port => config.get("beef.http.websocket.port") - server = WebSocketServer.new(:accepted_domains => "0.0.0.0",:port => 6666) #we get host and port server.run() do |ws| #@TODO debug print the path and who request for hooked browser mapping print_info("Path requested #{ws.path} Origins #{ws.origin}")