src/media/cast/transport/transport_video_sender.cc

   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
   3 // found in the LICENSE file.
   4
   5 #include "media/cast/transport/transport_video_sender.h"
   6
   7 #include <list>
   8
   9 #include "base/bind.h"
  10 #include "base/logging.h"
  11 #include "base/message_loop/message_loop.h"
  12 #include "media/base/video_frame.h"
  13 #include "media/cast/transport/pacing/paced_sender.h"
  14
  15 namespace media {
  16 namespace cast {
  17 namespace transport {
  18
  19 TransportVideoSender::TransportVideoSender(
  20     const CastTransportConfig& config,
  21     base::TickClock* clock,
  22     const scoped_refptr<base::TaskRunner>& transport_task_runner,
  23     PacedSender* const paced_packet_sender)
  24     : rtp_max_delay_(base::TimeDelta::FromMilliseconds(
  25           config.video_rtp_config.max_delay_ms)),
  26       encryptor_(),
  27       rtp_sender_(clock,
  28                   config,
  29                   false,
  30                   transport_task_runner,
  31                   paced_packet_sender) {
  32   initialized_ = encryptor_.Initialize(config.aes_key, config.aes_iv_mask);
  33 }
  34
  35 TransportVideoSender::~TransportVideoSender() {}
  36
  37 void TransportVideoSender::InsertCodedVideoFrame(
  38     const EncodedVideoFrame* coded_frame,
  39     const base::TimeTicks& capture_time) {
  40   if (encryptor_.initialized()) {
  41     EncodedVideoFrame encrypted_video_frame;
  42
  43     if (!EncryptVideoFrame(*coded_frame, &encrypted_video_frame))
  44       return;
  45
  46     rtp_sender_.IncomingEncodedVideoFrame(&encrypted_video_frame, capture_time);
  47   } else {
  48     rtp_sender_.IncomingEncodedVideoFrame(coded_frame, capture_time);
  49   }
  50   if (coded_frame->key_frame) {
  51     VLOG(1) << "Send encoded key frame; frame_id:"
  52             << static_cast<int>(coded_frame->frame_id);
  53   }
  54 }
  55
  56 bool TransportVideoSender::EncryptVideoFrame(
  57     const EncodedVideoFrame& video_frame,
  58     EncodedVideoFrame* encrypted_frame) {
  59   if (!encryptor_.Encrypt(
  60            video_frame.frame_id, video_frame.data, &(encrypted_frame->data)))
  61     return false;
  62
  63   encrypted_frame->codec = video_frame.codec;
  64   encrypted_frame->key_frame = video_frame.key_frame;
  65   encrypted_frame->frame_id = video_frame.frame_id;
  66   encrypted_frame->last_referenced_frame_id =
  67       video_frame.last_referenced_frame_id;
  68   return true;
  69 }
  70
  71 void TransportVideoSender::ResendPackets(
  72     const MissingFramesAndPacketsMap& missing_frames_and_packets) {
  73   rtp_sender_.ResendPackets(missing_frames_and_packets);
  74 }
  75
  76 void TransportVideoSender::SubscribeVideoRtpStatsCallback(
  77     const CastTransportRtpStatistics& callback) {
  78   rtp_sender_.SubscribeRtpStatsCallback(callback);
  79 }
  80
  81 }  // namespace transport
  82 }  // namespace cast
  83 }  // namespace media