aba8dc2a00
This CL is preparation for extraction of public API for network emulation layer. Bug: webrtc:10138 Change-Id: Id59204ea20a103dafce4122c59e51a354836c374 Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/126624 Commit-Queue: Artem Titov <titovartem@webrtc.org> Reviewed-by: Sebastian Jansson <srte@webrtc.org> Cr-Commit-Position: refs/heads/master@{#27050}
137 lines
4.3 KiB
C++
137 lines
4.3 KiB
C++
/*
|
|
* Copyright 2019 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 "absl/memory/memory.h"
|
|
#include "api/test/simulated_network.h"
|
|
#include "call/simulated_network.h"
|
|
#include "rtc_base/event.h"
|
|
#include "rtc_base/logging.h"
|
|
#include "test/gmock.h"
|
|
#include "test/gtest.h"
|
|
#include "test/scenario/network/network_emulation.h"
|
|
#include "test/scenario/network/network_emulation_manager.h"
|
|
|
|
namespace webrtc {
|
|
namespace test {
|
|
|
|
class SocketReader : public sigslot::has_slots<> {
|
|
public:
|
|
explicit SocketReader(rtc::AsyncSocket* socket) : socket_(socket) {
|
|
socket_->SignalReadEvent.connect(this, &SocketReader::OnReadEvent);
|
|
size_ = 128 * 1024;
|
|
buf_ = new char[size_];
|
|
}
|
|
~SocketReader() override { delete[] buf_; }
|
|
|
|
void OnReadEvent(rtc::AsyncSocket* socket) {
|
|
RTC_DCHECK(socket_ == socket);
|
|
int64_t timestamp;
|
|
len_ = socket_->Recv(buf_, size_, ×tamp);
|
|
{
|
|
rtc::CritScope crit(&lock_);
|
|
received_count_++;
|
|
}
|
|
}
|
|
|
|
int ReceivedCount() {
|
|
rtc::CritScope crit(&lock_);
|
|
return received_count_;
|
|
}
|
|
|
|
private:
|
|
rtc::AsyncSocket* socket_;
|
|
char* buf_;
|
|
size_t size_;
|
|
int len_;
|
|
|
|
rtc::CriticalSection lock_;
|
|
int received_count_ RTC_GUARDED_BY(lock_) = 0;
|
|
};
|
|
|
|
TEST(NetworkEmulationManagerTest, GeneratedIpv4AddressDoesNotCollide) {
|
|
NetworkEmulationManager network_manager;
|
|
std::set<rtc::IPAddress> ips;
|
|
EndpointConfig config;
|
|
config.generated_ip_family = EndpointConfig::IpAddressFamily::kIpv4;
|
|
for (int i = 0; i < 1000; i++) {
|
|
EmulatedEndpoint* endpoint = network_manager.CreateEndpoint(config);
|
|
ASSERT_EQ(endpoint->GetPeerLocalAddress().family(), AF_INET);
|
|
bool result = ips.insert(endpoint->GetPeerLocalAddress()).second;
|
|
ASSERT_TRUE(result);
|
|
}
|
|
}
|
|
|
|
TEST(NetworkEmulationManagerTest, GeneratedIpv6AddressDoesNotCollide) {
|
|
NetworkEmulationManager network_manager;
|
|
std::set<rtc::IPAddress> ips;
|
|
EndpointConfig config;
|
|
config.generated_ip_family = EndpointConfig::IpAddressFamily::kIpv6;
|
|
for (int i = 0; i < 1000; i++) {
|
|
EmulatedEndpoint* endpoint = network_manager.CreateEndpoint(config);
|
|
ASSERT_EQ(endpoint->GetPeerLocalAddress().family(), AF_INET6);
|
|
bool result = ips.insert(endpoint->GetPeerLocalAddress()).second;
|
|
ASSERT_TRUE(result);
|
|
}
|
|
}
|
|
|
|
TEST(NetworkEmulationManagerTest, Run) {
|
|
NetworkEmulationManager network_manager;
|
|
|
|
EmulatedNetworkNode* alice_node = network_manager.CreateEmulatedNode(
|
|
absl::make_unique<SimulatedNetwork>(BuiltInNetworkBehaviorConfig()));
|
|
EmulatedNetworkNode* bob_node = network_manager.CreateEmulatedNode(
|
|
absl::make_unique<SimulatedNetwork>(BuiltInNetworkBehaviorConfig()));
|
|
EmulatedEndpoint* alice_endpoint =
|
|
network_manager.CreateEndpoint(EndpointConfig());
|
|
EmulatedEndpoint* bob_endpoint =
|
|
network_manager.CreateEndpoint(EndpointConfig());
|
|
network_manager.CreateRoute(alice_endpoint, {alice_node}, bob_endpoint);
|
|
network_manager.CreateRoute(bob_endpoint, {bob_node}, alice_endpoint);
|
|
|
|
auto* nt1 = network_manager.CreateNetworkThread({alice_endpoint});
|
|
auto* nt2 = network_manager.CreateNetworkThread({bob_endpoint});
|
|
|
|
for (uint64_t j = 0; j < 2; j++) {
|
|
auto* s1 = nt1->socketserver()->CreateAsyncSocket(AF_INET, SOCK_DGRAM);
|
|
auto* s2 = nt2->socketserver()->CreateAsyncSocket(AF_INET, SOCK_DGRAM);
|
|
|
|
SocketReader r1(s1);
|
|
SocketReader r2(s2);
|
|
|
|
rtc::SocketAddress a1(alice_endpoint->GetPeerLocalAddress(), 0);
|
|
rtc::SocketAddress a2(bob_endpoint->GetPeerLocalAddress(), 0);
|
|
|
|
s1->Bind(a1);
|
|
s2->Bind(a2);
|
|
|
|
s1->Connect(s1->GetLocalAddress());
|
|
s2->Connect(s2->GetLocalAddress());
|
|
|
|
rtc::CopyOnWriteBuffer data("Hello");
|
|
for (uint64_t i = 0; i < 1000; i++) {
|
|
s1->Send(data.data(), data.size());
|
|
s2->Send(data.data(), data.size());
|
|
}
|
|
|
|
rtc::Event wait;
|
|
wait.Wait(1000);
|
|
ASSERT_EQ(r1.ReceivedCount(), 1000);
|
|
ASSERT_EQ(r2.ReceivedCount(), 1000);
|
|
|
|
delete s1;
|
|
delete s2;
|
|
}
|
|
}
|
|
|
|
} // namespace test
|
|
} // namespace webrtc
|