You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
682 lines
18 KiB
682 lines
18 KiB
package p2p
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"net"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/tendermint/tendermint/crypto"
|
|
"github.com/tendermint/tendermint/libs/log"
|
|
"github.com/tendermint/tendermint/libs/protoio"
|
|
"github.com/tendermint/tendermint/p2p/conn"
|
|
p2pproto "github.com/tendermint/tendermint/proto/tendermint/p2p"
|
|
|
|
"golang.org/x/net/netutil"
|
|
)
|
|
|
|
const (
|
|
defaultDialTimeout = time.Second
|
|
defaultFilterTimeout = 5 * time.Second
|
|
defaultHandshakeTimeout = 3 * time.Second
|
|
)
|
|
|
|
// MConnProtocol is the MConn protocol identifier.
|
|
const MConnProtocol Protocol = "mconn"
|
|
|
|
// MConnTransportOption sets an option for MConnTransport.
|
|
type MConnTransportOption func(*MConnTransport)
|
|
|
|
// MConnTransportMaxIncomingConnections sets the maximum number of
|
|
// simultaneous incoming connections. Default: 0 (unlimited)
|
|
func MConnTransportMaxIncomingConnections(max int) MConnTransportOption {
|
|
return func(mt *MConnTransport) { mt.maxIncomingConnections = max }
|
|
}
|
|
|
|
// MConnTransportFilterTimeout sets the timeout for filter callbacks.
|
|
func MConnTransportFilterTimeout(timeout time.Duration) MConnTransportOption {
|
|
return func(mt *MConnTransport) { mt.filterTimeout = timeout }
|
|
}
|
|
|
|
// MConnTransportConnFilters sets connection filters.
|
|
func MConnTransportConnFilters(filters ...ConnFilterFunc) MConnTransportOption {
|
|
return func(mt *MConnTransport) { mt.connFilters = filters }
|
|
}
|
|
|
|
// ConnFilterFunc is a callback for connection filtering. If it returns an
|
|
// error, the connection is rejected. The set of existing connections is passed
|
|
// along with the new connection and all resolved IPs.
|
|
type ConnFilterFunc func(ConnSet, net.Conn, []net.IP) error
|
|
|
|
// ConnDuplicateIPFilter resolves and keeps all ips for an incoming connection
|
|
// and refuses new ones if they come from a known ip.
|
|
var ConnDuplicateIPFilter ConnFilterFunc = func(cs ConnSet, c net.Conn, ips []net.IP) error {
|
|
for _, ip := range ips {
|
|
if cs.HasIP(ip) {
|
|
return ErrRejected{
|
|
conn: c,
|
|
err: fmt.Errorf("ip<%v> already connected", ip),
|
|
isDuplicate: true,
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// MConnTransport is a Transport implementation using the current multiplexed
|
|
// Tendermint protocol ("MConn"). It inherits lots of code and logic from the
|
|
// previous implementation for parity with the current P2P stack (such as
|
|
// connection filtering, peer verification, and panic handling), which should be
|
|
// moved out of the transport once the rest of the P2P stack is rewritten.
|
|
type MConnTransport struct {
|
|
privKey crypto.PrivKey
|
|
nodeInfo NodeInfo
|
|
channelDescs []*ChannelDescriptor
|
|
mConnConfig conn.MConnConfig
|
|
|
|
maxIncomingConnections int
|
|
dialTimeout time.Duration
|
|
handshakeTimeout time.Duration
|
|
filterTimeout time.Duration
|
|
|
|
logger log.Logger
|
|
listener net.Listener
|
|
|
|
closeOnce sync.Once
|
|
chAccept chan *mConnConnection
|
|
chError chan error
|
|
chClose chan struct{}
|
|
|
|
// FIXME: This is a vestige from the old transport, and should be managed
|
|
// by the router once we rewrite the P2P core.
|
|
conns ConnSet
|
|
connFilters []ConnFilterFunc
|
|
}
|
|
|
|
// NewMConnTransport sets up a new MConn transport.
|
|
func NewMConnTransport(
|
|
logger log.Logger,
|
|
nodeInfo NodeInfo,
|
|
privKey crypto.PrivKey,
|
|
mConnConfig conn.MConnConfig,
|
|
opts ...MConnTransportOption,
|
|
) *MConnTransport {
|
|
m := &MConnTransport{
|
|
privKey: privKey,
|
|
nodeInfo: nodeInfo,
|
|
mConnConfig: mConnConfig,
|
|
channelDescs: []*ChannelDescriptor{},
|
|
|
|
dialTimeout: defaultDialTimeout,
|
|
handshakeTimeout: defaultHandshakeTimeout,
|
|
filterTimeout: defaultFilterTimeout,
|
|
|
|
logger: logger,
|
|
chAccept: make(chan *mConnConnection),
|
|
chError: make(chan error),
|
|
chClose: make(chan struct{}),
|
|
|
|
conns: NewConnSet(),
|
|
connFilters: []ConnFilterFunc{},
|
|
}
|
|
for _, opt := range opts {
|
|
opt(m)
|
|
}
|
|
return m
|
|
}
|
|
|
|
// SetChannelDescriptors implements Transport.
|
|
//
|
|
// This is not concurrency-safe, and must be called before listening.
|
|
//
|
|
// FIXME: This is here for compatibility with existing switch code,
|
|
// it should be passed via the constructor instead.
|
|
func (m *MConnTransport) SetChannelDescriptors(chDescs []*conn.ChannelDescriptor) {
|
|
m.channelDescs = chDescs
|
|
}
|
|
|
|
// Listen asynchronously listens for inbound connections on the given endpoint.
|
|
// It must be called exactly once before calling Accept(), and the caller must
|
|
// call Close() to shut down the listener.
|
|
func (m *MConnTransport) Listen(endpoint Endpoint) error {
|
|
if m.listener != nil {
|
|
return errors.New("MConn transport is already listening")
|
|
}
|
|
err := m.normalizeEndpoint(&endpoint)
|
|
if err != nil {
|
|
return fmt.Errorf("invalid MConn listen endpoint %q: %w", endpoint, err)
|
|
}
|
|
|
|
m.listener, err = net.Listen("tcp", fmt.Sprintf("%v:%v", endpoint.IP, endpoint.Port))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if m.maxIncomingConnections > 0 {
|
|
m.listener = netutil.LimitListener(m.listener, m.maxIncomingConnections)
|
|
}
|
|
|
|
// Spawn a goroutine to accept inbound connections asynchronously.
|
|
go m.accept()
|
|
|
|
return nil
|
|
}
|
|
|
|
// accept accepts inbound connections in a loop, and asynchronously handshakes
|
|
// with the peer to avoid head-of-line blocking. Established connections are
|
|
// passed to Accept() via chAccept.
|
|
// See: https://github.com/tendermint/tendermint/issues/204
|
|
func (m *MConnTransport) accept() {
|
|
for {
|
|
tcpConn, err := m.listener.Accept()
|
|
if err != nil {
|
|
// We have to check for closure first, since we don't want to
|
|
// propagate "use of closed network connection" errors.
|
|
select {
|
|
case <-m.chClose:
|
|
default:
|
|
// We also select on chClose here, in case the transport closes
|
|
// while we're blocked on error propagation.
|
|
select {
|
|
case m.chError <- err:
|
|
case <-m.chClose:
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
go func() {
|
|
err := m.filterTCPConn(tcpConn)
|
|
if err != nil {
|
|
if err := tcpConn.Close(); err != nil {
|
|
m.logger.Debug("failed to close TCP connection", "err", err)
|
|
}
|
|
select {
|
|
case m.chError <- err:
|
|
case <-m.chClose:
|
|
}
|
|
return
|
|
}
|
|
|
|
conn, err := newMConnConnection(m, tcpConn, "")
|
|
if err != nil {
|
|
m.conns.Remove(tcpConn)
|
|
if err := tcpConn.Close(); err != nil {
|
|
m.logger.Debug("failed to close TCP connection", "err", err)
|
|
}
|
|
select {
|
|
case m.chError <- err:
|
|
case <-m.chClose:
|
|
}
|
|
} else {
|
|
select {
|
|
case m.chAccept <- conn:
|
|
case <-m.chClose:
|
|
if err := tcpConn.Close(); err != nil {
|
|
m.logger.Debug("failed to close TCP connection", "err", err)
|
|
}
|
|
}
|
|
}
|
|
}()
|
|
}
|
|
}
|
|
|
|
// Accept implements Transport.
|
|
//
|
|
// accept() runs a concurrent accept loop that accepts inbound connections
|
|
// and then handshakes in a non-blocking fashion. The handshaked and validated
|
|
// connections are returned via this call, picking them off of the chAccept
|
|
// channel (or the handshake error, if any).
|
|
func (m *MConnTransport) Accept(ctx context.Context) (Connection, error) {
|
|
select {
|
|
case conn := <-m.chAccept:
|
|
return conn, nil
|
|
case err := <-m.chError:
|
|
return nil, err
|
|
case <-m.chClose:
|
|
return nil, ErrTransportClosed{}
|
|
case <-ctx.Done():
|
|
return nil, nil
|
|
}
|
|
}
|
|
|
|
// Dial implements Transport.
|
|
func (m *MConnTransport) Dial(ctx context.Context, endpoint Endpoint) (Connection, error) {
|
|
err := m.normalizeEndpoint(&endpoint)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
ctx, cancel := context.WithTimeout(ctx, m.dialTimeout)
|
|
defer cancel()
|
|
|
|
dialer := net.Dialer{}
|
|
tcpConn, err := dialer.DialContext(ctx, "tcp",
|
|
net.JoinHostPort(endpoint.IP.String(), fmt.Sprintf("%v", endpoint.Port)))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
err = m.filterTCPConn(tcpConn)
|
|
if err != nil {
|
|
if err := tcpConn.Close(); err != nil {
|
|
m.logger.Debug("failed to close TCP connection", "err", err)
|
|
}
|
|
return nil, err
|
|
}
|
|
|
|
conn, err := newMConnConnection(m, tcpConn, endpoint.PeerID)
|
|
if err != nil {
|
|
m.conns.Remove(tcpConn)
|
|
if err := tcpConn.Close(); err != nil {
|
|
m.logger.Debug("failed to close TCP connection", "err", err)
|
|
}
|
|
return nil, err
|
|
}
|
|
|
|
return conn, nil
|
|
}
|
|
|
|
// Endpoints implements Transport.
|
|
func (m *MConnTransport) Endpoints() []Endpoint {
|
|
if m.listener == nil {
|
|
return []Endpoint{}
|
|
}
|
|
addr := m.listener.Addr().(*net.TCPAddr)
|
|
return []Endpoint{{
|
|
Protocol: MConnProtocol,
|
|
PeerID: m.nodeInfo.ID(),
|
|
IP: addr.IP,
|
|
Port: uint16(addr.Port),
|
|
}}
|
|
}
|
|
|
|
// Close implements Transport.
|
|
func (m *MConnTransport) Close() error {
|
|
var err error
|
|
m.closeOnce.Do(func() {
|
|
// We have to close chClose first, so that accept() will detect
|
|
// the closure and not propagate the error.
|
|
close(m.chClose)
|
|
if m.listener != nil {
|
|
err = m.listener.Close()
|
|
}
|
|
})
|
|
return err
|
|
}
|
|
|
|
// filterTCPConn filters a TCP connection, rejecting it if this function errors.
|
|
func (m *MConnTransport) filterTCPConn(tcpConn net.Conn) error {
|
|
if m.conns.Has(tcpConn) {
|
|
return ErrRejected{conn: tcpConn, isDuplicate: true}
|
|
}
|
|
|
|
host, _, err := net.SplitHostPort(tcpConn.RemoteAddr().String())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
ip := net.ParseIP(host)
|
|
if ip == nil {
|
|
return fmt.Errorf("connection address has invalid IP address %q", host)
|
|
}
|
|
|
|
// Apply filter callbacks.
|
|
chErr := make(chan error, len(m.connFilters))
|
|
for _, connFilter := range m.connFilters {
|
|
go func(connFilter ConnFilterFunc) {
|
|
chErr <- connFilter(m.conns, tcpConn, []net.IP{ip})
|
|
}(connFilter)
|
|
}
|
|
|
|
for i := 0; i < cap(chErr); i++ {
|
|
select {
|
|
case err := <-chErr:
|
|
if err != nil {
|
|
return ErrRejected{conn: tcpConn, err: err, isFiltered: true}
|
|
}
|
|
case <-time.After(m.filterTimeout):
|
|
return ErrFilterTimeout{}
|
|
}
|
|
|
|
}
|
|
|
|
// FIXME: Doesn't really make sense to set this here, but we preserve the
|
|
// behavior from the previous P2P transport implementation. This should
|
|
// be moved to the router.
|
|
m.conns.Set(tcpConn, []net.IP{ip})
|
|
return nil
|
|
}
|
|
|
|
// normalizeEndpoint normalizes and validates an endpoint.
|
|
func (m *MConnTransport) normalizeEndpoint(endpoint *Endpoint) error {
|
|
if endpoint == nil {
|
|
return errors.New("nil endpoint")
|
|
}
|
|
if err := endpoint.Validate(); err != nil {
|
|
return err
|
|
}
|
|
if endpoint.Protocol == "" {
|
|
endpoint.Protocol = MConnProtocol
|
|
}
|
|
if endpoint.Protocol != MConnProtocol {
|
|
return fmt.Errorf("unsupported protocol %q", endpoint.Protocol)
|
|
}
|
|
if len(endpoint.IP) == 0 {
|
|
return errors.New("endpoint must have an IP address")
|
|
}
|
|
if endpoint.Path != "" {
|
|
return fmt.Errorf("endpoint cannot have path (got %q)", endpoint.Path)
|
|
}
|
|
if endpoint.Port == 0 {
|
|
endpoint.Port = 26657
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// mConnConnection implements Connection for MConnTransport. It takes a base TCP
|
|
// connection and upgrades it to MConnection over an encrypted SecretConnection.
|
|
type mConnConnection struct {
|
|
logger log.Logger
|
|
transport *MConnTransport
|
|
secretConn *conn.SecretConnection
|
|
mConn *conn.MConnection
|
|
|
|
peerInfo NodeInfo
|
|
|
|
closeOnce sync.Once
|
|
chReceive chan mConnMessage
|
|
chError chan error
|
|
chClose chan struct{}
|
|
}
|
|
|
|
// mConnMessage passes MConnection messages through internal channels.
|
|
type mConnMessage struct {
|
|
channelID byte
|
|
payload []byte
|
|
}
|
|
|
|
// newMConnConnection creates a new mConnConnection by handshaking
|
|
// with a peer.
|
|
func newMConnConnection(
|
|
transport *MConnTransport,
|
|
tcpConn net.Conn,
|
|
expectPeerID NodeID,
|
|
) (c *mConnConnection, err error) {
|
|
// FIXME: Since the MConnection code panics, we need to recover here
|
|
// and turn it into an error. Be careful not to alias err, so we can
|
|
// update it from within this function. We should remove panics instead.
|
|
defer func() {
|
|
if r := recover(); r != nil {
|
|
err = ErrRejected{
|
|
conn: tcpConn,
|
|
err: fmt.Errorf("recovered from panic: %v", r),
|
|
isAuthFailure: true,
|
|
}
|
|
}
|
|
}()
|
|
|
|
err = tcpConn.SetDeadline(time.Now().Add(transport.handshakeTimeout))
|
|
if err != nil {
|
|
err = ErrRejected{
|
|
conn: tcpConn,
|
|
err: fmt.Errorf("secret conn failed: %v", err),
|
|
isAuthFailure: true,
|
|
}
|
|
return
|
|
}
|
|
|
|
c = &mConnConnection{
|
|
transport: transport,
|
|
chReceive: make(chan mConnMessage),
|
|
chError: make(chan error),
|
|
chClose: make(chan struct{}),
|
|
}
|
|
c.secretConn, err = conn.MakeSecretConnection(tcpConn, transport.privKey)
|
|
if err != nil {
|
|
err = ErrRejected{
|
|
conn: tcpConn,
|
|
err: fmt.Errorf("secret conn failed: %v", err),
|
|
isAuthFailure: true,
|
|
}
|
|
return
|
|
}
|
|
c.peerInfo, err = c.handshake()
|
|
if err != nil {
|
|
err = ErrRejected{
|
|
conn: tcpConn,
|
|
err: fmt.Errorf("handshake failed: %v", err),
|
|
isAuthFailure: true,
|
|
}
|
|
return
|
|
}
|
|
|
|
// Validate node info.
|
|
// FIXME: All of the ID verification code below should be moved to the
|
|
// router once implemented.
|
|
err = c.peerInfo.Validate()
|
|
if err != nil {
|
|
err = ErrRejected{
|
|
conn: tcpConn,
|
|
err: err,
|
|
isNodeInfoInvalid: true,
|
|
}
|
|
return
|
|
}
|
|
|
|
// For outgoing conns, ensure connection key matches dialed key.
|
|
if expectPeerID != "" {
|
|
peerID := NodeIDFromPubKey(c.PubKey())
|
|
if expectPeerID != peerID {
|
|
err = ErrRejected{
|
|
conn: tcpConn,
|
|
id: peerID,
|
|
err: fmt.Errorf(
|
|
"conn.ID (%v) dialed ID (%v) mismatch",
|
|
peerID,
|
|
expectPeerID,
|
|
),
|
|
isAuthFailure: true,
|
|
}
|
|
return
|
|
}
|
|
}
|
|
|
|
// Reject self.
|
|
if transport.nodeInfo.ID() == c.peerInfo.ID() {
|
|
err = ErrRejected{
|
|
addr: *NewNetAddress(c.peerInfo.ID(), c.secretConn.RemoteAddr()),
|
|
conn: tcpConn,
|
|
id: c.peerInfo.ID(),
|
|
isSelf: true,
|
|
}
|
|
return
|
|
}
|
|
|
|
err = transport.nodeInfo.CompatibleWith(c.peerInfo)
|
|
if err != nil {
|
|
err = ErrRejected{
|
|
conn: tcpConn,
|
|
err: err,
|
|
id: c.peerInfo.ID(),
|
|
isIncompatible: true,
|
|
}
|
|
return
|
|
}
|
|
|
|
err = tcpConn.SetDeadline(time.Time{})
|
|
if err != nil {
|
|
err = ErrRejected{
|
|
conn: tcpConn,
|
|
err: fmt.Errorf("secret conn failed: %v", err),
|
|
isAuthFailure: true,
|
|
}
|
|
return
|
|
}
|
|
|
|
// Set up the MConnection wrapper
|
|
c.mConn = conn.NewMConnectionWithConfig(
|
|
c.secretConn,
|
|
transport.channelDescs,
|
|
c.onReceive,
|
|
c.onError,
|
|
transport.mConnConfig,
|
|
)
|
|
// FIXME: Log format is set up for compatibility with existing peer code.
|
|
c.logger = transport.logger.With("peer", c.RemoteEndpoint().NetAddress())
|
|
c.mConn.SetLogger(c.logger)
|
|
err = c.mConn.Start()
|
|
return c, err
|
|
}
|
|
|
|
// handshake performs an MConn handshake, returning the peer's node info.
|
|
func (c *mConnConnection) handshake() (NodeInfo, error) {
|
|
var pbNodeInfo p2pproto.NodeInfo
|
|
chErr := make(chan error, 2)
|
|
go func() {
|
|
_, err := protoio.NewDelimitedWriter(c.secretConn).WriteMsg(c.transport.nodeInfo.ToProto())
|
|
chErr <- err
|
|
}()
|
|
go func() {
|
|
_, err := protoio.NewDelimitedReader(c.secretConn, MaxNodeInfoSize()).ReadMsg(&pbNodeInfo)
|
|
chErr <- err
|
|
}()
|
|
for i := 0; i < cap(chErr); i++ {
|
|
if err := <-chErr; err != nil {
|
|
return NodeInfo{}, err
|
|
}
|
|
}
|
|
|
|
return NodeInfoFromProto(&pbNodeInfo)
|
|
}
|
|
|
|
// onReceive is a callback for MConnection received messages.
|
|
func (c *mConnConnection) onReceive(channelID byte, payload []byte) {
|
|
select {
|
|
case c.chReceive <- mConnMessage{channelID: channelID, payload: payload}:
|
|
case <-c.chClose:
|
|
}
|
|
}
|
|
|
|
// onError is a callback for MConnection errors. The error is passed to
|
|
// chError, which is only consumed by ReceiveMessage() for parity with
|
|
// the old MConnection behavior.
|
|
func (c *mConnConnection) onError(e interface{}) {
|
|
err, ok := e.(error)
|
|
if !ok {
|
|
err = fmt.Errorf("%v", err)
|
|
}
|
|
select {
|
|
case c.chError <- err:
|
|
case <-c.chClose:
|
|
}
|
|
}
|
|
|
|
// String displays connection information.
|
|
// FIXME: This is here for backwards compatibility with existing code,
|
|
// it should probably just return RemoteEndpoint().String(), if anything.
|
|
func (c *mConnConnection) String() string {
|
|
endpoint := c.RemoteEndpoint()
|
|
return fmt.Sprintf("MConn{%v:%v}", endpoint.IP, endpoint.Port)
|
|
}
|
|
|
|
// SendMessage implements Connection.
|
|
func (c *mConnConnection) SendMessage(channelID byte, msg []byte) (bool, error) {
|
|
// We don't check chError here, to preserve old MConnection behavior.
|
|
select {
|
|
case <-c.chClose:
|
|
return false, io.EOF
|
|
default:
|
|
return c.mConn.Send(channelID, msg), nil
|
|
}
|
|
}
|
|
|
|
// TrySendMessage implements Connection.
|
|
func (c *mConnConnection) TrySendMessage(channelID byte, msg []byte) (bool, error) {
|
|
// We don't check chError here, to preserve old MConnection behavior.
|
|
select {
|
|
case <-c.chClose:
|
|
return false, io.EOF
|
|
default:
|
|
return c.mConn.TrySend(channelID, msg), nil
|
|
}
|
|
}
|
|
|
|
// ReceiveMessage implements Connection.
|
|
func (c *mConnConnection) ReceiveMessage() (byte, []byte, error) {
|
|
select {
|
|
case err := <-c.chError:
|
|
return 0, nil, err
|
|
case <-c.chClose:
|
|
return 0, nil, io.EOF
|
|
case msg := <-c.chReceive:
|
|
return msg.channelID, msg.payload, nil
|
|
}
|
|
}
|
|
|
|
// NodeInfo implements Connection.
|
|
func (c *mConnConnection) NodeInfo() NodeInfo {
|
|
return c.peerInfo
|
|
}
|
|
|
|
// PubKey implements Connection.
|
|
func (c *mConnConnection) PubKey() crypto.PubKey {
|
|
return c.secretConn.RemotePubKey()
|
|
}
|
|
|
|
// LocalEndpoint implements Connection.
|
|
func (c *mConnConnection) LocalEndpoint() Endpoint {
|
|
// FIXME: For compatibility with existing P2P tests we need to
|
|
// handle non-TCP connections. This should be removed.
|
|
endpoint := Endpoint{
|
|
Protocol: MConnProtocol,
|
|
PeerID: c.transport.nodeInfo.ID(),
|
|
}
|
|
if addr, ok := c.secretConn.LocalAddr().(*net.TCPAddr); ok {
|
|
endpoint.IP = addr.IP
|
|
endpoint.Port = uint16(addr.Port)
|
|
}
|
|
return endpoint
|
|
}
|
|
|
|
// RemoteEndpoint implements Connection.
|
|
func (c *mConnConnection) RemoteEndpoint() Endpoint {
|
|
// FIXME: For compatibility with existing P2P tests we need to
|
|
// handle non-TCP connections. This should be removed.
|
|
endpoint := Endpoint{
|
|
Protocol: MConnProtocol,
|
|
PeerID: c.peerInfo.ID(),
|
|
}
|
|
if addr, ok := c.secretConn.RemoteAddr().(*net.TCPAddr); ok {
|
|
endpoint.IP = addr.IP
|
|
endpoint.Port = uint16(addr.Port)
|
|
}
|
|
return endpoint
|
|
}
|
|
|
|
// Status implements Connection.
|
|
func (c *mConnConnection) Status() conn.ConnectionStatus {
|
|
return c.mConn.Status()
|
|
}
|
|
|
|
// Close implements Connection.
|
|
func (c *mConnConnection) Close() error {
|
|
c.transport.conns.RemoveAddr(c.secretConn.RemoteAddr())
|
|
var err error
|
|
c.closeOnce.Do(func() {
|
|
err = c.mConn.Stop()
|
|
close(c.chClose)
|
|
})
|
|
return err
|
|
}
|
|
|
|
// FlushClose implements Connection.
|
|
func (c *mConnConnection) FlushClose() error {
|
|
c.transport.conns.RemoveAddr(c.secretConn.RemoteAddr())
|
|
c.closeOnce.Do(func() {
|
|
c.mConn.FlushStop()
|
|
close(c.chClose)
|
|
})
|
|
return nil
|
|
}
|