Merge "DO NOT MERGE: MPEG4Extractor: ensure returned status is checked." into lmp-dev (61df326a) · Commits · e / os / android_frameworks_av

media/libstagefright/MPEG4Extractor.cpp

+51 −6

Original line number	Diff line number	Diff line
		@@ -67,6 +67,7 @@ public:
		Vector<SidxEntry> &sidx,
		const Trex *trex,
		off64_t firstMoofOffset);
		virtual status_t init();

		virtual status_t start(MetaData *params = NULL);
		virtual status_t stop();
		@@ -2121,7 +2122,10 @@ status_t MPEG4Extractor::parseChunk(off64_t *offset, int depth) {
		if (mLastTrack == NULL) {
		return ERROR_MALFORMED;
		}
		parseSegmentIndex(data_offset, chunk_data_size);
		status_t err = parseSegmentIndex(data_offset, chunk_data_size);
		if (err != OK) {
		return err;
		}
		*offset += chunk_size;
		return UNKNOWN_ERROR; // stop parsing after sidx
		}
		@@ -2763,9 +2767,13 @@ sp<MediaSource> MPEG4Extractor::getTrack(size_t index) {

		ALOGV("getTrack called, pssh: %zu", mPssh.size());

		return new MPEG4Source(this,
		sp<MPEG4Source> source = new MPEG4Source(this,
		track->meta, mDataSource, track->timescale, track->sampleTable,
		mSidxEntries, trex, mMoofOffset);
		if (source->init() != OK) {
		return NULL;
		}
		return source;
		}

		// static
		@@ -3129,6 +3137,7 @@ MPEG4Source::MPEG4Source(
		mTrex(trex),
		mFirstMoofOffset(firstMoofOffset),
		mCurrentMoofOffset(firstMoofOffset),
		mNextMoofOffset(-1),
		mCurrentTime(0),
		mCurrentSampleInfoAllocSize(0),
		mCurrentSampleInfoSizes(NULL),
		@@ -3193,10 +3202,14 @@ MPEG4Source::MPEG4Source(

		CHECK(format->findInt32(kKeyTrackID, &mTrackId));

		}

		status_t MPEG4Source::init() {
		if (mFirstMoofOffset != 0) {
		off64_t offset = mFirstMoofOffset;
		parseChunk(&offset);
		return parseChunk(&offset);
		}
		return OK;
		}

		MPEG4Source::~MPEG4Source() {
		@@ -3320,9 +3333,30 @@ status_t MPEG4Source::parseChunk(off64_t *offset) {
		}
		chunk_size = ntohl(hdr[0]);
		chunk_type = ntohl(hdr[1]);
		if (chunk_size == 1) {
		// ISO/IEC 14496-12:2012, 8.8.4 Movie Fragment Box, moof is a Box
		// which is defined in 4.2 Object Structure.
		// When chunk_size==1, 8 bytes follows as "largesize".
		if (mDataSource->readAt(*offset + 8, &chunk_size, 8) < 8) {
		return ERROR_IO;
		}
		chunk_size = ntoh64(chunk_size);
		if (chunk_size < 16) {
		// The smallest valid chunk is 16 bytes long in this case.
		return ERROR_MALFORMED;
		}
		} else if (chunk_size == 0) {
		// next box extends to end of file.
		} else if (chunk_size < 8) {
		// The smallest valid chunk is 8 bytes long in this case.
		return ERROR_MALFORMED;
		}

		if (chunk_type == FOURCC('m', 'o', 'o', 'f')) {
		mNextMoofOffset = *offset;
		break;
		} else if (chunk_size == 0) {
		break;
		}
		*offset += chunk_size;
		}
		@@ -4187,17 +4221,25 @@ status_t MPEG4Source::fragmentedRead(
		totalOffset += se->mSize;
		}
		mCurrentMoofOffset = totalOffset;
		mNextMoofOffset = -1;
		mCurrentSamples.clear();
		mCurrentSampleIndex = 0;
		parseChunk(&totalOffset);
		status_t err = parseChunk(&totalOffset);
		if (err != OK) {
		return err;
		}
		mCurrentTime = totalTime * mTimescale / 1000000ll;
		} else {
		// without sidx boxes, we can only seek to 0
		mCurrentMoofOffset = mFirstMoofOffset;
		mNextMoofOffset = -1;
		mCurrentSamples.clear();
		mCurrentSampleIndex = 0;
		off64_t tmp = mCurrentMoofOffset;
		parseChunk(&tmp);
		status_t err = parseChunk(&tmp);
		if (err != OK) {
		return err;
		}
		mCurrentTime = 0;
		}

		@@ -4226,7 +4268,10 @@ status_t MPEG4Source::fragmentedRead(
		mCurrentMoofOffset = nextMoof;
		mCurrentSamples.clear();
		mCurrentSampleIndex = 0;
		parseChunk(&nextMoof);
		status_t err = parseChunk(&nextMoof);
		if (err != OK) {
		return err;
		}
		if (mCurrentSampleIndex >= mCurrentSamples.size()) {
		return ERROR_END_OF_STREAM;
		}