Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(333)

Unified Diff: media/mp4/box_reader.cc

Issue 10823139: Set error on unrecognized top-level BMFF boxes. (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src
Patch Set: Update comment Created 8 years, 5 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « media/mp4/box_reader.h ('k') | media/mp4/box_reader_unittest.cc » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: media/mp4/box_reader.cc
diff --git a/media/mp4/box_reader.cc b/media/mp4/box_reader.cc
index 8892a43991cbd1e9b976134c1ca00f640baaca83..b352664a6ffe303ad3e337635ab9f061416b612d 100644
--- a/media/mp4/box_reader.cc
+++ b/media/mp4/box_reader.cc
@@ -10,6 +10,7 @@
#include <set>
#include "base/logging.h"
+#include "base/memory/scoped_ptr.h"
#include "media/mp4/box_definitions.h"
#include "media/mp4/rcheck.h"
@@ -95,14 +96,22 @@ BoxReader::~BoxReader() {
}
}
+// static
BoxReader* BoxReader::ReadTopLevelBox(const uint8* buf,
const int buf_size,
bool* err) {
- BoxReader* reader = new BoxReader(buf, buf_size);
- if (reader->ReadHeader(err) && reader->size() <= buf_size) {
- return reader;
+ scoped_ptr<BoxReader> reader(new BoxReader(buf, buf_size));
+ if (!reader->ReadHeader(err))
+ return NULL;
+
+ if (!IsValidTopLevelBox(reader->type())) {
+ *err = true;
+ return NULL;
}
- delete reader;
+
+ if (reader->size() <= buf_size)
+ return reader.release();
+
return NULL;
}
@@ -114,11 +123,41 @@ bool BoxReader::StartTopLevelBox(const uint8* buf,
bool* err) {
BoxReader reader(buf, buf_size);
if (!reader.ReadHeader(err)) return false;
+ if (!IsValidTopLevelBox(reader.type())) {
+ *err = true;
+ return false;
+ }
*type = reader.type();
*box_size = reader.size();
return true;
}
+// static
+bool BoxReader::IsValidTopLevelBox(const FourCC& type) {
+ switch (type) {
+ case FOURCC_FTYP:
+ case FOURCC_PDIN:
+ case FOURCC_MOOV:
+ case FOURCC_MOOF:
+ case FOURCC_MFRA:
+ case FOURCC_MDAT:
+ case FOURCC_FREE:
+ case FOURCC_SKIP:
+ case FOURCC_META:
+ case FOURCC_MECO:
+ case FOURCC_STYP:
+ case FOURCC_SIDX:
+ case FOURCC_SSIX:
+ case FOURCC_PRFT:
+ return true;
+ default:
+ // Hex is used to show nonprintable characters and aid in debugging
+ LOG(WARNING) << "Unrecognized top-level box type 0x"
+ << std::hex << type;
+ return false;
+ }
+}
+
bool BoxReader::ScanChildren() {
DCHECK(!scanned_);
scanned_ = true;
« no previous file with comments | « media/mp4/box_reader.h ('k') | media/mp4/box_reader_unittest.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698