From bc77deea1850af5f4aa78604e21e52611d12a981 Mon Sep 17 00:00:00 2001
From: nicolas <48695862+merklefruit@users.noreply.github.com>
Date: Fri, 11 Oct 2024 11:53:53 +0200
Subject: [PATCH] fix(builder): re-add constraint pubkey validation

---
 builder/builder/builder.go | 28 ++++++++--------------------
 1 file changed, 8 insertions(+), 20 deletions(-)

diff --git a/builder/builder/builder.go b/builder/builder/builder.go
index 65c65f144..e765b4bad 100644
--- a/builder/builder/builder.go
+++ b/builder/builder/builder.go
@@ -11,6 +11,7 @@ import (
 	"math/big"
 	"net/http"
 	_ "os"
+	"slices"
 	"strings"
 	"sync"
 	"time"
@@ -362,32 +363,19 @@ func (b *Builder) subscribeToRelayForConstraints(relayBaseEndpoint string) error
 		}
 
 		for _, constraint := range constraintsSigned {
-			// TODO: re-enable this once testing the devnet has ended
-			// oneValidSignature := false
-			// Check if the signature is valid against any of the authorized pubkeys
-			// for _, pubkey := range b.slotConstraintsPubkeys {
-			// 	valid, err := constraint.VerifySignature(pubkey, b.GetConstraintsDomain())
-			// 	if err != nil || !valid {
-			// 		log.Error("Failed to verify constraint signature", "err", err)
-			// 		continue
-			// 	}
-			//
-			// 	oneValidSignature = true
-			// }
-
-			// TODO: remove this once testing the devnet has ended, we should check for authorized keys
+			// Check that the constraints pubkey is authorized to sign constraints
+			if !slices.Contains(b.slotConstraintsPubkeys, constraint.Message.Pubkey) {
+				log.Warn("Received constraint from unauthorized pubkey", "pubkey", constraint.Message.Pubkey)
+				continue
+			}
+
+			// Verify the signature of the constraints message
 			valid, err := constraint.VerifySignature(constraint.Message.Pubkey, b.GetConstraintsDomain())
 			if err != nil || !valid {
 				log.Error("Failed to verify constraint signature", "err", err)
 				continue
 			}
 
-			// TODO: re-enable this once testing the devnet has ended
-			// If there is no valid signature, continue with the next constraint
-			// if !oneValidSignature {
-			// 	continue
-			// }
-
 			decodedConstraints, err := DecodeConstraints(constraint)
 			if err != nil {
 				log.Error("Failed to decode constraint: ", err)