io_request 2.0.0 → 2.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9a78b709d9b064cd44737d76a24c27f0df37e9f0d5897829043e9d884287219d
-  data.tar.gz: a2350c480d0f9cc883526c47021505bfd511ffa12622436ac5bafbe4d050c6df
+  metadata.gz: 3129d25d05cbc0292eb186416db04f305f3c8fa09c525daafe1926ea2c833a76
+  data.tar.gz: 417380ff03d49890bd4161a1db3811c3e4c105c5357082579801318aadb828cf
 SHA512:
-  metadata.gz: b1cc3c89258eaa0d4511fdcd1dc17cae4b82061ebdb747732a8cc60467a9a39c3414a337f6f85acc4dd6768f16bc45f9fe037405e5e7aeb19f0723deeb01759a
-  data.tar.gz: eb6918fb66bbaec142dd88abf4c00b69d5204fb88b03f00532e52d12d8a03822275efb3cac4584177a8f9ab40bc549bc5dff09977410f3cf6f99435f594e41ad
+  metadata.gz: e20f44b5bd79b67d7b0d33c5edd16ba34321f1815bf3641c7bf986798e9bfc0ed3495015625b58bc541385fb4b6f35e16c080eac1e9c92404b35db255c815d23
+  data.tar.gz: 3638cb26446495bf95154d1e3d836b2b5660023491bd53b45175fb8701012d5de6fc3e5832381f4a0fe165b3858bc52056f8087c8a8026f73c8f8c096e5683ae

data/Gemfile

@@ -11,3 +11,7 @@ gem 'json', '~> 2.3'
 gem 'timeout', '~> 0.1.0'
 
 gem 'logger', '~> 1.4'
+
+gem 'openssl', '~> 2.2'
+
+gem 'pry', '~> 0.13.1'

data/bin/console

@@ -4,12 +4,5 @@
 require 'bundler/setup'
 require 'io_request'
 
-# You can add fixtures and/or initialization code here to make experimenting
-# with your gem easier. You can also use a different console, if you like.
-
-# (If you use this, don't forget to add pry to your Gemfile!)
-# require "pry"
-# Pry.start
-
-require 'irb'
-IRB.start(__FILE__)
+require 'pry'
+Pry.start(__FILE__)

data/lib/io_request.rb

@@ -1,6 +1,10 @@
 # frozen_string_literal: true
 
+# Main module.
+module IORequest; end
+
 require_relative 'io_request/version'
+require_relative 'io_request/logging'
 require_relative 'io_request/utility/multi_thread'
 require_relative 'io_request/utility/with_id'
 require_relative 'io_request/utility/with_prog_name'
@@ -8,23 +12,3 @@ require_relative 'io_request/utility/with_prog_name'
 require_relative 'io_request/authorizer'
 require_relative 'io_request/message'
 require_relative 'io_request/client'
-
-require 'logger'
-
-# Main module.
-module IORequest
-  # @return [Logger]
-  def self.logger
-    @@logger ||= Logger.new( # rubocop:disable Style/ClassVars
-      STDOUT,
-      formatter: proc do |severity, datetime, progname, msg|
-        "[#{datetime}] #{severity} - #{progname}:\t #{msg}\n"
-      end
-    )
-  end
-
-  # @param new_logger [Logger]
-  def self.logger=(new_logger)
-    @@logger = new_logger # rubocop:disable Style/ClassVars
-  end
-end

data/lib/io_request/connection/ssl_sockets.rb

@@ -0,0 +1,155 @@
+# frozen_string_literal: true
+
+require_relative '../../io_request'
+
+require 'socket'
+require 'openssl'
+
+module IORequest
+  # Connection via SSL sockets
+  module SSLSockets
+    # SSL socket server.
+    class Server
+      include Utility::MultiThread
+
+      # Initalize new server.
+      # @param port [Integer] port of server.
+      # @param authorizer [Authorizer]
+      # @param certificate [String]
+      # @param key [String]
+      def initialize(
+        port: 8000,
+        authorizer: Authorizer.empty,
+        certificate: nil,
+        key: nil,
+        &requests_handler
+      )
+        @port = port
+        @authorizer = authorizer
+        @requests_handler = requests_handler
+
+        initialize_ssl_context(certificate, key)
+      end
+
+      # @return [Array<IORequest::Client>]
+      attr_reader :clients
+
+      # Start server.
+      def start
+        @clients = []
+
+        @server = TCPServer.new(@port)
+
+        @accept_thread = in_thread(name: 'accept_thr') { accept_loop }
+      end
+
+      # Fully stop server.
+      def stop
+        @clients.each(&:close)
+        @clients = []
+
+        @server.close
+        @server = nil
+
+        @accept_thread&.kill
+        @accept_thread = nil
+      end
+
+      private
+
+      def initialize_ssl_context(certificate, key)
+        @ctx = OpenSSL::SSL::SSLContext.new
+        @ctx.cert = OpenSSL::X509::Certificate.new certificate
+        @ctx.key = OpenSSL::PKey::RSA.new key
+        @ctx.ssl_version = :TLSv1_2
+      end
+
+      def accept_loop
+        while (socket = @server.accept)
+          handle_socket(socket)
+        end
+      rescue
+        stop
+      end
+
+      def handle_socket(socket)
+        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
+        end
+      rescue StandardError => e
+        IORequest.warn "Unknown error while handling sockets: #{e}"
+      end
+    end
+
+    # SSL socket client.
+    class Client
+      # Initialize new client.
+      # @param authorizer [Authorizer]
+      # @param certificate [String]
+      # @param key [String]
+      def initialize(
+        authorizer: Authorizer.empty,
+        certificate: nil,
+        key: nil,
+        &requests_handler
+      )
+        @authorizer = authorizer
+        @requests_handler = requests_handler
+
+        initialize_ssl_context(certificate, key)
+      end
+
+      # Connect to server.
+      # @param host [String] host of server.
+      # @param port [Integer] port of server.
+      def connect(host = 'localhost', port = 8000)
+        socket = TCPSocket.new(host, port)
+
+        ssl_socket = OpenSSL::SSL::SSLSocket.new(socket, @ctx)
+        ssl_socket.sync_close = true
+        ssl_socket.connect
+
+        @client = IORequest::Client.new authorizer: @authorizer
+        begin
+          @client.open read_write: ssl_socket
+          @client.respond(&@requests_handler)
+        rescue StandardError
+          IORequest.debug "Failed to open client: #{e}"
+          ssl_socket.close
+          @client = nil
+        end
+      end
+
+      # Closes connection to server.
+      def disconnect
+        return unless defined?(@client) && !@client.nil?
+
+        @client.close
+        @client = nil
+      end
+
+      # Wrapper over {IORequest::Client#request}
+      def request(*args, **options, &block)
+        @client.request(*args, **options, &block)
+      end
+
+      private
+
+      def initialize_ssl_context(certificate, key)
+        @ctx = OpenSSL::SSL::SSLContext.new
+        @ctx.cert = OpenSSL::X509::Certificate.new certificate
+        @ctx.key = OpenSSL::PKey::RSA.new key
+        @ctx.ssl_version = :TLSv1_2
+      end
+    end
+  end
+end

data/lib/io_request/logging.rb

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+require 'logger'
+
+module IORequest
+  # @return [Logger]
+  def self.logger
+    @@logger ||= Logger.new( # rubocop:disable Style/ClassVars
+      STDOUT,
+      formatter: proc do |severity, datetime, progname, msg|
+        "[#{datetime}] #{severity} - #{progname}:\t #{msg}\n"
+      end
+    )
+  end
+
+  # @param new_logger [Logger]
+  def self.logger=(new_logger)
+    @@logger = new_logger # rubocop:disable Style/ClassVars
+  end
+end

data/lib/io_request/message.rb

@@ -10,9 +10,10 @@ module IORequest
     # Create new message.
     # @param data [Hash]
     # @param type [Symbol] one of {TYPES} member.
-    # @param id [Utility::ExtendedID, String, nil] only should be filled if message is received from outside.
-    # @param to [Utility::ExtendedID, String, nil] if message is response, it should include integer
-    #   of original request.
+    # @param id [Utility::ExtendedID, String, nil] only should be filled if
+    #   message is received from outside.
+    # @param to [Utility::ExtendedID, String, nil] if message is response, it
+    #   should include integer of original request.
     def initialize(data, type: :request, id: nil, to: nil)
       @data = data
       @type = type

data/lib/io_request/version.rb

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

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: io_request
 version: !ruby/object:Gem::Version
-  version: 2.0.0
+  version: 2.1.0
 platform: ruby
 authors:
 - Fizvlad
@@ -110,11 +110,12 @@ files:
 - README.md
 - Rakefile
 - bin/console
-- examples/simple_example.rb
 - io_request.gemspec
 - lib/io_request.rb
 - lib/io_request/authorizer.rb
 - lib/io_request/client.rb
+- lib/io_request/connection/ssl_sockets.rb
+- lib/io_request/logging.rb
 - lib/io_request/message.rb
 - lib/io_request/utility/multi_thread.rb
 - lib/io_request/utility/with_id.rb

data/examples/simple_example.rb

@@ -1,54 +0,0 @@
-# frozen_string_literal: true
-
-require 'io_request'
-
-r1, w1 = IO.pipe
-r2, w2 = IO.pipe
-
-client_1 = IORequest::Client.new read: r1, write: w2
-client_2 = IORequest::Client.new read: r2, write: w1
-
-# Use
-# Set up responders
-# Authorization
-client_2.respond type: 'auth' do |request|
-  puts "Client 2: Authorization attempt as #{request.data[:username].inspect}"
-  sleep 2 # Some processing
-  { type: 'auth_success' }
-end
-
-# Default
-client_2.respond do |request|
-  puts "Client 2: #{request.data.inspect}"
-  { type: 'success' }
-end
-
-# Send requests
-auth = false
-auth_request = client_1.request(
-  data: { type: 'auth', username: 'mymail@example.com', password: "let's pretend password hash is here" },
-  sync: true
-) do |response|
-  unless response.data[:type] == 'auth_success'
-    puts "Client 1: Authorization failed. Response: #{response.data.inspect}"
-    next
-  end
-
-  auth = true
-  # Do something
-end
-exit unless auth
-puts 'Client 1: Authorized!'
-
-message = client_1.request(
-  data: { type: 'message', message: 'Hello!' },
-  sync: true
-) do |_response|
-  puts 'Client 1: Message responded'
-end
-
-# Close
-r1.close
-w1.close
-r2.close
-w2.close