Commit 8e7808b5 authored by Clément Bœsch's avatar Clément Bœsch

avcodec/realtextdec: add some memory checks

parent 27a9bee2
...@@ -59,6 +59,7 @@ static int rt_event_to_ass(AVBPrint *buf, const char *p) ...@@ -59,6 +59,7 @@ static int rt_event_to_ass(AVBPrint *buf, const char *p)
static int realtext_decode_frame(AVCodecContext *avctx, static int realtext_decode_frame(AVCodecContext *avctx,
void *data, int *got_sub_ptr, AVPacket *avpkt) void *data, int *got_sub_ptr, AVPacket *avpkt)
{ {
int ret = 0;
AVSubtitle *sub = data; AVSubtitle *sub = data;
const char *ptr = avpkt->data; const char *ptr = avpkt->data;
AVBPrint buf; AVBPrint buf;
...@@ -67,9 +68,11 @@ static int realtext_decode_frame(AVCodecContext *avctx, ...@@ -67,9 +68,11 @@ static int realtext_decode_frame(AVCodecContext *avctx,
// note: no need to rescale pts & duration since they are in the same // note: no need to rescale pts & duration since they are in the same
// timebase as ASS (1/100) // timebase as ASS (1/100)
if (ptr && avpkt->size > 0 && !rt_event_to_ass(&buf, ptr)) if (ptr && avpkt->size > 0 && !rt_event_to_ass(&buf, ptr))
ff_ass_add_rect(sub, buf.str, avpkt->pts, avpkt->duration, 0); ret = ff_ass_add_rect_bprint(sub, &buf, avpkt->pts, avpkt->duration, 0);
*got_sub_ptr = sub->num_rects > 0;
av_bprint_finalize(&buf, NULL); av_bprint_finalize(&buf, NULL);
if (ret < 0)
return ret;
*got_sub_ptr = sub->num_rects > 0;
return avpkt->size; return avpkt->size;
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment