OLD | NEW |
1 /* | 1 /* |
2 * Copyright (c) 2015 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2015 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 #include "webrtc/modules/pacing/packet_router.h" | 11 #include "webrtc/modules/pacing/packet_router.h" |
12 | 12 |
13 #include "webrtc/base/atomicops.h" | 13 #include "webrtc/base/atomicops.h" |
14 #include "webrtc/base/checks.h" | 14 #include "webrtc/base/checks.h" |
15 #include "webrtc/modules/rtp_rtcp/include/rtp_rtcp.h" | 15 #include "webrtc/modules/rtp_rtcp/include/rtp_rtcp.h" |
16 #include "webrtc/modules/rtp_rtcp/include/rtp_rtcp_defines.h" | 16 #include "webrtc/modules/rtp_rtcp/include/rtp_rtcp_defines.h" |
17 #include "webrtc/modules/rtp_rtcp/source/rtcp_packet/transport_feedback.h" | 17 #include "webrtc/modules/rtp_rtcp/source/rtcp_packet/transport_feedback.h" |
18 | 18 |
19 namespace webrtc { | 19 namespace webrtc { |
20 | 20 |
21 namespace { | |
22 void AddModule(RtpRtcp* rtp_module, std::list<RtpRtcp*>* rtp_modules) { | |
23 RTC_DCHECK(std::find(rtp_modules->begin(), rtp_modules->end(), rtp_module) == | |
24 rtp_modules->end()); | |
25 rtp_modules->push_back(rtp_module); | |
26 } | |
27 | |
28 void RemoveModule(RtpRtcp* rtp_module, std::list<RtpRtcp*>* rtp_modules) { | |
29 RTC_DCHECK(std::find(rtp_modules->begin(), rtp_modules->end(), rtp_module) != | |
30 rtp_modules->end()); | |
31 rtp_modules->remove(rtp_module); | |
32 } | |
33 | |
34 bool SendFeedback(rtcp::TransportFeedback* packet, | |
35 std::list<RtpRtcp*>* rtp_modules) { | |
36 for (auto* rtp_module : *rtp_modules) { | |
37 packet->WithPacketSenderSsrc(rtp_module->SSRC()); | |
38 if (rtp_module->SendFeedbackPacket(*packet)) | |
39 return true; | |
40 } | |
41 return false; | |
42 } | |
43 } // namespace | |
44 | |
45 PacketRouter::PacketRouter() : transport_seq_(0) { | 21 PacketRouter::PacketRouter() : transport_seq_(0) { |
46 pacer_thread_checker_.DetachFromThread(); | 22 pacer_thread_checker_.DetachFromThread(); |
47 } | 23 } |
48 | 24 |
49 PacketRouter::~PacketRouter() { | 25 PacketRouter::~PacketRouter() { |
50 RTC_DCHECK(send_rtp_modules_.empty()); | 26 RTC_DCHECK(rtp_modules_.empty()); |
51 RTC_DCHECK(recv_rtp_modules_.empty()); | |
52 } | 27 } |
53 | 28 |
54 void PacketRouter::AddRtpModule(RtpRtcp* rtp_module, bool sender) { | 29 void PacketRouter::AddRtpModule(RtpRtcp* rtp_module) { |
55 rtc::CritScope cs(&modules_crit_); | 30 rtc::CritScope cs(&modules_crit_); |
56 if (sender) { | 31 RTC_DCHECK(std::find(rtp_modules_.begin(), rtp_modules_.end(), rtp_module) == |
57 AddModule(rtp_module, &send_rtp_modules_); | 32 rtp_modules_.end()); |
58 } else { | 33 rtp_modules_.push_back(rtp_module); |
59 AddModule(rtp_module, &recv_rtp_modules_); | |
60 } | |
61 } | 34 } |
62 | 35 |
63 void PacketRouter::RemoveRtpModule(RtpRtcp* rtp_module, bool sender) { | 36 void PacketRouter::RemoveRtpModule(RtpRtcp* rtp_module) { |
64 rtc::CritScope cs(&modules_crit_); | 37 rtc::CritScope cs(&modules_crit_); |
65 if (sender) { | 38 RTC_DCHECK(std::find(rtp_modules_.begin(), rtp_modules_.end(), rtp_module) != |
66 RemoveModule(rtp_module, &send_rtp_modules_); | 39 rtp_modules_.end()); |
67 } else { | 40 rtp_modules_.remove(rtp_module); |
68 RemoveModule(rtp_module, &recv_rtp_modules_); | |
69 } | |
70 } | 41 } |
71 | 42 |
72 bool PacketRouter::TimeToSendPacket(uint32_t ssrc, | 43 bool PacketRouter::TimeToSendPacket(uint32_t ssrc, |
73 uint16_t sequence_number, | 44 uint16_t sequence_number, |
74 int64_t capture_timestamp, | 45 int64_t capture_timestamp, |
75 bool retransmission) { | 46 bool retransmission) { |
76 RTC_DCHECK(pacer_thread_checker_.CalledOnValidThread()); | 47 RTC_DCHECK(pacer_thread_checker_.CalledOnValidThread()); |
77 rtc::CritScope cs(&modules_crit_); | 48 rtc::CritScope cs(&modules_crit_); |
78 for (auto* rtp_module : send_rtp_modules_) { | 49 for (auto* rtp_module : rtp_modules_) { |
79 if (rtp_module->SendingMedia() && ssrc == rtp_module->SSRC()) { | 50 if (rtp_module->SendingMedia() && ssrc == rtp_module->SSRC()) { |
80 return rtp_module->TimeToSendPacket(ssrc, sequence_number, | 51 return rtp_module->TimeToSendPacket(ssrc, sequence_number, |
81 capture_timestamp, retransmission); | 52 capture_timestamp, retransmission); |
82 } | 53 } |
83 } | 54 } |
84 return true; | 55 return true; |
85 } | 56 } |
86 | 57 |
87 size_t PacketRouter::TimeToSendPadding(size_t bytes_to_send) { | 58 size_t PacketRouter::TimeToSendPadding(size_t bytes_to_send) { |
88 RTC_DCHECK(pacer_thread_checker_.CalledOnValidThread()); | 59 RTC_DCHECK(pacer_thread_checker_.CalledOnValidThread()); |
89 size_t total_bytes_sent = 0; | 60 size_t total_bytes_sent = 0; |
90 rtc::CritScope cs(&modules_crit_); | 61 rtc::CritScope cs(&modules_crit_); |
91 for (RtpRtcp* module : send_rtp_modules_) { | 62 for (RtpRtcp* module : rtp_modules_) { |
92 if (module->SendingMedia()) { | 63 if (module->SendingMedia()) { |
93 size_t bytes_sent = | 64 size_t bytes_sent = |
94 module->TimeToSendPadding(bytes_to_send - total_bytes_sent); | 65 module->TimeToSendPadding(bytes_to_send - total_bytes_sent); |
95 total_bytes_sent += bytes_sent; | 66 total_bytes_sent += bytes_sent; |
96 if (total_bytes_sent >= bytes_to_send) | 67 if (total_bytes_sent >= bytes_to_send) |
97 break; | 68 break; |
98 } | 69 } |
99 } | 70 } |
100 return total_bytes_sent; | 71 return total_bytes_sent; |
101 } | 72 } |
(...skipping 16 matching lines...) Expand all Loading... |
118 prev_seq = rtc::AtomicOps::CompareAndSwap(&transport_seq_, desired_prev_seq, | 89 prev_seq = rtc::AtomicOps::CompareAndSwap(&transport_seq_, desired_prev_seq, |
119 new_seq); | 90 new_seq); |
120 } while (prev_seq != desired_prev_seq); | 91 } while (prev_seq != desired_prev_seq); |
121 | 92 |
122 return new_seq; | 93 return new_seq; |
123 } | 94 } |
124 | 95 |
125 bool PacketRouter::SendFeedback(rtcp::TransportFeedback* packet) { | 96 bool PacketRouter::SendFeedback(rtcp::TransportFeedback* packet) { |
126 RTC_DCHECK(pacer_thread_checker_.CalledOnValidThread()); | 97 RTC_DCHECK(pacer_thread_checker_.CalledOnValidThread()); |
127 rtc::CritScope cs(&modules_crit_); | 98 rtc::CritScope cs(&modules_crit_); |
128 if (::webrtc::SendFeedback(packet, &recv_rtp_modules_)) | 99 for (auto* rtp_module : rtp_modules_) { |
129 return true; | 100 packet->WithPacketSenderSsrc(rtp_module->SSRC()); |
130 if (::webrtc::SendFeedback(packet, &send_rtp_modules_)) | 101 if (rtp_module->SendFeedbackPacket(*packet)) |
131 return true; | 102 return true; |
| 103 } |
132 return false; | 104 return false; |
133 } | 105 } |
134 | 106 |
135 } // namespace webrtc | 107 } // namespace webrtc |
OLD | NEW |