实现UDP IOCP

分手后的思念是犯贱 2022-09-18 02:52 204阅读 0赞

[http://h-lm.spaces.live.com/blog/cns!C523F565A10E3B66!824.entry][http_h-lm.spaces.live.com_blog_cns_C523F565A10E3B66_824.entry]

2008/11/11

#### 实现UDP IOCP心得 ####

当前许多资料都是介绍TCP的IOCP的实现，UDP的较少。

1.很多人在讨论UDP是否需要IOCP。

借 [http://jlbookworm.spaces.live.com/blog/cns!ef3e777c24481c39!240.entry][http_jlbookworm.spaces.live.com_blog_cns_ef3e777c24481c39_240.entry] 的文章来说明一下

#### 通宵奋战，搞定UDP+IOCP ####

总算是没有白费时间，搞定了UDP+IOCP。原来的程序采用的是多线程+阻塞的变种，效率一直不如意。本来想一开始就用IOCP的，但是在网上查阅资料的时候看到讨论说UDP不需要用IOCP的，自己想想也觉得有点道理。于是就用最简单的多线程+阻塞来实现了数据采集。后来测试的时候效率一直不如意，更改为IOCP模型时思路错了，一直更改不成功。昨晚突然想到了问题所在，一下就把UDP+IOCP实现了。呵呵。

总算是小有收获，工作线程更改用户界面的问题也解决了，数据采集效率也还过得去了，相关测试程序都可以勉强运转了，又恢复了些许信心了。感觉是度过了一个黎明前的黑夜了，只是不知道还有多少个更黑的夜晚需要经历。

我的实现还没有进行测试，之后再附上。

2.TCP的IOCP是在Accept之后，将Accept创建的套接字与完成端口绑定，而在UDP中，则是把WSASocket或Socket创建的套接字与完成端口绑定。

在实现UDP IOCP时，可以参考已有的TCP IOCP代码，例如 [http://www.cppblog.com/niewenlong/archive/2007/08/17/30224.html][http_www.cppblog.com_niewenlong_archive_2007_08_17_30224.html]

另外 [http://www.codeproject.com/KB/IP/iocp-multicast-udp.aspx][http_www.codeproject.com_KB_IP_iocp-multicast-udp.aspx] 可供下的源码中的客户端代码是UDP IOCP实现

3.以下数据结构非常重要。

typedef struct \_PER\_IO\_OPERATION\_DATA  
\{  
    OVERLAPPED Overlapped;  
    WSABUF DataBuff;  
    char Buff\[24\];  
    BOOL OperationType;  
\}PER\_IO\_OPERATION\_DATA,\* LPPER\_IO\_OPERATION\_DATA;

因为在UDP中每次RecvFrom获WSARecvFrom会传回UDP数据来源的IP，因此可以将以上数据结构修改成：

typedef struct \_PER\_IO\_OPERATION\_DATA  
\{  
    OVERLAPPED Overlapped;  
    WSABUF DataBuff;  
    char Buff\[24\];  
    unsigned long recvBytes;      //存储接收到的字节数  
    SOCKADDR\_IN remoteAddr; //存储数据来源IP地址  
    int remoteAddrLen;              //存储数据来源IP地址长度  
\}PER\_IO\_OPERATION\_DATA,\* LPPER\_IO\_OPERATION\_DATA;

4.实现过程

创建LPPER\_IO\_OPERATION\_DATA数据结构并进行初始化（初始化很重要）

LPPER\_IO\_OPERATION\_DATA ioperdata;  
    ioperdata = (LPPER\_IO\_OPERATION\_DATA)malloc(sizeof(PER\_IO\_OPERATION\_DATA));  
    memset(&(ioperdata->Overlapped), 0, sizeof(OVERLAPPED));  
      
    (ioperdata->DataBuff).len = 24;  
    (ioperdata->DataBuff).buf = ioperdata->Buff;  
    ioperdata->recvBytes = 24;  
    ioperdata->remoteAddrLen = sizeof(ioperdata->remoteAddr);

创建完成端口

hCompletionPort = CreateIoCompletionPort(INVALID\_HANDLE\_VALUE, NULL, 0, 0);

创建UDP socket

udpSocket = WSASocket(AF\_INET, SOCK\_DGRAM, IPPROTO\_IP, NULL, 0, WSA\_FLAG\_OVERLAPPED);

绑定UDP Socket

bind(udpSocket, (SOCKADDR\*) & addr, sizeof(SOCKADDR\_IN));

将完成端口与UDP Socket绑定

CreateIoCompletionPort(  
        (HANDLE)udpSocket,  
        hCompletionPort,  
        (DWORD)udpSocket,  
        5  
    );

根据CPU数量\*2+2的原则创建工作者线程------- ------- --------------一直循环

CreateWorkers(m\_dwThreads)

接收数据                                                                          获得当前完成状态  GetQueuedCompletionStatus(

WSARecvFrom(                                                                                     ComPort,  
                udpSocket,                                                                          &BytesTransferred,  
                &(ioperdata->DataBuff),                                                         (LPDWORD) & nSocket,  
                1,                                                                                       (LPOVERLAPPED \*) & PerIoData,  
                &(ioperdata->recvBytes),                                                        INFINITE   );                     
                &flags,                                                             
                (SOCKADDR\*) & (ioperdata->remoteAddr),                                  
                &(ioperdata->remoteAddrLen),                  
                &(ioperdata->Overlapped),                                  
                NULL);                                                            处理接收到的数据

继续执行投递操作（类同接收数据操作）

5.WSAGetLastError错误代码

通过WSAGetLastError的信息来测试程序中出现的问题，常见的错误有10055、10014、6等，最主要的是变量的初始化。

另附上 [http://www.cppblog.com/johndragon/archive/2008/09/16/21845.html][http_www.cppblog.com_johndragon_archive_2008_09_16_21845.html]的总结

1- 不要为每个小数据包发送一个IOCP请求,这样很容易耗尽IOCP的内部队列.....从而产生10055错误.  
2- 不要试图在发送出IOCP请求之后,收到完成通知之前修改请求中使用的数据缓冲的内容,因为在这段时间,系统可能会来读取这些缓冲.  
 3- 为了避免内存拷贝,可以尝试关闭SOCKET的发送和接收缓冲区,不过代价是,你需要更多的接收请求POST到一个数据流量比较大的SOCKET,从而保证系统一直可以找到BUFFER来收取到来的数据.  
4- 在发出多个接收请求的时候,如果你的WORKTHREAD不止一个,一定要使用一些手段来保证接收完成的数据按照发送接收请求的顺序处理,否则,你会遇到数据包用混乱的顺序排列在你的处理队列里.....  
5- 说起工作线程, 最好要根据MS的建议, 开 CPU个数\*2+2 个, 如果你不了解IOCP的工作原理的话.   
6- IOCP的工作线程是系统优化和调度的, 自己就不需要进行额外的工作了.如果您自信您的智慧和经验超过MS的工程师, 那你还需要IOCP么....  
7-发出一个Send请求之后，就不需要再去检测是否发送完整，因为iocp会帮你做这件事情，有些人说iocp没有做这件事情，这和iocp的高效能是相悖的，并且我做过的无数次测试表明，Iocp要么断开连接，要么就帮你把每个发送请求都发送完整。  
8- 出现数据错乱的时候，不要慌，要从多线程的角度检查你的解析和发送数据包的代码，看看是不是有顺序上的问题。  
9- 当遇到奇怪的内存问题时，逐渐的减少工作线程的数量，可以帮你更快的锁定问题发生的潜在位置。  
10-同样是遇到内存问题时，请先去检查你的客户端在服务器端内部映射对象的释放是否有问题。而且要小心的编写iocp完成失败的处理代码，防止引用一个错误的内部映射对象的地址。  
11- overlapped对象一定要保存在持久的位置，并且不到操作完成（不管成功还是失败）不要释放，否则可能会引发各种奇怪的问题。  
12- IOCP的所有工作都是在获取完成状态的那个函数内部进行调度和完成的，所以除了注意工作线程的数量之外，还要注意，尽量保持足够多的工作线程处在获取完成状态的那个等待里面，这样做就需要减少工作线程的负担，确保工作线程内部要处理费时的工作。（我的建议是工作线程和逻辑线程彻底区分开）  
13- 刚刚想起来，overlapped对象要为每次的send和recv操作都准备一个全新的，不能图方便重复利用。  
14- 尽量保持send和recv的缓冲的大小是系统页面大小的倍数，因为系统发送或者接收数据的时候，会锁用户内存的，比页面小的缓冲会浪费掉整个一个页面。（作为第一条的补充，建议把小包合并成大包发送）

KEY WORDS: UDP; IOCP; I/O; Completion Port

关键词: UDP; IOCP; I/O; 完成端口

// Udp_Server.cpp : Defines the entry point for the console application. // #include "stdafx.h" #include <winsock2.h> #include <ws2tcpip.h> #include "Stdio.h" #include <iostream> #define BUFSIZE 1024 //max size of incoming data buffer #define DEFAULT_GROUP_ADDRESS "192.168.0.121" #define DEFAULT_PORT 7125 typedef struct { OVERLAPPED Overlapped; WSABUF DataBuf; CHAR Buffer[BUFSIZE]; //DWORD BytesSEND; //DWORD BytesRECV; SOCKADDR_IN SAddr; } stnIocpData, *pStnIocpData; typedef struct { u_short l_onoff; u_short l_linger; }stnLinger; HANDLE g_hIocp; DWORD WINAPI WorkerThread(LPVOID lpParam); bool CreateNetSocket(); BOOL CreateWorkers(UINT); void InitWinsock2(); void UnInitWinsock2(); //SOCKET g_hSocket; u_short nPort = DEFAULT_PORT; void InitWinsock2() { WSADATA data; WORD version; int ret = 0; version = (MAKEWORD(2, 2)); ret = WSAStartup(version, &data); if (ret != 0) { ret = WSAGetLastError(); if (ret == WSANOTINITIALISED) { printf("not initialised"); } } } void UnInitWinsock2() { WSACleanup(); } bool CreateNetSocket(void) { BOOL fFlag = TRUE; int nRet=0; SOCKADDR_IN stLclAddr; SOCKET nSocket = socket(AF_INET, SOCK_DGRAM,0); if (nSocket == INVALID_SOCKET) { printf ("socket() failed, Err: %d\n", WSAGetLastError()); return false; } nRet = setsockopt(nSocket,SOL_SOCKET,SO_REUSEADDR, (char *)&fFlag, sizeof(fFlag)); if (nRet == SOCKET_ERROR) { printf ("setsockopt() SO_REUSEADDR failed, Err: %d\n",WSAGetLastError()); } /* stnLinger Linger; Linger.l_onoff = 1; Linger.l_linger = 0; */ /* struct linger lng; lng.l_onoff = 0; nRet = setsockopt(g_hSocket,SOL_SOCKET,SO_LINGER, (char*)&lng, sizeof(lng)); if (nRet == SOCKET_ERROR) { nRet = WSAGetLastError(); printf ("setsockopt() SO_REUSEADDR failed, Err: %d\n",WSAGetLastError()); } */ stLclAddr.sin_family = AF_INET; stLclAddr.sin_addr.s_addr = htonl(INADDR_ANY); stLclAddr.sin_port = htons(nPort); nRet = bind(nSocket,(struct sockaddr*) &stLclAddr,sizeof(stLclAddr)); if (nRet == SOCKET_ERROR) { printf ("bind() port: %d failed, Err: %d\n", nPort,WSAGetLastError()); } g_hIocp = CreateIoCompletionPort (INVALID_HANDLE_VALUE,NULL,0,10); if (!g_hIocp) { fprintf (stdout, "g_hCompletionPort Create Failed\n"); return false; } CreateIoCompletionPort ((HANDLE)nSocket,g_hIocp,(DWORD)nSocket,10); pStnIocpData PerIoData; if ((PerIoData = new stnIocpData()) == NULL) { printf("GlobalAlloc() failed with error %d\n", GetLastError()); return false; } ZeroMemory(&(PerIoData->Overlapped), sizeof(OVERLAPPED)); PerIoData->DataBuf.len = BUFSIZE; PerIoData->DataBuf.buf = PerIoData->Buffer; DWORD Flags = 0; DWORD RecvBytes; int iLen = sizeof(PerIoData->SAddr); if (WSARecvFrom(nSocket, &(PerIoData->DataBuf), 1, &RecvBytes, &Flags, (sockaddr*)&(PerIoData->SAddr),&iLen, (LPWSAOVERLAPPED)PerIoData, NULL) == SOCKET_ERROR) { if (WSAGetLastError() != ERROR_IO_PENDING) { printf("WSARecv() failed with error %d\n", WSAGetLastError()); } } return true; } BOOL CreateWorkers (UINT uiCount) { DWORD ThreadId; HANDLE ThreadHandle; DWORD i; for (i = 0; i < uiCount; i++) { ThreadHandle = CreateThread (NULL,0,WorkerThread,g_hIocp,0,&ThreadId); if (!ThreadHandle) { fprintf (stdout, "Create Worker Thread Failed\n"); return FALSE; } CloseHandle (ThreadHandle); } return TRUE; } DWORD WINAPI WorkerThread(LPVOID lpParam) { HANDLE CompletionPort = (HANDLE) lpParam; DWORD BytesTransferred; LPOVERLAPPED Overlapped; memset(&Overlapped,0,sizeof(Overlapped)); pStnIocpData PerIoData; DWORD RecvBytes; DWORD Flags; DWORD nSocket; while(TRUE) { if (GetQueuedCompletionStatus(CompletionPort, &BytesTransferred, &nSocket, (LPOVERLAPPED *) &PerIoData, INFINITE) == 0) { printf("GetQueuedCompletionStatus failed with error %d\n", GetLastError()); return 0; } if (BytesTransferred == 0) { printf("Closing socket %d\n", nSocket); if (closesocket(nSocket) == SOCKET_ERROR) { return 0; } GlobalFree(PerIoData); continue; } /* BytesTransferred = 0; cout << "nSocket = " << nSocket << endl; cout << "接收的数据: " << PerIoData->DataBuf.buf << "长度: " << PerIoData->DataBuf.len << endl; */ //char* szllg = "dfjksdjf"; sendto( nSocket, PerIoData->DataBuf.buf, BytesTransferred, 0 , (sockaddr*)&(PerIoData->SAddr), sizeof(PerIoData->SAddr)); Flags = 0; ZeroMemory(&(PerIoData->Overlapped), sizeof(OVERLAPPED)); PerIoData->DataBuf.len = BUFSIZE; PerIoData->DataBuf.buf = PerIoData->Buffer; int iLen = sizeof(PerIoData->SAddr); if (WSARecvFrom(nSocket, &(PerIoData->DataBuf), 1, &RecvBytes, &Flags,(sockaddr*)&(PerIoData->SAddr),&iLen, &(PerIoData->Overlapped), NULL) == SOCKET_ERROR) { if (WSAGetLastError() != ERROR_IO_PENDING) { printf("WSARecv() failed with error %d\n", WSAGetLastError()); } if ( WSAGetLastError() == 10054 ) { WSARecvFrom(nSocket, &(PerIoData->DataBuf), 1, &RecvBytes, &Flags,(sockaddr*)&(PerIoData->SAddr),&iLen, &(PerIoData->Overlapped), NULL); } } } return 0; } //----------------------------------------------------------------------------- int main(int argc, char* argv[]) { InitWinsock2(); HANDLE hWait2Exit = CreateEvent(NULL,FALSE,TRUE,"MCLIENT"); ResetEvent(hWait2Exit ); SetThreadPriority (GetCurrentThread (), THREAD_PRIORITY_TIME_CRITICAL); if ( !CreateNetSocket() ) { printf("Error condition @ CreateNetConnections , exiting\n"); return 1; } SYSTEM_INFO SystemInfo; GetSystemInfo(&SystemInfo); if (!CreateWorkers( SystemInfo.dwNumberOfProcessors*2) ) { printf("Error condition @CreateWorkers, exiting\n"); return 1; } WaitForSingleObject(hWait2Exit,INFINITE); UnInitWinsock2(); return 1; }

[http_h-lm.spaces.live.com_blog_cns_C523F565A10E3B66_824.entry]: http://h-lm.spaces.live.com/blog/cns!C523F565A10E3B66!824.entry
[http_jlbookworm.spaces.live.com_blog_cns_ef3e777c24481c39_240.entry]: http://jlbookworm.spaces.live.com/blog/cns%21ef3e777c24481c39%21240.entry
[http_www.cppblog.com_niewenlong_archive_2007_08_17_30224.html]: http://www.cppblog.com/niewenlong/archive/2007/08/17/30224.html
[http_www.codeproject.com_KB_IP_iocp-multicast-udp.aspx]: http://www.codeproject.com/KB/IP/iocp-multicast-udp.aspx
[http_www.cppblog.com_johndragon_archive_2008_09_16_21845.html]: http://www.cppblog.com/johndragon/archive/2008/09/16/21845.html