完成端口通信例子

代码贴上来以后慢慢修改

服务端代码：

#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 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;
}