Simple, Robust and Expandable Winsock Server for Multiple Clients with Easy to Add New Services

/*
Several wrapper class to automatically
free the resources in the destructor on the exit
from a function. They are common for the
client and the server.

Author Vadim Motorine  
email vad@friendly-ware.com
Copyright (C) 2006 Vadim Motorine 
*/

#pragma once

#ifndef VM_SAEWINSCK_H
#define VM_SAEWINSCK_H
#include "STDAFXCmmn.H"
#include "winsock2.h"
#include "../vmClnt/vmClnt/TaskPasportCr.h"
#include <crtdbg.h>

/* It automatically frees the resources allocated for
the WinSock library*/
class CSafeWSAStrtp
{
	bool isinit;
public:
	CSafeWSAStrtp():isinit(false){;}
// after the call of this function the cleanup is automatic
	void GetProtect(){
		isinit = true;}
	bool IsInit(){return isinit;}
	~CSafeWSAStrtp(){Detach();}
	void Detach(){
		if(isinit){
			WSACleanup();
			isinit = false;
		}
	;}
};

/*
If we define the variable of this class at the
the body  beginning of a function, this
function will be protected by the critical 
section. The critical secion should be initalized
slsewhere and supplied via the constructor.
*/
class CSafeCrctclSctn
{
	LPCRITICAL_SECTION csp;
public:
	CSafeCrctclSctn(LPCRITICAL_SECTION cspi):csp(0){csp = cspi;
	EnterCriticalSection(cspi);}
	~CSafeCrctclSctn(){LeaveCriticalSection(csp);}
//
};

//This class is actually used in the client only.
//The class variable should be defined at the beginning
//of the thread function.  Then the destuctor automatically
//reports about the thread exit.
class CSafeExtThrd
{
	CTaskPasportCr* tskPsprt;//the class to communicate with the thread
public:
	CSafeExtThrd(CTaskPasportCr* tskPsprtI):tskPsprt(tskPsprtI){;}
	~CSafeExtThrd(){
		try{//Though it is improbable,  we may delete tskPsprt before the 
		//the thread stops.
			tskPsprt->SetStopped();
			tskPsprt->SetIsItCnnctd(false);
		}
		catch(...){
		
		}
	}
};

//This class  automatically closes the socket in the destructor
//on the exit from the function
class CSafeSckt
{
	SOCKET sct;
public:
	CSafeSckt():sct(INVALID_SOCKET){;}
//The operator "=" puts the socket under the protection	
	void operator=(SOCKET scti){
		if(sct != INVALID_SOCKET){
			return;}
		sct = scti;}
//The destructor closes the socket		
	~CSafeSckt(){Detach();}
	void Detach(){if(sct != INVALID_SOCKET){
		int err;
		err = ::shutdown(sct,SD_BOTH);//returns 0 if no error
		err = ::closesocket(sct);//returns 0 if no error
		sct = INVALID_SOCKET;}
	;}

};

#endif