完成端口通信例子



代码贴上来以后慢慢修改

服务端代码：

#include <WinSock2.h>

#define BUFFER_SIZE 1024
#define OP_READ    18
#define OP_WRITE   28
#define OP_ACCEPT  38
#define CHECK_CODE 0x010110

BOOL bStopThread = false;

typedef struct _PER_HANDLE_DATA
{
	SOCKET s;
	sockaddr_in addr;      // 客户端地址
	char buf[BUFFER_SIZE];
	int nOperationType;
}PER_HANDLE_DATA, *PPER_HANDLE_DATA;

#pragma pack(1)
typedef struct MsgAsk
{
	int iCode;
	int iBodySize;
	char szBuffer[32];
}MSG_ASK, *PMSG_ASK;

typedef struct MsgBody
{
	int iBodySize;
	int iOpType;
	char szBuffer[64];
}MSG_BODY, *PMSG_BODY;

typedef struct MsgAck
{
	int  iCheckCode;
	char szBuffer[32];
}MSG_ACK, *PMSG_ACK;
#pragma pack()

DWORD WINAPI ServerThread(LPVOID lpParam)
{
	// 得到完成端口句柄
	HANDLE           hCompletion = (HANDLE)lpParam;
	DWORD            dwTrans;
	PPER_HANDLE_DATA pPerHandle;
	OVERLAPPED*      pOverLapped;

	while (!bStopThread)
	{
		// 在关联到此完成端口的所有套接字上等待I/O完成
		BOOL bOK = ::GetQueuedCompletionStatus(hCompletion,
			&dwTrans, (PULONG_PTR)&pPerHandle, &pOverLapped, WSA_INFINITE);
		if (!bOK)
		{
			::closesocket(pPerHandle->s);
			::GlobalFree(pPerHandle);
			::GlobalFree(pOverLapped);
			continue;
		}
		switch(pPerHandle->nOperationType)
		{
		case OP_READ:
			{
				MSG_ASK msgAsk = {0};
				memcpy(&msgAsk, pPerHandle->buf, sizeof(msgAsk));
				if (msgAsk.iCode != CHECK_CODE
					|| msgAsk.iBodySize != sizeof(msgAsk))
				{
					printf("error\n");
				}
				else
				{
					msgAsk.szBuffer[strlen(msgAsk.szBuffer) + 1] = '\n';
					printf(msgAsk.szBuffer);
					printf("Recv bytes = %d, msgAsk.size = %d\n", dwTrans, msgAsk.iBodySize);
				}

				MSG_BODY msgBody = {0};
				memcpy(&msgBody, pPerHandle->buf + msgAsk.iBodySize, sizeof(MSG_BODY));
				if (msgBody.iOpType == OP_READ && msgBody.iBodySize == sizeof(MSG_BODY))
				{
					printf("msgBody.szBuffer = %s\n", msgBody.szBuffer);
				}

				MSG_ACK msgAck = {0};
				msgAck.iCheckCode = CHECK_CODE;
				memcpy(msgAck.szBuffer, "This is the ack package",
					strlen("This is the ack package"));

				// 继续投递发送I/O请求
				pPerHandle->nOperationType = OP_WRITE;
				WSABUF buf;
				buf.buf = (char*)&msgAck;
				buf.len = sizeof(MSG_ACK);

				OVERLAPPED *pol = (OVERLAPPED *)::GlobalAlloc(GPTR, sizeof(OVERLAPPED));

				DWORD dwFlags = 0, dwSend = 0;
				::WSASend(pPerHandle->s, &buf, 1, &dwSend, dwFlags, pol, NULL); 
			}
			break;
		case OP_WRITE:
			{
				if (dwTrans == sizeof(MSG_ACK))
				{
					printf("Transfer successfully\n");
				}
				// 然后投递接收I/O请求
			}
			break;
		case OP_ACCEPT:
			break;
		}
	}

	return 0;
}

DWORD InitWinsock()
{
	DWORD dwRet = 0;

	WSADATA wsaData;  
	dwRet = WSAStartup(MAKEWORD(2,2), &wsaData);  
	if (dwRet != NO_ERROR)  
	{  
		printf("error code = %d\n", GetLastError());  
		dwRet = GetLastError();  
	}  

	return dwRet;
}

void UnInitWinsock()
{
	WSACleanup();
}

int main(int argc, _TCHAR* argv[])
{
	int nPort = 5000;

	InitWinsock();
	
	// 创建完成端口对象
	HANDLE hCompletion = ::CreateIoCompletionPort(INVALID_HANDLE_VALUE, 0, 0, 0);
	if (hCompletion == NULL)
	{
		DWORD dwRet = GetLastError();
		return dwRet;
	}
	::CreateThread(NULL, 0, ServerThread, (LPVOID)hCompletion, 0, 0);

	// 创建监听套接字
	SOCKET sListen = ::socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
	SOCKADDR_IN si;
	si.sin_family = AF_INET;
	si.sin_port = ::htons(nPort);
	si.sin_addr.s_addr = INADDR_ANY;
	::bind(sListen, (sockaddr*)&si, sizeof(si));
	::listen(sListen, 10);

	while (TRUE)
	{
		SOCKADDR_IN saRemote;
		int nRemoteLen = sizeof(saRemote);
		printf("Accepting...\n");
		SOCKET sNew = ::accept(sListen, (sockaddr*)&saRemote, &nRemoteLen);
		//SOCKET sNew = ::accept(sListen, NULL, NULL);
		if (sNew == INVALID_SOCKET)
		{
			continue;
		}
		printf("Accept one!\n");

		// 接受新连接后，创建一个per-handle数据，并关联到完成端口对象
		PPER_HANDLE_DATA pPerHandle = (PPER_HANDLE_DATA)::GlobalAlloc(GPTR, sizeof(PER_HANDLE_DATA));
		pPerHandle->s = sNew;
		memcpy(&pPerHandle->addr, &saRemote, nRemoteLen);
		pPerHandle->nOperationType = OP_READ;
		::CreateIoCompletionPort((HANDLE) pPerHandle->s, hCompletion, (ULONG_PTR)pPerHandle, 0);

		// 投递一个接收请求
		OVERLAPPED *pol = (OVERLAPPED *)::GlobalAlloc(GPTR, sizeof(OVERLAPPED));
		WSABUF buf;
		buf.buf = pPerHandle->buf;
		buf.len = BUFFER_SIZE;
		DWORD dwRecv = 0;
		DWORD dwFlags = 0;
		::WSARecv(pPerHandle->s, &buf, 1, &dwRecv, &dwFlags, pol, NULL);
	}

	return 0;
}

客户端代码(客户端没用完成端口)：

#include <WinSock2.h>

#define CHECK_CODE 0x010110
#define OP_READ   18
#define OP_WRITE  28
#define OP_ACCEPT 38

#pragma pack(1)

typedef struct MsgAsk
{
	int iCode;
	int iBodySize;
	char szBuffer[32];
}MSG_ASK, *PMSG_ASK;

typedef struct MsgBody
{
	int iBodySize;
	int iOpType;
	char szBuffer[64];
}MSG_BODY, *PMSG_BODY;

typedef struct MsgAck
{
	int  iCheckCode;
	char szBuffer[32];
}MSG_ACK, *PMSG_ACK;

#pragma pack()

DWORD SendAll(SOCKET &clientSock, char* buffer, int size)
{
	DWORD dwStatus = 0;
	char  *pTemp   = buffer;
	int   total    = 0, count = 0;

	while(total < size) 
	{
		count = send(clientSock, pTemp, size - total, 0);
		if(count < 0) 
		{
			dwStatus = WSAGetLastError();
			break;
		}
		total += count;
		pTemp += count;
	}

	return dwStatus ;
}

DWORD RecvAll(SOCKET &sock, char* buffer, int size)  
{  
	DWORD dwStatus = 0;  
	char *pTemp    = buffer;  
	int total      = 0, count = 0;  

	while (total < size)  
	{  
		count = recv(sock, pTemp, size-total, 0);  
		if (count < 0)  
		{  
			dwStatus = WSAGetLastError();  
			break;  
		}  

		total += count;  
		pTemp += count;  
	}  

	return dwStatus;  
}  

int _tmain(int argc, _TCHAR* argv[])
{
	WSADATA wsaData;  
	int iResult = WSAStartup(MAKEWORD(2,2), &wsaData);  
	if (iResult != NO_ERROR)  
	{  
		printf("error code = %d\n", GetLastError());  
		return -1;  
	}  

	sockaddr_in clientAddr;  
	clientAddr.sin_addr.s_addr = inet_addr("127.0.0.1");  
	clientAddr.sin_family      = AF_INET;  
	clientAddr.sin_port        = htons(5000);  

	SOCKET clientSock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);  
	if (clientSock == INVALID_SOCKET)  
	{  
		printf("Create socket failed, error code = %d\n", WSAGetLastError());  
		return -1;  
	}  

	//connect  
	while (connect(clientSock, (SOCKADDR *)&clientAddr, sizeof(SOCKADDR_IN)) == SOCKET_ERROR)  
	{  
		printf("Connecting...\n");  
		Sleep(1000);  
	}  

	MSG_ASK msgAsk = {0};
	msgAsk.iBodySize = sizeof(MSG_ASK);
	msgAsk.iCode = CHECK_CODE;
	memcpy(msgAsk.szBuffer, "This is a header", strlen("This is a header"));

	// 发送头部
	SendAll(clientSock, (char*)&msgAsk, msgAsk.iBodySize);

	MSG_BODY msgBody = {0};
	msgBody.iBodySize = sizeof(MSG_BODY);
	msgBody.iOpType = OP_READ;
	memcpy(msgBody.szBuffer, "This is the body", strlen("This is the body"));

	// 发送body
	SendAll(clientSock, (char*)&msgBody, msgBody.iBodySize);

	MSG_ACK msgAck = {0};

	RecvAll(clientSock, (char*)&msgAck, sizeof(msgAck));
	if (msgAck.iCheckCode == CHECK_CODE)
	{
		printf("The process is successful, msgAck.szBuffer = %s \n", msgAck.szBuffer);
	}
	else
	{
		printf("failed\n");
	}

	closesocket(clientSock);
	WSACleanup();

	return 0;
}