summaryrefslogtreecommitdiffstats
path: root/third_party/libwebrtc/rtc_base/test_echo_server.h
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-19 00:47:55 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-19 00:47:55 +0000
commit26a029d407be480d791972afb5975cf62c9360a6 (patch)
treef435a8308119effd964b339f76abb83a57c29483 /third_party/libwebrtc/rtc_base/test_echo_server.h
parentInitial commit. (diff)
downloadfirefox-26a029d407be480d791972afb5975cf62c9360a6.tar.xz
firefox-26a029d407be480d791972afb5975cf62c9360a6.zip
Adding upstream version 124.0.1.upstream/124.0.1
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'third_party/libwebrtc/rtc_base/test_echo_server.h')
-rw-r--r--third_party/libwebrtc/rtc_base/test_echo_server.h77
1 files changed, 77 insertions, 0 deletions
diff --git a/third_party/libwebrtc/rtc_base/test_echo_server.h b/third_party/libwebrtc/rtc_base/test_echo_server.h
new file mode 100644
index 0000000000..82817624a5
--- /dev/null
+++ b/third_party/libwebrtc/rtc_base/test_echo_server.h
@@ -0,0 +1,77 @@
+/*
+ * Copyright 2004 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 RTC_BASE_TEST_ECHO_SERVER_H_
+#define RTC_BASE_TEST_ECHO_SERVER_H_
+
+#include <stddef.h>
+#include <stdint.h>
+
+#include <list>
+#include <memory>
+
+#include "absl/algorithm/container.h"
+#include "absl/memory/memory.h"
+#include "rtc_base/async_packet_socket.h"
+#include "rtc_base/async_tcp_socket.h"
+#include "rtc_base/socket.h"
+#include "rtc_base/socket_address.h"
+#include "rtc_base/third_party/sigslot/sigslot.h"
+#include "rtc_base/thread.h"
+
+namespace rtc {
+
+// A test echo server, echoes back any packets sent to it.
+// Useful for unit tests.
+class TestEchoServer : public sigslot::has_slots<> {
+ public:
+ TestEchoServer(Thread* thread, const SocketAddress& addr);
+ ~TestEchoServer() override;
+
+ TestEchoServer(const TestEchoServer&) = delete;
+ TestEchoServer& operator=(const TestEchoServer&) = delete;
+
+ SocketAddress address() const { return server_socket_->GetLocalAddress(); }
+
+ private:
+ void OnAccept(Socket* socket) {
+ Socket* raw_socket = socket->Accept(nullptr);
+ if (raw_socket) {
+ AsyncTCPSocket* packet_socket = new AsyncTCPSocket(raw_socket);
+ packet_socket->SignalReadPacket.connect(this, &TestEchoServer::OnPacket);
+ packet_socket->SubscribeCloseEvent(
+ this, [this](AsyncPacketSocket* s, int err) { OnClose(s, err); });
+ client_sockets_.push_back(packet_socket);
+ }
+ }
+ void OnPacket(AsyncPacketSocket* socket,
+ const char* buf,
+ size_t size,
+ const SocketAddress& remote_addr,
+ const int64_t& /* packet_time_us */) {
+ rtc::PacketOptions options;
+ socket->Send(buf, size, options);
+ }
+ void OnClose(AsyncPacketSocket* socket, int err) {
+ ClientList::iterator it = absl::c_find(client_sockets_, socket);
+ client_sockets_.erase(it);
+ // `OnClose` is triggered by socket Close callback, deleting `socket` while
+ // processing that callback might be unsafe.
+ Thread::Current()->PostTask([socket = absl::WrapUnique(socket)] {});
+ }
+
+ typedef std::list<AsyncTCPSocket*> ClientList;
+ std::unique_ptr<Socket> server_socket_;
+ ClientList client_sockets_;
+};
+
+} // namespace rtc
+
+#endif // RTC_BASE_TEST_ECHO_SERVER_H_