156 lines
4.0 KiB
Go
156 lines
4.0 KiB
Go
package router
|
|
|
|
import (
|
|
"net"
|
|
)
|
|
|
|
type Overlay interface {
|
|
// Enhance a features map with overlay-related features
|
|
AddFeaturesTo(map[string]string)
|
|
|
|
// Prepare on overlay connection. The connection should remain
|
|
// passive until it has been Confirm()ed.
|
|
PrepareConnection(OverlayConnectionParams) (OverlayConnection, error)
|
|
|
|
// Obtain diagnostic information specific to the overlay
|
|
Diagnostics() interface{}
|
|
}
|
|
|
|
// Interface to overlay network packet handling
|
|
type NetworkOverlay interface {
|
|
Overlay
|
|
|
|
// The routes have changed, so any cached information should
|
|
// be discarded.
|
|
InvalidateRoutes()
|
|
|
|
// A mapping of a short id to a peer has changed
|
|
InvalidateShortIDs()
|
|
|
|
// Start consuming forwarded packets.
|
|
StartConsumingPackets(*Peer, *Peers, OverlayConsumer) error
|
|
}
|
|
|
|
type OverlayConnectionParams struct {
|
|
RemotePeer *Peer
|
|
|
|
// The local address of the corresponding TCP connection. Used to
|
|
// derive the local IP address for sending. May differ for
|
|
// different overlay connections.
|
|
LocalAddr *net.TCPAddr
|
|
|
|
// The remote address of the corresponding TCP connection. Used to
|
|
// determine the address to send to, but only if the TCP
|
|
// connection is outbound. Otherwise the Overlay needs to discover
|
|
// it (e.g. from incoming datagrams).
|
|
RemoteAddr *net.TCPAddr
|
|
|
|
// Is the corresponding TCP connection outbound?
|
|
Outbound bool
|
|
|
|
// Unique identifier for this connection
|
|
ConnUID uint64
|
|
|
|
// Session key, if connection is encrypted; nil otherwise
|
|
SessionKey *[32]byte
|
|
|
|
// Function to send a control message to the counterpart
|
|
// overlay connection.
|
|
SendControlMessage func(tag byte, msg []byte) error
|
|
|
|
// Features passed at connection initiation
|
|
Features map[string]string
|
|
}
|
|
|
|
// When a consumer is called, the decoder will already have been used
|
|
// to decode the frame.
|
|
type OverlayConsumer func(ForwardPacketKey) FlowOp
|
|
|
|
// All of the machinery to manage overlay connectivity to a particular
|
|
// peer
|
|
type OverlayConnection interface {
|
|
// Confirm that the connection is really wanted, and so the
|
|
// Overlay should begin heartbeats etc. to verify the operation of
|
|
// the overlay connection.
|
|
Confirm()
|
|
|
|
// A channel indicating that the overlay connection is
|
|
// established, i.e. its operation has been confirmed.
|
|
EstablishedChannel() <-chan struct{}
|
|
|
|
// A channel indicating an error from the overlay connection. The
|
|
// overlay connection is not expected to be operational after the
|
|
// first error, so the channel only needs to buffer a single
|
|
// error.
|
|
ErrorChannel() <-chan error
|
|
|
|
Stop()
|
|
|
|
// Handle a message from the peer. 'tag' exists for
|
|
// compatibility, and should always be
|
|
// ProtocolOverlayControlMessage for non-sleeve overlays.
|
|
ControlMessage(tag byte, msg []byte)
|
|
|
|
// User facing overlay name
|
|
DisplayName() string
|
|
}
|
|
|
|
// All of the machinery to forward packets to a particular peer
|
|
type OverlayForwarder interface {
|
|
OverlayConnection
|
|
// Forward a packet across the connection. May be called as soon
|
|
// as the overlay connection is created, in particular before
|
|
// Confirm(). The return value nil means the key could not be
|
|
// handled by this forwarder.
|
|
Forward(ForwardPacketKey) FlowOp
|
|
}
|
|
|
|
type NullOverlay struct{}
|
|
|
|
func (NullOverlay) AddFeaturesTo(map[string]string) {
|
|
}
|
|
|
|
func (NullOverlay) PrepareConnection(OverlayConnectionParams) (OverlayConnection, error) {
|
|
return NullOverlay{}, nil
|
|
}
|
|
|
|
func (NullOverlay) Diagnostics() interface{} {
|
|
return nil
|
|
}
|
|
func (NullOverlay) Confirm() {
|
|
}
|
|
|
|
func (NullOverlay) EstablishedChannel() <-chan struct{} {
|
|
return nil
|
|
}
|
|
|
|
func (NullOverlay) ErrorChannel() <-chan error {
|
|
return nil
|
|
}
|
|
|
|
func (NullOverlay) Stop() {
|
|
}
|
|
|
|
func (NullOverlay) ControlMessage(byte, []byte) {
|
|
}
|
|
|
|
func (NullOverlay) DisplayName() string {
|
|
return "null"
|
|
}
|
|
|
|
type NullNetworkOverlay struct{ NullOverlay }
|
|
|
|
func (NullNetworkOverlay) InvalidateRoutes() {
|
|
}
|
|
|
|
func (NullNetworkOverlay) InvalidateShortIDs() {
|
|
}
|
|
|
|
func (NullNetworkOverlay) StartConsumingPackets(*Peer, *Peers, OverlayConsumer) error {
|
|
return nil
|
|
}
|
|
|
|
func (NullNetworkOverlay) Forward(ForwardPacketKey) FlowOp {
|
|
return DiscardingFlowOp{}
|
|
}
|