WinSock IO模型四: 重叠I/O (完成例程)

完成例程模型相比与事件通知模型有个很大的优点就是不再受64个消息的限制,一个线程可以同时管理成百上千个socket连接,且保持较高的性能。 
完成例程相比与完成端口较为逊色,因为它的性能不能随着系统CPU数量的增长而线程增长,不过在我看来已经很强了,呵呵~! 
说白了,这些连接都是由系统来帮你管理的。你只需做的一件事就是:开启一个线程来accept进来的连接,剩下的工作交由系统来处理。而你,则需要提供给系统一个回调函数,发生新的网络事件的时候系统将执行这个函数: 

void CALLBACK WorkerRoutine(DWORD Error, DWORD BytesTransferred, LPWSAOVERLAPPED Overlapped, DWORD InFlags) 
然后告诉系统用WorkerRoutine函数处理接收到的数据: 
WSARecv( m_socket, @FBuf, 1, dwTemp, dwFlag, @m_overlap, WorkerRoutine ); 
然后......没有什么然后了,系统什么都给你做了! 

源码--------------------------------------------------------------------------------------------- 

#pragma comment(lib,"ws2_32.lib") 
#include <winsock2.h> 
#include <stdio.h> 
#define DATA_BUFSIZE 1024        // 接收缓冲区大小 
#define MAXSESSION 10000        // 最大连接数 
typedef struct _SOCKET_INFORMATION { 
   OVERLAPPED Overlapped; 
   SOCKET Socket; 
   WSABUF DataBuf; 
   DWORD BytesSEND; 
   DWORD BytesRECV; 
} SOCKET_INFORMATION, * LPSOCKET_INFORMATION; 


SOCKET        ListenSocket = INVALID_SOCKET; 
DWORD   Flags = 0;                                                                // WSARecv的参数 
void CALLBACK WorkerRoutine(DWORD Error, DWORD BytesTransferred,LPWSAOVERLAPPED Overlapped, DWORD InFlags); 
DWORD WINAPI AcceptThread(LPVOID lpParameter) 
{ 
        WSADATA wsaData; 
        WSAStartup(MAKEWORD(2,2),&wsaData); 
        ListenSocket = WSASocket(AF_INET,SOCK_STREAM,IPPROTO_TCP,NULL,NULL,WSA_FLAG_OVERLAPPED); 
        SOCKADDR_IN ServerAddr; 
        ServerAddr.sin_family = AF_INET; 
        ServerAddr.sin_addr.S_un.S_addr = htonl(INADDR_ANY); 
        ServerAddr.sin_port = htons(1234); 
        bind(ListenSocket,(LPSOCKADDR)&ServerAddr,sizeof(ServerAddr)); 
        listen(ListenSocket,100); 
        printf("listenning...\n"); 
        SOCKADDR_IN ClientAddr; 
        int addr_length=sizeof(ClientAddr); 
        while (TRUE) 
        { 
                LPSOCKET_INFORMATION  SI = new SOCKET_INFORMATION; 
                if ((SI->Socket = accept(ListenSocket,(SOCKADDR*)&ClientAddr, &addr_length)) != INVALID_SOCKET) 
                { 
                        printf("accept ip:%s port:%d\n",inet_ntoa(ClientAddr.sin_addr),ClientAddr.sin_port); 
                        memset(&SI->Overlapped,0,sizeof(WSAOVERLAPPED)); 
                        SI->DataBuf.buf = new char[DATA_BUFSIZE]; 
                        SI->DataBuf.len = DATA_BUFSIZE; 
                        memset(SI->DataBuf.buf,0,DATA_BUFSIZE); 
                        if(WSARecv(SI->Socket, &SI->DataBuf, 1, &SI->BytesRECV, &Flags, &SI->Overlapped, WorkerRoutine) == SOCKET_ERROR) 
                        { 
                                int err = WSAGetLastError(); 
                                if(WSAGetLastError() != WSA_IO_PENDING) 
                                { 
                                        printf("disconnect\n"); 
                                        closesocket(SI->Socket); 
                                        delete [] SI->DataBuf.buf; 
                                        delete SI; 
                                        continue; 
                                } 
                        } 
                } 
                 
        } 
return FALSE; 
} 


void CALLBACK WorkerRoutine(DWORD Error, DWORD BytesTransferred, LPWSAOVERLAPPED Overlapped, DWORD InFlags) 
{ 
        LPSOCKET_INFORMATION SI = (LPSOCKET_INFORMATION)Overlapped; 
        if (Error != 0 || BytesTransferred == 0) 
        { 
                printf("disconnect\n"); 
                closesocket(SI->Socket); 
                delete [] SI->DataBuf.buf; 
                delete SI; 
                return; 
        } 
        //使用数据 
        printf("call back:%s\n",SI->DataBuf.buf); 
        memset(SI->DataBuf.buf,0,DATA_BUFSIZE); 
         
        if(WSARecv(SI->Socket, &SI->DataBuf, 1, &SI->BytesRECV, &Flags, &SI->Overlapped, WorkerRoutine) == SOCKET_ERROR) 
        { 
                int err = WSAGetLastError(); 
                if(WSAGetLastError() != WSA_IO_PENDING) 
                { 
                        printf("disconnect\n"); 
                        closesocket(SI->Socket); 
                        delete [] SI->DataBuf.buf; 
                        delete SI; 
                        return; 
                } 
        } 
} 


void main()   
{ 
        HANDLE hThreads = CreateThread(NULL, 0, AcceptThread, NULL, NULL, NULL); 
           
        WaitForSingleObject(hThreads,INFINITE); 
        printf("exit\n"); 
        CloseHandle(hThreads); 
} 



 

  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
重叠IO模型之OverLapped完成例程模型WSACompletionRoutineServer VS2010 基础入门 客户端与服务器端 客户端向服务器端发送数据 可接收多个客户端 #include #include #pragma comment (lib, "ws2_32.lib") #define PORT 8088 #define MSG_SIZE 1024 SOCKET g_sConnect; bool g_bConnect = false; typedef struct { WSAOVERLAPPED overLap; WSABUF wsaBuf; char chMsg[MSG_SIZE]; DWORD nRecvNum; DWORD nFlags; SOCKET sClient; }PRE_IO_OPERATION_DATA, *LP_PER_IO_OPERATION_DATA; void CALLBACK CompletionRoutine(DWORD dwError, DWORD dwTrans, LPWSAOVERLAPPED lpOverlap, DWORD nFlags); DWORD WINAPI workThread(LPVOID lp) { LP_PER_IO_OPERATION_DATA lpData; while(TRUE) { if (g_bConnect) // 有新的连接 { // 为lpData分配空间并初始化 lpData = (LP_PER_IO_OPERATION_DATA)HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(PRE_IO_OPERATION_DATA)); lpData->wsaBuf.len = MSG_SIZE; lpData->wsaBuf.buf = lpData->chMsg; lpData->sClient = g_sConnect; WSARecv(lpData->sClient, &lpData->wsaBuf, 1, &lpData->nRecvNum, &lpData->nFlags, &lpData->overLap, CompletionRoutine); g_bConnect = false; // 处理完毕 } SleepEx(1000, TRUE); } return 0; } // 系统在WSARecv收到信息后,自动调用此函数,并传入参数--回调函数 void CALLBACK CompletionRoutine(DWORD dwError, DWORD dwTrans, LPWSAOVERLAPPED lpOverlap, DWORD nFlags) { LP_PER_IO_OPERATION_DATA lpData = (LP_PER_IO_OPERATION_DATA)lpOverlap; if (0 != dwError) // 接收失败 { printf("Socket %d Close!\n", lpData->sClient); closesocket(lpData->sClient); HeapFree(GetProcessHeap(), 0, lpData); } else // 接收成功 { lpData->chMsg[dwTrans] = '\0'; send(lpData->sClient, lpData->chMsg, dwTrans, 0); printf("Socket:%d MSG: %s \n", lpData->sClient, lpData->chMsg); memset(&lpData->overLap, 0, sizeof(WSAOVERLAPPED)); lpData->wsaBuf.len = MSG_SIZE; lpData->wsaBuf.buf = lpData->chMsg; // 继续接收来自客户端的数据 实现 WSARecv与CompletionRoutine循环 WSARecv(lpData->sClient, &lpData->wsaBuf,1, &lpData->nRecvNum, &lpData->nFlags, &lpData->overLap, CompletionRoutine); } } int main() { WSADATA wsaData; WSAStartup(0x0202, &wsaData); SOCKET sListen; sListen = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP); sockaddr_in addrListen; addrListen.sin_family = AF_INET; addrListen.sin_port = htons(PORT); addrListen.sin_addr.S_un.S_addr = htonl(ADDR_ANY); int nErrorCode = 0; nErrorCode = bind(sListen, (sockaddr*)&addrListen, sizeof(sockaddr)); nErrorCode = listen(sListen, 5); DWORD nThreadID; CreateThread(NULL, 0, workThread, NULL, 0, &nThreadID); sockaddr_in addrConnect; int nAddrLen = sizeof(sockaddr_in); printf("Server Started!\n"); while(TRUE) { g_sConnect= accept(sListen, (sockaddr*)&addrConnect, &nAddrLen); if (INVALID_SOCKET == g_sConnect) { return -1; } g_bConnect = true; // 连接成功 printf("Accept Client :%s -- PORT:%d\n", inet_ntoa(addrConnect.sin_addr), htons(addrConnect.sin_port)); } return 0; }
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值