Clean up

nginx · May 9, 2024 · 32cd885 · 32cd885
1 parent 902e5f8
commit 32cd885
Show file tree

Hide file tree

Showing 4 changed files with 60 additions and 4 deletions.
diff --git a/sdk/checksum/checksum.go b/sdk/checksum/checksum.go
@@ -10,6 +10,9 @@ package checksum
 import (
 	"crypto/sha256"
 	"fmt"
+	"math"
+
+	log "github.com/sirupsen/logrus"
 )
 
 // Checksum - calculate checksum from []byte
@@ -36,8 +39,19 @@ func Chunk(buf []byte, lim int) [][]byte {
 		return [][]byte{buf}
 	}
 
-	chunks := make([][]byte, 0)
+	if lim <= 0 {
+		log.Error("Unable to chuck payload, chunk size is too small")
+		return [][]byte{}
+	}
+
+	chuckSize := bufSize / lim
+
+	if chuckSize > math.MaxInt64-1 {
+		log.Error("Unable to chuck payload, data too large")
+		return [][]byte{}
+	}
 
+	chunks := make([][]byte, 0, chuckSize+1)
 	for len(buf) >= lim {
 		chunk, buf = buf[:lim], buf[lim:]
 		chunks = append(chunks, chunk)

diff --git a/test/integration/vendor/github.com/nginx/agent/sdk/v2/checksum/checksum.go b/test/integration/vendor/github.com/nginx/agent/sdk/v2/checksum/checksum.go
diff --git a/test/performance/vendor/github.com/nginx/agent/sdk/v2/checksum/checksum.go b/test/performance/vendor/github.com/nginx/agent/sdk/v2/checksum/checksum.go
diff --git a/vendor/github.com/nginx/agent/sdk/v2/checksum/checksum.go b/vendor/github.com/nginx/agent/sdk/v2/checksum/checksum.go