diff options
Diffstat (limited to 'chromium/media/formats/mp4/box_reader.h')
-rw-r--r-- | chromium/media/formats/mp4/box_reader.h | 222 |
1 files changed, 222 insertions, 0 deletions
diff --git a/chromium/media/formats/mp4/box_reader.h b/chromium/media/formats/mp4/box_reader.h new file mode 100644 index 00000000000..3360204ed54 --- /dev/null +++ b/chromium/media/formats/mp4/box_reader.h @@ -0,0 +1,222 @@ +// Copyright 2014 The Chromium Authors. All rights reserved. +// Use of this source code is governed by a BSD-style license that can be +// found in the LICENSE file. + +#ifndef MEDIA_FORMATS_MP4_BOX_READER_H_ +#define MEDIA_FORMATS_MP4_BOX_READER_H_ + +#include <map> +#include <vector> + +#include "base/compiler_specific.h" +#include "base/logging.h" +#include "media/base/media_export.h" +#include "media/base/media_log.h" +#include "media/formats/mp4/fourccs.h" +#include "media/formats/mp4/rcheck.h" + +namespace media { +namespace mp4 { + +class BoxReader; + +struct MEDIA_EXPORT Box { + virtual ~Box(); + virtual bool Parse(BoxReader* reader) = 0; + virtual FourCC BoxType() const = 0; +}; + +class MEDIA_EXPORT BufferReader { + public: + BufferReader(const uint8* buf, const int size) + : buf_(buf), size_(size), pos_(0) { + CHECK(buf); + CHECK_GE(size, 0); + } + + bool HasBytes(int count) { return (pos() + count <= size()); } + + // Read a value from the stream, perfoming endian correction, and advance the + // stream pointer. + bool Read1(uint8* v) WARN_UNUSED_RESULT; + bool Read2(uint16* v) WARN_UNUSED_RESULT; + bool Read2s(int16* v) WARN_UNUSED_RESULT; + bool Read4(uint32* v) WARN_UNUSED_RESULT; + bool Read4s(int32* v) WARN_UNUSED_RESULT; + bool Read8(uint64* v) WARN_UNUSED_RESULT; + bool Read8s(int64* v) WARN_UNUSED_RESULT; + + bool ReadFourCC(FourCC* v) WARN_UNUSED_RESULT; + + bool ReadVec(std::vector<uint8>* t, int count) WARN_UNUSED_RESULT; + + // These variants read a 4-byte integer of the corresponding signedness and + // store it in the 8-byte return type. + bool Read4Into8(uint64* v) WARN_UNUSED_RESULT; + bool Read4sInto8s(int64* v) WARN_UNUSED_RESULT; + + // Advance the stream by this many bytes. + bool SkipBytes(int nbytes) WARN_UNUSED_RESULT; + + const uint8* data() const { return buf_; } + int size() const { return size_; } + int pos() const { return pos_; } + + protected: + const uint8* buf_; + int size_; + int pos_; + + template<typename T> bool Read(T* t) WARN_UNUSED_RESULT; +}; + +class MEDIA_EXPORT BoxReader : public BufferReader { + public: + ~BoxReader(); + + // Create a BoxReader from a buffer. Note that this function may return NULL + // if an intact, complete box was not available in the buffer. If |*err| is + // set, there was a stream-level error when creating the box; otherwise, NULL + // values are only expected when insufficient data is available. + // + // |buf| is retained but not owned, and must outlive the BoxReader instance. + static BoxReader* ReadTopLevelBox(const uint8* buf, + const int buf_size, + const LogCB& log_cb, + bool* err); + + // Read the box header from the current buffer. This function returns true if + // there is enough data to read the header and the header is sane; that is, it + // does not check to ensure the entire box is in the buffer before returning + // true. The semantics of |*err| are the same as above. + // + // |buf| is not retained. + static bool StartTopLevelBox(const uint8* buf, + const int buf_size, + const LogCB& log_cb, + FourCC* type, + int* box_size, + bool* err) WARN_UNUSED_RESULT; + + // Returns true if |type| is recognized to be a top-level box, false + // otherwise. This returns true for some boxes which we do not parse. + // Helpful in debugging misaligned appends. + static bool IsValidTopLevelBox(const FourCC& type, + const LogCB& log_cb); + + // Scan through all boxes within the current box, starting at the current + // buffer position. Must be called before any of the *Child functions work. + bool ScanChildren() WARN_UNUSED_RESULT; + + // Return true if child with type |child.BoxType()| exists. + bool HasChild(Box* child) WARN_UNUSED_RESULT; + + // Read exactly one child box from the set of children. The type of the child + // will be determined by the BoxType() method of |child|. + bool ReadChild(Box* child) WARN_UNUSED_RESULT; + + // Read one child if available. Returns false on error, true on successful + // read or on child absent. + bool MaybeReadChild(Box* child) WARN_UNUSED_RESULT; + + // Read at least one child. False means error or no such child present. + template<typename T> bool ReadChildren( + std::vector<T>* children) WARN_UNUSED_RESULT; + + // Read any number of children. False means error. + template<typename T> bool MaybeReadChildren( + std::vector<T>* children) WARN_UNUSED_RESULT; + + // Read all children, regardless of FourCC. This is used from exactly one box, + // corresponding to a rather significant inconsistency in the BMFF spec. + // Note that this method is mutually exclusive with ScanChildren(). + template<typename T> bool ReadAllChildren( + std::vector<T>* children) WARN_UNUSED_RESULT; + + // Populate the values of 'version()' and 'flags()' from a full box header. + // Many boxes, but not all, use these values. This call should happen after + // the box has been initialized, and does not re-read the main box header. + bool ReadFullBoxHeader() WARN_UNUSED_RESULT; + + FourCC type() const { return type_; } + uint8 version() const { return version_; } + uint32 flags() const { return flags_; } + + const LogCB& log_cb() const { return log_cb_; } + + private: + BoxReader(const uint8* buf, const int size, const LogCB& log_cb); + + // Must be called immediately after init. If the return is false, this + // indicates that the box header and its contents were not available in the + // stream or were nonsensical, and that the box must not be used further. In + // this case, if |*err| is false, the problem was simply a lack of data, and + // should only be an error condition if some higher-level component knows that + // no more data is coming (i.e. EOS or end of containing box). If |*err| is + // true, the error is unrecoverable and the stream should be aborted. + bool ReadHeader(bool* err); + + LogCB log_cb_; + FourCC type_; + uint8 version_; + uint32 flags_; + + typedef std::multimap<FourCC, BoxReader> ChildMap; + + // The set of child box FourCCs and their corresponding buffer readers. Only + // valid if scanned_ is true. + ChildMap children_; + bool scanned_; +}; + +// Template definitions +template<typename T> bool BoxReader::ReadChildren(std::vector<T>* children) { + RCHECK(MaybeReadChildren(children) && !children->empty()); + return true; +} + +template<typename T> +bool BoxReader::MaybeReadChildren(std::vector<T>* children) { + DCHECK(scanned_); + DCHECK(children->empty()); + + children->resize(1); + FourCC child_type = (*children)[0].BoxType(); + + ChildMap::iterator start_itr = children_.lower_bound(child_type); + ChildMap::iterator end_itr = children_.upper_bound(child_type); + children->resize(std::distance(start_itr, end_itr)); + typename std::vector<T>::iterator child_itr = children->begin(); + for (ChildMap::iterator itr = start_itr; itr != end_itr; ++itr) { + RCHECK(child_itr->Parse(&itr->second)); + ++child_itr; + } + children_.erase(start_itr, end_itr); + + DVLOG(2) << "Found " << children->size() << " " + << FourCCToString(child_type) << " boxes."; + return true; +} + +template<typename T> +bool BoxReader::ReadAllChildren(std::vector<T>* children) { + DCHECK(!scanned_); + scanned_ = true; + + bool err = false; + while (pos() < size()) { + BoxReader child_reader(&buf_[pos_], size_ - pos_, log_cb_); + if (!child_reader.ReadHeader(&err)) break; + T child; + RCHECK(child.Parse(&child_reader)); + children->push_back(child); + pos_ += child_reader.size(); + } + + return !err; +} + +} // namespace mp4 +} // namespace media + +#endif // MEDIA_FORMATS_MP4_BOX_READER_H_ |