2018-10-19 23:34:09 +00:00
|
|
|
// Copyright 2018 Google LLC
|
2018-07-09 21:03:03 +00:00
|
|
|
//
|
|
|
|
// 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.
|
2018-05-02 05:50:55 +00:00
|
|
|
|
|
|
|
package ping
|
|
|
|
|
|
|
|
import (
|
|
|
|
"encoding/binary"
|
|
|
|
"sync"
|
|
|
|
|
|
|
|
"gvisor.googlesource.com/gvisor/pkg/sleep"
|
|
|
|
"gvisor.googlesource.com/gvisor/pkg/tcpip"
|
|
|
|
"gvisor.googlesource.com/gvisor/pkg/tcpip/buffer"
|
|
|
|
"gvisor.googlesource.com/gvisor/pkg/tcpip/header"
|
|
|
|
"gvisor.googlesource.com/gvisor/pkg/tcpip/stack"
|
|
|
|
"gvisor.googlesource.com/gvisor/pkg/waiter"
|
|
|
|
)
|
|
|
|
|
2018-08-02 17:41:44 +00:00
|
|
|
// +stateify savable
|
2018-05-02 05:50:55 +00:00
|
|
|
type pingPacket struct {
|
|
|
|
pingPacketEntry
|
|
|
|
senderAddress tcpip.FullAddress
|
|
|
|
data buffer.VectorisedView `state:".(buffer.VectorisedView)"`
|
|
|
|
timestamp int64
|
|
|
|
hasTimestamp bool
|
|
|
|
// views is used as buffer for data when its length is large
|
|
|
|
// enough to store a VectorisedView.
|
|
|
|
views [8]buffer.View `state:"nosave"`
|
|
|
|
}
|
|
|
|
|
|
|
|
type endpointState int
|
|
|
|
|
|
|
|
const (
|
|
|
|
stateInitial endpointState = iota
|
|
|
|
stateBound
|
|
|
|
stateConnected
|
|
|
|
stateClosed
|
|
|
|
)
|
|
|
|
|
|
|
|
// endpoint represents a ping endpoint. This struct serves as the interface
|
|
|
|
// between users of the endpoint and the protocol implementation; it is legal to
|
|
|
|
// have concurrent goroutines make calls into the endpoint, they are properly
|
|
|
|
// synchronized.
|
|
|
|
type endpoint struct {
|
|
|
|
// The following fields are initialized at creation time and do not
|
|
|
|
// change throughout the lifetime of the endpoint.
|
|
|
|
stack *stack.Stack `state:"manual"`
|
|
|
|
netProto tcpip.NetworkProtocolNumber
|
2018-09-04 21:30:15 +00:00
|
|
|
transProto tcpip.TransportProtocolNumber
|
2018-05-02 05:50:55 +00:00
|
|
|
waiterQueue *waiter.Queue
|
|
|
|
|
|
|
|
// The following fields are used to manage the receive queue, and are
|
|
|
|
// protected by rcvMu.
|
|
|
|
rcvMu sync.Mutex `state:"nosave"`
|
|
|
|
rcvReady bool
|
|
|
|
rcvList pingPacketList
|
2018-05-11 23:20:01 +00:00
|
|
|
rcvBufSizeMax int `state:".(int)"`
|
2018-05-02 05:50:55 +00:00
|
|
|
rcvBufSize int
|
|
|
|
rcvClosed bool
|
|
|
|
rcvTimestamp bool
|
|
|
|
|
|
|
|
// The following fields are protected by the mu mutex.
|
|
|
|
mu sync.RWMutex `state:"nosave"`
|
|
|
|
sndBufSize int
|
2018-08-07 14:56:08 +00:00
|
|
|
// shutdownFlags represent the current shutdown state of the endpoint.
|
|
|
|
shutdownFlags tcpip.ShutdownFlags
|
|
|
|
id stack.TransportEndpointID
|
|
|
|
state endpointState
|
|
|
|
bindNICID tcpip.NICID
|
|
|
|
bindAddr tcpip.Address
|
|
|
|
regNICID tcpip.NICID
|
|
|
|
route stack.Route `state:"manual"`
|
2018-05-02 05:50:55 +00:00
|
|
|
}
|
|
|
|
|
2018-09-04 21:30:15 +00:00
|
|
|
func newEndpoint(stack *stack.Stack, netProto tcpip.NetworkProtocolNumber, transProto tcpip.TransportProtocolNumber, waiterQueue *waiter.Queue) *endpoint {
|
2018-05-02 05:50:55 +00:00
|
|
|
return &endpoint{
|
|
|
|
stack: stack,
|
|
|
|
netProto: netProto,
|
2018-09-04 21:30:15 +00:00
|
|
|
transProto: transProto,
|
2018-05-02 05:50:55 +00:00
|
|
|
waiterQueue: waiterQueue,
|
|
|
|
rcvBufSizeMax: 32 * 1024,
|
|
|
|
sndBufSize: 32 * 1024,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Close puts the endpoint in a closed state and frees all resources
|
|
|
|
// associated with it.
|
|
|
|
func (e *endpoint) Close() {
|
|
|
|
e.mu.Lock()
|
2018-08-07 14:56:08 +00:00
|
|
|
e.shutdownFlags = tcpip.ShutdownRead | tcpip.ShutdownWrite
|
2018-05-02 05:50:55 +00:00
|
|
|
switch e.state {
|
|
|
|
case stateBound, stateConnected:
|
2018-09-04 21:30:15 +00:00
|
|
|
e.stack.UnregisterTransportEndpoint(e.regNICID, []tcpip.NetworkProtocolNumber{e.netProto}, e.transProto, e.id)
|
2018-05-02 05:50:55 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// Close the receive list and drain it.
|
|
|
|
e.rcvMu.Lock()
|
|
|
|
e.rcvClosed = true
|
|
|
|
e.rcvBufSize = 0
|
|
|
|
for !e.rcvList.Empty() {
|
|
|
|
p := e.rcvList.Front()
|
|
|
|
e.rcvList.Remove(p)
|
|
|
|
}
|
|
|
|
e.rcvMu.Unlock()
|
|
|
|
|
|
|
|
e.route.Release()
|
|
|
|
|
|
|
|
// Update the state.
|
|
|
|
e.state = stateClosed
|
2018-07-10 04:19:58 +00:00
|
|
|
|
|
|
|
e.mu.Unlock()
|
|
|
|
|
|
|
|
e.waiterQueue.Notify(waiter.EventHUp | waiter.EventErr | waiter.EventIn | waiter.EventOut)
|
2018-05-02 05:50:55 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// Read reads data from the endpoint. This method does not block if
|
|
|
|
// there is no data pending.
|
|
|
|
func (e *endpoint) Read(addr *tcpip.FullAddress) (buffer.View, tcpip.ControlMessages, *tcpip.Error) {
|
|
|
|
e.rcvMu.Lock()
|
|
|
|
|
|
|
|
if e.rcvList.Empty() {
|
|
|
|
err := tcpip.ErrWouldBlock
|
|
|
|
if e.rcvClosed {
|
|
|
|
err = tcpip.ErrClosedForReceive
|
|
|
|
}
|
|
|
|
e.rcvMu.Unlock()
|
|
|
|
return buffer.View{}, tcpip.ControlMessages{}, err
|
|
|
|
}
|
|
|
|
|
|
|
|
p := e.rcvList.Front()
|
|
|
|
e.rcvList.Remove(p)
|
|
|
|
e.rcvBufSize -= p.data.Size()
|
|
|
|
ts := e.rcvTimestamp
|
|
|
|
|
|
|
|
e.rcvMu.Unlock()
|
|
|
|
|
|
|
|
if addr != nil {
|
|
|
|
*addr = p.senderAddress
|
|
|
|
}
|
|
|
|
|
|
|
|
if ts && !p.hasTimestamp {
|
|
|
|
// Linux uses the current time.
|
|
|
|
p.timestamp = e.stack.NowNanoseconds()
|
|
|
|
}
|
|
|
|
|
|
|
|
return p.data.ToView(), tcpip.ControlMessages{HasTimestamp: ts, Timestamp: p.timestamp}, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// prepareForWrite prepares the endpoint for sending data. In particular, it
|
|
|
|
// binds it if it's still in the initial state. To do so, it must first
|
|
|
|
// reacquire the mutex in exclusive mode.
|
|
|
|
//
|
|
|
|
// Returns true for retry if preparation should be retried.
|
|
|
|
func (e *endpoint) prepareForWrite(to *tcpip.FullAddress) (retry bool, err *tcpip.Error) {
|
|
|
|
switch e.state {
|
|
|
|
case stateInitial:
|
|
|
|
case stateConnected:
|
|
|
|
return false, nil
|
|
|
|
|
|
|
|
case stateBound:
|
|
|
|
if to == nil {
|
|
|
|
return false, tcpip.ErrDestinationRequired
|
|
|
|
}
|
|
|
|
return false, nil
|
|
|
|
default:
|
|
|
|
return false, tcpip.ErrInvalidEndpointState
|
|
|
|
}
|
|
|
|
|
|
|
|
e.mu.RUnlock()
|
|
|
|
defer e.mu.RLock()
|
|
|
|
|
|
|
|
e.mu.Lock()
|
|
|
|
defer e.mu.Unlock()
|
|
|
|
|
|
|
|
// The state changed when we released the shared locked and re-acquired
|
|
|
|
// it in exclusive mode. Try again.
|
|
|
|
if e.state != stateInitial {
|
|
|
|
return true, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// The state is still 'initial', so try to bind the endpoint.
|
|
|
|
if err := e.bindLocked(tcpip.FullAddress{}, nil); err != nil {
|
|
|
|
return false, err
|
|
|
|
}
|
|
|
|
|
|
|
|
return true, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// Write writes data to the endpoint's peer. This method does not block
|
|
|
|
// if the data cannot be written.
|
2018-09-28 17:59:21 +00:00
|
|
|
func (e *endpoint) Write(p tcpip.Payload, opts tcpip.WriteOptions) (uintptr, <-chan struct{}, *tcpip.Error) {
|
2018-05-02 05:50:55 +00:00
|
|
|
// MSG_MORE is unimplemented. (This also means that MSG_EOR is a no-op.)
|
|
|
|
if opts.More {
|
2018-09-28 17:59:21 +00:00
|
|
|
return 0, nil, tcpip.ErrInvalidOptionValue
|
2018-05-02 05:50:55 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
to := opts.To
|
|
|
|
|
|
|
|
e.mu.RLock()
|
|
|
|
defer e.mu.RUnlock()
|
|
|
|
|
2018-08-07 14:56:08 +00:00
|
|
|
// If we've shutdown with SHUT_WR we are in an invalid state for sending.
|
|
|
|
if e.shutdownFlags&tcpip.ShutdownWrite != 0 {
|
2018-09-28 17:59:21 +00:00
|
|
|
return 0, nil, tcpip.ErrClosedForSend
|
2018-08-07 14:56:08 +00:00
|
|
|
}
|
|
|
|
|
2018-05-02 05:50:55 +00:00
|
|
|
// Prepare for write.
|
|
|
|
for {
|
|
|
|
retry, err := e.prepareForWrite(to)
|
|
|
|
if err != nil {
|
2018-09-28 17:59:21 +00:00
|
|
|
return 0, nil, err
|
2018-05-02 05:50:55 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
if !retry {
|
|
|
|
break
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
var route *stack.Route
|
|
|
|
if to == nil {
|
|
|
|
route = &e.route
|
|
|
|
|
|
|
|
if route.IsResolutionRequired() {
|
|
|
|
// Promote lock to exclusive if using a shared route, given that it may
|
|
|
|
// need to change in Route.Resolve() call below.
|
|
|
|
e.mu.RUnlock()
|
|
|
|
defer e.mu.RLock()
|
|
|
|
|
|
|
|
e.mu.Lock()
|
|
|
|
defer e.mu.Unlock()
|
|
|
|
|
|
|
|
// Recheck state after lock was re-acquired.
|
|
|
|
if e.state != stateConnected {
|
2018-09-28 17:59:21 +00:00
|
|
|
return 0, nil, tcpip.ErrInvalidEndpointState
|
2018-05-02 05:50:55 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
// Reject destination address if it goes through a different
|
|
|
|
// NIC than the endpoint was bound to.
|
|
|
|
nicid := to.NIC
|
|
|
|
if e.bindNICID != 0 {
|
|
|
|
if nicid != 0 && nicid != e.bindNICID {
|
2018-09-28 17:59:21 +00:00
|
|
|
return 0, nil, tcpip.ErrNoRoute
|
2018-05-02 05:50:55 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
nicid = e.bindNICID
|
|
|
|
}
|
|
|
|
|
|
|
|
toCopy := *to
|
|
|
|
to = &toCopy
|
|
|
|
netProto, err := e.checkV4Mapped(to, true)
|
|
|
|
if err != nil {
|
2018-09-28 17:59:21 +00:00
|
|
|
return 0, nil, err
|
2018-05-02 05:50:55 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// Find the enpoint.
|
|
|
|
r, err := e.stack.FindRoute(nicid, e.bindAddr, to.Addr, netProto)
|
|
|
|
if err != nil {
|
2018-09-28 17:59:21 +00:00
|
|
|
return 0, nil, err
|
2018-05-02 05:50:55 +00:00
|
|
|
}
|
|
|
|
defer r.Release()
|
|
|
|
|
|
|
|
route = &r
|
|
|
|
}
|
|
|
|
|
|
|
|
if route.IsResolutionRequired() {
|
|
|
|
waker := &sleep.Waker{}
|
2018-09-28 17:59:21 +00:00
|
|
|
if ch, err := route.Resolve(waker); err != nil {
|
2018-05-02 05:50:55 +00:00
|
|
|
if err == tcpip.ErrWouldBlock {
|
|
|
|
// Link address needs to be resolved. Resolution was triggered the
|
|
|
|
// background. Better luck next time.
|
|
|
|
route.RemoveWaker(waker)
|
2018-09-28 17:59:21 +00:00
|
|
|
return 0, ch, tcpip.ErrNoLinkAddress
|
2018-05-02 05:50:55 +00:00
|
|
|
}
|
2018-09-28 17:59:21 +00:00
|
|
|
return 0, nil, err
|
2018-05-02 05:50:55 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
v, err := p.Get(p.Size())
|
|
|
|
if err != nil {
|
2018-09-28 17:59:21 +00:00
|
|
|
return 0, nil, err
|
2018-05-02 05:50:55 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
switch e.netProto {
|
|
|
|
case header.IPv4ProtocolNumber:
|
|
|
|
err = sendPing4(route, e.id.LocalPort, v)
|
|
|
|
|
|
|
|
case header.IPv6ProtocolNumber:
|
2018-09-04 21:30:15 +00:00
|
|
|
err = sendPing6(route, e.id.LocalPort, v)
|
2018-05-02 05:50:55 +00:00
|
|
|
}
|
|
|
|
|
2018-12-06 19:40:39 +00:00
|
|
|
if err != nil {
|
|
|
|
return 0, nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
return uintptr(len(v)), nil, nil
|
2018-05-02 05:50:55 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// Peek only returns data from a single datagram, so do nothing here.
|
|
|
|
func (e *endpoint) Peek([][]byte) (uintptr, tcpip.ControlMessages, *tcpip.Error) {
|
|
|
|
return 0, tcpip.ControlMessages{}, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// SetSockOpt sets a socket option. Currently not supported.
|
|
|
|
func (e *endpoint) SetSockOpt(opt interface{}) *tcpip.Error {
|
|
|
|
switch v := opt.(type) {
|
|
|
|
case tcpip.TimestampOption:
|
|
|
|
e.rcvMu.Lock()
|
|
|
|
e.rcvTimestamp = v != 0
|
|
|
|
e.rcvMu.Unlock()
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// GetSockOpt implements tcpip.Endpoint.GetSockOpt.
|
|
|
|
func (e *endpoint) GetSockOpt(opt interface{}) *tcpip.Error {
|
|
|
|
switch o := opt.(type) {
|
|
|
|
case tcpip.ErrorOption:
|
|
|
|
return nil
|
|
|
|
|
|
|
|
case *tcpip.SendBufferSizeOption:
|
|
|
|
e.mu.Lock()
|
|
|
|
*o = tcpip.SendBufferSizeOption(e.sndBufSize)
|
|
|
|
e.mu.Unlock()
|
|
|
|
return nil
|
|
|
|
|
|
|
|
case *tcpip.ReceiveBufferSizeOption:
|
|
|
|
e.rcvMu.Lock()
|
|
|
|
*o = tcpip.ReceiveBufferSizeOption(e.rcvBufSizeMax)
|
|
|
|
e.rcvMu.Unlock()
|
|
|
|
return nil
|
|
|
|
|
|
|
|
case *tcpip.ReceiveQueueSizeOption:
|
|
|
|
e.rcvMu.Lock()
|
|
|
|
if e.rcvList.Empty() {
|
|
|
|
*o = 0
|
|
|
|
} else {
|
|
|
|
p := e.rcvList.Front()
|
|
|
|
*o = tcpip.ReceiveQueueSizeOption(p.data.Size())
|
|
|
|
}
|
|
|
|
e.rcvMu.Unlock()
|
|
|
|
return nil
|
|
|
|
|
|
|
|
case *tcpip.TimestampOption:
|
|
|
|
e.rcvMu.Lock()
|
|
|
|
*o = 0
|
|
|
|
if e.rcvTimestamp {
|
|
|
|
*o = 1
|
|
|
|
}
|
|
|
|
e.rcvMu.Unlock()
|
2018-12-21 21:12:32 +00:00
|
|
|
return nil
|
2018-05-02 05:50:55 +00:00
|
|
|
|
2018-12-21 21:12:32 +00:00
|
|
|
case *tcpip.KeepaliveEnabledOption:
|
|
|
|
*o = 0
|
|
|
|
return nil
|
|
|
|
|
|
|
|
default:
|
|
|
|
return tcpip.ErrUnknownProtocolOption
|
|
|
|
}
|
2018-05-02 05:50:55 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func sendPing4(r *stack.Route, ident uint16, data buffer.View) *tcpip.Error {
|
|
|
|
if len(data) < header.ICMPv4EchoMinimumSize {
|
|
|
|
return tcpip.ErrInvalidEndpointState
|
|
|
|
}
|
|
|
|
|
|
|
|
// Set the ident. Sequence number is provided by the user.
|
|
|
|
binary.BigEndian.PutUint16(data[header.ICMPv4MinimumSize:], ident)
|
|
|
|
|
|
|
|
hdr := buffer.NewPrependable(header.ICMPv4EchoMinimumSize + int(r.MaxHeaderLength()))
|
|
|
|
|
|
|
|
icmpv4 := header.ICMPv4(hdr.Prepend(header.ICMPv4EchoMinimumSize))
|
|
|
|
copy(icmpv4, data)
|
|
|
|
data = data[header.ICMPv4EchoMinimumSize:]
|
|
|
|
|
|
|
|
// Linux performs these basic checks.
|
|
|
|
if icmpv4.Type() != header.ICMPv4Echo || icmpv4.Code() != 0 {
|
|
|
|
return tcpip.ErrInvalidEndpointState
|
|
|
|
}
|
|
|
|
|
|
|
|
icmpv4.SetChecksum(0)
|
|
|
|
icmpv4.SetChecksum(^header.Checksum(icmpv4, header.Checksum(data, 0)))
|
|
|
|
|
2018-09-14 22:22:42 +00:00
|
|
|
return r.WritePacket(hdr, data.ToVectorisedView(), header.ICMPv4ProtocolNumber, r.DefaultTTL())
|
2018-05-02 05:50:55 +00:00
|
|
|
}
|
|
|
|
|
2018-09-04 21:30:15 +00:00
|
|
|
func sendPing6(r *stack.Route, ident uint16, data buffer.View) *tcpip.Error {
|
|
|
|
if len(data) < header.ICMPv6EchoMinimumSize {
|
|
|
|
return tcpip.ErrInvalidEndpointState
|
|
|
|
}
|
|
|
|
|
|
|
|
// Set the ident. Sequence number is provided by the user.
|
|
|
|
binary.BigEndian.PutUint16(data[header.ICMPv6MinimumSize:], ident)
|
|
|
|
|
|
|
|
hdr := buffer.NewPrependable(header.ICMPv6EchoMinimumSize + int(r.MaxHeaderLength()))
|
|
|
|
|
|
|
|
icmpv6 := header.ICMPv6(hdr.Prepend(header.ICMPv6EchoMinimumSize))
|
|
|
|
copy(icmpv6, data)
|
|
|
|
data = data[header.ICMPv6EchoMinimumSize:]
|
|
|
|
|
|
|
|
if icmpv6.Type() != header.ICMPv6EchoRequest || icmpv6.Code() != 0 {
|
|
|
|
return tcpip.ErrInvalidEndpointState
|
|
|
|
}
|
|
|
|
|
|
|
|
icmpv6.SetChecksum(0)
|
|
|
|
icmpv6.SetChecksum(^header.Checksum(icmpv6, header.Checksum(data, 0)))
|
2018-09-13 03:38:27 +00:00
|
|
|
|
2018-09-14 22:22:42 +00:00
|
|
|
return r.WritePacket(hdr, data.ToVectorisedView(), header.ICMPv6ProtocolNumber, r.DefaultTTL())
|
2018-09-04 21:30:15 +00:00
|
|
|
}
|
|
|
|
|
2018-05-02 05:50:55 +00:00
|
|
|
func (e *endpoint) checkV4Mapped(addr *tcpip.FullAddress, allowMismatch bool) (tcpip.NetworkProtocolNumber, *tcpip.Error) {
|
|
|
|
netProto := e.netProto
|
|
|
|
if header.IsV4MappedAddress(addr.Addr) {
|
|
|
|
return 0, tcpip.ErrNoRoute
|
|
|
|
}
|
|
|
|
|
|
|
|
// Fail if we're bound to an address length different from the one we're
|
|
|
|
// checking.
|
|
|
|
if l := len(e.id.LocalAddress); !allowMismatch && l != 0 && l != len(addr.Addr) {
|
|
|
|
return 0, tcpip.ErrInvalidEndpointState
|
|
|
|
}
|
|
|
|
|
|
|
|
return netProto, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// Connect connects the endpoint to its peer. Specifying a NIC is optional.
|
|
|
|
func (e *endpoint) Connect(addr tcpip.FullAddress) *tcpip.Error {
|
|
|
|
e.mu.Lock()
|
|
|
|
defer e.mu.Unlock()
|
|
|
|
|
|
|
|
nicid := addr.NIC
|
|
|
|
localPort := uint16(0)
|
|
|
|
switch e.state {
|
|
|
|
case stateBound, stateConnected:
|
|
|
|
localPort = e.id.LocalPort
|
|
|
|
if e.bindNICID == 0 {
|
|
|
|
break
|
|
|
|
}
|
|
|
|
|
|
|
|
if nicid != 0 && nicid != e.bindNICID {
|
|
|
|
return tcpip.ErrInvalidEndpointState
|
|
|
|
}
|
|
|
|
|
|
|
|
nicid = e.bindNICID
|
|
|
|
default:
|
|
|
|
return tcpip.ErrInvalidEndpointState
|
|
|
|
}
|
|
|
|
|
|
|
|
netProto, err := e.checkV4Mapped(&addr, false)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
// Find a route to the desired destination.
|
|
|
|
r, err := e.stack.FindRoute(nicid, e.bindAddr, addr.Addr, netProto)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
defer r.Release()
|
|
|
|
|
|
|
|
id := stack.TransportEndpointID{
|
|
|
|
LocalAddress: r.LocalAddress,
|
|
|
|
LocalPort: localPort,
|
|
|
|
RemoteAddress: r.RemoteAddress,
|
|
|
|
}
|
|
|
|
|
|
|
|
// Even if we're connected, this endpoint can still be used to send
|
|
|
|
// packets on a different network protocol, so we register both even if
|
|
|
|
// v6only is set to false and this is an ipv6 endpoint.
|
|
|
|
netProtos := []tcpip.NetworkProtocolNumber{netProto}
|
|
|
|
|
|
|
|
id, err = e.registerWithStack(nicid, netProtos, id)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
e.id = id
|
|
|
|
e.route = r.Clone()
|
|
|
|
e.regNICID = nicid
|
|
|
|
|
|
|
|
e.state = stateConnected
|
|
|
|
|
|
|
|
e.rcvMu.Lock()
|
|
|
|
e.rcvReady = true
|
|
|
|
e.rcvMu.Unlock()
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// ConnectEndpoint is not supported.
|
|
|
|
func (*endpoint) ConnectEndpoint(tcpip.Endpoint) *tcpip.Error {
|
|
|
|
return tcpip.ErrInvalidEndpointState
|
|
|
|
}
|
|
|
|
|
|
|
|
// Shutdown closes the read and/or write end of the endpoint connection
|
|
|
|
// to its peer.
|
|
|
|
func (e *endpoint) Shutdown(flags tcpip.ShutdownFlags) *tcpip.Error {
|
2018-08-07 14:56:08 +00:00
|
|
|
e.mu.Lock()
|
|
|
|
defer e.mu.Unlock()
|
|
|
|
e.shutdownFlags |= flags
|
2018-05-02 05:50:55 +00:00
|
|
|
|
|
|
|
if e.state != stateConnected {
|
|
|
|
return tcpip.ErrNotConnected
|
|
|
|
}
|
|
|
|
|
|
|
|
if flags&tcpip.ShutdownRead != 0 {
|
|
|
|
e.rcvMu.Lock()
|
|
|
|
wasClosed := e.rcvClosed
|
|
|
|
e.rcvClosed = true
|
|
|
|
e.rcvMu.Unlock()
|
|
|
|
|
|
|
|
if !wasClosed {
|
|
|
|
e.waiterQueue.Notify(waiter.EventIn)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// Listen is not supported by UDP, it just fails.
|
|
|
|
func (*endpoint) Listen(int) *tcpip.Error {
|
|
|
|
return tcpip.ErrNotSupported
|
|
|
|
}
|
|
|
|
|
|
|
|
// Accept is not supported by UDP, it just fails.
|
|
|
|
func (*endpoint) Accept() (tcpip.Endpoint, *waiter.Queue, *tcpip.Error) {
|
|
|
|
return nil, nil, tcpip.ErrNotSupported
|
|
|
|
}
|
|
|
|
|
|
|
|
func (e *endpoint) registerWithStack(nicid tcpip.NICID, netProtos []tcpip.NetworkProtocolNumber, id stack.TransportEndpointID) (stack.TransportEndpointID, *tcpip.Error) {
|
|
|
|
if id.LocalPort != 0 {
|
|
|
|
// The endpoint already has a local port, just attempt to
|
|
|
|
// register it.
|
2018-09-04 21:30:15 +00:00
|
|
|
err := e.stack.RegisterTransportEndpoint(nicid, netProtos, e.transProto, id, e)
|
2018-05-02 05:50:55 +00:00
|
|
|
return id, err
|
|
|
|
}
|
|
|
|
|
|
|
|
// We need to find a port for the endpoint.
|
|
|
|
_, err := e.stack.PickEphemeralPort(func(p uint16) (bool, *tcpip.Error) {
|
|
|
|
id.LocalPort = p
|
2018-09-04 21:30:15 +00:00
|
|
|
err := e.stack.RegisterTransportEndpoint(nicid, netProtos, e.transProto, id, e)
|
2018-05-02 05:50:55 +00:00
|
|
|
switch err {
|
|
|
|
case nil:
|
|
|
|
return true, nil
|
|
|
|
case tcpip.ErrPortInUse:
|
|
|
|
return false, nil
|
|
|
|
default:
|
|
|
|
return false, err
|
|
|
|
}
|
|
|
|
})
|
|
|
|
|
|
|
|
return id, err
|
|
|
|
}
|
|
|
|
|
|
|
|
func (e *endpoint) bindLocked(addr tcpip.FullAddress, commit func() *tcpip.Error) *tcpip.Error {
|
|
|
|
// Don't allow binding once endpoint is not in the initial state
|
|
|
|
// anymore.
|
|
|
|
if e.state != stateInitial {
|
|
|
|
return tcpip.ErrInvalidEndpointState
|
|
|
|
}
|
|
|
|
|
|
|
|
netProto, err := e.checkV4Mapped(&addr, false)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
// Expand netProtos to include v4 and v6 if the caller is binding to a
|
|
|
|
// wildcard (empty) address, and this is an IPv6 endpoint with v6only
|
|
|
|
// set to false.
|
|
|
|
netProtos := []tcpip.NetworkProtocolNumber{netProto}
|
|
|
|
|
|
|
|
if len(addr.Addr) != 0 {
|
|
|
|
// A local address was specified, verify that it's valid.
|
|
|
|
if e.stack.CheckLocalAddress(addr.NIC, netProto, addr.Addr) == 0 {
|
|
|
|
return tcpip.ErrBadLocalAddress
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
id := stack.TransportEndpointID{
|
|
|
|
LocalPort: addr.Port,
|
|
|
|
LocalAddress: addr.Addr,
|
|
|
|
}
|
|
|
|
id, err = e.registerWithStack(addr.NIC, netProtos, id)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
if commit != nil {
|
|
|
|
if err := commit(); err != nil {
|
|
|
|
// Unregister, the commit failed.
|
2018-09-04 21:30:15 +00:00
|
|
|
e.stack.UnregisterTransportEndpoint(addr.NIC, netProtos, e.transProto, id)
|
2018-05-02 05:50:55 +00:00
|
|
|
return err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
e.id = id
|
|
|
|
e.regNICID = addr.NIC
|
|
|
|
|
|
|
|
// Mark endpoint as bound.
|
|
|
|
e.state = stateBound
|
|
|
|
|
|
|
|
e.rcvMu.Lock()
|
|
|
|
e.rcvReady = true
|
|
|
|
e.rcvMu.Unlock()
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// Bind binds the endpoint to a specific local address and port.
|
|
|
|
// Specifying a NIC is optional.
|
|
|
|
func (e *endpoint) Bind(addr tcpip.FullAddress, commit func() *tcpip.Error) *tcpip.Error {
|
|
|
|
e.mu.Lock()
|
|
|
|
defer e.mu.Unlock()
|
|
|
|
|
|
|
|
err := e.bindLocked(addr, commit)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
e.bindNICID = addr.NIC
|
|
|
|
e.bindAddr = addr.Addr
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// GetLocalAddress returns the address to which the endpoint is bound.
|
|
|
|
func (e *endpoint) GetLocalAddress() (tcpip.FullAddress, *tcpip.Error) {
|
|
|
|
e.mu.RLock()
|
|
|
|
defer e.mu.RUnlock()
|
|
|
|
|
|
|
|
return tcpip.FullAddress{
|
|
|
|
NIC: e.regNICID,
|
|
|
|
Addr: e.id.LocalAddress,
|
|
|
|
Port: e.id.LocalPort,
|
|
|
|
}, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// GetRemoteAddress returns the address to which the endpoint is connected.
|
|
|
|
func (e *endpoint) GetRemoteAddress() (tcpip.FullAddress, *tcpip.Error) {
|
|
|
|
e.mu.RLock()
|
|
|
|
defer e.mu.RUnlock()
|
|
|
|
|
|
|
|
if e.state != stateConnected {
|
|
|
|
return tcpip.FullAddress{}, tcpip.ErrNotConnected
|
|
|
|
}
|
|
|
|
|
|
|
|
return tcpip.FullAddress{
|
|
|
|
NIC: e.regNICID,
|
|
|
|
Addr: e.id.RemoteAddress,
|
|
|
|
Port: e.id.RemotePort,
|
|
|
|
}, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// Readiness returns the current readiness of the endpoint. For example, if
|
|
|
|
// waiter.EventIn is set, the endpoint is immediately readable.
|
|
|
|
func (e *endpoint) Readiness(mask waiter.EventMask) waiter.EventMask {
|
|
|
|
// The endpoint is always writable.
|
|
|
|
result := waiter.EventOut & mask
|
|
|
|
|
|
|
|
// Determine if the endpoint is readable if requested.
|
|
|
|
if (mask & waiter.EventIn) != 0 {
|
|
|
|
e.rcvMu.Lock()
|
|
|
|
if !e.rcvList.Empty() || e.rcvClosed {
|
|
|
|
result |= waiter.EventIn
|
|
|
|
}
|
|
|
|
e.rcvMu.Unlock()
|
|
|
|
}
|
|
|
|
|
|
|
|
return result
|
|
|
|
}
|
|
|
|
|
|
|
|
// HandlePacket is called by the stack when new packets arrive to this transport
|
|
|
|
// endpoint.
|
2018-09-13 04:57:04 +00:00
|
|
|
func (e *endpoint) HandlePacket(r *stack.Route, id stack.TransportEndpointID, vv buffer.VectorisedView) {
|
2018-05-02 05:50:55 +00:00
|
|
|
e.rcvMu.Lock()
|
|
|
|
|
|
|
|
// Drop the packet if our buffer is currently full.
|
|
|
|
if !e.rcvReady || e.rcvClosed || e.rcvBufSize >= e.rcvBufSizeMax {
|
|
|
|
e.rcvMu.Unlock()
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
wasEmpty := e.rcvBufSize == 0
|
|
|
|
|
|
|
|
// Push new packet into receive list and increment the buffer size.
|
|
|
|
pkt := &pingPacket{
|
|
|
|
senderAddress: tcpip.FullAddress{
|
|
|
|
NIC: r.NICID(),
|
|
|
|
Addr: id.RemoteAddress,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
pkt.data = vv.Clone(pkt.views[:])
|
|
|
|
e.rcvList.PushBack(pkt)
|
|
|
|
e.rcvBufSize += vv.Size()
|
|
|
|
|
|
|
|
if e.rcvTimestamp {
|
|
|
|
pkt.timestamp = e.stack.NowNanoseconds()
|
|
|
|
pkt.hasTimestamp = true
|
|
|
|
}
|
|
|
|
|
|
|
|
e.rcvMu.Unlock()
|
|
|
|
|
|
|
|
// Notify any waiters that there's data to be read now.
|
|
|
|
if wasEmpty {
|
|
|
|
e.waiterQueue.Notify(waiter.EventIn)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// HandleControlPacket implements stack.TransportEndpoint.HandleControlPacket.
|
2018-09-13 04:57:04 +00:00
|
|
|
func (e *endpoint) HandleControlPacket(id stack.TransportEndpointID, typ stack.ControlType, extra uint32, vv buffer.VectorisedView) {
|
2018-05-02 05:50:55 +00:00
|
|
|
}
|