Click here to Skip to main content
Click here to Skip to main content
Add your own
alternative version

Push Framework - A C++ toolkit for high performance server development

, 23 May 2012
Write asynchronous, multithreaded servers in a few lines of code. Monitor realtime activity with a deploy-only dashboard.
Prize winner in Competition "Best C++/MFC article of March 2011"
ChatApplication.zip
ChatAPI
ChatClient
res
.svn
entries
prop-base
ChatClient.ico.svn-base
props
text-base
ChatClient.ico.svn-base
ChatClient.rc2.svn-base
tmp
prop-base
props
text-base
ChatClient.ico
ChatPackets
ChatProtocol
ChatServer
ChatServer.vcproj.INTERNAL.Ahmed.Charfeddine.user
output
ChatServer.ini
TCPSocket
TCPSocket.zip
XMLProtocol
XMLProtocol.zip
ChatRobots.zip
ChatRobots
ChatRobots.ini
ProtoBufExample.zip
ProtoBufExampleClient
ProtoBufExampleProtocol
requests.pb.cc
responses.pb.cc
ProtoBufExampleServer
PushFramework_Essentials.zip
include
PushFramework_Sources.zip
private
QoS.zip
QoSExampleClient
QoSExampleProtocol
QoSExampleServer
/********************************************************************
	File :			Listener.cpp
	Creation date :	2010/6/27
		
	License :			Copyright 2010 Ahmed Charfeddine, http://www.pushframework.com

				   Licensed under the Apache License, Version 2.0 (the "License");
				   you may not use this file except in compliance with the License.
				   You may obtain a copy of the License at
				
					   http://www.apache.org/licenses/LICENSE-2.0
				
				   Unless required by applicable law or agreed to in writing, software
				   distributed under the License is distributed on an "AS IS" BASIS,
				   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
				   See the License for the specific language governing permissions and
				   limitations under the License.
	
	
*********************************************************************/
#include "StdAfx.h"
#include "Listener.h"



namespace PushFramework{


Listener::Listener()
{
}
Listener::~Listener(void)
{
}


bool Listener::startListening()
{
	//Create the socket :
	hSocket = WSASocket(AF_INET, SOCK_STREAM, 0, NULL, 0, WSA_FLAG_OVERLAPPED);
	if (hSocket == INVALID_SOCKET)
	{
		return false;
	}

	//Prepare address to bind to : 
	SOCKADDR_IN		saServer;		
	saServer.sin_port = htons(uPort);
	saServer.sin_family = AF_INET;
	if(options.interfaceAddress != NULL)
	{
		u_long nInterfaceAddr = inet_addr(options.interfaceAddress);
		if (nInterfaceAddr != INADDR_NONE){
			closesocket(hSocket);
			return false;
		}
		saServer.sin_addr.s_addr = nInterfaceAddr;
	}
	else
		saServer.sin_addr.s_addr = INADDR_ANY;


	//Bind the socket :
	int nRet = bind(hSocket, (LPSOCKADDR)&saServer, sizeof(struct sockaddr));
	if (nRet == SOCKET_ERROR)
	{
		closesocket(hSocket);
		return false;
	}


	//Create socket IO event
	hSocketEvent = WSACreateEvent();
	if (hSocketEvent == WSA_INVALID_EVENT)
	{
		closesocket(hSocket);
		return false;
	}


	//I am interested in accept event only :
	nRet = WSAEventSelect(hSocket, hSocketEvent, FD_ACCEPT);
	if (nRet == SOCKET_ERROR)
	{
		WSACloseEvent(hSocketEvent);
		closesocket(hSocket);
		return false;
	}

	
	//Set socket to listening mode :
	nRet = ::listen(hSocket, options.listeningBackLog);
	if (nRet == SOCKET_ERROR)
	{
		WSACloseEvent(hSocketEvent);
		closesocket(hSocket);
		return false;
	}


	//Create kill event used by ::stop to terminate the listening thread
	hKillEvent	= CreateEvent(NULL, TRUE, FALSE, NULL);


	UINT	dwThreadId = 0;
	hThread = (HANDLE)_beginthreadex(NULL, // Security
		0,					// default stack size
		threadProc,  // thread proc
		(void*) this,	    
		0,					// init flag
		&dwThreadId);	// 
	
	
	if (hThread == INVALID_HANDLE_VALUE)
	{
		WSACloseEvent(hSocketEvent);
		closesocket(hSocket);
		CloseHandle(hKillEvent);
		return false;
	}
	return true;
}

void Listener::stopListening()
{
	::SetEvent(hKillEvent);

	WaitForSingleObject(hThread, INFINITE);

	CloseHandle(hThread);
	CloseHandle(hKillEvent);
	closesocket(hSocket);	
	WSACloseEvent(hSocketEvent);
}

unsigned __stdcall Listener::threadProc( LPVOID lpVoid )
{
	Listener* pThis = reinterpret_cast<Listener*>(lpVoid);
	pThis->doListen();
	return 0;
}

bool Listener::acceptConnectionRequest()
{
	// accept socket :
	SOCKADDR_IN	SockAddr;
	int nLen = sizeof(SOCKADDR_IN);
	SOCKET	clientSocket = accept(hSocket, (LPSOCKADDR)&SockAddr, &nLen); 
	if (clientSocket == INVALID_SOCKET)
		return false;
	
	//TCP no delay option :
	const char optValue = options.isTcpNoDelay ? 1 : 0;
	int nRet = setsockopt(clientSocket, IPPROTO_TCP, TCP_NODELAY, &optValue, sizeof(char));
	if (nRet == -1)
	{
		shutdown(clientSocket, 2);
		closesocket(clientSocket);
		return true;
	}

	//Start servicing new connection
	if (!handleAcceptedSocket(clientSocket, SockAddr))
	{
		shutdown(hSocket, 2);
		closesocket(hSocket);
		return true;
	}
	return true;
}

void Listener::setListeningPort( unsigned int uPort )
{
	this->uPort = uPort;
}

void Listener::doListen()
{
	HANDLE			hWaits[2];
	hWaits[0]		= hKillEvent;
	hWaits[1]		= hSocketEvent;


	WSANETWORKEVENTS events;
	while(true)
	{
		// Wait for either the kill event or  an accept event :
		DWORD dwRet = WSAWaitForMultipleEvents(2, hWaits, FALSE, INFINITE, FALSE);

		//If kill event then quit loop :
		if (dwRet == WAIT_OBJECT_0)
		{
			break; 
		}

		int nRet = WSAEnumNetworkEvents(hSocket, hSocketEvent, &events);
		if (nRet == SOCKET_ERROR)
		{
			//Error
			break;
		}

		//Deal with the new incoming connection
		if ((events.lNetworkEvents & FD_ACCEPT) && (events.iErrorCode[FD_ACCEPT_BIT] == 0))
		{
			if(!acceptConnectionRequest())
				break;
		}
		else
			break;		
	} 
}

ListenerOptions& Listener::getOptions()
{
	return options;
}

}

By viewing downloads associated with this article you agree to the Terms of Service and the article's licence.

If a file you wish to view isn't highlighted, and is a text file (not binary), please let us know and we'll add colourisation support for it.

License

This article, along with any associated source code and files, is licensed under The Apache License, Version 2.0

About the Author

Ahmed Charfeddine
Technical Lead
Tunisia Tunisia
Engineer in telecommunications, C++ developer and Amateur Chess Programmer.
Websites:
http://www.pushframework.com
http://www.batchdocument.com
http://www.virtualglobe3d.com
http://www.chesscomposer.com
 
I am open to any collaboration or freelance project.
Follow on   Twitter

| Advertise | Privacy | Mobile
Web02 | 2.8.140721.1 | Last Updated 23 May 2012
Article Copyright 2011 by Ahmed Charfeddine
Everything else Copyright © CodeProject, 1999-2014
Terms of Service
Layout: fixed | fluid