From 22772c7822f7b9b24cb2017f8f796905d615f6a8 Mon Sep 17 00:00:00 2001 From: antisnatchor Date: Mon, 9 Apr 2012 11:16:21 +0100 Subject: [PATCH] Issue 654: the RESTful api classes are now extending the Router class --- core/main/rest/handlers/hookedbrowsers.rb | 4 +--- core/main/rest/handlers/logs.rb | 4 +--- core/main/rest/handlers/modules.rb | 4 +--- 3 files changed, 3 insertions(+), 9 deletions(-) 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')