diff -r 000000000000 -r 7f656887cf89 plugins/consoles/rcons/server/win32/ServerSocket.cpp --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/plugins/consoles/rcons/server/win32/ServerSocket.cpp Wed Jun 23 15:52:26 2010 +0100 @@ -0,0 +1,86 @@ +// ServerSocket.cpp +// +// Copyright (c) 2010 Accenture. All rights reserved. +// This component and the accompanying materials are made available +// under the terms of the "Eclipse Public License v1.0" +// which accompanies this distribution, and is available +// at the URL "http://www.eclipse.org/legal/epl-v10.html". +// +// Initial Contributors: +// Accenture - Initial contribution +// + +#include "Misc.h" +#include "stdafx.h" +#include "console_host.h" +#include "ServerSocket.h" +#include "ClientSocket.h" + + +CServerSocket* CServerSocket::New(CWindow& aWindow) + { + std::auto_ptr self(new(EThrow) CServerSocket(aWindow)); + self->Construct(); + return self.release(); + } + +CServerSocket::CServerSocket(CWindow& aWindow) + : CSocket(aWindow), iClientAcceptor(NULL) + { + } + +void CServerSocket::Listen(unsigned short aPort, unsigned int aBacklogSize, MSocketClientAcceptor& aClientAcceptor) + { + SOCKADDR_IN sockAddr; + memset(&sockAddr, 0, sizeof(sockAddr)); + sockAddr.sin_family = AF_INET; + sockAddr.sin_addr.s_addr = htonl(INADDR_ANY); + sockAddr.sin_port = htons((u_short)aPort); + + if (bind(iSocket, (SOCKADDR*)&sockAddr, sizeof(sockAddr)) == SOCKET_ERROR) + { + throw KExceptionSocketBindFailed; + } + + if (listen(iSocket, aBacklogSize) == SOCKET_ERROR) + { + throw KExceptionSocketListenFailed; + } + + iClientAcceptor = &aClientAcceptor; + } + +CServerSocket::~CServerSocket() + { + } + +void CServerSocket::Construct() + { + CSocket::Construct(); + + if (WSAAsyncSelect(iSocket, iWindow.Handle(), KSocketMessage, FD_ACCEPT | FD_CLOSE) == SOCKET_ERROR) + { + throw KExceptionSocketSelectFailed; + } + } + +void CServerSocket::HandleAccept() + { + ASSERT(iClientAcceptor); + SOCKET newSocket = WSAAccept(iSocket, NULL, NULL, NULL, 0); + if (newSocket == INVALID_SOCKET) + { + int err = WSAGetLastError(); + throw KExceptionSocketAcceptFailed; + } + std::auto_ptr clientSocket(CClientSocket::New(iWindow, newSocket)); + iClientAcceptor->HandleNewClient(*clientSocket.get()); + clientSocket.release(); + } + +LRESULT CServerSocket::HandleSocketMessage(LPARAM /*aLParam*/) + { + HandleAccept(); + return 0; + } +