61 lines
1.8 KiB
Go
61 lines
1.8 KiB
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
|
|
}
|