From 9d9e9b22c5fb509fe6501642ae3dc1572acb808e Mon Sep 17 00:00:00 2001
From: shugo <shugo@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>
Date: Sat, 5 Nov 2016 06:47:36 +0000
Subject: * lib/net/smtp.rb (tlsconnect): support timeout for TLS handshake.  
 [ruby-core:76893] [Bug #12678]

* lib/net/protocol.rb (ssl_socket_connect): new method to implement
  timeout for TLS handshake.

* lib/net/http.rb (connect): use Net::Protocol#ssl_socket_connect.

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@56576 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
---
 test/net/smtp/test_smtp.rb       | 56 ++++++++++++++++++++++++++++++++++++++++
 test/net/smtp/test_ssl_socket.rb |  5 ++++
 2 files changed, 61 insertions(+)

(limited to 'test')

diff --git a/test/net/smtp/test_smtp.rb b/test/net/smtp/test_smtp.rb
index 3bcceb6fc5..656e77dc86 100644
--- a/test/net/smtp/test_smtp.rb
+++ b/test/net/smtp/test_smtp.rb
@@ -5,6 +5,10 @@ require 'test/unit'
 
 module Net
   class TestSMTP < Test::Unit::TestCase
+    CA_FILE = File.expand_path("../imap/cacert.pem", __dir__)
+    SERVER_KEY = File.expand_path("../imap/server.key", __dir__)
+    SERVER_CERT = File.expand_path("../imap/server.crt", __dir__)
+
     class FakeSocket
       attr_reader :write_io
 
@@ -98,5 +102,57 @@ module Net
         smtp.rcptto("foo\r\nbar")
       end
     end
+
+    def test_tls_connect
+      server = TCPServer.new("127.0.0.1", 0)
+      ctx = OpenSSL::SSL::SSLContext.new
+      ctx.ca_file = CA_FILE
+      ctx.key = File.open(SERVER_KEY) { |f|
+        OpenSSL::PKey::RSA.new(f)
+      }
+      ctx.cert = File.open(SERVER_CERT) { |f|
+        OpenSSL::X509::Certificate.new(f)
+      }
+      ssl_server = OpenSSL::SSL::SSLServer.new(server, ctx)
+      begin
+        sock = nil
+        Thread.start do
+          sock = ssl_server.accept
+          sock.write("220 localhost Service ready\r\n")
+          sock.gets
+          sock.write("250 localhost\r\n")
+          sock.gets
+          sock.write("221 localhost Service closing transmission channel\r\n")
+        end
+        smtp = Net::SMTP.new("localhost", server.addr[1])
+        smtp.enable_tls
+        smtp.open_timeout = 0.1
+        smtp.start do
+        end
+      ensure
+        sock.close if sock
+        ssl_server.close
+      end
+    end
+
+    def test_tls_connect_timeout
+      server = TCPServer.new("127.0.0.1", 0)
+      begin
+        sock = nil
+        Thread.start do
+          sock = server.accept
+        end
+        smtp = Net::SMTP.new("127.0.0.1", server.addr[1])
+        smtp.enable_tls
+        smtp.open_timeout = 0.1
+        assert_raise(Net::OpenTimeout) do
+          smtp.start do
+          end
+        end
+      ensure
+        sock.close if sock
+        server.close
+      end
+    end
   end
 end
diff --git a/test/net/smtp/test_ssl_socket.rb b/test/net/smtp/test_ssl_socket.rb
index 20792190c8..354f413040 100644
--- a/test/net/smtp/test_ssl_socket.rb
+++ b/test/net/smtp/test_ssl_socket.rb
@@ -7,6 +7,11 @@ module Net
     class MySMTP < SMTP
       attr_accessor :fake_tcp, :fake_ssl
 
+      def initialize(*args)
+        super(*args)
+        @open_timeout = nil
+      end
+
       def tcp_socket address, port
         fake_tcp
       end
-- 
cgit v1.2.3