mirror of
https://git.ffmpeg.org/ffmpeg.git
synced 2024-12-19 05:55:07 +00:00
xsub: Check memory allocation
This commit is contained in:
parent
83797da6e3
commit
522d971c7f
@ -95,7 +95,13 @@ static int decode_frame(AVCodecContext *avctx, void *data, int *data_size,
|
|||||||
|
|
||||||
// allocate sub and set values
|
// allocate sub and set values
|
||||||
sub->rects = av_mallocz(sizeof(*sub->rects));
|
sub->rects = av_mallocz(sizeof(*sub->rects));
|
||||||
|
if (!sub->rects)
|
||||||
|
return AVERROR(ENOMEM);
|
||||||
sub->rects[0] = av_mallocz(sizeof(*sub->rects[0]));
|
sub->rects[0] = av_mallocz(sizeof(*sub->rects[0]));
|
||||||
|
if (!sub->rects[0]) {
|
||||||
|
av_freep(&sub->rects);
|
||||||
|
return AVERROR(ENOMEM);
|
||||||
|
}
|
||||||
sub->num_rects = 1;
|
sub->num_rects = 1;
|
||||||
sub->rects[0]->x = x; sub->rects[0]->y = y;
|
sub->rects[0]->x = x; sub->rects[0]->y = y;
|
||||||
sub->rects[0]->w = w; sub->rects[0]->h = h;
|
sub->rects[0]->w = w; sub->rects[0]->h = h;
|
||||||
@ -104,6 +110,13 @@ static int decode_frame(AVCodecContext *avctx, void *data, int *data_size,
|
|||||||
sub->rects[0]->pict.data[0] = av_malloc(w * h);
|
sub->rects[0]->pict.data[0] = av_malloc(w * h);
|
||||||
sub->rects[0]->nb_colors = 4;
|
sub->rects[0]->nb_colors = 4;
|
||||||
sub->rects[0]->pict.data[1] = av_mallocz(AVPALETTE_SIZE);
|
sub->rects[0]->pict.data[1] = av_mallocz(AVPALETTE_SIZE);
|
||||||
|
if (!sub->rects[0]->pict.data[0] || !sub->rects[0]->pict.data[1]) {
|
||||||
|
av_freep(&sub->rects[0]->pict.data[1]);
|
||||||
|
av_freep(&sub->rects[0]->pict.data[0]);
|
||||||
|
av_freep(&sub->rects[0]);
|
||||||
|
av_freep(&sub->rects);
|
||||||
|
return AVERROR(ENOMEM);
|
||||||
|
}
|
||||||
|
|
||||||
// read palette
|
// read palette
|
||||||
for (i = 0; i < sub->rects[0]->nb_colors; i++)
|
for (i = 0; i < sub->rects[0]->nb_colors; i++)
|
||||||
|
Loading…
Reference in New Issue
Block a user