2a4d70cc93
Add a method to DtlsSrtpTransport to cache the RTP Absolute Send Time extension id. The method would be called when using DTLS-SRTP with external authentication. TBR=pthatcher@webrtc.org Bug: webrtc:7013 Change-Id: Ie9edb9382cbb4cf43eea5da3030991a0d20293a5 Reviewed-on: https://webrtc-review.googlesource.com/27260 Commit-Queue: Zhi Huang <zhihuang@webrtc.org> Reviewed-by: Steve Anton <steveanton@webrtc.org> Cr-Commit-Position: refs/heads/master@{#20947}
106 lines
3.9 KiB
C++
106 lines
3.9 KiB
C++
/*
|
|
* 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 UpdateSendEncryptedHeaderExtensionIds(
|
|
const std::vector<int>& send_extension_ids);
|
|
|
|
void UpdateRecvEncryptedHeaderExtensionIds(
|
|
const std::vector<int>& recv_extension_ids);
|
|
|
|
bool IsActive() { return srtp_transport_->IsActive(); }
|
|
|
|
// Cache RTP Absoulute SendTime extension header ID. This is only used when
|
|
// external authentication is enabled.
|
|
void CacheRtpAbsSendTimeHeaderExtension(int rtp_abs_sendtime_extn_id) {
|
|
srtp_transport_->CacheRtpAbsSendTimeHeaderExtension(
|
|
rtp_abs_sendtime_extn_id);
|
|
}
|
|
|
|
// 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(bool writable);
|
|
void OnSentPacket(const rtc::SentPacket& sent_packet);
|
|
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_
|