soft_quay/core/installer/limits.go

package installer

import (
	"errors"
	"fmt"
	"math"
)

var ErrInvalidLimits = errors.New("invalid ZIP extraction limits")

const (
	DefaultMaxEntries               = 10_000
	DefaultMaxArchiveBytes          = int64(4 * 1024 * 1024 * 1024)
	DefaultMaxCentralDirectoryBytes = int64(64 * 1024 * 1024)
	DefaultMaxUncompressedBytes     = int64(4 * 1024 * 1024 * 1024)
	DefaultMaxCompressionRatio      = 200.0
)

// Limits bounds archive metadata and decompressed output.
type Limits struct {
	MaxEntries               int
	MaxArchiveBytes          int64
	MaxCentralDirectoryBytes int64
	MaxUncompressedBytes     int64
	MaxCompressionRatio      float64
}

func DefaultLimits() Limits {
	return Limits{
		MaxEntries:               DefaultMaxEntries,
		MaxArchiveBytes:          DefaultMaxArchiveBytes,
		MaxCentralDirectoryBytes: DefaultMaxCentralDirectoryBytes,
		MaxUncompressedBytes:     DefaultMaxUncompressedBytes,
		MaxCompressionRatio:      DefaultMaxCompressionRatio,
	}
}

func (limits Limits) validate() error {
	if limits.MaxEntries <= 0 {
		return fmt.Errorf("%w: MaxEntries must be positive", ErrInvalidLimits)
	}
	if limits.MaxArchiveBytes <= 0 {
		return fmt.Errorf("%w: MaxArchiveBytes must be positive", ErrInvalidLimits)
	}
	if limits.MaxCentralDirectoryBytes <= 0 {
		return fmt.Errorf("%w: MaxCentralDirectoryBytes must be positive", ErrInvalidLimits)
	}
	if limits.MaxCentralDirectoryBytes > limits.MaxArchiveBytes {
		return fmt.Errorf("%w: MaxCentralDirectoryBytes exceeds MaxArchiveBytes", ErrInvalidLimits)
	}
	if limits.MaxUncompressedBytes <= 0 {
		return fmt.Errorf("%w: MaxUncompressedBytes must be positive", ErrInvalidLimits)
	}
	if limits.MaxCompressionRatio <= 0 ||
		math.IsNaN(limits.MaxCompressionRatio) ||
		math.IsInf(limits.MaxCompressionRatio, 0) {
		return fmt.Errorf("%w: MaxCompressionRatio must be finite and positive", ErrInvalidLimits)
	}
	return nil
}