You've already forked FFmpeg
mirror of
https://github.com/FFmpeg/FFmpeg.git
synced 2025-08-04 22:03:09 +02:00
avcodec/dvbsubenc: Check nb_colors before using it
Avoids a potential overflow when multiplying nb_colors by 6. Also make the nb_colors check a bit more strict. Signed-off-by: Andreas Rheinhardt <andreas.rheinhardt@outlook.com>
This commit is contained in:
@ -329,24 +329,23 @@ static int dvbsub_encode(AVCodecContext *avctx, uint8_t *outbuf, int buf_size,
|
||||
|
||||
if (h->num_rects) {
|
||||
for (clut_id = 0; clut_id < h->num_rects; clut_id++) {
|
||||
if (buf_size < 6 + h->rects[clut_id]->nb_colors * 6)
|
||||
return AVERROR_BUFFER_TOO_SMALL;
|
||||
|
||||
/* CLUT segment */
|
||||
|
||||
if (h->rects[clut_id]->nb_colors <= 4) {
|
||||
if (h->rects[clut_id]->nb_colors <= 4U) {
|
||||
/* 2 bpp, some decoders do not support it correctly */
|
||||
bpp_index = 0;
|
||||
} else if (h->rects[clut_id]->nb_colors <= 16) {
|
||||
} else if (h->rects[clut_id]->nb_colors <= 16U) {
|
||||
/* 4 bpp, standard encoding */
|
||||
bpp_index = 1;
|
||||
} else if (h->rects[clut_id]->nb_colors <= 256) {
|
||||
} else if (h->rects[clut_id]->nb_colors <= 256U) {
|
||||
/* 8 bpp, standard encoding */
|
||||
bpp_index = 2;
|
||||
} else {
|
||||
return AVERROR(EINVAL);
|
||||
}
|
||||
|
||||
if (buf_size < 6 + h->rects[clut_id]->nb_colors * 6)
|
||||
return AVERROR_BUFFER_TOO_SMALL;
|
||||
|
||||
/* CLUT segment */
|
||||
*q++ = 0x0f; /* sync byte */
|
||||
|
Reference in New Issue
Block a user