Chinaunix首页 | 论坛 | 博客
  • 博客访问: 244025
  • 博文数量: 62
  • 博客积分: 973
  • 博客等级: 准尉
  • 技术积分: 530
  • 用 户 组: 普通用户
  • 注册时间: 2011-11-16 23:25
文章分类

全部博文(62)

文章存档

2013年(1)

2012年(14)

2011年(47)

分类: C/C++

2011-12-04 22:24:16

amy3701229所理解的完成端口原理是:创建一个完成端口与一个SOCKET绑定,所有通过该SOCKET通讯的数据都由完成端口机制来处理,完成端口是异步I/O,接收和发送都是提交后不管结果,所有结果都由完成端口的消息机制来做相应的通知,并且完成端口内部有线程池的概念
 
//www.cnblogs.com/jiayongzhi/archive/2011/05/17/2048286.html

首先来说为什么要使用完成端口:

原因还是因为为了解决recv方法为阻塞式的问题,WinSocket封装的WSARecv方法为非堵塞的方法。

int WSARecv(

SOCKET lpCompletionRoutine

);

WSARecv为非阻塞的方法,其中第二个参数是I/O请求成功时,数据保存的地址。

Socket的触发是属于网卡硬件的中断信号,只是此信号CPU不能直接获取状态,此时我们可以使之绑定Event事件,Event内核对象的状态时可以监听到的。

这也就是WSAEventSelect模型的原理,当然重叠模型的最终原理也是如此。但Event的方法有着其弊病:当模型处理多线程事件时要调用WSAWaitForMultipleEvents函数,

WSAWaitForMultipleEvents函数一次最多只能等待64个事件对象。所以当海量客户端连接服务器时,服务器将没有能力应对,于是我们使用完成端口。

完成端口:

HANDLE CreateIoCompletionPort(

    HANDLE FileHandle, //要链接的Socket

    HANDLE ExistingCompletionPort, //全局完成端口

//同完成端口关联到一起的句柄,此处可为链接的socket,或是id等等(目地使接收到的socket知道是哪个socket)

    DWORD CompletionKey,

    DWORD NumberOfConcurrentThreads

);

此函数创建创建Socket与完成端口的链接,CreateIoCompletionPort函数被用于完成两个工作:

  • 用于创建个完成端口对象。
  • 将一个句柄同完成端口关联到一起。

用函数GetQueuedCompletionStatus等待全局完成端口的完成队列。 

BOOL GetQueuedCompletionStatus(
    
    HANDLE                CompletionPort,

    LPDWORD               lpNumberOfBytes,

    PULONG_PTR        lpCompletionKey,  //此参数为CreateIoCompletionPort第三个参数传过来的句柄,通过此参数获得socket
     
    LPOVERLAPPED*         lpOverlapped,

    DWORD                dwMilliseconds

);   

完成端口的工作原理是,把Socket和完成端口绑定,通过关联句柄传递传递参数,使得获取到的Socket能得知是那个socket,参数可以自定义可以是socket本身也可以是id等等。

  1. #include "WinSock2.h"

  2. #pragma comment(lib, "ws2_32.lib")


  3. #define MESSAGESIZE 1024

  4. SOCKET serverSocket;

  5. DWORD WINAPI SocketProcAccept(LPVOID pParam);

  6. DWORD WINAPI SocketProcMain(LPVOID pParam);

  7. enum SOCKETOPERATE
  8. {
  9.     soREVC
  10. };

  11. struct SOCKETDATA
  12. {

  13.     WSAOVERLAPPED overlapped;

  14.     WSABUF buf;

  15.     char sMessage[MESSAGESIZE];

  16.     DWORD dwBytes;

  17.     DWORD flag;

  18.     SOCKETOPERATE socketType;

  19.     void Clear(SOCKETOPERATE type)
  20.     {
  21.         ZeroMemory(this, sizeof(SOCKETDATA));

  22.         buf.buf = sMessage;

  23.         buf.len = MESSAGESIZE;

  24.         socketType = type;
  25.     }
  26. };

  27. SOCKET CreateServiceSocket(int Port)
  28. {

  29.     int iError;

  30.     WSAData data;

  31.     iError = WSAStartup(0x0202, &data);

  32.     SOCKET tmp = socket(AF_INET,SOCK_STREAM,0);

  33.     if(tmp == INVALID_SOCKET)
  34.     {
  35.         return INVALID_SOCKET;
  36.     }

  37.     SOCKADDR_IN addr;

  38.     addr.sin_addr.s_addr = inet_addr("127.0.0.1");

  39.     addr.sin_family = AF_INET;

  40.     addr.sin_port = htons(Port);

  41.     if((bind(tmp, (sockaddr*)&addr, sizeof(addr))) != 0)
  42.     {
  43.         closesocket(tmp);

  44.         return INVALID_SOCKET;
  45.     }

  46.     if((listen(tmp, INFINITE)) != 0)
  47.     {
  48.         closesocket(tmp);

  49.         return INVALID_SOCKET;
  50.     }
  51.     return tmp;
  52. }

  53. int _tmain(int argc, _TCHAR* argv[])
  54. {
  55.     HANDLE CP = INVALID_HANDLE_VALUE;

  56.     CP = CreateIoCompletionPort(INVALID_HANDLE_VALUE, NULL, 0, 0);

  57.     SYSTEM_INFO systemInfo;

  58.     GetSystemInfo(&systemInfo);

  59.     for (int i = 0; i<systemInfo.dwNumberOfProcessors; i++)
  60.     {
  61.         CreateThread(NULL, NULL, &SocketProcMain, CP, NULL, NULL);
  62.     }

  63.     serverSocket = CreateServiceSocket(6565);

  64.     if (serverSocket == INVALID_SOCKET)
  65.     {
  66.         return 0;
  67.     }

  68.     CreateThread(NULL, NULL, &SocketProcAccept, CP, NULL, NULL);

  69.     while(1)
  70.     {
  71.         Sleep(10000);
  72.     }

  73.     CloseHandle(CP);

  74.     closesocket(serverSocket);

  75.     WSACleanup();

  76.     return 0;
  77. }

  78. DWORD WINAPI SocketProcAccept(LPVOID pParam)
  79. {
  80.     HANDLE CP = (HANDLE)pParam;

  81.     SOCKADDR_IN addr;

  82.     int len = sizeof(SOCKADDR_IN);

  83.     SOCKET tmp;

  84.     SOCKETDATA *lpSocketData;

  85.     while(1)
  86.     {
  87.         tmp = accept(serverSocket, (sockaddr*)&addr, &len);

  88.         printf("Client Accept:%s\t:%d\n", inet_ntoa(addr.sin_addr), htons(addr.sin_port));

  89.         CreateIoCompletionPort((HANDLE)tmp, CP, (DWORD)tmp, INFINITE);

  90.         lpSocketData = (SOCKETDATA *)HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(SOCKETDATA));

  91.         lpSocketData->Clear(soREVC);

  92.         WSARecv(tmp, &lpSocketData->buf, 1,&lpSocketData->dwBytes, &lpSocketData->flag, &lpSocketData->overlapped, NULL);
  93.     }
  94. }

  95. DWORD WINAPI SocketProcMain(LPVOID pParam)
  96. {
  97.     HANDLE CP = (HANDLE)pParam;

  98.     SOCKADDR_IN addr;

  99.     DWORD dwBytes;

  100.     SOCKETDATA *lpSocketData;

  101.     SOCKET clientSocket;
  102.     
  103.     while(1)
  104.     {
  105.         GetQueuedCompletionStatus(CP, &dwBytes, (PULONG_PTR)&clientSocket, (LPOVERLAPPED*)&lpSocketData, INFINITE);

  106.         if(dwBytes == 0xFFFFFFFF)
  107.         {
  108.             return 0;
  109.         }

  110.         if(lpSocketData->socketType == soREVC)
  111.         {
  112.             if(dwBytes == 0)
  113.             {

  114.                 closesocket(clientSocket);

  115.                 HeapFree(GetProcessHeap(), 0, lpSocketData);

  116.             }else
  117.             {

  118.                 lpSocketData->sMessage[dwBytes] = '\0';

  119.                 printf("%x\t:%s\n", (DWORD)clientSocket, lpSocketData->sMessage);

  120.                 lpSocketData->Clear(soREVC);

  121.                 WSARecv(clientSocket, &lpSocketData->buf, 1, &lpSocketData->dwBytes, &lpSocketData->flag, &lpSocketData->overlapped, NULL);

  122.             }

  123.         }

  124.     }
  125. }

Socket 重叠I/O 完成端口模型详解

//blog.sina.com.cn/s/blog_3fb7f7270100fesw.html

 

  1. 2009年02月23日 星期一 下午 10:07
  2. #include "stdafx.h"

  3. #include <WINSOCK2.h>
  4. #include <stdio.h>

  5. #define PORT 5150
  6. #define MSGSIZE 1024

  7. #pragma comment(lib, "ws2_32.lib")

  8. typedef enum
  9. {
  10.    RECV_POSTED
  11. }OPERATION_TYPE; //枚举,表示状态

  12. typedef struct
  13. {
  14. WSAOVERLAPPED overlap;
  15. WSABUF Buffer;
  16.    char szMessage[MSGSIZE];
  17. DWORD NumberOfBytesRecvd;
  18. DWORD Flags;
  19. OPERATION_TYPE OperationType;
  20. }PER_IO_OPERATION_DATA, *LPPER_IO_OPERATION_DATA; //定义一个结构体保存IO数据

  21. DWORD WINAPI WorkerThread(LPVOID);

  22. int main()
  23. {
  24.    WSADATA wsaData;
  25.    SOCKET sListen, sClient;
  26.    SOCKADDR_IN local, client;
  27.    DWORD i, dwThreadId;
  28.    int iaddrSize = sizeof(SOCKADDR_IN);
  29.    HANDLE CompletionPort = INVALID_HANDLE_VALUE;
  30.    SYSTEM_INFO systeminfo;
  31.    LPPER_IO_OPERATION_DATA lpPerIOData = NULL;

  32.    // Initialize Windows Socket library
  33.    WSAStartup(0x0202, &wsaData);

  34.    // 初始化完成端口
  35.    CompletionPort = CreateIoCompletionPort(INVALID_HANDLE_VALUE, NULL, 0, 0);

  36.    // 有几个CPU就创建几个工作者线程
  37.    GetSystemInfo(&systeminfo);
  38.    for (i = 0; i < systeminfo.dwNumberOfProcessors; i++)
  39.    {
  40.      CreateThread(NULL, 0, WorkerThread, CompletionPort, 0, &dwThreadId);
  41.    }

  42.    // 创建套接字
  43.    sListen = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);

  44.    // 绑定套接字
  45.    local.sin_addr.S_un.S_addr = htonl(INADDR_ANY);
  46. local.sin_family = AF_INET;
  47. local.sin_port = htons(PORT);
  48.    bind(sListen, (struct sockaddr *)&local, sizeof(SOCKADDR_IN));

  49.    // 开始监听!
  50.    listen(sListen, 3);

  51.    while (TRUE)//主进程的这个循环中循环等待客户端连接,若有连接,则将该客户套接字于完成端口绑定到一起
  52.       //然后开始异步等待接收客户传来的数据。
  53.    {
  54.      // 如果接到客户请求连接,则继续,否则等待。
  55.      sClient = accept(sListen, (struct sockaddr *)&client, &iaddrSize);
  56. //client中保存用户信息。
  57.      printf("Accepted client:%s:%d\n", inet_ntoa(client.sin_addr), ntohs(client.sin_port));

  58. //将这个最新到来的客户套接字和完成端口绑定到一起。
  59.      CreateIoCompletionPort((HANDLE)sClient, CompletionPort, ( ULONG_PTR)sClient, 0);
  60. //第三个参数表示传递的参数,这里就传递的客户套接字地址。最后一个参数为0 表示有和CPU一样的进程数。即1个CPU一个线程
  61.    
  62.      // 初始化结构体
  63.      lpPerIOData = (LPPER_IO_OPERATION_DATA)HeapAlloc(
  64.        GetProcessHeap(),
  65.        HEAP_ZERO_MEMORY,
  66.        sizeof(PER_IO_OPERATION_DATA));
  67.      lpPerIOData->Buffer.len = MSGSIZE; // len=1024
  68.      lpPerIOData->Buffer.buf = lpPerIOData->szMessage;
  69.      lpPerIOData->OperationType = RECV_POSTED; //操作类型
  70.      WSARecv(sClient, //异步接收消息,立刻返回。
  71.        &lpPerIOData->Buffer, //获得接收的数据
  72.        1, //The number of WSABUF structures in the lpBuffers array.
  73.        &lpPerIOData->NumberOfBytesRecvd, //接收到的字节数,如果错误返回0
  74.        &lpPerIOData->Flags, //参数,先不管
  75.        &lpPerIOData->overlap, //输入这个结构体咯。
  76.        NULL);
  77.    }


  78. //posts an I/O completion packet to an I/O completion port.
  79.    PostQueuedCompletionStatus(CompletionPort, 0xFFFFFFFF, 0, NULL);
  80. CloseHandle(CompletionPort);
  81. closesocket(sListen);
  82. WSACleanup();
  83. return 0;
  84. }
  85. //工作者线程有一个参数,是指向完成端口的句柄
  86. DWORD WINAPI WorkerThread(LPVOID CompletionPortID)
  87. {
  88.    HANDLE CompletionPort=(HANDLE)CompletionPortID;
  89.    DWORD dwBytesTransferred;
  90.    SOCKET sClient;
  91.    LPPER_IO_OPERATION_DATA lpPerIOData = NULL;

  92.    while (TRUE)
  93.    {
  94.      GetQueuedCompletionStatus( //遇到可以接收数据则返回,否则等待
  95.        CompletionPort,
  96.        &dwBytesTransferred, //返回的字数
  97.     (PULONG_PTR&)sClient, //是响应的哪个客户套接字?
  98.        (LPOVERLAPPED *)&lpPerIOData, //得到该套接字保存的IO信息
  99.        INFINITE); //无限等待咯。不超时的那种。
  100.      if (dwBytesTransferred == 0xFFFFFFFF)
  101.      {
  102.        return 0;
  103.      }
  104.    
  105.      if (lpPerIOData->OperationType == RECV_POSTED) //如果受到数据
  106.      {
  107.        if (dwBytesTransferred == 0)
  108.        {
  109.          // Connection was closed by client
  110.          closesocket(sClient);
  111.          HeapFree(GetProcessHeap(), 0, lpPerIOData); //释放结构体
  112.        }
  113.        else
  114.        {
  115.          lpPerIOData->szMessage[dwBytesTransferred] = '\0';
  116.          send(sClient, lpPerIOData->szMessage, dwBytesTransferred, 0); //将接收到的消息返回
  117.        
  118.          // Launch another asynchronous operation for sClient
  119.          memset(lpPerIOData, 0, sizeof(PER_IO_OPERATION_DATA));
  120.          lpPerIOData->Buffer.len = MSGSIZE;
  121.          lpPerIOData->Buffer.buf = lpPerIOData->szMessage;
  122.          lpPerIOData->OperationType = RECV_POSTED;
  123.          WSARecv(sClient, //循环接收
  124.     &lpPerIOData->Buffer,
  125.     1,
  126.     &lpPerIOData->NumberOfBytesRecvd,
  127.     &lpPerIOData->Flags,
  128.     &lpPerIOData->overlap,
  129.     NULL);
  130.        }
  131.      }
  132.    }
  133.    return 0;
  134. }

首先,说说主线程:
1.创建完成端口对象
2.创建工作者线程(这里工作者线程的数量是按照CPU的个数来决定的,这样可以达到最佳性能)
3.创建监听套接字,绑定,监听,然后程序进入循环
4.在循环中,我做了以下几件事情:
(1).接受一个客户端连接
(2).将该客户端套接字与完成端口绑定到一起(还是调用CreateIoCompletionPort,但这次的作用不同),
注意,按道理来讲,此时传递给CreateIoCompletionPort的第三个参数应该是一个完成键,
一般来讲,程序都是传递一个单句柄数据结构的地址,该单句柄数据包含了和该客户端连接有关的信息,
由于我们只关心套接字句柄,所以直接将套接字句柄作为完成键传递;
(3).触发一个WSARecv异步调用,用到了“尾随数据”,使接收数据所用的缓冲区紧跟在WSAOVERLAPPED对象之后,
此外,还有操作类型等重要信息。

在工作者线程的循环中,我们
1.调用GetQueuedCompletionStatus取得本次I/O的相关信息(例如套接字句柄、传送的字节数、单I/O数据结构的地址等等)
2.通过单I/O数据结构找到接收数据缓冲区,然后将数据原封不动的发送到客户端
3.再次触发一个WSARecv异步操作

自己的理解,有任何问题都可以与我联系

 

阅读(2999) | 评论(0) | 转发(0) |
0

上一篇:C语言编译全过程

下一篇:深拷贝与浅拷贝

给主人留下些什么吧!~~