mirror of
https://github.com/refraction-networking/uquic.git
synced 2025-04-04 20:57:36 +03:00
889dc81333
* qtls: migrate tests away from Ginkgo * qtls: simplify the quic.EncryptionLevel <-> tls.EncryptionLevel test * qtls: improve naming of tests
84 lines
2 KiB
Go
84 lines
2 KiB
Go
package qtls
|
|
|
|
import (
|
|
"crypto/tls"
|
|
"fmt"
|
|
"net"
|
|
"testing"
|
|
|
|
"github.com/quic-go/quic-go/internal/testdata"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestClientSessionCacheAddAndRestoreData(t *testing.T) {
|
|
ln, err := tls.Listen("tcp4", "localhost:0", testdata.GetTLSConfig())
|
|
require.NoError(t, err)
|
|
|
|
done := make(chan struct{})
|
|
go func() {
|
|
defer close(done)
|
|
|
|
for {
|
|
conn, err := ln.Accept()
|
|
if err != nil {
|
|
return
|
|
}
|
|
_, err = conn.Read(make([]byte, 10))
|
|
require.NoError(t, err)
|
|
_, err = conn.Write([]byte("foobar"))
|
|
require.NoError(t, err)
|
|
}
|
|
}()
|
|
|
|
restored := make(chan []byte, 1)
|
|
clientConf := &tls.Config{
|
|
RootCAs: testdata.GetRootCA(),
|
|
ClientSessionCache: &clientSessionCache{
|
|
wrapped: tls.NewLRUClientSessionCache(10),
|
|
getData: func(bool) []byte { return []byte("session") },
|
|
setData: func(data []byte, earlyData bool) bool {
|
|
require.False(t, earlyData) // running on top of TCP, we can only test non-0-RTT here
|
|
restored <- data
|
|
return true
|
|
},
|
|
},
|
|
}
|
|
conn, err := tls.Dial(
|
|
"tcp4",
|
|
fmt.Sprintf("localhost:%d", ln.Addr().(*net.TCPAddr).Port),
|
|
clientConf,
|
|
)
|
|
require.NoError(t, err)
|
|
_, err = conn.Write([]byte("foobar"))
|
|
require.NoError(t, err)
|
|
require.False(t, conn.ConnectionState().DidResume)
|
|
require.Len(t, restored, 0)
|
|
_, err = conn.Read(make([]byte, 10))
|
|
require.NoError(t, err)
|
|
require.NoError(t, conn.Close())
|
|
|
|
// make sure the cache can deal with nonsensical inputs
|
|
clientConf.ClientSessionCache.Put("foo", nil)
|
|
clientConf.ClientSessionCache.Put("bar", &tls.ClientSessionState{})
|
|
|
|
conn, err = tls.Dial(
|
|
"tcp4",
|
|
fmt.Sprintf("localhost:%d", ln.Addr().(*net.TCPAddr).Port),
|
|
clientConf,
|
|
)
|
|
require.NoError(t, err)
|
|
_, err = conn.Write([]byte("foobar"))
|
|
require.NoError(t, err)
|
|
require.True(t, conn.ConnectionState().DidResume)
|
|
var restoredData []byte
|
|
select {
|
|
case restoredData = <-restored:
|
|
default:
|
|
t.Fatal("no data restored")
|
|
}
|
|
require.Equal(t, []byte("session"), restoredData)
|
|
require.NoError(t, conn.Close())
|
|
|
|
require.NoError(t, ln.Close())
|
|
<-done
|
|
}
|