OLD | NEW |
1 /* | 1 /* |
2 * Copyright 2004 The WebRTC Project Authors. All rights reserved. | 2 * Copyright 2004 The WebRTC Project Authors. All rights reserved. |
3 * | 3 * |
4 * Use of this source code is governed by a BSD-style license | 4 * Use of this source code is governed by a BSD-style license |
5 * that can be found in the LICENSE file in the root of the source | 5 * that can be found in the LICENSE file in the root of the source |
6 * tree. An additional intellectual property rights grant can be found | 6 * tree. An additional intellectual property rights grant can be found |
7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
9 */ | 9 */ |
10 | 10 |
11 #ifndef WEBRTC_BASE_TESTECHOSERVER_H_ | 11 #ifndef WEBRTC_BASE_TESTECHOSERVER_H_ |
12 #define WEBRTC_BASE_TESTECHOSERVER_H_ | 12 #define WEBRTC_BASE_TESTECHOSERVER_H_ |
13 | 13 |
14 #include <list> | |
15 #include <memory> | |
16 #include "webrtc/base/asynctcpsocket.h" | |
17 #include "webrtc/base/constructormagic.h" | |
18 #include "webrtc/base/socketaddress.h" | |
19 #include "webrtc/base/sigslot.h" | |
20 #include "webrtc/base/thread.h" | |
21 | 14 |
22 namespace rtc { | 15 // This header is deprecated and is just left here temporarily during |
23 | 16 // refactoring. See https://bugs.webrtc.org/7634 for more details. |
24 // A test echo server, echoes back any packets sent to it. | 17 #include "webrtc/rtc_base/testechoserver.h" |
25 // Useful for unit tests. | |
26 class TestEchoServer : public sigslot::has_slots<> { | |
27 public: | |
28 TestEchoServer(Thread* thread, const SocketAddress& addr) | |
29 : server_socket_(thread->socketserver()->CreateAsyncSocket(addr.family(), | |
30 SOCK_STREAM)) { | |
31 server_socket_->Bind(addr); | |
32 server_socket_->Listen(5); | |
33 server_socket_->SignalReadEvent.connect(this, &TestEchoServer::OnAccept); | |
34 } | |
35 ~TestEchoServer() { | |
36 for (ClientList::iterator it = client_sockets_.begin(); | |
37 it != client_sockets_.end(); ++it) { | |
38 delete *it; | |
39 } | |
40 } | |
41 | |
42 SocketAddress address() const { return server_socket_->GetLocalAddress(); } | |
43 | |
44 private: | |
45 void OnAccept(AsyncSocket* socket) { | |
46 AsyncSocket* raw_socket = socket->Accept(nullptr); | |
47 if (raw_socket) { | |
48 AsyncTCPSocket* packet_socket = new AsyncTCPSocket(raw_socket, false); | |
49 packet_socket->SignalReadPacket.connect(this, &TestEchoServer::OnPacket); | |
50 packet_socket->SignalClose.connect(this, &TestEchoServer::OnClose); | |
51 client_sockets_.push_back(packet_socket); | |
52 } | |
53 } | |
54 void OnPacket(AsyncPacketSocket* socket, const char* buf, size_t size, | |
55 const SocketAddress& remote_addr, | |
56 const PacketTime& packet_time) { | |
57 rtc::PacketOptions options; | |
58 socket->Send(buf, size, options); | |
59 } | |
60 void OnClose(AsyncPacketSocket* socket, int err) { | |
61 ClientList::iterator it = | |
62 std::find(client_sockets_.begin(), client_sockets_.end(), socket); | |
63 client_sockets_.erase(it); | |
64 Thread::Current()->Dispose(socket); | |
65 } | |
66 | |
67 typedef std::list<AsyncTCPSocket*> ClientList; | |
68 std::unique_ptr<AsyncSocket> server_socket_; | |
69 ClientList client_sockets_; | |
70 RTC_DISALLOW_COPY_AND_ASSIGN(TestEchoServer); | |
71 }; | |
72 | |
73 } // namespace rtc | |
74 | 18 |
75 #endif // WEBRTC_BASE_TESTECHOSERVER_H_ | 19 #endif // WEBRTC_BASE_TESTECHOSERVER_H_ |
OLD | NEW |