mirror of
https://github.com/cloudflare/cloudflared.git
synced 2025-05-11 04:46:35 +00:00
495f9fb8bd
Remove send and return methods from Funnel interface. Users of Funnel can provide their own send and return methods without wrapper to comply with the interface. Move packet router to ingress package to avoid circular dependency
79 lines
1.6 KiB
Go
79 lines
1.6 KiB
Go
//go:build darwin || linux
|
|
|
|
package ingress
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/google/gopacket/layers"
|
|
"github.com/rs/zerolog"
|
|
"github.com/stretchr/testify/require"
|
|
"golang.org/x/net/icmp"
|
|
"golang.org/x/net/ipv4"
|
|
|
|
"github.com/cloudflare/cloudflared/packet"
|
|
)
|
|
|
|
func TestFunnelIdleTimeout(t *testing.T) {
|
|
const (
|
|
idleTimeout = time.Second
|
|
echoID = 42573
|
|
startSeq = 8129
|
|
)
|
|
logger := zerolog.New(os.Stderr)
|
|
proxy, err := newICMPProxy(localhostIP, "", &logger, idleTimeout)
|
|
require.NoError(t, err)
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
|
|
proxyDone := make(chan struct{})
|
|
go func() {
|
|
proxy.Serve(ctx)
|
|
close(proxyDone)
|
|
}()
|
|
|
|
// Send a packet to register the flow
|
|
pk := packet.ICMP{
|
|
IP: &packet.IP{
|
|
Src: localhostIP,
|
|
Dst: localhostIP,
|
|
Protocol: layers.IPProtocolICMPv4,
|
|
},
|
|
Message: &icmp.Message{
|
|
Type: ipv4.ICMPTypeEcho,
|
|
Code: 0,
|
|
Body: &icmp.Echo{
|
|
ID: echoID,
|
|
Seq: startSeq,
|
|
Data: []byte(t.Name()),
|
|
},
|
|
},
|
|
}
|
|
muxer := newMockMuxer(0)
|
|
responder := packetResponder{
|
|
datagramMuxer: muxer,
|
|
}
|
|
require.NoError(t, proxy.Request(ctx, &pk, &responder))
|
|
validateEchoFlow(t, muxer, &pk)
|
|
|
|
// Send second request, should reuse the funnel
|
|
require.NoError(t, proxy.Request(ctx, &pk, &packetResponder{
|
|
datagramMuxer: nil,
|
|
}))
|
|
validateEchoFlow(t, muxer, &pk)
|
|
|
|
time.Sleep(idleTimeout * 2)
|
|
newMuxer := newMockMuxer(0)
|
|
newResponder := packetResponder{
|
|
datagramMuxer: newMuxer,
|
|
}
|
|
require.NoError(t, proxy.Request(ctx, &pk, &newResponder))
|
|
validateEchoFlow(t, newMuxer, &pk)
|
|
|
|
cancel()
|
|
<-proxyDone
|
|
}
|