dnrops
/
metasploit-framework
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add support for getting IPv6 addresses from hosts files and saving them into the cache

This commit is contained in:
Grant Willcox 2023-04-28 14:47:56 -05:00
parent 708da64a68
commit 3e81408a24
No known key found for this signature in database
GPG Key ID: D35E05C0F2B81E83
1 changed files with 16 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
lib/rex/proto/dns/cached_resolver.rb
View File

@ -43,7 +43,13 @@ module DNS
hostname = ent.pop
next unless MATCH_HOSTNAME.match hostname
begin
self.cache.add_static(hostname, ent.first)
if Rex::Socket.is_ipv4?(ent.first)
self.cache.add_static(hostname, ent.first, Dnsruby::Types::A)
elsif Rex::Socket.is_ipv6?(ent.first)
self.cache.add_static(hostname, ent.first, Dnsruby::Types::AAAA)
else
raise "Unknown IP address format #{ent.first} in hosts file!"
end
rescue => e
# Deal with edge-cases in users' hostsfile
@logger.error(e)
@ -51,7 +57,15 @@ module DNS
end
hostname = ent.pop
begin
self.cache.add_static(hostname, ent.first) if MATCH_HOSTNAME.match hostname
if MATCH_HOSTNAME.match hostname
if Rex::Socket.is_ipv4?(ent.first)
self.cache.add_static(hostname, ent.first, Dnsruby::Types::A)
elsif Rex::Socket.is_ipv6?(ent.first)
self.cache.add_static(hostname, ent.first, Dnsruby::Types::AAAA)
else
raise "Unknown IP address format #{ent.first} in hosts file!"
end
end
rescue => e
# Deal with edge-cases in users' hostsfile
@logger.error(e)