diff --git a/core/main/rest/handlers/hookedbrowsers.rb b/core/main/rest/handlers/hookedbrowsers.rb index b56fffbe4..aa269ff88 100644 --- a/core/main/rest/handlers/hookedbrowsers.rb +++ b/core/main/rest/handlers/hookedbrowsers.rb @@ -17,11 +17,9 @@ module BeEF module Core module Rest - class HookedBrowsers < Sinatra::Base + class HookedBrowsers < BeEF::Core::Router::Router config = BeEF::Core::Configuration.instance - configure do set :show_exceptions, false end - not_found do 'Not Found.' end before do error 401 unless params[:token] == config.get('beef.api_token') diff --git a/core/main/rest/handlers/logs.rb b/core/main/rest/handlers/logs.rb index d4ee2bc0d..52a64e089 100644 --- a/core/main/rest/handlers/logs.rb +++ b/core/main/rest/handlers/logs.rb @@ -17,11 +17,9 @@ module BeEF module Core module Rest - class Logs < Sinatra::Base + class Logs < BeEF::Core::Router::Router config = BeEF::Core::Configuration.instance - configure do set :show_exceptions, false end - not_found do 'Not Found.' end before do error 401 unless params[:token] == config.get('beef.api_token') diff --git a/core/main/rest/handlers/modules.rb b/core/main/rest/handlers/modules.rb index 0d2707089..5afa85482 100644 --- a/core/main/rest/handlers/modules.rb +++ b/core/main/rest/handlers/modules.rb @@ -17,11 +17,9 @@ module BeEF module Core module Rest - class Modules < Sinatra::Base + class Modules < BeEF::Core::Router::Router config = BeEF::Core::Configuration.instance - configure do set :show_exceptions, false end - not_found do 'Not Found.' end before do error 401 unless params[:token] == config.get('beef.api_token')