1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
/*
* Copyright (c) 2012 The WebRTC project authors. All Rights Reserved.
*
* Use of this source code is governed by a BSD-style license
* that can be found in the LICENSE file in the root of the source
* tree. An additional intellectual property rights grant can be found
* in the file PATENTS. All contributing project authors may
* be found in the AUTHORS file in the root of the source tree.
*/
#ifndef MODULES_RTP_RTCP_TEST_TESTAPI_TEST_API_H_
#define MODULES_RTP_RTCP_TEST_TESTAPI_TEST_API_H_
#include "api/call/transport.h"
#include "common_types.h" // NOLINT(build/include)
#include "modules/rtp_rtcp/include/receive_statistics.h"
#include "modules/rtp_rtcp/include/rtp_header_parser.h"
#include "modules/rtp_rtcp/include/rtp_payload_registry.h"
#include "modules/rtp_rtcp/include/rtp_receiver.h"
#include "modules/rtp_rtcp/include/rtp_rtcp.h"
#include "modules/rtp_rtcp/include/rtp_rtcp_defines.h"
#include "test/gtest.h"
namespace webrtc {
// This class sends all its packet straight to the provided RtpRtcp module.
// with optional packet loss.
class LoopBackTransport : public Transport {
public:
LoopBackTransport()
: count_(0),
packet_loss_(0),
rtp_payload_registry_(NULL),
rtp_receiver_(NULL),
rtp_rtcp_module_(NULL) {}
void SetSendModule(RtpRtcp* rtp_rtcp_module,
RTPPayloadRegistry* payload_registry,
RtpReceiver* receiver,
ReceiveStatistics* receive_statistics);
void DropEveryNthPacket(int n);
bool SendRtp(const uint8_t* data,
size_t len,
const PacketOptions& options) override;
bool SendRtcp(const uint8_t* data, size_t len) override;
private:
int count_;
int packet_loss_;
ReceiveStatistics* receive_statistics_;
RTPPayloadRegistry* rtp_payload_registry_;
RtpReceiver* rtp_receiver_;
RtpRtcp* rtp_rtcp_module_;
};
class TestRtpReceiver : public RtpData {
public:
int32_t OnReceivedPayloadData(
const uint8_t* payload_data,
size_t payload_size,
const webrtc::WebRtcRTPHeader* rtp_header) override;
const uint8_t* payload_data() const { return payload_data_; }
size_t payload_size() const { return payload_size_; }
webrtc::WebRtcRTPHeader rtp_header() const { return rtp_header_; }
private:
uint8_t payload_data_[1500];
size_t payload_size_;
webrtc::WebRtcRTPHeader rtp_header_;
};
} // namespace webrtc
#endif // MODULES_RTP_RTCP_TEST_TESTAPI_TEST_API_H_
|
