github.com/onflow/flow-go@v0.35.7-crescendo-preview.23-atree-inlining/engine/collection/ingest/engine.go

// Package ingest implements an engine for receiving transactions that need
// to be packaged into a collection.
package ingest

import (
	"context"
	"errors"
	"fmt"

	"github.com/rs/zerolog"

	"github.com/onflow/flow-go/access"
	"github.com/onflow/flow-go/engine"
	"github.com/onflow/flow-go/engine/common/fifoqueue"
	"github.com/onflow/flow-go/model/flow"
	"github.com/onflow/flow-go/module"
	"github.com/onflow/flow-go/module/component"
	"github.com/onflow/flow-go/module/irrecoverable"
	"github.com/onflow/flow-go/module/mempool/epochs"
	"github.com/onflow/flow-go/module/metrics"
	"github.com/onflow/flow-go/network"
	"github.com/onflow/flow-go/network/channels"
	"github.com/onflow/flow-go/state/protocol"
	"github.com/onflow/flow-go/utils/logging"
)

// Engine is the transaction ingestion engine, which ensures that new
// transactions are delegated to the correct collection cluster, and prepared
// to be included in a collection.
type Engine struct {
	*component.ComponentManager
	log                  zerolog.Logger
	engMetrics           module.EngineMetrics
	colMetrics           module.CollectionMetrics
	conduit              network.Conduit
	me                   module.Local
	state                protocol.State
	pendingTransactions  engine.MessageStore
	messageHandler       *engine.MessageHandler
	pools                *epochs.TransactionPools
	transactionValidator *access.TransactionValidator

	config Config
}

// New creates a new collection ingest engine.
func New(
	log zerolog.Logger,
	net network.EngineRegistry,
	state protocol.State,
	engMetrics module.EngineMetrics,
	mempoolMetrics module.MempoolMetrics,
	colMetrics module.CollectionMetrics,
	me module.Local,
	chain flow.Chain,
	pools *epochs.TransactionPools,
	config Config,
	limiter *AddressRateLimiter,
) (*Engine, error) {

	logger := log.With().Str("engine", "ingest").Logger()

	transactionValidator := access.NewTransactionValidatorWithLimiter(
		access.NewProtocolStateBlocks(state),
		chain,
		access.TransactionValidationOptions{
			Expiry:                 flow.DefaultTransactionExpiry,
			ExpiryBuffer:           config.ExpiryBuffer,
			MaxGasLimit:            config.MaxGasLimit,
			CheckScriptsParse:      config.CheckScriptsParse,
			MaxTransactionByteSize: config.MaxTransactionByteSize,
			MaxCollectionByteSize:  config.MaxCollectionByteSize,
		},
		limiter,
	)

	// FIFO queue for transactions
	queue, err := fifoqueue.NewFifoQueue(
		int(config.MaxMessageQueueSize),
		fifoqueue.WithLengthObserver(func(len int) {
			mempoolMetrics.MempoolEntries(metrics.ResourceTransactionIngestQueue, uint(len))
		}),
	)
	if err != nil {
		return nil, fmt.Errorf("could not create transaction message queue: %w", err)
	}
	pendingTransactions := &engine.FifoMessageStore{FifoQueue: queue}

	// define how inbound messages are mapped to message queues
	handler := engine.NewMessageHandler(
		logger,
		engine.NewNotifier(),
		engine.Pattern{
			Match: func(msg *engine.Message) bool {
				_, ok := msg.Payload.(*flow.TransactionBody)
				if ok {
					engMetrics.MessageReceived(metrics.EngineCollectionIngest, metrics.MessageTransaction)
				}
				return ok
			},
			Store: pendingTransactions,
		},
	)

	e := &Engine{
		log:                  logger,
		engMetrics:           engMetrics,
		colMetrics:           colMetrics,
		me:                   me,
		state:                state,
		pendingTransactions:  pendingTransactions,
		messageHandler:       handler,
		pools:                pools,
		config:               config,
		transactionValidator: transactionValidator,
	}

	e.ComponentManager = component.NewComponentManagerBuilder().
		AddWorker(e.processQueuedTransactions).
		Build()

	conduit, err := net.Register(channels.PushTransactions, e)
	if err != nil {
		return nil, fmt.Errorf("could not register engine: %w", err)
	}
	e.conduit = conduit

	return e, nil
}

// Process processes a transaction message from the network and enqueues the
// message. Validation and ingestion is performed in the processQueuedTransactions
// worker.
func (e *Engine) Process(channel channels.Channel, originID flow.Identifier, event interface{}) error {
	select {
	case <-e.ComponentManager.ShutdownSignal():
		e.log.Warn().Msgf("received message from %x after shut down", originID)
		return nil
	default:
	}

	err := e.messageHandler.Process(originID, event)
	if err != nil {
		if engine.IsIncompatibleInputTypeError(err) {
			e.log.Warn().Msgf("%v delivered unsupported message %T through %v", originID, event, channel)
			return nil
		}
		return fmt.Errorf("unexpected error while processing engine message: %w", err)
	}
	return nil
}

// ProcessTransaction processes a transaction message submitted from another
// local component. The transaction is validated and ingested synchronously.
// This is used by the GRPC API, for transactions from Access nodes.
func (e *Engine) ProcessTransaction(tx *flow.TransactionBody) error {
	// do not process transactions after the engine has shut down
	select {
	case <-e.ComponentManager.ShutdownSignal():
		return component.ErrComponentShutdown
	default:
	}

	return e.onTransaction(e.me.NodeID(), tx)
}

// processQueuedTransactions is the main message processing loop for transaction messages.
func (e *Engine) processQueuedTransactions(ctx irrecoverable.SignalerContext, ready component.ReadyFunc) {
	ready()

	for {
		select {
		case <-ctx.Done():
			return
		case <-e.messageHandler.GetNotifier():
			err := e.processAvailableMessages(ctx)
			if err != nil {
				// if an error reaches this point, it is unexpected
				ctx.Throw(err)
				return
			}
		}
	}
}

// processAvailableMessages is called when the message queue is non-empty. It
// will process transactions while the queue is non-empty, then return.
//
// All expected error conditions are handled within this function. Unexpected
// errors which should cause the component to stop are passed up.
func (e *Engine) processAvailableMessages(ctx context.Context) error {
	for {
		select {
		case <-ctx.Done():
			return nil
		default:
		}

		msg, ok := e.pendingTransactions.Get()
		if ok {
			err := e.onTransaction(msg.OriginID, msg.Payload.(*flow.TransactionBody))
			// log warnings for expected error conditions
			if engine.IsUnverifiableInputError(err) {
				e.log.Warn().Err(err).Msg("unable to process unverifiable transaction")
			} else if engine.IsInvalidInputError(err) {
				e.log.Warn().Err(err).Msg("discarding invalid transaction")
			} else if err != nil {
				// bubble up unexpected error
				return fmt.Errorf("unexpected error handling transaction: %w", err)
			}
			continue
		}

		// when there is no more messages in the queue, back to the loop to wait
		// for the next incoming message to arrive.
		return nil
	}
}

// onTransaction handles receipt of a new transaction. This can be submitted
// from outside the system or routed from another collection node.
//
// Returns:
//   - engine.UnverifiableInputError if the reference block is unknown or if the
//     node is not a member of any cluster in the reference epoch.
//   - engine.InvalidInputError if the transaction is invalid.
//   - other error for any other unexpected error condition.
func (e *Engine) onTransaction(originID flow.Identifier, tx *flow.TransactionBody) error {

	defer e.engMetrics.MessageHandled(metrics.EngineCollectionIngest, metrics.MessageTransaction)

	txID := tx.ID()
	log := e.log.With().
		Hex("origin_id", originID[:]).
		Hex("tx_id", txID[:]).
		Hex("ref_block_id", tx.ReferenceBlockID[:]).
		Logger()

	log.Info().Msg("transaction message received")

	// get the state snapshot w.r.t. the reference block
	refSnapshot := e.state.AtBlockID(tx.ReferenceBlockID)
	// fail fast if this is an unknown reference
	_, err := refSnapshot.Head()
	if err != nil {
		return engine.NewUnverifiableInputError("could not get reference block for transaction (%x): %w", txID, err)
	}

	// using the transaction's reference block, determine which cluster we're in.
	// if we don't know the reference block, we will fail when attempting to query the epoch.
	refEpoch := refSnapshot.Epochs().Current()

	localCluster, err := e.getLocalCluster(refEpoch)
	if err != nil {
		return fmt.Errorf("could not get local cluster: %w", err)
	}
	clusters, err := refEpoch.Clustering()
	if err != nil {
		return fmt.Errorf("could not get clusters for reference epoch: %w", err)
	}
	txCluster, ok := clusters.ByTxID(txID)
	if !ok {
		return fmt.Errorf("could not get cluster responsible for tx: %x", txID)
	}

	localClusterFingerPrint := localCluster.ID()
	txClusterFingerPrint := txCluster.ID()
	log = log.With().
		Hex("local_cluster", logging.ID(localClusterFingerPrint)).
		Hex("tx_cluster", logging.ID(txClusterFingerPrint)).
		Logger()

	// validate and ingest the transaction, so it is eligible for inclusion in
	// a future collection proposed by this node
	err = e.ingestTransaction(log, refEpoch, tx, txID, localClusterFingerPrint, txClusterFingerPrint)
	if err != nil {
		return fmt.Errorf("could not ingest transaction: %w", err)
	}

	// if the message was submitted internally (ie. via the Access API)
	// propagate it to members of the responsible cluster (either our cluster
	// or a different cluster)
	if originID == e.me.NodeID() {
		e.propagateTransaction(log, tx, txCluster)
	}

	log.Info().Msg("transaction processed")
	return nil
}

// getLocalCluster returns the cluster this node is a part of for the given reference epoch.
// In cases where the node is not a part of any cluster, this function will differentiate
// between expected and unexpected cases.
//
// Returns:
//   - engine.UnverifiableInputError when this node is not in any cluster because it is not
//     a member of the reference epoch. This is an expected condition and the transaction
//     should be discarded.
//   - other error for any other, unexpected error condition.
func (e *Engine) getLocalCluster(refEpoch protocol.Epoch) (flow.IdentitySkeletonList, error) {
	epochCounter, err := refEpoch.Counter()
	if err != nil {
		return nil, fmt.Errorf("could not get counter for reference epoch: %w", err)
	}
	clusters, err := refEpoch.Clustering()
	if err != nil {
		return nil, fmt.Errorf("could not get clusters for reference epoch: %w", err)
	}

	localCluster, _, ok := clusters.ByNodeID(e.me.NodeID())
	if !ok {
		// if we aren't assigned to a cluster, check that we are a member of
		// the reference epoch
		refIdentities, err := refEpoch.InitialIdentities()
		if err != nil {
			return nil, fmt.Errorf("could not get initial identities for reference epoch: %w", err)
		}

		if _, ok := refIdentities.ByNodeID(e.me.NodeID()); ok {
			// CAUTION: we are a member of the epoch, but have no assigned cluster!
			// This is an unexpected condition and indicates a protocol state invariant has been broken
			return nil, fmt.Errorf("this node should have an assigned cluster in epoch (counter=%d), but has none", epochCounter)
		}
		return nil, engine.NewUnverifiableInputError("this node is not assigned a cluster in epoch (counter=%d)", epochCounter)
	}

	return localCluster, nil
}

// ingestTransaction validates and ingests the transaction, if it is routed to
// our local cluster, is valid, and has not been seen previously.
//
// Returns:
// * engine.InvalidInputError if the transaction is invalid.
// * other error for any other unexpected error condition.
func (e *Engine) ingestTransaction(
	log zerolog.Logger,
	refEpoch protocol.Epoch,
	tx *flow.TransactionBody,
	txID flow.Identifier,
	localClusterFingerprint flow.Identifier,
	txClusterFingerprint flow.Identifier,
) error {
	epochCounter, err := refEpoch.Counter()
	if err != nil {
		return fmt.Errorf("could not get counter for reference epoch: %w", err)
	}

	// use the transaction pool for the epoch the reference block is part of
	pool := e.pools.ForEpoch(epochCounter)

	// short-circuit if we have already stored the transaction
	if pool.Has(txID) {
		log.Debug().Msg("received dupe transaction")
		return nil
	}

	// check if the transaction is valid
	err = e.transactionValidator.Validate(tx)
	if err != nil {
		return engine.NewInvalidInputErrorf("invalid transaction (%x): %w", txID, err)
	}

	// if our cluster is responsible for the transaction, add it to our local mempool
	if localClusterFingerprint == txClusterFingerprint {
		_ = pool.Add(tx)
		e.colMetrics.TransactionIngested(txID)
	}

	return nil
}

// propagateTransaction propagates the transaction to a number of the responsible
// cluster's members. Any unexpected networking errors are logged.
func (e *Engine) propagateTransaction(log zerolog.Logger, tx *flow.TransactionBody, txCluster flow.IdentitySkeletonList) {
	log.Debug().Msg("propagating transaction to cluster")

	err := e.conduit.Multicast(tx, e.config.PropagationRedundancy+1, txCluster.NodeIDs()...)
	if err != nil && !errors.Is(err, network.EmptyTargetList) {
		// if multicast to a target cluster with at least one node failed, log an error and exit
		e.log.Error().Err(err).Msg("could not route transaction to cluster")
	}
	if err == nil {
		e.engMetrics.MessageSent(metrics.EngineCollectionIngest, metrics.MessageTransaction)
	}
}