2005-03-06 00:36:11 +03:00
|
|
|
begin
|
|
|
|
require "openssl"
|
|
|
|
rescue LoadError
|
|
|
|
end
|
|
|
|
require 'test/unit'
|
|
|
|
|
|
|
|
if defined?(OpenSSL)
|
|
|
|
|
|
|
|
require 'socket'
|
|
|
|
dir = File.expand_path(__FILE__)
|
|
|
|
2.times {dir = File.dirname(dir)}
|
|
|
|
$:.replace([File.join(dir, "ruby")] | $:)
|
|
|
|
require 'ut_eof'
|
|
|
|
|
|
|
|
module SSLPair
|
|
|
|
def server
|
|
|
|
host = "127.0.0.1"
|
|
|
|
port = 0
|
|
|
|
ctx = OpenSSL::SSL::SSLContext.new()
|
2005-03-09 13:45:42 +03:00
|
|
|
ctx.ciphers = "ADH"
|
2005-03-06 00:36:11 +03:00
|
|
|
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()
|
2005-03-09 13:45:42 +03:00
|
|
|
ctx.ciphers = "ADH"
|
2005-03-06 00:36:11 +03:00
|
|
|
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
|
2007-12-15 18:00:52 +03:00
|
|
|
ensure
|
|
|
|
if th && th.alive?
|
|
|
|
th.kill
|
2009-03-06 06:56:38 +03:00
|
|
|
th.join
|
2007-12-15 18:00:52 +03:00
|
|
|
end
|
2005-03-06 00:36:11 +03:00
|
|
|
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
|
|
|
|
|
2008-12-28 19:09:59 +03:00
|
|
|
def test_read_nonblock
|
|
|
|
ssl_pair {|s1, s2|
|
2008-12-31 11:18:12 +03:00
|
|
|
err = nil
|
|
|
|
assert_raise(Errno::EWOULDBLOCK) {
|
|
|
|
begin
|
|
|
|
s2.read_nonblock(10)
|
|
|
|
ensure
|
|
|
|
err = $!
|
|
|
|
end
|
|
|
|
}
|
|
|
|
assert_match(/SSL_ERROR_WANT_READ/, err.message)
|
2008-12-28 19:09:59 +03:00
|
|
|
s1.write "abc\ndef\n"
|
2009-02-08 05:01:22 +03:00
|
|
|
IO.select([s2])
|
2008-12-28 19:09:59 +03:00
|
|
|
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
|
|
|
|
|
2005-03-06 00:36:11 +03:00
|
|
|
end
|
|
|
|
|
|
|
|
end
|