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

Fix another data handling issue

Browse Source
This commit is contained in:
zinduolis
2024-10-27 16:32:51 +10:00
parent 740d056142
commit 1cd9d0d8ad
1 changed files with 1 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
spec/beef/modules/debug/test_beef_debugs_spec.rb
View File

@@ -89,18 +89,12 @@ RSpec.describe 'BeEF Debug Command Modules:', run_on_browserstack: true do
# Grab Command Module IDs as they can differ from machine to machine # Grab Command Module IDs as they can differ from machine to machine
@debug_mod_ids = JSON.parse(RestClient.get("#{RESTAPI_MODULES}?token=#{@token}")) @debug_mod_ids = JSON.parse(RestClient.get("#{RESTAPI_MODULES}?token=#{@token}"))
@debug_mod_names_ids = {} @debug_mod_names_ids = {}
@debug_mods = @debug_mod_ids.to_a.select do |cmd_mod| @debug_mods = @debug_mod_ids.to_a.select do |cmd_mod|
category = Array(cmd_mod[1]['category']) # Ensure category is always an array category = Array(cmd_mod[1]['category']) # Ensure category is always an array
category.include?('Debug') category.include?('Debug')
end.map do |debug_mod| end.map do |debug_mod|
# Only process if debug_mod[1] exists and has the expected keys @debug_mod_names_ids[debug_mod[1]['class']] = debug_mod[1]['id']
if debug_mod[1] && debug_mod[1].key?('class') && debug_mod[1].key?('id')
@debug_mod_names_ids[debug_mod[1]['class']] = debug_mod[1]['id']
else
puts "Warning: Incomplete data found for debug_mod: #{debug_mod.inspect}"
end
end end
rescue StandardError => e rescue StandardError => e
print_info "Exception: #{e}" print_info "Exception: #{e}"