Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(166)

Unified Diff: webrtc/call/rtp_demuxer.cc

Issue 2957763002: Revert of Create RtcpDemuxer (Closed)
Patch Set: Created 3 years, 6 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « webrtc/call/rtp_demuxer.h ('k') | webrtc/call/rtp_demuxer_unittest.cc » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: webrtc/call/rtp_demuxer.cc
diff --git a/webrtc/call/rtp_demuxer.cc b/webrtc/call/rtp_demuxer.cc
index e1957ed554baf6955848cd869637b1276eb57d32..620d4b17a928ffe5f7c7bd9dbd1748a96c005c3e 100644
--- a/webrtc/call/rtp_demuxer.cc
+++ b/webrtc/call/rtp_demuxer.cc
@@ -12,27 +12,50 @@
#include "webrtc/base/checks.h"
#include "webrtc/base/logging.h"
-#include "webrtc/call/rsid_resolution_observer.h"
#include "webrtc/call/rtp_packet_sink_interface.h"
-#include "webrtc/call/rtp_rtcp_demuxer_helper.h"
#include "webrtc/modules/rtp_rtcp/source/rtp_header_extensions.h"
#include "webrtc/modules/rtp_rtcp/source/rtp_packet_received.h"
namespace webrtc {
namespace {
+
constexpr size_t kMaxProcessedSsrcs = 1000; // Prevent memory overuse.
+
+template <typename Key, typename Value>
+bool MultimapAssociationExists(const std::multimap<Key, Value>& multimap,
+ Key key,
+ Value val) {
+ auto it_range = multimap.equal_range(key);
+ using Reference = typename std::multimap<Key, Value>::const_reference;
+ return std::any_of(it_range.first, it_range.second,
+ [val](Reference elem) { return elem.second == val; });
+}
+
+template <typename Key, typename Value>
+size_t RemoveFromMultimapByValue(std::multimap<Key, Value*>* multimap,
+ const Value* value) {
+ size_t count = 0;
+ for (auto it = multimap->begin(); it != multimap->end();) {
+ if (it->second == value) {
+ it = multimap->erase(it);
+ ++count;
+ } else {
+ ++it;
+ }
+ }
+ return count;
+}
+
} // namespace
-RtpDemuxer::RtpDemuxer() = default;
+RtpDemuxer::RtpDemuxer() {}
RtpDemuxer::~RtpDemuxer() {
RTC_DCHECK(sinks_.empty());
- RTC_DCHECK(rsid_sinks_.empty());
}
void RtpDemuxer::AddSink(uint32_t ssrc, RtpPacketSinkInterface* sink) {
- RTC_DCHECK(sink);
RecordSsrcToSinkAssociation(ssrc, sink);
}
@@ -65,7 +88,7 @@
}
bool RtpDemuxer::OnRtpPacket(const RtpPacketReceived& packet) {
- ResolveAssociations(packet);
+ FindSsrcAssociations(packet);
auto it_range = sinks_.equal_range(packet.Ssrc());
for (auto it = it_range.first; it != it_range.second; ++it) {
it->second->OnRtpPacket(packet);
@@ -73,33 +96,27 @@
return it_range.first != it_range.second;
}
-void RtpDemuxer::RegisterRsidResolutionObserver(
- RsidResolutionObserver* observer) {
- RTC_DCHECK(observer);
- RTC_DCHECK(!ContainerHasKey(rsid_resolution_observers_, observer));
-
- rsid_resolution_observers_.push_back(observer);
-
- processed_ssrcs_.clear(); // New observer requires new notifications.
-}
-
-void RtpDemuxer::DeregisterRsidResolutionObserver(
- const RsidResolutionObserver* observer) {
- RTC_DCHECK(observer);
- auto it = std::find(rsid_resolution_observers_.begin(),
- rsid_resolution_observers_.end(), observer);
- RTC_DCHECK(it != rsid_resolution_observers_.end());
- rsid_resolution_observers_.erase(it);
-}
-
-void RtpDemuxer::ResolveAssociations(const RtpPacketReceived& packet) {
+void RtpDemuxer::FindSsrcAssociations(const RtpPacketReceived& packet) {
// Avoid expensive string comparisons for RSID by looking the sinks up only
// by SSRC whenever possible.
if (processed_ssrcs_.find(packet.Ssrc()) != processed_ssrcs_.cend()) {
return;
}
- ResolveRsidToSsrcAssociations(packet);
+ // RSID-based associations:
+ std::string rsid;
+ if (packet.GetExtension<RtpStreamId>(&rsid)) {
+ // All streams associated with this RSID need to be marked as associated
+ // with this SSRC (if they aren't already).
+ auto it_range = rsid_sinks_.equal_range(rsid);
+ for (auto it = it_range.first; it != it_range.second; ++it) {
+ RecordSsrcToSinkAssociation(packet.Ssrc(), it->second);
+ }
+
+ // To prevent memory-overuse attacks, forget this RSID. Future packets
+ // with this RSID, but a different SSRC, will not spawn new associations.
+ rsid_sinks_.erase(it_range.first, it_range.second);
+ }
if (processed_ssrcs_.size() < kMaxProcessedSsrcs) { // Prevent memory overuse
processed_ssrcs_.insert(packet.Ssrc()); // Avoid re-examining in-depth.
@@ -110,30 +127,4 @@
}
}
-void RtpDemuxer::ResolveRsidToSsrcAssociations(
- const RtpPacketReceived& packet) {
- std::string rsid;
- if (packet.GetExtension<RtpStreamId>(&rsid)) {
- // All streams associated with this RSID need to be marked as associated
- // with this SSRC (if they aren't already).
- auto it_range = rsid_sinks_.equal_range(rsid);
- for (auto it = it_range.first; it != it_range.second; ++it) {
- RecordSsrcToSinkAssociation(packet.Ssrc(), it->second);
- }
-
- NotifyObserversOfRsidResolution(rsid, packet.Ssrc());
-
- // To prevent memory-overuse attacks, forget this RSID. Future packets
- // with this RSID, but a different SSRC, will not spawn new associations.
- rsid_sinks_.erase(it_range.first, it_range.second);
- }
-}
-
-void RtpDemuxer::NotifyObserversOfRsidResolution(const std::string& rsid,
- uint32_t ssrc) {
- for (auto* observer : rsid_resolution_observers_) {
- observer->OnRsidResolved(rsid, ssrc);
- }
-}
-
} // namespace webrtc
« no previous file with comments | « webrtc/call/rtp_demuxer.h ('k') | webrtc/call/rtp_demuxer_unittest.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698