1 files changed, 83 insertions, 0 deletions
diff --git a/third_party/libwebrtc/test/network/emulated_network_manager.h b/third_party/libwebrtc/test/network/emulated_network_manager.h
new file mode 100644
index 0000000000..fb4ee1ee85
--- /dev/null
+++ b/third_party/libwebrtc/test/network/emulated_network_manager.h
@@ -0,0 +1,83 @@
+/*
+ *  Copyright (c) 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.
+ */
+
+#ifndef TEST_NETWORK_EMULATED_NETWORK_MANAGER_H_
+#define TEST_NETWORK_EMULATED_NETWORK_MANAGER_H_
+
+#include <functional>
+#include <memory>
+#include <vector>
+
+#include "api/sequence_checker.h"
+#include "api/test/network_emulation_manager.h"
+#include "api/test/time_controller.h"
+#include "rtc_base/ip_address.h"
+#include "rtc_base/network.h"
+#include "rtc_base/socket_server.h"
+#include "rtc_base/thread.h"
+#include "test/network/network_emulation.h"
+
+namespace webrtc {
+namespace test {
+
+// Framework assumes that rtc::NetworkManager is called from network thread.
+class EmulatedNetworkManager : public rtc::NetworkManagerBase,
+                               public sigslot::has_slots<>,
+                               public EmulatedNetworkManagerInterface {
+ public:
+  EmulatedNetworkManager(TimeController* time_controller,
+                         TaskQueueForTest* task_queue,
+                         EndpointsContainer* endpoints_container);
+
+  void EnableEndpoint(EmulatedEndpointImpl* endpoint);
+  void DisableEndpoint(EmulatedEndpointImpl* endpoint);
+
+  // NetworkManager interface. All these methods are supposed to be called from
+  // the same thread.
+  void StartUpdating() override;
+  void StopUpdating() override;
+
+  // We don't support any address interfaces in the network emulation framework.
+  std::vector<const rtc::Network*> GetAnyAddressNetworks() override {
+    return {};
+  }
+
+  // EmulatedNetworkManagerInterface API
+  rtc::Thread* network_thread() override { return network_thread_.get(); }
+  rtc::NetworkManager* network_manager() override { return this; }
+  rtc::PacketSocketFactory* packet_socket_factory() override {
+    return packet_socket_factory_.get();
+  }
+  std::vector<EmulatedEndpoint*> endpoints() const override {
+    return endpoints_container_->GetEndpoints();
+  }
+  void GetStats(
+      std::function<void(EmulatedNetworkStats)> stats_callback) const override;
+
+ private:
+  void UpdateNetworksOnce();
+  void MaybeSignalNetworksChanged();
+
+  TaskQueueForTest* const task_queue_;
+  const EndpointsContainer* const endpoints_container_;
+  // The `network_thread_` must outlive `packet_socket_factory_`, because they
+  // both refer to a socket server that is owned by `network_thread_`. Both
+  // pointers are assigned only in the constructor, but the way they are
+  // initialized unfortunately doesn't work with const std::unique_ptr<...>.
+  std::unique_ptr<rtc::Thread> network_thread_;
+  std::unique_ptr<rtc::PacketSocketFactory> packet_socket_factory_;
+  bool sent_first_update_ RTC_GUARDED_BY(network_thread_);
+  int start_count_ RTC_GUARDED_BY(network_thread_);
+};
+
+}  // namespace test
+}  // namespace webrtc
+
+#endif  // TEST_NETWORK_EMULATED_NETWORK_MANAGER_H_