diff options
author | shyouhei <shyouhei@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2007-08-15 23:23:39 +0000 |
---|---|---|
committer | shyouhei <shyouhei@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2007-08-15 23:23:39 +0000 |
commit | 6175ca03be6d0d51359f9017123708987d0f5eb7 (patch) | |
tree | ecfcf6e79a21b1d25c3f6f42dd68ea0a14add89c /ruby_1_8_5/lib/resolv-replace.rb | |
parent | 80a56b248b2e9cfc95622aed98750df05a19f667 (diff) |
add tag v1_8_5_91v1_8_5_91
git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/tags/v1_8_5_91@13046 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
Diffstat (limited to 'ruby_1_8_5/lib/resolv-replace.rb')
-rw-r--r-- | ruby_1_8_5/lib/resolv-replace.rb | 62 |
1 files changed, 62 insertions, 0 deletions
diff --git a/ruby_1_8_5/lib/resolv-replace.rb b/ruby_1_8_5/lib/resolv-replace.rb new file mode 100644 index 0000000000..5d15b4577c --- /dev/null +++ b/ruby_1_8_5/lib/resolv-replace.rb @@ -0,0 +1,62 @@ +require 'socket' +require 'resolv' + +class << IPSocket + alias original_resolv_getaddress getaddress + def getaddress(host) + begin + return Resolv.getaddress(host).to_s + rescue Resolv::ResolvError + raise SocketError, "Hostname not known: #{host}" + end + end +end + +class TCPSocket + alias original_resolv_initialize initialize + def initialize(host, 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(IPSocket.getaddress(host), port) + end + + alias original_resolv_connect connect + def connect(host, port) + original_resolv_connect(IPSocket.getaddress(host), port) + end + + alias original_resolv_send send + def send(mesg, flags, *rest) + if rest.length == 2 + host, port = rest + begin + addrs = Resolv.getaddresses(host) + rescue Resolv::ResolvError + raise SocketError, "Hostname 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(IPSocket.getaddress(host), port) + end +end if defined? SOCKSSocket |