1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
|
// Copyright 2016 The Netstack Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package ipv4_test
import (
"context"
"testing"
"time"
"gvisor.googlesource.com/gvisor/pkg/tcpip"
"gvisor.googlesource.com/gvisor/pkg/tcpip/buffer"
"gvisor.googlesource.com/gvisor/pkg/tcpip/link/channel"
"gvisor.googlesource.com/gvisor/pkg/tcpip/link/sniffer"
"gvisor.googlesource.com/gvisor/pkg/tcpip/network/ipv4"
"gvisor.googlesource.com/gvisor/pkg/tcpip/stack"
)
const stackAddr = "\x0a\x00\x00\x01"
type testContext struct {
t *testing.T
linkEP *channel.Endpoint
s *stack.Stack
}
func newTestContext(t *testing.T) *testContext {
s := stack.New([]string{ipv4.ProtocolName}, []string{ipv4.PingProtocolName})
const defaultMTU = 65536
id, linkEP := channel.New(256, defaultMTU, "")
if testing.Verbose() {
id = sniffer.New(id)
}
if err := s.CreateNIC(1, id); err != nil {
t.Fatalf("CreateNIC failed: %v", err)
}
if err := s.AddAddress(1, ipv4.ProtocolNumber, stackAddr); err != nil {
t.Fatalf("AddAddress failed: %v", err)
}
s.SetRouteTable([]tcpip.Route{{
Destination: "\x00\x00\x00\x00",
Mask: "\x00\x00\x00\x00",
Gateway: "",
NIC: 1,
}})
return &testContext{
t: t,
s: s,
linkEP: linkEP,
}
}
func (c *testContext) cleanup() {
close(c.linkEP.C)
}
func (c *testContext) loopback() {
go func() {
for pkt := range c.linkEP.C {
v := make(buffer.View, len(pkt.Header)+len(pkt.Payload))
copy(v, pkt.Header)
copy(v[len(pkt.Header):], pkt.Payload)
vv := v.ToVectorisedView([1]buffer.View{})
c.linkEP.Inject(pkt.Proto, &vv)
}
}()
}
func TestEcho(t *testing.T) {
c := newTestContext(t)
defer c.cleanup()
c.loopback()
ch := make(chan ipv4.PingReply, 1)
p := ipv4.Pinger{
Stack: c.s,
NICID: 1,
Addr: stackAddr,
Wait: 10 * time.Millisecond,
Count: 1, // one ping only
}
if err := p.Ping(context.Background(), ch); err != nil {
t.Fatalf("icmp.Ping failed: %v", err)
}
ping := <-ch
if ping.Error != nil {
t.Errorf("bad ping response: %v", ping.Error)
}
}
func TestEchoSequence(t *testing.T) {
c := newTestContext(t)
defer c.cleanup()
c.loopback()
const numPings = 3
ch := make(chan ipv4.PingReply, numPings)
p := ipv4.Pinger{
Stack: c.s,
NICID: 1,
Addr: stackAddr,
Wait: 10 * time.Millisecond,
Count: numPings,
}
if err := p.Ping(context.Background(), ch); err != nil {
t.Fatalf("icmp.Ping failed: %v", err)
}
for i := uint16(0); i < numPings; i++ {
ping := <-ch
if ping.Error != nil {
t.Errorf("i=%d bad ping response: %v", i, ping.Error)
}
if ping.SeqNumber != i {
t.Errorf("SeqNumber=%d, want %d", ping.SeqNumber, i)
}
}
}
|
