mirror of
https://github.com/refraction-networking/uquic.git
synced 2025-04-02 19:57:35 +03:00
95575f5fe7
uTLS is not yet bumped to the new version, so this commit breaks the dependencies relationship by getting rid of the local replace.
99 lines
3.3 KiB
Go
99 lines
3.3 KiB
Go
package self_test
|
|
|
|
import (
|
|
"bufio"
|
|
"bytes"
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
mrand "math/rand"
|
|
"net"
|
|
|
|
quic "github.com/refraction-networking/uquic"
|
|
"github.com/refraction-networking/uquic/internal/protocol"
|
|
"github.com/refraction-networking/uquic/internal/utils"
|
|
"github.com/refraction-networking/uquic/logging"
|
|
"github.com/refraction-networking/uquic/qlog"
|
|
|
|
. "github.com/onsi/ginkgo/v2"
|
|
. "github.com/onsi/gomega"
|
|
)
|
|
|
|
var _ = Describe("Handshake tests", func() {
|
|
addTracers := func(pers protocol.Perspective, conf *quic.Config) *quic.Config {
|
|
enableQlog := mrand.Int()%3 != 0
|
|
enableCustomTracer := mrand.Int()%3 != 0
|
|
|
|
fmt.Fprintf(GinkgoWriter, "%s using qlog: %t, custom: %t\n", pers, enableQlog, enableCustomTracer)
|
|
|
|
var tracerConstructors []func(context.Context, logging.Perspective, quic.ConnectionID) logging.ConnectionTracer
|
|
if enableQlog {
|
|
tracerConstructors = append(tracerConstructors, func(_ context.Context, p logging.Perspective, connID quic.ConnectionID) logging.ConnectionTracer {
|
|
if mrand.Int()%2 == 0 { // simulate that a qlog collector might only want to log some connections
|
|
fmt.Fprintf(GinkgoWriter, "%s qlog tracer deciding to not trace connection %x\n", p, connID)
|
|
return nil
|
|
}
|
|
fmt.Fprintf(GinkgoWriter, "%s qlog tracing connection %x\n", p, connID)
|
|
return qlog.NewConnectionTracer(utils.NewBufferedWriteCloser(bufio.NewWriter(&bytes.Buffer{}), io.NopCloser(nil)), p, connID)
|
|
})
|
|
}
|
|
if enableCustomTracer {
|
|
tracerConstructors = append(tracerConstructors, func(context.Context, logging.Perspective, quic.ConnectionID) logging.ConnectionTracer {
|
|
return logging.NullConnectionTracer{}
|
|
})
|
|
}
|
|
c := conf.Clone()
|
|
c.Tracer = func(ctx context.Context, p logging.Perspective, connID quic.ConnectionID) logging.ConnectionTracer {
|
|
tracers := make([]logging.ConnectionTracer, 0, len(tracerConstructors))
|
|
for _, c := range tracerConstructors {
|
|
if tr := c(ctx, p, connID); tr != nil {
|
|
tracers = append(tracers, tr)
|
|
}
|
|
}
|
|
return logging.NewMultiplexedConnectionTracer(tracers...)
|
|
}
|
|
return c
|
|
}
|
|
|
|
for i := 0; i < 3; i++ {
|
|
It("handshakes with a random combination of tracers", func() {
|
|
if enableQlog {
|
|
Skip("This test sets tracers and won't produce any qlogs.")
|
|
}
|
|
quicClientConf := addTracers(protocol.PerspectiveClient, getQuicConfig(nil))
|
|
quicServerConf := addTracers(protocol.PerspectiveServer, getQuicConfig(nil))
|
|
|
|
serverChan := make(chan *quic.Listener)
|
|
go func() {
|
|
defer GinkgoRecover()
|
|
ln, err := quic.ListenAddr("localhost:0", getTLSConfig(), quicServerConf)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
serverChan <- ln
|
|
conn, err := ln.Accept(context.Background())
|
|
Expect(err).ToNot(HaveOccurred())
|
|
str, err := conn.OpenUniStream()
|
|
Expect(err).ToNot(HaveOccurred())
|
|
_, err = str.Write(PRData)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
Expect(str.Close()).To(Succeed())
|
|
}()
|
|
|
|
ln := <-serverChan
|
|
defer ln.Close()
|
|
|
|
conn, err := quic.DialAddr(
|
|
context.Background(),
|
|
fmt.Sprintf("localhost:%d", ln.Addr().(*net.UDPAddr).Port),
|
|
getTLSClientConfig(),
|
|
quicClientConf,
|
|
)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
defer conn.CloseWithError(0, "")
|
|
str, err := conn.AcceptUniStream(context.Background())
|
|
Expect(err).ToNot(HaveOccurred())
|
|
data, err := io.ReadAll(str)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
Expect(data).To(Equal(PRData))
|
|
})
|
|
}
|
|
})
|