Winsock IOCP模型

来源：互联网发布：java 调用chart.js 编辑：程序博客网时间：2024/04/28 16:35

// IOCP.cpp : Defines the entry point for the console application.
//
//

#include "stdafx.h"

#include <WinSock2.h>
#include <Windows.h>
#include <process.h>
#pragma comment(lib, "WS2_32.lib")

#define MAX_BUFFER 256
#define MAX_TIMEOUT 1000
#define MAX_SOCKET 1024
#define MAX_THREAD 64

typedef enum _OPERATION_INFO_
{
    OP_NULL,
    OP_READ,
    OP_WRITE
}OPERATIONINFO;

typedef struct _PER_HANDLE_DATA_
{
public:
    _PER_HANDLE_DATA_()
    {
        clean();
    }
    ~_PER_HANDLE_DATA_()
    {
        clean();
    }
protected:
    void clean()
    {
        sock = INVALID_SOCKET;
        memset(&addr, 0, sizeof(addr));
        addr.sin_addr.S_un.S_addr = INADDR_ANY;
        addr.sin_port = htons(0);
        addr.sin_family = AF_INET;
    }
public:
    SOCKET sock;
    SOCKADDR_IN addr;

}PERHANDLEDATA, *PPERHANDLEDATA;

typedef struct _PER_IO_DTATA_
{
public:
    _PER_IO_DTATA_()
    {
        clean();
    }
    ~_PER_IO_DTATA_()
    {
        clean();
    }
private:
    void clean()
    {
        ZeroMemory(&ol, sizeof(ol));
        memset(buf, 0, sizeof(buf));
        wsaBuf.buf = buf;
        wsaBuf.len = MAX_BUFFER;
        opType = OP_NULL;
    }
public:
    WSAOVERLAPPED ol;
    WSABUF wsaBuf;
    char buf[MAX_BUFFER];
    OPERATIONINFO opType;
}PERIODATA, *PPERIODATA;

HANDLE hThread[MAX_THREAD] = {0};
int g_nThread = 0;
BOOL g_bExitThread = FALSE;

unsigned __stdcall ThreadProc(LPVOID lParam);

int _tmain(int argc, _TCHAR* argv[])
{
    WSADATA wsaData;
    if(0 != WSAStartup(MAKEWORD(2, 2), &wsaData))
    {
        printf("WSAStartup failed with error code: %d/n", GetLastError());
        return EXIT_FAILURE;
    }
    if(2 != HIBYTE(wsaData.wVersion) || 2 != LOBYTE(wsaData.wVersion))
    {
        printf("Socket version not supported./n");
        WSACleanup();
        return EXIT_FAILURE;
    }

    // Create I/O Completion Port
    HANDLE hIOCP = CreateIoCompletionPort(INVALID_HANDLE_VALUE, NULL, NULL, 0);
    if(NULL == hIOCP)
    {
        printf("CreateIoCompletionPort failed with error code: %d/n", GetLastError());
        WSACleanup();
        return EXIT_FAILURE;
    }
    // Create worker thread
    SYSTEM_INFO si = {0};
    GetSystemInfo(&si);
    for(int i = 0; i < (int)si.dwNumberOfProcessors+2; i++)
    {
        hThread[g_nThread] = (HANDLE)_beginthreadex(NULL, 0, ThreadProc, (LPVOID)hIOCP, 0, NULL);
        if(NULL == hThread[g_nThread])
        {
            printf("_beginthreadex failed with error code: %d/n", GetLastError());
            continue;
        }
        ++g_nThread;

        if(g_nThread > MAX_THREAD)
        {
            break;
        }
    }

    // Create socket
    SOCKET sListen = WSASocket(AF_INET, SOCK_STREAM, IPPROTO_TCP, NULL, 0, WSA_FLAG_OVERLAPPED);
    if(INVALID_SOCKET == sListen)
    {
        printf("WSASocket failed with error code: %d/n", WSAGetLastError());
        goto EXIT_CODE;
    }
    SOCKADDR_IN addr;
    memset(&addr, 0, sizeof(addr));
    addr.sin_addr.S_un.S_addr = inet_addr("127.0.0.1");
    addr.sin_family = AF_INET;
    addr.sin_port = htons(5050);
    if(SOCKET_ERROR == bind(sListen, (LPSOCKADDR)&addr, sizeof(addr)))
    {
        printf("bind failed with error code: %d/n", WSAGetLastError());
        closesocket(sListen);
        sListen = INVALID_SOCKET;
        goto EXIT_CODE;
    }
    if(SOCKET_ERROR == listen(sListen, 5))
    {
        printf("listen failed with error code: %d/n", WSAGetLastError());
        closesocket(sListen);
        sListen = INVALID_SOCKET;
        goto EXIT_CODE;
    }

    printf("Server start, wait for client to connect .../n");
    while(TRUE)
    {
        SOCKADDR_IN remote;
        memset(&remote, 0, sizeof(remote));
        int len = sizeof(remote);

        SOCKET sNew = WSAAccept(sListen, (LPSOCKADDR)&remote, &len, NULL, NULL);
        if(INVALID_SOCKET == sNew)
        {
            printf("WSAAccept failed with error code: %d/n", WSAGetLastError());
            continue;
        }
        printf("Client <%s : %d> come in./n", inet_ntoa(remote.sin_addr), ntohs(remote.sin_port));
        PERHANDLEDATA* pPerHandleData = new PERHANDLEDATA;
        pPerHandleData->sock = sNew;
        memcpy(&(pPerHandleData->addr), &remote, sizeof(remote));
        // Associate with IOCP
        if(NULL == CreateIoCompletionPort((HANDLE)(pPerHandleData->sock), hIOCP, (ULONG_PTR)pPerHandleData, 0))
        {
            printf("CreateIoCompletionPort failed with error code: %d/n", GetLastError());
            closesocket(pPerHandleData->sock);
            delete pPerHandleData;
            continue;
        }
        // Post Receive
        PERIODATA* pPerIoData = new PERIODATA;
        pPerIoData->opType = OP_READ;
        DWORD dwTrans = pPerIoData->wsaBuf.len;
        DWORD dwFlags = 0;
        if(SOCKET_ERROR == WSARecv(pPerHandleData->sock, &(pPerIoData->wsaBuf), 1,
            &dwTrans, &dwFlags, &(pPerIoData->ol), NULL))
        {
            if(WSA_IO_PENDING != WSAGetLastError())
            {
                printf("WSARecv failed with error code: %d/n", WSAGetLastError());
                closesocket(pPerHandleData->sock);
                delete pPerHandleData;
                delete pPerIoData;
                continue;
            }
        }
    }
    closesocket(sListen);
    sListen = INVALID_SOCKET;

EXIT_CODE:
    g_bExitThread = TRUE;
    PostQueuedCompletionStatus(hIOCP, 0, NULL, NULL);
    WaitForMultipleObjects(g_nThread, hThread, TRUE, INFINITE);
    for(int i = 0; i < g_nThread; i++)
    {
        CloseHandle(hThread[g_nThread]);
    }
    CloseHandle(hIOCP); // Close IOCP
    WSACleanup();
    return 0;
}

unsigned __stdcall ThreadProc(LPVOID lParam)
{
    HANDLE hIOCP = (HANDLE)lParam;

    PERHANDLEDATA* pPerHandleData = NULL;
    PERIODATA* pPerIoData = NULL;
    WSAOVERLAPPED* lpOverlapped = NULL;
    DWORD dwTrans = 0;
    DWORD dwFlags = 0;
    while(!g_bExitThread)
    {
        BOOL bRet = GetQueuedCompletionStatus(hIOCP, &dwTrans, (PULONG_PTR)&pPerHandleData, &lpOverlapped, INFINITE);
        if(!bRet)
        {
            printf("GetQueuedCompletionStatus failed with error: %d/n", WSAGetLastError());
            continue;
        }
        else
        {
            pPerIoData = CONTAINING_RECORD(lpOverlapped, PERIODATA, ol);
            if(0 == dwTrans)
            {
                printf("Client: <%s : %d> leave./n", inet_ntoa(pPerHandleData->addr.sin_addr), ntohs(pPerHandleData->addr.sin_port));
                closesocket(pPerHandleData->sock);
                delete pPerHandleData;
                delete pPerIoData;
                continue;
            }
            else
            {
                switch(pPerIoData->opType)
                {
                case OP_READ:
                    printf("recv client <%s : %d> data: %s/n", inet_ntoa(pPerHandleData->addr.sin_addr), ntohs(pPerHandleData->addr.sin_port), pPerIoData->buf);
                    pPerIoData->opType = OP_WRITE;
                    memset(&(pPerIoData->ol), 0, sizeof(pPerIoData->ol));
                    if(SOCKET_ERROR == WSASend(pPerHandleData->sock, &(pPerIoData->wsaBuf), 1, &dwTrans, dwFlags, &(pPerIoData->ol), NULL))
                    {
                        if(WSA_IO_PENDING != WSAGetLastError())
                        {
                            printf("WSASend failed with error code: %d./n", WSAGetLastError());
                            continue;
                        }
                    }
                    break;

                case OP_WRITE:
                    {
                        pPerIoData->opType = OP_READ;
                        dwFlags = 0;
                        memset(&(pPerIoData->ol), 0, sizeof(pPerIoData->ol));
                        memset(pPerIoData->buf, 0, sizeof(pPerIoData->buf));
                        pPerIoData->wsaBuf.buf = pPerIoData->buf;
                        dwTrans = pPerIoData->wsaBuf.len = MAX_BUFFER;
                        if(SOCKET_ERROR == WSARecv(pPerHandleData->sock, &(pPerIoData->wsaBuf), 1, &dwTrans, &dwFlags, &(pPerIoData->ol), NULL))
                        {
                            if(WSA_IO_PENDING != WSAGetLastError())
                            {
                                printf("WSARecv failed with error code: %d./n", WSAGetLastError());
                                continue;
                            }
                        }
                    }
                    break;

                default:
                    break;
                }
            }
        }
    }
    return 0;
}
// IOCP.cpp : Defines the entry point for the console application.
//
//

#include "stdafx.h"

#include <WinSock2.h>
#include <Windows.h>
#include <process.h>
#pragma comment(lib, "WS2_32.lib")

#define MAX_BUFFER 256
#define MAX_TIMEOUT 1000
#define MAX_SOCKET 1024
#define MAX_THREAD 64

typedef enum _OPERATION_INFO_
{
OP_NULL,
OP_READ,
OP_WRITE
}OPERATIONINFO;

typedef struct _PER_HANDLE_DATA_
{
public:
_PER_HANDLE_DATA_()
{
  clean();
}
~_PER_HANDLE_DATA_()
{
  clean();
}
protected:
void clean()
{
  sock = INVALID_SOCKET;
  memset(&addr, 0, sizeof(addr));
  addr.sin_addr.S_un.S_addr = INADDR_ANY;
  addr.sin_port = htons(0);
  addr.sin_family = AF_INET;
}
public:
SOCKET sock;
SOCKADDR_IN addr;

}PERHANDLEDATA, *PPERHANDLEDATA;

typedef struct _PER_IO_DTATA_
{
public:
_PER_IO_DTATA_()
{
  clean();
}
~_PER_IO_DTATA_()
{
  clean();
}
private:
void clean()
{
  ZeroMemory(&ol, sizeof(ol));
  memset(buf, 0, sizeof(buf));
  wsaBuf.buf = buf;
  wsaBuf.len = MAX_BUFFER;
  opType = OP_NULL;
}
public:
WSAOVERLAPPED ol;
WSABUF wsaBuf;
char buf[MAX_BUFFER];
OPERATIONINFO opType;
}PERIODATA, *PPERIODATA;

HANDLE hThread[MAX_THREAD] = {0};
int g_nThread = 0;
BOOL g_bExitThread = FALSE;

unsigned __stdcall ThreadProc(LPVOID lParam);

int _tmain(int argc, _TCHAR* argv[])
{
WSADATA wsaData;
if(0 != WSAStartup(MAKEWORD(2, 2), &wsaData))
{
  printf("WSAStartup failed with error code: %d/n", GetLastError());
  return EXIT_FAILURE;
}
if(2 != HIBYTE(wsaData.wVersion) || 2 != LOBYTE(wsaData.wVersion))
{
  printf("Socket version not supported./n");
  WSACleanup();
  return EXIT_FAILURE;
}

// Create I/O Completion Port
HANDLE hIOCP = CreateIoCompletionPort(INVALID_HANDLE_VALUE, NULL, NULL, 0);
if(NULL == hIOCP)
{
  printf("CreateIoCompletionPort failed with error code: %d/n", GetLastError());
  WSACleanup();
  return EXIT_FAILURE;
}
// Create worker thread
SYSTEM_INFO si = {0};
GetSystemInfo(&si);
for(int i = 0; i < (int)si.dwNumberOfProcessors+2; i++)
{
  hThread[g_nThread] = (HANDLE)_beginthreadex(NULL, 0, ThreadProc, (LPVOID)hIOCP, 0, NULL);
  if(NULL == hThread[g_nThread])
  {
   printf("_beginthreadex failed with error code: %d/n", GetLastError());
   continue;
  }
  ++g_nThread;

  if(g_nThread > MAX_THREAD)
  {
   break;
  }
}

// Create socket
SOCKET sListen = WSASocket(AF_INET, SOCK_STREAM, IPPROTO_TCP, NULL, 0, WSA_FLAG_OVERLAPPED);
if(INVALID_SOCKET == sListen)
{
  printf("WSASocket failed with error code: %d/n", WSAGetLastError());
  goto EXIT_CODE;
}
SOCKADDR_IN addr;
memset(&addr, 0, sizeof(addr));
addr.sin_addr.S_un.S_addr = inet_addr("127.0.0.1");
addr.sin_family = AF_INET;
addr.sin_port = htons(5050);
if(SOCKET_ERROR == bind(sListen, (LPSOCKADDR)&addr, sizeof(addr)))
{
  printf("bind failed with error code: %d/n", WSAGetLastError());
  closesocket(sListen);
  sListen = INVALID_SOCKET;
  goto EXIT_CODE;
}
if(SOCKET_ERROR == listen(sListen, 5))
{
  printf("listen failed with error code: %d/n", WSAGetLastError());
  closesocket(sListen);
  sListen = INVALID_SOCKET;
  goto EXIT_CODE;
}

printf("Server start, wait for client to connect .../n");
while(TRUE)
{
  SOCKADDR_IN remote;
  memset(&remote, 0, sizeof(remote));
  int len = sizeof(remote);

  SOCKET sNew = WSAAccept(sListen, (LPSOCKADDR)&remote, &len, NULL, NULL);
  if(INVALID_SOCKET == sNew)
  {
   printf("WSAAccept failed with error code: %d/n", WSAGetLastError());
   continue;
  }
  printf("Client <%s : %d> come in./n", inet_ntoa(remote.sin_addr), ntohs(remote.sin_port));
  PERHANDLEDATA* pPerHandleData = new PERHANDLEDATA;
  pPerHandleData->sock = sNew;
  memcpy(&(pPerHandleData->addr), &remote, sizeof(remote));
  // Associate with IOCP
  if(NULL == CreateIoCompletionPort((HANDLE)(pPerHandleData->sock), hIOCP, (ULONG_PTR)pPerHandleData, 0))
  {
   printf("CreateIoCompletionPort failed with error code: %d/n", GetLastError());
   closesocket(pPerHandleData->sock);
   delete pPerHandleData;
   continue;
  }
  // Post Receive
  PERIODATA* pPerIoData = new PERIODATA;
  pPerIoData->opType = OP_READ;
  DWORD dwTrans = pPerIoData->wsaBuf.len;
  DWORD dwFlags = 0;
  if(SOCKET_ERROR == WSARecv(pPerHandleData->sock, &(pPerIoData->wsaBuf), 1,
   &dwTrans, &dwFlags, &(pPerIoData->ol), NULL))
  {
   if(WSA_IO_PENDING != WSAGetLastError())
   {
    printf("WSARecv failed with error code: %d/n", WSAGetLastError());
    closesocket(pPerHandleData->sock);
    delete pPerHandleData;
    delete pPerIoData;
    continue;
   }
  }
}
closesocket(sListen);
sListen = INVALID_SOCKET;

EXIT_CODE:
g_bExitThread = TRUE;
PostQueuedCompletionStatus(hIOCP, 0, NULL, NULL);
WaitForMultipleObjects(g_nThread, hThread, TRUE, INFINITE);
for(int i = 0; i < g_nThread; i++)
{
CloseHandle(hThread[g_nThread]);
}
CloseHandle(hIOCP); // Close IOCP
WSACleanup();
return 0;
}

unsigned __stdcall ThreadProc(LPVOID lParam)
{
HANDLE hIOCP = (HANDLE)lParam;

PERHANDLEDATA* pPerHandleData = NULL;
PERIODATA* pPerIoData = NULL;
WSAOVERLAPPED* lpOverlapped = NULL;
DWORD dwTrans = 0;
DWORD dwFlags = 0;
while(!g_bExitThread)
{
  BOOL bRet = GetQueuedCompletionStatus(hIOCP, &dwTrans, (PULONG_PTR)&pPerHandleData, &lpOverlapped, INFINITE);
  if(!bRet)
  {
   printf("GetQueuedCompletionStatus failed with error: %d/n", WSAGetLastError());
   continue;
  }
  else
  {
   pPerIoData = CONTAINING_RECORD(lpOverlapped, PERIODATA, ol);
   if(0 == dwTrans)
   {
    printf("Client: <%s : %d> leave./n", inet_ntoa(pPerHandleData->addr.sin_addr), ntohs(pPerHandleData->addr.sin_port));
    closesocket(pPerHandleData->sock);
    delete pPerHandleData;
    delete pPerIoData;
    continue;
   }
   else
   {
    switch(pPerIoData->opType)
    {
    case OP_READ:
     printf("recv client <%s : %d> data: %s/n", inet_ntoa(pPerHandleData->addr.sin_addr), ntohs(pPerHandleData->addr.sin_port), pPerIoData->buf);
     pPerIoData->opType = OP_WRITE;
     memset(&(pPerIoData->ol), 0, sizeof(pPerIoData->ol));
     if(SOCKET_ERROR == WSASend(pPerHandleData->sock, &(pPerIoData->wsaBuf), 1, &dwTrans, dwFlags, &(pPerIoData->ol), NULL))
     {
      if(WSA_IO_PENDING != WSAGetLastError())
      {
       printf("WSASend failed with error code: %d./n", WSAGetLastError());
       continue;
      }
     }
     break;

    case OP_WRITE:
     {
      pPerIoData->opType = OP_READ;
      dwFlags = 0;
      memset(&(pPerIoData->ol), 0, sizeof(pPerIoData->ol));
      memset(pPerIoData->buf, 0, sizeof(pPerIoData->buf));
      pPerIoData->wsaBuf.buf = pPerIoData->buf;
      dwTrans = pPerIoData->wsaBuf.len = MAX_BUFFER;
      if(SOCKET_ERROR == WSARecv(pPerHandleData->sock, &(pPerIoData->wsaBuf), 1, &dwTrans, &dwFlags, &(pPerIoData->ol), NULL))
      {
       if(WSA_IO_PENDING != WSAGetLastError())
       {
        printf("WSARecv failed with error code: %d./n", WSAGetLastError());
        continue;
       }
      }
     }
     break;

    default:
     break;
    }
   }
  }
}
return 0;
}

本文来自CSDN博客，转载请标明出处：http://blog.csdn.net/VisualEleven/archive/2010/11/29/6041893.aspx