IOCP模型總結
IOCP(I/O Completion Port,I/O完成端口)是性能最好的一種I/O模型。它是應用程序使用線程池處理異步I/O請求的一種機制。在處理多個并發的異步I/O請求時,以往的模型都是在接收請求是創建一個線程來應答請求。這樣就有很多的線程并行地運行在系統中。而這些線程都是可運行的,Windows內核花費大量的時間在進行線程的上下文切換,并沒有多少時間花在線程運行上。再加上創建新線程的開銷比較大,所以造成了效率的低下。
調用的步驟如下:
抽象出一個完成端口大概的處理流程:
1:創建一個完成端口。
2:創建一個線程A。
3:A線程循環調用GetQueuedCompletionStatus()函數來得到IO操作結果,這個函數是個阻塞函數。
4:主線程循環里調用accept等待客戶端連接上來。
5:主線程里accept返回新連接建立以后,把這個新的套接字句柄用CreateIoCompletionPort關聯到完成端口,然后發出一個異步的 WSASend或者WSARecv調用,因為是異步函數,WSASend/WSARecv會馬上返回,實際的發送或者接收數據的操作由WINDOWS系統去做。
6:主線程繼續下一次循環,阻塞在accept這里等待客戶端連接。
7:WINDOWS系統完成WSASend或者WSArecv的操作,把結果發到完成端口。
8:A線程里的GetQueuedCompletionStatus()馬上返回,并從完成端口取得剛完成的WSASend/WSARecv的結果。
9:在A線程里對這些數據進行處理(如果處理過程很耗時,需要新開線程處理),然后接著發出WSASend/WSARecv,并繼續下一次循環阻塞在GetQueuedCompletionStatus()這里。
歸根到底概括完成端口模型一句話:
我們不停地發出異步的WSASend/WSARecv IO操作,具體的IO處理過程由WINDOWS系統完成,WINDOWS系統完成實際的IO處理后,把結果送到完成端口上(如果有多個IO都完成了,那么就在完成端口那里排成一個隊列)。我們在另外一個線程里從完成端口不斷地取出IO操作結果,然后根據需要再發出WSASend/WSARecv IO操作。
而IOCP模型是事先開好了N個線程,存儲在線程池中,讓他們hold。然后將所有用戶的請求都投遞到一個完成端口上,然后N個工作線程逐一地從完成端口中取得用戶消息并加以處理。這樣就避免了為每個用戶開一個線程。既減少了線程資源,又提高了線程的利用率。
完成端口模型是怎樣實現的呢?我們先創建一個完成端口(::CreateIoCompletioPort())。然后再創建一個或多個工作線程,并指定他們到這個完成端口上去讀取數據。我們再將遠程
連接的套接字句柄關聯到這個完成端口(還是用::CreateIoCompletionPort())。一切就 OK了。
工作線程都干些什么呢?首先是調用::GetQueuedCompletionStatus()函數在關聯到這個完成端口上的所有套接字上等待I/O的完成。再判斷完成了什么類型的I/O。一般來說,有三種類型的I/O,OP_ACCEPT,OP_READ和OP_WIRTE。我們到數據緩沖區內讀取數據后,再投遞一個或是多個同類型的I/O即可(::AcceptEx()、::WSARecv()、::WSASend())。對讀取到的數據,我們可以按照自己的需要來進行相應的處理。
為此,我們需要一個以OVERLAPPED(重疊I/O)結構為第一個字段的per-I/O數據自定義結構。
1 typedef struct _PER_IO_DATA
2 {
3 OVERLAPPED ol; // 重疊I/O結構
4 char buf[BUFFER_SIZE]; // 數據緩沖區
5 int nOperationType; //I/O操作類型
6 #define OP_READ 1
7 #define OP_WRITE 2
8 #define OP_ACCEPT 3
9 } PER_IO_DATA, *PPER_IO_DATA;
將一個PER_IO_DATA結構強制轉化成一個OVERLAPPED結構傳給::GetQueuedCompletionStatus()函數,返回的這個PER_IO_DATA結構的的nOperationType就是I/O操作的類型。當然,這些類型都是在投遞I/O請求時自己設置的。
這樣一個IOCP服務器的框架就出來了。當然,要做一個好的IOCP服務器,還有考慮很多問題,如內存資源管理、接受連接的方法、惡意的客戶連接、包的重排序等等。以上是個人對于IOCP模型的一些理解與看法,還有待完善。另外各Winsock API的用法參見MSDN。
補充IOCP模型的實現:
//創建一個完成端口
HANDLE FCompletPort = CreateIoCompletionPort( INVALID_HANDLE_VALUE, 0,0,0 );
//接受遠程連接,并把這個連接的socket句柄綁定到剛才創建的IOCP上
AConnect = accept( FListenSock, addr, len);
CreateIoCompletionPort( AConnect, FCompletPort, NULL, 0 );
//創建CPU數*2 + 2個線程
SYSTEM_INFO si;
GetSystemInfo(&si);
for (int i=1;si.dwNumberOfProcessors*2+2;i++)
{
AThread = TRecvSendThread.Create( false );
AThread.CompletPort = FCompletPort;//告訴這個線程,你要去這個IOCP去訪問數據
}
OK,就這么簡單,我們要做的就是建立一個IOCP,把遠程連接的socket句柄綁定到剛才創建的IOCP上,最后創建n個線程,并告訴這n個線程到這個IOCP上去訪問數據就可以了。
再看一下TRecvSendThread線程都干些什么:
void TRecvSendThread.Execute(...)
{
while (!self.Terminated)
{
//查詢IOCP狀態(數據讀寫操作是否完成)
GetQueuedCompletionStatus( CompletPort, BytesTransd, CompletKey, POVERLAPPED(pPerIoDat), TIME_OUT );
if (BytesTransd !=0) .......
....;//數據讀寫操作完成
//再投遞一個讀數據請求
WSARecv( CompletKey, &(pPerIoDat->BufData), 1, BytesRecv, Flags, &(pPerIoDat->Overlap), NULL );
}
}
讀寫線程只是簡單地檢查IOCP是否完成了我們投遞的讀寫操作,如果完成了則再投遞一個新的讀寫請求。應該注意到,我們創建的所有TRecvSendThread都在訪問同一個IOCP(因為我們只創建了一個IOCP),并且我們沒有使用臨界區!難道不會產生沖突嗎?不用考慮同步問題嗎?呵呵,這正是IOCP的奧妙所在。IOCP不是一個普通的對象,不需要考慮線程安全問題。它會自動調配訪問它的線程:如果某個socket上有一個線程A正在訪問,那么線程B的訪問請求會被分配到另外一個socket。這一切都是由系統自動調配的,我們無需過問。
實例:
簡單實現,適合IOCP入門
參考:《WINDOWS網絡與通信程序設計》
1 /******************************************************************
2 *
3 * Copyright (c) 2008, xxxxx有限公司
4 * All rights reserved.
5 *
6 * 文件名稱:IOCPHeader.h
7 * 摘 要: IOCP定義文件
8 *
9 * 當前版本:1.0
10 * 作 者:吳會然
11 * 完成日期:2008-9-16
12 *
13 * 取代版本:
14 * 原 作者:
15 * 完成日期:
16 *
17 ******************************************************************/
18
19 #ifndef _IOCPHEADER_H_20080916_
20 #define _IOCPHEADER_H_20080916_
21
22 #include <WINSOCK2.H>
23 #include <windows.h>
24
25 #define BUFFER_SIZE 1024
26
27 /******************************************************************
28 * per_handle 數據
29 *******************************************************************/
30 typedef struct _PER_HANDLE_DATA
31 {
32 SOCKET s; // 對應的套接字句柄
33 sockaddr_in addr; // 對方的地址
34
35 }PER_HANDLE_DATA, *PPER_HANDLE_DATA;
36
37 /******************************************************************
38 * per_io 數據
39 *******************************************************************/
40 typedef struct _PER_IO_DATA
41 {
42 OVERLAPPED ol; // 重疊結構
43 char buf[BUFFER_SIZE]; // 數據緩沖區
44 int nOperationType; // 操作類型
45
46 #define OP_READ 1
47 #define OP_WRITE 2
48 #define OP_ACCEPT 3
49
50 }PER_IO_DATA, *PPER_IO_DATA;
51
52 #endif
53
54 /******************************************************************
55 *
56 * Copyright (c) 2008, xxxxx有限公司
57 * All rights reserved.
58 *
59 * 文件名稱:main.cpp
60 * 摘 要: iocp demo
61 *
62 * 當前版本:1.0
63 * 作 者:吳會然
64 * 完成日期:2008-9-16
65 *
66 * 取代版本:
67 * 原 作者:
68 * 完成日期:
69 *
70 ******************************************************************/
71
72 #include <iostream>
73 #include <string>
74 #include "IOCPHeader.h"
75 using namespace std;
76
77 DWORD WINAPI ServerThread( LPVOID lpParam );
78
79 int main( int argc, char *argv[] )
80 {
81 //////////////////////////////////////////////////////////////////////////
82 WSADATA wsaData;
83
84 if( 0 != WSAStartup( MAKEWORD( 2, 2 ), &wsaData ) )
85 {
86 printf( "Using %s (Status:%s)/n", wsaData.szDescription, wsaData.szSystemStatus );
89 printf( "with API versions: %d.%d to %d.%d",
90 LOBYTE( wsaData.wVersion), HIBYTE( wsaData.wVersion ),
91 LOBYTE( wsaData.wHighVersion), HIBYTE( wsaData.wHighVersion) );
92
93 return -1;
94 }
95 else
96 {
97 printf("Windows sockets 2.2 startup/n");
98 }
99 //////////////////////////////////////////////////////////////////////////
100
101 int nPort = 20055;
102
103 // 創建完成端口對象
104 // 創建工作線程處理完成端口對象的事件
105 HANDLE hIocp = ::CreateIoCompletionPort( INVALID_HANDLE_VALUE, 0, 0, 0 );
106 ::CreateThread( NULL, 0, ServerThread, (LPVOID)hIocp, 0, 0 );
107
108 // 創建監聽套接字,綁定本地端口,開始監聽
109 SOCKET sListen = ::socket( AF_INET,-SOCK_STREAM, 0 );
110
111 SOCKADDR_IN addr;
112 addr.sin_family = AF_INET;
113 addr.sin_port = ::htons( nPort );
114 addr.sin_addr.S_un.S_addr = INADDR_ANY;
115 ::bind( sListen, (sockaddr *)&addr, sizeof( addr ) );
116 ::listen( sListen, 5 );
117
118 printf( "iocp demo start
/n" );
119
120 // 循環處理到來的請求
121 while ( TRUE )
122 {
123 // 等待接受未決的連接請求
124 SOCKADDR_IN saRemote;
125 int nRemoteLen = sizeof( saRemote );
126 SOCKET sRemote = ::accept( sListen, (sockaddr *)&saRemote, &nRemoteLen );
127
128 // 接受到新連接之后,為它創建一個per_handle數據,并將他們關聯到完成端口對象
131 PPER_HANDLE_DATA pPerHandle = ( PPER_HANDLE_DATA )::GlobalAlloc( GPTR, sizeof( PPER_HANDLE_DATA ) );
134 if( pPerHandle == NULL )
135 {
136 break;
137 }
138
139 pPerHandle->s = sRemote;
140 memcpy( &pPerHandle->addr, &saRemote, nRemoteLen );
141
142 ::CreateIoCompletionPort( ( HANDLE)pPerHandle->s, hIocp, (DWORD)pPerHandle, 0 );
145
146 // 投遞一個接受請求
147 PPER_IO_DATA pIoData = ( PPER_IO_DATA )::GlobalAlloc( GPTR, sizeof( PPER_IO_DATA ) );
150 if( pIoData == NULL )
151 {
152 break;
153 }
154
155 pIoData->nOperationType = OP_READ;
156 WSABUF buf;
157 buf.buf = pIoData->buf;
158 buf.len = BUFFER_SIZE;
159
160 DWORD dwRecv = 0;
161 DWORD dwFlags = 0;
162
163 ::WSARecv( pPerHandle->s, &buf, 1, &dwRecv, &dwFlags, &pIoData->ol, NULL );
167 }
168
169 //////////////////////////////////////////////////////////////////////////
170 ERROR_PROC:
171 WSACleanup();
172 //////////////////////////////////////////////////////////////////////////
173
174 return 0;
175 }
176
177 /******************************************************************
178 * 函數介紹:處理完成端口對象事件的線程
179 * 輸入參數:
180 * 輸出參數:
181 * 返回值 :
182 *******************************************************************/
183 DWORD WINAPI ServerThread( LPVOID lpParam )
184 {
185 HANDLE hIocp = ( HANDLE )lpParam;
186 if( hIocp == NULL )
187 {
188 return -1;
189 }
190
191 DWORD dwTrans = 0;
192 PPER_HANDLE_DATA pPerHandle;
193 PPER_IO_DATA pPerIo;
194
195 while( TRUE )
196 {
197 // 在關聯到此完成端口的所有套接字上等待I/O完成
198 BOOL bRet = ::GetQueuedCompletionStatus( hIocp, &dwTrans, (LPDWORD)&pPerHandle, (LPOVERLAPPED*)&pPerIo, WSA_INFINITE );
201 if( !bRet ) // 發生錯誤
202 {
203 ::closesocket( pPerHandle->s );
204 ::GlobalFree( pPerHandle );
205 ::GlobalFree( pPerIo );
206
207 cout << "error" << endl;
208 continue;
209 }
210
211 // 套接字被對方關閉
212 if( dwTrans == 0 && ( pPerIo->nOperationType == OP_READ || pPerIo->nOperationType&nb-sp;== OP_WRITE ) )
215 {
216 ::closesocket( pPerHandle->s );
217 ::GlobalFree( pPerHandle );
218 ::GlobalFree( pPerIo );
219
220 cout << "client closed" << endl;
221 continue;
222 }
223
224 switch ( pPerIo->nOperationType )
225 {
226 case OP_READ: // 完成一個接收請求
227 {
228 pPerIo->buf[dwTrans] = '/0';
229 printf( "%s/n", pPerIo->buf );
230
231 // 繼續投遞接受操作
232 WSABUF buf;
233 buf.buf = pPerIo->buf;
234 buf.len = BUFFER_SIZE;
235 pPerIo->nOperationType = OP_READ;
236
237 DWORD dwRecv = 0;
238 DWORD dwFlags = 0;
239
240 ::WSARecv( pPerHandle->s, &buf, 1, &dwRecv, &dwFlags, &pPerIo->ol, NULL );
244 }
245 break;
246 case OP_WRITE:
247 case OP_ACCEPT:
248 break;
249
250 }
251
252 }
253
254 return 0;
255 }
本文轉自:http://blog.csdn.net/shigaopb/article/details/6114163
posted on 2012-10-23 14:46
王海光 閱讀(765)
評論(0) 編輯 收藏 引用 所屬分類:
C++