Files
Ayush Ranjan 138e98fb7d nvproxy: Refactor DriverVersion out to nvconf package.
This allows for runsc to be able to use DriverVersion without having to depend
on the entirety of nvproxy.

PiperOrigin-RevId: 733912696
2025-03-05 16:43:03 -08:00

751 lines
26 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 cmd
import (
"context"
"fmt"
"os"
"os/exec"
"path/filepath"
"runtime"
"runtime/debug"
"strconv"
"strings"
"time"
"github.com/google/subcommands"
specs "github.com/opencontainers/runtime-spec/specs-go"
"github.com/syndtr/gocapability/capability"
"golang.org/x/sys/unix"
"gvisor.dev/gvisor/pkg/coretag"
"gvisor.dev/gvisor/pkg/cpuid"
"gvisor.dev/gvisor/pkg/fd"
"gvisor.dev/gvisor/pkg/log"
"gvisor.dev/gvisor/pkg/metric"
"gvisor.dev/gvisor/pkg/prometheus"
"gvisor.dev/gvisor/pkg/ring0"
"gvisor.dev/gvisor/pkg/sentry/devices/nvproxy/nvconf"
"gvisor.dev/gvisor/pkg/sentry/hostmm"
"gvisor.dev/gvisor/pkg/sentry/platform"
"gvisor.dev/gvisor/runsc/boot"
"gvisor.dev/gvisor/runsc/cmd/util"
"gvisor.dev/gvisor/runsc/config"
"gvisor.dev/gvisor/runsc/flag"
"gvisor.dev/gvisor/runsc/profile"
"gvisor.dev/gvisor/runsc/specutils"
)
// Note that directfsSandboxCaps is the same as caps defined in gofer.go
// except CAP_SYS_CHROOT because we don't need to chroot in directfs mode.
var (
directfsSandboxCaps = []string{
"CAP_CHOWN",
"CAP_DAC_OVERRIDE",
"CAP_DAC_READ_SEARCH",
"CAP_FOWNER",
"CAP_FSETID",
}
// directfsSandboxLinuxCaps is the minimal set of capabilities needed by the
// sandbox to operate on files in directfs mode.
directfsSandboxLinuxCaps = &specs.LinuxCapabilities{
Bounding: directfsSandboxCaps,
Effective: directfsSandboxCaps,
Permitted: directfsSandboxCaps,
}
hostnetSandboxLinuxCaps = map[capability.Cap]string{
capability.CAP_NET_ADMIN: "CAP_NET_ADMIN",
capability.CAP_NET_BIND_SERVICE: "CAP_NET_BIND_SERVICE",
capability.CAP_NET_RAW: "CAP_NET_RAW",
}
)
// Boot implements subcommands.Command for the "boot" command which starts a
// new sandbox. It should not be called directly.
type Boot struct {
// bundleDir is the directory containing the OCI spec.
bundleDir string
// specFD is the file descriptor that the spec will be read from.
specFD int
// controllerFD is the file descriptor of a stream socket for the
// control server that is donated to this process.
controllerFD int
// deviceFD is the file descriptor for the platform device file.
deviceFD int
// ioFDs is the list of FDs used to connect to FS gofers.
ioFDs intFlags
// devIoFD is the FD to connect to dev gofer.
devIoFD int
// goferFilestoreFDs are FDs to the regular files that will back the tmpfs or
// overlayfs mount for certain gofer mounts.
goferFilestoreFDs intFlags
// goferMountConfs contains information about how the gofer mounts have been
// configured. The first entry is for rootfs and the following entries are
// for bind mounts in Spec.Mounts (in the same order).
goferMountConfs boot.GoferMountConfFlags
// stdioFDs are the fds for stdin, stdout, and stderr. They must be
// provided in that order.
stdioFDs intFlags
// passFDs are mappings of user-supplied host to guest file descriptors.
passFDs fdMappings
// execFD is the host file descriptor used for program execution.
execFD int
// applyCaps determines if capabilities defined in the spec should be applied
// to the process.
applyCaps bool
// setUpChroot is set to true if the sandbox is started in an empty root.
setUpRoot bool
// cpuNum number of CPUs to create inside the sandbox.
cpuNum int
// totalMem sets the initial amount of total memory to report back to the
// container.
totalMem uint64
// totalHostMem is the total memory reported by host /proc/meminfo.
totalHostMem uint64
// userLogFD is the file descriptor to write user logs to.
userLogFD int
// startSyncFD is the file descriptor to synchronize runsc and sandbox.
startSyncFD int
// mountsFD is the file descriptor to read list of mounts after they have
// been resolved (direct paths, no symlinks). They are resolved outside the
// sandbox (e.g. gofer) and sent through this FD. When mountsFD is not
// provided, there is no cleaning required for mounts and the mounts in
// the spec can be used as is.
mountsFD int
podInitConfigFD int
sinkFDs intFlags
saveFDs intFlags
// attached is set to true to kill the sandbox process when the parent process
// terminates. This flag is set when the command execve's itself because
// parent death signal doesn't propagate through execve when uid/gid changes.
attached bool
// productName is the value to show in
// /sys/devices/virtual/dmi/id/product_name.
productName string
hostTHP boot.HostTHP
// FDs for profile data.
profileFDs profile.FDArgs
// finalMetricsFD is a file descriptor to write metric data to upon sandbox
// termination.
finalMetricsFD int
// profilingMetricsFD is a file descriptor to write Sentry metrics data to.
profilingMetricsFD int
// profilingMetricsLossy sets whether profilingMetricsFD is a lossy channel.
// If so, the format used to write to it will contain a checksum.
profilingMetricsLossy bool
// procMountSyncFD is a file descriptor that has to be closed when the
// procfs mount isn't needed anymore.
procMountSyncFD int
// syncUsernsFD is the file descriptor that has to be closed when the
// boot process should invoke setuid/setgid for root user. This is mainly
// used to synchronize rootless user namespace initialization.
syncUsernsFD int
// nvidiaDriverVersion is the Nvidia driver version on the host.
nvidiaDriverVersion string
}
// Name implements subcommands.Command.Name.
func (*Boot) Name() string {
return "boot"
}
// Synopsis implements subcommands.Command.Synopsis.
func (*Boot) Synopsis() string {
return "launch a sandbox process"
}
// Usage implements subcommands.Command.Usage.
func (*Boot) Usage() string {
return "boot [flags] <container id>\n"
}
// SetFlags implements subcommands.Command.SetFlags.
func (b *Boot) SetFlags(f *flag.FlagSet) {
f.StringVar(&b.bundleDir, "bundle", "", "required path to the root of the bundle directory")
f.BoolVar(&b.applyCaps, "apply-caps", false, "if true, apply capabilities defined in the spec to the process")
f.BoolVar(&b.setUpRoot, "setup-root", false, "if true, set up an empty root for the process")
f.IntVar(&b.cpuNum, "cpu-num", 0, "number of CPUs to create inside the sandbox")
f.IntVar(&b.procMountSyncFD, "proc-mount-sync-fd", -1, "file descriptor that has to be written to when /proc isn't needed anymore and can be unmounted")
f.IntVar(&b.syncUsernsFD, "sync-userns-fd", -1, "file descriptor used to synchronize rootless user namespace initialization.")
f.Uint64Var(&b.totalMem, "total-memory", 0, "sets the initial amount of total memory to report back to the container")
f.Uint64Var(&b.totalHostMem, "total-host-memory", 0, "total memory reported by host /proc/meminfo")
f.BoolVar(&b.attached, "attached", false, "if attached is true, kills the sandbox process when the parent process terminates")
f.StringVar(&b.productName, "product-name", "", "value to show in /sys/devices/virtual/dmi/id/product_name")
f.StringVar(&b.nvidiaDriverVersion, "nvidia-driver-version", "", "Nvidia driver version on the host")
f.StringVar(&b.hostTHP.ShmemEnabled, "host-thp-shmem-enabled", "", "value of /sys/kernel/mm/transparent_hugepage/shmem_enabled on the host")
f.StringVar(&b.hostTHP.Defrag, "host-thp-defrag", "", "value of /sys/kernel/mm/transparent_hugepage/defrag on the host")
// Open FDs that are donated to the sandbox.
f.IntVar(&b.specFD, "spec-fd", -1, "required fd with the container spec")
f.IntVar(&b.controllerFD, "controller-fd", -1, "required FD of a stream socket for the control server that must be donated to this process")
f.IntVar(&b.deviceFD, "device-fd", -1, "FD for the platform device file")
f.Var(&b.ioFDs, "io-fds", "list of image FDs and/or socket FDs to connect gofer clients. They must follow this order: root first, then mounts as defined in the spec")
f.IntVar(&b.devIoFD, "dev-io-fd", -1, "FD to connect dev gofer client")
f.Var(&b.stdioFDs, "stdio-fds", "list of FDs containing sandbox stdin, stdout, and stderr in that order")
f.Var(&b.passFDs, "pass-fd", "mapping of host to guest FDs. They must be in M:N format. M is the host and N the guest descriptor.")
f.IntVar(&b.execFD, "exec-fd", -1, "host file descriptor used for program execution.")
f.Var(&b.goferFilestoreFDs, "gofer-filestore-fds", "FDs to the regular files that will back the overlayfs or tmpfs mount if a gofer mount is to be overlaid.")
f.Var(&b.goferMountConfs, "gofer-mount-confs", "information about how the gofer mounts have been configured.")
f.IntVar(&b.userLogFD, "user-log-fd", 0, "file descriptor to write user logs to. 0 means no logging.")
f.IntVar(&b.startSyncFD, "start-sync-fd", -1, "required FD to used to synchronize sandbox startup")
f.IntVar(&b.mountsFD, "mounts-fd", -1, "mountsFD is an optional file descriptor to read list of mounts after they have been resolved (direct paths, no symlinks).")
f.IntVar(&b.podInitConfigFD, "pod-init-config-fd", -1, "file descriptor to the pod init configuration file.")
f.Var(&b.sinkFDs, "sink-fds", "ordered list of file descriptors to be used by the sinks defined in --pod-init-config.")
f.Var(&b.saveFDs, "save-fds", "ordered list of file descriptors to be used save checkpoints. Order: kernel state, page metadata, page file")
// Profiling flags.
b.profileFDs.SetFromFlags(f)
f.IntVar(&b.finalMetricsFD, "final-metrics-log-fd", -1, "file descriptor to write metrics to upon sandbox termination.")
f.IntVar(&b.profilingMetricsFD, "profiling-metrics-fd", -1, "file descriptor to write sentry profiling metrics.")
f.BoolVar(&b.profilingMetricsLossy, "profiling-metrics-fd-lossy", false, "if true, treat the sentry profiling metrics FD as lossy and write a checksum to it.")
}
// Execute implements subcommands.Command.Execute. It starts a sandbox in a
// waiting state.
func (b *Boot) Execute(_ context.Context, f *flag.FlagSet, args ...any) subcommands.ExitStatus {
if b.specFD == -1 || b.controllerFD == -1 || b.startSyncFD == -1 || f.NArg() != 1 {
f.Usage()
return subcommands.ExitUsageError
}
conf := args[0].(*config.Config)
// Set traceback level
debug.SetTraceback(conf.Traceback)
// Initialize CPUID information.
cpuid.Initialize()
// Initialize ring0 library.
ring0.InitDefault()
argOverride := make(map[string]string)
// Do these before chroot takes effect, otherwise we can't read /sys.
if len(b.productName) == 0 {
if product, err := os.ReadFile("/sys/devices/virtual/dmi/id/product_name"); err != nil {
log.Warningf("Not setting product_name: %v", err)
} else {
b.productName = strings.TrimSpace(string(product))
log.Infof("Setting product_name: %q", b.productName)
argOverride["product-name"] = b.productName
}
}
if conf.AppHugePages {
if len(b.hostTHP.ShmemEnabled) == 0 {
val, err := hostmm.ReadTransparentHugepageEnum("shmem_enabled")
if err != nil {
log.Warningf("Failed to infer --host-thp-shmem-enabled: %v", err)
} else {
b.hostTHP.ShmemEnabled = val
log.Infof("Setting host-thp-shmem-enabled: %q", b.hostTHP.ShmemEnabled)
argOverride["host-thp-shmem-enabled"] = b.hostTHP.ShmemEnabled
}
}
if len(b.hostTHP.Defrag) == 0 {
val, err := hostmm.ReadTransparentHugepageEnum("defrag")
if err != nil {
log.Warningf("Failed to infer --host-thp-defrag", err)
} else {
b.hostTHP.Defrag = val
log.Infof("Setting host-thp-defrag: %q", b.hostTHP.Defrag)
argOverride["host-thp-defrag"] = b.hostTHP.Defrag
}
}
}
if b.attached {
// Ensure this process is killed after parent process terminates when
// attached mode is enabled. In the unfortunate event that the parent
// terminates before this point, this process leaks.
if err := unix.Prctl(unix.PR_SET_PDEATHSIG, uintptr(unix.SIGKILL), 0, 0, 0); err != nil {
util.Fatalf("error setting parent death signal: %v", err)
}
}
if b.syncUsernsFD >= 0 {
syncUsernsForRootless(b.syncUsernsFD)
argOverride["sync-userns-fd"] = "-1"
}
// Get the spec from the specFD. We *must* keep this os.File alive past
// the call setCapsAndCallSelf, otherwise the FD will be closed and the
// child process cannot read it
specFile := os.NewFile(uintptr(b.specFD), "spec file")
spec, err := specutils.ReadSpecFromFile(b.bundleDir, specFile, conf)
if err != nil {
util.Fatalf("reading spec: %v", err)
}
if b.setUpRoot {
if err := setUpChroot(spec, conf); err != nil {
util.Fatalf("error setting up chroot: %v", err)
}
argOverride["setup-root"] = "false"
if !conf.Rootless {
// /proc is umounted from a forked process, because the
// current one is going to re-execute itself without
// capabilities.
cmd, w := execProcUmounter()
defer cmd.Wait()
defer w.Close()
if b.procMountSyncFD != -1 {
panic("procMountSyncFD is set")
}
b.procMountSyncFD = int(w.Fd())
argOverride["proc-mount-sync-fd"] = strconv.Itoa(b.procMountSyncFD)
// Clear FD_CLOEXEC. Regardless of b.applyCaps, this process will be
// re-executed. procMountSyncFD should remain open.
if _, _, errno := unix.RawSyscall(unix.SYS_FCNTL, w.Fd(), unix.F_SETFD, 0); errno != 0 {
util.Fatalf("error clearing CLOEXEC: %v", errno)
}
if !b.applyCaps {
// Remove the args that have already been done before calling self.
args := prepareArgs(b.Name(), f, argOverride)
// Note that we've already read the spec from the spec FD, and
// we will read it again after the exec call. This works
// because the ReadSpecFromFile function seeks to the beginning
// of the file before reading.
util.Fatalf("callSelfAsNobody(%v): %v", args, callSelfAsNobody(args))
// This prevents the specFile finalizer from running and closed
// the specFD, which we have passed to ourselves when
// re-execing.
runtime.KeepAlive(specFile)
panic("unreachable")
}
}
}
specutils.LogSpecDebug(spec, conf.OCISeccomp)
if b.applyCaps {
caps := &specs.LinuxCapabilities{}
gPlatform, err := platform.Lookup(conf.Platform)
if err != nil {
util.Fatalf("loading platform: %v", err)
}
if gPlatform.Requirements().RequiresCapSysPtrace {
// Ptrace platform requires extra capabilities.
const c = "CAP_SYS_PTRACE"
caps.Bounding = append(caps.Bounding, c)
caps.Effective = append(caps.Effective, c)
caps.Permitted = append(caps.Permitted, c)
}
if conf.DirectFS {
caps = specutils.MergeCapabilities(caps, directfsSandboxLinuxCaps)
}
if conf.Network == config.NetworkHost {
curCaps, err := capability.NewPid2(0)
if err != nil {
util.Fatalf("capability.NewPid2(0) failed: %v", err)
}
if err := curCaps.Load(); err != nil {
util.Fatalf("unable to load capabilities: %v", err)
}
addCaps := []string{}
for c, strCap := range hostnetSandboxLinuxCaps {
if c == capability.CAP_NET_RAW && !conf.EnableRaw {
continue
}
if curCaps.Get(capability.PERMITTED, c) {
addCaps = append(addCaps, strCap)
}
}
if len(addCaps) != 0 {
caps = specutils.MergeCapabilities(caps, &specs.LinuxCapabilities{
Bounding: addCaps,
Effective: addCaps,
Permitted: addCaps,
})
}
}
argOverride["apply-caps"] = "false"
// Remove the args that have already been done before calling self.
args := prepareArgs(b.Name(), f, argOverride)
// Note that we've already read the spec from the spec FD, and
// we will read it again after the exec call. This works
// because the ReadSpecFromFile function seeks to the beginning
// of the file before reading.
util.Fatalf("setCapsAndCallSelf(%v, %v): %v", args, caps, setCapsAndCallSelf(args, caps))
// This prevents the specFile finalizer from running and closed
// the specFD, which we have passed to ourselves when
// re-execing.
runtime.KeepAlive(specFile)
panic("unreachable")
}
if b.syncUsernsFD >= 0 {
// syncUsernsFD is set, but runsc hasn't been re-executed with a new UID and GID.
// We expect that setCapsAndCallSelf has to be called in this case.
panic("unreachable")
}
// Close specFile to avoid exposing it to the sandbox.
if err := specFile.Close(); err != nil {
util.Fatalf("closing specFile: %v", err)
}
// At this point we won't re-execute, so it's safe to limit via rlimits. Any
// limit >= 0 works. If the limit is lower than the current number of open
// files, then Setrlimit will succeed, and the next open will fail.
if conf.FDLimit > -1 {
rlimit := unix.Rlimit{
Cur: uint64(conf.FDLimit),
Max: uint64(conf.FDLimit),
}
switch err := unix.Setrlimit(unix.RLIMIT_NOFILE, &rlimit); err {
case nil:
case unix.EPERM:
log.Warningf("FD limit %d is higher than the current hard limit or system-wide maximum", conf.FDLimit)
default:
util.Fatalf("Failed to set RLIMIT_NOFILE: %v", err)
}
}
// When mountsFD is not provided, there is no cleaning required.
if b.mountsFD >= 0 {
// Read resolved mount list and replace the original one from the spec.
mountsFile := os.NewFile(uintptr(b.mountsFD), "mounts file")
cleanMounts, err := specutils.ReadMounts(mountsFile)
if err != nil {
mountsFile.Close()
util.Fatalf("Error reading mounts file: %v", err)
}
mountsFile.Close()
spec.Mounts = cleanMounts
}
if conf.DirectFS {
// sandbox should run with a umask of 0, because we want to preserve file
// modes exactly as sent by the sentry, which would have already applied
// the application umask.
unix.Umask(0)
}
if conf.EnableCoreTags {
if err := coretag.Enable(); err != nil {
util.Fatalf("Failed to core tag sentry: %v", err)
}
// Verify that all sentry threads are properly core tagged, and log
// current core tag.
coreTags, err := coretag.GetAllCoreTags(0)
if err != nil {
util.Fatalf("Failed read current core tags: %v", err)
}
if len(coreTags) != 1 {
util.Fatalf("Not all child threads were core tagged the same. Tags=%v", coreTags)
}
log.Infof("Core tag enabled (core tag=%d)", coreTags[0])
}
var nvidiaDriverVersion nvconf.DriverVersion
if b.nvidiaDriverVersion != "" {
nvidiaDriverVersion, err = nvconf.DriverVersionFrom(b.nvidiaDriverVersion)
if err != nil {
util.Fatalf("Failed to parse nvidia driver version: %v", err)
}
}
// Create the loader.
bootArgs := boot.Args{
ID: f.Arg(0),
Spec: spec,
Conf: conf,
ControllerFD: b.controllerFD,
Device: fd.New(b.deviceFD),
GoferFDs: b.ioFDs.GetArray(),
DevGoferFD: b.devIoFD,
StdioFDs: b.stdioFDs.GetArray(),
PassFDs: b.passFDs.GetArray(),
ExecFD: b.execFD,
GoferFilestoreFDs: b.goferFilestoreFDs.GetArray(),
GoferMountConfs: b.goferMountConfs.GetArray(),
NumCPU: b.cpuNum,
TotalMem: b.totalMem,
TotalHostMem: b.totalHostMem,
UserLogFD: b.userLogFD,
ProductName: b.productName,
PodInitConfigFD: b.podInitConfigFD,
SinkFDs: b.sinkFDs.GetArray(),
ProfileOpts: b.profileFDs.ToOpts(),
NvidiaDriverVersion: nvidiaDriverVersion,
HostTHP: b.hostTHP,
SaveFDs: b.saveFDs.GetFDs(),
}
l, err := boot.New(bootArgs)
if err != nil {
util.Fatalf("creating loader: %v", err)
}
// Fatalf exits the process and doesn't run defers.
// 'l' must be destroyed explicitly after this point!
if b.procMountSyncFD != -1 {
l.PreSeccompCallback = func() {
// Call validateOpenFDs() before umounting /proc.
validateOpenFDs(bootArgs.PassFDs)
// Umount /proc right before installing seccomp filters.
umountProc(b.procMountSyncFD)
}
}
// Prepare metrics.
// This needs to happen after the kernel is initialized (such that all metrics are registered)
// but before the start-sync file is notified, as the parent process needs to query for
// registered metrics prior to sending the start signal.
metric.Initialize()
var finalMetricsFile *os.File
if b.finalMetricsFD != -1 {
finalMetricsFile = os.NewFile(uintptr(b.finalMetricsFD), "final metrics file")
}
if b.profilingMetricsFD != -1 {
if err := metric.StartProfilingMetrics(metric.ProfilingMetricsOptions[*os.File]{
Sink: os.NewFile(uintptr(b.profilingMetricsFD), "metrics file"),
Lossy: b.profilingMetricsLossy,
Metrics: conf.ProfilingMetrics,
Rate: time.Duration(conf.ProfilingMetricsRate) * time.Microsecond,
}); err != nil {
l.Destroy()
util.Fatalf("unable to start profiling metrics: %v", err)
}
defer metric.StopProfilingMetrics()
}
// Notify the parent process the sandbox has booted (and that the controller
// is up).
startSyncFile := os.NewFile(uintptr(b.startSyncFD), "start-sync file")
buf := make([]byte, 1)
if w, err := startSyncFile.Write(buf); err != nil || w != 1 {
l.Destroy()
util.Fatalf("unable to write into the start-sync descriptor: %v", err)
}
// Closes startSyncFile because 'l.Run()' only returns when the sandbox exits.
startSyncFile.Close()
// Wait for the start signal from runsc.
l.WaitForStartSignal()
// Run the application and wait for it to finish.
if err := l.Run(); err != nil {
l.Destroy()
util.Fatalf("running sandbox: %v", err)
}
ws := l.WaitExit()
log.Infof("application exiting with %+v", ws)
waitStatus := args[1].(*unix.WaitStatus)
*waitStatus = unix.WaitStatus(ws)
if finalMetricsFile != nil {
if err := exportFinalMetrics(finalMetricsFile); err != nil {
l.Destroy()
util.Fatalf("unable to export final metrics: %v", err)
}
}
l.Destroy()
return subcommands.ExitSuccess
}
// prepareArgs returns the args that can be used to re-execute the current
// program. It manipulates the flags of the subcommands.Command identified by
// subCmdName and fSet is the flag.FlagSet of this subcommand. It applies the
// flags specified by override map. In case of conflict, flag is overridden.
//
// Postcondition: prepareArgs() takes ownership of override map.
func prepareArgs(subCmdName string, fSet *flag.FlagSet, override map[string]string) []string {
var args []string
// Add all args up until (and including) the sub command.
for _, arg := range os.Args {
args = append(args, arg)
if arg == subCmdName {
break
}
}
// Set sub command flags. Iterate through all the explicitly set flags.
fSet.Visit(func(gf *flag.Flag) {
// If a conflict is found with override, then prefer override flag.
if ov, ok := override[gf.Name]; ok {
args = append(args, fmt.Sprintf("--%s=%s", gf.Name, ov))
delete(override, gf.Name)
return
}
// Otherwise pass through the original flag.
args = append(args, fmt.Sprintf("--%s=%s", gf.Name, gf.Value))
})
// Apply remaining override flags (that didn't conflict above).
for of, ov := range override {
args = append(args, fmt.Sprintf("--%s=%s", of, ov))
}
// Add the non-flag arguments at the end.
args = append(args, fSet.Args()...)
return args
}
// execProcUmounter execute a child process that umounts /proc when the
// returned pipe is closed.
func execProcUmounter() (*exec.Cmd, *os.File) {
r, w, err := os.Pipe()
if err != nil {
util.Fatalf("error creating a pipe: %v", err)
}
defer r.Close()
cmd := exec.Command(specutils.ExePath)
cmd.Args = append(cmd.Args, "umount", "--sync-fd=3", "/proc")
cmd.ExtraFiles = append(cmd.ExtraFiles, r)
cmd.Stdin = os.Stdin
cmd.Stdout = os.Stdout
cmd.Stderr = os.Stderr
if err := cmd.Start(); err != nil {
util.Fatalf("error executing umounter: %v", err)
}
return cmd, w
}
// umountProc writes to syncFD signalling the process started by
// execProcUmounter() to umount /proc.
func umountProc(syncFD int) {
syncFile := os.NewFile(uintptr(syncFD), "procfs umount sync FD")
buf := make([]byte, 1)
if w, err := syncFile.Write(buf); err != nil || w != 1 {
util.Fatalf("unable to write into the proc umounter descriptor: %v", err)
}
syncFile.Close()
var waitStatus unix.WaitStatus
if _, err := unix.Wait4(0, &waitStatus, 0, nil); err != nil {
util.Fatalf("error waiting for the proc umounter process: %v", err)
}
if !waitStatus.Exited() || waitStatus.ExitStatus() != 0 {
util.Fatalf("the proc umounter process failed: %v", waitStatus)
}
if err := unix.Access("/proc/self", unix.F_OK); err != unix.ENOENT {
util.Fatalf("/proc is still accessible")
}
}
// validateOpenFDs checks that the sandbox process does not have any open
// directory FDs.
func validateOpenFDs(passFDs []boot.FDMapping) {
passHostFDs := make(map[int]struct{})
for _, passFD := range passFDs {
passHostFDs[passFD.Host] = struct{}{}
}
const selfFDDir = "/proc/self/fd"
if err := filepath.WalkDir(selfFDDir, func(path string, d os.DirEntry, err error) error {
if err != nil {
return err
}
if d.Type() != os.ModeSymlink {
// All entries are symlinks. Ignore the callback for fd directory itself.
return nil
}
if fdInfo, err := os.Stat(path); err != nil {
if os.IsNotExist(err) {
// Ignore FDs that are now closed. For example, the FD to selfFDDir that
// was opened by filepath.WalkDir() to read dirents.
return nil
}
return fmt.Errorf("os.Stat(%s) failed: %v", path, err)
} else if !fdInfo.IsDir() {
return nil
}
// Uh-oh. This is a directory FD.
fdNo, err := strconv.Atoi(d.Name())
if err != nil {
return fmt.Errorf("strconv.Atoi(%s) failed: %v", d.Name(), err)
}
dirLink, err := os.Readlink(path)
if err != nil {
return fmt.Errorf("os.Readlink(%s) failed: %v", path, err)
}
if _, ok := passHostFDs[fdNo]; ok {
// Passed FDs are allowed to be directories. The user must be knowing
// what they are doing. Log a warning regardless.
log.Warningf("Sandbox has access to FD %d, which is a directory for %s", fdNo, dirLink)
return nil
}
return fmt.Errorf("FD %d is a directory for %s", fdNo, dirLink)
}); err != nil {
util.Fatalf("WalkDir(%s) failed: %v", selfFDDir, err)
}
}
// exportFinalMetrics exports all metrics to the given file.
// The file is closed as part of this function.
func exportFinalMetrics(f *os.File) error {
snapshot, err := metric.GetSnapshot(metric.SnapshotOptions{})
if err != nil {
return fmt.Errorf("getting metrics snapshot: %w", err)
}
_, err = prometheus.Write(f, prometheus.ExportOptions{}, map[*prometheus.Snapshot]prometheus.SnapshotExportOptions{
snapshot: {
ExporterPrefix: "runsc_",
},
})
if err != nil {
return fmt.Errorf("writing metrics snapshot: %w", err)
}
if err = f.Close(); err != nil {
return fmt.Errorf("closing metrics snapshot file: %w", err)
}
return nil
}