add: add check bad format when opening bad size file

jeromelesaux · May 1, 2024 · e7785ed · e7785ed
1 parent 788d9c7
commit e7785ed
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/convert/screen/screen.go b/convert/screen/screen.go
@@ -1,6 +1,7 @@
 package screen
 
 import (
+	"errors"
 	"image"
 	"image/color"
 
@@ -369,6 +370,9 @@ func ScrToImg(scrPath string, mode uint8, p color.Palette) (*image.NRGBA, error)
 	if err != nil {
 		return nil, err
 	}
+	if len(d) < (m.Width * m.Height) {
+		return nil, errors.New("bad format")
+	}
 	switch mode {
 	case 0:
 		out = setImageMode0(out, p, d, m.Width, m.Height)