【网络编程】之十二、wsaeventselect+线程池 服务器实现

最新推荐文章于 2020-08-17 23:45:32 发布
最新推荐文章于 2020-08-17 23:45:32 发布
阅读量2k 收藏 6
点赞数 1
分类专栏： WinSocket
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/Jofranks/article/details/9392023
版权 
#include<WinSock2.h>
#include<iostream>
using namespace std;


#pragma comment(lib, "WS2_32.lib")

typedef struct _SOCKET_OBJ{
	SOCKET s;
	HANDLE event;
	sockaddr_in addrRemote;
	_SOCKET_OBJ *pNext;
}SOCKET_OBJ, *PSOCKET_OBJ;

typedef struct _THREAD_OBJ{
	HANDLE events[WSA_MAXIMUM_WAIT_EVENTS];
	int nSocketCount;
	PSOCKET_OBJ pSocketHeader;
	PSOCKET_OBJ pSocketTail;
	CRITICAL_SECTION cs;
	_THREAD_OBJ *pNext;
}THREAD_OBJ, *PTHREAD_OBJ;

PTHREAD_OBJ g_pThreadList;
CRITICAL_SECTION g_cs;

LONG g_nTatolConnections;
LONG g_nCurrentConnections;

DWORD WINAPI ServerThread(LPVOID lpParam);
//******************************************************************************//
PSOCKET_OBJ GetSocketObj(SOCKET s)
{
	PSOCKET_OBJ pSocket = (PSOCKET_OBJ)::GlobalAlloc(GPTR, sizeof(SOCKET_OBJ));
	if(pSocket != NULL)
	{
		pSocket->s = s;
		pSocket->event = ::WSACreateEvent();
	}

	return pSocket;
}

void FreeSocketObj(PSOCKET_OBJ pSocket)
{
	::CloseHandle(pSocket->event);
	if(pSocket->s != INVALID_SOCKET)
	{
		closesocket(pSocket->s);
	}
	::GlobalFree(pSocket);
}

//*************************************************************************//

PTHREAD_OBJ GetThreadObj()
{
	PTHREAD_OBJ pThread = (PTHREAD_OBJ)::GlobalAlloc(GPTR, sizeof(THREAD_OBJ));
	if(pThread != NULL)
	{
		::InitializeCriticalSection(&pThread->cs);

		pThread->events[0] = ::WSACreateEvent();

		::EnterCriticalSection(&g_cs);
		pThread->pNext = g_pThreadList;
		g_pThreadList = pThread;
		::LeaveCriticalSection(&g_cs);
	}

	return pThread;
}

void FreeThreadObj(PTHREAD_OBJ pThread)
{
	::EnterCriticalSection(&g_cs);
	PTHREAD_OBJ p = g_pThreadList;
	if(p == pThread)
	{
		g_pThreadList = p->pNext;
	}
	else
	{
		while(p != NULL && p->pNext != pThread)
		{
			p = p->pNext;
		}
		if(p != NULL)
		{
			p->pNext = pThread->pNext;
		}
	}
	::LeaveCriticalSection(&g_cs);

	::CloseHandle(pThread->events[0]);
	::DeleteCriticalSection(&pThread->cs);
	::GlobalFree(pThread);
}


void RebulidArray(PTHREAD_OBJ pThread)
{
	::EnterCriticalSection(&pThread->cs);
	PSOCKET_OBJ pSocket = pThread->pSocketHeader;
	int n = 1;
	while(pSocket != NULL)
	{
		pThread->events[n++] = pSocket->event;
		pSocket = pSocket->pNext;
	}

	::LeaveCriticalSection(&pThread->cs);
}

//********************************************************************//

BOOL insertSocketObj(PTHREAD_OBJ pThread, PSOCKET_OBJ pSocket)
{
	BOOL bRet = FALSE;
	::EnterCriticalSection(&pThread->cs);
	if(pThread->nSocketCount < WSA_MAXIMUM_WAIT_EVENTS - 1)
	{
		if(pThread->pSocketHeader == NULL)
		{
			pThread->pSocketHeader = pThread->pSocketTail = pSocket;
		}
		else
		{
			pThread->pSocketTail->pNext = pSocket;
			pThread->pSocketTail = pSocket;
		}
		pThread->nSocketCount++;
		bRet = TRUE;
	}

	::LeaveCriticalSection(&pThread->cs);

	if(bRet)
	{
		::InterlockedIncrement(&g_nTatolConnections);
		::InterlockedIncrement(&g_nCurrentConnections);
	}
	return bRet;
}

void AssignToFreeThread(PSOCKET_OBJ pSocket)
{
	pSocket->pNext = NULL;
	::EnterCriticalSection(&g_cs);
	PTHREAD_OBJ pThread = g_pThreadList;

	while(pThread != NULL)
	{
		if(insertSocketObj(pThread, pSocket))
			break;
		pThread = pThread->pNext;
	}


	if(pThread == NULL)
	{
		pThread = GetThreadObj();
		insertSocketObj(pThread, pSocket);
		::CreateThread(NULL, 0, ServerThread, pThread, 0, NULL);

	}

	::LeaveCriticalSection(&g_cs);

	::WSASetEvent(pThread->events[0]);
}


void RemoveSocketObj(PTHREAD_OBJ pThread, PSOCKET_OBJ pSocket)
{
	::EnterCriticalSection(&pThread->cs);

	PSOCKET_OBJ pTest = pThread->pSocketHeader;
	if(pTest == pSocket)
	{
		if(pThread->pSocketHeader == pThread->pSocketTail)
			pThread->pSocketTail = pThread->pSocketHeader = pTest->pNext;
		else
			pThread->pSocketHeader = pTest->pNext;
	}
	else
	{
		while(pTest != NULL && pTest->pNext != pSocket)
			pTest = pTest->pNext;
		if(pTest != NULL)
		{
			if(pThread->pSocketTail == pSocket)
				pThread->pSocketTail = pTest;
			pTest->pNext = pSocket->pNext;
		}
	}

	pThread->nSocketCount--;

	::LeaveCriticalSection(&pThread->cs);


	::WSASetEvent(pThread->events[0]);
	::InterlockedDecrement(&g_nCurrentConnections);
}

//********************************************************************//
PSOCKET_OBJ FindSocketObj(PTHREAD_OBJ pThread, int nIndex)
{
	PSOCKET_OBJ pSocket = pThread->pSocketHeader;
	while(--nIndex)
	{
		if(pSocket == NULL)
			return NULL;
		pSocket = pSocket->pNext;
	}
	return pSocket;
}

BOOL HandleIO(PTHREAD_OBJ pThread, PSOCKET_OBJ pSocket)
{
	WSANETWORKEVENTS event;
	::WSAEnumNetworkEvents(pSocket->s, pSocket->event, &event);
	do{
		if(event.lNetworkEvents & FD_READ)
		{
			if(event.iErrorCode[FD_READ_BIT] == 0)
			{
				char szText[256];
				int nRecv = ::recv(pSocket->s, szText, strlen(szText), 0);
				if(nRecv > 0)
				{
					szText[nRecv] = '\0';
					cout << "接收到数据：" << szText << endl;
				}
			}
			else
				break;
		}
		else
		{
			if(event.lNetworkEvents &FD_CLOSE)
			{
				break;
			}
			else
			{
				if(event.lNetworkEvents & FD_WRITE)
				{
					if(event.iErrorCode[FD_READ_BIT] == 0)
					{
						char szText[256];
						int nRecv = ::recv(pSocket->s, szText, strlen(szText), 0);
						if(nRecv > 0)
						{
							szText[nRecv] = '\0';
							cout << "接收到数据：" << szText << endl;
						}
					}
					else
						break;
				}
			}
		}
		return TRUE;

	}while(FALSE);


	RemoveSocketObj(pThread, pSocket);
	FreeSocketObj(pSocket);
	return FALSE;
}


DWORD WINAPI ServerThread(LPVOID lpParam)
{
	PTHREAD_OBJ pThread = (PTHREAD_OBJ)lpParam;
	while(TRUE)
	{
		int nIndex = ::WSAWaitForMultipleEvents(pThread->nSocketCount + 1, pThread->events, FALSE, WSA_INFINITE, FALSE);
		nIndex = nIndex - WSA_WAIT_EVENT_0;

		for(int i = nIndex; i < pThread->nSocketCount + 1; ++i)
		{
			nIndex = ::WSAWaitForMultipleEvents(1, &pThread->events[i], TRUE, 1000, FALSE);
			if(nIndex == WSA_WAIT_FAILED || nIndex == WSA_WAIT_TIMEOUT)
			{
				continue;
			}
			else
			{
				if(i == 0)
				{
					RebulidArray(pThread);

					if(pThread->nSocketCount == 0)
					{
						FreeThreadObj(pThread);
						return 0;
					}
					::WSAResetEvent(pThread->events[0]);
				}
				else
				{
					PSOCKET_OBJ pSocket = (PSOCKET_OBJ)FindSocketObj(pThread, i);
					if(pSocket != NULL)
					{
						if(!HandleIO(pThread, pSocket))
							RebulidArray(pThread);
					}
					else
						cout << "Unable to find socket object" << endl;
				}
			}
		}
	}
	return 0;
}


//******************************************************************************//

int main(void)
{

	WSADATA wsaData;  
	WORD sockVersion = MAKEWORD(2,0);//指定版本号  
	::WSAStartup(sockVersion, &wsaData);//载入winsock的dll  
	//创建套接字基于TCP  
	SOCKET sListen = ::socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);  
	if(sListen == INVALID_SOCKET)  
	{  
		printf("error");  
		::WSACleanup();//清理，释放资源  
		return 0;  
	}  

	sockaddr_in sin;  
	sin.sin_family = AF_INET;  
	sin.sin_port = htons(8888);//端口号8888  
	sin.sin_addr.S_un.S_addr = INADDR_ANY;//地址全是0，也就是所有的地址  
	//绑定socket  
	if(::bind(sListen, (LPSOCKADDR)&sin, sizeof(sin)) == SOCKET_ERROR)  
	{  
		printf("error");  
		::WSACleanup();//清理释放资源  
		return 0;  
	}  
	//监听socket  
	if(::listen(sListen, 2) == SOCKET_ERROR)  
	{  
		printf("error");  
		::WSACleanup();//释放资源  
		return 0;  
	}  

	WSAEVENT event = ::WSACreateEvent();
	::WSAEventSelect(sListen, event, FD_ACCEPT | FD_CLOSE);
	::InitializeCriticalSection(&g_cs);

	//处理请求
	while(TRUE)
	{
		int nRet = ::WaitForSingleObject(event, 5 * 1000);
		if(nRet == WAIT_FAILED)
		{
			cout << "failed waitforsingleobject" << endl;
			break;
		}
		else if(nRet == WSA_WAIT_TIMEOUT)
		{
			cout << endl;
			cout << " tatolconnections:" << g_nTatolConnections << endl;
			cout << " currentconnections: " << g_nCurrentConnections << endl;
			continue;
		}
		else
		{
			::ResetEvent(event);   //新连接

			while(TRUE)
			{
				sockaddr_in si;
				int nLen = sizeof(si);
				SOCKET sNew = ::accept(sListen, (sockaddr*)&si, &nLen);
				if(sNew == SOCKET_ERROR)
					break;
				PSOCKET_OBJ pSocket = GetSocketObj(sNew);
				pSocket->addrRemote = si;
				::WSAEventSelect(pSocket->s, pSocket->event, FD_READ | FD_CLOSE | FD_WRITE);
				AssignToFreeThread(pSocket);
			}
		}
	}

	::DeleteCriticalSection(&g_cs);
	::WSACleanup();
	return 0;
}

jofranks
关注
专栏目录
WsaEventSelect封装的类，多线程
yuanxiaobo007的专栏
03-08 1876
最近工作上需要一个小的tcp服务器，就想到了用WsaEventSelect去管理socket, 遂在网上找了下资料，有个多线程的Wsaeventselect的C源码，就想将其封装为类，用了两天（好吧，水平有限）。其中有个问题，是客户端如果closesocket太快的话，服务端这边会收不到FD_CLOSE消息，在客户端closesocket前加了个sleep就正常了，查了资料貌似是和tcp关闭时候的
Windows Sockets网络编程（3）WSAEventSelect模型开发
机器学习
07-25 2232
WSAEventSelect模型是非阻塞的，该模型允许在一个或者多个套接字上接收以事件为基础的网络事件通知。Windows Sockets应用程序在创建套接字后，调用WSAEventSelect()函数，将一个事件对象与网络事件集合关联在一起。当网络事件发生时，应用程序以事件的形式接收网络事件通知。 创建TCP 这里不再赘述了，能来到本文的，相信基本功已经不用多讲了。实在不明白的可以阅读《Windows Sockets网络编程（0）TCP In Action》一文，该文详细的叙述了TCP创建的整个过
WSAEventSelect 线程池 实现服务器示例
09-25
用WSAEventSelect结合线程池实现的服务器示例.希望能够帮助到那些热爱网络,但又迷失方向的朋友.
一个基于线程池的WSAEventSelect模型服务器设计
weixin_33716941的博客
07-08 167
//WSAEventSelect.h #include "CInitSocket.h" typedef struct _SOCKET_OBJ { SOCKET sock; //套接字句柄 HANDLE hEvent; //与次套接字相关的事件对象句柄 sockaddr_in addrRemote; //客户端地址信息 _SOCKET_OBJ *pN...
基于WSAEventSelect的多线程服务器C++程序
01-05
总之，基于`WSAEventSelect`的多线程服务器程序设计是一项复杂而重要的任务，它涉及到网络编程、并发控制和资源管理等多个方面。通过合理的设计和实现，我们可以构建出一个高效、稳定的C++服务器，能够处理大量并发...
windows 网络通信 socket编程详解 快速入门
04-08
在Windows平台上进行网络通信，Socket编程是不可或缺的一部分。Socket提供了进程间通信的一种方式，尤其在网络环境中，它使得不同计算机上的程序能够相互通信。本文主要介绍Windows下的Socket编程，包括TCP/IP体系...
vc网络编程源代码（十二章完整源代码）
08-14
8. **网络编程设计模式**：例如，服务器的并发模型（单线程、多线程、线程池、异步I/O等），客户端的同步与异步设计，以及异常处理机制，这些都是提升网络程序设计水平的重要知识点。 通过深入研究这些源代码，...
WSAEventSelect模型例子（使用线程池）
yhtppp的专栏
04-08 1995
<br />以下来自《Windows 网络与通信程序设计（第2版）》3.4.3<br /> 1.<br /> WinDef.h中的声明：<br /><br /> #ifndef NULL<br /> #ifdef __cplusplus<br />#define NULL    0<br /> #else<br />#define NULL    ((void *)0)<br /> #endif<br /> #endif<br /><br /><br /> 2.<br /> WinNT.h
WSAEventSelect多线程服务器
10-01
由于王艳平老师的代码用纯C实现 全局函数太多 阅读难以理解 本人使用C++重写 思路清晰注释到位 特点： 1 使用多线程 目的是监控连接数量和线程数目 2 界面基于对话框 使用WSAAsyncSelect和WSAEventSelect两个模型 这是因为使用了界面 前者对消息开放 后者完成网络服务 3 有文档 有测试程序 4 代码只有200多行 比书上少了100行 容易阅读多了 三个类SockectOjb ThreadOjb ThreadPool ">由于王艳平老师的代码用纯C实现 全局函数太多 阅读难以理解 本人使用C++重写 思路清晰注释到位 特点： 1 使用多线程 目的是监控连接数量和线程数目 2 界面基于对话框 使用WSAAsyncSelect和WSAEventSelect两个模型 这 [更多]
多线程下的WSAEventSelect模型
08-10
http://blog.csdn.net/wangjieest/article/details/7042108 http://blog.csdn.net/ithzhang/article/details/8476556
Winsock模型之WSAEventSelect(事件通知)
05-25
Winsock模型之WSAEventSelect(事件通知)
【网络编程】之十二、wsaeventselect+线程池 服务器实现
liujiayu2的专栏
08-21 651
[cpp] view plaincopy #include   #include   using namespace std;         #pragma comment(lib, "WS2_32.lib")      typedef struct _SOCKET_OBJ{       SOCKET s;       HANDLE event; 
WSAEventSelect多线程服务器实现示例
生命因你而精彩！明天更美好！
10-12 4754
从CSDN的一个xd那里看到这个代码，觉得对WSAEventSelect模型的多线程实现做的非常好，代码行文风格和代码质量都是非常不错。我做了些小的改动学习之用。如果你也感兴趣不妨一起留言讨论讨论。
Windows WSAEventSelect 网络通信模型
qq_34462436的博客
08-17 815
文章目录Windows WSAEventSelect 网络通信模型WSAEventSelect模型介绍WSAEventSelect 用于服务器端**WSAWaitForMultipleEvents**WSAEnumNetworkEvents函数完整代码 Windows WSAEventSelect 网络通信模型 WSAEventSelect 网络通信模型是 Windows 系统上常用的一种异步 socket 通信模型，下面来详细介绍下其用法。 WSAEventSelect模型介绍 ​ WSAEventSel
Windows socket之WSAEventSelect模型
系统运维
01-07 362
WSAEventSelect模型 WSAEventSelect模型是Windowssocekts提供的另一个有用异步IO模型。该模型允许在一个或多个套接字上接收以事件为基础的网络事件通知。Windowssockets应用程序可以通过调用WSAEventSelect函数，将一个事件与网络事件集合关联起来。当网络事件发生时，应用程序以事件的形式接收网络事件通知。 WSAEventSe...
WSAWaitForMultipleEvents的用法
热门推荐
mandi1829的博客
10-16 1万+
// wsawaitformultipleeventsTest.cpp : 定义控制台应用程序的入口点。 // //程序说明：这个示例程序演示了如使用winsock的等待事件函数。 /* 等事件函数的示例如下： WSAWaitForMultipleEvents(4, EventArray, FALSE, WSA_INFINITE, FALSE); 其中，第一个参数为事件的个数。第二个参
Windows服务器网络通信：完成端口实现与异步I/O模型解析
在Windows网络编程中，完成端口（Completion Port，简称CP）是一种高效的多线程I/O模型，尤其适用于高并发的服务器应用。实现一个基于完成端口的网络服务需要遵循一系列步骤，下面将详细解释这些步骤以及相关知识点...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值