1 files changed, 58 insertions, 21 deletions

diff --git a/media/cast/net/cast_transport_sender_impl.h b/media/cast/net/cast_transport_sender_impl.h
index 5d34fde..9cc3b46 100644
--- a/media/cast/net/cast_transport_sender_impl.h
+++ b/media/cast/net/cast_transport_sender_impl.h
@@ -2,12 +2,32 @@
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
+// This class maintains a send transport for audio and video in a Cast
+// Streaming session.
+// Audio, video frames and RTCP messages are submitted to this object
+// and then packetized and paced to the underlying UDP socket.
+//
+// The hierarchy of send transport in a Cast Streaming session:
+//
+// CastTransportSender              RTP                      RTCP
+// ------------------------------------------------------------------
+//                      TransportEncryptionHandler (A/V)
+//                      RtpSender (A/V)                   Rtcp (A/V)
+//                                      PacedSender (Shared)
+//                                      UdpTransport (Shared)
+//
+// There are objects of TransportEncryptionHandler, RtpSender and Rtcp
+// for each audio and video stream.
+// PacedSender and UdpTransport are shared between all RTP and RTCP
+// streams.
+
 #ifndef MEDIA_CAST_NET_CAST_TRANSPORT_IMPL_H_
 #define MEDIA_CAST_NET_CAST_TRANSPORT_IMPL_H_
 
 #include "base/callback.h"
 #include "base/memory/ref_counted.h"
 #include "base/memory/scoped_ptr.h"
+#include "base/memory/weak_ptr.h"
 #include "base/time/tick_clock.h"
 #include "base/time/time.h"
 #include "base/timer/timer.h"
@@ -17,7 +37,7 @@
 #include "media/cast/net/cast_transport_config.h"
 #include "media/cast/net/cast_transport_sender.h"
 #include "media/cast/net/pacing/paced_sender.h"
-#include "media/cast/net/rtcp/rtcp_builder.h"
+#include "media/cast/net/rtcp/rtcp.h"
 #include "media/cast/net/rtp/rtp_sender.h"
 
 namespace media {
@@ -27,9 +47,7 @@ class UdpTransport;
 
 class CastTransportSenderImpl : public CastTransportSender {
  public:
-  // external_transport is only used for testing.
-  // Note that SetPacketReceiver does not work if an external
-  // transport is provided.
+  // |external_transport| is only used for testing.
   // |raw_events_callback|: Raw events will be returned on this callback
   // which will be invoked every |raw_events_callback_interval|.
   // This can be a null callback, i.e. if user is not interested in raw events.
@@ -47,23 +65,19 @@ class CastTransportSenderImpl : public CastTransportSender {
 
   virtual ~CastTransportSenderImpl();
 
-  virtual void InitializeAudio(const CastTransportRtpConfig& config) OVERRIDE;
-  virtual void InitializeVideo(const CastTransportRtpConfig& config) OVERRIDE;
-
-  // CastTransportSender implementation.
-  virtual void SetPacketReceiver(const PacketReceiverCallback& packet_receiver)
-      OVERRIDE;
-
+  virtual void InitializeAudio(const CastTransportRtpConfig& config,
+                               const RtcpCastMessageCallback& cast_message_cb,
+                               const RtcpRttCallback& rtt_cb) OVERRIDE;
+  virtual void InitializeVideo(const CastTransportRtpConfig& config,
+                               const RtcpCastMessageCallback& cast_message_cb,
+                               const RtcpRttCallback& rtt_cb) OVERRIDE;
   virtual void InsertCodedAudioFrame(const EncodedFrame& audio_frame) OVERRIDE;
   virtual void InsertCodedVideoFrame(const EncodedFrame& video_frame) OVERRIDE;
 
-  virtual void SendRtcpFromRtpSender(uint32 packet_type_flags,
-                                     uint32 ntp_seconds,
-                                     uint32 ntp_fraction,
-                                     uint32 rtp_timestamp,
-                                     const RtcpDlrrReportBlock& dlrr,
-                                     uint32 sending_ssrc,
-                                     const std::string& c_name) OVERRIDE;
+  virtual void SendSenderReport(
+      uint32 ssrc,
+      base::TimeTicks current_time,
+      uint32 current_time_as_rtp_timestamp) OVERRIDE;
 
   virtual void ResendPackets(bool is_audio,
                              const MissingFramesAndPacketsMap& missing_packets,
@@ -71,22 +85,43 @@ class CastTransportSenderImpl : public CastTransportSender {
                              base::TimeDelta dedupe_window)
       OVERRIDE;
 
+  virtual PacketReceiverCallback PacketReceiverForTesting() OVERRIDE;
+
  private:
   // If |raw_events_callback_| is non-null, calls it with events collected
   // by |event_subscriber_| since last call.
   void SendRawEvents();
 
+  // Start receiving packets.
+  void StartReceiving();
+
+  // Called when a packet is received.
+  void OnReceivedPacket(scoped_ptr<Packet> packet);
+
+  // Called when a log message is received.
+  void OnReceivedLogMessage(EventMediaType media_type,
+                            const RtcpReceiverLogMessage& log);
+
   base::TickClock* clock_;  // Not owned by this class.
   CastTransportStatusCallback status_callback_;
   scoped_refptr<base::SingleThreadTaskRunner> transport_task_runner_;
 
-  scoped_ptr<UdpTransport> transport_;
   LoggingImpl logging_;
+
+  // Interface to a UDP socket.
+  scoped_ptr<UdpTransport> transport_;
+
+  // Packet sender that performs pacing.
   PacedSender pacer_;
-  RtcpBuilder rtcp_builder_;
+
+  // Packetizer for audio and video frames.
   scoped_ptr<RtpSender> audio_sender_;
   scoped_ptr<RtpSender> video_sender_;
 
+  // Maintains RTCP session for audio and video.
+  scoped_ptr<Rtcp> audio_rtcp_session_;
+  scoped_ptr<Rtcp> video_rtcp_session_;
+
   // Encrypts data in EncodedFrames before they are sent.  Note that it's
   // important for the encryption to happen here, in code that would execute in
   // the main browser process, for security reasons.  This helps to mitigate
@@ -96,9 +131,11 @@ class CastTransportSenderImpl : public CastTransportSender {
 
   // This is non-null iff |raw_events_callback_| is non-null.
   scoped_ptr<SimpleEventSubscriber> event_subscriber_;
-  base::RepeatingTimer<CastTransportSenderImpl> raw_events_timer_;
 
   BulkRawEventsCallback raw_events_callback_;
+  base::TimeDelta raw_events_callback_interval_;
+
+  base::WeakPtrFactory<CastTransportSenderImpl> weak_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(CastTransportSenderImpl);
 };