feat(lib): limit concurrency of wasm-based verifierswasm

Signed-off-by: Xe Iaso <me@xeiaso.net>

2 files changed, 27 insertions, 1 deletions

diff --git a/lib/anubis.go b/lib/anubis.go
index 762c6aa..269df5d 100644
--- a/lib/anubis.go
+++ b/lib/anubis.go
@@ -152,7 +152,11 @@ func New(opts Options) (*Server, error) {
 			return nil, fmt.Errorf("can't load static/wasm/%s: %w", finfo.Name(), err)
 		}
 
-		result.validators[name] = runner
+		var concurrentLimit int64 = 4
+
+		cv := NewConcurrentVerifier(runner, concurrentLimit)
+
+		result.validators[name] = cv
 	}
 
 	mux := http.NewServeMux()
diff --git a/lib/verifier.go b/lib/verifier.go
index 3c28c14..c41cb93 100644
--- a/lib/verifier.go
+++ b/lib/verifier.go
@@ -6,6 +6,8 @@ import (
 	"crypto/subtle"
 	"errors"
 	"fmt"
+
+	"golang.org/x/sync/semaphore"
 )
 
 var (
@@ -68,3 +70,23 @@ func hasLeadingZeroNibbles(data []byte, n uint32) bool {
 	}
 	return count >= n
 }
+
+type ConcurrentVerifier struct {
+	Verifier
+	sem *semaphore.Weighted
+}
+
+func NewConcurrentVerifier(v Verifier, maxConcurrent int64) *ConcurrentVerifier {
+	return &ConcurrentVerifier{
+		Verifier: v,
+		sem:      semaphore.NewWeighted(maxConcurrent),
+	}
+}
+
+func (cv *ConcurrentVerifier) Verify(ctx context.Context, challenge, verify []byte, nonce, difficulty uint32) (bool, error) {
+	if err := cv.sem.Acquire(ctx, 1); err != nil {
+		return false, fmt.Errorf("can't verify solution: %w", err)
+	}
+
+	return cv.Verifier.Verify(ctx, challenge, verify, nonce, difficulty)
+}