initial commit

Author: shrimalmadhur

.github/workflows/tests.yml

@@ -0,0 +1,19 @@
+name: tests
+
+on:
+  push:
+    branches:
+      - master
+  pull_request:
+
+jobs:
+  Test:
+    name: Unit Test
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/setup-go@v4
+        with:
+          go-version: '1.22'
+      - name: Unit Test
+        run: make tests

.gitignore

@@ -0,0 +1,5 @@
+bin/
+
+.env
+
+**/operators/

.gitmodules

@@ -0,0 +1,12 @@
+[submodule "example/contracts/lib/forge-std"]
+	path = example/contracts/lib/forge-std
+	url = https://github.com/foundry-rs/forge-std
+[submodule "example/contracts/lib/openzeppelin-contracts"]
+	path = example/contracts/lib/openzeppelin-contracts
+	url = https://github.com/OpenZeppelin/openzeppelin-contracts
+[submodule "example/contracts/lib/openzeppelin-contracts-upgradeable"]
+	path = example/contracts/lib/openzeppelin-contracts-upgradeable
+	url = https://github.com/OpenZeppelin/openzeppelin-contracts-upgradeable
+[submodule "example/contracts/lib/eigenlayer-middleware"]
+	path = example/contracts/lib/eigenlayer-middleware
+	url = https://github.com/Layr-Labs/eigenlayer-middleware

LICENSE

@@ -0,0 +1,120 @@
+Business Source License 1.1
+
+License text copyright (c) 2017 MariaDB Corporation Ab, All Rights Reserved.
+"Business Source License" is a trademark of MariaDB Corporation Ab.
+
+-----------------------------------------------------------------------------
+
+Parameters
+
+Licensor:             Layr Labs, Inc.
+
+Licensed Work:        Teal
+                      The Licensed Work is (c) 2023 Layr Labs, Inc.
+
+Additional Use Grant: 
+
+You may additionally use any of the software included in the following repositories
+[here](https://docs.google.com/spreadsheets/d/1PlJRow5C0GMqXZlIxRm5CEnkhH-gMV1wIdq1pCfbZco/edit?usp=sharing)
+(“Additional Use Grant Software”) for production commercial uses, but only if such
+uses are (i) built on or using the EigenLayer Protocol or EigenDA, and (ii) not
+Competing Uses. 
+
+“Competing Use” means any use of the Additional Use Grant Software in any product,
+protocol, application or service that is made available to third parties and that
+(i) substitutes for use of EigenLayer Protocol or EigenDA, (ii) offers the same or
+substantially similar functionality as the EigenLayer Protocol or EigenDA or
+(iii) is built on or using a protocol with substantially similar functionality as
+the EigenLayer Protocol.
+
+EigenLayer Protocol means the restaking protocol as further described in the
+documentation [here](https://docs.eigenlayer.xyz/), as updated from time to time. 
+
+EigenDA means the data availability protocol built on top of the EigenLayer
+Protocol as further described in the documentation
+[here](https://docs.eigenlayer.xyz/eigenda/overview), as updated from time to time. 
+
+Change Dates:
+
+- All commits before have a change date of 2029-01-20 (January 20th, 2029)
+
+Change License:       MIT
+
+-----------------------------------------------------------------------------
+
+Terms
+
+The Licensor hereby grants you the right to copy, modify, create derivative
+works, redistribute, and make non-production use of the Licensed Work. The
+Licensor may make an Additional Use Grant, above, permitting limited
+production use.
+
+Effective on the Change Date, or the fourth anniversary of the first publicly
+available distribution of a specific version of the Licensed Work under this
+License, whichever comes first, the Licensor hereby grants you rights under
+the terms of the Change License, and the rights granted in the paragraph
+above terminate.
+
+If your use of the Licensed Work does not comply with the requirements
+currently in effect as described in this License, you must purchase a
+commercial license from the Licensor, its affiliated entities, or authorized
+resellers, or you must refrain from using the Licensed Work.
+
+All copies of the original and modified Licensed Work, and derivative works
+of the Licensed Work, are subject to this License. This License applies
+separately for each version of the Licensed Work and the Change Date may vary
+for each version of the Licensed Work released by Licensor.
+
+You must conspicuously display this License on each original or modified copy
+of the Licensed Work. If you receive the Licensed Work in original or
+modified form from a third party, the terms and conditions set forth in this
+License apply to your use of that work.
+
+Any use of the Licensed Work in violation of this License will automatically
+terminate your rights under this License for the current and all other
+versions of the Licensed Work.
+
+This License does not grant you any right in any trademark or logo of
+Licensor or its affiliates (provided that you may use a trademark or logo of
+Licensor as expressly required by this License).
+
+TO THE EXTENT PERMITTED BY APPLICABLE LAW, THE LICENSED WORK IS PROVIDED ON
+AN "AS IS" BASIS. LICENSOR HEREBY DISCLAIMS ALL WARRANTIES AND CONDITIONS,
+EXPRESS OR IMPLIED, INCLUDING (WITHOUT LIMITATION) WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, NON-INFRINGEMENT, AND
+TITLE.
+
+MariaDB hereby grants you permission to use this License’s text to license
+your works, and to refer to it using the trademark "Business Source License",
+as long as you comply with the Covenants of Licensor below.
+
+-----------------------------------------------------------------------------
+
+Covenants of Licensor
+
+In consideration of the right to use this License’s text and the "Business
+Source License" name and trademark, Licensor covenants to MariaDB, and to all
+other recipients of the licensed work to be provided by Licensor:
+
+1. To specify as the Change License the GPL Version 2.0 or any later version,
+   or a license that is compatible with GPL Version 2.0 or a later version,
+   where "compatible" means that software provided under the Change License can
+   be included in a program with software provided under GPL Version 2.0 or a
+   later version. Licensor may specify additional Change Licenses without
+   limitation.
+
+2. To either: (a) specify an additional grant of rights to use that does not
+   impose any additional restriction on the right granted in this License, as
+   the Additional Use Grant; or (b) insert the text "None".
+
+3. To specify a Change Date.
+
+4. Not to modify this License in any other way.
+
+-----------------------------------------------------------------------------
+
+Notice
+
+The Business Source License (this document, or the "License") is not an Open
+Source license. However, the Licensed Work will eventually be made available
+under an Open Source License, as stated in this License.

Makefile

@@ -0,0 +1,29 @@
+.PHONY: install
+install-macos-linux:
+	@echo "Installing dependencies"
+	brew install bufbuild/buf/buf
+	go install github.com/fullstorydev/grpcurl/cmd/grpcurl@latest
+	go install github.com/golang/protobuf/protoc-gen-go@latest
+	go install github.com/grpc-ecosystem/grpc-gateway/v2/protoc-gen-grpc-gateway@latest
+	go install github.com/grpc-ecosystem/grpc-gateway/v2/protoc-gen-openapiv2@latest
+	@echo "Done"
+
+.PHONY: generate
+generate:
+	@echo "Generating go bindings"
+	cd api/proto && buf generate
+	@echo "Done"
+
+.PHONY: build-node
+build-node:
+	@echo "Building node"
+	go build -o bin/node node/cmd/main.go
+	@echo "Done"
+
+.PHONY: fmt
+fmt: ## formats all go files
+	go fmt ./...
+
+.PHONY: test
+tests:
+	go test ./...

README.md

@@ -0,0 +1,7 @@
+# teal
+
+## Install Dependencies
+
+```
+make install-macos-linux
+```

aggregator/aggregator.go

@@ -0,0 +1,127 @@
+package aggregator
+
+import (
+	"context"
+	"fmt"
+	"math/big"
+	"sync"
+	"time"
+
+	"github.com/Layr-Labs/eigensdk-go/crypto/bls"
+	"github.com/Layr-Labs/eigensdk-go/logging"
+	"github.com/Layr-Labs/eigensdk-go/services/avsregistry"
+	blsagg "github.com/Layr-Labs/eigensdk-go/services/bls_aggregation"
+	"github.com/Layr-Labs/eigensdk-go/types"
+	operatorrequester "github.com/Layr-Labs/teal/aggregator/operator_requester"
+)
+
+type AggregatorService struct {
+	logger            logging.Logger
+	avsRegistryReader avsregistry.AvsRegistryService
+	blsAggService     blsagg.BlsAggregationService
+	operatorRequester operatorrequester.OperatorRequester
+
+	mu sync.Mutex
+}
+
+func NewAggregatorService(
+	logger logging.Logger,
+	avsRegistryReader avsregistry.AvsRegistryService,
+	blsAggService blsagg.BlsAggregationService,
+	operatorRequester operatorrequester.OperatorRequester,
+) *AggregatorService {
+	return &AggregatorService{
+		logger:            logger,
+		avsRegistryReader: avsRegistryReader,
+		blsAggService:     blsAggService,
+		operatorRequester: operatorRequester,
+	}
+}
+
+// GetCertificate sends a task to all registered nodes and aggregates their responses
+// Only works for single quorum for simplicity
+func (s *AggregatorService) GetCertificate(
+	ctx context.Context,
+	taskIndex types.TaskIndex,
+	taskCreatedBlock uint32,
+	quorumNumber types.QuorumNum,
+	quorumThresholdPercentage types.QuorumThresholdPercentage,
+	data []byte,
+	timeToExpiry time.Duration,
+) (*blsagg.BlsAggregationServiceResponse, error) {
+	// Only allow one task at a time
+	s.mu.Lock()
+	defer s.mu.Unlock()
+
+	quorumNumbers := types.QuorumNums{quorumNumber}
+	quorumThresholdPercentages := types.QuorumThresholdPercentages{quorumThresholdPercentage}
+
+	// Initialize task in BLS aggregation service
+	err := s.blsAggService.InitializeNewTaskWithWindow(
+		taskIndex,
+		taskCreatedBlock,
+		quorumNumbers,
+		quorumThresholdPercentages,
+		timeToExpiry,
+		1*time.Second,
+	)
+	if err != nil {
+		return nil, fmt.Errorf("failed to initialize task: %w", err)
+	}
+
+	// Get operators from registry
+	operators, err := s.avsRegistryReader.GetOperatorsAvsStateAtBlock(ctx, quorumNumbers, taskCreatedBlock)
+	if err != nil {
+		return nil, fmt.Errorf("failed to get operators: %w", err)
+	}
+
+	// Send task to all operators in parallel
+	for operatorId, operator := range operators {
+		go func(operatorId types.OperatorId, operator types.OperatorAvsState) {
+			s.logger.Info("Requesting certification from operator", "operatorId", operatorId, "socket", operator.OperatorInfo.Socket)
+			// Create connection for this operator
+			resp, err := s.operatorRequester.RequestCertification(ctx, operator, taskIndex, data)
+			if err != nil {
+				return
+			}
+
+			signature := &bls.Signature{G1Point: bls.NewG1Point(big.NewInt(0), big.NewInt(0))}
+			_, err = signature.SetBytes(resp.Signature)
+			if err != nil {
+				s.logger.Error("Failed to unmarshal signature",
+					"operatorId", operatorId,
+					"error", err)
+				return
+			}
+
+			s.logger.Info("Received signature from operator", "operatorId", operatorId)
+
+			// Process signature from node
+			err = s.blsAggService.ProcessNewSignature(
+				ctx,
+				taskIndex,
+				types.TaskResponse(resp.Data),
+				signature,
+				operatorId,
+			)
+			if err != nil {
+				s.logger.Error("Failed to process signature",
+					"operatorId", operatorId,
+					"error", err)
+				return
+			}
+			s.logger.Info("Processed signature from operator", "operatorId", operatorId)
+		}(operatorId, operator)
+	}
+
+	// Wait for aggregated response
+	select {
+	case resp := <-s.blsAggService.GetResponseChannel():
+		if resp.Err != nil {
+			return nil, fmt.Errorf("aggregation failed: %w", resp.Err)
+		}
+		return &resp, nil
+	case <-ctx.Done():
+		return nil, ctx.Err()
+	}
+}