mirror of
https://github.com/ivuorinen/gibidify.git
synced 2026-01-26 11:34:03 +00:00
e9bd694685
Add overflow checks before converting uint64 memory values to int64 to prevent potential integer overflow issues identified by gosec (G115). - Add math.MaxInt64 checks in fileproc/backpressure.go - Add math.MaxInt64 checks in fileproc/resource_monitor_validation.go - Add math.MaxInt64 checks in fileproc/resource_monitor_metrics.go - Add math.MaxInt64 check in benchmark/benchmark.go with nosec annotation Co-authored-by: ivuorinen <11024+ivuorinen@users.noreply.github.com>
78 lines
2.0 KiB
Go
78 lines
2.0 KiB
Go
package cli
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
|
|
"github.com/ivuorinen/gibidify/config"
|
|
"github.com/ivuorinen/gibidify/fileproc"
|
|
"github.com/ivuorinen/gibidify/utils"
|
|
)
|
|
|
|
// collectFiles collects all files to be processed.
|
|
func (p *Processor) collectFiles() ([]string, error) {
|
|
files, err := fileproc.CollectFiles(p.flags.SourceDir)
|
|
if err != nil {
|
|
return nil, utils.WrapError(err, utils.ErrorTypeProcessing, utils.CodeProcessingCollection, "error collecting files")
|
|
}
|
|
logrus.Infof("Found %d files to process", len(files))
|
|
return files, nil
|
|
}
|
|
|
|
// validateFileCollection validates the collected files against resource limits.
|
|
func (p *Processor) validateFileCollection(files []string) error {
|
|
if !config.GetResourceLimitsEnabled() {
|
|
return nil
|
|
}
|
|
|
|
// Check file count limit
|
|
maxFiles := config.GetMaxFiles()
|
|
if len(files) > maxFiles {
|
|
return utils.NewStructuredError(
|
|
utils.ErrorTypeValidation,
|
|
utils.CodeResourceLimitFiles,
|
|
fmt.Sprintf("file count (%d) exceeds maximum limit (%d)", len(files), maxFiles),
|
|
"",
|
|
map[string]interface{}{
|
|
"file_count": len(files),
|
|
"max_files": maxFiles,
|
|
},
|
|
)
|
|
}
|
|
|
|
// Check total size limit (estimate)
|
|
maxTotalSize := config.GetMaxTotalSize()
|
|
totalSize := int64(0)
|
|
oversizedFiles := 0
|
|
|
|
for _, filePath := range files {
|
|
if fileInfo, err := os.Stat(filePath); err == nil {
|
|
totalSize += fileInfo.Size()
|
|
if totalSize > maxTotalSize {
|
|
return utils.NewStructuredError(
|
|
utils.ErrorTypeValidation,
|
|
utils.CodeResourceLimitTotalSize,
|
|
fmt.Sprintf("total file size (%d bytes) would exceed maximum limit (%d bytes)", totalSize, maxTotalSize),
|
|
"",
|
|
map[string]interface{}{
|
|
"total_size": totalSize,
|
|
"max_total_size": maxTotalSize,
|
|
"files_checked": len(files),
|
|
},
|
|
)
|
|
}
|
|
} else {
|
|
oversizedFiles++
|
|
}
|
|
}
|
|
|
|
if oversizedFiles > 0 {
|
|
logrus.Warnf("Could not stat %d files during pre-validation", oversizedFiles)
|
|
}
|
|
|
|
logrus.Infof("Pre-validation passed: %d files, %d MB total", len(files), totalSize/1024/1024)
|
|
return nil
|
|
}
|