From 26d3119f8fa6fbfa50a47df26fcbf9899eae141d Mon Sep 17 00:00:00 2001
From: Chris Schinnerl <chris@sia.tech>
Date: Wed, 21 Feb 2024 18:02:49 +0100
Subject: [PATCH] worker: encrypt using multiple goroutines

---
 object/slab.go | 15 +++++++++++----
 1 file changed, 11 insertions(+), 4 deletions(-)

diff --git a/object/slab.go b/object/slab.go
index 9c3afa6088..aa8bb7d451 100644
--- a/object/slab.go
+++ b/object/slab.go
@@ -3,6 +3,7 @@ package object
 import (
 	"bytes"
 	"io"
+	"sync"
 
 	"github.com/klauspost/reedsolomon"
 	rhpv2 "go.sia.tech/core/rhp/v2"
@@ -79,11 +80,17 @@ func (s Slab) Length() int {
 // Encrypt xors shards with the keystream derived from s.Key, using a
 // different nonce for each shard.
 func (s Slab) Encrypt(shards [][]byte) {
-	for i, shard := range shards {
-		nonce := [24]byte{1: byte(i)}
-		c, _ := chacha20.NewUnauthenticatedCipher(s.Key.entropy[:], nonce[:])
-		c.XORKeyStream(shard, shard)
+	var wg sync.WaitGroup
+	for i := range shards {
+		wg.Add(1)
+		go func(i int) {
+			nonce := [24]byte{1: byte(i)}
+			c, _ := chacha20.NewUnauthenticatedCipher(s.Key.entropy[:], nonce[:])
+			c.XORKeyStream(shards[i], shards[i])
+			wg.Done()
+		}(i)
 	}
+	wg.Wait()
 }
 
 // Encode encodes slab data into sector-sized shards. The supplied shards should