1 /*
2  *  Copyright 2004 The WebRTC Project Authors. All rights reserved.
3  *
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
6  *  tree. An additional intellectual property rights grant can be found
7  *  in the file PATENTS.  All contributing project authors may
8  *  be found in the AUTHORS file in the root of the source tree.
9  */
10 
11 #ifndef WEBRTC_BASE_TESTECHOSERVER_H_
12 #define WEBRTC_BASE_TESTECHOSERVER_H_
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 
22 namespace rtc {
23 
24 // A test echo server, echoes back any packets sent to it.
25 // Useful for unit tests.
26 class TestEchoServer : public sigslot::has_slots<> {
27  public:
TestEchoServer(Thread * thread,const SocketAddress & addr)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   }
~TestEchoServer()35   ~TestEchoServer() {
36     for (ClientList::iterator it = client_sockets_.begin();
37          it != client_sockets_.end(); ++it) {
38       delete *it;
39     }
40   }
41 
address()42   SocketAddress address() const { return server_socket_->GetLocalAddress(); }
43 
44  private:
OnAccept(AsyncSocket * socket)45   void OnAccept(AsyncSocket* socket) {
46     AsyncSocket* raw_socket = socket->Accept(NULL);
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   }
OnPacket(AsyncPacketSocket * socket,const char * buf,size_t size,const SocketAddress & remote_addr,const PacketTime & packet_time)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   }
OnClose(AsyncPacketSocket * socket,int err)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 
75 #endif  // WEBRTC_BASE_TESTECHOSERVER_H_
76