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
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
|
// Copyright 2020 The gVisor Authors.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package network
import (
"context"
"fmt"
"os"
"regexp"
"strconv"
"strings"
"testing"
"gvisor.dev/gvisor/pkg/test/dockerutil"
"gvisor.dev/gvisor/test/benchmarks/harness"
)
var h harness.Harness
func BenchmarkIperf(b *testing.B) {
// Get two machines
clientMachine, err := h.GetMachine()
if err != nil {
b.Fatalf("failed to get machine: %v", err)
}
serverMachine, err := h.GetMachine()
if err != nil {
b.Fatalf("failed to get machine: %v", err)
}
for _, bm := range []struct {
name string
clientRuntime string
serverRuntime string
}{
// We are either measuring the server or the client. The other should be
// runc. e.g. Upload sees how fast the runtime under test uploads to a native
// server.
{name: "Upload", clientRuntime: dockerutil.Runtime(), serverRuntime: "runc"},
{name: "Download", clientRuntime: "runc", serverRuntime: dockerutil.Runtime()},
} {
b.Run(bm.name, func(b *testing.B) {
// Get a container from the server and set its runtime.
ctx := context.Background()
server := serverMachine.GetContainer(ctx, b)
defer server.CleanUp(ctx)
server.Runtime = bm.serverRuntime
// Get a container from the client and set its runtime.
client := clientMachine.GetContainer(ctx, b)
defer client.CleanUp(ctx)
client.Runtime = bm.clientRuntime
// iperf serves on port 5001 by default.
port := 5001
// Start the server.
if err := server.Spawn(ctx, dockerutil.RunOpts{
Image: "benchmarks/iperf",
Ports: []int{port},
}, "iperf", "-s"); err != nil {
b.Fatalf("failed to start server with: %v", err)
}
ip, err := serverMachine.IPAddress()
if err != nil {
b.Fatalf("failed to find server ip: %v", err)
}
servingPort, err := server.FindPort(ctx, port)
if err != nil {
b.Fatalf("failed to find port %d: %v", port, err)
}
// Make sure the server is up and serving before we run.
if err := harness.WaitUntilServing(ctx, clientMachine, ip, servingPort); err != nil {
b.Fatalf("failed to wait for server: %v", err)
}
// iperf report in Kb realtime
cmd := fmt.Sprintf("iperf -f K --realtime -c %s -p %d", ip.String(), servingPort)
// Run the client.
b.ResetTimer()
for i := 0; i < b.N; i++ {
out, err := client.Run(ctx, dockerutil.RunOpts{
Image: "benchmarks/iperf",
}, strings.Split(cmd, " ")...)
if err != nil {
b.Fatalf("failed to run client: %v", err)
}
b.StopTimer()
// Parse bandwidth and report it.
bW, err := bandwidth(out)
if err != nil {
b.Fatalf("failed to parse bandwitdth from %s: %v", out, err)
}
b.ReportMetric(bW, "KBytes/sec")
b.StartTimer()
}
})
}
}
// bandwidth parses the Bandwidth number from an iperf report. A sample is below.
func bandwidth(data string) (float64, error) {
re := regexp.MustCompile(`\[\s*\d+\][^\n]+\s+(\d+\.?\d*)\s+KBytes/sec`)
match := re.FindStringSubmatch(data)
if len(match) < 1 {
return 0, fmt.Errorf("failed get bandwidth: %s", data)
}
return strconv.ParseFloat(match[1], 64)
}
func TestParser(t *testing.T) {
sampleData := `
------------------------------------------------------------
Client connecting to 10.138.15.215, TCP port 32779
TCP window size: 45.0 KByte (default)
------------------------------------------------------------
[ 3] local 10.138.15.216 port 32866 connected with 10.138.15.215 port 32779
[ ID] Interval Transfer Bandwidth
[ 3] 0.0-10.0 sec 459520 KBytes 45900 KBytes/sec
`
bandwidth, err := bandwidth(sampleData)
if err != nil || bandwidth != 45900 {
t.Fatalf("failed with: %v and %f", err, bandwidth)
}
}
func TestMain(m *testing.M) {
h.Init()
os.Exit(m.Run())
}
|
