move the pool values as well

Signed-off-by: egibs <20933572+egibs@users.noreply.github.com>

Author: egibs

pkg/file/file.go

@@ -2,8 +2,10 @@ package file
 
 // common values used across malcontent for extracting and reading files.
 const (
-	ExtractBuffer       = 64 * 1024 // 64KB
-	MaxBytes            = 1 << 32   // 2048MB
-	ReadBuffer    int64 = 64 * 1024 // 64KB
-	ZipBuffer           = 2 * 1024  // 2KB
+	DefaultPoolBuffer int   = 4 * 1024   // 4KB
+	ExtractBuffer           = 64 * 1024  // 64KB
+	MaxPoolBuffer     int   = 128 * 1024 // 128KB
+	MaxBytes                = 1 << 32    // 2048MB
+	ReadBuffer        int64 = 64 * 1024  // 64KB
+	ZipBuffer               = 2 * 1024   // 2KB
 )

pkg/pool/pool.go

@@ -6,11 +6,7 @@ import (
 	"sync"
 
 	yarax "github.com/VirusTotal/yara-x/go"
-)
-
-const (
-	defaultBuffer int = 4 * 1024   // 4KB
-	maxBuffer     int = 128 * 1024 // 128KB
+	"github.com/chainguard-dev/malcontent/pkg/file"
 )
 
 // BufferPool provides a pool of byte slices for use as buffers.
@@ -24,13 +20,13 @@ func NewBufferPool(count int) *BufferPool {
 
 	bp.pool = sync.Pool{
 		New: func() any {
-			buffer := make([]byte, defaultBuffer)
+			buffer := make([]byte, file.DefaultPoolBuffer)
 			return &buffer
 		},
 	}
 
 	for range count {
-		buffer := make([]byte, defaultBuffer)
+		buffer := make([]byte, file.DefaultPoolBuffer)
 		bp.pool.Put(&buffer)
 	}
 
@@ -66,7 +62,7 @@ func (bp *BufferPool) Put(buf []byte) {
 
 	clear(buf)
 	bufPtr := &buf
-	if cap(*bufPtr) <= maxBuffer {
+	if cap(*bufPtr) <= file.MaxPoolBuffer {
 		bp.pool.Put(bufPtr)
 	}
 }