OLD | NEW |
1 /* | 1 /* |
2 * Copyright (c) 2017 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2017 The WebRTC project authors. All Rights Reserved. |
3 * | 3 * |
4 * Use of this source code is governed by a BSD-style license | 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 | 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 | 6 * tree. An additional intellectual property rights grant can be found |
7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
9 */ | 9 */ |
10 | 10 |
11 #ifndef WEBRTC_CALL_RTP_DEMUXER_H_ | 11 #ifndef WEBRTC_CALL_RTP_DEMUXER_H_ |
12 #define WEBRTC_CALL_RTP_DEMUXER_H_ | 12 #define WEBRTC_CALL_RTP_DEMUXER_H_ |
13 | 13 |
14 #include <map> | 14 #include <map> |
15 #include <set> | 15 #include <set> |
16 #include <string> | 16 #include <string> |
17 #include <vector> | |
18 | 17 |
19 namespace webrtc { | 18 namespace webrtc { |
20 | 19 |
21 class RsidResolutionObserver; | |
22 class RtpPacketReceived; | 20 class RtpPacketReceived; |
23 class RtpPacketSinkInterface; | 21 class RtpPacketSinkInterface; |
24 | 22 |
25 // This class represents the RTP demuxing, for a single RTP session (i.e., one | 23 // This class represents the RTP demuxing, for a single RTP session (i.e., one |
26 // ssrc space, see RFC 7656). It isn't thread aware, leaving responsibility of | 24 // ssrc space, see RFC 7656). It isn't thread aware, leaving responsibility of |
27 // multithreading issues to the user of this class. | 25 // multithreading issues to the user of this class. |
28 // TODO(nisse): Should be extended to also do MID-based demux and payload-type | 26 // TODO(nisse): Should be extended to also do MID-based demux and payload-type |
29 // demux. | 27 // demux. |
30 class RtpDemuxer { | 28 class RtpDemuxer { |
31 public: | 29 public: |
32 RtpDemuxer(); | 30 RtpDemuxer(); |
33 ~RtpDemuxer(); | 31 ~RtpDemuxer(); |
34 | 32 |
35 // Registers a sink. The same sink can be registered for multiple ssrcs, and | 33 // Registers a sink. The same sink can be registered for multiple ssrcs, and |
36 // the same ssrc can have multiple sinks. Null pointer is not allowed. | 34 // the same ssrc can have multiple sinks. Null pointer is not allowed. |
37 void AddSink(uint32_t ssrc, RtpPacketSinkInterface* sink); | 35 void AddSink(uint32_t ssrc, RtpPacketSinkInterface* sink); |
38 | 36 |
39 // Registers a sink's association to an RSID. Null pointer is not allowed. | 37 // Registers a sink's association to an RSID. Null pointer is not allowed. |
40 void AddSink(const std::string& rsid, RtpPacketSinkInterface* sink); | 38 void AddSink(const std::string& rsid, RtpPacketSinkInterface* sink); |
41 | 39 |
42 // Removes a sink. Return value reports if anything was actually removed. | 40 // Removes a sink. Return value reports if anything was actually removed. |
43 // Null pointer is not allowed. | 41 // Null pointer is not allowed. |
44 bool RemoveSink(const RtpPacketSinkInterface* sink); | 42 bool RemoveSink(const RtpPacketSinkInterface* sink); |
45 | 43 |
46 // Returns true if at least one matching sink was found. | 44 // Returns true if at least one matching sink was found, otherwise false. |
47 bool OnRtpPacket(const RtpPacketReceived& packet); | 45 bool OnRtpPacket(const RtpPacketReceived& packet); |
48 | 46 |
49 // Allows other objects to be notified when RSID-SSRC associations are | |
50 // resolved by this object. | |
51 void RegisterRsidResolutionObserver(RsidResolutionObserver* observer); | |
52 | |
53 // Undo a previous RegisterRsidResolutionObserver(). | |
54 void DeregisterRsidResolutionObserver(const RsidResolutionObserver* observer); | |
55 | |
56 private: | 47 private: |
57 // Records a sink<->SSRC association. This can happen by explicit | 48 // Records a sink<->SSRC association. This can happen by explicit |
58 // configuration by AddSink(ssrc...), or by inferred configuration from an | 49 // configuration by AddSink(ssrc...), or by inferred configuration from an |
59 // RSID-based configuration which is resolved to an SSRC upon | 50 // RSID-based configuration which is resolved to an SSRC upon |
60 // packet reception. | 51 // packet reception. |
61 void RecordSsrcToSinkAssociation(uint32_t ssrc, RtpPacketSinkInterface* sink); | 52 void RecordSsrcToSinkAssociation(uint32_t ssrc, RtpPacketSinkInterface* sink); |
62 | 53 |
63 // When a new packet arrives, we attempt to resolve extra associations. | 54 // When a new packet arrives, we attempt to resolve extra associations, |
64 void ResolveAssociations(const RtpPacketReceived& packet); | 55 // such as which RSIDs are associated with which SSRCs. |
65 | 56 void FindSsrcAssociations(const RtpPacketReceived& packet); |
66 // Find the associations of RSID to SSRCs. | |
67 void ResolveRsidToSsrcAssociations(const RtpPacketReceived& packet); | |
68 | |
69 // Notify observers of the resolution of an RSID to an SSRC. | |
70 void NotifyObserversOfRsidResolution(const std::string& rsid, uint32_t ssrc); | |
71 | 57 |
72 // This records the association SSRCs to sinks. Other associations, such | 58 // This records the association SSRCs to sinks. Other associations, such |
73 // as by RSID, also end up here once the RSID, etc., is resolved to an SSRC. | 59 // as by RSID, also end up here once the RSID, etc., is resolved to an SSRC. |
74 std::multimap<uint32_t, RtpPacketSinkInterface*> sinks_; | 60 std::multimap<uint32_t, RtpPacketSinkInterface*> sinks_; |
75 | 61 |
76 // A sink may be associated with an RSID - RTP Stream ID. This tag has a | 62 // A sink may be associated with an RSID - RTP Stream ID. This tag has a |
77 // one-to-one association with an SSRC, but that SSRC is not yet known. | 63 // one-to-one association with an SSRC, but that SSRC is not yet known. |
78 // When it becomes known, the association of the sink to the RSID is deleted | 64 // When it becomes known, the association of the sink to the RSID is deleted |
79 // from this container, and moved into |sinks_|. | 65 // from this container, and moved into |sinks_|. |
80 std::multimap<std::string, RtpPacketSinkInterface*> rsid_sinks_; | 66 std::multimap<std::string, RtpPacketSinkInterface*> rsid_sinks_; |
81 | 67 |
82 // Iterating over |rsid_sinks_| for each incoming and performing multiple | 68 // Iterating over |rsid_sinks_| for each incoming and performing multiple |
83 // string comparisons is of non-trivial cost. To avoid this cost, we only | 69 // string comparisons is of non-trivial cost. To avoid this cost, we only |
84 // check RSIDs for the first packet on each incoming SSRC stream. | 70 // check RSIDs for the first packet on each incoming SSRC stream. |
85 // (If RSID associations are added later, we check again.) | 71 // (If RSID associations are added later, we check again.) |
86 std::set<uint32_t> processed_ssrcs_; | 72 std::set<uint32_t> processed_ssrcs_; |
87 | 73 |
88 // Avoid an attack that would create excessive logging. | 74 // Avoid an attack that would create excessive logging. |
89 bool logged_max_processed_ssrcs_exceeded_ = false; | 75 bool logged_max_processed_ssrcs_exceeded_ = false; |
90 | |
91 // Observers which will be notified when an RSID association to an SSRC is | |
92 // resolved by this object. | |
93 std::vector<RsidResolutionObserver*> rsid_resolution_observers_; | |
94 }; | 76 }; |
95 | 77 |
96 } // namespace webrtc | 78 } // namespace webrtc |
97 | 79 |
98 #endif // WEBRTC_CALL_RTP_DEMUXER_H_ | 80 #endif // WEBRTC_CALL_RTP_DEMUXER_H_ |
OLD | NEW |