Repair an integer division by zero vulnerability in ggufPadding

This commit is contained in:
ylwango613 2026-01-03 14:17:17 +08:00
parent 18fdcc94e5
commit 047b86d0f4
1 changed files with 5 additions and 0 deletions

View File

@ -237,6 +237,11 @@ func (llm *gguf) Decode(rs io.ReadSeeker) error {
alignment := llm.kv.Uint("general.alignment", 32)
// Validate alignment
if alignment == 0 {
return fmt.Errorf("invalid general.alignment: cannot be zero")
}
offset, err := rs.Seek(0, io.SeekCurrent)
if err != nil {
return err