nebula/service/listener.go

package service

import (
	"net"
	"sync"
)

type tcpListener struct {
	port   uint16
	s      *Service
	addr   *net.TCPAddr
	accept chan net.Conn

	// closed is closed instead of accept so a handler blocked handing off a
	// connection can't send on a closed channel.
	closed    chan struct{}
	closeOnce sync.Once
}

func (l *tcpListener) Accept() (net.Conn, error) {
	// select picks randomly among ready cases, so without this a handler still
	// waiting to hand off could win over closed after Close has returned.
	select {
	case <-l.closed:
		return nil, net.ErrClosed
	default:
	}

	select {
	case conn := <-l.accept:
		return conn, nil
	case <-l.closed:
		return nil, net.ErrClosed
	}
}

func (l *tcpListener) Close() error {
	l.closeOnce.Do(func() {
		l.s.mu.Lock()
		delete(l.s.mu.listeners, l.port)
		l.s.mu.Unlock()

		close(l.closed)
	})
	return nil
}

// Addr returns the listener's network address.
func (l *tcpListener) Addr() net.Addr {
	return l.addr
}