plugins/consoles/rcons/server/win32/ServerSocket.cpp
changeset 0 7f656887cf89
--- /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<CServerSocket> 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<CClientSocket> clientSocket(CClientSocket::New(iWindow, newSocket));
+	iClientAcceptor->HandleNewClient(*clientSocket.get());
+	clientSocket.release();
+	}
+
+LRESULT CServerSocket::HandleSocketMessage(LPARAM /*aLParam*/)
+	{
+	HandleAccept();
+	return 0;
+	}
+