diff --git a/lib/private_address_check.rb b/lib/private_address_check.rb index b1d2995..3b82a43 100644 --- a/lib/private_address_check.rb +++ b/lib/private_address_check.rb @@ -31,6 +31,8 @@ def private_address?(address) def resolves_to_private_address?(hostname) ips = Resolv.getaddresses(hostname) + return true if ips.empty? + ips.any? do |ip| private_address?(ip) end diff --git a/test/private_address_check_test.rb b/test/private_address_check_test.rb index 6560751..2afdf79 100644 --- a/test/private_address_check_test.rb +++ b/test/private_address_check_test.rb @@ -35,4 +35,8 @@ def test_private_hostname_for_public_addresses def test_private_hostname_for_private_addresses assert PrivateAddressCheck.resolves_to_private_address?("localhost") end + + def test_private_address_for_malformed_addresses + assert PrivateAddressCheck.resolves_to_private_address?("127.1") + end end