net

package
v0.2.2-pipeline-release.2 Latest Latest
Warning

This package is not in the latest version of its module.

 Go to latest Published: Sep 22, 2021 License: MIT Imports: 21 Imported by: 0

Documentation

Overview

Package net manages the accepting network connections/messages and routing the data upward for the protocols to consume.

Index

Constants

View Source 
const (
	// ReadBufferSize const is the default value used to set the socket option SO_RCVBUF
	ReadBufferSize = 5 * 1024 * 1024
	// WriteBufferSize const is the default value used to set the socket option SO_SNDBUF
	WriteBufferSize = 5 * 1024 * 1024
	// TCPKeepAlive sets whether KeepAlive is active or not on the socket
	TCPKeepAlive = true
	// TCPKeepAlivePeriod sets the interval of KeepAlive
	TCPKeepAlivePeriod = 10 * time.Second
)
View Source 
const DefaultMessageQueueSize uint = 5120

DefaultMessageQueueSize is the buffer size of each queue mentioned above. (queues are buffered channels)

View Source 
const DefaultQueueCount uint = 6

DefaultQueueCount is the default number of messages queue we hold. messages queues are used to serialize message receiving

View Source 
const MessageQueueSize = 250

MessageQueueSize is the size for queue of messages before pushing them on the socket

Variables

View Source 
var (
	// ErrAlreadyClosed is an error for when `Close` is called on a closed connection.
	ErrAlreadyClosed = errors.New("p2p: connection is already closed")
	// ErrClosed is returned when connection already closed.
	ErrClosed = errors.New("p2p: connection closed")
	// ErrQueueFull is returned when the outbound message queue is full.
	ErrQueueFull = errors.New("p2p: outbound message queue is full, dropping peer")
	// ErrTriedToSetupExistingConn occurs when handshake packet is sent twice on a connection
	ErrTriedToSetupExistingConn = errors.New("p2p: tried to setup existing connection")
	// ErrMsgExceededLimit occurs when a received message size exceeds the defined message size
	ErrMsgExceededLimit = errors.New("p2p: message size exceeded limit")
)
View Source 
var IPv4LoopbackAddress = net.IP{127, 0, 0, 1}

IPv4LoopbackAddress is a local IPv4 loopback

Functions

func NodeAddr 

func NodeAddr(info *node.Info) *net.UDPAddr

NodeAddr makes a UDPAddr from a Info struct

func Temporary 

func Temporary(err error) bool

Temporary checks whether the given error should be considered temporary.

Types

type Connection 

type Connection interface {
	fmt.Stringer

	ID() string
	RemotePublicKey() p2pcrypto.PublicKey
	SetRemotePublicKey(key p2pcrypto.PublicKey)
	Created() time.Time

	RemoteAddr() net.Addr

	Session() NetworkSession
	SetSession(session NetworkSession)

	Send(ctx context.Context, m []byte) error
	Close() error
	Closed() bool
}

Connection is an interface stating the API of all secured connections in the system

type ConnectionMock 

type ConnectionMock struct {
	Addr net.Addr
	// contains filtered or unexported fields
}

ConnectionMock mocks connections.

func NewConnectionMock 

func NewConnectionMock(key p2pcrypto.PublicKey) *ConnectionMock

NewConnectionMock creates a ConnectionMock.

func (*ConnectionMock) Close 

func (cm *ConnectionMock) Close() error

Close mocks the interface.

func (ConnectionMock) Closed 

func (cm ConnectionMock) Closed() bool

Closed mocks the interface.

func (ConnectionMock) Created added in v0.1.2 

func (cm ConnectionMock) Created() time.Time

Created mocks the connection interface.

func (ConnectionMock) ID 

func (cm ConnectionMock) ID() string

ID mocks the connection interface.

func (ConnectionMock) IncomingChannel 

func (cm ConnectionMock) IncomingChannel() chan []byte

IncomingChannel mocks the interface.

func (*ConnectionMock) RemoteAddr 

func (cm *ConnectionMock) RemoteAddr() net.Addr

RemoteAddr mocks the interface.

func (ConnectionMock) RemotePublicKey 

func (cm ConnectionMock) RemotePublicKey() p2pcrypto.PublicKey

RemotePublicKey mocks the interface.

func (*ConnectionMock) Send 

func (cm *ConnectionMock) Send(context.Context, []byte) error

Send mocks the interface.

func (ConnectionMock) SendCount 

func (cm ConnectionMock) SendCount() int32

SendCount mutates the mock.

func (ConnectionMock) Session 

func (cm ConnectionMock) Session() NetworkSession

Session mocks the interface.

func (*ConnectionMock) SetCreated added in v0.1.2 

func (cm *ConnectionMock) SetCreated(time2 time.Time)

SetCreated mutates the mock.

func (*ConnectionMock) SetRemotePublicKey 

func (cm *ConnectionMock) SetRemotePublicKey(key p2pcrypto.PublicKey)

SetRemotePublicKey mutates the mock.

func (*ConnectionMock) SetSendDelay 

func (cm *ConnectionMock) SetSendDelay(delay time.Duration)

SetSendDelay mutates the mock.

func (*ConnectionMock) SetSendResult 

func (cm *ConnectionMock) SetSendResult(err error)

SetSendResult mutates the mock.

func (*ConnectionMock) SetSession 

func (cm *ConnectionMock) SetSession(session NetworkSession)

SetSession mutates the mock.

func (ConnectionMock) String 

func (cm ConnectionMock) String() string

String mocks the interface

type ConnectionSource 

type ConnectionSource int

ConnectionSource specifies the connection originator - local or remote node. 

const (
	Local ConnectionSource = iota
	Remote
)

ConnectionSource values

type ConnectionWithErr 

type ConnectionWithErr struct {
	Conn Connection
	Err  error
}

ConnectionWithErr is a pair of Connection and an error occurred within the connection

type FormattedConnection 

type FormattedConnection struct {
	// contains filtered or unexported fields
}

FormattedConnection is an io.Writer and an io.Closer A network connection supporting full-duplex messaging

func (*FormattedConnection) Close 

func (c *FormattedConnection) Close() error

Close closes the connection (implements io.Closer). It is go safe.

func (*FormattedConnection) Closed 

func (c *FormattedConnection) Closed() bool

Closed returns whether the connection is closed

func (*FormattedConnection) Created added in v0.1.2 

func (c *FormattedConnection) Created() time.Time

Created is the time the connection was created

func (*FormattedConnection) ID 

func (c *FormattedConnection) ID() string

ID returns the channel's ID

func (*FormattedConnection) RemoteAddr 

func (c *FormattedConnection) RemoteAddr() net.Addr

RemoteAddr returns the channel's remote peer address

func (*FormattedConnection) RemotePublicKey 

func (c *FormattedConnection) RemotePublicKey() p2pcrypto.PublicKey

RemotePublicKey returns the remote peer's public key

func (*FormattedConnection) Send 

func (c *FormattedConnection) Send(ctx context.Context, m []byte) error

Send pushes a message into the queue if the connection is not closed.

func (*FormattedConnection) Session 

func (c *FormattedConnection) Session() NetworkSession

Session returns the network session

func (*FormattedConnection) SetRemotePublicKey 

func (c *FormattedConnection) SetRemotePublicKey(key p2pcrypto.PublicKey)

SetRemotePublicKey sets the remote peer's public key

func (*FormattedConnection) SetSession 

func (c *FormattedConnection) SetSession(session NetworkSession)

SetSession sets the network session

func (*FormattedConnection) String 

func (c *FormattedConnection) String() string

String returns a string describing the connection

type HandshakeData 

type HandshakeData struct {
	ClientVersion string
	NetworkID     uint32
	Port          uint16
}

HandshakeData is the handshake message struct

type IncomingMessageEvent 

type IncomingMessageEvent struct {
	Conn      Connection
	Message   []byte
	RequestID string
}

IncomingMessageEvent is the event reported on new incoming message, it contains the message and the Connection carrying the message

type ManagedConnection 

type ManagedConnection interface {
	Connection
	// contains filtered or unexported methods
}

ManagedConnection in an interface extending Connection with some internal methods that are required for Net to manage Connections

type MsgConnection added in v0.1.2 

type MsgConnection struct {
	// contains filtered or unexported fields
}

MsgConnection is an io.Writer and an io.Closer A network connection supporting full-duplex messaging It resembles the Connection interface but suits a packet oriented socket.

func (*MsgConnection) Close added in v0.1.2 

func (c *MsgConnection) Close() error

Close closes the connection (implements io.Closer). It is go safe.

func (*MsgConnection) Closed added in v0.1.2 

func (c *MsgConnection) Closed() bool

Closed returns whether the connection is closed

func (*MsgConnection) Created added in v0.1.2 

func (c *MsgConnection) Created() time.Time

Created saves the time when the connection was created

func (*MsgConnection) ID added in v0.1.2 

func (c *MsgConnection) ID() string

ID returns the channel's ID

func (*MsgConnection) RemoteAddr added in v0.1.2 

func (c *MsgConnection) RemoteAddr() net.Addr

RemoteAddr returns the channel's remote peer address

func (*MsgConnection) RemotePublicKey added in v0.1.2 

func (c *MsgConnection) RemotePublicKey() p2pcrypto.PublicKey

RemotePublicKey returns the remote peer's public key

func (*MsgConnection) Send added in v0.1.2 

func (c *MsgConnection) Send(ctx context.Context, m []byte) error

Send pushes a message to the messages queue

func (*MsgConnection) Session added in v0.1.2 

func (c *MsgConnection) Session() NetworkSession

Session returns the network session

func (*MsgConnection) SetRemotePublicKey added in v0.1.2 

func (c *MsgConnection) SetRemotePublicKey(key p2pcrypto.PublicKey)

SetRemotePublicKey sets the remote peer's public key

func (*MsgConnection) SetSession added in v0.1.2 

func (c *MsgConnection) SetSession(session NetworkSession)

SetSession sets the network session

func (*MsgConnection) String added in v0.1.2 

func (c *MsgConnection) String() string

String returns a string describing the connection

type Net 

type Net struct {
	// contains filtered or unexported fields
}

Net is a connection factory able to dial remote endpoints Net clients should register all callbacks Connections may be initiated by Dial() or by remote clients connecting to the listen address It provides full duplex messaging functionality over the same tcp/ip connection Net has no channel events processing loops - clients are responsible for polling these channels and popping events from them

func NewNet 

func NewNet(ctx context.Context, conf config.Config, localEntity node.LocalNode, logger log.Log) (*Net, error)

NewNet creates a new network. It attempts to tcp listen on address. e.g. localhost:1234 .

func (*Net) Dial 

func (n *Net) Dial(ctx context.Context, address net.Addr, remotePubkey p2pcrypto.PublicKey) (Connection, error)

Dial a remote server with provided time out address:: net.Addr Returns established connection that local clients can send messages to or error if failed to establish a connection, currently only secured connections are supported

func (*Net) EnqueueMessage 

func (n *Net) EnqueueMessage(ctx context.Context, event IncomingMessageEvent)

EnqueueMessage inserts a message into a queue, to decide on which queue to send the message to it sum the remote public key bytes as integer to segment to queueCount queues.

func (*Net) HandlePreSessionIncomingMessage 

func (n *Net) HandlePreSessionIncomingMessage(c Connection, message []byte) error

HandlePreSessionIncomingMessage establishes session with the remote peer and update the Connection with the new session

func (*Net) IncomingMessages 

func (n *Net) IncomingMessages() []chan IncomingMessageEvent

IncomingMessages returns a slice of channels which incoming messages are delivered on the receiver should iterate on all the channels and read all messages. to sync messages order but enable parallel messages handling.

func (*Net) LocalAddr 

func (n *Net) LocalAddr() net.Addr

LocalAddr returns the local listening address. panics before calling Start or if Start errored

func (*Net) LocalNode 

func (n *Net) LocalNode() node.LocalNode

LocalNode return's the local node descriptor

func (*Net) Logger 

func (n *Net) Logger() log.Log

Logger returns a reference to logger

func (*Net) NetworkID 

func (n *Net) NetworkID() uint32

NetworkID retuers Net's network ID

func (*Net) Shutdown 

func (n *Net) Shutdown()

Shutdown initiate a graceful closing of the TCP listener and all other internal routines

func (*Net) Start 

func (n *Net) Start(ctx context.Context, listener net.Listener)

Start begins accepting connections from the listener socket

func (*Net) SubscribeClosingConnections 

func (n *Net) SubscribeClosingConnections(f func(context.Context, ConnectionWithErr))

SubscribeClosingConnections registers a callback for a new connection event. all registered callbacks are called before moving.

func (*Net) SubscribeOnNewRemoteConnections 

func (n *Net) SubscribeOnNewRemoteConnections(f func(event NewConnectionEvent))

SubscribeOnNewRemoteConnections registers a callback for a new connection event. all registered callbacks are called before moving.

type NetworkMock 

type NetworkMock struct {
	// contains filtered or unexported fields
}

NetworkMock is a mock struct

func NewNetworkMock 

func NewNetworkMock(tb testing.TB) *NetworkMock

NewNetworkMock is a mock

func (*NetworkMock) Dial 

func (n *NetworkMock) Dial(ctx context.Context, address net.Addr, remotePublicKey p2pcrypto.PublicKey) (Connection, error)

Dial dials

func (*NetworkMock) DialCount 

func (n *NetworkMock) DialCount() int32

DialCount gets the dial count

func (*NetworkMock) EnqueueMessage 

func (n *NetworkMock) EnqueueMessage(ctx context.Context, event IncomingMessageEvent)

EnqueueMessage return channel of IncomingMessages

func (*NetworkMock) HandlePreSessionIncomingMessage 

func (n *NetworkMock) HandlePreSessionIncomingMessage(c Connection, msg []byte) error

HandlePreSessionIncomingMessage and stuff

func (*NetworkMock) IncomingMessages 

func (n *NetworkMock) IncomingMessages() []chan IncomingMessageEvent

IncomingMessages return channel of IncomingMessages

func (*NetworkMock) Logger 

func (n *NetworkMock) Logger() log.Log

Logger return the logger

func (*NetworkMock) NetworkID 

func (n *NetworkMock) NetworkID() uint32

NetworkID is netid

func (NetworkMock) PreSessionCount 

func (n NetworkMock) PreSessionCount() int32

PreSessionCount counts

func (NetworkMock) PublishClosingConnection 

func (n NetworkMock) PublishClosingConnection(con ConnectionWithErr)

PublishClosingConnection is a hack to expose the above method in the mock but still impl the same interface

func (NetworkMock) PublishNewRemoteConnection 

func (n NetworkMock) PublishNewRemoteConnection(nce NewConnectionEvent)

PublishNewRemoteConnection and stuff

func (*NetworkMock) SetDialDelayMs 

func (n *NetworkMock) SetDialDelayMs(delay int8)

SetDialDelayMs sets delay

func (*NetworkMock) SetDialResult 

func (n *NetworkMock) SetDialResult(err error)

SetDialResult is a mock

func (*NetworkMock) SetNextDialSessionID 

func (n *NetworkMock) SetNextDialSessionID(sID []byte)

SetNextDialSessionID mutates the mock to change the next returned session id

func (*NetworkMock) SetPreSessionResult 

func (n *NetworkMock) SetPreSessionResult(err error)

SetPreSessionResult does this

func (*NetworkMock) SubscribeClosingConnections 

func (n *NetworkMock) SubscribeClosingConnections(f func(context.Context, ConnectionWithErr))

SubscribeClosingConnections subscribes on new connections

func (*NetworkMock) SubscribeOnNewRemoteConnections 

func (n *NetworkMock) SubscribeOnNewRemoteConnections(f func(event NewConnectionEvent))

SubscribeOnNewRemoteConnections subscribes on new connections

type NetworkSession 

type NetworkSession interface {
	ID() p2pcrypto.PublicKey // Unique session id, currently the peer pubkey TODO: @noam use pubkey from conn and remove this

	OpenMessage(boxedMessage []byte) ([]byte, error) // decrypt data using session dec key
	SealMessage(message []byte) []byte               // encrypt data using session enc key
}

NetworkSession is an authenticated network session between 2 peers. Sessions may be used between 'connections' until they expire. Session provides the encryptor/decryptor for all messages exchanged between 2 peers. enc/dec is using an ephemeral sym key exchanged securely between the peers via the handshake protocol The handshake protocol goal is to create an authenticated network session.

func NewNetworkSession 

func NewNetworkSession(sharedSecret p2pcrypto.SharedSecret, peerPubkey p2pcrypto.PublicKey) NetworkSession

NewNetworkSession creates a new network session based on provided data

type NewConnectionEvent 

type NewConnectionEvent struct {
	Conn Connection
	Node *node.Info
}

NewConnectionEvent is a struct holding a new created connection and a node info.

type ReadWriteCloserMock 

type ReadWriteCloserMock struct {
}

ReadWriteCloserMock is a mock of ReadWriteCloserMock

func (ReadWriteCloserMock) Close 

func (m ReadWriteCloserMock) Close() error

Close mocks close

func (ReadWriteCloserMock) Read 

func (m ReadWriteCloserMock) Read(p []byte) (n int, err error)

Read reads something

func (ReadWriteCloserMock) RemoteAddr 

func (m ReadWriteCloserMock) RemoteAddr() net.Addr

RemoteAddr mocks remote addr return

func (ReadWriteCloserMock) Write 

func (m ReadWriteCloserMock) Write(p []byte) (n int, err error)

Write mocks write

type SessionMock 

type SessionMock struct {
	SealMessageFunc func(message []byte) []byte
	OpenMessageFunc func(boxedMessage []byte) ([]byte, error)
	// contains filtered or unexported fields
}

SessionMock mocks NetworkSession.

func NewSessionMock 

func NewSessionMock(pubkey p2pcrypto.PublicKey) *SessionMock

NewSessionMock creates a new mock for given public key.

func (SessionMock) ID 

func (sm SessionMock) ID() p2pcrypto.PublicKey

ID is a mock.

func (SessionMock) OpenMessage 

func (sm SessionMock) OpenMessage(boxedMessage []byte) ([]byte, error)

OpenMessage is a mock.

func (SessionMock) SealMessage 

func (sm SessionMock) SealMessage(message []byte) []byte

SealMessage is a mock.

type UDPListener added in v0.1.6 

type UDPListener interface {
	LocalAddr() net.Addr
	Close() error
	WriteToUDP(final []byte, addr *net.UDPAddr) (int, error)
	ReadFrom(p []byte) (n int, addr net.Addr, err error)
	WriteTo(p []byte, addr net.Addr) (n int, err error)
	SetDeadline(t time.Time) error
	SetReadDeadline(t time.Time) error
	SetWriteDeadline(t time.Time) error
}

UDPListener is the api required for listening on udp messages.

type UDPMessageEvent 

type UDPMessageEvent struct {
	From     p2pcrypto.PublicKey
	FromAddr net.Addr
	Message  []byte
}

UDPMessageEvent is an event about a udp message. passed through a channel

type UDPNet 

type UDPNet struct {
	// contains filtered or unexported fields
}

UDPNet is used to listen on or send udp messages

func NewUDPNet 

func NewUDPNet(ctx context.Context, config config.Config, localEntity node.LocalNode, log log.Log) (*UDPNet, error)

NewUDPNet creates a UDPNet

func (*UDPNet) Dial added in v0.1.2 

func (n *UDPNet) Dial(ctx context.Context, address net.Addr, remotePublicKey p2pcrypto.PublicKey) (Connection, error)

Dial creates a Connection interface which is wrapped around a udp socket with a session and start listening on messages from it. it uses the `connect` syscall

func (*UDPNet) EnqueueMessage added in v0.1.2 

func (n *UDPNet) EnqueueMessage(ctx context.Context, ime IncomingMessageEvent)

EnqueueMessage pushes an incoming message event into the queue

func (*UDPNet) HandlePreSessionIncomingMessage added in v0.1.2 

func (n *UDPNet) HandlePreSessionIncomingMessage(c Connection, msg []byte) error

HandlePreSessionIncomingMessage is used to satisfy an api similar to the tcpnet. not used here.

func (*UDPNet) IncomingMessages 

func (n *UDPNet) IncomingMessages() chan IncomingMessageEvent

IncomingMessages is a channel where incoming UDPMessagesEvents will stream

func (*UDPNet) LocalAddr 

func (n *UDPNet) LocalAddr() net.Addr

LocalAddr returns the local listening addr, will panic before running Start. or if start errored

func (*UDPNet) NetworkID added in v0.1.2 

func (n *UDPNet) NetworkID() uint32

NetworkID returns the network id given and used for creating a session.

func (*UDPNet) Send 

func (n *UDPNet) Send(to *node.Info, data []byte) error

Send writes a udp packet to the target with the given data

func (*UDPNet) Shutdown 

func (n *UDPNet) Shutdown()

Shutdown stops listening and closes the connection

func (*UDPNet) Start 

func (n *UDPNet) Start(ctx context.Context, listener UDPListener)

Start will start reading messages from the udp socket and pass them up the channels

func (*UDPNet) SubscribeClosingConnections added in v0.1.2 

func (n *UDPNet) SubscribeClosingConnections(f func(ctx context.Context, connection ConnectionWithErr))

SubscribeClosingConnections registers a callback for a new connection event. all registered callbacks are called before moving.

func (*UDPNet) SubscribeOnNewRemoteConnections added in v0.1.2 

func (n *UDPNet) SubscribeOnNewRemoteConnections(f func(event NewConnectionEvent))

SubscribeOnNewRemoteConnections registers a callback for a new connection event. all registered callbacks are called before moving.

Source Files

View all Source files

Directories

Path Synopsis
delimited
Package delimited implements a reader and writer for simple streams of length-delimited byte records.
 Package delimited implements a reader and writer for simple streams of length-delimited byte records.

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL
go.dev uses cookies from Google to deliver and enhance the quality of its services and to analyze traffic. Learn more.