netbird/gvisor
0
0
Fork 0
mirror of https://github.com/netbirdio/gvisor.git synced 2026-05-22 17:12:49 -07:00
Code Issues Packages Projects Releases Wiki Activity

roll forward cl/558267882

Browse Source 
This change is correct regardless of the rolled-back changes to IP address
selection.

PiperOrigin-RevId: 572061466
This commit is contained in:
Kevin Krakauer
2023-10-09 15:49:17 -07:00
committed by gVisor bot
parent 79da11aa57
commit b3de4851a6
1 changed files with 20 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/tcpip/transport/udp/udp_test.go
+20 -9
View File
@@ -725,20 +725,31 @@ func TestDualWriteConnectedToV4Mapped(t *testing.T) {
}
func TestPreflightBindsEndpoint(t *testing.T) {
protocols := map[string]tcpip.NetworkProtocolNumber{
"ipv4": ipv4.ProtocolNumber,
"ipv6": ipv6.ProtocolNumber,
tcs := []struct {
name string
proto tcpip.NetworkProtocolNumber
flow context.TestFlow
}{
{
name: "ipv4",
proto: ipv4.ProtocolNumber,
flow: context.UnicastV4,
},
{
name: "ipv6",
proto: ipv6.ProtocolNumber,
flow: context.UnicastV6,
},
}
for name, ipProtocolNumber := range protocols {
t.Run(name, func(t *testing.T) {
for _, tc := range tcs {
t.Run(tc.name, func(t *testing.T) {
c := context.New(t, []stack.TransportProtocolFactory{udp.NewProtocol})
defer c.Cleanup()
c.CreateEndpoint(ipProtocolNumber, udp.ProtocolNumber)
c.CreateEndpoint(tc.proto, udp.ProtocolNumber)
flow := context.UnicastV6
h := flow.MakeHeader4Tuple(context.Outgoing)
writeDstAddr := flow.MapAddrIfApplicable(h.Dst.Addr)
h := tc.flow.MakeHeader4Tuple(context.Outgoing)
writeDstAddr := tc.flow.MapAddrIfApplicable(h.Dst.Addr)
writeOpts := tcpip.WriteOptions{
To: &tcpip.FullAddress{Addr: writeDstAddr, Port: h.Dst.Port},
}