a678a3baee
TEST=video_coding_unittests, video_coding_integrationtests, rtp_rtcp_unittests, trybots Review URL: https://webrtc-codereview.appspot.com/1044004 git-svn-id: http://webrtc.googlecode.com/svn/trunk@3393 4adac7df-926f-26a2-2b94-8c16560cd09d
91 lines
2.3 KiB
C++
91 lines
2.3 KiB
C++
/*
|
|
* 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.
|
|
*/
|
|
|
|
#include "common_types.h"
|
|
#include "rtp_rtcp.h"
|
|
#include "rtp_rtcp_defines.h"
|
|
|
|
namespace webrtc {
|
|
|
|
// This class sends all its packet straight to the provided RtpRtcp module.
|
|
// with optional packet loss.
|
|
class LoopBackTransport : public webrtc::Transport {
|
|
public:
|
|
LoopBackTransport()
|
|
: _count(0),
|
|
_packetLoss(0),
|
|
_rtpRtcpModule(NULL) {
|
|
}
|
|
void SetSendModule(RtpRtcp* rtpRtcpModule) {
|
|
_rtpRtcpModule = rtpRtcpModule;
|
|
}
|
|
void DropEveryNthPacket(int n) {
|
|
_packetLoss = n;
|
|
}
|
|
virtual int SendPacket(int channel, const void *data, int len) {
|
|
_count++;
|
|
if (_packetLoss > 0) {
|
|
if ((_count % _packetLoss) == 0) {
|
|
return len;
|
|
}
|
|
}
|
|
if (_rtpRtcpModule->IncomingPacket((const WebRtc_UWord8*)data, len) == 0) {
|
|
return len;
|
|
}
|
|
return -1;
|
|
}
|
|
virtual int SendRTCPPacket(int channel, const void *data, int len) {
|
|
if (_rtpRtcpModule->IncomingPacket((const WebRtc_UWord8*)data, len) == 0) {
|
|
return len;
|
|
}
|
|
return -1;
|
|
}
|
|
private:
|
|
int _count;
|
|
int _packetLoss;
|
|
RtpRtcp* _rtpRtcpModule;
|
|
};
|
|
|
|
class RtpReceiver : public RtpData {
|
|
public:
|
|
enum { kMaxPayloadSize = 1500 };
|
|
|
|
virtual WebRtc_Word32 OnReceivedPayloadData(
|
|
const WebRtc_UWord8* payloadData,
|
|
const WebRtc_UWord16 payloadSize,
|
|
const webrtc::WebRtcRTPHeader* rtpHeader) {
|
|
EXPECT_LE(payloadSize, kMaxPayloadSize);
|
|
memcpy(_payloadData, payloadData, payloadSize);
|
|
memcpy(&_rtpHeader, rtpHeader, sizeof(_rtpHeader));
|
|
_payloadSize = payloadSize;
|
|
return 0;
|
|
}
|
|
|
|
const WebRtc_UWord8* payload_data() const {
|
|
return _payloadData;
|
|
}
|
|
|
|
WebRtc_UWord16 payload_size() const {
|
|
return _payloadSize;
|
|
}
|
|
|
|
webrtc::WebRtcRTPHeader rtp_header() const {
|
|
return _rtpHeader;
|
|
}
|
|
|
|
private:
|
|
WebRtc_UWord8 _payloadData[kMaxPayloadSize];
|
|
WebRtc_UWord16 _payloadSize;
|
|
webrtc::WebRtcRTPHeader _rtpHeader;
|
|
};
|
|
|
|
} // namespace webrtc
|
|
|