Added stubs and registered classes for the 3 main RESTful API endpoints: hooks, modules, logs

core/bootstrap.rb

@@ -42,5 +42,7 @@ require 'core/extensions'
 require 'core/hbmanager'
 
 ## @note Include RESTful API
-require 'core/main/rest/handlers/rest'
-require 'core/main/rest/api'
+require 'core/main/rest/handlers/hookedbrowsers'
+require 'core/main/rest/handlers/modules'
+require 'core/main/rest/handlers/logs'
+require 'core/main/rest/api'

core/main/rest/api.rb

@@ -17,14 +17,27 @@ module BeEF
   module Core
     module Rest
 
-      module RegisterHttpHandler
-
+      module RegisterHooksHandler
         def self.mount_handler(server)
-          server.mount('/api', BeEF::Core::Rest::Rest.new)
+          server.mount('/api/hooks', BeEF::Core::Rest::HookedBrowsers.new)
         end
       end
 
-      BeEF::API::Registrar.instance.register(BeEF::Core::Rest::RegisterHttpHandler, BeEF::API::Server, 'mount_handler')
+      module RegisterModulesHandler
+        def self.mount_handler(server)
+          server.mount('/api/modules', BeEF::Core::Rest::Modules.new)
+        end
+      end
+
+      module RegisterLogsHandler
+        def self.mount_handler(server)
+          server.mount('/api/logs', BeEF::Core::Rest::Logs.new)
+        end
+      end
+
+      BeEF::API::Registrar.instance.register(BeEF::Core::Rest::RegisterHooksHandler, BeEF::API::Server, 'mount_handler')
+      BeEF::API::Registrar.instance.register(BeEF::Core::Rest::RegisterModulesHandler, BeEF::API::Server, 'mount_handler')
+      BeEF::API::Registrar.instance.register(BeEF::Core::Rest::RegisterLogsHandler, BeEF::API::Server, 'mount_handler')
 
     end
   end

core/main/rest/handlers/hookedbrowsers.rb

@@ -17,7 +17,7 @@
 module BeEF
   module Core
     module Rest
-      class Rest < Sinatra::Base
+      class HookedBrowsers < Sinatra::Base
 
         config = BeEF::Core::Configuration.instance
 
@@ -25,13 +25,8 @@ module BeEF
           error 401 unless params[:token] == config.get('beef.api_token')
         end
 
-        get '/sinatra' do
-          "Sinatra! v.#{Sinatra::VERSION}"
-        end
-
-        get '/modules' do
-          puts Sinatra::VERSION
-          "List all modules!"
+        get '/' do
+          "return hooked browsers"
         end
 
       end

core/main/rest/handlers/logs.rb

@@ -0,0 +1,35 @@
+#
+#   Copyright 2012 Wade Alcorn wade@bindshell.net
+#
+#   Licensed under the Apache License, Version 2.0 (the "License");
+#   you may not use this file except in compliance with the License.
+#   You may obtain a copy of the License at
+#
+#       http://www.apache.org/licenses/LICENSE-2.0
+#
+#   Unless required by applicable law or agreed to in writing, software
+#   distributed under the License is distributed on an "AS IS" BASIS,
+#   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#   See the License for the specific language governing permissions and
+#   limitations under the License.
+#
+
+module BeEF
+  module Core
+    module Rest
+      class Logs < Sinatra::Base
+
+        config = BeEF::Core::Configuration.instance
+
+        before do
+          error 401 unless params[:token] == config.get('beef.api_token')
+        end
+
+        get '/' do
+          "return global logs or HB logs"
+        end
+
+      end
+    end
+  end
+end

core/main/rest/handlers/modules.rb

@@ -0,0 +1,35 @@
+#
+#   Copyright 2012 Wade Alcorn wade@bindshell.net
+#
+#   Licensed under the Apache License, Version 2.0 (the "License");
+#   you may not use this file except in compliance with the License.
+#   You may obtain a copy of the License at
+#
+#       http://www.apache.org/licenses/LICENSE-2.0
+#
+#   Unless required by applicable law or agreed to in writing, software
+#   distributed under the License is distributed on an "AS IS" BASIS,
+#   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#   See the License for the specific language governing permissions and
+#   limitations under the License.
+#
+
+module BeEF
+  module Core
+    module Rest
+      class Modules < Sinatra::Base
+
+        config = BeEF::Core::Configuration.instance
+
+        before do
+          error 401 unless params[:token] == config.get('beef.api_token')
+        end
+
+        get '/' do
+          "return available modules"
+        end
+
+      end
+    end
+  end
+end