POC Signature aggregation API

config/config.go

@@ -246,3 +246,13 @@ func (c *Config) GetWarpQuorum(blockchainID ids.ID) (WarpQuorum, error) {
 	}
 	return WarpQuorum{}, errFailedToGetWarpQuorum
 }
+
+// Config implements the peers.Config interface
+func (c *Config) GetPChainAPI() *APIConfig {
+	return c.PChainAPI
+}
+
+// Config implements the peers.Config interface
+func (c *Config) GetInfoAPI() *APIConfig {
+	return c.InfoAPI
+}

main/main.go

@@ -17,6 +17,7 @@ import (
 	"github.com/ava-labs/avalanchego/message"
 	"github.com/ava-labs/avalanchego/utils/constants"
 	"github.com/ava-labs/avalanchego/utils/logging"
+	"github.com/ava-labs/avalanchego/utils/set"
 	"github.com/ava-labs/awm-relayer/api"
 	"github.com/ava-labs/awm-relayer/config"
 	"github.com/ava-labs/awm-relayer/database"
@@ -26,6 +27,7 @@ import (
 	"github.com/ava-labs/awm-relayer/peers"
 	"github.com/ava-labs/awm-relayer/relayer"
 	"github.com/ava-labs/awm-relayer/relayer/checkpoint"
+	"github.com/ava-labs/awm-relayer/signature-aggregator/aggregator"
 	"github.com/ava-labs/awm-relayer/utils"
 	"github.com/ava-labs/awm-relayer/vms"
 	"github.com/ava-labs/subnet-evm/ethclient"
@@ -134,11 +136,23 @@ func main() {
 	if logLevel <= logging.Debug {
 		networkLogLevel = logLevel
 	}
+	var trackedSubnets set.Set[ids.ID]
+	// trackedSubnets is no longer strictly required but keeping it here for now
+	// to keep full parity with existing AWM relayer for now
+	// TODO: remove this from here once trackedSubnets are no longer referenced
+	// by ping messages in avalanchego
+	for _, sourceBlockchain := range cfg.SourceBlockchains {
+		trackedSubnets.Add(sourceBlockchain.GetSubnetID())
+	}
+
 	network, err := peers.NewNetwork(
 		networkLogLevel,
 		prometheus.DefaultRegisterer,
+		trackedSubnets,
 		&cfg,
 	)
+	network.InitializeConnectionsAndCheckStake(&cfg)
+
 	if err != nil {
 		logger.Fatal("Failed to create app request network", zap.Error(err))
 		panic(err)
@@ -197,17 +211,19 @@ func main() {
 		panic(err)
 	}
 
+	signatureAggregator := aggregator.NewSignatureAggregator(network, logger, messageCreator)
+
 	applicationRelayers, minHeights, err := createApplicationRelayers(
 		context.Background(),
 		logger,
 		relayerMetrics,
 		db,
 		ticker,
 		network,
-		messageCreator,
 		&cfg,
 		sourceClients,
 		destinationClients,
+		signatureAggregator,
 	)
 	if err != nil {
 		logger.Fatal("Failed to create application relayers", zap.Error(err))
@@ -332,10 +348,10 @@ func createApplicationRelayers(
 	db database.RelayerDatabase,
 	ticker *utils.Ticker,
 	network *peers.AppRequestNetwork,
-	messageCreator message.Creator,
 	cfg *config.Config,
 	sourceClients map[ids.ID]ethclient.Client,
 	destinationClients map[ids.ID]vms.DestinationClient,
+	signatureAggregator *aggregator.SignatureAggregator,
 ) (map[common.Hash]*relayer.ApplicationRelayer, map[ids.ID]uint64, error) {
 	applicationRelayers := make(map[common.Hash]*relayer.ApplicationRelayer)
 	minHeights := make(map[ids.ID]uint64)
@@ -355,10 +371,10 @@ func createApplicationRelayers(
 			ticker,
 			*sourceBlockchain,
 			network,
-			messageCreator,
 			cfg,
 			currentHeight,
 			destinationClients,
+			signatureAggregator,
 		)
 		if err != nil {
 			logger.Error(
@@ -391,10 +407,10 @@ func createApplicationRelayersForSourceChain(
 	ticker *utils.Ticker,
 	sourceBlockchain config.SourceBlockchain,
 	network *peers.AppRequestNetwork,
-	messageCreator message.Creator,
 	cfg *config.Config,
 	currentHeight uint64,
 	destinationClients map[ids.ID]vms.DestinationClient,
+	signatureAggregator *aggregator.SignatureAggregator,
 ) (map[common.Hash]*relayer.ApplicationRelayer, uint64, error) {
 	// Create the ApplicationRelayers
 	logger.Info(
@@ -438,12 +454,12 @@ func createApplicationRelayersForSourceChain(
 			logger,
 			metrics,
 			network,
-			messageCreator,
 			relayerID,
 			destinationClients[relayerID.DestinationBlockchainID],
 			sourceBlockchain,
 			checkpointManager,
 			cfg,
+			signatureAggregator,
 		)
 		if err != nil {
 			logger.Error(

peers/app_request_network.go

@@ -5,14 +5,18 @@ package peers
 
 import (
 	"context"
+	"fmt"
 	"math/big"
 	"os"
 	"sync"
 	"time"
 
 	"github.com/ava-labs/avalanchego/ids"
+	"github.com/ava-labs/avalanchego/message"
 	"github.com/ava-labs/avalanchego/network"
+	avagoCommon "github.com/ava-labs/avalanchego/snow/engine/common"
 	snowVdrs "github.com/ava-labs/avalanchego/snow/validators"
+	"github.com/ava-labs/avalanchego/subnets"
 	"github.com/ava-labs/avalanchego/utils/constants"
 	"github.com/ava-labs/avalanchego/utils/logging"
 	"github.com/ava-labs/avalanchego/utils/set"
@@ -30,8 +34,8 @@ const (
 )
 
 type AppRequestNetwork struct {
-	Network         network.Network
-	Handler         *RelayerExternalHandler
+	network         network.Network
+	handler         *RelayerExternalHandler
 	infoAPI         *InfoAPI
 	logger          logging.Logger
 	lock            *sync.Mutex
@@ -42,7 +46,8 @@ type AppRequestNetwork struct {
 func NewNetwork(
 	logLevel logging.Level,
 	registerer prometheus.Registerer,
-	cfg *config.Config,
+	trackedSubnets set.Set[ids.ID],
+	cfg Config,
 ) (*AppRequestNetwork, error) {
 	logger := logging.NewLogger(
 		"awm-relayer-p2p",
@@ -63,7 +68,7 @@ func NewNetwork(
 		return nil, err
 	}
 
-	infoAPI, err := NewInfoAPI(cfg.InfoAPI)
+	infoAPI, err := NewInfoAPI(cfg.GetInfoAPI())
 	if err != nil {
 		logger.Error(
 			"Failed to create info API",
@@ -80,11 +85,6 @@ func NewNetwork(
 		return nil, err
 	}
 
-	var trackedSubnets set.Set[ids.ID]
-	for _, sourceBlockchain := range cfg.SourceBlockchains {
-		trackedSubnets.Add(sourceBlockchain.GetSubnetID())
-	}
-
 	testNetwork, err := network.NewTestNetwork(logger, networkID, snowVdrs.NewManager(), trackedSubnets, handler)
 	if err != nil {
 		logger.Error(
@@ -94,38 +94,47 @@ func NewNetwork(
 		return nil, err
 	}
 
-	validatorClient := validators.NewCanonicalValidatorClient(logger, cfg.PChainAPI)
+	validatorClient := validators.NewCanonicalValidatorClient(logger, cfg.GetPChainAPI())
 
 	arNetwork := &AppRequestNetwork{
-		Network:         testNetwork,
-		Handler:         handler,
+		network:         testNetwork,
+		handler:         handler,
 		infoAPI:         infoAPI,
 		logger:          logger,
 		lock:            new(sync.Mutex),
 		validatorClient: validatorClient,
 	}
+	go logger.RecoverAndPanic(func() {
+		testNetwork.Dispatch()
+	})
+
+	return arNetwork, nil
+}
 
+// TODO: remove dependence on Relayer specific config since this is meant to be a generic AppRequestNetwork file
+func (n *AppRequestNetwork) InitializeConnectionsAndCheckStake(cfg *config.Config) error {
 	// Manually connect to the validators of each of the source subnets.
 	// We return an error if we are unable to connect to sufficient stake on any of the subnets.
 	// Sufficient stake is determined by the Warp quora of the configured supported destinations,
 	// or if the subnet supports all destinations, by the quora of all configured destinations.
 	for _, sourceBlockchain := range cfg.SourceBlockchains {
 		if sourceBlockchain.GetSubnetID() == constants.PrimaryNetworkID {
-			if err := arNetwork.connectToPrimaryNetworkPeers(cfg, sourceBlockchain); err != nil {
-				return nil, err
+			if err := n.connectToPrimaryNetworkPeers(cfg, sourceBlockchain); err != nil {
+				return fmt.Errorf(
+					"failed to connect to primary network peers: %w",
+					err,
+				)
 			}
 		} else {
-			if err := arNetwork.connectToNonPrimaryNetworkPeers(cfg, sourceBlockchain); err != nil {
-				return nil, err
+			if err := n.connectToNonPrimaryNetworkPeers(cfg, sourceBlockchain); err != nil {
+				return fmt.Errorf(
+					"failed to connect to non-primary network peers: %w",
+					err,
+				)
 			}
 		}
 	}
-
-	go logger.RecoverAndPanic(func() {
-		testNetwork.Dispatch()
-	})
-
-	return arNetwork, nil
+	return nil
 }
 
 // ConnectPeers connects the network to peers with the given nodeIDs.
@@ -135,7 +144,7 @@ func (n *AppRequestNetwork) ConnectPeers(nodeIDs set.Set[ids.NodeID]) set.Set[id
 	defer n.lock.Unlock()
 
 	// First, check if we are already connected to all the peers
-	connectedPeers := n.Network.PeerInfo(nodeIDs.List())
+	connectedPeers := n.network.PeerInfo(nodeIDs.List())
 	if len(connectedPeers) == nodeIDs.Len() {
 		return nodeIDs
 	}
@@ -160,7 +169,7 @@ func (n *AppRequestNetwork) ConnectPeers(nodeIDs set.Set[ids.NodeID]) set.Set[id
 	for _, peer := range peers {
 		if nodeIDs.Contains(peer.ID) {
 			trackedNodes.Add(peer.ID)
-			n.Network.ManuallyTrack(peer.ID, peer.PublicIP)
+			n.network.ManuallyTrack(peer.ID, peer.PublicIP)
 			if len(trackedNodes) == nodeIDs.Len() {
 				return trackedNodes
 			}
@@ -182,7 +191,7 @@ func (n *AppRequestNetwork) ConnectPeers(nodeIDs set.Set[ids.NodeID]) set.Set[id
 			)
 		} else {
 			trackedNodes.Add(apiNodeID)
-			n.Network.ManuallyTrack(apiNodeID, apiNodeIPPort)
+			n.network.ManuallyTrack(apiNodeID, apiNodeIPPort)
 		}
 	}
 
@@ -245,6 +254,25 @@ func (n *AppRequestNetwork) ConnectToCanonicalValidators(subnetID ids.ID) (*Conn
 	}, nil
 }
 
+func (n *AppRequestNetwork) Send(
+	msg message.OutboundMessage,
+	nodeIDs set.Set[ids.NodeID],
+	subnetID ids.ID,
+	allower subnets.Allower,
+) set.Set[ids.NodeID] {
+	return n.network.Send(msg, avagoCommon.SendConfig{NodeIDs: nodeIDs}, subnetID, allower)
+}
+
+func (n *AppRequestNetwork) RegisterAppRequest(requestID ids.RequestID) {
+	n.handler.RegisterAppRequest(requestID)
+}
+func (n *AppRequestNetwork) RegisterRequestID(requestID uint32, numExpectedResponse int) chan message.InboundMessage {
+	return n.handler.RegisterRequestID(requestID, numExpectedResponse)
+}
+func (n *AppRequestNetwork) GetSubnetID(blockchainID ids.ID) (ids.ID, error) {
+	return n.validatorClient.GetSubnetID(context.Background(), blockchainID)
+}
+
 // Private helpers
 
 // Connect to the validators of the source blockchain. For each destination blockchain,

peers/config.go

@@ -0,0 +1,12 @@
+// Copyright (C) 2024, Ava Labs, Inc. All rights reserved.
+// See the file LICENSE for licensing terms.
+
+package peers
+
+import "github.com/ava-labs/awm-relayer/config"
+
+// Config defines a common interface necessary for standing up an AppRequestNetwork.
+type Config interface {
+	GetInfoAPI() *config.APIConfig
+	GetPChainAPI() *config.APIConfig
+}