tcp-client 0.7.0 → 0.9.2

data/lib/tcp-client/errors.rb

@@ -1,76 +1,153 @@
 # frozen_string_literal: true
 
 class TCPClient
+  #
+  # Raised when a SSL connection should be establshed but the OpenSSL gem is not available.
+  #
   class NoOpenSSLError < RuntimeError
     def initialize
       super('OpenSSL is not available')
     end
   end
 
+  #
+  # Raised when a method requires a callback block but no such block is specified.
+  #
   class NoBlockGivenError < ArgumentError
     def initialize
       super('no block given')
     end
   end
 
+  #
+  # Raised when an invalid timeout value was specified.
+  #
   class InvalidDeadLineError < ArgumentError
+    #
+    # @param timeout [Object] the invalid value
+    #
     def initialize(timeout)
       super("invalid deadline - #{timeout}")
     end
   end
 
+  #
+  # Raised by {Configuration} when an undefined attribute should  be set.
+  #
   class UnknownAttributeError < ArgumentError
+    #
+    # @param attribute [Object] the undefined atttribute
+    #
     def initialize(attribute)
       super("unknown attribute - #{attribute}")
     end
   end
 
+  #
+  # Raised when a given timeout exception parameter is not an exception class.
+  #
   class NotAnExceptionError < TypeError
+    #
+    # @param object [Object] the invalid object
+    #
     def initialize(object)
       super("exception class required - #{object.inspect}")
     end
   end
 
-  class NotConnectedError < IOError
+  #
+  # Base exception class for all network related errors.
+  #
+  # Will be raised for any system level network error when {Configuration.normalize_network_errors} is configured.
+  #
+  # You should catch this exception class when you like to handle any relevant {TCPClient} error.
+  #
+  class NetworkError < StandardError
+  end
+
+  #
+  # Raised when a {TCPClient} instance should read/write from/to the network but is not connected.
+  #
+  class NotConnectedError < NetworkError
     def initialize
       super('client not connected')
     end
   end
 
-  class TimeoutError < IOError
+  #
+  # Base exception class for a detected timeout.
+  #
+  # You should catch this exception class when you like to handle any timeout error.
+  #
+  class TimeoutError < NetworkError
+    #
+    # Initializes the instance with an optional message.
+    #
+    # The message will be generated from {#action} when not specified.
+    #
+    # @overload initialize
+    # @overload initialize(message)
+    #
+    # @param message [#to_s] the error message
+    #
     def initialize(message = nil)
       super(message || "unable to #{action} in time")
     end
 
+    #
+    # @attribute [r] action
+    # @return [Symbol] the action which timed out
+    #
     def action
       :process
     end
   end
 
+  #
+  # Raised by default whenever a {TCPClient.connect} timed out.
+  #
   class ConnectTimeoutError < TimeoutError
+    #
+    # @attribute [r] action
+    # @return [Symbol] the action which timed out: `:connect`
+    #
     def action
       :connect
     end
   end
 
+  #
+  # Raised by default whenever a {TCPClient#read} timed out.
+  #
   class ReadTimeoutError < TimeoutError
+    #
+    # @attribute [r] action
+    # @return [Symbol] the action which timed out: :read`
+    #
     def action
       :read
     end
   end
 
+  #
+  # Raised by default whenever a {TCPClient#write} timed out.
+  #
   class WriteTimeoutError < TimeoutError
+    #
+    # @attribute [r] action
+    # @return [Symbol] the action which timed out: `:write`
+    #
     def action
       :write
     end
   end
 
-  NoOpenSSL = NoOpenSSLError
-  NoBlockGiven = NoBlockGivenError
-  InvalidDeadLine = InvalidDeadLineError
-  UnknownAttribute = UnknownAttributeError
-  NotAnException = NotAnExceptionError
-  NotConnected = NotConnectedError
+  NoOpenSSL = NoOpenSSLError # @!visibility private
+  NoBlockGiven = NoBlockGivenError # @!visibility private
+  InvalidDeadLine = InvalidDeadLineError # @!visibility private
+  UnknownAttribute = UnknownAttributeError # @!visibility private
+  NotAnException = NotAnExceptionError # @!visibility private
+  NotConnected = NotConnectedError # @!visibility private
   deprecate_constant(
     :NoOpenSSL,
     :NoBlockGiven,

data/lib/tcp-client/mixin/io_with_deadline.rb

@@ -1,6 +1,7 @@
 # frozen_string_literal: true
 
-module IOWithDeadlineMixin
+# @!visibility private
+module IOWithDeadlineMixin # :nodoc:
   def self.included(mod)
     methods = mod.instance_methods
     if methods.index(:wait_writable) && methods.index(:wait_readable)

data/lib/tcp-client/ssl_socket.rb

@@ -18,6 +18,7 @@ class TCPClient
       super(socket, create_context(ssl_params))
       self.sync_close = true
       self.hostname = address.hostname
+      check_new_session if @new_session
       deadline.valid? ? connect_with_deadline(deadline, exception) : connect
       post_connection_check(address.hostname) if should_verify?(ssl_params)
     end
@@ -25,7 +26,19 @@ class TCPClient
     private
 
     def create_context(ssl_params)
-      OpenSSL::SSL::SSLContext.new.tap { |ctx| ctx.set_params(ssl_params) }
+      @new_session = nil
+      ::OpenSSL::SSL::SSLContext.new.tap do |ctx|
+        ctx.set_params(ssl_params)
+        ctx.session_cache_mode = CONTEXT_CACHE_MODE
+        ctx.session_new_cb = proc { |_, sess| @new_session = sess }
+      end
+    end
+
+    def check_new_session
+      time = @new_session.time.to_f + @new_session.timeout
+      if Process.clock_gettime(Process::CLOCK_REALTIME) < time
+        self.session = @new_session
+      end
     end
 
     def connect_with_deadline(deadline, exception)
@@ -33,9 +46,13 @@ class TCPClient
     end
 
     def should_verify?(ssl_params)
-      ssl_params[:verify_mode] != OpenSSL::SSL::VERIFY_NONE &&
+      ssl_params[:verify_mode] != ::OpenSSL::SSL::VERIFY_NONE &&
         context.verify_hostname
     end
+
+    CONTEXT_CACHE_MODE =
+      ::OpenSSL::SSL::SSLContext::SESSION_CACHE_CLIENT |
+        ::OpenSSL::SSL::SSLContext::SESSION_CACHE_NO_INTERNAL_STORE
   end
 
   private_constant(:SSLSocket)

data/lib/tcp-client/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 class TCPClient
-  VERSION = '0.7.0'
+  VERSION = '0.9.2'
 end

data/lib/tcp-client.rb

@@ -9,8 +9,48 @@ require_relative 'tcp-client/configuration'
 require_relative 'tcp-client/default_configuration'
 require_relative 'tcp-client/version'
 
+#
+# Client class to communicate with a server via TCP w/o SSL.
+#
+# All connect/read/write actions can be monitored to ensure that all actions
+# terminate before given time limits - or raise an exception.
+#
+# @example request to Google.com and limit network interactions to 1.5 seconds
+#   TCPClient.with_deadline(1.5, 'www.google.com:443') do |client|
+#     client.write("GET / HTTP/1.1\r\nHost: www.google.com\r\n\r\n")
+#     client.read(12)
+#   end
+#   # => "HTTP/1.1 200"
+#
+#
 class TCPClient
-  def self.open(address, configuration = Configuration.default)
+  #
+  # Creates a new instance which is connected to the server on the given
+  # `address`.
+  #
+  # If no `configuration` is given, the {.default_configuration} will be used.
+  #
+  # If an optional block is given, then the block's result is returned and the
+  # connection will be closed when the block execution ends.
+  # This can be used to create an ad-hoc connection which is garanteed to be
+  # closed.
+  #
+  # If no block is giiven the connected client instance is returned.
+  # This can be used as a shorthand to create & connect a client.
+  #
+  # @param address [Address, String, Addrinfo, Integer] the address to connect
+  #   to, see {Address#initialize} for valid formats
+  # @param configuration [Configuration] the {Configuration} to be used for
+  #   this instance
+  #
+  # @yieldparam client [TCPClient] the connected client
+  # @yieldreturn [Object] any result
+  #
+  # @return [Object, TCPClient] the block result or the connected client
+  #
+  # @see #connect
+  #
+  def self.open(address, configuration = nil)
     client = new
     client.connect(Address.new(address), configuration)
     block_given? ? yield(client) : client
@@ -18,11 +58,35 @@ class TCPClient
     client.close if block_given?
   end
 
-  def self.with_deadline(
-    timeout,
-    address,
-    configuration = Configuration.default
-  )
+  #
+  # Yields a new instance which is connected to the server on the given
+  # `address`.It limits all {#read} and {#write} actions within the block to
+  # the given time.
+  #
+  # It ensures to close the connection when the block execution ends and returns
+  # the block`s result.
+  #
+  # This can be used to create an ad-hoc connection which is garanteed to be
+  # closed and which {#read}/{#write} call sequence should not last longer than
+  # the `timeout`.
+  #
+  # If no `configuration` is given, the {.default_configuration} will be used.
+  #
+  # @param timeout [Numeric] maximum time in seconds for all {#read} and
+  #   {#write} calls within the block
+  # @param address [Address, String, Addrinfo, Integer] the address to connect
+  #   to, see {Address#initialize} for valid formats
+  # @param configuration [Configuration] the {Configuration} to be used for
+  #   this instance
+  #
+  # @yieldparam client [TCPClient] the connected client
+  # @yieldreturn [Object] any result
+  #
+  # @return [Object] the block's result
+  #
+  # @see #with_deadline
+  #
+  def self.with_deadline(timeout, address, configuration = nil)
     client = nil
     raise(NoBlockGivenError) unless block_given?
     address = Address.new(address)
@@ -34,38 +98,142 @@ class TCPClient
     client&.close
   end
 
-  attr_reader :address, :configuration
+  #
+  # @return [Address] the address used by this client instance
+  #
+  attr_reader :address
 
-  def initialize
-    @socket = @address = @deadline = @configuration = nil
+  #
+  # @return [Configuration] the configuration used by this client instance
+  #
+  attr_reader :configuration
+
+  #
+  # @!parse attr_reader :closed?
+  # @return [Boolean] true when the connection is closed, false when connected
+  #
+  def closed?
+    @socket.nil? || @socket.closed?
   end
 
-  def to_s
-    @address&.to_s || ''
+  #
+  # Close the current connection if connected.
+  #
+  # @return [self]
+  #
+  def close
+    @socket&.close
+    self
+  rescue *NETWORK_ERRORS
+    self
+  ensure
+    @socket = @deadline = nil
   end
 
-  def connect(address, configuration, timeout: nil, exception: nil)
+  #
+  # Establishes a new connection to a given `address`.
+  #
+  # It accepts a connection-specific configuration or uses the
+  # {.default_configuration}. The {#configuration} used by this instance will
+  # be a copy of the configuration used for this method call. This allows to
+  # configure the behavior per connection.
+  #
+  # The optional `timeout` and `exception` parameters allow to override the
+  # `connect_timeout` and `connect_timeout_error` values.
+  #
+  # @param address [Address, String, Addrinfo, Integer] the address to connect
+  #   to, see {Address#initialize} for valid formats
+  # @param configuration [Configuration] the {Configuration} to be used for
+  #   this instance
+  # @param timeout [Numeric] maximum time in seconds to connect
+  # @param exception [Class] exception class to be used when the connect timeout
+  #   reached
+  #
+  # @return [self]
+  #
+  # @raise {NoOpenSSLError} if SSL should be used but OpenSSL is not avail
+  #
+  # @see NetworkError
+  #
+  def connect(address, configuration = nil, timeout: nil, exception: nil)
     close if @socket
-    raise(NoOpenSSLError) if configuration.ssl? && !defined?(SSLSocket)
     @address = Address.new(address)
-    @configuration = configuration.dup.freeze
+    @configuration = (configuration || Configuration.default).dup
+    raise(NoOpenSSLError) if @configuration.ssl? && !defined?(SSLSocket)
     @socket = create_socket(timeout, exception)
     self
   end
 
-  def close
-    @socket&.close
-    self
-  rescue IOError
+  #
+  # Flush all internal buffers (write all through).
+  #
+  # @return [self]
+  #
+  def flush
+    stem_errors { @socket&.flush }
     self
-  ensure
-    @socket = @deadline = nil
   end
 
-  def closed?
-    @socket.nil? || @socket.closed?
+  #
+  # Read the given `nbytes` or the next available buffer from server.
+  #
+  # The optional `timeout` and `exception` parameters allow to override the
+  # `read_timeout` and `read_timeout_error` values of the used {#configuration}.
+  #
+  # @param nbytes [Integer] the number of bytes to read
+  # @param timeout [Numeric] maximum time in seconds to read
+  # @param exception [Class] exception class to be used when the read timeout
+  #   reached
+  #
+  # @return [String] the read buffer
+  #
+  # @raise [NotConnectedError] if {#connect} was not called before
+  #
+  # @see NetworkError
+  #
+  def read(nbytes = nil, timeout: nil, exception: nil)
+    raise(NotConnectedError) if closed?
+    deadline = create_deadline(timeout, configuration.read_timeout)
+    return stem_errors { @socket.read(nbytes) } unless deadline.valid?
+    exception ||= configuration.read_timeout_error
+    stem_errors(exception) do
+      @socket.read_with_deadline(nbytes, deadline, exception)
+    end
   end
 
+  #
+  # @return [String] the currently used address as text.
+  #
+  # @see Address#to_s
+  #
+  def to_s
+    @address&.to_s || ''
+  end
+
+  #
+  # Execute a block with a given overall time limit.
+  #
+  # When you like to ensure that a complete {#read}/{#write} communication
+  # sequence with the server is finished before a given amount of time you use
+  # this method.
+  #
+  # @example ensure to send SMTP welcome message and receive a 4 byte answer
+  #   answer = client.with_deadline(2.5) do
+  #     client.write('HELO')
+  #     client.read(4)
+  #   end
+  #   # answer is EHLO when server speaks fluent SMPT
+  #
+  # @param timeout [Numeric] maximum time in seconds for all {#read} and
+  #   {#write} calls within the block
+  #
+  # @yieldparam client [TCPClient] self
+  # @yieldreturn [Object] any result
+  #
+  # @return [Object] the block`s result
+  #
+  # @raise [NoBlockGivenError] if the block is missing
+  #
   def with_deadline(timeout)
     previous_deadline = @deadline
     raise(NoBlockGivenError) unless block_given?
@@ -76,29 +244,34 @@ class TCPClient
     @deadline = previous_deadline
   end
 
-  def read(nbytes = nil, timeout: nil, exception: nil)
-    raise(NotConnectedError) if closed?
-    deadline = create_deadline(timeout, configuration.read_timeout)
-    return @socket.read(nbytes) unless deadline.valid?
-    exception ||= configuration.read_timeout_error
-    @socket.read_with_deadline(nbytes, deadline, exception)
-  end
-
-  def write(*msg, timeout: nil, exception: nil)
+  #
+  # Write the given `messages` to the server.
+  #
+  # The optional `timeout` and `exception` parameters allow to override the
+  # `write_timeout` and `write_timeout_error` values of the used
+  # {#configuration}.
+  #
+  # @param messages [String] one or more messages to write
+  # @param timeout [Numeric] maximum time in seconds to write
+  # @param exception [Class] exception class to be used when the write timeout
+  #   reached
+  #
+  # @return [Integer] bytes written
+  #
+  # @raise [NotConnectedError] if {#connect} was not called before
+  #
+  def write(*messages, timeout: nil, exception: nil)
     raise(NotConnectedError) if closed?
     deadline = create_deadline(timeout, configuration.write_timeout)
-    return @socket.write(*msg) unless deadline.valid?
+    return stem_errors { @socket.write(*messages) } unless deadline.valid?
     exception ||= configuration.write_timeout_error
-    msg.sum do |chunk|
-      @socket.write_with_deadline(chunk.b, deadline, exception)
+    stem_errors(exception) do
+      messages.sum do |chunk|
+        @socket.write_with_deadline(chunk.b, deadline, exception)
+      end
     end
   end
 
-  def flush
-    @socket&.flush
-    self
-  end
-
   private
 
   def create_deadline(timeout, default)
@@ -108,8 +281,34 @@ class TCPClient
   def create_socket(timeout, exception)
     deadline = create_deadline(timeout, configuration.connect_timeout)
     exception ||= configuration.connect_timeout_error
-    @socket = TCPSocket.new(address, configuration, deadline, exception)
-    return @socket unless configuration.ssl?
-    SSLSocket.new(@socket, address, configuration, deadline, exception)
+    stem_errors(exception) do
+      @socket = TCPSocket.new(address, configuration, deadline, exception)
+      return @socket unless configuration.ssl?
+      SSLSocket.new(@socket, address, configuration, deadline, exception)
+    end
   end
+
+  def stem_errors(except = nil)
+    yield
+  rescue *NETWORK_ERRORS => e
+    raise unless configuration.normalize_network_errors
+    (except && e.is_a?(except)) ? raise : raise(NetworkError, e)
+  end
+
+  NETWORK_ERRORS =
+    [
+      Errno::EADDRNOTAVAIL,
+      Errno::ECONNABORTED,
+      Errno::ECONNREFUSED,
+      Errno::ECONNRESET,
+      Errno::EHOSTUNREACH,
+      Errno::EINVAL,
+      Errno::ENETUNREACH,
+      Errno::EPIPE,
+      IOError,
+      SocketError
+    ].tap do |errors|
+      errors << ::OpenSSL::SSL::SSLError if defined?(::OpenSSL::SSL::SSLError)
+    end.freeze
+  private_constant(:NETWORK_ERRORS)
 end

data/rakefile.rb

@@ -1,16 +1,12 @@
 # frozen_string_literal: true
 
 require 'rake/clean'
-require 'rake/testtask'
 require 'bundler/gem_tasks'
+require 'rspec/core/rake_task'
+require 'yard'
 
 $stdout.sync = $stderr.sync = true
-
-CLOBBER << 'prj'
-
+CLOBBER << 'prj' << 'doc' << '.yardoc'
 task(:default) { exec('rake --tasks') }
-
-Rake::TestTask.new(:test) do |task|
-  task.pattern = 'test/**/*_test.rb'
-  task.warning = task.verbose = true
-end
+RSpec::Core::RakeTask.new { |task| task.ruby_opts = %w[-w] }
+YARD::Rake::YardocTask.new { |task| task.stats_options = %w[--list-undoc] }

data/sample/google_ssl.rb

@@ -2,23 +2,24 @@
 
 require_relative '../lib/tcp-client'
 
-# create a configuration.
-# - use TLS 1.2
+# create a configuration:
 # - don't use internal buffering
+# - use TLS 1.2 or TLS 1.3
 cfg =
   TCPClient::Configuration.create(
     buffered: false,
     ssl_params: {
-      ssl_version: :TLSv1_2
+      min_version: :TLS1_2,
+      max_version: :TLS1_3
     }
   )
 
-# request to Google:
-# - limit all interactions to 0.5 seconds
+# request to Google.com:
+# - limit all network interactions to 1.5 seconds
 # - use the Configuration cfg
 # - send a simple HTTP get request
 # - read 12 byte: "HTTP/1.1 " + 3 byte HTTP status code
-TCPClient.with_deadline(0.5, 'www.google.com:443', cfg) do |client|
+TCPClient.with_deadline(1.5, 'www.google.com:443', cfg) do |client|
   p client.write("GET / HTTP/1.1\r\nHost: www.google.com\r\n\r\n")
   p client.read(12)
 end

data/spec/helper.rb

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+
+require 'rspec/core'
+require_relative '../lib/tcp-client'
+
+$stdout.sync = $stderr.sync = true
+
+RSpec.configure do |config|
+  config.disable_monkey_patching!
+  config.warnings = true
+  config.order = :random
+end