Moving src/webrtc into src/.
In order to eliminate the WebRTC Subtree mirror in Chromium, WebRTC is moving the content of the src/webrtc directory up to the src/ directory. NOPRESUBMIT=true NOTREECHECKS=true NOTRY=true TBR=tommi@webrtc.org Bug: chromium:611808 Change-Id: Iac59c5b51b950f174119565bac87955a7994bc38 Reviewed-on: https://webrtc-review.googlesource.com/1560 Commit-Queue: Mirko Bonadei <mbonadei@webrtc.org> Reviewed-by: Henrik Kjellander <kjellander@webrtc.org> Cr-Commit-Position: refs/heads/master@{#19845}
This commit is contained in:
committed by
Commit Bot
parent
6674846b4a
commit
bb547203bf
714
pc/datachannel_unittest.cc
Normal file
714
pc/datachannel_unittest.cc
Normal file
@ -0,0 +1,714 @@
|
||||
/*
|
||||
* Copyright 2013 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 <memory>
|
||||
|
||||
#include "webrtc/pc/datachannel.h"
|
||||
#include "webrtc/pc/sctputils.h"
|
||||
#include "webrtc/pc/test/fakedatachannelprovider.h"
|
||||
#include "webrtc/rtc_base/gunit.h"
|
||||
|
||||
using webrtc::DataChannel;
|
||||
using webrtc::SctpSidAllocator;
|
||||
|
||||
static constexpr int kDefaultTimeout = 10000;
|
||||
|
||||
class FakeDataChannelObserver : public webrtc::DataChannelObserver {
|
||||
public:
|
||||
FakeDataChannelObserver()
|
||||
: messages_received_(0),
|
||||
on_state_change_count_(0),
|
||||
on_buffered_amount_change_count_(0) {}
|
||||
|
||||
void OnStateChange() {
|
||||
++on_state_change_count_;
|
||||
}
|
||||
|
||||
void OnBufferedAmountChange(uint64_t previous_amount) {
|
||||
++on_buffered_amount_change_count_;
|
||||
}
|
||||
|
||||
void OnMessage(const webrtc::DataBuffer& buffer) {
|
||||
++messages_received_;
|
||||
}
|
||||
|
||||
size_t messages_received() const {
|
||||
return messages_received_;
|
||||
}
|
||||
|
||||
void ResetOnStateChangeCount() {
|
||||
on_state_change_count_ = 0;
|
||||
}
|
||||
|
||||
void ResetOnBufferedAmountChangeCount() {
|
||||
on_buffered_amount_change_count_ = 0;
|
||||
}
|
||||
|
||||
size_t on_state_change_count() const {
|
||||
return on_state_change_count_;
|
||||
}
|
||||
|
||||
size_t on_buffered_amount_change_count() const {
|
||||
return on_buffered_amount_change_count_;
|
||||
}
|
||||
|
||||
private:
|
||||
size_t messages_received_;
|
||||
size_t on_state_change_count_;
|
||||
size_t on_buffered_amount_change_count_;
|
||||
};
|
||||
|
||||
class SctpDataChannelTest : public testing::Test {
|
||||
protected:
|
||||
SctpDataChannelTest()
|
||||
: provider_(new FakeDataChannelProvider()),
|
||||
webrtc_data_channel_(DataChannel::Create(provider_.get(),
|
||||
cricket::DCT_SCTP,
|
||||
"test",
|
||||
init_)) {}
|
||||
|
||||
void SetChannelReady() {
|
||||
provider_->set_transport_available(true);
|
||||
webrtc_data_channel_->OnTransportChannelCreated();
|
||||
if (webrtc_data_channel_->id() < 0) {
|
||||
webrtc_data_channel_->SetSctpSid(0);
|
||||
}
|
||||
provider_->set_ready_to_send(true);
|
||||
}
|
||||
|
||||
void AddObserver() {
|
||||
observer_.reset(new FakeDataChannelObserver());
|
||||
webrtc_data_channel_->RegisterObserver(observer_.get());
|
||||
}
|
||||
|
||||
webrtc::InternalDataChannelInit init_;
|
||||
std::unique_ptr<FakeDataChannelProvider> provider_;
|
||||
std::unique_ptr<FakeDataChannelObserver> observer_;
|
||||
rtc::scoped_refptr<DataChannel> webrtc_data_channel_;
|
||||
};
|
||||
|
||||
class StateSignalsListener : public sigslot::has_slots<> {
|
||||
public:
|
||||
int opened_count() const { return opened_count_; }
|
||||
int closed_count() const { return closed_count_; }
|
||||
|
||||
void OnSignalOpened(DataChannel* data_channel) {
|
||||
++opened_count_;
|
||||
}
|
||||
|
||||
void OnSignalClosed(DataChannel* data_channel) {
|
||||
++closed_count_;
|
||||
}
|
||||
|
||||
private:
|
||||
int opened_count_ = 0;
|
||||
int closed_count_ = 0;
|
||||
};
|
||||
|
||||
// Verifies that the data channel is connected to the transport after creation.
|
||||
TEST_F(SctpDataChannelTest, ConnectedToTransportOnCreated) {
|
||||
provider_->set_transport_available(true);
|
||||
rtc::scoped_refptr<DataChannel> dc =
|
||||
DataChannel::Create(provider_.get(), cricket::DCT_SCTP, "test1", init_);
|
||||
|
||||
EXPECT_TRUE(provider_->IsConnected(dc.get()));
|
||||
// The sid is not set yet, so it should not have added the streams.
|
||||
EXPECT_FALSE(provider_->IsSendStreamAdded(dc->id()));
|
||||
EXPECT_FALSE(provider_->IsRecvStreamAdded(dc->id()));
|
||||
|
||||
dc->SetSctpSid(0);
|
||||
EXPECT_TRUE(provider_->IsSendStreamAdded(dc->id()));
|
||||
EXPECT_TRUE(provider_->IsRecvStreamAdded(dc->id()));
|
||||
}
|
||||
|
||||
// Verifies that the data channel is connected to the transport if the transport
|
||||
// is not available initially and becomes available later.
|
||||
TEST_F(SctpDataChannelTest, ConnectedAfterTransportBecomesAvailable) {
|
||||
EXPECT_FALSE(provider_->IsConnected(webrtc_data_channel_.get()));
|
||||
|
||||
provider_->set_transport_available(true);
|
||||
webrtc_data_channel_->OnTransportChannelCreated();
|
||||
EXPECT_TRUE(provider_->IsConnected(webrtc_data_channel_.get()));
|
||||
}
|
||||
|
||||
// Tests the state of the data channel.
|
||||
TEST_F(SctpDataChannelTest, StateTransition) {
|
||||
StateSignalsListener state_signals_listener;
|
||||
webrtc_data_channel_->SignalOpened.connect(
|
||||
&state_signals_listener, &StateSignalsListener::OnSignalOpened);
|
||||
webrtc_data_channel_->SignalClosed.connect(
|
||||
&state_signals_listener, &StateSignalsListener::OnSignalClosed);
|
||||
EXPECT_EQ(webrtc::DataChannelInterface::kConnecting,
|
||||
webrtc_data_channel_->state());
|
||||
EXPECT_EQ(state_signals_listener.opened_count(), 0);
|
||||
EXPECT_EQ(state_signals_listener.closed_count(), 0);
|
||||
SetChannelReady();
|
||||
|
||||
EXPECT_EQ(webrtc::DataChannelInterface::kOpen, webrtc_data_channel_->state());
|
||||
EXPECT_EQ(state_signals_listener.opened_count(), 1);
|
||||
EXPECT_EQ(state_signals_listener.closed_count(), 0);
|
||||
webrtc_data_channel_->Close();
|
||||
EXPECT_EQ(webrtc::DataChannelInterface::kClosed,
|
||||
webrtc_data_channel_->state());
|
||||
EXPECT_EQ(state_signals_listener.opened_count(), 1);
|
||||
EXPECT_EQ(state_signals_listener.closed_count(), 1);
|
||||
// Verifies that it's disconnected from the transport.
|
||||
EXPECT_FALSE(provider_->IsConnected(webrtc_data_channel_.get()));
|
||||
}
|
||||
|
||||
// Tests that DataChannel::buffered_amount() is correct after the channel is
|
||||
// blocked.
|
||||
TEST_F(SctpDataChannelTest, BufferedAmountWhenBlocked) {
|
||||
AddObserver();
|
||||
SetChannelReady();
|
||||
webrtc::DataBuffer buffer("abcd");
|
||||
EXPECT_TRUE(webrtc_data_channel_->Send(buffer));
|
||||
|
||||
EXPECT_EQ(0U, webrtc_data_channel_->buffered_amount());
|
||||
EXPECT_EQ(0U, observer_->on_buffered_amount_change_count());
|
||||
|
||||
provider_->set_send_blocked(true);
|
||||
|
||||
const int number_of_packets = 3;
|
||||
for (int i = 0; i < number_of_packets; ++i) {
|
||||
EXPECT_TRUE(webrtc_data_channel_->Send(buffer));
|
||||
}
|
||||
EXPECT_EQ(buffer.data.size() * number_of_packets,
|
||||
webrtc_data_channel_->buffered_amount());
|
||||
EXPECT_EQ(number_of_packets, observer_->on_buffered_amount_change_count());
|
||||
}
|
||||
|
||||
// Tests that the queued data are sent when the channel transitions from blocked
|
||||
// to unblocked.
|
||||
TEST_F(SctpDataChannelTest, QueuedDataSentWhenUnblocked) {
|
||||
AddObserver();
|
||||
SetChannelReady();
|
||||
webrtc::DataBuffer buffer("abcd");
|
||||
provider_->set_send_blocked(true);
|
||||
EXPECT_TRUE(webrtc_data_channel_->Send(buffer));
|
||||
|
||||
EXPECT_EQ(1U, observer_->on_buffered_amount_change_count());
|
||||
|
||||
provider_->set_send_blocked(false);
|
||||
SetChannelReady();
|
||||
EXPECT_EQ(0U, webrtc_data_channel_->buffered_amount());
|
||||
EXPECT_EQ(2U, observer_->on_buffered_amount_change_count());
|
||||
}
|
||||
|
||||
// Tests that no crash when the channel is blocked right away while trying to
|
||||
// send queued data.
|
||||
TEST_F(SctpDataChannelTest, BlockedWhenSendQueuedDataNoCrash) {
|
||||
AddObserver();
|
||||
SetChannelReady();
|
||||
webrtc::DataBuffer buffer("abcd");
|
||||
provider_->set_send_blocked(true);
|
||||
EXPECT_TRUE(webrtc_data_channel_->Send(buffer));
|
||||
EXPECT_EQ(1U, observer_->on_buffered_amount_change_count());
|
||||
|
||||
// Set channel ready while it is still blocked.
|
||||
SetChannelReady();
|
||||
EXPECT_EQ(buffer.size(), webrtc_data_channel_->buffered_amount());
|
||||
EXPECT_EQ(1U, observer_->on_buffered_amount_change_count());
|
||||
|
||||
// Unblock the channel to send queued data again, there should be no crash.
|
||||
provider_->set_send_blocked(false);
|
||||
SetChannelReady();
|
||||
EXPECT_EQ(0U, webrtc_data_channel_->buffered_amount());
|
||||
EXPECT_EQ(2U, observer_->on_buffered_amount_change_count());
|
||||
}
|
||||
|
||||
// Tests that DataChannel::messages_sent() and DataChannel::bytes_sent() are
|
||||
// correct, sending data both while unblocked and while blocked.
|
||||
TEST_F(SctpDataChannelTest, VerifyMessagesAndBytesSent) {
|
||||
AddObserver();
|
||||
SetChannelReady();
|
||||
std::vector<webrtc::DataBuffer> buffers({
|
||||
webrtc::DataBuffer("message 1"),
|
||||
webrtc::DataBuffer("msg 2"),
|
||||
webrtc::DataBuffer("message three"),
|
||||
webrtc::DataBuffer("quadra message"),
|
||||
webrtc::DataBuffer("fifthmsg"),
|
||||
webrtc::DataBuffer("message of the beast"),
|
||||
});
|
||||
|
||||
// Default values.
|
||||
EXPECT_EQ(0U, webrtc_data_channel_->messages_sent());
|
||||
EXPECT_EQ(0U, webrtc_data_channel_->bytes_sent());
|
||||
|
||||
// Send three buffers while not blocked.
|
||||
provider_->set_send_blocked(false);
|
||||
EXPECT_TRUE(webrtc_data_channel_->Send(buffers[0]));
|
||||
EXPECT_TRUE(webrtc_data_channel_->Send(buffers[1]));
|
||||
EXPECT_TRUE(webrtc_data_channel_->Send(buffers[2]));
|
||||
size_t bytes_sent = buffers[0].size() + buffers[1].size() + buffers[2].size();
|
||||
EXPECT_EQ_WAIT(0U, webrtc_data_channel_->buffered_amount(), kDefaultTimeout);
|
||||
EXPECT_EQ(3U, webrtc_data_channel_->messages_sent());
|
||||
EXPECT_EQ(bytes_sent, webrtc_data_channel_->bytes_sent());
|
||||
|
||||
// Send three buffers while blocked, queuing the buffers.
|
||||
provider_->set_send_blocked(true);
|
||||
EXPECT_TRUE(webrtc_data_channel_->Send(buffers[3]));
|
||||
EXPECT_TRUE(webrtc_data_channel_->Send(buffers[4]));
|
||||
EXPECT_TRUE(webrtc_data_channel_->Send(buffers[5]));
|
||||
size_t bytes_queued =
|
||||
buffers[3].size() + buffers[4].size() + buffers[5].size();
|
||||
EXPECT_EQ(bytes_queued, webrtc_data_channel_->buffered_amount());
|
||||
EXPECT_EQ(3U, webrtc_data_channel_->messages_sent());
|
||||
EXPECT_EQ(bytes_sent, webrtc_data_channel_->bytes_sent());
|
||||
|
||||
// Unblock and make sure everything was sent.
|
||||
provider_->set_send_blocked(false);
|
||||
EXPECT_EQ_WAIT(0U, webrtc_data_channel_->buffered_amount(), kDefaultTimeout);
|
||||
bytes_sent += bytes_queued;
|
||||
EXPECT_EQ(6U, webrtc_data_channel_->messages_sent());
|
||||
EXPECT_EQ(bytes_sent, webrtc_data_channel_->bytes_sent());
|
||||
}
|
||||
|
||||
// Tests that the queued control message is sent when channel is ready.
|
||||
TEST_F(SctpDataChannelTest, OpenMessageSent) {
|
||||
// Initially the id is unassigned.
|
||||
EXPECT_EQ(-1, webrtc_data_channel_->id());
|
||||
|
||||
SetChannelReady();
|
||||
EXPECT_GE(webrtc_data_channel_->id(), 0);
|
||||
EXPECT_EQ(cricket::DMT_CONTROL, provider_->last_send_data_params().type);
|
||||
EXPECT_EQ(provider_->last_send_data_params().ssrc,
|
||||
static_cast<uint32_t>(webrtc_data_channel_->id()));
|
||||
}
|
||||
|
||||
TEST_F(SctpDataChannelTest, QueuedOpenMessageSent) {
|
||||
provider_->set_send_blocked(true);
|
||||
SetChannelReady();
|
||||
provider_->set_send_blocked(false);
|
||||
|
||||
EXPECT_EQ(cricket::DMT_CONTROL, provider_->last_send_data_params().type);
|
||||
EXPECT_EQ(provider_->last_send_data_params().ssrc,
|
||||
static_cast<uint32_t>(webrtc_data_channel_->id()));
|
||||
}
|
||||
|
||||
// Tests that the DataChannel created after transport gets ready can enter OPEN
|
||||
// state.
|
||||
TEST_F(SctpDataChannelTest, LateCreatedChannelTransitionToOpen) {
|
||||
SetChannelReady();
|
||||
webrtc::InternalDataChannelInit init;
|
||||
init.id = 1;
|
||||
rtc::scoped_refptr<DataChannel> dc =
|
||||
DataChannel::Create(provider_.get(), cricket::DCT_SCTP, "test1", init);
|
||||
EXPECT_EQ(webrtc::DataChannelInterface::kConnecting, dc->state());
|
||||
EXPECT_TRUE_WAIT(webrtc::DataChannelInterface::kOpen == dc->state(),
|
||||
1000);
|
||||
}
|
||||
|
||||
// Tests that an unordered DataChannel sends data as ordered until the OPEN_ACK
|
||||
// message is received.
|
||||
TEST_F(SctpDataChannelTest, SendUnorderedAfterReceivesOpenAck) {
|
||||
SetChannelReady();
|
||||
webrtc::InternalDataChannelInit init;
|
||||
init.id = 1;
|
||||
init.ordered = false;
|
||||
rtc::scoped_refptr<DataChannel> dc =
|
||||
DataChannel::Create(provider_.get(), cricket::DCT_SCTP, "test1", init);
|
||||
|
||||
EXPECT_EQ_WAIT(webrtc::DataChannelInterface::kOpen, dc->state(), 1000);
|
||||
|
||||
// Sends a message and verifies it's ordered.
|
||||
webrtc::DataBuffer buffer("some data");
|
||||
ASSERT_TRUE(dc->Send(buffer));
|
||||
EXPECT_TRUE(provider_->last_send_data_params().ordered);
|
||||
|
||||
// Emulates receiving an OPEN_ACK message.
|
||||
cricket::ReceiveDataParams params;
|
||||
params.ssrc = init.id;
|
||||
params.type = cricket::DMT_CONTROL;
|
||||
rtc::CopyOnWriteBuffer payload;
|
||||
webrtc::WriteDataChannelOpenAckMessage(&payload);
|
||||
dc->OnDataReceived(params, payload);
|
||||
|
||||
// Sends another message and verifies it's unordered.
|
||||
ASSERT_TRUE(dc->Send(buffer));
|
||||
EXPECT_FALSE(provider_->last_send_data_params().ordered);
|
||||
}
|
||||
|
||||
// Tests that an unordered DataChannel sends unordered data after any DATA
|
||||
// message is received.
|
||||
TEST_F(SctpDataChannelTest, SendUnorderedAfterReceiveData) {
|
||||
SetChannelReady();
|
||||
webrtc::InternalDataChannelInit init;
|
||||
init.id = 1;
|
||||
init.ordered = false;
|
||||
rtc::scoped_refptr<DataChannel> dc =
|
||||
DataChannel::Create(provider_.get(), cricket::DCT_SCTP, "test1", init);
|
||||
|
||||
EXPECT_EQ_WAIT(webrtc::DataChannelInterface::kOpen, dc->state(), 1000);
|
||||
|
||||
// Emulates receiving a DATA message.
|
||||
cricket::ReceiveDataParams params;
|
||||
params.ssrc = init.id;
|
||||
params.type = cricket::DMT_TEXT;
|
||||
webrtc::DataBuffer buffer("data");
|
||||
dc->OnDataReceived(params, buffer.data);
|
||||
|
||||
// Sends a message and verifies it's unordered.
|
||||
ASSERT_TRUE(dc->Send(buffer));
|
||||
EXPECT_FALSE(provider_->last_send_data_params().ordered);
|
||||
}
|
||||
|
||||
// Tests that the channel can't open until it's successfully sent the OPEN
|
||||
// message.
|
||||
TEST_F(SctpDataChannelTest, OpenWaitsForOpenMesssage) {
|
||||
webrtc::DataBuffer buffer("foo");
|
||||
|
||||
provider_->set_send_blocked(true);
|
||||
SetChannelReady();
|
||||
EXPECT_EQ(webrtc::DataChannelInterface::kConnecting,
|
||||
webrtc_data_channel_->state());
|
||||
provider_->set_send_blocked(false);
|
||||
EXPECT_EQ_WAIT(webrtc::DataChannelInterface::kOpen,
|
||||
webrtc_data_channel_->state(), 1000);
|
||||
EXPECT_EQ(cricket::DMT_CONTROL, provider_->last_send_data_params().type);
|
||||
}
|
||||
|
||||
// Tests that close first makes sure all queued data gets sent.
|
||||
TEST_F(SctpDataChannelTest, QueuedCloseFlushes) {
|
||||
webrtc::DataBuffer buffer("foo");
|
||||
|
||||
provider_->set_send_blocked(true);
|
||||
SetChannelReady();
|
||||
EXPECT_EQ(webrtc::DataChannelInterface::kConnecting,
|
||||
webrtc_data_channel_->state());
|
||||
provider_->set_send_blocked(false);
|
||||
EXPECT_EQ_WAIT(webrtc::DataChannelInterface::kOpen,
|
||||
webrtc_data_channel_->state(), 1000);
|
||||
provider_->set_send_blocked(true);
|
||||
webrtc_data_channel_->Send(buffer);
|
||||
webrtc_data_channel_->Close();
|
||||
provider_->set_send_blocked(false);
|
||||
EXPECT_EQ_WAIT(webrtc::DataChannelInterface::kClosed,
|
||||
webrtc_data_channel_->state(), 1000);
|
||||
EXPECT_EQ(cricket::DMT_TEXT, provider_->last_send_data_params().type);
|
||||
}
|
||||
|
||||
// Tests that messages are sent with the right ssrc.
|
||||
TEST_F(SctpDataChannelTest, SendDataSsrc) {
|
||||
webrtc_data_channel_->SetSctpSid(1);
|
||||
SetChannelReady();
|
||||
webrtc::DataBuffer buffer("data");
|
||||
EXPECT_TRUE(webrtc_data_channel_->Send(buffer));
|
||||
EXPECT_EQ(1U, provider_->last_send_data_params().ssrc);
|
||||
}
|
||||
|
||||
// Tests that the incoming messages with wrong ssrcs are rejected.
|
||||
TEST_F(SctpDataChannelTest, ReceiveDataWithInvalidSsrc) {
|
||||
webrtc_data_channel_->SetSctpSid(1);
|
||||
SetChannelReady();
|
||||
|
||||
AddObserver();
|
||||
|
||||
cricket::ReceiveDataParams params;
|
||||
params.ssrc = 0;
|
||||
webrtc::DataBuffer buffer("abcd");
|
||||
webrtc_data_channel_->OnDataReceived(params, buffer.data);
|
||||
|
||||
EXPECT_EQ(0U, observer_->messages_received());
|
||||
}
|
||||
|
||||
// Tests that the incoming messages with right ssrcs are acceted.
|
||||
TEST_F(SctpDataChannelTest, ReceiveDataWithValidSsrc) {
|
||||
webrtc_data_channel_->SetSctpSid(1);
|
||||
SetChannelReady();
|
||||
|
||||
AddObserver();
|
||||
|
||||
cricket::ReceiveDataParams params;
|
||||
params.ssrc = 1;
|
||||
webrtc::DataBuffer buffer("abcd");
|
||||
|
||||
webrtc_data_channel_->OnDataReceived(params, buffer.data);
|
||||
EXPECT_EQ(1U, observer_->messages_received());
|
||||
}
|
||||
|
||||
// Tests that no CONTROL message is sent if the datachannel is negotiated and
|
||||
// not created from an OPEN message.
|
||||
TEST_F(SctpDataChannelTest, NoMsgSentIfNegotiatedAndNotFromOpenMsg) {
|
||||
webrtc::InternalDataChannelInit config;
|
||||
config.id = 1;
|
||||
config.negotiated = true;
|
||||
config.open_handshake_role = webrtc::InternalDataChannelInit::kNone;
|
||||
|
||||
SetChannelReady();
|
||||
rtc::scoped_refptr<DataChannel> dc =
|
||||
DataChannel::Create(provider_.get(), cricket::DCT_SCTP, "test1", config);
|
||||
|
||||
EXPECT_EQ_WAIT(webrtc::DataChannelInterface::kOpen, dc->state(), 1000);
|
||||
EXPECT_EQ(0U, provider_->last_send_data_params().ssrc);
|
||||
}
|
||||
|
||||
// Tests that DataChannel::messages_received() and DataChannel::bytes_received()
|
||||
// are correct, receiving data both while not open and while open.
|
||||
TEST_F(SctpDataChannelTest, VerifyMessagesAndBytesReceived) {
|
||||
AddObserver();
|
||||
std::vector<webrtc::DataBuffer> buffers({
|
||||
webrtc::DataBuffer("message 1"),
|
||||
webrtc::DataBuffer("msg 2"),
|
||||
webrtc::DataBuffer("message three"),
|
||||
webrtc::DataBuffer("quadra message"),
|
||||
webrtc::DataBuffer("fifthmsg"),
|
||||
webrtc::DataBuffer("message of the beast"),
|
||||
});
|
||||
|
||||
webrtc_data_channel_->SetSctpSid(1);
|
||||
cricket::ReceiveDataParams params;
|
||||
params.ssrc = 1;
|
||||
|
||||
// Default values.
|
||||
EXPECT_EQ(0U, webrtc_data_channel_->messages_received());
|
||||
EXPECT_EQ(0U, webrtc_data_channel_->bytes_received());
|
||||
|
||||
// Receive three buffers while data channel isn't open.
|
||||
webrtc_data_channel_->OnDataReceived(params, buffers[0].data);
|
||||
webrtc_data_channel_->OnDataReceived(params, buffers[1].data);
|
||||
webrtc_data_channel_->OnDataReceived(params, buffers[2].data);
|
||||
EXPECT_EQ(0U, observer_->messages_received());
|
||||
EXPECT_EQ(0U, webrtc_data_channel_->messages_received());
|
||||
EXPECT_EQ(0U, webrtc_data_channel_->bytes_received());
|
||||
|
||||
// Open channel and make sure everything was received.
|
||||
SetChannelReady();
|
||||
size_t bytes_received =
|
||||
buffers[0].size() + buffers[1].size() + buffers[2].size();
|
||||
EXPECT_EQ(3U, observer_->messages_received());
|
||||
EXPECT_EQ(3U, webrtc_data_channel_->messages_received());
|
||||
EXPECT_EQ(bytes_received, webrtc_data_channel_->bytes_received());
|
||||
|
||||
// Receive three buffers while open.
|
||||
webrtc_data_channel_->OnDataReceived(params, buffers[3].data);
|
||||
webrtc_data_channel_->OnDataReceived(params, buffers[4].data);
|
||||
webrtc_data_channel_->OnDataReceived(params, buffers[5].data);
|
||||
bytes_received += buffers[3].size() + buffers[4].size() + buffers[5].size();
|
||||
EXPECT_EQ(6U, observer_->messages_received());
|
||||
EXPECT_EQ(6U, webrtc_data_channel_->messages_received());
|
||||
EXPECT_EQ(bytes_received, webrtc_data_channel_->bytes_received());
|
||||
}
|
||||
|
||||
// Tests that OPEN_ACK message is sent if the datachannel is created from an
|
||||
// OPEN message.
|
||||
TEST_F(SctpDataChannelTest, OpenAckSentIfCreatedFromOpenMessage) {
|
||||
webrtc::InternalDataChannelInit config;
|
||||
config.id = 1;
|
||||
config.negotiated = true;
|
||||
config.open_handshake_role = webrtc::InternalDataChannelInit::kAcker;
|
||||
|
||||
SetChannelReady();
|
||||
rtc::scoped_refptr<DataChannel> dc =
|
||||
DataChannel::Create(provider_.get(), cricket::DCT_SCTP, "test1", config);
|
||||
|
||||
EXPECT_EQ_WAIT(webrtc::DataChannelInterface::kOpen, dc->state(), 1000);
|
||||
|
||||
EXPECT_EQ(static_cast<unsigned int>(config.id),
|
||||
provider_->last_send_data_params().ssrc);
|
||||
EXPECT_EQ(cricket::DMT_CONTROL, provider_->last_send_data_params().type);
|
||||
}
|
||||
|
||||
// Tests the OPEN_ACK role assigned by InternalDataChannelInit.
|
||||
TEST_F(SctpDataChannelTest, OpenAckRoleInitialization) {
|
||||
webrtc::InternalDataChannelInit init;
|
||||
EXPECT_EQ(webrtc::InternalDataChannelInit::kOpener, init.open_handshake_role);
|
||||
EXPECT_FALSE(init.negotiated);
|
||||
|
||||
webrtc::DataChannelInit base;
|
||||
base.negotiated = true;
|
||||
webrtc::InternalDataChannelInit init2(base);
|
||||
EXPECT_EQ(webrtc::InternalDataChannelInit::kNone, init2.open_handshake_role);
|
||||
}
|
||||
|
||||
// Tests that the DataChannel is closed if the sending buffer is full.
|
||||
TEST_F(SctpDataChannelTest, ClosedWhenSendBufferFull) {
|
||||
SetChannelReady();
|
||||
|
||||
rtc::CopyOnWriteBuffer buffer(1024);
|
||||
memset(buffer.data(), 0, buffer.size());
|
||||
|
||||
webrtc::DataBuffer packet(buffer, true);
|
||||
provider_->set_send_blocked(true);
|
||||
|
||||
for (size_t i = 0; i < 16 * 1024 + 1; ++i) {
|
||||
EXPECT_TRUE(webrtc_data_channel_->Send(packet));
|
||||
}
|
||||
|
||||
EXPECT_TRUE(
|
||||
webrtc::DataChannelInterface::kClosed == webrtc_data_channel_->state() ||
|
||||
webrtc::DataChannelInterface::kClosing == webrtc_data_channel_->state());
|
||||
}
|
||||
|
||||
// Tests that the DataChannel is closed on transport errors.
|
||||
TEST_F(SctpDataChannelTest, ClosedOnTransportError) {
|
||||
SetChannelReady();
|
||||
webrtc::DataBuffer buffer("abcd");
|
||||
provider_->set_transport_error();
|
||||
|
||||
EXPECT_TRUE(webrtc_data_channel_->Send(buffer));
|
||||
|
||||
EXPECT_EQ(webrtc::DataChannelInterface::kClosed,
|
||||
webrtc_data_channel_->state());
|
||||
}
|
||||
|
||||
// Tests that a already closed DataChannel does not fire onStateChange again.
|
||||
TEST_F(SctpDataChannelTest, ClosedDataChannelDoesNotFireOnStateChange) {
|
||||
AddObserver();
|
||||
webrtc_data_channel_->Close();
|
||||
// OnStateChange called for kClosing and kClosed.
|
||||
EXPECT_EQ(2U, observer_->on_state_change_count());
|
||||
|
||||
observer_->ResetOnStateChangeCount();
|
||||
webrtc_data_channel_->RemotePeerRequestClose();
|
||||
EXPECT_EQ(0U, observer_->on_state_change_count());
|
||||
}
|
||||
|
||||
// Tests that RemotePeerRequestClose closes the local DataChannel.
|
||||
TEST_F(SctpDataChannelTest, RemotePeerRequestClose) {
|
||||
AddObserver();
|
||||
webrtc_data_channel_->RemotePeerRequestClose();
|
||||
|
||||
// OnStateChange called for kClosing and kClosed.
|
||||
EXPECT_EQ(2U, observer_->on_state_change_count());
|
||||
EXPECT_EQ(webrtc::DataChannelInterface::kClosed,
|
||||
webrtc_data_channel_->state());
|
||||
}
|
||||
|
||||
// Tests that the DataChannel is closed if the received buffer is full.
|
||||
TEST_F(SctpDataChannelTest, ClosedWhenReceivedBufferFull) {
|
||||
SetChannelReady();
|
||||
rtc::CopyOnWriteBuffer buffer(1024);
|
||||
memset(buffer.data(), 0, buffer.size());
|
||||
|
||||
cricket::ReceiveDataParams params;
|
||||
params.ssrc = 0;
|
||||
|
||||
// Receiving data without having an observer will overflow the buffer.
|
||||
for (size_t i = 0; i < 16 * 1024 + 1; ++i) {
|
||||
webrtc_data_channel_->OnDataReceived(params, buffer);
|
||||
}
|
||||
EXPECT_EQ(webrtc::DataChannelInterface::kClosed,
|
||||
webrtc_data_channel_->state());
|
||||
}
|
||||
|
||||
// Tests that sending empty data returns no error and keeps the channel open.
|
||||
TEST_F(SctpDataChannelTest, SendEmptyData) {
|
||||
webrtc_data_channel_->SetSctpSid(1);
|
||||
SetChannelReady();
|
||||
EXPECT_EQ(webrtc::DataChannelInterface::kOpen,
|
||||
webrtc_data_channel_->state());
|
||||
|
||||
webrtc::DataBuffer buffer("");
|
||||
EXPECT_TRUE(webrtc_data_channel_->Send(buffer));
|
||||
EXPECT_EQ(webrtc::DataChannelInterface::kOpen,
|
||||
webrtc_data_channel_->state());
|
||||
}
|
||||
|
||||
// Tests that a channel can be closed without being opened or assigned an sid.
|
||||
TEST_F(SctpDataChannelTest, NeverOpened) {
|
||||
provider_->set_transport_available(true);
|
||||
webrtc_data_channel_->OnTransportChannelCreated();
|
||||
webrtc_data_channel_->Close();
|
||||
}
|
||||
|
||||
// Test that the data channel goes to the "closed" state (and doesn't crash)
|
||||
// when its transport goes away, even while data is buffered.
|
||||
TEST_F(SctpDataChannelTest, TransportDestroyedWhileDataBuffered) {
|
||||
SetChannelReady();
|
||||
|
||||
rtc::CopyOnWriteBuffer buffer(1024);
|
||||
memset(buffer.data(), 0, buffer.size());
|
||||
webrtc::DataBuffer packet(buffer, true);
|
||||
|
||||
// Send a packet while sending is blocked so it ends up buffered.
|
||||
provider_->set_send_blocked(true);
|
||||
EXPECT_TRUE(webrtc_data_channel_->Send(packet));
|
||||
|
||||
// Tell the data channel that its tranpsort is being destroyed.
|
||||
// It should then stop using the transport (allowing us to delete it) and
|
||||
// transition to the "closed" state.
|
||||
webrtc_data_channel_->OnTransportChannelDestroyed();
|
||||
provider_.reset(nullptr);
|
||||
EXPECT_EQ_WAIT(webrtc::DataChannelInterface::kClosed,
|
||||
webrtc_data_channel_->state(), kDefaultTimeout);
|
||||
}
|
||||
|
||||
class SctpSidAllocatorTest : public testing::Test {
|
||||
protected:
|
||||
SctpSidAllocator allocator_;
|
||||
};
|
||||
|
||||
// Verifies that an even SCTP id is allocated for SSL_CLIENT and an odd id for
|
||||
// SSL_SERVER.
|
||||
TEST_F(SctpSidAllocatorTest, SctpIdAllocationBasedOnRole) {
|
||||
int id;
|
||||
EXPECT_TRUE(allocator_.AllocateSid(rtc::SSL_SERVER, &id));
|
||||
EXPECT_EQ(1, id);
|
||||
EXPECT_TRUE(allocator_.AllocateSid(rtc::SSL_CLIENT, &id));
|
||||
EXPECT_EQ(0, id);
|
||||
EXPECT_TRUE(allocator_.AllocateSid(rtc::SSL_SERVER, &id));
|
||||
EXPECT_EQ(3, id);
|
||||
EXPECT_TRUE(allocator_.AllocateSid(rtc::SSL_CLIENT, &id));
|
||||
EXPECT_EQ(2, id);
|
||||
}
|
||||
|
||||
// Verifies that SCTP ids of existing DataChannels are not reused.
|
||||
TEST_F(SctpSidAllocatorTest, SctpIdAllocationNoReuse) {
|
||||
int old_id = 1;
|
||||
EXPECT_TRUE(allocator_.ReserveSid(old_id));
|
||||
|
||||
int new_id;
|
||||
EXPECT_TRUE(allocator_.AllocateSid(rtc::SSL_SERVER, &new_id));
|
||||
EXPECT_NE(old_id, new_id);
|
||||
|
||||
old_id = 0;
|
||||
EXPECT_TRUE(allocator_.ReserveSid(old_id));
|
||||
EXPECT_TRUE(allocator_.AllocateSid(rtc::SSL_CLIENT, &new_id));
|
||||
EXPECT_NE(old_id, new_id);
|
||||
}
|
||||
|
||||
// Verifies that SCTP ids of removed DataChannels can be reused.
|
||||
TEST_F(SctpSidAllocatorTest, SctpIdReusedForRemovedDataChannel) {
|
||||
int odd_id = 1;
|
||||
int even_id = 0;
|
||||
EXPECT_TRUE(allocator_.ReserveSid(odd_id));
|
||||
EXPECT_TRUE(allocator_.ReserveSid(even_id));
|
||||
|
||||
int allocated_id = -1;
|
||||
EXPECT_TRUE(allocator_.AllocateSid(rtc::SSL_SERVER, &allocated_id));
|
||||
EXPECT_EQ(odd_id + 2, allocated_id);
|
||||
|
||||
EXPECT_TRUE(allocator_.AllocateSid(rtc::SSL_CLIENT, &allocated_id));
|
||||
EXPECT_EQ(even_id + 2, allocated_id);
|
||||
|
||||
EXPECT_TRUE(allocator_.AllocateSid(rtc::SSL_SERVER, &allocated_id));
|
||||
EXPECT_EQ(odd_id + 4, allocated_id);
|
||||
|
||||
EXPECT_TRUE(allocator_.AllocateSid(rtc::SSL_CLIENT, &allocated_id));
|
||||
EXPECT_EQ(even_id + 4, allocated_id);
|
||||
|
||||
allocator_.ReleaseSid(odd_id);
|
||||
allocator_.ReleaseSid(even_id);
|
||||
|
||||
// Verifies that removed ids are reused.
|
||||
EXPECT_TRUE(allocator_.AllocateSid(rtc::SSL_SERVER, &allocated_id));
|
||||
EXPECT_EQ(odd_id, allocated_id);
|
||||
|
||||
EXPECT_TRUE(allocator_.AllocateSid(rtc::SSL_CLIENT, &allocated_id));
|
||||
EXPECT_EQ(even_id, allocated_id);
|
||||
|
||||
// Verifies that used higher ids are not reused.
|
||||
EXPECT_TRUE(allocator_.AllocateSid(rtc::SSL_SERVER, &allocated_id));
|
||||
EXPECT_EQ(odd_id + 6, allocated_id);
|
||||
|
||||
EXPECT_TRUE(allocator_.AllocateSid(rtc::SSL_CLIENT, &allocated_id));
|
||||
EXPECT_EQ(even_id + 6, allocated_id);
|
||||
}
|
||||
Reference in New Issue
Block a user