From 8e4ce1b3868bb45c26e758fccc7a710e21a6f9a0 Mon Sep 17 00:00:00 2001
From: normal <normal@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>
Date: Fri, 7 Jul 2017 17:09:39 +0000
Subject: webrick: add Server Name Indication (SNI)

* lib/webrick/https.rb: servername_cb implementation.
* lib/webrick/ssl.rb: abstract servername_cb.
* test/webrick/test_https.rb: test.
  [ruby-dev:50165] [Feature #13729]
  Author: Tietew <tietew@gmail.com>

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@59281 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
---
 test/webrick/test_https.rb | 84 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 84 insertions(+)
 create mode 100644 test/webrick/test_https.rb

(limited to 'test/webrick')

diff --git a/test/webrick/test_https.rb b/test/webrick/test_https.rb
new file mode 100644
index 0000000000..30c8019c92
--- /dev/null
+++ b/test/webrick/test_https.rb
@@ -0,0 +1,84 @@
+# frozen_string_literal: false
+require "test/unit"
+require "net/http"
+require "webrick"
+require "webrick/https"
+require "webrick/utils"
+require_relative "utils"
+
+class TestWEBrickHTTPS < Test::Unit::TestCase
+  empty_log = Object.new
+  def empty_log.<<(str)
+    assert_equal('', str)
+    self
+  end
+  NoLog = WEBrick::Log.new(empty_log, WEBrick::BasicLog::WARN)
+
+  class HTTPSNITest < ::Net::HTTP
+    attr_accessor :sni_hostname
+
+    def ssl_socket_connect(s, timeout)
+      s.hostname = sni_hostname
+      super
+    end
+  end
+
+  def teardown
+    WEBrick::Utils::TimeoutHandler.terminate
+    super
+  end
+
+  def https_get(addr, port, hostname, path)
+    http = HTTPSNITest.new(addr, port)
+    http.use_ssl = true
+    http.verify_mode = OpenSSL::SSL::VERIFY_NONE
+    http.sni_hostname = hostname
+    req = Net::HTTP::Get.new(path)
+    req["Host"] = "#{hostname}:#{port}"
+    http.request(req).body
+  end
+
+  def test_sni
+    config = {
+      :ServerName => "localhost",
+      :SSLEnable => true,
+      :SSLCertName => "/CN=locahost",
+    }
+    TestWEBrick.start_httpserver(config){|server, addr, port, log|
+      server.mount_proc("/") {|req, res| res.body = "master" }
+
+      vhost_config1 = {
+        :ServerName => "vhost1",
+        :Port => port,
+        :DoNotListen => true,
+        :Logger => NoLog,
+        :AccessLog => [],
+        :SSLEnable => true,
+        :SSLCertName => "/CN=vhost1",
+      }
+      vhost1 = WEBrick::HTTPServer.new(vhost_config1)
+      vhost1.mount_proc("/") {|req, res| res.body = "vhost1" }
+      server.virtual_host(vhost1)
+
+      vhost_config2 = {
+        :ServerName => "vhost2",
+        :ServerAlias => ["vhost2alias"],
+        :Port => port,
+        :DoNotListen => true,
+        :Logger => NoLog,
+        :AccessLog => [],
+        :SSLEnable => true,
+        :SSLCertName => "/CN=vhost2",
+      }
+      vhost2 = WEBrick::HTTPServer.new(vhost_config2)
+      vhost2.mount_proc("/") {|req, res| res.body = "vhost2" }
+      server.virtual_host(vhost2)
+
+      assert_equal("master", https_get(addr, port, "localhost", "/localhost"))
+      assert_equal("master", https_get(addr, port, "unknown", "/unknown"))
+      assert_equal("vhost1", https_get(addr, port, "vhost1", "/vhost1"))
+      assert_equal("vhost2", https_get(addr, port, "vhost2", "/vhost2"))
+      assert_equal("vhost2", https_get(addr, port, "vhost2alias", "/vhost2alias"))
+    }
+  end
+end
-- 
cgit v1.2.3