tls_test.go

package vhost

import (
	"crypto/tls"
	"net"
	"testing"

	"github.com/quic-go/quic-go"
)

func TestSNI(t *testing.T) {
	var testHostname string = "foo.example.com"

	l, err := net.Listen("tcp", "127.0.0.1:12345")
	if err != nil {
		panic(err)
	}
	defer l.Close()

	go func() {
		conf := &tls.Config{ServerName: testHostname}
		conn, err := tls.Dial("tcp", "127.0.0.1:12345", conf)
		if err != nil {
			panic(err)
		}
		conn.Close()
	}()

	conn, err := l.Accept()
	if err != nil {
		panic(err)
	}
	c, err := TLS(conn)
	if err != nil {
		panic(err)
	}

	if c.Host() != testHostname {
		t.Errorf("Connection Host() is %s, expected %s", c.Host(), testHostname)
	}
}

func TestQuicSNI(t *testing.T) {
	var testHostname string = "foo.example.com"

	udpConn, err := net.ListenUDP("udp", nil)
	if err != nil {
		panic(err)
	}
	defer udpConn.Close()

	go func() {
		tlsConf := &tls.Config{
			InsecureSkipVerify: true,
			NextProtos:         []string{"quic"},
		}
		conn, err := net.ListenUDP("udp", nil)
		if err != nil {
			t.Fatal(err)
		}
		remoteAddr, err := net.ResolveUDPAddr("udp", udpConn.LocalAddr().String())
		if err != nil {
			t.Fatal(err)
		}
		ea, err := quic.Dial(conn, remoteAddr, testHostname, tlsConf, nil)
		if err != nil {
			t.Fatal(err)
		}
		s, err := ea.OpenStream()
		if err != nil {
			t.Fatal(err)
		}
		s.Close()
	}()

	c, err := QUIC(udpConn)
	if err != nil {
		t.Fatal(err)
	}

	if c.Host() != testHostname {
		t.Errorf("Connection Host() is %s, expected %s", c.Host(), testHostname)
	}
}