From 188602e662868a79c7f85e9193e9aedc9ba1a170 Mon Sep 17 00:00:00 2001
From: Sasi Inguva <isasi@google.com>
Date: Wed, 18 Oct 2017 20:11:16 -0700
Subject: [PATCH] lavf/mov.c: Fix parsing of edit list atoms with invalid elst
entry count.
Signed-off-by: Sasi Inguva <isasi@google.com>
---
libavformat/mov.c | 21 ++++++++++-
tests/fate/mov.mak | 4 ++
tests/ref/fate/mov-invalid-elst-entry-count | 57 +++++++++++++++++++++++++++++
3 files changed, 81 insertions(+), 1 deletion(-)
create mode 100644 tests/ref/fate/mov-invalid-elst-entry-count
@@ -4597,6 +4597,7 @@ static int mov_read_elst(MOVContext *c, AVIOContext *pb, MOVAtom atom)
{
MOVStreamContext *sc;
int i, edit_count, version;
+ int64_t elst_entry_size;
if (c->fc->nb_streams < 1 || c->ignore_editlist)
return 0;
@@ -4605,6 +4606,21 @@ static int mov_read_elst(MOVContext *c, AVIOContext *pb, MOVAtom atom)
version = avio_r8(pb); /* version */
avio_rb24(pb); /* flags */
edit_count = avio_rb32(pb); /* entries */
+ atom.size -= 8;
+
+ elst_entry_size = version == 1 ? 20 : 12;
+ if (atom.size != edit_count * elst_entry_size) {
+ if (c->fc->strict_std_compliance >= FF_COMPLIANCE_STRICT) {
+ av_log(c->fc, AV_LOG_ERROR, "Invalid edit list entry_count: %d for elst atom of size: %"PRId64" bytes.\n",
+ edit_count, atom.size + 8);
+ return AVERROR_INVALIDDATA;
+ } else {
+ edit_count = atom.size / elst_entry_size;
+ if (edit_count * elst_entry_size != atom.size) {
+ av_log(c->fc, AV_LOG_WARNING, "ELST atom of %"PRId64" bytes, bigger than %d entries.", atom.size, edit_count);
+ }
+ }
+ }
if (!edit_count)
return 0;
@@ -4617,17 +4633,20 @@ static int mov_read_elst(MOVContext *c, AVIOContext *pb, MOVAtom atom)
return AVERROR(ENOMEM);
av_log(c->fc, AV_LOG_TRACE, "track[%u].edit_count = %i\n", c->fc->nb_streams - 1, edit_count);
- for (i = 0; i < edit_count && !pb->eof_reached; i++) {
+ for (i = 0; i < edit_count && atom.size > 0 && !pb->eof_reached; i++) {
MOVElst *e = &sc->elst_data[i];
if (version == 1) {
e->duration = avio_rb64(pb);
e->time = avio_rb64(pb);
+ atom.size -= 16;
} else {
e->duration = avio_rb32(pb); /* segment duration */
e->time = (int32_t)avio_rb32(pb); /* media time */
+ atom.size -= 8;
}
e->rate = avio_rb32(pb) / 65536.0;
+ atom.size -= 4;
av_log(c->fc, AV_LOG_TRACE, "duration=%"PRId64" time=%"PRId64" rate=%f\n",
e->duration, e->time, e->rate);
@@ -6,6 +6,7 @@ FATE_MOV = fate-mov-3elist \
fate-mov-1elist-ends-last-bframe \
fate-mov-2elist-elist1-ends-bframe \
fate-mov-3elist-encrypted \
+ fate-mov-invalid-elst-entry-count \
fate-mov-gpmf-remux \
FATE_MOV_FFPROBE = fate-mov-aac-2048-priming \
@@ -39,6 +40,9 @@ fate-mov-1elist-ends-last-bframe: CMD = framemd5 -i $(TARGET_SAMPLES)/mov/mov-1e
# Makes sure that we handle timestamps of packets in case of multiple edit lists with one of them ending on a B-frame correctly.
fate-mov-2elist-elist1-ends-bframe: CMD = framemd5 -i $(TARGET_SAMPLES)/mov/mov-2elist-elist1-ends-bframe.mov
+# Makes sure that we handle invalid edit list entry count correctly.
+fate-mov-invalid-elst-entry-count: CMD = framemd5 -i $(TARGET_SAMPLES)/mov/invalid_elst_entry_count.mov
+
fate-mov-aac-2048-priming: CMD = run ffprobe$(PROGSSUF)$(EXESUF) -show_packets -print_format compact $(TARGET_SAMPLES)/mov/aac-2048-priming.mov
fate-mov-zombie: CMD = run ffprobe$(PROGSSUF)$(EXESUF) -show_streams -show_packets -show_frames -bitexact -print_format compact $(TARGET_SAMPLES)/mov/white_zombie_scrunch-part.mov
new file mode 100644
@@ -0,0 +1,57 @@
+#format: frame checksums
+#version: 2
+#hash: MD5
+#tb 0: 1/24
+#media_type 0: video
+#codec_id 0: rawvideo
+#dimensions 0: 640x480
+#sar 0: 1/1
+#stream#, dts, pts, duration, size, hash
+0, 0, 0, 1, 460800, 549730883a0b56e6accaf021903daecf
+0, 1, 1, 1, 460800, 783389b4342d4be925fc5244791e760a
+0, 2, 2, 1, 460800, 8384af6426d94a2077930c93013e09ad
+0, 3, 3, 1, 460800, 9380a1d9ecacf5b3105383c1c8083188
+0, 4, 4, 1, 460800, eb28174acfceb868b9058757bed049c5
+0, 5, 5, 1, 460800, 9732bd4a58884dbf2be48d819433130f
+0, 6, 6, 1, 460800, 0c553fb530cf042eb84f5b13817a96a6
+0, 7, 7, 1, 460800, 621f02aded5e35fa1f373afd3ed283bd
+0, 8, 8, 1, 460800, c76167c6bda91f657708c88252ea315d
+0, 9, 9, 1, 460800, 872df2d8c522e2440ddd04bca7dce497
+0, 10, 10, 1, 460800, 6ee9154e48c5132ad4ba122b255bd2bb
+0, 11, 11, 1, 460800, 362e61629795702ebe9183ce3786d7f2
+0, 12, 12, 1, 460800, f3ec59e6fc4e3c2e75f42bef34ca73b5
+0, 13, 13, 1, 460800, 68d9caea8697736dd716cba43b614919
+0, 14, 14, 1, 460800, 4a4efb0201a64236db4330725758c139
+0, 15, 15, 1, 460800, f32f8997dcdd87ad910dea886a0de17d
+0, 16, 16, 1, 460800, 51a8549d7b4aaacaf6050bc07a82b440
+0, 17, 17, 1, 460800, 5145aa05bbb0c3faab40fc8fa233af1d
+0, 18, 18, 1, 460800, bbfcbe3c9600b2a0f413057d7959e9e7
+0, 19, 19, 1, 460800, 02cfd4a350fa274e12fce8352001bf21
+0, 20, 20, 1, 460800, 20dd372da9e656add433f31e3e9c1fb8
+0, 21, 21, 1, 460800, 3b885593f8b42676ce40c329a63f62bf
+0, 22, 22, 1, 460800, c38b453b56c3ea14f7d8691d83752486
+0, 23, 23, 1, 460800, 79643132988dabc9dc1ba3af0aeaebc5
+0, 24, 24, 1, 460800, 60a099be31244b2f69ca6107cdbd7e06
+0, 25, 25, 1, 460800, 1de6ff4e0aa81216e4b7b1c8e74fb143
+0, 26, 26, 1, 460800, 5223a81e6964c28cf42593f259397aa1
+0, 27, 27, 1, 460800, 2dfcf01c86aa712cd6f1c7656eeb17db
+0, 28, 28, 1, 460800, 8c86ee0f02fabccaed8d8fc8babd031e
+0, 29, 29, 1, 460800, b3ea1983f7efeec11306445d9ae5d477
+0, 30, 30, 1, 460800, 86a4cc9fa7db5ff5ca2be69ad191179f
+0, 31, 31, 1, 460800, 8194715afe23ae34a019797a53a6ee2c
+0, 32, 32, 1, 460800, 447a153f1c6bb703eff62edfd14e08e0
+0, 33, 33, 1, 460800, 092257082789b898dbb14d1f19e79347
+0, 34, 34, 1, 460800, d6320d204a119cfeef5645a4118bc600
+0, 35, 35, 1, 460800, 2ee710deae4bb0d156528797ad1c4981
+0, 36, 36, 1, 460800, 1256eac030985c04c4501ad5a72e9d66
+0, 37, 37, 1, 460800, f16ad8c1aa572be7666c7907ce4aebbc
+0, 38, 38, 1, 460800, 865088cbd47d0151b62a45d5426c8216
+0, 39, 39, 1, 460800, 26c78ca43d93c6da153f3dea5d945e0e
+0, 40, 40, 1, 460800, d775d6705c965401ccc143d5ae432938
+0, 41, 41, 1, 460800, f9837d514753c59e6776452d9043524f
+0, 42, 42, 1, 460800, 8463f5172914828abcc770f888bfd183
+0, 43, 43, 1, 460800, 3108557748cfb7965b33b16b35359de0
+0, 44, 44, 1, 460800, 477d596944e028dd72c207bb6e6b22de
+0, 45, 45, 1, 460800, 69e4ffbd600c8d8bc070d7d7324ee2b1
+0, 46, 46, 1, 460800, 2211c57bc9ec1788217002900f9102b1
+0, 47, 47, 1, 460800, bcfb5f0a7f88da3ec8c6b447ed4b67eb
--
2.15.0.rc2.357.g7e34df9404-goog