diff -r 000000000000 -r 7f656887cf89 plugins/consoles/rcons/server/win32/Socket.cpp --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/plugins/consoles/rcons/server/win32/Socket.cpp Wed Jun 23 15:52:26 2010 +0100 @@ -0,0 +1,84 @@ +// Socket.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 "Socket.h" +#include "DynamicBuffer.h" + + +CSocket::CSocket(CWindow& aWindow) + : iWindow(aWindow), iSocket(INVALID_SOCKET), iClosureObserver(NULL) + { + } + +CSocket::CSocket(CWindow& aWindow, SOCKET aSocket) + : iWindow(aWindow), iSocket(aSocket), iClosureObserver(NULL) + { + } + +CSocket::~CSocket() + { + if (iSocket != INVALID_SOCKET) + { + closesocket(iSocket); + } + + iWindow.RemoveMessageHandler(*this); + } + +void CSocket::SetClosureObserver(MSocketClosureObserver* aClosureObserver) + { + iClosureObserver = aClosureObserver; + } + +void CSocket::Construct() + { + if (iSocket == INVALID_SOCKET) + { + if ((iSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP)) == INVALID_SOCKET) + { + throw KExceptionSocketConstructFailed; + } + BOOL b = 1; + setsockopt(iSocket, IPPROTO_TCP, TCP_NODELAY, (char*)&b, sizeof(BOOL)); + } + + iWindow.AddMessageHandler(*this); + } + +void CSocket::HandleClosure() + { + if (iClosureObserver) + { + iClosureObserver->HandleSocketClosure(*this); + } + } + +LRESULT CSocket::HandleWindowMessage(UINT aMessage, WPARAM aWParam, LPARAM aLParam) + { + if ((aMessage == KSocketMessage) && (aWParam == iSocket)) + { + if (WSAGETSELECTEVENT(aLParam) == FD_CLOSE) + { + HandleClosure(); + } + else + { + HandleSocketMessage(aLParam); + } + return 0; + } + return 1; + } +