garfield/wpscan
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Memory improvements during wp-items enumerations

Browse Source
This commit is contained in:
erwanlr
2014-03-29 12:16:23 +01:00
parent 7053a1e42a
commit c4c8a8b703
1 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
lib/common/collections/wp_items/detectable.rb
View File

@@ -17,6 +17,7 @@ class WpItems < Array
hydra = browser.hydra hydra = browser.hydra
targets = targets_items(wp_target, options) targets = targets_items(wp_target, options)
progress_bar = progress_bar(targets.size, options) progress_bar = progress_bar(targets.size, options)
queue_count = 0
exist_options = { exist_options = {
error_404_hash: wp_target.error_404_hash, error_404_hash: wp_target.error_404_hash,
homepage_hash: wp_target.homepage_hash, homepage_hash: wp_target.homepage_hash,
@@ -43,8 +44,16 @@ class WpItems < Array
end end
hydra.queue(request) hydra.queue(request)
queue_count += 1
if queue_count >= browser.max_threads
hydra.run
queue_count = 0
puts "Sent #{browser.max_threads} requests ..." if options[:verbose]
end
end end
# run the remaining requests
hydra.run hydra.run
results.sort! results.sort!
results # can't just return results.sort because the #sort returns an array, and we want a WpItems results # can't just return results.sort because the #sort returns an array, and we want a WpItems