126 lines
3.5 KiB
Go
126 lines
3.5 KiB
Go
// Copyright 2018 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 proc
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"gvisor.dev/gvisor/pkg/sentry/context"
|
|
"gvisor.dev/gvisor/pkg/sentry/inet"
|
|
"gvisor.dev/gvisor/pkg/sentry/usermem"
|
|
)
|
|
|
|
func TestQuerySendBufferSize(t *testing.T) {
|
|
ctx := context.Background()
|
|
s := inet.NewTestStack()
|
|
s.TCPSendBufSize = inet.TCPBufferSize{100, 200, 300}
|
|
tmi := &tcpMemInode{s: s, dir: tcpWMem}
|
|
tmf := &tcpMemFile{tcpMemInode: tmi}
|
|
|
|
buf := make([]byte, 100)
|
|
dst := usermem.BytesIOSequence(buf)
|
|
n, err := tmf.Read(ctx, nil, dst, 0)
|
|
if err != nil {
|
|
t.Fatalf("Read failed: %v", err)
|
|
}
|
|
|
|
if got, want := string(buf[:n]), "100\t200\t300\n"; got != want {
|
|
t.Fatalf("Bad string: got %v, want %v", got, want)
|
|
}
|
|
}
|
|
|
|
func TestQueryRecvBufferSize(t *testing.T) {
|
|
ctx := context.Background()
|
|
s := inet.NewTestStack()
|
|
s.TCPRecvBufSize = inet.TCPBufferSize{100, 200, 300}
|
|
tmi := &tcpMemInode{s: s, dir: tcpRMem}
|
|
tmf := &tcpMemFile{tcpMemInode: tmi}
|
|
|
|
buf := make([]byte, 100)
|
|
dst := usermem.BytesIOSequence(buf)
|
|
n, err := tmf.Read(ctx, nil, dst, 0)
|
|
if err != nil {
|
|
t.Fatalf("Read failed: %v", err)
|
|
}
|
|
|
|
if got, want := string(buf[:n]), "100\t200\t300\n"; got != want {
|
|
t.Fatalf("Bad string: got %v, want %v", got, want)
|
|
}
|
|
}
|
|
|
|
var cases = []struct {
|
|
str string
|
|
initial inet.TCPBufferSize
|
|
final inet.TCPBufferSize
|
|
}{
|
|
{
|
|
str: "",
|
|
initial: inet.TCPBufferSize{1, 2, 3},
|
|
final: inet.TCPBufferSize{1, 2, 3},
|
|
},
|
|
{
|
|
str: "100\n",
|
|
initial: inet.TCPBufferSize{1, 100, 200},
|
|
final: inet.TCPBufferSize{100, 100, 200},
|
|
},
|
|
{
|
|
str: "100 200 300\n",
|
|
initial: inet.TCPBufferSize{1, 2, 3},
|
|
final: inet.TCPBufferSize{100, 200, 300},
|
|
},
|
|
}
|
|
|
|
func TestConfigureSendBufferSize(t *testing.T) {
|
|
ctx := context.Background()
|
|
s := inet.NewTestStack()
|
|
for _, c := range cases {
|
|
s.TCPSendBufSize = c.initial
|
|
tmi := &tcpMemInode{s: s, dir: tcpWMem}
|
|
tmf := &tcpMemFile{tcpMemInode: tmi}
|
|
|
|
// Write the values.
|
|
src := usermem.BytesIOSequence([]byte(c.str))
|
|
if n, err := tmf.Write(ctx, nil, src, 0); n != int64(len(c.str)) || err != nil {
|
|
t.Errorf("Write, case = %q: got (%d, %v), wanted (%d, nil)", c.str, n, err, len(c.str))
|
|
}
|
|
|
|
// Read the values from the stack and check them.
|
|
if s.TCPSendBufSize != c.final {
|
|
t.Errorf("TCPSendBufferSize, case = %q: got %v, wanted %v", c.str, s.TCPSendBufSize, c.final)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestConfigureRecvBufferSize(t *testing.T) {
|
|
ctx := context.Background()
|
|
s := inet.NewTestStack()
|
|
for _, c := range cases {
|
|
s.TCPRecvBufSize = c.initial
|
|
tmi := &tcpMemInode{s: s, dir: tcpRMem}
|
|
tmf := &tcpMemFile{tcpMemInode: tmi}
|
|
|
|
// Write the values.
|
|
src := usermem.BytesIOSequence([]byte(c.str))
|
|
if n, err := tmf.Write(ctx, nil, src, 0); n != int64(len(c.str)) || err != nil {
|
|
t.Errorf("Write, case = %q: got (%d, %v), wanted (%d, nil)", c.str, n, err, len(c.str))
|
|
}
|
|
|
|
// Read the values from the stack and check them.
|
|
if s.TCPRecvBufSize != c.final {
|
|
t.Errorf("TCPRecvBufferSize, case = %q: got %v, wanted %v", c.str, s.TCPRecvBufSize, c.final)
|
|
}
|
|
}
|
|
}
|