avformat/matroskaenc: Check ff_vorbiscomment_length in put_flac_codecpriv()

Its currently guaranteed to be smaller but its safer to check anyway

Signed-off-by: Michael Niedermayer <michaelni@gmx.at>
This commit is contained in:
Michael Niedermayer 2015-05-11 15:37:38 +02:00
parent 0db5b2b9f8
commit 66f26b3e8e

View File

@ -546,12 +546,15 @@ static int put_flac_codecpriv(AVFormatContext *s,
"Lavf" : LIBAVFORMAT_IDENT; "Lavf" : LIBAVFORMAT_IDENT;
AVDictionary *dict = NULL; AVDictionary *dict = NULL;
uint8_t buf[32], *data, *p; uint8_t buf[32], *data, *p;
int len; int64_t len;
snprintf(buf, sizeof(buf), "0x%"PRIx64, codec->channel_layout); snprintf(buf, sizeof(buf), "0x%"PRIx64, codec->channel_layout);
av_dict_set(&dict, "WAVEFORMATEXTENSIBLE_CHANNEL_MASK", buf, 0); av_dict_set(&dict, "WAVEFORMATEXTENSIBLE_CHANNEL_MASK", buf, 0);
len = ff_vorbiscomment_length(dict, vendor); len = ff_vorbiscomment_length(dict, vendor);
if (len >= ((1<<24) - 4))
return AVERROR(EINVAL);
data = av_malloc(len + 4); data = av_malloc(len + 4);
if (!data) { if (!data) {
av_dict_free(&dict); av_dict_free(&dict);