summaryrefslogtreecommitdiff
path: root/test/openssl/test_partial_record_read.rb
blob: 1899a300df1a7f3b0e1ff1d99d1302d9639d34ce (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
require_relative "utils"

if defined?(OpenSSL::TestUtils)

  class OpenSSL::TestPartialRecordRead < OpenSSL::SSLTestCase
    def test_partial_tls_record_read_nonblock
      start_server(OpenSSL::SSL::VERIFY_NONE, true, :server_proc =>
          Proc.new do |server_ctx, server_ssl|
            begin
              server_ssl.io.write("\x01") # the beginning of a TLS record
              sleep 6                     # do not finish prematurely before the read by the client is attempted
            ensure
              server_ssl.close
            end
          end
      ) do |server, port|
        sock = TCPSocket.new("127.0.0.1", port)
        ssl = OpenSSL::SSL::SSLSocket.new(sock)
        ssl.sync_close = true
        begin
          ssl.connect
          sleep 3  # wait is required for the (incomplete) TLS record to arrive at the client socket

          # Should raise a IO::WaitReadable since a full TLS record is not available for reading.
          assert_raise(IO::WaitReadable) { ssl.read_nonblock(1) }
        ensure
          ssl.close
        end
      end
    end

  end

end