mirror of
https://github.com/gravitational/teleport.git
synced 2026-10-31 20:39:37 +08:00
Zac Bergquist
2ae9d770fc
* Move configuration from lib/service to lib/service/servicecfg The new servicecfg package will hold only configuration for services. This will allow other packages (like tctl and tsh) to depend on servicecfg without pulling in all of lib/service (which has a number of platform-specific details). This is the first step towards being able to build tctl for Windows. * Move PAM and BPF config into servicecfg This breaks a compile-time dependency on BPF/PAM for tctl.
117 lines
3.2 KiB
Go
117 lines
3.2 KiB
Go
/*
|
|
Copyright 2019 Gravitational, Inc.
|
|
|
|
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 bpf
|
|
|
|
import (
|
|
"io"
|
|
"os"
|
|
osexec "os/exec"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/gravitational/teleport/lib/defaults"
|
|
"github.com/gravitational/teleport/lib/service/servicecfg"
|
|
"github.com/gravitational/teleport/lib/utils"
|
|
)
|
|
|
|
// reexecInCGroupCmd is a cmd argument used to re-exec the test binary.
|
|
const reexecInCGroupCmd = "reexecCgroup"
|
|
|
|
func TestMain(m *testing.M) {
|
|
utils.InitLoggerForTests()
|
|
|
|
// Check if the re-exec was requested.
|
|
if len(os.Args) >= 3 && os.Args[1] == reexecInCGroupCmd {
|
|
// Get the command to run passed as the 3rd argument.
|
|
cmd := os.Args[2]
|
|
|
|
if err := waitAndRun(cmd); err != nil {
|
|
// Something went wrong, exit with error.
|
|
os.Exit(1)
|
|
}
|
|
|
|
// The rexec was handled and nothing bad happened.
|
|
os.Exit(0)
|
|
}
|
|
|
|
os.Exit(m.Run())
|
|
}
|
|
|
|
// TestCheckAndSetDefaults makes sure defaults are set when the user does not
|
|
// provide values for the page sizes and hard coded values (like zero or a
|
|
// specific page size) are respected when given.
|
|
func TestCheckAndSetDefaults(t *testing.T) {
|
|
var perfBufferPageCount = defaults.PerfBufferPageCount
|
|
var openPerfBufferPageCount = defaults.OpenPerfBufferPageCount
|
|
var zeroPageCount = 0
|
|
|
|
var tests = []struct {
|
|
inConfig *servicecfg.BPFConfig
|
|
outConfig *servicecfg.BPFConfig
|
|
}{
|
|
// Empty values get defaults.
|
|
{
|
|
inConfig: &servicecfg.BPFConfig{
|
|
CommandBufferSize: nil,
|
|
DiskBufferSize: nil,
|
|
NetworkBufferSize: nil,
|
|
},
|
|
outConfig: &servicecfg.BPFConfig{
|
|
CommandBufferSize: &perfBufferPageCount,
|
|
DiskBufferSize: &openPerfBufferPageCount,
|
|
NetworkBufferSize: &perfBufferPageCount,
|
|
},
|
|
},
|
|
// Values are not wiped out with defaults.
|
|
{
|
|
inConfig: &servicecfg.BPFConfig{
|
|
CommandBufferSize: &zeroPageCount,
|
|
DiskBufferSize: &zeroPageCount,
|
|
NetworkBufferSize: &perfBufferPageCount,
|
|
},
|
|
outConfig: &servicecfg.BPFConfig{
|
|
CommandBufferSize: &zeroPageCount,
|
|
DiskBufferSize: &zeroPageCount,
|
|
NetworkBufferSize: &perfBufferPageCount,
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
err := tt.inConfig.CheckAndSetDefaults()
|
|
require.NoError(t, err)
|
|
require.Equal(t, *tt.outConfig.CommandBufferSize, *tt.inConfig.CommandBufferSize)
|
|
require.Equal(t, *tt.outConfig.DiskBufferSize, *tt.inConfig.DiskBufferSize)
|
|
require.Equal(t, *tt.outConfig.NetworkBufferSize, *tt.inConfig.NetworkBufferSize)
|
|
}
|
|
}
|
|
|
|
// waitAndRun opens FD 3 and waits for at least one byte. After it runs the
|
|
// passed command and waits until returns.
|
|
func waitAndRun(cmd string) error {
|
|
waitFD := os.NewFile(3, "/proc/self/fd/3")
|
|
defer waitFD.Close()
|
|
|
|
buff := make([]byte, 1)
|
|
if _, err := waitFD.Read(buff); err != nil && err != io.EOF {
|
|
return err
|
|
}
|
|
|
|
return osexec.Command(cmd).Run()
|
|
}
|