Hiddenden/beef
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update api token

Browse Source
This commit is contained in:
Josh
2020-04-26 23:31:59 -07:00
parent d257109a9d
commit 3318b1e059
1 changed files with 34 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

67
spec/beef/extensions/websocket_hooked_browser_spec.rb
View File

@@ -22,44 +22,45 @@ RSpec.describe 'BeEF WebSockets: Browser Hooking', :run_on_browserstack => true
@secure_port = @config.get('beef.http.websocket.secure_port') @secure_port = @config.get('beef.http.websocket.secure_port')
@config.set('beef.http.websocket.secure', true) @config.set('beef.http.websocket.secure', true)
@config.set('beef.http.websocket.enable', true) @config.set('beef.http.websocket.enable', true)
#set config parameters #set config parameters
@config.set('beef.credentials.user', "beef") @config.set('beef.credentials.user', "beef")
@config.set('beef.credentials.passwd', "beef") @config.set('beef.credentials.passwd', "beef")
@username = @config.get('beef.credentials.user') @username = @config.get('beef.credentials.user')
@password = @config.get('beef.credentials.passwd') @password = @config.get('beef.credentials.passwd')
#load extensions, best practice is to reload as previous tests can potentially cause issues. #load extensions, best practice is to reload as previous tests can potentially cause issues.
print_info "Loading in BeEF::Extensions" print_info "Loading in BeEF::Extensions"
BeEF::Extensions.load BeEF::Extensions.load
sleep 2 sleep 2
if @config.get('beef.module').nil? if @config.get('beef.module').nil?
puts "loading modules" puts "loading modules"
BeEF::Modules.load BeEF::Modules.load
sleep 2 sleep 2
end end
#generate token for the api to use
@token = BeEF::Core::Crypto::api_token # load up DB
# load up DB # Connect to DB
# Connect to DB ActiveRecord::Base.logger = nil
ActiveRecord::Base.logger = nil OTR::ActiveRecord.migrations_paths = [File.join('core', 'main', 'ar-migrations')]
OTR::ActiveRecord.migrations_paths = [File.join('core', 'main', 'ar-migrations')] OTR::ActiveRecord.configure_from_hash!(adapter:'sqlite3', database:'beef.db')
OTR::ActiveRecord.configure_from_hash!(adapter:'sqlite3', database:'beef.db')
# Migrate (if required) # Migrate (if required)
context = ActiveRecord::Migration.new.migration_context context = ActiveRecord::Migration.new.migration_context
if context.needs_migration? if context.needs_migration?
puts "migrating db" puts "migrating db"
ActiveRecord::Migrator.new(:up, context.migrations, context.schema_migration).migrate ActiveRecord::Migrator.new(:up, context.migrations, context.schema_migration).migrate
end end
http_hook_server = BeEF::Core::Server.instance http_hook_server = BeEF::Core::Server.instance
http_hook_server.prepare http_hook_server.prepare
@pids = fork do #generate token for the api to use
@token = BeEF::Core::Crypto::api_token
@pids = fork do
BeEF::API::Registrar.instance.fire(BeEF::API::Server, 'pre_http_start', http_hook_server) BeEF::API::Registrar.instance.fire(BeEF::API::Server, 'pre_http_start', http_hook_server)
end end
@pid = fork do @pid = fork do
http_hook_server.start http_hook_server.start
end end
# wait for server to start # wait for server to start
sleep 1 sleep 1
end end