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 DISALLOW_EVIL_CONSTRUCTORS(TestEchoServer); 69 }; 70 71 } // namespace rtc 72 73 #endif // WEBRTC_BASE_TESTECHOSERVER_H_ 74