forked from quic-go/quic-go
08cf78d2b0
The new QUIC proxy allows to listen on “:0”, which allows us to get rid of all “address already in use” errors. The constructor now takes an Opts struct, which makes configuring it easier.
79 lines
2.1 KiB
Go
79 lines
2.1 KiB
Go
package integrationtests
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"os/exec"
|
|
"strconv"
|
|
|
|
_ "github.com/lucas-clemente/quic-clients" // download clients
|
|
"github.com/lucas-clemente/quic-go/integrationtests/proxy"
|
|
"github.com/lucas-clemente/quic-go/protocol"
|
|
|
|
. "github.com/onsi/ginkgo"
|
|
. "github.com/onsi/gomega"
|
|
. "github.com/onsi/gomega/gexec"
|
|
)
|
|
|
|
var _ = Describe("Drop Proxy", func() {
|
|
BeforeEach(func() {
|
|
dataMan.GenerateData(dataLen)
|
|
})
|
|
|
|
var proxy *quicproxy.QuicProxy
|
|
|
|
runDropTest := func(dropCallback quicproxy.DropCallback, version protocol.VersionNumber) {
|
|
var err error
|
|
proxy, err = quicproxy.NewQuicProxy("localhost:0", quicproxy.Opts{
|
|
RemoteAddr: "localhost:" + port,
|
|
DropPacket: dropCallback,
|
|
})
|
|
Expect(err).ToNot(HaveOccurred())
|
|
|
|
command := exec.Command(
|
|
clientPath,
|
|
"--quic-version="+strconv.Itoa(int(version)),
|
|
"--host=127.0.0.1",
|
|
"--port="+strconv.Itoa(proxy.LocalPort()),
|
|
"https://quic.clemente.io/data",
|
|
)
|
|
|
|
session, err := Start(command, GinkgoWriter, GinkgoWriter)
|
|
Expect(err).NotTo(HaveOccurred())
|
|
defer session.Kill()
|
|
Eventually(session, 20).Should(Exit(0))
|
|
Expect(bytes.Contains(session.Out.Contents(), dataMan.GetData())).To(BeTrue())
|
|
}
|
|
|
|
AfterEach(func() {
|
|
Expect(proxy.Close()).To(Succeed())
|
|
})
|
|
|
|
for i := range protocol.SupportedVersions {
|
|
version := protocol.SupportedVersions[i]
|
|
|
|
Context(fmt.Sprintf("with quic version %d", version), func() {
|
|
Context("dropping every 4th packet after the crypto handshake", func() {
|
|
dropper := func(p protocol.PacketNumber) bool {
|
|
if p <= 10 { // don't interfere with the crypto handshake
|
|
return false
|
|
}
|
|
return p%4 == 0
|
|
}
|
|
|
|
It("gets a file when many outgoing packets are dropped", func() {
|
|
runDropTest(func(d quicproxy.Direction, p protocol.PacketNumber) bool {
|
|
return d == quicproxy.DirectionOutgoing && dropper(p)
|
|
}, version)
|
|
})
|
|
|
|
It("gets a file when many incoming packets are dropped", func() {
|
|
runDropTest(func(d quicproxy.Direction, p protocol.PacketNumber) bool {
|
|
return d == quicproxy.DirectionIncoming && dropper(p)
|
|
}, version)
|
|
})
|
|
})
|
|
})
|
|
}
|
|
})
|