Created the DtlsSrtpTransport.
The DtlsSrtpTransport is designed to take DTLS responsibilities from BaseChannel.
DtlsSrtpTransport is responsible for exporting keys from DtlsTransport
and setting up the wrapped SrtpTransport.
The DtlsSrtpTransport is not hooked up to BaseChannel yet in this CL.
Bug: webrtc:7013
Change-Id: I318c00dadf9b1e033ec842de6e1536e9227ab713
Reviewed-on: https://webrtc-review.googlesource.com/6700
Commit-Queue: Zhi Huang <zhihuang@webrtc.org>
Reviewed-by: Peter Thatcher <pthatcher@webrtc.org>
Cr-Commit-Position: refs/heads/master@{#20804}diff --git a/pc/dtlssrtptransport.h b/pc/dtlssrtptransport.h
new file mode 100644
index 0000000..43f072a
--- /dev/null
+++ b/pc/dtlssrtptransport.h
@@ -0,0 +1,97 @@
+/*
+ * 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 PC_DTLSSRTPTRANSPORT_H_
+#define PC_DTLSSRTPTRANSPORT_H_
+
+#include <memory>
+#include <string>
+#include <vector>
+
+#include "p2p/base/dtlstransportinternal.h"
+#include "pc/rtptransportinternaladapter.h"
+#include "pc/srtptransport.h"
+
+namespace webrtc {
+
+// This class is intended to be used as an RtpTransport and it wraps both an
+// SrtpTransport and DtlsTransports(RTP/RTCP). When the DTLS handshake is
+// finished, it extracts the keying materials from DtlsTransport and sets them
+// to SrtpTransport.
+class DtlsSrtpTransport : public RtpTransportInternalAdapter {
+ public:
+ explicit DtlsSrtpTransport(
+ std::unique_ptr<webrtc::SrtpTransport> srtp_transport);
+
+ // Set P2P layer RTP/RTCP DtlsTransports. When using RTCP-muxing,
+ // |rtcp_dtls_transport| is null.
+ void SetDtlsTransports(cricket::DtlsTransportInternal* rtp_dtls_transport,
+ cricket::DtlsTransportInternal* rtcp_dtls_transport);
+
+ void SetRtcpMuxEnabled(bool enable) override;
+
+ // Set the header extension ids that should be encrypted.
+ void SetSendEncryptedHeaderExtensionIds(
+ const std::vector<int>& send_extension_ids);
+
+ void SetRecvEncryptedHeaderExtensionIds(
+ const std::vector<int>& recv_extension_ids);
+
+ bool IsActive() { return srtp_transport_->IsActive(); }
+
+ // TODO(zhihuang): Remove this when we remove RtpTransportAdapter.
+ RtpTransportAdapter* GetInternal() override { return nullptr; }
+
+ sigslot::signal2<DtlsSrtpTransport*, bool> SignalDtlsSrtpSetupFailure;
+
+ private:
+ bool IsDtlsActive();
+ bool IsDtlsConnected();
+ bool IsDtlsWritable();
+ bool DtlsHandshakeCompleted();
+ void MaybeSetupDtlsSrtp();
+ void SetupRtpDtlsSrtp();
+ void SetupRtcpDtlsSrtp();
+ bool ExtractParams(cricket::DtlsTransportInternal* dtls_transport,
+ int* selected_crypto_suite,
+ std::vector<unsigned char>* send_key,
+ std::vector<unsigned char>* recv_key);
+ void SetDtlsTransport(cricket::DtlsTransportInternal* new_dtls_transport,
+ cricket::DtlsTransportInternal** old_dtls_transport);
+ void SetRtpDtlsTransport(cricket::DtlsTransportInternal* rtp_dtls_transport);
+ void SetRtcpDtlsTransport(
+ cricket::DtlsTransportInternal* rtcp_dtls_transport);
+ void UpdateWritableStateAndMaybeSetupDtlsSrtp();
+ // Set the writability and fire the SignalWritableState if the writability
+ // changes.
+ void SetWritable(bool writable);
+
+ void OnDtlsState(cricket::DtlsTransportInternal* dtls_transport,
+ cricket::DtlsTransportState state);
+ void OnWritableState(rtc::PacketTransportInternal* transport);
+ void OnPacketReceived(bool rtcp,
+ rtc::CopyOnWriteBuffer* packet,
+ const rtc::PacketTime& packet_time);
+ void OnReadyToSend(bool ready);
+
+ bool writable_ = false;
+ std::unique_ptr<SrtpTransport> srtp_transport_;
+ // Owned by the TransportController.
+ cricket::DtlsTransportInternal* rtp_dtls_transport_ = nullptr;
+ cricket::DtlsTransportInternal* rtcp_dtls_transport_ = nullptr;
+
+ // The encrypted header extension IDs.
+ rtc::Optional<std::vector<int>> send_extension_ids_;
+ rtc::Optional<std::vector<int>> recv_extension_ids_;
+};
+
+} // namespace webrtc
+
+#endif // PC_DTLSSRTPTRANSPORT_H_
