diff options
author | Michael Niedermayer <michaelni@gmx.at> | 2014-03-17 20:06:30 +0400 |
---|---|---|
committer | Michael Niedermayer <michaelni@gmx.at> | 2014-03-17 20:06:30 +0400 |
commit | f6f3c85c0eff54cebc3a45caa7a1e0454b5c9ff8 (patch) | |
tree | 822b6a5afe08d0446ad57a62181348518fd0ab92 /libavformat | |
parent | b1eb92a6bcdaa57a150a7493bc29f82cbc36ef99 (diff) |
avformat/isom: Simplify mov_rewrite_dvd_sub_extradata()
This avoids the 2nd buffer and fixes the palette truncated check
Signed-off-by: Michael Niedermayer <michaelni@gmx.at>
Diffstat (limited to 'libavformat')
-rw-r--r-- | libavformat/isom.c | 18 |
1 files changed, 8 insertions, 10 deletions
diff --git a/libavformat/isom.c b/libavformat/isom.c index ffe369e19d..d897808391 100644 --- a/libavformat/isom.c +++ b/libavformat/isom.c @@ -26,6 +26,7 @@ #include "isom.h" #include "libavcodec/mpeg4audio.h" #include "libavcodec/mpegaudiodata.h" +#include "libavutil/avstring.h" #include "libavutil/intreadwrite.h" /* http://www.mp4ra.org */ @@ -456,28 +457,25 @@ static uint32_t yuv_to_rgba(uint32_t ycbcr) static int mov_rewrite_dvd_sub_extradata(AVStream *st) { - char pal_s[256]; - char buf[256]; - int pal_s_pos = 0; + char buf[256] = {0}; uint8_t *src = st->codec->extradata; int i; if (st->codec->extradata_size != 64) return 0; + snprintf(buf, sizeof(buf), "size: %dx%d\npalette: ", + st->codec->width, st->codec->height); + for (i = 0; i < 16; i++) { uint32_t yuv = AV_RB32(src + i * 4); uint32_t rgba = yuv_to_rgba(yuv); - snprintf(pal_s + pal_s_pos, sizeof(pal_s) - pal_s_pos, "%06x%s", rgba, - i != 15 ? ", " : ""); - pal_s_pos = strlen(pal_s); - if (pal_s_pos >= sizeof(pal_s)) - return 0; + av_strlcatf(buf, sizeof(buf), "%06x%s", rgba, i != 15 ? ", " : ""); } - snprintf(buf, sizeof(buf), "size: %dx%d\npalette: %s\n", - st->codec->width, st->codec->height, pal_s); + if (av_strlcat(buf, "\n", sizeof(buf)) >= sizeof(buf)) + return 0; av_freep(&st->codec->extradata); st->codec->extradata_size = 0; |