Select网络模型

MySockt.h
#pragma once
#define  MAXNUM 64
#include<WinSock2.h>
#pragma comment(lib,"ws2_32.lib")
#define UM_USER WM_USER+1   //向窗口发消息
enum NETTYPE
{
	NETRECV,NETACCEPT,NETSEND
};
class CMySocket
{
public:
	CMySocket(void);
	~CMySocket(void);
public :
	static DWORD WINAPI ThreadAccept(LPVOID lpvoid);
	static DWORD WINAPI ThreadRecv(LPVOID lpvoid);
public :
	bool StartSevere(HWND m_hWnd,short nPort=1234,CString nIp=_T("127.0.0.1"));
	void StopSevere();
	bool SendData(TCHAR *tzbuf,int nsize);
	bool SelectSocket(SOCKET SocketWaiter,NETTYPE type);
public :
	HANDLE m_hThreadAccept;
	HANDLE m_hThreadRecv;
	HANDLE m_hEventQuit;
	HANDLE m_hThreadAcceptCheck;
	HANDLE m_hThreadRecvCheck;
	HWND m_hParrentWnd;
public :
	int m_num;
	SOCKET SocketListen;
	NETTYPE type;
	SOCKET SocketWaiter[MAXNUM];
};

MySocket.cpp
#include "stdafx.h"
#include "MySocket.h"


CMySocket::CMySocket(void)
{
	m_hThreadAccept = NULL;
	m_hThreadRecv= NULL;
	m_hEventQuit= NULL;
	m_hThreadAcceptCheck= NULL;
	m_hThreadRecvCheck= NULL;
	m_hParrentWnd=NULL;
	m_num=0;
	ZeroMemory(SocketWaiter,sizeof(SocketWaiter));
}


CMySocket::~CMySocket(void)
{
	StopSevere();
}
void CMySocket::StopSevere()
{
	WSACleanup();
	for(int i=0;i<m_num;i++)
	{
		if(SocketWaiter[i])
		{
			closesocket(SocketWaiter[i]);
			SocketWaiter[i]=NULL;
		}
	}
	if(SocketListen)
	{
		closesocket(SocketListen);
		SocketListen=NULL;
	}
	SetEvent(m_hEventQuit);
	if(WAIT_OBJECT_0 !=WaitForSingleObject(m_hThreadAcceptCheck,100))
	{
		::TerminateThread(m_hThreadAccept,-1);
		m_hThreadAccept=NULL;
	}
	if(WAIT_OBJECT_0 !=WaitForSingleObject(m_hThreadRecvCheck,100))
	{
		::TerminateThread(m_hThreadRecv,-1);
		m_hThreadRecv=NULL;
	}
	if(m_hEventQuit)
	{
		CloseHandle(m_hEventQuit);
		m_hEventQuit=NULL;
	}
	if(m_hThreadAcceptCheck)
	{
		CloseHandle(m_hThreadAcceptCheck);
		m_hThreadAcceptCheck=NULL;
	}
	if(m_hThreadRecvCheck)
	{
		CloseHandle(m_hThreadRecvCheck);
		m_hThreadRecvCheck=NULL;
	}
}
DWORD WINAPI CMySocket::ThreadAccept(LPVOID lpvoid)
{
	CMySocket *pthis=(CMySocket*)lpvoid;
	while(1)
	{
		if(  WAIT_OBJECT_0 ==  WaitForSingleObject(pthis->m_hEventQuit,100))
		{
			break;
		}
		pthis->SocketWaiter[pthis->m_num]=accept(pthis->SocketListen,NULL,NULL);
		if(INVALID_SOCKET==pthis->SocketWaiter[pthis->m_num])
		{
			pthis->StopSevere();
		}
		pthis->m_num++;
		TCHAR szbuf[1024]=_T("服务器链接成功");
		SendMessage(pthis->m_hParrentWnd,UM_USER,(WPARAM)szbuf,1024);
		//pthis->m_lstbox.AddString(szbuf);
	}
	SetEvent(pthis->m_hThreadAcceptCheck);
	return 0;
}
DWORD WINAPI CMySocket::ThreadRecv(LPVOID lpvoid)
{
	CMySocket *pthis=(CMySocket*)lpvoid;
	while(1)
	{
		if(  WAIT_OBJECT_0 ==  WaitForSingleObject(pthis->m_hEventQuit,100))
		{
			break;
		}
		char szbuf[1024]={0};
		TCHAR tzbuf[1024]={0};
		for(int i=0;i<pthis->m_num;i++)
		{		
			if(pthis->SelectSocket(pthis->SocketWaiter[i],NETRECV))
			{
				int nres =recv(pthis->SocketWaiter[i],szbuf,1024,0);
				if(0!=nres &&SOCKET_ERROR !=nres)
				{
				#ifdef  _UNICODE
					MultiByteToWideChar(CP_ACP,0,szbuf,-1,tzbuf,1024);
				#else
					strcpy_s(tzbuf,1024,szbuf);
				#endif
					SendMessage(pthis->m_hParrentWnd,UM_USER,(WPARAM)tzbuf,1024);
					//pthis->m_lstbox.AddString(tzbuf);
			}
			
			}
		}

	}
	SetEvent(pthis->m_hThreadRecvCheck);
	return 0;
}
bool CMySocket::StartSevere(HWND m_hWnd,short nPort,CString nIp)
{
	m_hParrentWnd=m_hWnd;
	//0.加载库
	WORD wVersionRequested;
	WSADATA wsaData;
	int err;
	wVersionRequested = MAKEWORD( 2, 2 );
	err = WSAStartup( wVersionRequested, &wsaData );
	if ( err != 0 ) 
	{
		return 0;
	}
 
	if ( LOBYTE( wsaData.wVersion ) != 2 || HIBYTE( wsaData.wVersion ) != 2 )
	{
		WSACleanup( );
		return 0; 
	}
	//1.创建socket与外界通信的接口
	SocketListen=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);
	if(INVALID_SOCKET==SocketListen)
	{
		StopSevere();
		return 0;
	}
	//2.bind
	sockaddr_in addrsock;
	addrsock.sin_family=AF_INET;
	addrsock.sin_port=htons(nPort);
	char szbuf[1024];
#ifdef _UNICODE
	WideCharToMultiByte(CP_ACP,0,nIp,-1,szbuf,1024,0,0);
#else
	strcpy_s(szbuf,1024,nIP);
#endif
	addrsock.sin_addr.s_addr=inet_addr(szbuf);
	int nsize=sizeof(sockaddr_in);
	if(SOCKET_ERROR==bind(SocketListen,(const struct sockaddr*)&addrsock,nsize))
	{
		StopSevere();
		return 0;
	}
	//3.listen 监听
	if(SOCKET_ERROR==listen(SocketListen,10))
	{
		StopSevere();
		return 0;
	}
	m_hEventQuit=::CreateEvent(NULL,TRUE,FALSE,NULL);
	m_hThreadAcceptCheck=::CreateEvent(NULL,TRUE,FALSE,NULL);
	m_hThreadRecvCheck=::CreateEvent(NULL,TRUE,FALSE,NULL);
	m_hThreadAccept=::CreateThread(NULL,0,&CMySocket::ThreadAccept,this,0,NULL);
	m_hThreadRecv=::CreateThread(NULL,0,&CMySocket::ThreadRecv,this,0,NULL);
	if(!m_hEventQuit||!m_hThreadAcceptCheck||!m_hThreadRecvCheck||!m_hThreadAccept||!m_hThreadRecv)
	{
		StopSevere();
		return 0;
	}
	return 1;
}
bool CMySocket::SendData(TCHAR *tzbuf,int nsize)
{
	if(!tzbuf||nsize<0)
	{
		return false;
	}
	char szbuf[1024]={0};
#ifdef _UNICODE
	WideCharToMultiByte(CP_ACP,0,tzbuf,-1,szbuf,1024,0,0);
#else
	strcpy_s(szbuf,1024,tzbuf);
#endif
	for(int i=0;i<m_num;i++)
	{	
		if(SOCKET_ERROR==send(SocketWaiter[i],szbuf,nsize,0))
		{
			return false;
		}
	}
	return true;
}
bool CMySocket::SelectSocket(SOCKET SocketWaiter,NETTYPE type)
{
	if(!SocketWaiter)
	{
		return false;
	}
	TIMEVAL tv;
	tv.tv_sec=0;
	tv.tv_usec=100;
	//定义一个集合
	fd_set fdsets;
	//清空集合
	FD_ZERO(&fdsets);
	//将SocketWaiter加入到集合
	FD_SET(SocketWaiter,&fdsets);
	//将集合交由select管理
	if(type==NETACCEPT||type==NETRECV)
	{
		select(0,&fdsets,0,0,&tv);
	}
	else if(type==NETSEND)
	{
		select(0,0,&fdsets,0,&tv);
	}
	//校验SocketWaiter是否在集合
	if(FD_ISSET(SocketWaiter,&fdsets))
	{
		return TRUE;
	}
	else
	{
		return FALSE;
	}
	return true;

}