Skip to content

Commit

Permalink
Merge b9be7b3 into e3dde5d
Browse files Browse the repository at this point in the history
  • Loading branch information
ninoseki committed Oct 3, 2020
2 parents e3dde5d + b9be7b3 commit a5773a1
Showing 1 changed file with 7 additions and 3 deletions.
10 changes: 7 additions & 3 deletions lib/rogue_one/detector.rb
Expand Up @@ -89,13 +89,13 @@ def inspect
return unless @memo.empty?

# read domains outside of the async blocks
domains
load_domains

normal = bulk_resolve(normal_resolver, domains)
normal_resolutions = bulk_resolve(normal_resolver, domains)
resolutions = bulk_resolve(target_resolver, domains)

results = resolutions.map do |domain, addresses|
normal_addresses = normal.dig(domain) || []
normal_addresses = normal_resolutions.dig(domain) || []
address = (addresses || []).first
[domain, address] if address && !normal_addresses.include?(address)
end.compact.to_h
Expand All @@ -104,6 +104,10 @@ def inspect
@verbose_memo = results if verbose
end

def load_domains
domains
end

def domains
@domains ||= custom_list ? custom_domains : top_100_domains
end
Expand Down

0 comments on commit a5773a1

Please sign in to comment.