diff --git a/core/main/rest/handlers/admin.rb b/core/main/rest/handlers/admin.rb new file mode 100644 index 000000000..51fa7555e --- /dev/null +++ b/core/main/rest/handlers/admin.rb @@ -0,0 +1,42 @@ +# +# 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 Admin < BeEF::Core::Router::Router + + config = BeEF::Core::Configuration.instance + + before do + # error 401 unless params[:token] == config.get('beef.api_token') + halt 401 if not BeEF::Core::Rest.permitted_source?(request.ip) + headers 'Content-Type' => 'application/json; charset=UTF-8', + 'Pragma' => 'no-cache', + 'Cache-Control' => 'no-cache', + 'Expires' => '0' + end + + get '/' do + "Hiya" + end + + private + + end + end + end +end \ No newline at end of file