Sha256: 56c402c76ff27264db460d37f82361d58f3fca558186daf8ddf4385ce7ee319e
Contents?: true
Size: 1.85 KB
Versions: 10
Compression:
Stored size: 1.85 KB
Contents
module CMSScanner
# Scope system logic
class Target < WebSite
# @return [ Array<PublicSuffix::Domain, String> ]
def scope
@scope ||= Scope.new
end
# @param [ String ] url An absolute URL
#
# @return [ Boolean ] true if the url given is in scope
def in_scope?(url)
scope.include?(Addressable::URI.parse(url.strip).host)
rescue StandardError
false
end
# @param [ Typhoeus::Response ] res
# @param [ String ] xpath
# @param [ Array<String> ] attributes
#
# @yield [ String, Nokogiri::XML::Element ] The in scope url and its associated tag
#
# @return [ Array<String> ] The in scope absolute URLs detected in the response's body
def in_scope_urls(res, xpath = '//link|//script|//style|//img|//a', attributes = %w[href src])
found = []
urls_from_page(res, xpath, attributes) do |url, tag|
next unless in_scope?(url)
yield url, tag if block_given?
found << url
end
found
end
# Scope Implementation
class Scope
# @return [ Array<PublicSuffix::Domain ] The valid domains in scope
def domains
@domains ||= []
end
# @return [ Array<String> ] The invalid domains in scope (such as IP addresses etc)
def invalid_domains
@invalid_domains ||= []
end
def <<(element)
if PublicSuffix.valid?(element)
domains << PublicSuffix.parse(element)
else
invalid_domains << element
end
end
# @return [ Boolean ] Wether or not the host is in the scope
def include?(host)
if PublicSuffix.valid?(host)
domain = PublicSuffix.parse(host)
domains.each { |d| return true if domain.match(d) }
else
invalid_domains.each { |d| return true if host == d }
end
false
end
end
end
end
Version data entries
10 entries across 10 versions & 1 rubygems