| 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 "chromecast/media/cma/ipc_streamer/decoder_buffer_base_marshaller.h" | 5 #include "chromecast/media/cma/ipc_streamer/decoder_buffer_base_marshaller.h" |
| 6 | 6 |
| 7 #include "base/logging.h" | 7 #include "base/logging.h" |
| 8 #include "chromecast/media/cma/base/decoder_buffer_base.h" | 8 #include "chromecast/media/cma/base/decoder_buffer_base.h" |
| 9 #include "chromecast/media/cma/base/decrypt_config_impl.h" |
| 9 #include "chromecast/media/cma/ipc/media_message.h" | 10 #include "chromecast/media/cma/ipc/media_message.h" |
| 10 #include "chromecast/media/cma/ipc/media_message_type.h" | 11 #include "chromecast/media/cma/ipc/media_message_type.h" |
| 11 #include "chromecast/media/cma/ipc_streamer/decrypt_config_marshaller.h" | 12 #include "chromecast/media/cma/ipc_streamer/decrypt_config_marshaller.h" |
| 13 #include "chromecast/public/media/decrypt_config.h" |
| 12 #include "media/base/decrypt_config.h" | 14 #include "media/base/decrypt_config.h" |
| 13 | 15 |
| 14 namespace chromecast { | 16 namespace chromecast { |
| 15 namespace media { | 17 namespace media { |
| 16 | 18 |
| 17 namespace { | 19 namespace { |
| 18 const size_t kMaxFrameSize = 4 * 1024 * 1024; | 20 const size_t kMaxFrameSize = 4 * 1024 * 1024; |
| 19 | 21 |
| 20 class DecoderBufferFromMsg : public DecoderBufferBase { | 22 class DecoderBufferFromMsg : public DecoderBufferBase { |
| 21 public: | 23 public: |
| 22 explicit DecoderBufferFromMsg(scoped_ptr<MediaMessage> msg); | 24 explicit DecoderBufferFromMsg(scoped_ptr<MediaMessage> msg); |
| 23 | 25 |
| 24 void Initialize(); | 26 void Initialize(); |
| 25 | 27 |
| 26 // DecoderBufferBase implementation. | 28 // DecoderBufferBase implementation. |
| 27 StreamId stream_id() const override; | 29 StreamId stream_id() const override; |
| 28 base::TimeDelta timestamp() const override; | 30 TimeDelta timestamp() const override; |
| 29 void set_timestamp(const base::TimeDelta& timestamp) override; | 31 void set_timestamp(TimeDelta timestamp) override; |
| 30 const uint8* data() const override; | 32 const uint8* data() const override; |
| 31 uint8* writable_data() const override; | 33 uint8* writable_data() const override; |
| 32 size_t data_size() const override; | 34 size_t data_size() const override; |
| 33 const ::media::DecryptConfig* decrypt_config() const override; | 35 const DecryptConfig* decrypt_config() const override; |
| 34 bool end_of_stream() const override; | 36 bool end_of_stream() const override; |
| 35 | 37 |
| 36 private: | 38 private: |
| 37 ~DecoderBufferFromMsg() override; | 39 ~DecoderBufferFromMsg() override; |
| 38 | 40 |
| 39 // Indicates whether this is an end of stream frame. | 41 // Indicates whether this is an end of stream frame. |
| 40 bool is_eos_; | 42 bool is_eos_; |
| 41 | 43 |
| 42 // Stream Id this decoder buffer belongs to. | 44 // Stream Id this decoder buffer belongs to. |
| 43 StreamId stream_id_; | 45 StreamId stream_id_; |
| 44 | 46 |
| 45 // Frame timestamp. | 47 // Frame timestamp. |
| 46 base::TimeDelta pts_; | 48 TimeDelta pts_; |
| 47 | 49 |
| 48 // CENC parameters. | 50 // CENC parameters. |
| 49 scoped_ptr< ::media::DecryptConfig> decrypt_config_; | 51 scoped_ptr<DecryptConfig> decrypt_config_; |
| 50 | 52 |
| 51 // Size of the frame. | 53 // Size of the frame. |
| 52 size_t data_size_; | 54 size_t data_size_; |
| 53 | 55 |
| 54 // Keeps the message since frame data is not copied. | 56 // Keeps the message since frame data is not copied. |
| 55 scoped_ptr<MediaMessage> msg_; | 57 scoped_ptr<MediaMessage> msg_; |
| 56 uint8* data_; | 58 uint8* data_; |
| 57 | 59 |
| 58 DISALLOW_COPY_AND_ASSIGN(DecoderBufferFromMsg); | 60 DISALLOW_COPY_AND_ASSIGN(DecoderBufferFromMsg); |
| 59 }; | 61 }; |
| (...skipping 14 matching lines...) Expand all Loading... |
| 74 CHECK_EQ(msg_->type(), FrameMediaMsg); | 76 CHECK_EQ(msg_->type(), FrameMediaMsg); |
| 75 | 77 |
| 76 CHECK(msg_->ReadPod(&is_eos_)); | 78 CHECK(msg_->ReadPod(&is_eos_)); |
| 77 if (is_eos_) | 79 if (is_eos_) |
| 78 return; | 80 return; |
| 79 | 81 |
| 80 CHECK(msg_->ReadPod(&stream_id_)); | 82 CHECK(msg_->ReadPod(&stream_id_)); |
| 81 | 83 |
| 82 int64 pts_internal = 0; | 84 int64 pts_internal = 0; |
| 83 CHECK(msg_->ReadPod(&pts_internal)); | 85 CHECK(msg_->ReadPod(&pts_internal)); |
| 84 pts_ = base::TimeDelta::FromInternalValue(pts_internal); | 86 pts_ = TimeDelta::FromMicroseconds(pts_internal); |
| 85 | 87 |
| 86 bool has_decrypt_config = false; | 88 bool has_decrypt_config = false; |
| 87 CHECK(msg_->ReadPod(&has_decrypt_config)); | 89 CHECK(msg_->ReadPod(&has_decrypt_config)); |
| 88 if (has_decrypt_config) | 90 if (has_decrypt_config) |
| 89 decrypt_config_.reset(DecryptConfigMarshaller::Read(msg_.get()).release()); | 91 decrypt_config_.reset(DecryptConfigMarshaller::Read(msg_.get()).release()); |
| 90 | 92 |
| 91 CHECK(msg_->ReadPod(&data_size_)); | 93 CHECK(msg_->ReadPod(&data_size_)); |
| 92 CHECK_GT(data_size_, 0u); | 94 CHECK_GT(data_size_, 0u); |
| 93 CHECK_LT(data_size_, kMaxFrameSize); | 95 CHECK_LT(data_size_, kMaxFrameSize); |
| 94 | 96 |
| 95 // Get a pointer to the frame data inside the message. | 97 // Get a pointer to the frame data inside the message. |
| 96 // Avoid copying the frame data here. | 98 // Avoid copying the frame data here. |
| 97 data_ = static_cast<uint8*>(msg_->GetWritableBuffer(data_size_)); | 99 data_ = static_cast<uint8*>(msg_->GetWritableBuffer(data_size_)); |
| 98 CHECK(data_); | 100 CHECK(data_); |
| 99 | 101 |
| 100 if (decrypt_config_) { | 102 if (decrypt_config_) { |
| 101 uint32 subsample_total_size = 0; | 103 uint32 subsample_total_size = 0; |
| 102 for (size_t k = 0; k < decrypt_config_->subsamples().size(); k++) { | 104 for (size_t k = 0; k < decrypt_config_->subsamples().size(); k++) { |
| 103 subsample_total_size += decrypt_config_->subsamples()[k].clear_bytes; | 105 subsample_total_size += decrypt_config_->subsamples()[k].clear_bytes; |
| 104 subsample_total_size += decrypt_config_->subsamples()[k].cypher_bytes; | 106 subsample_total_size += decrypt_config_->subsamples()[k].cypher_bytes; |
| 105 } | 107 } |
| 106 CHECK_EQ(subsample_total_size, data_size_); | 108 CHECK_EQ(subsample_total_size, data_size_); |
| 107 } | 109 } |
| 108 } | 110 } |
| 109 | 111 |
| 110 StreamId DecoderBufferFromMsg::stream_id() const { | 112 StreamId DecoderBufferFromMsg::stream_id() const { |
| 111 return stream_id_; | 113 return stream_id_; |
| 112 } | 114 } |
| 113 | 115 |
| 114 base::TimeDelta DecoderBufferFromMsg::timestamp() const { | 116 TimeDelta DecoderBufferFromMsg::timestamp() const { |
| 115 return pts_; | 117 return pts_; |
| 116 } | 118 } |
| 117 | 119 |
| 118 void DecoderBufferFromMsg::set_timestamp(const base::TimeDelta& timestamp) { | 120 void DecoderBufferFromMsg::set_timestamp(TimeDelta timestamp) { |
| 119 pts_ = timestamp; | 121 pts_ = timestamp; |
| 120 } | 122 } |
| 121 | 123 |
| 122 const uint8* DecoderBufferFromMsg::data() const { | 124 const uint8* DecoderBufferFromMsg::data() const { |
| 123 CHECK(msg_->IsSerializedMsgAvailable()); | 125 CHECK(msg_->IsSerializedMsgAvailable()); |
| 124 return data_; | 126 return data_; |
| 125 } | 127 } |
| 126 | 128 |
| 127 uint8* DecoderBufferFromMsg::writable_data() const { | 129 uint8* DecoderBufferFromMsg::writable_data() const { |
| 128 CHECK(msg_->IsSerializedMsgAvailable()); | 130 CHECK(msg_->IsSerializedMsgAvailable()); |
| 129 return data_; | 131 return data_; |
| 130 } | 132 } |
| 131 | 133 |
| 132 size_t DecoderBufferFromMsg::data_size() const { | 134 size_t DecoderBufferFromMsg::data_size() const { |
| 133 return data_size_; | 135 return data_size_; |
| 134 } | 136 } |
| 135 | 137 |
| 136 const ::media::DecryptConfig* DecoderBufferFromMsg::decrypt_config() const { | 138 const DecryptConfig* DecoderBufferFromMsg::decrypt_config() const { |
| 137 return decrypt_config_.get(); | 139 return decrypt_config_.get(); |
| 138 } | 140 } |
| 139 | 141 |
| 140 bool DecoderBufferFromMsg::end_of_stream() const { | 142 bool DecoderBufferFromMsg::end_of_stream() const { |
| 141 return is_eos_; | 143 return is_eos_; |
| 142 } | 144 } |
| 143 | 145 |
| 144 } // namespace | 146 } // namespace |
| 145 | 147 |
| 146 // static | 148 // static |
| 147 void DecoderBufferBaseMarshaller::Write( | 149 void DecoderBufferBaseMarshaller::Write( |
| 148 const scoped_refptr<DecoderBufferBase>& buffer, | 150 const scoped_refptr<DecoderBufferBase>& buffer, |
| 149 MediaMessage* msg) { | 151 MediaMessage* msg) { |
| 150 CHECK(msg->WritePod(buffer->end_of_stream())); | 152 CHECK(msg->WritePod(buffer->end_of_stream())); |
| 151 if (buffer->end_of_stream()) | 153 if (buffer->end_of_stream()) |
| 152 return; | 154 return; |
| 153 | 155 |
| 154 CHECK(msg->WritePod(buffer->stream_id())); | 156 CHECK(msg->WritePod(buffer->stream_id())); |
| 155 CHECK(msg->WritePod(buffer->timestamp().ToInternalValue())); | 157 CHECK(msg->WritePod(buffer->timestamp().InMicroseconds())); |
| 156 | 158 |
| 157 bool has_decrypt_config = | 159 bool has_decrypt_config = |
| 158 (buffer->decrypt_config() != NULL && | 160 (buffer->decrypt_config() != NULL && |
| 159 buffer->decrypt_config()->iv().size() > 0); | 161 buffer->decrypt_config()->iv().size() > 0); |
| 160 CHECK(msg->WritePod(has_decrypt_config)); | 162 CHECK(msg->WritePod(has_decrypt_config)); |
| 161 | 163 |
| 162 if (has_decrypt_config) { | 164 if (has_decrypt_config) { |
| 163 // DecryptConfig may contain 0 subsamples if all content is encrypted. | 165 // DecryptConfig may contain 0 subsamples if all content is encrypted. |
| 164 // Map this case to a single fully-encrypted "subsample" for more consistent | 166 // Map this case to a single fully-encrypted "subsample" for more consistent |
| 165 // backend handling. | 167 // backend handling. |
| 166 if (buffer->decrypt_config()->subsamples().empty()) { | 168 if (buffer->decrypt_config()->subsamples().empty()) { |
| 167 std::vector< ::media::SubsampleEntry> encrypted_subsample_list(1); | 169 std::vector<SubsampleEntry> encrypted_subsample_list(1); |
| 168 encrypted_subsample_list[0].clear_bytes = 0; | 170 encrypted_subsample_list[0].clear_bytes = 0; |
| 169 encrypted_subsample_list[0].cypher_bytes = buffer->data_size(); | 171 encrypted_subsample_list[0].cypher_bytes = buffer->data_size(); |
| 170 ::media::DecryptConfig full_sample_config( | 172 DecryptConfigImpl full_sample_config(buffer->decrypt_config()->key_id(), |
| 171 buffer->decrypt_config()->key_id(), | 173 buffer->decrypt_config()->iv(), |
| 172 buffer->decrypt_config()->iv(), | 174 encrypted_subsample_list); |
| 173 encrypted_subsample_list); | |
| 174 DecryptConfigMarshaller::Write(full_sample_config, msg); | 175 DecryptConfigMarshaller::Write(full_sample_config, msg); |
| 175 } else { | 176 } else { |
| 176 DecryptConfigMarshaller::Write(*buffer->decrypt_config(), msg); | 177 DecryptConfigMarshaller::Write(*buffer->decrypt_config(), msg); |
| 177 } | 178 } |
| 178 } | 179 } |
| 179 | 180 |
| 180 CHECK(msg->WritePod(buffer->data_size())); | 181 CHECK(msg->WritePod(buffer->data_size())); |
| 181 CHECK(msg->WriteBuffer(buffer->data(), buffer->data_size())); | 182 CHECK(msg->WriteBuffer(buffer->data(), buffer->data_size())); |
| 182 } | 183 } |
| 183 | 184 |
| 184 // static | 185 // static |
| 185 scoped_refptr<DecoderBufferBase> DecoderBufferBaseMarshaller::Read( | 186 scoped_refptr<DecoderBufferBase> DecoderBufferBaseMarshaller::Read( |
| 186 scoped_ptr<MediaMessage> msg) { | 187 scoped_ptr<MediaMessage> msg) { |
| 187 scoped_refptr<DecoderBufferFromMsg> buffer( | 188 scoped_refptr<DecoderBufferFromMsg> buffer( |
| 188 new DecoderBufferFromMsg(msg.Pass())); | 189 new DecoderBufferFromMsg(msg.Pass())); |
| 189 buffer->Initialize(); | 190 buffer->Initialize(); |
| 190 return buffer; | 191 return buffer; |
| 191 } | 192 } |
| 192 | 193 |
| 193 } // namespace media | 194 } // namespace media |
| 194 } // namespace chromecast | 195 } // namespace chromecast |
| OLD | NEW |