pbos@webrtc.org | 8ce445e | 2013-08-19 16:09:34 +0000 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (c) 2013 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 | #ifndef WEBRTC_VIDEO_ENGINE_TEST_COMMON_RTP_RTCP_OBSERVER_H_ |
| 11 | #define WEBRTC_VIDEO_ENGINE_TEST_COMMON_RTP_RTCP_OBSERVER_H_ |
| 12 | |
| 13 | #include <map> |
| 14 | #include <vector> |
| 15 | |
pbos@webrtc.org | c71929d | 2014-01-24 09:30:53 +0000 | [diff] [blame] | 16 | #include "testing/gtest/include/gtest/gtest.h" |
| 17 | |
stefan@webrtc.org | 4985c7b | 2013-11-15 12:32:15 +0000 | [diff] [blame] | 18 | #include "webrtc/modules/rtp_rtcp/interface/rtp_header_parser.h" |
pbos@webrtc.org | 8ce445e | 2013-08-19 16:09:34 +0000 | [diff] [blame] | 19 | #include "webrtc/typedefs.h" |
pbos@webrtc.org | 24e2089 | 2013-10-28 16:32:01 +0000 | [diff] [blame] | 20 | #include "webrtc/video_send_stream.h" |
pbos@webrtc.org | 8ce445e | 2013-08-19 16:09:34 +0000 | [diff] [blame] | 21 | |
| 22 | namespace webrtc { |
| 23 | namespace test { |
| 24 | |
| 25 | class RtpRtcpObserver { |
| 26 | public: |
pbos@webrtc.org | e1e050e | 2013-11-21 11:42:02 +0000 | [diff] [blame] | 27 | virtual ~RtpRtcpObserver() {} |
pbos@webrtc.org | 8ce445e | 2013-08-19 16:09:34 +0000 | [diff] [blame] | 28 | newapi::Transport* SendTransport() { |
| 29 | return &send_transport_; |
| 30 | } |
| 31 | |
| 32 | newapi::Transport* ReceiveTransport() { |
| 33 | return &receive_transport_; |
| 34 | } |
| 35 | |
pbos@webrtc.org | c179706 | 2013-08-23 09:19:30 +0000 | [diff] [blame] | 36 | void SetReceivers(PacketReceiver* send_transport_receiver, |
| 37 | PacketReceiver* receive_transport_receiver) { |
pbos@webrtc.org | 8ce445e | 2013-08-19 16:09:34 +0000 | [diff] [blame] | 38 | send_transport_.SetReceiver(send_transport_receiver); |
| 39 | receive_transport_.SetReceiver(receive_transport_receiver); |
| 40 | } |
| 41 | |
| 42 | void StopSending() { |
| 43 | send_transport_.StopSending(); |
| 44 | receive_transport_.StopSending(); |
| 45 | } |
| 46 | |
stefan@webrtc.org | 4985c7b | 2013-11-15 12:32:15 +0000 | [diff] [blame] | 47 | virtual EventTypeWrapper Wait() { |
pbos@webrtc.org | 66f4394 | 2013-11-28 11:59:31 +0000 | [diff] [blame] | 48 | EventTypeWrapper result = observation_complete_->Wait(timeout_ms_); |
| 49 | observation_complete_->Reset(); |
| 50 | return result; |
stefan@webrtc.org | 4985c7b | 2013-11-15 12:32:15 +0000 | [diff] [blame] | 51 | } |
pbos@webrtc.org | 28a1166 | 2013-09-19 14:22:12 +0000 | [diff] [blame] | 52 | |
pbos@webrtc.org | 8ce445e | 2013-08-19 16:09:34 +0000 | [diff] [blame] | 53 | protected: |
stefan@webrtc.org | aacdb9f | 2013-12-18 20:28:25 +0000 | [diff] [blame] | 54 | RtpRtcpObserver(unsigned int event_timeout_ms, |
| 55 | const FakeNetworkPipe::Config& configuration) |
pbos@webrtc.org | 8ce445e | 2013-08-19 16:09:34 +0000 | [diff] [blame] | 56 | : lock_(CriticalSectionWrapper::CreateCriticalSection()), |
pbos@webrtc.org | 28a1166 | 2013-09-19 14:22:12 +0000 | [diff] [blame] | 57 | observation_complete_(EventWrapper::Create()), |
stefan@webrtc.org | 4985c7b | 2013-11-15 12:32:15 +0000 | [diff] [blame] | 58 | parser_(RtpHeaderParser::Create()), |
pbos@webrtc.org | 8ce445e | 2013-08-19 16:09:34 +0000 | [diff] [blame] | 59 | send_transport_(lock_.get(), |
pbos@webrtc.org | 28a1166 | 2013-09-19 14:22:12 +0000 | [diff] [blame] | 60 | this, |
| 61 | &RtpRtcpObserver::OnSendRtp, |
stefan@webrtc.org | e028410 | 2013-11-18 11:45:11 +0000 | [diff] [blame] | 62 | &RtpRtcpObserver::OnSendRtcp, |
stefan@webrtc.org | aacdb9f | 2013-12-18 20:28:25 +0000 | [diff] [blame] | 63 | configuration), |
pbos@webrtc.org | 8ce445e | 2013-08-19 16:09:34 +0000 | [diff] [blame] | 64 | receive_transport_(lock_.get(), |
pbos@webrtc.org | 28a1166 | 2013-09-19 14:22:12 +0000 | [diff] [blame] | 65 | this, |
| 66 | &RtpRtcpObserver::OnReceiveRtp, |
stefan@webrtc.org | e028410 | 2013-11-18 11:45:11 +0000 | [diff] [blame] | 67 | &RtpRtcpObserver::OnReceiveRtcp, |
stefan@webrtc.org | aacdb9f | 2013-12-18 20:28:25 +0000 | [diff] [blame] | 68 | configuration), |
stefan@webrtc.org | e028410 | 2013-11-18 11:45:11 +0000 | [diff] [blame] | 69 | timeout_ms_(event_timeout_ms) {} |
| 70 | |
| 71 | explicit RtpRtcpObserver(unsigned int event_timeout_ms) |
| 72 | : lock_(CriticalSectionWrapper::CreateCriticalSection()), |
| 73 | observation_complete_(EventWrapper::Create()), |
stefan@webrtc.org | 09f84e5 | 2013-11-18 11:58:24 +0000 | [diff] [blame] | 74 | parser_(RtpHeaderParser::Create()), |
stefan@webrtc.org | e028410 | 2013-11-18 11:45:11 +0000 | [diff] [blame] | 75 | send_transport_(lock_.get(), |
| 76 | this, |
| 77 | &RtpRtcpObserver::OnSendRtp, |
| 78 | &RtpRtcpObserver::OnSendRtcp, |
stefan@webrtc.org | aacdb9f | 2013-12-18 20:28:25 +0000 | [diff] [blame] | 79 | FakeNetworkPipe::Config()), |
stefan@webrtc.org | e028410 | 2013-11-18 11:45:11 +0000 | [diff] [blame] | 80 | receive_transport_(lock_.get(), |
| 81 | this, |
| 82 | &RtpRtcpObserver::OnReceiveRtp, |
| 83 | &RtpRtcpObserver::OnReceiveRtcp, |
stefan@webrtc.org | aacdb9f | 2013-12-18 20:28:25 +0000 | [diff] [blame] | 84 | FakeNetworkPipe::Config()), |
pbos@webrtc.org | 28a1166 | 2013-09-19 14:22:12 +0000 | [diff] [blame] | 85 | timeout_ms_(event_timeout_ms) {} |
pbos@webrtc.org | 8ce445e | 2013-08-19 16:09:34 +0000 | [diff] [blame] | 86 | |
| 87 | enum Action { |
| 88 | SEND_PACKET, |
| 89 | DROP_PACKET, |
| 90 | }; |
| 91 | |
| 92 | virtual Action OnSendRtp(const uint8_t* packet, size_t length) { |
| 93 | return SEND_PACKET; |
| 94 | } |
| 95 | |
| 96 | virtual Action OnSendRtcp(const uint8_t* packet, size_t length) { |
| 97 | return SEND_PACKET; |
| 98 | } |
| 99 | |
| 100 | virtual Action OnReceiveRtp(const uint8_t* packet, size_t length) { |
| 101 | return SEND_PACKET; |
| 102 | } |
| 103 | |
| 104 | virtual Action OnReceiveRtcp(const uint8_t* packet, size_t length) { |
| 105 | return SEND_PACKET; |
| 106 | } |
| 107 | |
| 108 | |
| 109 | private: |
| 110 | class PacketTransport : public test::DirectTransport { |
| 111 | public: |
| 112 | typedef Action (RtpRtcpObserver::*PacketTransportAction)(const uint8_t*, |
| 113 | size_t); |
stefan@webrtc.org | e028410 | 2013-11-18 11:45:11 +0000 | [diff] [blame] | 114 | |
pbos@webrtc.org | 8ce445e | 2013-08-19 16:09:34 +0000 | [diff] [blame] | 115 | PacketTransport(CriticalSectionWrapper* lock, |
| 116 | RtpRtcpObserver* observer, |
| 117 | PacketTransportAction on_rtp, |
stefan@webrtc.org | e028410 | 2013-11-18 11:45:11 +0000 | [diff] [blame] | 118 | PacketTransportAction on_rtcp, |
stefan@webrtc.org | aacdb9f | 2013-12-18 20:28:25 +0000 | [diff] [blame] | 119 | const FakeNetworkPipe::Config& configuration) |
| 120 | : test::DirectTransport(configuration), |
stefan@webrtc.org | e028410 | 2013-11-18 11:45:11 +0000 | [diff] [blame] | 121 | lock_(lock), |
pbos@webrtc.org | 8ce445e | 2013-08-19 16:09:34 +0000 | [diff] [blame] | 122 | observer_(observer), |
| 123 | on_rtp_(on_rtp), |
| 124 | on_rtcp_(on_rtcp) {} |
| 125 | |
| 126 | private: |
pbos@webrtc.org | 3009c81 | 2013-11-20 12:17:04 +0000 | [diff] [blame] | 127 | virtual bool SendRtp(const uint8_t* packet, size_t length) OVERRIDE { |
pbos@webrtc.org | c71929d | 2014-01-24 09:30:53 +0000 | [diff] [blame] | 128 | EXPECT_FALSE(RtpHeaderParser::IsRtcp(packet, static_cast<int>(length))); |
pbos@webrtc.org | 8ce445e | 2013-08-19 16:09:34 +0000 | [diff] [blame] | 129 | Action action; |
| 130 | { |
| 131 | CriticalSectionScoped crit_(lock_); |
| 132 | action = (observer_->*on_rtp_)(packet, length); |
| 133 | } |
| 134 | switch (action) { |
| 135 | case DROP_PACKET: |
| 136 | // Drop packet silently. |
| 137 | return true; |
| 138 | case SEND_PACKET: |
pbos@webrtc.org | 3009c81 | 2013-11-20 12:17:04 +0000 | [diff] [blame] | 139 | return test::DirectTransport::SendRtp(packet, length); |
pbos@webrtc.org | 8ce445e | 2013-08-19 16:09:34 +0000 | [diff] [blame] | 140 | } |
pbos@webrtc.org | 96ff6ab | 2013-08-19 16:35:36 +0000 | [diff] [blame] | 141 | return true; // Will never happen, makes compiler happy. |
pbos@webrtc.org | 8ce445e | 2013-08-19 16:09:34 +0000 | [diff] [blame] | 142 | } |
pbos@webrtc.org | 96ff6ab | 2013-08-19 16:35:36 +0000 | [diff] [blame] | 143 | |
pbos@webrtc.org | 3009c81 | 2013-11-20 12:17:04 +0000 | [diff] [blame] | 144 | virtual bool SendRtcp(const uint8_t* packet, size_t length) OVERRIDE { |
pbos@webrtc.org | c71929d | 2014-01-24 09:30:53 +0000 | [diff] [blame] | 145 | EXPECT_TRUE(RtpHeaderParser::IsRtcp(packet, static_cast<int>(length))); |
pbos@webrtc.org | 8ce445e | 2013-08-19 16:09:34 +0000 | [diff] [blame] | 146 | Action action; |
| 147 | { |
| 148 | CriticalSectionScoped crit_(lock_); |
| 149 | action = (observer_->*on_rtcp_)(packet, length); |
| 150 | } |
| 151 | switch (action) { |
| 152 | case DROP_PACKET: |
| 153 | // Drop packet silently. |
| 154 | return true; |
| 155 | case SEND_PACKET: |
pbos@webrtc.org | 3009c81 | 2013-11-20 12:17:04 +0000 | [diff] [blame] | 156 | return test::DirectTransport::SendRtcp(packet, length); |
pbos@webrtc.org | 8ce445e | 2013-08-19 16:09:34 +0000 | [diff] [blame] | 157 | } |
pbos@webrtc.org | 96ff6ab | 2013-08-19 16:35:36 +0000 | [diff] [blame] | 158 | return true; // Will never happen, makes compiler happy. |
pbos@webrtc.org | 8ce445e | 2013-08-19 16:09:34 +0000 | [diff] [blame] | 159 | } |
| 160 | |
| 161 | // Pointer to shared lock instance protecting on_rtp_/on_rtcp_ calls. |
| 162 | CriticalSectionWrapper* lock_; |
| 163 | |
| 164 | RtpRtcpObserver* observer_; |
| 165 | PacketTransportAction on_rtp_, on_rtcp_; |
| 166 | }; |
| 167 | |
| 168 | protected: |
| 169 | scoped_ptr<CriticalSectionWrapper> lock_; |
pbos@webrtc.org | 28a1166 | 2013-09-19 14:22:12 +0000 | [diff] [blame] | 170 | scoped_ptr<EventWrapper> observation_complete_; |
stefan@webrtc.org | 4985c7b | 2013-11-15 12:32:15 +0000 | [diff] [blame] | 171 | scoped_ptr<RtpHeaderParser> parser_; |
pbos@webrtc.org | 8ce445e | 2013-08-19 16:09:34 +0000 | [diff] [blame] | 172 | |
| 173 | private: |
| 174 | PacketTransport send_transport_, receive_transport_; |
pbos@webrtc.org | 28a1166 | 2013-09-19 14:22:12 +0000 | [diff] [blame] | 175 | unsigned int timeout_ms_; |
pbos@webrtc.org | 8ce445e | 2013-08-19 16:09:34 +0000 | [diff] [blame] | 176 | }; |
| 177 | } // namespace test |
| 178 | } // namespace webrtc |
| 179 | |
| 180 | #endif // WEBRTC_VIDEO_ENGINE_TEST_COMMON_RTP_RTCP_OBSERVER_H_ |