This commit is contained in:
erwanlr
2021-05-03 10:15:42 +02:00
parent 75d6a16298
commit 1d18514ab5
3 changed files with 18 additions and 2 deletions

View File

@@ -39,8 +39,7 @@ module WPScan
#
# @return [ Hash ]
def potential_urls(opts = {})
urls = {}
domain_name = (PublicSuffix.domain(target.uri.host) || target.uri.host)[/(^[\w|-]+)/, 1]
urls = {}
File.open(opts[:list]).each_with_index do |path, index|
path.gsub!('{domain_name}', domain_name)
@@ -51,6 +50,14 @@ module WPScan
urls
end
def domain_name
@domain_name ||= if Resolv::AddressRegex.match?(target.uri.host)
target.uri.host
else
(PublicSuffix.domain(target.uri.host) || target.uri.host)[/(^[\w|-]+)/, 1]
end
end
def create_progress_bar(opts = {})
super(opts.merge(title: ' Checking DB Exports -'))
end

View File

@@ -13,6 +13,7 @@ require 'uri'
require 'time'
require 'readline'
require 'securerandom'
require 'resolv'
# Monkey Patches/Fixes/Override
require 'wpscan/typhoeus/response' # Adds a from_vuln_api? method
# Custom Libs

View File

@@ -64,6 +64,14 @@ describe WPScan::Finders::DbExports::KnownLocations do
expect(finder.potential_urls(opts).keys).to include "#{url}/dc-2.sql"
end
end
context 'when an IP address' do
let(:url) { 'http://192.168.1.12' }
it 'replaces {domain_name} by the IP address' do
expect(finder.potential_urls(opts).keys).to include "#{url}/192.168.1.12.sql"
end
end
end
describe '#aggressive' do