| OLD | NEW |
| 1 /* | 1 /* |
| 2 * Copyright (c) 2013 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2013 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 #ifndef WEBRTC_TEST_RTP_RTCP_OBSERVER_H_ | 10 #ifndef WEBRTC_TEST_RTP_RTCP_OBSERVER_H_ |
| 11 #define WEBRTC_TEST_RTP_RTCP_OBSERVER_H_ | 11 #define WEBRTC_TEST_RTP_RTCP_OBSERVER_H_ |
| 12 | 12 |
| 13 #include <map> | 13 #include <map> |
| 14 #include <vector> | 14 #include <vector> |
| 15 | 15 |
| 16 #include "testing/gtest/include/gtest/gtest.h" | 16 #include "testing/gtest/include/gtest/gtest.h" |
| 17 | 17 |
| 18 #include "webrtc/base/criticalsection.h" | 18 #include "webrtc/base/criticalsection.h" |
| 19 #include "webrtc/base/event.h" |
| 19 #include "webrtc/modules/rtp_rtcp/include/rtp_header_parser.h" | 20 #include "webrtc/modules/rtp_rtcp/include/rtp_header_parser.h" |
| 20 #include "webrtc/test/constants.h" | 21 #include "webrtc/test/constants.h" |
| 21 #include "webrtc/test/direct_transport.h" | 22 #include "webrtc/test/direct_transport.h" |
| 22 #include "webrtc/typedefs.h" | 23 #include "webrtc/typedefs.h" |
| 23 #include "webrtc/video_send_stream.h" | 24 #include "webrtc/video_send_stream.h" |
| 24 | 25 |
| 25 namespace webrtc { | 26 namespace webrtc { |
| 26 namespace test { | 27 namespace test { |
| 27 | 28 |
| 28 class PacketTransport; | 29 class PacketTransport; |
| 29 | 30 |
| 30 class RtpRtcpObserver { | 31 class RtpRtcpObserver { |
| 31 public: | 32 public: |
| 32 enum Action { | 33 enum Action { |
| 33 SEND_PACKET, | 34 SEND_PACKET, |
| 34 DROP_PACKET, | 35 DROP_PACKET, |
| 35 }; | 36 }; |
| 36 | 37 |
| 37 virtual ~RtpRtcpObserver() {} | 38 virtual ~RtpRtcpObserver() {} |
| 38 | 39 |
| 39 virtual EventTypeWrapper Wait() { | 40 virtual bool Wait() { return observation_complete_.Wait(timeout_ms_); } |
| 40 EventTypeWrapper result = observation_complete_->Wait(timeout_ms_); | |
| 41 return result; | |
| 42 } | |
| 43 | 41 |
| 44 virtual Action OnSendRtp(const uint8_t* packet, size_t length) { | 42 virtual Action OnSendRtp(const uint8_t* packet, size_t length) { |
| 45 return SEND_PACKET; | 43 return SEND_PACKET; |
| 46 } | 44 } |
| 47 | 45 |
| 48 virtual Action OnSendRtcp(const uint8_t* packet, size_t length) { | 46 virtual Action OnSendRtcp(const uint8_t* packet, size_t length) { |
| 49 return SEND_PACKET; | 47 return SEND_PACKET; |
| 50 } | 48 } |
| 51 | 49 |
| 52 virtual Action OnReceiveRtp(const uint8_t* packet, size_t length) { | 50 virtual Action OnReceiveRtp(const uint8_t* packet, size_t length) { |
| 53 return SEND_PACKET; | 51 return SEND_PACKET; |
| 54 } | 52 } |
| 55 | 53 |
| 56 virtual Action OnReceiveRtcp(const uint8_t* packet, size_t length) { | 54 virtual Action OnReceiveRtcp(const uint8_t* packet, size_t length) { |
| 57 return SEND_PACKET; | 55 return SEND_PACKET; |
| 58 } | 56 } |
| 59 | 57 |
| 60 protected: | 58 protected: |
| 61 explicit RtpRtcpObserver(unsigned int event_timeout_ms) | 59 explicit RtpRtcpObserver(int event_timeout_ms) |
| 62 : observation_complete_(EventWrapper::Create()), | 60 : observation_complete_(false, false), |
| 63 parser_(RtpHeaderParser::Create()), | 61 parser_(RtpHeaderParser::Create()), |
| 64 timeout_ms_(event_timeout_ms) { | 62 timeout_ms_(event_timeout_ms) { |
| 65 parser_->RegisterRtpHeaderExtension(kRtpExtensionTransmissionTimeOffset, | 63 parser_->RegisterRtpHeaderExtension(kRtpExtensionTransmissionTimeOffset, |
| 66 kTOffsetExtensionId); | 64 kTOffsetExtensionId); |
| 67 parser_->RegisterRtpHeaderExtension(kRtpExtensionAbsoluteSendTime, | 65 parser_->RegisterRtpHeaderExtension(kRtpExtensionAbsoluteSendTime, |
| 68 kAbsSendTimeExtensionId); | 66 kAbsSendTimeExtensionId); |
| 69 parser_->RegisterRtpHeaderExtension(kRtpExtensionTransportSequenceNumber, | 67 parser_->RegisterRtpHeaderExtension(kRtpExtensionTransportSequenceNumber, |
| 70 kTransportSequenceNumberExtensionId); | 68 kTransportSequenceNumberExtensionId); |
| 71 } | 69 } |
| 72 | 70 |
| 73 const rtc::scoped_ptr<EventWrapper> observation_complete_; | 71 rtc::Event observation_complete_; |
| 74 const rtc::scoped_ptr<RtpHeaderParser> parser_; | 72 const rtc::scoped_ptr<RtpHeaderParser> parser_; |
| 75 | 73 |
| 76 private: | 74 private: |
| 77 unsigned int timeout_ms_; | 75 const int timeout_ms_; |
| 78 }; | 76 }; |
| 79 | 77 |
| 80 class PacketTransport : public test::DirectTransport { | 78 class PacketTransport : public test::DirectTransport { |
| 81 public: | 79 public: |
| 82 enum TransportType { kReceiver, kSender }; | 80 enum TransportType { kReceiver, kSender }; |
| 83 | 81 |
| 84 PacketTransport(Call* send_call, | 82 PacketTransport(Call* send_call, |
| 85 RtpRtcpObserver* observer, | 83 RtpRtcpObserver* observer, |
| 86 TransportType transport_type, | 84 TransportType transport_type, |
| 87 const FakeNetworkPipe::Config& configuration) | 85 const FakeNetworkPipe::Config& configuration) |
| (...skipping 44 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 132 return true; // Will never happen, makes compiler happy. | 130 return true; // Will never happen, makes compiler happy. |
| 133 } | 131 } |
| 134 | 132 |
| 135 RtpRtcpObserver* const observer_; | 133 RtpRtcpObserver* const observer_; |
| 136 TransportType transport_type_; | 134 TransportType transport_type_; |
| 137 }; | 135 }; |
| 138 } // namespace test | 136 } // namespace test |
| 139 } // namespace webrtc | 137 } // namespace webrtc |
| 140 | 138 |
| 141 #endif // WEBRTC_TEST_RTP_RTCP_OBSERVER_H_ | 139 #endif // WEBRTC_TEST_RTP_RTCP_OBSERVER_H_ |
| OLD | NEW |