io_request 2.1.0 → 2.4.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 3129d25d05cbc0292eb186416db04f305f3c8fa09c525daafe1926ea2c833a76
-  data.tar.gz: 417380ff03d49890bd4161a1db3811c3e4c105c5357082579801318aadb828cf
+  metadata.gz: 987e7926f719a3479bf88cc0bdc76701edbc35e3d77d3c25d0f6ff8009be971c
+  data.tar.gz: 7ad37eab2d0718cdd100a9f16c7f272e538e39ae97939aa3e5858dbeb5f05413
 SHA512:
-  metadata.gz: e20f44b5bd79b67d7b0d33c5edd16ba34321f1815bf3641c7bf986798e9bfc0ed3495015625b58bc541385fb4b6f35e16c080eac1e9c92404b35db255c815d23
-  data.tar.gz: 3638cb26446495bf95154d1e3d836b2b5660023491bd53b45175fb8701012d5de6fc3e5832381f4a0fe165b3858bc52056f8087c8a8026f73c8f8c096e5683ae
+  metadata.gz: '089626846b0ce20d3cc9ae7c7dfeb9c4b457de3f3500cd75b7456a112e6f1f1304763a6288c4349b220c7b69c24ef5e4b5f34124235359184b5a7e5f34e50b66'
+  data.tar.gz: 71e6692045f5752e5bd5eda60eeea755a8ac542f465ed6a19804569f5816a3599b615986a81793a8a4248a22000696ac4a362160efc246236c701beb2dcc0383

data/lib/io_request.rb

@@ -1,7 +1,13 @@
 # frozen_string_literal: true
 
 # Main module.
-module IORequest; end
+module IORequest
+  # Client received message of zero size.
+  class ZeroSizeMessageError < RuntimeError; end
+
+  # Authorization failed.
+  class AuthorizationFailureError < RuntimeError; end
+end
 
 require_relative 'io_request/version'
 require_relative 'io_request/logging'

data/lib/io_request/client.rb

@@ -45,15 +45,17 @@ module IORequest
     # @param r [IO] object to read from.
     # @param w [IO] object to write to.
     # @param rw [IO] read-write object (replaces `r` and `w` arguments).
+    # @return [Object] data from {Authorizer}
     def open(read: nil, write: nil, read_write: nil)
       @io_r = read_write || read
       @io_w = read_write || write
 
       IORequest.logger.debug(prog_name) { 'Starting connection' }
 
-      authorization
+      auth_data = authorization
       @open = true
       @data_transition_thread = in_thread(name: 'connection') { data_transition_loop }
+      auth_data
     end
 
     def open?
@@ -63,15 +65,21 @@ module IORequest
     # Close connection.
     def close
       close_internal
+
       join_threads
-      @open = false
     end
 
     # @yieldparam [Hash]
     # @yieldreturn [Hash]
-    def respond(&block)
-      IORequest.logger.debug(prog_name) { 'Saved responder block' }
-      @responder = block
+    def on_request(&block)
+      IORequest.logger.debug(prog_name) { 'Saved on_request block' }
+      @on_request = block
+    end
+    alias respond on_request
+
+    def on_close(&block)
+      IORequest.logger.debug(prog_name) { 'Saved on_close block' }
+      @on_close = block
     end
 
     # If callback block is provided, request will be sent asynchroniously.
@@ -96,50 +104,52 @@ module IORequest
 
     def close_internal
       IORequest.logger.debug(prog_name) { 'Closing connection' }
-      begin
-        send_zero_size_request
-      rescue StandardError
-        IORequest.logger.debug(prog_name) { 'Failed to send zero-sized message. Closing anyway' }
-      end
-      stop_data_transition
+      send_zero_size_request
       close_io
-    end
-
-    def stop_data_transition
-      return unless defined?(@data_transition_thread) && !@data_transition_thread.nil?
-
-      IORequest.logger.debug(prog_name) { 'Killing data transition thread' }
-      @data_transition_thread.kill
       @data_transition_thread = nil
+      @open = false
+      @on_close&.call if defined?(@on_close)
     end
 
     def close_io
-      IORequest.logger.debug(prog_name) { 'Closing IO' }
-      @mutex_r.synchronize { @io_r&.close }
-      @mutex_w.synchronize { @io_w&.close }
+      begin
+        @io_r&.close
+      rescue StandardError => e
+        IORequest.logger.debug "Failed to close read IO: #{e}"
+      end
+      begin
+        @io_w&.close
+      rescue StandardError => e
+        IORequest.logger.debug "Failed to close write IO: #{e}"
+      end
+      IORequest.logger.debug(prog_name) { 'Closed IO streams' }
     end
 
     def authorization
-      auth_successful = @mutex_r.synchronize do
+      auth_successful = false
+      data = nil
+      @mutex_r.synchronize do
         @mutex_w.synchronize do
           IORequest.logger.debug(prog_name) { 'Authorizing new client' }
-          @authorizer.authorize(@io_r, @io_w)
+          auth_successful = @authorizer.authorize(@io_r, @io_w)
+          data = @authorizer.data
         end
       end
-      unless auth_successful
-        IORequest.logger.debug(prog_name) { 'Authorization failed' }
-        raise 'Authorization failed'
-      end
+      raise AuthorizationFailureError unless auth_successful
 
-      IORequest.logger.debug(prog_name) { "New client authorized with data #{@authorizer.data}" }
+      IORequest.logger.debug(prog_name) { "New client authorized with data #{data}" }
+      data
     end
 
     def data_transition_loop
       IORequest.logger.debug(prog_name) { 'Starting data transition loop' }
       loop do
         data_transition_iteration
+      rescue ZeroSizeMessageError
+        IORequest.logger.debug(prog_name) { 'Connection was closed from the other side' }
+        break
       rescue StandardError => e
-        IORequest.logger.debug(prog_name) { "Data transition iteration failed: #{e}" }
+        IORequest.logger.debug(prog_name) { "Data transition unknown error: #{e}" }
         break
       end
       close_internal
@@ -156,7 +166,8 @@ module IORequest
     end
 
     def handle_request(message)
-      data = @responder&.call(message.data) || {}
+      data = {}
+      data = @on_request&.call(message.data) if defined?(@on_request)
       response = Message.new(data, type: :response, to: message.id)
       send_response(response)
     end
@@ -180,6 +191,8 @@ module IORequest
         IORequest.logger.debug(prog_name) { 'Sending zero size message' }
         @io_w.write([0].pack('S'))
       end
+    rescue StandardError => e
+      IORequest.logger.debug(prog_name) { "Failed to send zero-sized message(#{e})" }
     end
 
     def send_request_and_wait_for_response(request)

data/lib/io_request/connection/ssl_sockets.rb

@@ -32,11 +32,20 @@ module IORequest
       end
 
       # @return [Array<IORequest::Client>]
-      attr_reader :clients
+      def clients
+        @clients_data.keys
+      end
+
+      # @param client [IORequest::Client]
+      # @return [Hash, nil] you are free to store anything you want in hash.
+      #   Only field you will find in it is `auth` with authenticator data.
+      def data(client)
+        @clients_data[client]
+      end
 
       # Start server.
       def start
-        @clients = []
+        @clients_data = {}
 
         @server = TCPServer.new(@port)
 
@@ -45,8 +54,8 @@ module IORequest
 
       # Fully stop server.
       def stop
-        @clients.each(&:close)
-        @clients = []
+        clients.each(&:close)
+        @clients_data.clear
 
         @server.close
         @server = nil
@@ -68,7 +77,7 @@ module IORequest
         while (socket = @server.accept)
           handle_socket(socket)
         end
-      rescue
+      rescue StandardError
         stop
       end
 
@@ -76,17 +85,21 @@ module IORequest
         ssl_socket = OpenSSL::SSL::SSLSocket.new(socket, @ctx)
         ssl_socket.accept
 
-        client = IORequest::Client.new authorizer: @authorizer
-        begin
-          client.open read_write: ssl_socket
-          client.respond(&@requests_handler)
-          @clients << client
-        rescue StandardError
-          IORequest.debug "Failed to open client: #{e}"
-          ssl_socket.close
+        handle_client(ssl_socket, IORequest::Client.new(authorizer: @authorizer))
+      rescue StandardError => e
+        IORequest.logger.warn "Unknown error while handling sockets: #{e}"
+      end
+
+      def handle_client(ssl_socket, client)
+        auth_data = client.open read_write: ssl_socket
+        client.on_request { |data| @requests_handler.call(data, client) }
+        @clients_data[client] = { auth: auth_data }
+        client.on_close do
+          @clients_data.select! { |c, _d| c.open? }
         end
       rescue StandardError => e
-        IORequest.warn "Unknown error while handling sockets: #{e}"
+        IORequest.logger.debug "Failed to open client: #{e}"
+        ssl_socket.close
       end
     end
 
@@ -105,9 +118,15 @@ module IORequest
         @authorizer = authorizer
         @requests_handler = requests_handler
 
+        @client = nil
+
         initialize_ssl_context(certificate, key)
       end
 
+      def connected?
+        !@client.nil?
+      end
+
       # Connect to server.
       # @param host [String] host of server.
       # @param port [Integer] port of server.
@@ -121,9 +140,11 @@ module IORequest
         @client = IORequest::Client.new authorizer: @authorizer
         begin
           @client.open read_write: ssl_socket
-          @client.respond(&@requests_handler)
+          @client.on_request(&@requests_handler)
+          @client.on_close do
+            @client = nil
+          end
         rescue StandardError
-          IORequest.debug "Failed to open client: #{e}"
           ssl_socket.close
           @client = nil
         end

data/lib/io_request/message.rb

@@ -81,8 +81,8 @@ module IORequest
     # @param io_r [:read]
     # @return [Message]
     def self.read_from(io_r)
-      size = io_r.read(2).unpack1('S')
-      raise '0 size received' if size.zero?
+      size = io_r.read(2)&.unpack1('S') || 0
+      raise ZeroSizeMessageError if size.zero?
 
       json_string = io_r.read(size).unpack1("a#{size}")
       msg = JSON.parse(json_string, symbolize_names: true)

data/lib/io_request/version.rb

@@ -2,5 +2,5 @@
 
 module IORequest
   # Gem version.
-  VERSION = '2.1.0'
+  VERSION = '2.4.1'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: io_request
 version: !ruby/object:Gem::Version
-  version: 2.1.0
+  version: 2.4.1
 platform: ruby
 authors:
 - Fizvlad
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2020-07-18 00:00:00.000000000 Z
+date: 2020-07-23 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler