cc3bfc071e
Added beef_rest_client based on the beef rest api found in tools/lib Added mass auth attempts Adjusted test_constants to use environment variables -- to use with rake. Eventually should be a test
43 lines
1.3 KiB
Ruby
43 lines
1.3 KiB
Ruby
#
|
|
# Copyright (c) 2006-2017 Wade Alcorn - wade@bindshell.net
|
|
# Browser Exploitation Framework (BeEF) - http://beefproject.com
|
|
# See the file 'doc/COPYING' for copying permission
|
|
#
|
|
# less noisy verson of BeeRestAPI found in tools.
|
|
class BeefRestClient
|
|
def initialize proto, host, port, user, pass
|
|
@user = user
|
|
@pass = pass
|
|
@url = "#{proto}://#{host}:#{port}/api/"
|
|
@token = nil
|
|
end
|
|
|
|
def auth
|
|
begin
|
|
response = RestClient.post "#{@url}admin/login",
|
|
{ 'username' => "#{@user}",
|
|
'password' => "#{@pass}" }.to_json,
|
|
:content_type => :json,
|
|
:accept => :json
|
|
result = JSON.parse(response.body)
|
|
@token = result['token']
|
|
{:success => result['success'], :payload => result}
|
|
rescue => e
|
|
{:success => false, :payload => e.message }
|
|
end
|
|
end
|
|
|
|
def version
|
|
return {:success => false, :payload => 'no token'} if @token.nil?
|
|
begin
|
|
response = RestClient.get "#{@url}server/version", {:params => {:token => @token}}
|
|
result = JSON.parse(response.body)
|
|
|
|
{:success => result['success'], :payload => result}
|
|
rescue => e
|
|
print_error "Could not retrieve BeEF version: #{e.message}"
|
|
{:success => false, :payload => e.message}
|
|
end
|
|
end
|
|
end
|