module CMSScanner module Finders # Unique Finders container # # This class is designed to return a unique result such as a version # Note: Finders contained can return multiple results but the #run will only # returned the best finding class UniqueFinders < IndependentFinders # @param [ Hash ] opts # @option opts [ Symbol ] :mode :mixed, :passive or :aggressive # @option opts [ Int ] :confidence_threshold If a finding's confidence reaches this value, # it will be returned as the best finding. # Default is 100. # If <= 0, all finders will be ran. # # @return [ Object ] The best finding def run(opts = {}) opts[:confidence_threshold] ||= 100 symbols_from_mode(opts[:mode]).each do |symbol| each do |finder| [*finder.send(symbol, opts)].compact.each { |found| findings << found } next if opts[:confidence_threshold] <= 0 findings.each { |f| return f if f.confidence >= opts[:confidence_threshold] } end end best_finding(findings) end # @param [ Array ] findings # # @return [ Object ] The best finding def best_finding(findings) # results are sorted by confidence ASC findings.sort_by!(&:confidence) # If all findings have the same confidence, nil is returned return if findings.size > 1 && findings.first.confidence == findings.last.confidence findings.last end end end end