OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "media/formats/mp2t/es_parser_h264.h" | 5 #include "media/formats/mp2t/es_parser_h264.h" |
6 | 6 |
7 #include <limits> | 7 #include <limits> |
8 | 8 |
9 #include "base/logging.h" | 9 #include "base/logging.h" |
10 #include "base/numerics/safe_conversions.h" | 10 #include "base/numerics/safe_conversions.h" |
(...skipping 418 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
429 | 429 |
430 // Emit a frame. | 430 // Emit a frame. |
431 DVLOG(LOG_LEVEL_ES) << "Emit frame: stream_pos=" << current_access_unit_pos_ | 431 DVLOG(LOG_LEVEL_ES) << "Emit frame: stream_pos=" << current_access_unit_pos_ |
432 << " size=" << access_unit_size; | 432 << " size=" << access_unit_size; |
433 int es_size; | 433 int es_size; |
434 const uint8_t* es; | 434 const uint8_t* es; |
435 es_queue_->PeekAt(current_access_unit_pos_, &es, &es_size); | 435 es_queue_->PeekAt(current_access_unit_pos_, &es, &es_size); |
436 CHECK_GE(es_size, access_unit_size); | 436 CHECK_GE(es_size, access_unit_size); |
437 | 437 |
438 #if BUILDFLAG(ENABLE_HLS_SAMPLE_AES) | 438 #if BUILDFLAG(ENABLE_HLS_SAMPLE_AES) |
| 439 const DecryptConfig* base_decrypt_config = nullptr; |
| 440 if (use_hls_sample_aes_) { |
| 441 DCHECK(!get_decrypt_config_cb_.is_null()); |
| 442 base_decrypt_config = get_decrypt_config_cb_.Run(); |
| 443 } |
| 444 |
439 std::unique_ptr<uint8_t[]> adjusted_au; | 445 std::unique_ptr<uint8_t[]> adjusted_au; |
440 std::vector<SubsampleEntry> subsamples; | 446 std::vector<SubsampleEntry> subsamples; |
441 if (use_hls_sample_aes_) { | 447 if (use_hls_sample_aes_ && base_decrypt_config) { |
442 adjusted_au = AdjustAUForSampleAES(es, &access_unit_size, protected_blocks_, | 448 adjusted_au = AdjustAUForSampleAES(es, &access_unit_size, protected_blocks_, |
443 &subsamples); | 449 &subsamples); |
444 protected_blocks_.clear(); | 450 protected_blocks_.clear(); |
445 if (adjusted_au) | 451 if (adjusted_au) |
446 es = adjusted_au.get(); | 452 es = adjusted_au.get(); |
447 } | 453 } |
448 #endif | 454 #endif |
449 | 455 |
450 // TODO(wolenetz/acolwell): Validate and use a common cross-parser TrackId | 456 // TODO(wolenetz/acolwell): Validate and use a common cross-parser TrackId |
451 // type and allow multiple video tracks. See https://crbug.com/341581. | 457 // type and allow multiple video tracks. See https://crbug.com/341581. |
452 scoped_refptr<StreamParserBuffer> stream_parser_buffer = | 458 scoped_refptr<StreamParserBuffer> stream_parser_buffer = |
453 StreamParserBuffer::CopyFrom(es, access_unit_size, is_key_frame, | 459 StreamParserBuffer::CopyFrom(es, access_unit_size, is_key_frame, |
454 DemuxerStream::VIDEO, kMp2tVideoTrackId); | 460 DemuxerStream::VIDEO, kMp2tVideoTrackId); |
455 stream_parser_buffer->SetDecodeTimestamp(current_timing_desc.dts); | 461 stream_parser_buffer->SetDecodeTimestamp(current_timing_desc.dts); |
456 stream_parser_buffer->set_timestamp(current_timing_desc.pts); | 462 stream_parser_buffer->set_timestamp(current_timing_desc.pts); |
457 #if BUILDFLAG(ENABLE_HLS_SAMPLE_AES) | 463 #if BUILDFLAG(ENABLE_HLS_SAMPLE_AES) |
458 if (use_hls_sample_aes_) { | 464 if (use_hls_sample_aes_ && base_decrypt_config) { |
459 DCHECK(!get_decrypt_config_cb_.is_null()); | |
460 const DecryptConfig* base_decrypt_config = get_decrypt_config_cb_.Run(); | |
461 RCHECK(base_decrypt_config); | |
462 std::unique_ptr<DecryptConfig> decrypt_config(new DecryptConfig( | 465 std::unique_ptr<DecryptConfig> decrypt_config(new DecryptConfig( |
463 base_decrypt_config->key_id(), base_decrypt_config->iv(), subsamples)); | 466 base_decrypt_config->key_id(), base_decrypt_config->iv(), subsamples)); |
464 stream_parser_buffer->set_decrypt_config(std::move(decrypt_config)); | 467 stream_parser_buffer->set_decrypt_config(std::move(decrypt_config)); |
465 } | 468 } |
466 #endif | 469 #endif |
467 return es_adapter_.OnNewBuffer(stream_parser_buffer); | 470 return es_adapter_.OnNewBuffer(stream_parser_buffer); |
468 } | 471 } |
469 | 472 |
470 bool EsParserH264::UpdateVideoDecoderConfig(const H264SPS* sps, | 473 bool EsParserH264::UpdateVideoDecoderConfig(const H264SPS* sps, |
471 const EncryptionScheme& scheme) { | 474 const EncryptionScheme& scheme) { |
(...skipping 43 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
515 << " height=" << sps->sar_height; | 518 << " height=" << sps->sar_height; |
516 last_video_decoder_config_ = video_decoder_config; | 519 last_video_decoder_config_ = video_decoder_config; |
517 es_adapter_.OnConfigChanged(video_decoder_config); | 520 es_adapter_.OnConfigChanged(video_decoder_config); |
518 } | 521 } |
519 | 522 |
520 return true; | 523 return true; |
521 } | 524 } |
522 | 525 |
523 } // namespace mp2t | 526 } // namespace mp2t |
524 } // namespace media | 527 } // namespace media |
OLD | NEW |