diff options
Diffstat (limited to 'lib/resolv-replace.rb')
-rw-r--r-- | lib/resolv-replace.rb | 44 |
1 files changed, 28 insertions, 16 deletions
diff --git a/lib/resolv-replace.rb b/lib/resolv-replace.rb index 560a813d66..eb66baddf0 100644 --- a/lib/resolv-replace.rb +++ b/lib/resolv-replace.rb @@ -1,50 +1,62 @@ require 'socket' require 'resolv' -class BasicSocket - alias original_resolv_send send - def send(mesg, flags, *rest) - rest[0] = Resolv.getaddress(rest[0]).to_s unless rest.empty? - original_resolv_send(mesg, flags, *rest) - end -end - class << IPSocket alias original_resolv_getaddress getaddress def getaddress(host) - return Resolv.getaddress(host).to_s + begin + return Resolv.getaddress(host).to_s + rescue Resolv::ResolvError + raise SocketError, "Name or service not known: #{host}" + end end end class TCPSocket alias original_resolv_initialize initialize def initialize(host, serv, *rest) - rest[0] = Resolv.getaddress(rest[0]).to_s unless rest.empty? - original_resolv_initialize(Resolv.getaddress(host).to_s, serv, *rest) + rest[0] = IPSocket.getaddress(rest[0]) unless rest.empty? + original_resolv_initialize(IPSocket.getaddress(host), serv, *rest) end end class UDPSocket alias original_resolv_bind bind def bind(host, port) - original_resolv_bind(Resolv.getaddress(host).to_s, port) + original_resolv_bind(IPSocket.getaddress(host), port) end alias original_resolv_connect connect def connect(host, port) - original_resolv_connect(Resolv.getaddress(host).to_s, port) + original_resolv_connect(IPSocket.getaddress(host), port) end alias original_resolv_send send def send(mesg, flags, *rest) - rest[0] = Resolv.getaddress(rest[0]).to_s unless rest.empty? - original_resolv_send(mesg, flags, *rest) + if rest.length == 2 + host, port = rest + begin + addrs = Resolv.getaddresses(host) + rescue Resolv::ResolvError + raise SocketError, "Name or service not known: #{host}" + end + err = nil + addrs[0...-1].each {|addr| + begin + return original_resolv_send(mesg, flags, addr, port) + rescue SystemCallError + end + } + original_resolv_send(mesg, flags, addrs[-1], port) + else + original_resolv_send(mesg, flags, *rest) + end end end class SOCKSSocket alias original_resolv_initialize initialize def initialize(host, serv) - original_resolv_initialize(Resolv.getaddress(host).to_s, port) + original_resolv_initialize(IPSocket.getaddress(host), port) end end if defined? SOCKSSocket |