diff options
| -rw-r--r-- | media/libstagefright/MPEG4Extractor.cpp | 16 | 
1 files changed, 12 insertions, 4 deletions
diff --git a/media/libstagefright/MPEG4Extractor.cpp b/media/libstagefright/MPEG4Extractor.cpp index e009a0e..66a3793 100644 --- a/media/libstagefright/MPEG4Extractor.cpp +++ b/media/libstagefright/MPEG4Extractor.cpp @@ -3089,16 +3089,24 @@ status_t MPEG4Source::start(MetaData *params) {          mWantsNALFragments = false;      } +    int32_t tmp; +    CHECK(mFormat->findInt32(kKeyMaxInputSize, &tmp)); +    size_t max_size = tmp; + +    // A somewhat arbitrary limit that should be sufficient for 8k video frames +    // If you see the message below for a valid input stream: increase the limit +    if (max_size > 64 * 1024 * 1024) { +        ALOGE("bogus max input size: %zu", max_size); +        return ERROR_MALFORMED; +    }      mGroup = new MediaBufferGroup; - -    int32_t max_size; -    CHECK(mFormat->findInt32(kKeyMaxInputSize, &max_size)); -      mGroup->add_buffer(new MediaBuffer(max_size));      mSrcBuffer = new (std::nothrow) uint8_t[max_size];      if (mSrcBuffer == NULL) {          // file probably specified a bad max size +        delete mGroup; +        mGroup = NULL;          return ERROR_MALFORMED;      }  | 
