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 "webrtc/base/asynctcpsocket.h"
16 #include "webrtc/base/socketaddress.h"
17 #include "webrtc/base/sigslot.h"
18 #include "webrtc/base/thread.h"
19 
20 namespace rtc {
21 
22 // A test echo server, echoes back any packets sent to it.
23 // Useful for unit tests.
24 class TestEchoServer : public sigslot::has_slots<> {
25  public:
TestEchoServer(Thread * thread,const SocketAddress & addr)26   TestEchoServer(Thread* thread, const SocketAddress& addr)
27       : server_socket_(thread->socketserver()->CreateAsyncSocket(addr.family(),
28                                                                  SOCK_STREAM)) {
29     server_socket_->Bind(addr);
30     server_socket_->Listen(5);
31     server_socket_->SignalReadEvent.connect(this, &TestEchoServer::OnAccept);
32   }
~TestEchoServer()33   ~TestEchoServer() {
34     for (ClientList::iterator it = client_sockets_.begin();
35          it != client_sockets_.end(); ++it) {
36       delete *it;
37     }
38   }
39 
address()40   SocketAddress address() const { return server_socket_->GetLocalAddress(); }
41 
42  private:
OnAccept(AsyncSocket * socket)43   void OnAccept(AsyncSocket* socket) {
44     AsyncSocket* raw_socket = socket->Accept(NULL);
45     if (raw_socket) {
46       AsyncTCPSocket* packet_socket = new AsyncTCPSocket(raw_socket, false);
47       packet_socket->SignalReadPacket.connect(this, &TestEchoServer::OnPacket);
48       packet_socket->SignalClose.connect(this, &TestEchoServer::OnClose);
49       client_sockets_.push_back(packet_socket);
50     }
51   }
OnPacket(AsyncPacketSocket * socket,const char * buf,size_t size,const SocketAddress & remote_addr,const PacketTime & packet_time)52   void OnPacket(AsyncPacketSocket* socket, const char* buf, size_t size,
53                 const SocketAddress& remote_addr,
54                 const PacketTime& packet_time) {
55     rtc::PacketOptions options;
56     socket->Send(buf, size, options);
57   }
OnClose(AsyncPacketSocket * socket,int err)58   void OnClose(AsyncPacketSocket* socket, int err) {
59     ClientList::iterator it =
60         std::find(client_sockets_.begin(), client_sockets_.end(), socket);
61     client_sockets_.erase(it);
62     Thread::Current()->Dispose(socket);
63   }
64 
65   typedef std::list<AsyncTCPSocket*> ClientList;
66   scoped_ptr<AsyncSocket> server_socket_;
67   ClientList client_sockets_;
68   RTC_DISALLOW_COPY_AND_ASSIGN(TestEchoServer);
69 };
70 
71 }  // namespace rtc
72 
73 #endif  // WEBRTC_BASE_TESTECHOSERVER_H_
74