diff options
Diffstat (limited to 'lib')
-rw-r--r-- | lib/net/protocol.rb | 36 |
1 files changed, 31 insertions, 5 deletions
diff --git a/lib/net/protocol.rb b/lib/net/protocol.rb index 71378af804..3610abcacc 100644 --- a/lib/net/protocol.rb +++ b/lib/net/protocol.rb @@ -75,13 +75,39 @@ module Net # :nodoc: # ReadTimeout, a subclass of Timeout::Error, is raised if a chunk of the # response cannot be read within the read_timeout. - class ReadTimeout < Timeout::Error; end + class ReadTimeout < Timeout::Error + def initialize(io = nil) + @io = io + end + attr_reader :io + + def message + msg = super + if @io + msg = "#{msg} with #{@io.inspect}" + end + msg + end + end ## # WriteTimeout, a subclass of Timeout::Error, is raised if a chunk of the # response cannot be written within the write_timeout. Not raised on Windows. - class WriteTimeout < Timeout::Error; end + class WriteTimeout < Timeout::Error + def initialize(io = nil) + @io = io + end + attr_reader :io + + def message + msg = super + if @io + msg = "#{msg} with #{@io.inspect}" + end + msg + end + end class BufferedIO #:nodoc: internal use only @@ -188,12 +214,12 @@ module Net # :nodoc: rv.clear return when :wait_readable - @io.to_io.wait_readable(@read_timeout) or raise Net::ReadTimeout + (io = @io.to_io).wait_readable(@read_timeout) or raise Net::ReadTimeout.new(io) # continue looping when :wait_writable # OpenSSL::Buffering#read_nonblock may fail with IO::WaitWritable. # http://www.openssl.org/support/faq.html#PROG10 - @io.to_io.wait_writable(@read_timeout) or raise Net::ReadTimeout + (io = @io.to_io).wait_writable(@read_timeout) or raise Net::ReadTimeout.new(io) # continue looping when nil raise EOFError, 'end of file reached' @@ -267,7 +293,7 @@ module Net # :nodoc: end # continue looping when :wait_writable - @io.to_io.wait_writable(@write_timeout) or raise Net::WriteTimeout + (io = @io.to_io).wait_writable(@write_timeout) or raise Net::WriteTimeout.new(io) # continue looping end while need_retry end |