netbird/gvisor
0
0
Fork 0
mirror of https://github.com/netbirdio/gvisor.git synced 2026-05-22 17:12:49 -07:00
Code Issues Packages Projects Releases Wiki Activity

Make vfio-pci devices mmappable backed by a host FD.

Browse Source 
PiperOrigin-RevId: 619265846
This commit is contained in:
Jing Chen
2024-03-26 12:00:42 -07:00
committed by gVisor bot
parent edbc2af9f7
commit e902007771
3 changed files with 8 additions and 57 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/sentry/devices/tpuproxy/BUILD
+1
View File
@@ -31,6 +31,7 @@ go_library(
"//pkg/seccomp",
"//pkg/sentry/arch",
"//pkg/sentry/fsimpl/eventfd",
"//pkg/sentry/fsimpl/kernfs",
"//pkg/sentry/kernel",
"//pkg/sentry/memmap",
"//pkg/sentry/vfs",
pkg/sentry/devices/tpuproxy/tpu.go
+6 -3
View File
@@ -27,6 +27,7 @@ import (
"gvisor.dev/gvisor/pkg/marshal/primitive"
"gvisor.dev/gvisor/pkg/sentry/arch"
"gvisor.dev/gvisor/pkg/sentry/fsimpl/eventfd"
"gvisor.dev/gvisor/pkg/sentry/fsimpl/kernfs"
"gvisor.dev/gvisor/pkg/sentry/kernel"
"gvisor.dev/gvisor/pkg/sentry/vfs"
"gvisor.dev/gvisor/pkg/usermem"
@@ -174,6 +175,8 @@ func (fd *tpuFD) getPciDeviceFd(t *kernel.Task, arg hostarch.Addr) (uintptr, fun
if err != nil {
return 0, cleanup, err
}
// Initialize a mapping that is backed by a host FD.
pciDevFD.CachedMappable.Init(int(hostFD))
return uintptr(newFD), func() {}, nil
}
@@ -183,10 +186,10 @@ type pciDeviceFD struct {
vfs.FileDescriptionDefaultImpl
vfs.DentryMetadataFileDescriptionImpl
vfs.NoLockFD
kernfs.CachedMappable
hostFD int32
queue waiter.Queue
memmapFile tpuFDMemmapFile
hostFD int32
queue waiter.Queue
}
// Release implements vfs.FileDescriptionImpl.Release.
pkg/sentry/devices/tpuproxy/tpu_mmap.go
+1 -54
View File
@@ -85,59 +85,6 @@ func (mf *tpuFDMemmapFile) FD() int {
// ConfigureMMap implements vfs.FileDescriptionImpl.ConfigureMMap.
func (fd *pciDeviceFD) ConfigureMMap(ctx context.Context, opts *memmap.MMapOpts) error {
fd.CachedMappable.InitFileMapperOnce()
return vfs.GenericConfigureMMap(&fd.vfsfd, fd, opts)
}
// AddMapping implements memmap.Mappable.AddMapping.
func (fd *pciDeviceFD) AddMapping(ctx context.Context, ms memmap.MappingSpace, ar hostarch.AddrRange, offset uint64, writable bool) error {
return nil
}
// RemoveMapping implements memmap.Mappable.RemoveMapping.
func (fd *pciDeviceFD) RemoveMapping(ctx context.Context, ms memmap.MappingSpace, ar hostarch.AddrRange, offset uint64, writable bool) {
}
// CopyMapping implements memmap.Mappable.CopyMapping.
func (fd *pciDeviceFD) CopyMapping(ctx context.Context, ms memmap.MappingSpace, srcAR, dstAR hostarch.AddrRange, offset uint64, writable bool) error {
return nil
}
// Translate implements memmap.Mappable.Translate.
func (fd *pciDeviceFD) Translate(ctx context.Context, required, optional memmap.MappableRange, at hostarch.AccessType) ([]memmap.Translation, error) {
return []memmap.Translation{
{
Source: optional,
File: &fd.memmapFile,
Offset: optional.Start,
Perms: at,
},
}, nil
}
// InvalidateUnsavable implements memmap.Mappable.InvalidateUnsavable.
func (fd *pciDeviceFD) InvalidateUnsavable(ctx context.Context) error {
return nil
}
type pciDeviceFdMemmapFile struct {
fd *pciDeviceFD
}
// IncRef implements memmap.File.IncRef.
func (mf *pciDeviceFdMemmapFile) IncRef(memmap.FileRange, uint32) {
}
// DecRef implements memmap.File.DecRef.
func (mf *pciDeviceFdMemmapFile) DecRef(fr memmap.FileRange) {
}
// MapInternal implements memmap.File.MapInternal.
func (mf *pciDeviceFdMemmapFile) MapInternal(fr memmap.FileRange, at hostarch.AccessType) (safemem.BlockSeq, error) {
log.Traceback("tpuproxy: rejecting pciDeviceFdMemmapFile.MapInternal")
return safemem.BlockSeq{}, linuxerr.EINVAL
}
// FD implements memmap.File.FD.
func (mf *pciDeviceFdMemmapFile) FD() int {
return int(mf.fd.hostFD)
}