/TDDBC_Yokohama2_PCUnit/Ruby/doc/ruby/ruby-1.9.2/test/openssl/test_pair.rb
Ruby | 250 lines | 222 code | 28 blank | 0 comment | 10 complexity | 6b83eed09ca943d64de1355b5526c6f0 MD5 | raw file
Possible License(s): BSD-3-Clause, GPL-2.0, AGPL-3.0, 0BSD, Unlicense
- begin
- require "openssl"
- rescue LoadError
- end
- require 'test/unit'
- if defined?(OpenSSL)
- require 'socket'
- require_relative '../ruby/ut_eof'
- module SSLPair
- def server
- host = "127.0.0.1"
- port = 0
- ctx = OpenSSL::SSL::SSLContext.new()
- ctx.ciphers = "ADH"
- tcps = TCPServer.new(host, port)
- ssls = OpenSSL::SSL::SSLServer.new(tcps, ctx)
- return ssls
- end
- def client(port)
- host = "127.0.0.1"
- ctx = OpenSSL::SSL::SSLContext.new()
- ctx.ciphers = "ADH"
- s = TCPSocket.new(host, port)
- ssl = OpenSSL::SSL::SSLSocket.new(s, ctx)
- ssl.connect
- ssl.sync_close = true
- ssl
- end
- def ssl_pair
- ssls = server
- th = Thread.new {
- ns = ssls.accept
- ssls.close
- ns
- }
- port = ssls.to_io.addr[1]
- c = client(port)
- s = th.value
- if block_given?
- begin
- yield c, s
- ensure
- c.close unless c.closed?
- s.close unless s.closed?
- end
- else
- return c, s
- end
- ensure
- if th && th.alive?
- th.kill
- th.join
- end
- end
- end
- class OpenSSL::TestEOF1 < Test::Unit::TestCase
- include TestEOF
- include SSLPair
- def open_file(content)
- s1, s2 = ssl_pair
- Thread.new { s2 << content; s2.close }
- yield s1
- end
- end
- class OpenSSL::TestEOF2 < Test::Unit::TestCase
- include TestEOF
- include SSLPair
- def open_file(content)
- s1, s2 = ssl_pair
- Thread.new { s1 << content; s1.close }
- yield s2
- end
- end
- class OpenSSL::TestPair < Test::Unit::TestCase
- include SSLPair
- def test_getc
- ssl_pair {|s1, s2|
- s1 << "a"
- assert_equal(?a, s2.getc)
- }
- end
- def test_readpartial
- ssl_pair {|s1, s2|
- s2.write "a\nbcd"
- assert_equal("a\n", s1.gets)
- assert_equal("bcd", s1.readpartial(10))
- s2.write "efg"
- assert_equal("efg", s1.readpartial(10))
- s2.close
- assert_raise(EOFError) { s1.readpartial(10) }
- assert_raise(EOFError) { s1.readpartial(10) }
- assert_equal("", s1.readpartial(0))
- }
- end
- def test_readall
- ssl_pair {|s1, s2|
- s2.close
- assert_equal("", s1.read)
- }
- end
- def test_readline
- ssl_pair {|s1, s2|
- s2.close
- assert_raise(EOFError) { s1.readline }
- }
- end
- def test_puts_meta
- ssl_pair {|s1, s2|
- begin
- old = $/
- $/ = '*'
- s1.puts 'a'
- ensure
- $/ = old
- end
- s1.close
- assert_equal("a\n", s2.read)
- }
- end
- def test_puts_empty
- ssl_pair {|s1, s2|
- s1.puts
- s1.close
- assert_equal("\n", s2.read)
- }
- end
- def test_read_nonblock
- ssl_pair {|s1, s2|
- err = nil
- assert_raise(OpenSSL::SSL::SSLError) {
- begin
- s2.read_nonblock(10)
- ensure
- err = $!
- end
- }
- assert_kind_of(IO::WaitReadable, err)
- s1.write "abc\ndef\n"
- IO.select([s2])
- assert_equal("ab", s2.read_nonblock(2))
- assert_equal("c\n", s2.gets)
- ret = nil
- assert_nothing_raised("[ruby-core:20298]") { ret = s2.read_nonblock(10) }
- assert_equal("def\n", ret)
- }
- end
- def test_write_nonblock
- ssl_pair {|s1, s2|
- n = 0
- begin
- n += s1.write_nonblock("a" * 100000)
- n += s1.write_nonblock("b" * 100000)
- n += s1.write_nonblock("c" * 100000)
- n += s1.write_nonblock("d" * 100000)
- n += s1.write_nonblock("e" * 100000)
- n += s1.write_nonblock("f" * 100000)
- rescue IO::WaitWritable
- end
- s1.close
- assert_equal(n, s2.read.length)
- }
- end
- def test_write_nonblock_with_buffered_data
- ssl_pair {|s1, s2|
- s1.write "foo"
- s1.write_nonblock("bar")
- s1.write "baz"
- s1.close
- assert_equal("foobarbaz", s2.read)
- }
- end
- def test_connect_accept_nonblock
- host = "127.0.0.1"
- port = 0
- ctx = OpenSSL::SSL::SSLContext.new()
- ctx.ciphers = "ADH"
- serv = TCPServer.new(host, port)
- ssls = OpenSSL::SSL::SSLServer.new(serv, ctx)
- port = serv.connect_address.ip_port
- sock1 = TCPSocket.new(host, port)
- sock2 = serv.accept
- serv.close
- th = Thread.new {
- s2 = OpenSSL::SSL::SSLSocket.new(sock2, ctx)
- s2.sync_close = true
- begin
- sleep 0.2
- s2.accept_nonblock
- rescue IO::WaitReadable
- IO.select([s2])
- retry
- rescue IO::WaitWritable
- IO.select(nil, [s2])
- retry
- end
- s2
- }
- sleep 0.1
- ctx = OpenSSL::SSL::SSLContext.new()
- ctx.ciphers = "ADH"
- s1 = OpenSSL::SSL::SSLSocket.new(sock1, ctx)
- begin
- sleep 0.2
- s1.connect_nonblock
- rescue IO::WaitReadable
- IO.select([s1])
- retry
- rescue IO::WaitWritable
- IO.select(nil, [s1])
- retry
- end
- s1.sync_close = true
- s2 = th.value
- s1.print "a\ndef"
- assert_equal("a\n", s2.gets)
- ensure
- serv.close if serv && !serv.closed?
- sock1.close if sock1 && !sock1.closed?
- sock2.close if sock2 && !sock2.closed?
- end
- end
- end