| // Copyright 2008 Google Inc. All Rights Reserved. |
| |
| |
| #ifndef TALK_BASE_TESTECHOSERVER_H_ |
| #define TALK_BASE_TESTECHOSERVER_H_ |
| |
| #include <list> |
| #include "talk/base/asynctcpsocket.h" |
| #include "talk/base/socketaddress.h" |
| #include "talk/base/sigslot.h" |
| #include "talk/base/thread.h" |
| |
| namespace talk_base { |
| |
| // 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) |
| : server_socket_(thread->socketserver()->CreateAsyncSocket(SOCK_STREAM)) { |
| server_socket_->Bind(addr); |
| server_socket_->Listen(5); |
| server_socket_->SignalReadEvent.connect(this, &TestEchoServer::OnAccept); |
| } |
| ~TestEchoServer() { |
| for (ClientList::iterator it = client_sockets_.begin(); |
| it != client_sockets_.end(); ++it) { |
| delete *it; |
| } |
| } |
| |
| SocketAddress address() const { return server_socket_->GetLocalAddress(); } |
| |
| private: |
| void OnAccept(AsyncSocket* socket) { |
| AsyncSocket* raw_socket = socket->Accept(NULL); |
| if (raw_socket) { |
| AsyncTCPSocket* packet_socket = new AsyncTCPSocket(raw_socket, false); |
| packet_socket->SignalReadPacket.connect(this, &TestEchoServer::OnPacket); |
| packet_socket->SignalClose.connect(this, &TestEchoServer::OnClose); |
| client_sockets_.push_back(packet_socket); |
| } |
| } |
| void OnPacket(AsyncPacketSocket* socket, const char* buf, size_t size, |
| const SocketAddress& remote_addr) { |
| socket->Send(buf, size); |
| } |
| void OnClose(AsyncPacketSocket* socket, int err) { |
| ClientList::iterator it = |
| std::find(client_sockets_.begin(), client_sockets_.end(), socket); |
| client_sockets_.erase(it); |
| Thread::Current()->Dispose(socket); |
| } |
| |
| typedef std::list<AsyncTCPSocket*> ClientList; |
| scoped_ptr<AsyncSocket> server_socket_; |
| ClientList client_sockets_; |
| DISALLOW_EVIL_CONSTRUCTORS(TestEchoServer); |
| }; |
| |
| } // namespace talk_base |
| |
| #endif // TALK_BASE_TESTECHOSERVER_H_ |