blob: 3bfa475f73c6aff71db02bdbe5030f32617d7853 [file] [log] [blame]
pbos@webrtc.orge7f056e2013-08-19 16:09:341/*
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 */
Mirko Bonadei92ea95e2017-09-15 04:47:3110#ifndef TEST_RTP_RTCP_OBSERVER_H_
11#define TEST_RTP_RTCP_OBSERVER_H_
pbos@webrtc.orge7f056e2013-08-19 16:09:3412
13#include <map>
kwibergbfefb032016-05-01 21:53:4614#include <memory>
Elad Alond8d32482019-02-18 22:45:5715#include <utility>
pbos@webrtc.orge7f056e2013-08-19 16:09:3416#include <vector>
17
Artem Titov46c4e602018-08-17 12:26:5418#include "api/test/simulated_network.h"
19#include "call/simulated_packet_receiver.h"
Yves Gerey665174f2018-06-19 13:03:0520#include "call/video_send_stream.h"
Steve Anton10542f22019-01-11 17:11:0021#include "rtc_base/critical_section.h"
Mirko Bonadei92ea95e2017-09-15 04:47:3122#include "rtc_base/event.h"
23#include "system_wrappers/include/field_trial.h"
Mirko Bonadei92ea95e2017-09-15 04:47:3124#include "test/direct_transport.h"
25#include "test/gtest.h"
Tommi25eb47c2019-08-29 14:39:0526#include "test/rtp_header_parser.h"
pbos@webrtc.orge7f056e2013-08-19 16:09:3427
ilnik5328b9e2017-02-21 13:20:2828namespace {
29const int kShortTimeoutMs = 500;
30}
31
pbos@webrtc.orge7f056e2013-08-19 16:09:3432namespace webrtc {
33namespace test {
34
stefanf116bd02015-10-27 15:29:4235class PacketTransport;
36
pbos@webrtc.orge7f056e2013-08-19 16:09:3437class RtpRtcpObserver {
38 public:
stefanf116bd02015-10-27 15:29:4239 enum Action {
40 SEND_PACKET,
41 DROP_PACKET,
42 };
43
pbos@webrtc.orgb3cc78d2013-11-21 11:42:0244 virtual ~RtpRtcpObserver() {}
pbos@webrtc.orge7f056e2013-08-19 16:09:3445
ilnik5328b9e2017-02-21 13:20:2846 virtual bool Wait() {
sprangc1b57a12017-02-28 16:50:4747 if (field_trial::IsEnabled("WebRTC-QuickPerfTest")) {
ilnik5328b9e2017-02-21 13:20:2848 observation_complete_.Wait(kShortTimeoutMs);
49 return true;
50 }
51 return observation_complete_.Wait(timeout_ms_);
52 }
pbos@webrtc.org6917e192013-09-19 14:22:1253
stefanf116bd02015-10-27 15:29:4254 virtual Action OnSendRtp(const uint8_t* packet, size_t length) {
55 return SEND_PACKET;
56 }
stefan@webrtc.orgb082ade2013-11-18 11:45:1157
stefanf116bd02015-10-27 15:29:4258 virtual Action OnSendRtcp(const uint8_t* packet, size_t length) {
59 return SEND_PACKET;
60 }
61
62 virtual Action OnReceiveRtp(const uint8_t* packet, size_t length) {
63 return SEND_PACKET;
64 }
65
66 virtual Action OnReceiveRtcp(const uint8_t* packet, size_t length) {
67 return SEND_PACKET;
68 }
69
70 protected:
philipele828c962017-03-21 10:24:2771 RtpRtcpObserver() : RtpRtcpObserver(0) {}
Peter Boström5811a392015-12-10 12:02:5072 explicit RtpRtcpObserver(int event_timeout_ms)
Danil Chapovalov1b4e4bf2019-12-06 11:34:5773 : timeout_ms_(event_timeout_ms) {}
pbos@webrtc.orge7f056e2013-08-19 16:09:3474
Peter Boström5811a392015-12-10 12:02:5075 rtc::Event observation_complete_;
pbos@webrtc.orge7f056e2013-08-19 16:09:3476
77 private:
Peter Boström5811a392015-12-10 12:02:5078 const int timeout_ms_;
pbos@webrtc.orge7f056e2013-08-19 16:09:3479};
stefanf116bd02015-10-27 15:29:4280
81class PacketTransport : public test::DirectTransport {
82 public:
83 enum TransportType { kReceiver, kSender };
84
Danil Chapovalov44db4362019-09-30 02:16:2885 PacketTransport(TaskQueueBase* task_queue,
Sebastian Jansson09408112018-04-24 12:41:2286 Call* send_call,
87 RtpRtcpObserver* observer,
Christoffer Rodbrod2817d82017-10-24 14:26:4988 TransportType transport_type,
Sebastian Jansson09408112018-04-24 12:41:2289 const std::map<uint8_t, MediaType>& payload_type_map,
Artem Titov46c4e602018-08-17 12:26:5490 std::unique_ptr<SimulatedPacketReceiverInterface> nw_pipe)
Sebastian Jansson09408112018-04-24 12:41:2291 : test::DirectTransport(task_queue,
92 std::move(nw_pipe),
93 send_call,
94 payload_type_map),
Christoffer Rodbrod2817d82017-10-24 14:26:4995 observer_(observer),
96 transport_type_(transport_type) {}
97
stefanf116bd02015-10-27 15:29:4298 private:
99 bool SendRtp(const uint8_t* packet,
100 size_t length,
101 const PacketOptions& options) override {
102 EXPECT_FALSE(RtpHeaderParser::IsRtcp(packet, length));
103 RtpRtcpObserver::Action action;
104 {
105 if (transport_type_ == kSender) {
106 action = observer_->OnSendRtp(packet, length);
107 } else {
108 action = observer_->OnReceiveRtp(packet, length);
109 }
110 }
111 switch (action) {
112 case RtpRtcpObserver::DROP_PACKET:
113 // Drop packet silently.
114 return true;
115 case RtpRtcpObserver::SEND_PACKET:
116 return test::DirectTransport::SendRtp(packet, length, options);
117 }
118 return true; // Will never happen, makes compiler happy.
119 }
120
121 bool SendRtcp(const uint8_t* packet, size_t length) override {
122 EXPECT_TRUE(RtpHeaderParser::IsRtcp(packet, length));
123 RtpRtcpObserver::Action action;
124 {
125 if (transport_type_ == kSender) {
126 action = observer_->OnSendRtcp(packet, length);
127 } else {
128 action = observer_->OnReceiveRtcp(packet, length);
129 }
130 }
131 switch (action) {
132 case RtpRtcpObserver::DROP_PACKET:
133 // Drop packet silently.
134 return true;
135 case RtpRtcpObserver::SEND_PACKET:
136 return test::DirectTransport::SendRtcp(packet, length);
137 }
138 return true; // Will never happen, makes compiler happy.
139 }
140
141 RtpRtcpObserver* const observer_;
142 TransportType transport_type_;
143};
pbos@webrtc.orge7f056e2013-08-19 16:09:34144} // namespace test
145} // namespace webrtc
146
Mirko Bonadei92ea95e2017-09-15 04:47:31147#endif // TEST_RTP_RTCP_OBSERVER_H_