Server : Apache/2.4.41 (Ubuntu) System : Linux absol.cf 5.4.0-198-generic #218-Ubuntu SMP Fri Sep 27 20:18:53 UTC 2024 x86_64 User : www-data ( 33) PHP Version : 7.4.33 Disable Function : pcntl_alarm,pcntl_fork,pcntl_waitpid,pcntl_wait,pcntl_wifexited,pcntl_wifstopped,pcntl_wifsignaled,pcntl_wifcontinued,pcntl_wexitstatus,pcntl_wtermsig,pcntl_wstopsig,pcntl_signal,pcntl_signal_get_handler,pcntl_signal_dispatch,pcntl_get_last_error,pcntl_strerror,pcntl_sigprocmask,pcntl_sigwaitinfo,pcntl_sigtimedwait,pcntl_exec,pcntl_getpriority,pcntl_setpriority,pcntl_async_signals,pcntl_unshare, Directory :  /var/lib/gems/2.5.0/gems/eventmachine-1.2.7/tests/

Upload File : current_dir [ Writeable ] document_root [ Writeable ]

require 'em_test_helper'

require 'socket'
require 'openssl'

if EM.ssl?
  class TestSslProtocols < Test::Unit::TestCase

    module Client
      def ssl_handshake_completed
        $client_handshake_completed = true
        close_connection
      end

      def unbind
        EM.stop_event_loop
      end
    end

    module Server
      def ssl_handshake_completed
        $server_handshake_completed = true
      end
    end

    module ClientAny
      include Client
      def post_init
        start_tls(:ssl_version => %w(sslv2 sslv3 tlsv1 tlsv1_1 tlsv1_2))
      end
    end

    module ClientDefault
      include Client
      def post_init
        start_tls
      end
    end

    module ClientSSLv3
      include Client
      def post_init
        start_tls(:ssl_version => %w(SSLv3))
      end
    end

    module ServerSSLv3
      include Server
      def post_init
        start_tls(:ssl_version => %w(SSLv3))
      end
    end

    module ServerTLSv1CaseInsensitive
      include Server
      def post_init
        start_tls(:ssl_version => %w(tlsv1))
      end
    end

    module ServerAny
      include Server
      def post_init
        start_tls(:ssl_version => %w(sslv2 sslv3 tlsv1 tlsv1_1 tlsv1_2))
      end
    end

    module ServerDefault
      include Server
      def post_init
        start_tls
      end
    end

    module InvalidProtocol
      include Client
      def post_init
        start_tls(:ssl_version => %w(tlsv1 badinput))
      end
    end

    def test_invalid_ssl_version
      assert_raises(RuntimeError, "Unrecognized SSL/TLS Version: badinput") do
        EM.run do
          EM.start_server("127.0.0.1", 16784, InvalidProtocol)
          EM.connect("127.0.0.1", 16784, InvalidProtocol)
        end
      end
    end

    def test_any_to_v3
      $client_handshake_completed, $server_handshake_completed = false, false
      EM.run do
        EM.start_server("127.0.0.1", 16784, ServerSSLv3)
        EM.connect("127.0.0.1", 16784, ClientAny)
      end

      assert($client_handshake_completed)
      assert($server_handshake_completed)
    end

    def test_case_insensitivity
      $client_handshake_completed, $server_handshake_completed = false, false
      EM.run do
        EM.start_server("127.0.0.1", 16784, ServerTLSv1CaseInsensitive)
        EM.connect("127.0.0.1", 16784, ClientAny)
      end

      assert($client_handshake_completed)
      assert($server_handshake_completed)
    end

    def test_v3_to_any
      $client_handshake_completed, $server_handshake_completed = false, false
      EM.run do
        EM.start_server("127.0.0.1", 16784, ServerAny)
        EM.connect("127.0.0.1", 16784, ClientSSLv3)
      end

      assert($client_handshake_completed)
      assert($server_handshake_completed)
    end

    def test_v3_to_v3
      $client_handshake_completed, $server_handshake_completed = false, false
      EM.run do
        EM.start_server("127.0.0.1", 16784, ServerSSLv3)
        EM.connect("127.0.0.1", 16784, ClientSSLv3)
      end

      assert($client_handshake_completed)
      assert($server_handshake_completed)
    end

    def test_any_to_any
      $client_handshake_completed, $server_handshake_completed = false, false
      EM.run do
        EM.start_server("127.0.0.1", 16784, ServerAny)
        EM.connect("127.0.0.1", 16784, ClientAny)
      end

      assert($client_handshake_completed)
      assert($server_handshake_completed)
    end

    def test_default_to_default
      $client_handshake_completed, $server_handshake_completed = false, false
      EM.run do
        EM.start_server("127.0.0.1", 16784, ServerDefault)
        EM.connect("127.0.0.1", 16784, ClientDefault)
      end

      assert($client_handshake_completed)
      assert($server_handshake_completed)
    end

    module ServerV3StopAfterHandshake
      def post_init
        start_tls(:ssl_version => %w(SSLv3))
      end

      def ssl_handshake_completed
        $server_handshake_completed = true
        EM.stop_event_loop
      end
    end

    module ServerTLSv1StopAfterHandshake
      def post_init
        start_tls(:ssl_version => %w(TLSv1))
      end

      def ssl_handshake_completed
        $server_handshake_completed = true
        EM.stop_event_loop
      end
    end

    def test_v3_with_external_client
      $server_handshake_completed = false
      EM.run do
        setup_timeout(2)
        EM.start_server("127.0.0.1", 16784, ServerV3StopAfterHandshake)
        EM.defer do
          sock = TCPSocket.new("127.0.0.1", 16784)
          ctx = OpenSSL::SSL::SSLContext.new
          ctx.ssl_version = :SSLv3_client
          ssl = OpenSSL::SSL::SSLSocket.new(sock, ctx)
          ssl.connect
          ssl.close rescue nil
          sock.close rescue nil
        end
      end

      assert($server_handshake_completed)
    end

    def test_tlsv1_with_external_client
      $server_handshake_completed = false
      EM.run do
        setup_timeout(2)
        EM.start_server("127.0.0.1", 16784, ServerTLSv1StopAfterHandshake)
        EM.defer do
          sock = TCPSocket.new("127.0.0.1", 16784)
          ctx = OpenSSL::SSL::SSLContext.new
          ctx.ssl_version = :TLSv1_client
          ssl = OpenSSL::SSL::SSLSocket.new(sock, ctx)
          ssl.connect
          ssl.close rescue nil
          sock.close rescue nil
        end
      end

      assert($server_handshake_completed)
    end

    def test_tlsv1_required_with_external_client
      $server_handshake_completed = false

      EM.run do
        n = 0
        EM.add_periodic_timer(0.5) do
          n += 1
          (EM.stop rescue nil) if n == 2
        end
        EM.start_server("127.0.0.1", 16784, ServerTLSv1StopAfterHandshake)
        EM.defer do
          sock = TCPSocket.new("127.0.0.1", 16784)
          ctx = OpenSSL::SSL::SSLContext.new
          ctx.ssl_version = :SSLv3_client
          ssl = OpenSSL::SSL::SSLSocket.new(sock, ctx)
          assert_raise OpenSSL::SSL::SSLError do
            ssl.connect
          end
          ssl.close rescue nil
          sock.close rescue nil
          EM.stop rescue nil
        end
      end

      assert(!$server_handshake_completed)
    end
  end
else
  warn "EM built without SSL support, skipping tests in #{__FILE__}"
end