windows重叠I/O模型

 1 #include <stdio.h>
 2 #include <stdlib.h>
 3 #include <WinSock2.h>
 4 
 5 #define BUF_SIZE 1024
 6 void ErrorHandling(char *message);
 7 
 8 int main(int argc,char *argv[])
 9 {
10     if (argc != 2) {
11         printf("Usage : %s <port>n",argv[0]);
12         exit(1);
13     }
14 
15     WSADATA wsaData;
16     SOCKET hLinSock,hCRecvSock;
17     SOCKADDR_IN lisnAdr,recvAdr;
18     int recvAdrSz;
19     
20     WSABUF dataBuf;
21     WSAEVENT evObj;
22     WSAOVERLAPPED overlapped;
23 
24     char buf[BUF_SIZE];
25     int recvBytes = 0,flags = 0;
26 
27     if (WSAStartup(MAKEWORD(2,2),&wsaData) != 0)
28         ErrorHandling("WSAStartup() error.");
29 
30     hLinSock = socket(PF_INET,SOCK_STREAM,0);
31     memset(&lisnAdr,0,sizeof(lisnAdr));
32     lisnAdr.sin_family = AF_INET;
33     lisnAdr.sin_addr.s_addr = htonl(INADDR_ANY);
34     lisnAdr.sin_port = htons(atoi(argv[1]));
35 
36     if (bind(hLinSock,(SOCKADDR*)&lisnAdr,sizeof(lisnAdr)) == SOCKET_ERROR)
37         ErrorHandling("bind() error.");
38     if (listen(hLinSock,5) == SOCKET_ERROR)
39         ErrorHandling("listen() error.");
40 
41     recvAdrSz = sizeof(recvAdr);
42     hCRecvSock = accept(hLinSock,(SOCKADDR*)&recvAdr,&recvAdrSz);
43 
44     evObj = WSACreateEvent();
45     memset(&overlapped,sizeof(overlapped));
46     overlapped.hEvent = evObj;
47     dataBuf.len = BUF_SIZE;
48     dataBuf.buf = buf;
49     
50     if (WSARecv(hCRecvSock,&dataBuf,1,(LPDWORD)&recvBytes,(LPDWORD)&flags,&overlapped,NULL) == SOCKET_ERROR)
51     {
52         if (WSAGetLastError() == WSA_IO_PENDING)
53         {
54             puts("Background data receive");
55             WSAWaitForMultipleEvents(1,&evObj,TRUE,WSA_INFINITE,FALSE);
56             WSAGetOverlappedResult(hCRecvSock,(LPDWORD)&recvBytes,FALSE,NULL);
57         }
58         else{
59             ErrorHandling("WSARECV() error");
60         }
61     }
62 
63     printf("Received message: %sn",buf);
64     WSACloseEvent(evObj);
65     closesocket(hCRecvSock);
66     closesocket(hLinSock);
67     WSACleanup();
68     return 0;
69 }
70 
71 void ErrorHandling(char *message) {
72     fputs(message,stderr);
73     fputc(‘n‘,stderr);
74 }
75 
76 void CompressSockets(SOCKET hSockArr[],int idx,int total){
77     for (int i = idx; i < total; i++)
78         hSockArr[i] = hSockArr[i + 1];
79 }
80 
81 void CompressEvents(WSAEVENT hEventArr[],int total){
82     for (int i = idx; i < total; i++)
83         hEventArr[i] = hEventArr[i + 1];
84 }

 1 #pragma warning(disable:4996)
 2 #include <stdio.h>
 3 #include <stdlib.h>
 4 #include <string.h>
 5 #include <winsock2.h>
 6 
 7 void ErrorHandling(char *message);
 8 
 9 int main(int argc,char* argv[]) {
10     if (argc != 3) {
11         printf("Usage : %s <IP> <port>n",argv[0]);
12         exit(1);
13     }
14 
15     WSADATA wsaData;
16     SOCKET hSocket;
17     SOCKADDR_IN sendAdr;
18 
19     WSABUF dataBuf;
20     char msg[] = "Network is Computer!";
21     int sendBytes = 0;
22 
23     WSAEVENT evObj;
24     WSAOVERLAPPED overlapped;
25 
26     if (WSAStartup(MAKEWORD(2,&wsaData) != 0)
27         ErrorHandling("WSAStartup() error");
28 
29     hSocket = WSASocket(PF_INET,NULL,0,WSA_FLAG_OVERLAPPED);
30     memset(&sendAdr,sizeof(sendAdr));
31     sendAdr.sin_family = AF_INET;
32     sendAdr.sin_addr.s_addr = inet_addr(argv[1]);
33     sendAdr.sin_port = htons(atoi(argv[2]));
34 
35     if (connect(hSocket,(SOCKADDR*)&sendAdr,sizeof(sendAdr)) == SOCKET_ERROR)
36         ErrorHandling("connect() error");
37     else
38         printf("Connected.......n");
39 
40     evObj = WSACreateEvent();
41     memset(&overlapped,sizeof(overlapped));
42     overlapped.hEvent = evObj;
43     dataBuf.len = strlen(msg) + 1;
44     dataBuf.buf = msg;
45 
46     if (WSASend(hSocket,(LPDWORD)&sendBytes,NULL) == SOCKET_ERROR){
47         if (WSAGetLastError() == WSA_IO_PENDING){
48             puts("Background data send");
49             WSAWaitForMultipleEvents(1,FALSE);
50             WSAGetOverlappedResult(hSocket,(LPDWORD)&sendBytes,NULL);
51         }
52         else{
53             ErrorHandling("WSASend() error");
54         }
55     }
56     printf("Send data size:%dn",sendBytes);
57     WSACloseEvent(evObj);
58     closesocket(hSocket);
59     WSACleanup();
60     return 0;
61 }
62 
63 void ErrorHandling(char *message) {
64     fputs(message,stderr);
65     fputc(‘n‘,stderr);
66 }

 1 #include <stdio.h>
 2 #include <stdlib.h>
 3 #include <WinSock2.h>
 4 
 5 #define BUF_SIZE 1024
 6 void ErrorHandling(char *message);
 7 void CALLBACK CompRoutine(DWORD,DWORD,LPWSAOVERLAPPED,DWORD);
 8 
 9 WSABUF dataBuf;
10 char buf[BUF_SIZE];
11 int recvBytes = 0;
12 
13 int main(int argc,char *argv[])
14 {
15     if (argc != 2) {
16         printf("Usage : %s <port>n",argv[0]);
17         exit(1);
18     }
19 
20     WSADATA wsaData;
21     SOCKET hLinSock,hRecvSock;
22     SOCKADDR_IN lisnAdr,recvAdr;
23     
24     WSAEVENT evObj;
25     WSAOVERLAPPED overlapped;
26 
27     int idx,recvAdrSz,flags = 0;
28 
29     if (WSAStartup(MAKEWORD(2,&wsaData) != 0)
30         ErrorHandling("WSAStartup() error.");
31 
32     hLinSock = socket(PF_INET,0);
33     memset(&lisnAdr,sizeof(lisnAdr));
34     lisnAdr.sin_family = AF_INET;
35     lisnAdr.sin_addr.s_addr = htonl(INADDR_ANY);
36     lisnAdr.sin_port = htons(atoi(argv[1]));
37 
38     if (bind(hLinSock,sizeof(lisnAdr)) == SOCKET_ERROR)
39         ErrorHandling("bind() error.");
40     if (listen(hLinSock,5) == SOCKET_ERROR)
41         ErrorHandling("listen() error.");
42 
43     recvAdrSz = sizeof(recvAdr);
44     hRecvSock = accept(hLinSock,&recvAdrSz);
45     if (hRecvSock == INVALID_SOCKET)
46         ErrorHandling("accept() error");
47 
48     
49     memset(&overlapped,sizeof(overlapped));
50 
51     dataBuf.len = BUF_SIZE;
52     dataBuf.buf = buf;
53 
54     evObj = WSACreateEvent();
55     
56     if (WSARecv(hRecvSock,CompRoutine) == SOCKET_ERROR)
57     {
58         if (WSAGetLastError() == WSA_IO_PENDING)
59         {
60             puts("Background data receive");
61         }
62         else{
63             ErrorHandling("WSARECV() error");
64         }
65     }
66 
67     idx = WSAWaitForMultipleEvents(1,TRUE);
68     if (idx == WAIT_IO_COMPLETION)
69         puts("Overlapped I/O Completed");
70     else
71         ErrorHandling("WSARecv() error");
72     WSACloseEvent(evObj);
73     closesocket(hRecvSock);
74     closesocket(hLinSock);
75     WSACleanup();
76     return 0;
77 }
78 
79 void CALLBACK CompRoutine(DWORD dwError,DWORD szRecvBytes,LPWSAOVERLAPPED lpOverlapped,DWORD flags){
80 
81     if (dwError != 0){
82         ErrorHandling("CompRoutine error");
83     }
84     else{
85         recvBytes = szRecvBytes;
86         printf("Received message: %sn",buf);
87     }
88 }
89 
90 void ErrorHandling(char *message) {
91     fputs(message,stderr);
92     fputc(‘n‘,stderr);
93 }