| OLD | NEW |
| (Empty) |
| 1 /* | |
| 2 * Copyright (c) 2017 The WebRTC project authors. All Rights Reserved. | |
| 3 * | |
| 4 * Use of this source code is governed by a BSD-style license | |
| 5 * that can be found in the LICENSE file in the root of the source | |
| 6 * tree. An additional intellectual property rights grant can be found | |
| 7 * in the file PATENTS. All contributing project authors may | |
| 8 * be found in the AUTHORS file in the root of the source tree. | |
| 9 */ | |
| 10 | |
| 11 #include "webrtc/call/rtcp_demuxer.h" | |
| 12 | |
| 13 #include "webrtc/base/checks.h" | |
| 14 #include "webrtc/call/rtcp_packet_sink_interface.h" | |
| 15 #include "webrtc/call/rtp_rtcp_demuxer_helper.h" | |
| 16 #include "webrtc/common_types.h" | |
| 17 | |
| 18 namespace webrtc { | |
| 19 | |
| 20 RtcpDemuxer::RtcpDemuxer() = default; | |
| 21 | |
| 22 RtcpDemuxer::~RtcpDemuxer() { | |
| 23 RTC_DCHECK(ssrc_sinks_.empty()); | |
| 24 RTC_DCHECK(rsid_sinks_.empty()); | |
| 25 RTC_DCHECK(broadcast_sinks_.empty()); | |
| 26 } | |
| 27 | |
| 28 void RtcpDemuxer::AddSink(uint32_t sender_ssrc, RtcpPacketSinkInterface* sink) { | |
| 29 RTC_DCHECK(sink); | |
| 30 RTC_DCHECK(!ContainerHasKey(broadcast_sinks_, sink)); | |
| 31 RTC_DCHECK(!MultimapAssociationExists(ssrc_sinks_, sender_ssrc, sink)); | |
| 32 ssrc_sinks_.emplace(sender_ssrc, sink); | |
| 33 } | |
| 34 | |
| 35 void RtcpDemuxer::AddSink(const std::string& rsid, | |
| 36 RtcpPacketSinkInterface* sink) { | |
| 37 RTC_DCHECK(StreamId::IsLegalName(rsid)); | |
| 38 RTC_DCHECK(sink); | |
| 39 RTC_DCHECK(!ContainerHasKey(broadcast_sinks_, sink)); | |
| 40 RTC_DCHECK(!MultimapAssociationExists(rsid_sinks_, rsid, sink)); | |
| 41 rsid_sinks_.emplace(rsid, sink); | |
| 42 } | |
| 43 | |
| 44 void RtcpDemuxer::AddBroadcastSink(RtcpPacketSinkInterface* sink) { | |
| 45 RTC_DCHECK(sink); | |
| 46 RTC_DCHECK(!MultimapHasValue(ssrc_sinks_, sink)); | |
| 47 RTC_DCHECK(!MultimapHasValue(rsid_sinks_, sink)); | |
| 48 RTC_DCHECK(!ContainerHasKey(broadcast_sinks_, sink)); | |
| 49 broadcast_sinks_.push_back(sink); | |
| 50 } | |
| 51 | |
| 52 void RtcpDemuxer::RemoveSink(const RtcpPacketSinkInterface* sink) { | |
| 53 RTC_DCHECK(sink); | |
| 54 size_t removal_count = RemoveFromMultimapByValue(&ssrc_sinks_, sink) + | |
| 55 RemoveFromMultimapByValue(&rsid_sinks_, sink); | |
| 56 RTC_DCHECK_GT(removal_count, 0); | |
| 57 } | |
| 58 | |
| 59 void RtcpDemuxer::RemoveBroadcastSink(const RtcpPacketSinkInterface* sink) { | |
| 60 RTC_DCHECK(sink); | |
| 61 auto it = std::find(broadcast_sinks_.begin(), broadcast_sinks_.end(), sink); | |
| 62 RTC_DCHECK(it != broadcast_sinks_.end()); | |
| 63 broadcast_sinks_.erase(it); | |
| 64 } | |
| 65 | |
| 66 void RtcpDemuxer::OnRtcpPacket(rtc::ArrayView<const uint8_t> packet) { | |
| 67 // Perform sender-SSRC-based demuxing for packets with a sender-SSRC. | |
| 68 rtc::Optional<uint32_t> sender_ssrc = ParseRtcpPacketSenderSsrc(packet); | |
| 69 if (sender_ssrc) { | |
| 70 auto it_range = ssrc_sinks_.equal_range(*sender_ssrc); | |
| 71 for (auto it = it_range.first; it != it_range.second; ++it) { | |
| 72 it->second->OnRtcpPacket(packet); | |
| 73 } | |
| 74 } | |
| 75 | |
| 76 // All packets, even those without a sender-SSRC, are broadcast to sinks | |
| 77 // which listen to broadcasts. | |
| 78 for (RtcpPacketSinkInterface* sink : broadcast_sinks_) { | |
| 79 sink->OnRtcpPacket(packet); | |
| 80 } | |
| 81 } | |
| 82 | |
| 83 void RtcpDemuxer::OnRsidResolved(const std::string& rsid, uint32_t ssrc) { | |
| 84 // Record the new SSRC association for all of the sinks that were associated | |
| 85 // with the RSID. | |
| 86 auto it_range = rsid_sinks_.equal_range(rsid); | |
| 87 for (auto it = it_range.first; it != it_range.second; ++it) { | |
| 88 RtcpPacketSinkInterface* sink = it->second; | |
| 89 // Watch out for pre-existing SSRC-based associations. | |
| 90 if (!MultimapAssociationExists(ssrc_sinks_, ssrc, sink)) { | |
| 91 AddSink(ssrc, sink); | |
| 92 } | |
| 93 } | |
| 94 | |
| 95 // RSIDs are uniquely associated with SSRCs; no need to keep in memory | |
| 96 // the RSID-to-sink association of resolved RSIDs. | |
| 97 rsid_sinks_.erase(it_range.first, it_range.second); | |
| 98 } | |
| 99 | |
| 100 } // namespace webrtc | |
| OLD | NEW |