mirror of
https://github.com/hanzoai/visor.git
synced 2026-09-24 17:04:14 +00:00
Shailend ChandandgVisor bot
f739fd243b
Three nearly identical names covered three unrelated concepts (ConnectedEndpoint, connectedEndpoint, connectionedEndpoint), and the notification callbacks were named for the opposite of what they do (RecvNotify woke writers, SendNotify woke readers). - ConnectedEndpoint -> Sender, the write half of a connection (implementations queueSender, HostSender, SCMSender); the baseEndpoint field becomes connected -> peer: the type is named for what you do with it, the field for its role to the holder. - SendNotify -> NotifyDataReady, RecvNotify -> NotifyWriteSpace, CloseNotify -> NotifyStateChange; each named for the wakeup it delivers. NotifyStateChange must wake hangup-only pollers. - queue's embedded waiter queues become readerWaiters/writerWaiters, and baseEndpoint's embedded *waiter.Queue becomes a named field, WaitQueue, so wakeup call sites name their target. - writeShutdown and wrShutdown/rdShutdown -> sendShutdown/recvShutdown, the shutdown(2) directions they track. - A package doc block describes how the pieces fit together. No behavior change. PiperOrigin-RevId: 966732452
131 lines
3.5 KiB
Go
131 lines
3.5 KiB
Go
// 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 gofer
|
|
|
|
import (
|
|
"golang.org/x/sys/unix"
|
|
"gvisor.dev/gvisor/pkg/abi/linux"
|
|
"gvisor.dev/gvisor/pkg/context"
|
|
"gvisor.dev/gvisor/pkg/log"
|
|
"gvisor.dev/gvisor/pkg/sentry/socket/unix/transport"
|
|
"gvisor.dev/gvisor/pkg/syserr"
|
|
"gvisor.dev/gvisor/pkg/waiter"
|
|
)
|
|
|
|
func (i *inode) isSocket() bool {
|
|
return i.fileType() == linux.S_IFSOCK
|
|
}
|
|
|
|
func isSocketTypeSupported(sockType linux.SockType) bool {
|
|
switch sockType {
|
|
case unix.SOCK_STREAM, unix.SOCK_DGRAM, unix.SOCK_SEQPACKET:
|
|
return true
|
|
default:
|
|
return false
|
|
}
|
|
}
|
|
|
|
// endpoint is a Gofer-backed transport.BoundEndpoint.
|
|
//
|
|
// An endpoint's lifetime is the time between when filesystem.BoundEndpointAt()
|
|
// is called and either BoundEndpoint.BidirectionalConnect or
|
|
// BoundEndpoint.UnidirectionalConnect is called.
|
|
//
|
|
// +stateify savable
|
|
type endpoint struct {
|
|
// dentry is the filesystem dentry which produced this endpoint. dentry is
|
|
// not synthetic.
|
|
dentry *dentry
|
|
|
|
// path is the sentry path where this endpoint is bound.
|
|
path string
|
|
}
|
|
|
|
// BidirectionalConnect implements BoundEndpoint.BidirectionalConnect.
|
|
func (e *endpoint) BidirectionalConnect(ctx context.Context, ce transport.ConnectingEndpoint, returnConnect func(transport.Receiver, transport.Sender)) *syserr.Error {
|
|
// No lock ordering required as only the ConnectingEndpoint has a mutex.
|
|
ce.Lock()
|
|
|
|
// Check connecting state.
|
|
if ce.Connected() {
|
|
ce.Unlock()
|
|
return syserr.ErrAlreadyConnected
|
|
}
|
|
if ce.ListeningLocked() {
|
|
ce.Unlock()
|
|
return syserr.ErrInvalidEndpointState
|
|
}
|
|
|
|
c, err := e.newSender(ctx, ce.Type(), ce.WaiterQueue())
|
|
if err != nil {
|
|
ce.Unlock()
|
|
return err
|
|
}
|
|
|
|
returnConnect(c, c)
|
|
ce.Unlock()
|
|
if err := c.Init(); err != nil {
|
|
return syserr.FromError(err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// UnidirectionalConnect implements
|
|
// transport.BoundEndpoint.UnidirectionalConnect.
|
|
func (e *endpoint) UnidirectionalConnect(ctx context.Context) (transport.Sender, *syserr.Error) {
|
|
c, err := e.newSender(ctx, linux.SOCK_DGRAM, &waiter.Queue{})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if err := c.Init(); err != nil {
|
|
return nil, syserr.FromError(err)
|
|
}
|
|
|
|
// We don't need the receiver.
|
|
c.CloseRecv()
|
|
c.Release(ctx)
|
|
|
|
return c, nil
|
|
}
|
|
|
|
func (e *endpoint) newSender(ctx context.Context, sockType linux.SockType, queue *waiter.Queue) (*transport.SCMSender, *syserr.Error) {
|
|
e.dentry.inode.fs.renameMu.RLock()
|
|
hostSockFD, err := e.dentry.connect(ctx, sockType)
|
|
e.dentry.inode.fs.renameMu.RUnlock()
|
|
if err != nil {
|
|
return nil, syserr.ErrConnectionRefused
|
|
}
|
|
|
|
c, serr := transport.NewSCMSender(hostSockFD, queue, e.path)
|
|
if serr != nil {
|
|
unix.Close(hostSockFD)
|
|
log.Warningf("NewSCMSender failed: path=%q, err=%v", e.path, serr)
|
|
return nil, serr
|
|
}
|
|
return c, nil
|
|
}
|
|
|
|
// Release implements transport.BoundEndpoint.Release.
|
|
func (e *endpoint) Release(ctx context.Context) {
|
|
e.dentry.DecRef(ctx)
|
|
}
|
|
|
|
// Passcred implements transport.BoundEndpoint.Passcred.
|
|
func (e *endpoint) Passcred() bool {
|
|
return false
|
|
}
|