IOCP程序

学习的时候，在网上找了好久，但是都没找到满意的。自己会了，写一个可以运行的吧！

memset(&pPer_IO_Data->overlapped,0,sizeof(pPer_IO_Data->overlapped));
这一句非常重要，因为开始几次没有写这句话，老是出现问题，而且还找不到原因，查了好多的资料，总算是解决了！！

也希望大家能避免这个问题！

[html] view plain copy

1. #include <WINSOCK2.H>  
2. #include <windows.h>  
3. #include <iostream>  
4. #include <basetsd.h>   
5. using namespace std;  
6.   
7. #pragma comment(lib,"ws2_32.lib")  
8.   
9. #define BUFFER_SIZE 4096  
10. #define PORT 12345  
11.   
12.   
13. #define  OP_READ 1  
14. #define  OP_WRITE 2  
15. #define  OP_ACCEPT 3  
16.   
17. static int index=0;  
18.   
19. HANDLE hCompletion;  
20. typedef struct _PER_HANDLE_DATA  
21. {  
22.     SOCKET sock;  
23.     sockaddr_in si;  
24. }PER_HANDLE_DATA,*PPER_HANDLE_DATA;  
25.   
26.   
27. typedef struct _PER_IO_DATA  
28. {  
29.     OVERLAPPED overlapped;  
30.     WSABUF wsabuf;  
31.     char buffer[BUFFER_SIZE];  
32.     int operationType;  
33. }PER_IO_DATA,*PPER_IO_DATA;  
34. DWORD WINAPI ServerThread(LPVOID lParam);  
35. int main()  
36. {  
37.     WSADATA wsa;  
38.     WSAStartup(MAKEWORD(2,2),&wsa);  
39.       
40.       
41.     hCompletion=::CreateIoCompletionPort(INVALID_HANDLE_VALUE,NULL,0,0);  
42.     HANDLE h_DoThread=CreateThread(NULL,0,ServerThread,(LPVOID)hCompletion,0,0);  
43.       
44.     SOCKET sListen=socket(AF_INET,SOCK_STREAM,0);  
45.     sockaddr_in si;  
46.     si.sin_family=AF_INET;  
47.     si.sin_port=htons(PORT);  
48.     si.sin_addr.S_un.S_addr=INADDR_ANY;  
49.     bind(sListen,(SOCKADDR *)&si,sizeof(SOCKADDR));  
50.     listen(sListen,5);  
51.       
52.     while(TRUE)  
53.     {  
54.         sockaddr_in RemoteInfo;  
55.         int RemoteLen=sizeof(sockaddr_in);  
56.         SOCKET AcceptClient=accept(sListen,(sockaddr*)&RemoteInfo,&RemoteLen);  
57.         //每当有新连接时，将为其创建PER_HANDLE_DATA数据和PER_IO_DATA数据。  
58.         //PER_HANDLE_DATA  
59.         PPER_HANDLE_DATA pPerHandleData=new PER_HANDLE_DATA();  
60.         pPerHandleData->sock=AcceptClient;  
61.         memcpy(&(pPerHandleData->si),&RemoteInfo,sizeof(sockaddr));  
62.         //将套接字句柄与完成端口相关联  
63.         CreateIoCompletionPort((HANDLE)pPerHandleData->sock,hCompletion,  
64.             (DWORD)pPerHandleData,0);  
65.           
66.           
67.         //PER_IO_DATA  
68.         PPER_IO_DATA pPer_IO_Data=new PER_IO_DATA();  
69.         pPer_IO_Data->operationType=OP_READ;  
70.         //WSARecv等函数中，缓冲区必须是WSABUF，下面对WSABUF进行初始化  
71.   
72.         pPer_IO_Data->wsabuf.buf=pPer_IO_Data->buffer;  
73.         pPer_IO_Data->wsabuf.len=BUFFER_SIZE;  
74.         DWORD dwRecv;  
75.         DWORD dwFlags=0;  
76.   
77.         //  
78.         <span style="color:#FF0000;">memset(&pPer_IO_Data->overlapped,0,sizeof(pPer_IO_Data->overlapped));</span>  
79.         //投递一个接收请求  
80.         ::WSARecv(pPerHandleData->sock,&(pPer_IO_Data->wsabuf),  
81.             1,&dwRecv,&dwFlags,&(pPer_IO_Data->overlapped),NULL);  
82.               
83.     }  
84.       
85.     return 0;  
86. }  
87.   
88. DWORD WINAPI ServerThread(LPVOID lParam)  
89. {  
90.       
91.     HANDLE hCompletion=(HANDLE)lParam;  
92.     DWORD dwTrans;  
93.     PPER_HANDLE_DATA pPer_Handle_Data;  
94.     PPER_IO_DATA pPer_IO_Data;  
95.     cout<<"ServerThread"<<endl;  
96.     while (TRUE)  
97.     {  
98.           
99.         bool bOk=GetQueuedCompletionStatus(hCompletion,&dwTrans,(LPDWORD)&pPer_Handle_Data,(LPOVERLAPPED*)&pPer_IO_Data,INFINITE);  
100.         if(!bOk)  
101.         {  
102.             cout<<GetLastError()<<endl;  
103.             //在此套接字上有错误发生  
104.             closesocket(pPer_Handle_Data->sock);  
105.               
106.             delete pPer_Handle_Data;  
107.             delete pPer_IO_Data;  
108.             continue;  
109.         }  
110.           
111.         //套接字被对方关闭  
112.         if(dwTrans==0 && pPer_IO_Data->operationType==OP_READ || pPer_IO_Data->operationType==OP_WRITE )  
113.         {  
114.             closesocket(pPer_Handle_Data->sock);  
115.             delete pPer_Handle_Data;  
116.             delete pPer_IO_Data;  
117.             continue;  
118.         }  
119.       
120.           
121.         switch(pPer_IO_Data->operationType)  
122.         {  
123.             cout<<"Switch"<<endl;  
124.         case OP_READ:  
125.             {  
126.   
127.                 pPer_IO_Data->buffer[dwTrans]='\0';  
128.                 cout<<pPer_IO_Data->buffer<<endl;  
129.                 index++;  
130.                 cout<<index<<endl;  
131.                   
132.                   
133.                 //继续投递IO操作  
134.                 WSABUF wsabuf;  
135.                 wsabuf.buf=pPer_IO_Data->buffer;  
136.                 wsabuf.len=BUFFER_SIZE;  
137.                 pPer_IO_Data->operationType=OP_READ;  
138.                 DWORD flags=0;  
139.                 WSARecv(pPer_Handle_Data->sock,&wsabuf,1,&dwTrans,&flags,&(pPer_IO_Data->overlapped),NULL);  
140.                   
141.             }  
142.         case OP_WRITE:  
143.             break;  
144.         case OP_ACCEPT:  
145.             break;  
146.         }  
147.     }  
148.     return 0;  
149. }