Create the SrtpTransportInterface.

webrtc/ortc/rtptransportadapter.h

 /*
  *  Copyright 2017 The WebRTC project authors. All Rights Reserved.
  *
  *  Use of this source code is governed by a BSD-style license
  *  that can be found in the LICENSE file in the root of the source
  *  tree. An additional intellectual property rights grant can be found
  *  in the file PATENTS.  All contributing project authors may
  *  be found in the AUTHORS file in the root of the source tree.
  */
 
 #ifndef WEBRTC_ORTC_RTPTRANSPORTADAPTER_H_
 #define WEBRTC_ORTC_RTPTRANSPORTADAPTER_H_
 
 #include <memory>
 #include <vector>
 
-#include "webrtc/api/ortc/rtptransportinterface.h"
+#include "webrtc/api/ortc/srtptransportinterface.h"
 #include "webrtc/api/rtcerror.h"
 #include "webrtc/base/constructormagic.h"
 #include "webrtc/base/sigslot.h"
 #include "webrtc/media/base/streamparams.h"
 #include "webrtc/ortc/rtptransportcontrolleradapter.h"
 #include "webrtc/pc/channel.h"
 
 namespace webrtc {
 
-// Implementation of RtpTransportInterface to be used with RtpSenderAdapter,
-// RtpReceiverAdapter, and RtpTransportControllerAdapter classes.
+// Implementation of SrtpTransportInterface to be used with RtpSenderAdapter,
+// RtpReceiverAdapter, and RtpTransportControllerAdapter classes. This class
+// is used to implement both a secure and insecure RTP transport.
 //
 // TODO(deadbeef): When BaseChannel is split apart into separate
 // "RtpTransport"/"RtpTransceiver"/"RtpSender"/"RtpReceiver" objects, this
 // adapter object can be removed.
-class RtpTransportAdapter : public RtpTransportInterface {
+class RtpTransportAdapter : public SrtpTransportInterface {
  public:
   // |rtp| can't be null. |rtcp| can if RTCP muxing is used immediately (meaning
   // |rtcp_parameters.mux| is also true).
   static RTCErrorOr<std::unique_ptr<RtpTransportInterface>> CreateProxied(
       const RtcpParameters& rtcp_parameters,
       PacketTransportInterface* rtp,
       PacketTransportInterface* rtcp,
       RtpTransportControllerAdapter* rtp_transport_controller);
+
+  static RTCErrorOr<std::unique_ptr<SrtpTransportInterface>> CreateSrtpProxied(
+      const RtcpParameters& rtcp_parameters,
+      PacketTransportInterface* rtp,
+      PacketTransportInterface* rtcp,
+      RtpTransportControllerAdapter* rtp_transport_controller);
+
   ~RtpTransportAdapter() override;
 
   // RtpTransportInterface implementation.
   PacketTransportInterface* GetRtpPacketTransport() const override;
   PacketTransportInterface* GetRtcpPacketTransport() const override;
   RTCError SetRtcpParameters(const RtcpParameters& parameters) override;
   RtcpParameters GetRtcpParameters() const override { return rtcp_parameters_; }
 
+  // SRTP specific implementation.
+  RTCError SetSrtpSendKey(const cricket::CryptoParams& params) override;
+  RTCError SetSrtpReceiveKey(const cricket::CryptoParams& params) override;
+
   // Methods used internally by OrtcFactory.
   RtpTransportControllerAdapter* rtp_transport_controller() {
     return rtp_transport_controller_;
   }
   void TakeOwnershipOfRtpTransportController(
       std::unique_ptr<RtpTransportControllerInterface> controller);
 
   // Used by RtpTransportControllerAdapter to tell when it should stop
   // returning this transport from GetTransports().
   sigslot::signal1<RtpTransportAdapter*> SignalDestroyed;
 
+  // Used by the RtpTransportControllerAdapter to tell if an rtp sender or
+  // receiver should be created.

[Taylor Brandstetter, 2017/03/01 19:18:08]

nit: "can be created" instead of "should be"?

[Zhi Huang, 2017/03/02 23:35:39]

Done.

+  bool is_srtp_transport() { return is_srtp_transport_; }
+  bool key_set() { return have_send_key_ && have_receive_key_; }

[Taylor Brandstetter, 2017/03/01 19:18:08]

nit: "keys_set"?

[Zhi Huang, 2017/03/02 23:35:39]

Done.

+  // Used by the RtpTransportControllerAdapter to set keys for senders and
+  // receivers.
+  cricket::CryptoParams send_key() { return send_key_; }
+  cricket::CryptoParams receive_key() { return receive_key_; }
+
  protected:
   RtpTransportAdapter* GetInternal() override { return this; }
 
  private:
   RtpTransportAdapter(const RtcpParameters& rtcp_parameters,
                       PacketTransportInterface* rtp,
                       PacketTransportInterface* rtcp,
-                      RtpTransportControllerAdapter* rtp_transport_controller);
+                      RtpTransportControllerAdapter* rtp_transport_controller,
+                      bool is_srtp_transport);
 
   PacketTransportInterface* rtp_packet_transport_;
   PacketTransportInterface* rtcp_packet_transport_;
   RtpTransportControllerAdapter* rtp_transport_controller_;
   // Non-null if this class owns the transport controller.
   std::unique_ptr<RtpTransportControllerInterface>
       owned_rtp_transport_controller_;
   RtcpParameters rtcp_parameters_;
 
+  // SRTP specific members.
+  cricket::CryptoParams send_key_;
+  cricket::CryptoParams receive_key_;
+  bool is_srtp_transport_;
+  bool have_send_key_;
+  bool have_receive_key_;

[Taylor Brandstetter, 2017/03/01 19:18:08]

nit: have_send_key_ and have_receive_key_ can default to false here

[Zhi Huang, 2017/03/02 23:35:39]

Done.

+
   RTC_DISALLOW_IMPLICIT_CONSTRUCTORS(RtpTransportAdapter);
 };
 
 }  // namespace webrtc
 
 #endif  // WEBRTC_ORTC_RTPTRANSPORTADAPTER_H_