DASH Media Packaging SDK
 All Classes Namespaces Functions Variables Typedefs Enumerator
mp4_media_parser.cc
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4 
5 #include "packager/media/formats/mp4/mp4_media_parser.h"
6 
7 #include <limits>
8 
9 #include "packager/base/callback.h"
10 #include "packager/base/callback_helpers.h"
11 #include "packager/base/logging.h"
12 #include "packager/base/memory/ref_counted.h"
13 #include "packager/base/strings/string_number_conversions.h"
14 #include "packager/media/base/audio_stream_info.h"
15 #include "packager/media/base/buffer_reader.h"
16 #include "packager/media/base/decrypt_config.h"
17 #include "packager/media/base/key_source.h"
18 #include "packager/media/base/macros.h"
19 #include "packager/media/base/media_sample.h"
20 #include "packager/media/base/video_stream_info.h"
21 #include "packager/media/file/file.h"
22 #include "packager/media/file/file_closer.h"
23 #include "packager/media/filters/avc_decoder_configuration.h"
24 #include "packager/media/filters/hevc_decoder_configuration.h"
25 #include "packager/media/filters/vp_codec_configuration.h"
26 #include "packager/media/formats/mp4/box_definitions.h"
27 #include "packager/media/formats/mp4/box_reader.h"
28 #include "packager/media/formats/mp4/es_descriptor.h"
29 #include "packager/media/formats/mp4/rcheck.h"
30 #include "packager/media/formats/mp4/track_run_iterator.h"
31 
32 namespace edash_packager {
33 namespace media {
34 namespace mp4 {
35 namespace {
36 
37 uint64_t Rescale(uint64_t time_in_old_scale,
38  uint32_t old_scale,
39  uint32_t new_scale) {
40  return (static_cast<double>(time_in_old_scale) / old_scale) * new_scale;
41 }
42 
43 VideoCodec FourCCToVideoCodec(FourCC fourcc) {
44  switch (fourcc) {
45  case FOURCC_AVC1:
46  return kCodecH264;
47  case FOURCC_HEV1:
48  return kCodecHEV1;
49  case FOURCC_HVC1:
50  return kCodecHVC1;
51  case FOURCC_VP08:
52  return kCodecVP8;
53  case FOURCC_VP09:
54  return kCodecVP9;
55  case FOURCC_VP10:
56  return kCodecVP10;
57  default:
58  return kUnknownVideoCodec;
59  }
60 }
61 
62 AudioCodec FourCCToAudioCodec(FourCC fourcc) {
63  switch(fourcc) {
64  case FOURCC_DTSC:
65  return kCodecDTSC;
66  case FOURCC_DTSH:
67  return kCodecDTSH;
68  case FOURCC_DTSL:
69  return kCodecDTSL;
70  case FOURCC_DTSE:
71  return kCodecDTSE;
72  case FOURCC_DTSP:
73  return kCodecDTSP;
74  case FOURCC_DTSM:
75  return kCodecDTSM;
76  case FOURCC_AC3:
77  return kCodecAC3;
78  case FOURCC_EAC3:
79  return kCodecEAC3;
80  default:
81  return kUnknownAudioCodec;
82  }
83 }
84 
85 const char kWidevineKeySystemId[] = "edef8ba979d64acea3c827dcd51d21ed";
86 // Default DTS audio number of channels for 5.1 channel layout.
87 const uint8_t kDtsAudioNumChannels = 6;
88 
89 } // namespace
90 
91 MP4MediaParser::MP4MediaParser()
92  : state_(kWaitingForInit),
93  decryption_key_source_(NULL),
94  moof_head_(0),
95  mdat_tail_(0) {}
96 
97 MP4MediaParser::~MP4MediaParser() {}
98 
99 void MP4MediaParser::Init(const InitCB& init_cb,
100  const NewSampleCB& new_sample_cb,
101  KeySource* decryption_key_source) {
102  DCHECK_EQ(state_, kWaitingForInit);
103  DCHECK(init_cb_.is_null());
104  DCHECK(!init_cb.is_null());
105  DCHECK(!new_sample_cb.is_null());
106 
107  ChangeState(kParsingBoxes);
108  init_cb_ = init_cb;
109  new_sample_cb_ = new_sample_cb;
110  decryption_key_source_ = decryption_key_source;
111  if (decryption_key_source)
112  decryptor_source_.reset(new DecryptorSource(decryption_key_source));
113 }
114 
115 void MP4MediaParser::Reset() {
116  queue_.Reset();
117  runs_.reset();
118  moof_head_ = 0;
119  mdat_tail_ = 0;
120 }
121 
123  DCHECK_NE(state_, kWaitingForInit);
124  Reset();
125  ChangeState(kParsingBoxes);
126  return true;
127 }
128 
129 bool MP4MediaParser::Parse(const uint8_t* buf, int size) {
130  DCHECK_NE(state_, kWaitingForInit);
131 
132  if (state_ == kError)
133  return false;
134 
135  queue_.Push(buf, size);
136 
137  bool result, err = false;
138 
139  do {
140  if (state_ == kParsingBoxes) {
141  result = ParseBox(&err);
142  } else {
143  DCHECK_EQ(kEmittingSamples, state_);
144  result = EnqueueSample(&err);
145  if (result) {
146  int64_t max_clear = runs_->GetMaxClearOffset() + moof_head_;
147  err = !ReadAndDiscardMDATsUntil(max_clear);
148  }
149  }
150  } while (result && !err);
151 
152  if (err) {
153  DLOG(ERROR) << "Error while parsing MP4";
154  moov_.reset();
155  Reset();
156  ChangeState(kError);
157  return false;
158  }
159 
160  return true;
161 }
162 
163 bool MP4MediaParser::LoadMoov(const std::string& file_path) {
164  scoped_ptr<File, FileCloser> file(
165  File::OpenWithNoBuffering(file_path.c_str(), "r"));
166  if (!file) {
167  LOG(ERROR) << "Unable to open media file '" << file_path << "'";
168  return false;
169  }
170  if (!file->Seek(0)) {
171  LOG(WARNING) << "Filesystem does not support seeking on file '" << file_path
172  << "'";
173  return false;
174  }
175 
176  uint64_t file_position(0);
177  bool mdat_seen(false);
178  while (true) {
179  const uint32_t kBoxHeaderReadSize(16);
180  std::vector<uint8_t> buffer(kBoxHeaderReadSize);
181  int64_t bytes_read = file->Read(&buffer[0], kBoxHeaderReadSize);
182  if (bytes_read == 0) {
183  LOG(ERROR) << "Could not find 'moov' box in file '" << file_path << "'";
184  return false;
185  }
186  if (bytes_read < kBoxHeaderReadSize) {
187  LOG(ERROR) << "Error reading media file '" << file_path << "'";
188  return false;
189  }
190  uint64_t box_size;
191  FourCC box_type;
192  bool err;
193  if (!BoxReader::StartTopLevelBox(&buffer[0], kBoxHeaderReadSize, &box_type,
194  &box_size, &err)) {
195  LOG(ERROR) << "Could not start top level box from file '" << file_path
196  << "'";
197  return false;
198  }
199  if (box_type == FOURCC_MDAT) {
200  mdat_seen = true;
201  } else if (box_type == FOURCC_MOOV) {
202  if (!mdat_seen) {
203  // 'moov' is before 'mdat'. Nothing to do.
204  break;
205  }
206  // 'mdat' before 'moov'. Read and parse 'moov'.
207  if (!Parse(&buffer[0], bytes_read)) {
208  LOG(ERROR) << "Error parsing mp4 file '" << file_path << "'";
209  return false;
210  }
211  uint64_t bytes_to_read = box_size - bytes_read;
212  buffer.resize(bytes_to_read);
213  while (bytes_to_read > 0) {
214  bytes_read = file->Read(&buffer[0], bytes_to_read);
215  if (bytes_read <= 0) {
216  LOG(ERROR) << "Error reading 'moov' contents from file '" << file_path
217  << "'";
218  return false;
219  }
220  if (!Parse(&buffer[0], bytes_read)) {
221  LOG(ERROR) << "Error parsing mp4 file '" << file_path << "'";
222  return false;
223  }
224  bytes_to_read -= bytes_read;
225  }
226  queue_.Reset(); // So that we don't need to adjust data offsets.
227  mdat_tail_ = 0; // So it will skip boxes until mdat.
228  break; // Done.
229  }
230  file_position += box_size;
231  if (!file->Seek(file_position)) {
232  LOG(ERROR) << "Error skipping box in mp4 file '" << file_path << "'";
233  return false;
234  }
235  }
236  return true;
237 }
238 
239 bool MP4MediaParser::ParseBox(bool* err) {
240  const uint8_t* buf;
241  int size;
242  queue_.Peek(&buf, &size);
243  if (!size)
244  return false;
245 
246  scoped_ptr<BoxReader> reader(BoxReader::ReadTopLevelBox(buf, size, err));
247  if (reader.get() == NULL)
248  return false;
249 
250  if (reader->type() == FOURCC_MDAT) {
251  // The code ends up here only if a MOOV box is not yet seen.
252  DCHECK(!moov_);
253 
254  NOTIMPLEMENTED() << " Files with MDAT before MOOV is not supported yet.";
255  *err = true;
256  return false;
257  }
258 
259  // Set up mdat offset for ReadMDATsUntil().
260  mdat_tail_ = queue_.head() + reader->size();
261 
262  if (reader->type() == FOURCC_MOOV) {
263  *err = !ParseMoov(reader.get());
264  } else if (reader->type() == FOURCC_MOOF) {
265  moof_head_ = queue_.head();
266  *err = !ParseMoof(reader.get());
267 
268  // Return early to avoid evicting 'moof' data from queue. Auxiliary info may
269  // be located anywhere in the file, including inside the 'moof' itself.
270  // (Since 'default-base-is-moof' is mandated, no data references can come
271  // before the head of the 'moof', so keeping this box around is sufficient.)
272  return !(*err);
273  } else {
274  VLOG(2) << "Skipping top-level box: " << FourCCToString(reader->type());
275  }
276 
277  queue_.Pop(reader->size());
278  return !(*err);
279 }
280 
281 bool MP4MediaParser::ParseMoov(BoxReader* reader) {
282  if (moov_)
283  return true; // Already parsed the 'moov' box.
284 
285  moov_.reset(new Movie);
286  RCHECK(moov_->Parse(reader));
287  runs_.reset();
288 
289  std::vector<scoped_refptr<StreamInfo> > streams;
290 
291  for (std::vector<Track>::const_iterator track = moov_->tracks.begin();
292  track != moov_->tracks.end(); ++track) {
293  const uint32_t timescale = track->media.header.timescale;
294 
295  // Calculate duration (based on timescale).
296  uint64_t duration = 0;
297  if (track->media.header.duration > 0) {
298  duration = track->media.header.duration;
299  } else if (moov_->extends.header.fragment_duration > 0) {
300  DCHECK(moov_->header.timescale != 0);
301  duration = Rescale(moov_->extends.header.fragment_duration,
302  moov_->header.timescale,
303  timescale);
304  } else if (moov_->header.duration > 0 &&
305  moov_->header.duration != std::numeric_limits<uint64_t>::max()) {
306  DCHECK(moov_->header.timescale != 0);
307  duration =
308  Rescale(moov_->header.duration, moov_->header.timescale, timescale);
309  }
310 
311  const SampleDescription& samp_descr =
312  track->media.information.sample_table.description;
313 
314  size_t desc_idx = 0;
315 
316  // Read sample description index from mvex if it exists otherwise read
317  // from the first entry in Sample To Chunk box.
318  if (moov_->extends.tracks.size() > 0) {
319  for (size_t t = 0; t < moov_->extends.tracks.size(); t++) {
320  const TrackExtends& trex = moov_->extends.tracks[t];
321  if (trex.track_id == track->header.track_id) {
322  desc_idx = trex.default_sample_description_index;
323  break;
324  }
325  }
326  } else {
327  const std::vector<ChunkInfo>& chunk_info =
328  track->media.information.sample_table.sample_to_chunk.chunk_info;
329  RCHECK(chunk_info.size() > 0);
330  desc_idx = chunk_info[0].sample_description_index;
331  }
332  RCHECK(desc_idx > 0);
333  desc_idx -= 1; // BMFF descriptor index is one-based
334 
335  if (samp_descr.type == kAudio) {
336  RCHECK(!samp_descr.audio_entries.empty());
337 
338  // It is not uncommon to find otherwise-valid files with incorrect sample
339  // description indices, so we fail gracefully in that case.
340  if (desc_idx >= samp_descr.audio_entries.size())
341  desc_idx = 0;
342 
343  const AudioSampleEntry& entry = samp_descr.audio_entries[desc_idx];
344  const FourCC actual_format = entry.GetActualFormat();
345  AudioCodec codec = FourCCToAudioCodec(actual_format);
346  uint8_t num_channels = 0;
347  uint32_t sampling_frequency = 0;
348  uint8_t audio_object_type = 0;
349  uint32_t max_bitrate = 0;
350  uint32_t avg_bitrate = 0;
351  std::vector<uint8_t> extra_data;
352 
353  switch (actual_format) {
354  case FOURCC_MP4A:
355  // Check if it is MPEG4 AAC defined in ISO 14496 Part 3 or
356  // supported MPEG2 AAC variants.
357  if (entry.esds.es_descriptor.IsAAC()) {
358  codec = kCodecAAC;
359  const AACAudioSpecificConfig& aac_audio_specific_config =
360  entry.esds.aac_audio_specific_config;
361  num_channels = aac_audio_specific_config.num_channels();
362  sampling_frequency = aac_audio_specific_config.frequency();
363  audio_object_type = aac_audio_specific_config.audio_object_type();
364  extra_data = entry.esds.es_descriptor.decoder_specific_info();
365  break;
366  } else if (entry.esds.es_descriptor.IsDTS()) {
367  ObjectType audio_type = entry.esds.es_descriptor.object_type();
368  switch (audio_type) {
369  case kDTSC:
370  codec = kCodecDTSC;
371  break;
372  case kDTSE:
373  codec = kCodecDTSE;
374  break;
375  case kDTSH:
376  codec = kCodecDTSH;
377  break;
378  case kDTSL:
379  codec = kCodecDTSL;
380  break;
381  default:
382  LOG(ERROR) << "Unsupported audio type " << audio_type
383  << " in stsd box.";
384  return false;
385  }
386  num_channels = entry.esds.aac_audio_specific_config.num_channels();
387  // For dts audio in esds, current supported number of channels is 6
388  // as the only supported channel layout is 5.1.
389  if (num_channels != kDtsAudioNumChannels) {
390  LOG(ERROR) << "Unsupported channel count " << num_channels
391  << " for audio type " << audio_type << ".";
392  return false;
393  }
394  sampling_frequency = entry.samplerate;
395  max_bitrate = entry.esds.es_descriptor.max_bitrate();
396  avg_bitrate = entry.esds.es_descriptor.avg_bitrate();
397  } else {
398  LOG(ERROR) << "Unsupported audio format 0x" << std::hex
399  << actual_format << " in stsd box.";
400  return false;
401  }
402  break;
403  case FOURCC_DTSC:
404  FALLTHROUGH_INTENDED;
405  case FOURCC_DTSH:
406  FALLTHROUGH_INTENDED;
407  case FOURCC_DTSL:
408  FALLTHROUGH_INTENDED;
409  case FOURCC_DTSE:
410  FALLTHROUGH_INTENDED;
411  case FOURCC_DTSM:
412  extra_data = entry.ddts.extra_data;
413  max_bitrate = entry.ddts.max_bitrate;
414  avg_bitrate = entry.ddts.avg_bitrate;
415  num_channels = entry.channelcount;
416  sampling_frequency = entry.samplerate;
417  break;
418  case FOURCC_AC3:
419  extra_data = entry.dac3.data;
420  num_channels = entry.channelcount;
421  sampling_frequency = entry.samplerate;
422  break;
423  case FOURCC_EAC3:
424  extra_data = entry.dec3.data;
425  num_channels = entry.channelcount;
426  sampling_frequency = entry.samplerate;
427  break;
428  default:
429  LOG(ERROR) << "Unsupported audio format 0x" << std::hex
430  << actual_format << " in stsd box.";
431  return false;
432  }
433 
434  bool is_encrypted = entry.sinf.info.track_encryption.is_encrypted;
435  DVLOG(1) << "is_audio_track_encrypted_: " << is_encrypted;
436  streams.push_back(new AudioStreamInfo(
437  track->header.track_id,
438  timescale,
439  duration,
440  codec,
441  AudioStreamInfo::GetCodecString(codec, audio_object_type),
442  track->media.header.language.code,
443  entry.samplesize,
444  num_channels,
445  sampling_frequency,
446  max_bitrate,
447  avg_bitrate,
448  vector_as_array(&extra_data),
449  extra_data.size(),
450  is_encrypted));
451  }
452 
453  if (samp_descr.type == kVideo) {
454  RCHECK(!samp_descr.video_entries.empty());
455  if (desc_idx >= samp_descr.video_entries.size())
456  desc_idx = 0;
457  const VideoSampleEntry& entry = samp_descr.video_entries[desc_idx];
458 
459  uint32_t coded_width = entry.width;
460  uint32_t coded_height = entry.height;
461  uint32_t pixel_width = entry.pixel_aspect.h_spacing;
462  uint32_t pixel_height = entry.pixel_aspect.v_spacing;
463  if (pixel_width == 0 && pixel_height == 0) {
464  pixel_width = 1;
465  pixel_height = 1;
466  }
467  std::string codec_string;
468  uint8_t nalu_length_size = 0;
469 
470  const FourCC actual_format = entry.GetActualFormat();
471  const VideoCodec video_codec = FourCCToVideoCodec(actual_format);
472  switch (actual_format) {
473  case FOURCC_AVC1: {
474  AVCDecoderConfiguration avc_config;
475  if (!avc_config.Parse(entry.codec_config_record.data)) {
476  LOG(ERROR) << "Failed to parse avcc.";
477  return false;
478  }
479  codec_string = avc_config.GetCodecString();
480  nalu_length_size = avc_config.length_size();
481 
482  if (coded_width != avc_config.coded_width() ||
483  coded_height != avc_config.coded_height()) {
484  LOG(WARNING) << "Resolution in VisualSampleEntry (" << coded_width
485  << "," << coded_height
486  << ") does not match with resolution in "
487  "AVCDecoderConfigurationRecord ("
488  << avc_config.coded_width() << ","
489  << avc_config.coded_height()
490  << "). Use AVCDecoderConfigurationRecord.";
491  coded_width = avc_config.coded_width();
492  coded_height = avc_config.coded_height();
493  }
494 
495  if (pixel_width != avc_config.pixel_width() ||
496  pixel_height != avc_config.pixel_height()) {
497  LOG_IF(WARNING, pixel_width != 1 || pixel_height != 1)
498  << "Pixel aspect ratio in PASP box (" << pixel_width << ","
499  << pixel_height
500  << ") does not match with SAR in AVCDecoderConfigurationRecord "
501  "("
502  << avc_config.pixel_width() << "," << avc_config.pixel_height()
503  << "). Use AVCDecoderConfigurationRecord.";
504  pixel_width = avc_config.pixel_width();
505  pixel_height = avc_config.pixel_height();
506  }
507  break;
508  }
509  case FOURCC_HEV1:
510  case FOURCC_HVC1: {
511  HEVCDecoderConfiguration hevc_config;
512  if (!hevc_config.Parse(entry.codec_config_record.data)) {
513  LOG(ERROR) << "Failed to parse hevc.";
514  return false;
515  }
516  codec_string = hevc_config.GetCodecString(video_codec);
517  nalu_length_size = hevc_config.length_size();
518  break;
519  }
520  case FOURCC_VP08:
521  case FOURCC_VP09:
522  case FOURCC_VP10: {
523  VPCodecConfiguration vp_config;
524  if (!vp_config.Parse(entry.codec_config_record.data)) {
525  LOG(ERROR) << "Failed to parse vpcc.";
526  return false;
527  }
528  codec_string = vp_config.GetCodecString(video_codec);
529  break;
530  }
531  default:
532  LOG(ERROR) << "Unsupported video format "
533  << FourCCToString(actual_format) << " in stsd box.";
534  return false;
535  }
536 
537  bool is_encrypted = entry.sinf.info.track_encryption.is_encrypted;
538  DVLOG(1) << "is_video_track_encrypted_: " << is_encrypted;
539  streams.push_back(new VideoStreamInfo(
540  track->header.track_id, timescale, duration, video_codec,
541  codec_string, track->media.header.language.code, coded_width,
542  coded_height, pixel_width, pixel_height,
543  0, // trick_play_rate
544  nalu_length_size, vector_as_array(&entry.codec_config_record.data),
545  entry.codec_config_record.data.size(), is_encrypted));
546  }
547  }
548 
549  init_cb_.Run(streams);
550  if (!FetchKeysIfNecessary(moov_->pssh))
551  return false;
552  runs_.reset(new TrackRunIterator(moov_.get()));
553  RCHECK(runs_->Init());
554  ChangeState(kEmittingSamples);
555  return true;
556 }
557 
558 bool MP4MediaParser::ParseMoof(BoxReader* reader) {
559  // Must already have initialization segment.
560  RCHECK(moov_.get());
561  MovieFragment moof;
562  RCHECK(moof.Parse(reader));
563  if (!runs_)
564  runs_.reset(new TrackRunIterator(moov_.get()));
565  RCHECK(runs_->Init(moof));
566  if (!FetchKeysIfNecessary(moof.pssh))
567  return false;
568  ChangeState(kEmittingSamples);
569  return true;
570 }
571 
572 bool MP4MediaParser::FetchKeysIfNecessary(
573  const std::vector<ProtectionSystemSpecificHeader>& headers) {
574  if (headers.empty())
575  return true;
576 
577  // An error will be returned later if the samples need to be decrypted.
578  if (!decryption_key_source_)
579  return true;
580 
581  // TODO(tinskip): Pass in raw 'pssh' boxes to FetchKeys. This will allow
582  // supporting multiple keysystems. Move this to KeySource.
583  std::vector<uint8_t> widevine_system_id;
584  base::HexStringToBytes(kWidevineKeySystemId, &widevine_system_id);
585  for (std::vector<ProtectionSystemSpecificHeader>::const_iterator iter =
586  headers.begin(); iter != headers.end(); ++iter) {
587  if (iter->system_id == widevine_system_id) {
588  Status status = decryption_key_source_->FetchKeys(iter->data);
589  if (!status.ok()) {
590  LOG(ERROR) << "Error fetching decryption keys: " << status;
591  return false;
592  }
593  return true;
594  }
595  }
596 
597  LOG(ERROR) << "No viable 'pssh' box found for content decryption.";
598  return false;
599 }
600 
601 bool MP4MediaParser::EnqueueSample(bool* err) {
602  if (!runs_->IsRunValid()) {
603  // Remain in kEnqueueingSamples state, discarding data, until the end of
604  // the current 'mdat' box has been appended to the queue.
605  if (!queue_.Trim(mdat_tail_))
606  return false;
607 
608  ChangeState(kParsingBoxes);
609  return true;
610  }
611 
612  if (!runs_->IsSampleValid()) {
613  runs_->AdvanceRun();
614  return true;
615  }
616 
617  DCHECK(!(*err));
618 
619  const uint8_t* buf;
620  int buf_size;
621  queue_.Peek(&buf, &buf_size);
622  if (!buf_size)
623  return false;
624 
625  // Skip this entire track if it is not audio nor video.
626  if (!runs_->is_audio() && !runs_->is_video())
627  runs_->AdvanceRun();
628 
629  // Attempt to cache the auxiliary information first. Aux info is usually
630  // placed in a contiguous block before the sample data, rather than being
631  // interleaved. If we didn't cache it, this would require that we retain the
632  // start of the segment buffer while reading samples. Aux info is typically
633  // quite small compared to sample data, so this pattern is useful on
634  // memory-constrained devices where the source buffer consumes a substantial
635  // portion of the total system memory.
636  if (runs_->AuxInfoNeedsToBeCached()) {
637  queue_.PeekAt(runs_->aux_info_offset() + moof_head_, &buf, &buf_size);
638  if (buf_size < runs_->aux_info_size())
639  return false;
640  *err = !runs_->CacheAuxInfo(buf, buf_size);
641  return !*err;
642  }
643 
644  int64_t sample_offset = runs_->sample_offset() + moof_head_;
645  queue_.PeekAt(sample_offset, &buf, &buf_size);
646  if (buf_size < runs_->sample_size()) {
647  if (sample_offset < queue_.head()) {
648  LOG(ERROR) << "Incorrect sample offset " << sample_offset
649  << " < " << queue_.head();
650  *err = true;
651  }
652  return false;
653  }
654 
655  scoped_refptr<MediaSample> stream_sample(MediaSample::CopyFrom(
656  buf, runs_->sample_size(), runs_->is_keyframe()));
657  if (runs_->is_encrypted()) {
658  if (!decryptor_source_) {
659  *err = true;
660  LOG(ERROR) << "Encrypted media sample encountered, but decryption is not "
661  "enabled";
662  return false;
663  }
664 
665  scoped_ptr<DecryptConfig> decrypt_config = runs_->GetDecryptConfig();
666  if (!decrypt_config ||
667  !decryptor_source_->DecryptSampleBuffer(decrypt_config.get(),
668  stream_sample->writable_data(),
669  stream_sample->data_size())) {
670  *err = true;
671  LOG(ERROR) << "Cannot decrypt samples.";
672  return false;
673  }
674  }
675 
676  stream_sample->set_dts(runs_->dts());
677  stream_sample->set_pts(runs_->cts());
678  stream_sample->set_duration(runs_->duration());
679 
680  DVLOG(3) << "Pushing frame: "
681  << ", key=" << runs_->is_keyframe()
682  << ", dur=" << runs_->duration()
683  << ", dts=" << runs_->dts()
684  << ", cts=" << runs_->cts()
685  << ", size=" << runs_->sample_size();
686 
687  if (!new_sample_cb_.Run(runs_->track_id(), stream_sample)) {
688  *err = true;
689  LOG(ERROR) << "Failed to process the sample.";
690  return false;
691  }
692 
693  runs_->AdvanceSample();
694  return true;
695 }
696 
697 bool MP4MediaParser::ReadAndDiscardMDATsUntil(const int64_t offset) {
698  bool err = false;
699  while (mdat_tail_ < offset) {
700  const uint8_t* buf;
701  int size;
702  queue_.PeekAt(mdat_tail_, &buf, &size);
703 
704  FourCC type;
705  uint64_t box_sz;
706  if (!BoxReader::StartTopLevelBox(buf, size, &type, &box_sz, &err))
707  break;
708 
709  mdat_tail_ += box_sz;
710  }
711  queue_.Trim(std::min(mdat_tail_, offset));
712  return !err;
713 }
714 
715 void MP4MediaParser::ChangeState(State new_state) {
716  DVLOG(2) << "Changing state: " << new_state;
717  state_ = new_state;
718 }
719 
720 } // namespace mp4
721 } // namespace media
722 } // namespace edash_packager
static BoxReader * ReadTopLevelBox(const uint8_t *buf, const size_t buf_size, bool *err)
Definition: box_reader.cc:37
DecryptorSource wraps KeySource and is responsible for decryptor management.
static scoped_refptr< MediaSample > CopyFrom(const uint8_t *data, size_t size, bool is_key_frame)
Definition: media_sample.cc:45
void PeekAt(int64_t offset, const uint8_t **buf, int *size)
bool LoadMoov(const std::string &file_path)
virtual Status FetchKeys(const std::vector< uint8_t > &content_id, const std::string &policy)
Definition: key_source.cc:30
KeySource is responsible for encryption key acquisition.
Definition: key_source.h:29
static bool StartTopLevelBox(const uint8_t *buf, const size_t buf_size, FourCC *type, uint64_t *box_size, bool *err) WARN_UNUSED_RESULT
Definition: box_reader.cc:60
bool Flush() override WARN_UNUSED_RESULT
void Init(const InitCB &init_cb, const NewSampleCB &new_sample_cb, KeySource *decryption_key_source) override
bool Parse(const uint8_t *buf, int size) override WARN_UNUSED_RESULT
static File * OpenWithNoBuffering(const char *file_name, const char *mode)
Definition: file.cc:151
static std::string GetCodecString(AudioCodec codec, uint8_t audio_object_type)