簡單實現，適合IOCP入門
參考：《WINDOWS網路與通訊程式設計》
- /******************************************************************
- *
- * Copyright (c) 2008, xxxxx有限公司
- * All rights reserved.
- *
- * 檔案名稱：IOCPHeader.h
- * 摘 要： IOCP定義檔案
- *
- * 目前的版本：1.0
- * 作 者：吳會然
- * 完成日期：2008-9-16
- *
- * 取代版本：
- * 原 作者：
- * 完成日期：
- *
- ******************************************************************/
- #ifndef _IOCPHEADER_H_20080916_
- #define _IOCPHEADER_H_20080916_
- #include <WINSOCK2.H>
- #include <windows.h>
- #define BUFFER_SIZE 1024
- /******************************************************************
- * per_handle 資料
- *******************************************************************/
- typedef struct _PER_HANDLE_DATA
- {
- SOCKET s; // 對應的通訊端控制代碼
- sockaddr_in addr; // 對方的地址
- }PER_HANDLE_DATA, *PPER_HANDLE_DATA;
- /******************************************************************
- * per_io 資料
- *******************************************************************/
- typedef struct _PER_IO_DATA
- {
- OVERLAPPED ol; // 重疊結構
- char buf[BUFFER_SIZE]; // 資料緩衝區
- int nOperationType; // 操作類型
- #define OP_READ 1
- #define OP_WRITE 2
- #define OP_ACCEPT 3
- }PER_IO_DATA, *PPER_IO_DATA;
- #endif
- /******************************************************************
- *
- * Copyright (c) 2008, xxxxx有限公司
- * All rights reserved.
- *
- * 檔案名稱：main.cpp
- * 摘 要： iocp demo
- *
- * 目前的版本：1.0
- * 作 者：吳會然
- * 完成日期：2008-9-16
- *
- * 取代版本：
- * 原 作者：
- * 完成日期：
- *
- ******************************************************************/
- #include <iostream>
- #include <string>
- #include "IOCPHeader.h"
- using namespace std;
- DWORD WINAPI ServerThread( LPVOID lpParam );
- int main( int argc, char *argv[] )
- {
- //////////////////////////////////////////////////////////////////////////
- WSADATA wsaData;
- if( 0 != WSAStartup( MAKEWORD( 2, 2 ), &wsaData ) )
- {
- printf( "Using %s (Status:%s)/n", wsaData.szDescription, wsaData.szSystemStatus );
- printf( "with API versions: %d.%d to %d.%d",
- LOBYTE( wsaData.wVersion), HIBYTE( wsaData.wVersion ),
- LOBYTE( wsaData.wHighVersion), HIBYTE( wsaData.wHighVersion) );
- return -1;
- }
- else
- {
- printf("Windows sockets 2.2 startup/n");
- }
- //////////////////////////////////////////////////////////////////////////
- int nPort = 20055;
- // 建立完成連接埠對象
- // 建立背景工作執行緒處理完成連接埠對象的事件
- HANDLE hIocp = ::CreateIoCompletionPort( INVALID_HANDLE_VALUE, 0, 0, 0 );
- ::CreateThread( NULL, 0, ServerThread, (LPVOID)hIocp, 0, 0 );
- // 建立監聽通訊端，綁定本地連接埠，開始監聽
- SOCKET sListen = ::socket( AF_INET, SOCK_STREAM, 0 );
- SOCKADDR_IN addr;
- addr.sin_family = AF_INET;
- addr.sin_port = ::htons( nPort );
- addr.sin_addr.S_un.S_addr = INADDR_ANY;
- ::bind( sListen, (sockaddr *)&addr, sizeof( addr ) );
- ::listen( sListen, 5 );
- printf( "iocp demo start....../n" );
- // 迴圈處理到來的請求
- while ( TRUE )
- {
- // 等待接受未決的串連請求
- SOCKADDR_IN saRemote;
- int nRemoteLen = sizeof( saRemote );
- SOCKET sRemote = ::accept( sListen, (sockaddr *)&saRemote, &nRemoteLen );
- // 接受到新串連之後，為它建立一個per_handle資料，並將他們關聯到完成連接埠對象
- PPER_HANDLE_DATA pPerHandle = ( PPER_HANDLE_DATA )::GlobalAlloc( GPTR, sizeof( PPER_HANDLE_DATA ) );
- if( pPerHandle == NULL )
- {
- break;
- }
- pPerHandle->s = sRemote;
- memcpy( &pPerHandle->addr, &saRemote, nRemoteLen );
- ::CreateIoCompletionPort( ( HANDLE)pPerHandle->s, hIocp, (DWORD)pPerHandle, 0 );
- // 投遞一個接受請求
- PPER_IO_DATA pIoData = ( PPER_IO_DATA )::GlobalAlloc( GPTR, sizeof( PPER_IO_DATA ) );
- if( pIoData == NULL )
- {
- break;
- }
- pIoData->nOperationType = OP_READ;
- WSABUF buf;
- buf.buf = pIoData->buf;
- buf.len = BUFFER_SIZE;
-
- DWORD dwRecv = 0;
- DWORD dwFlags = 0;
- ::WSARecv( pPerHandle->s, &buf, 1, &dwRecv, &dwFlags, &pIoData->ol, NULL );
- }
- //////////////////////////////////////////////////////////////////////////
- ERROR_PROC:
- WSACleanup();
- //////////////////////////////////////////////////////////////////////////
- return 0;
- }
- /******************************************************************
- * 函數介紹：處理完成連接埠對象事件的線程
- * 輸入參數：
- * 輸出參數：
- * 傳回值 ：
- *******************************************************************/
- DWORD WINAPI ServerThread( LPVOID lpParam )
- {
- HANDLE hIocp = ( HANDLE )lpParam;
- if( hIocp == NULL )
- {
- return -1;
- }
- DWORD dwTrans = 0;
- PPER_HANDLE_DATA pPerHandle;
- PPER_IO_DATA pPerIo;
-
- while( TRUE )
- {
- // 在關聯到此完成連接埠的所有通訊端上等待I/O完成
- BOOL bRet = ::GetQueuedCompletionStatus( hIocp, &dwTrans, (LPDWORD)&pPerHandle, (LPOVERLAPPED*)&pPerIo, WSA_INFINITE );
- if( !bRet ) // 發生錯誤
- {
- ::closesocket( pPerHandle->s );
- ::GlobalFree( pPerHandle );
- ::GlobalFree( pPerIo );
- cout << "error" << endl;
- continue;
- }
- // 通訊端被對方關閉
- if( dwTrans == 0 && ( pPerIo->nOperationType == OP_READ || pPerIo->nOperationType == OP_WRITE ) )
- {
- ::closesocket( pPerHandle->s );
- ::GlobalFree( pPerHandle );
- ::GlobalFree( pPerIo );
- cout << "client closed" << endl;
- continue;
- }
- switch ( pPerIo->nOperationType )
- {
- case OP_READ: // 完成一個接收請求
- {
- pPerIo->buf[dwTrans] = '/0';
- printf( "%s/n", pPerIo->buf );
- // 繼續投遞接受操作
- WSABUF buf;
- buf.buf = pPerIo->buf;
- buf.len = BUFFER_SIZE;
- pPerIo->nOperationType = OP_READ;
-
- DWORD dwRecv = 0;
- DWORD dwFlags = 0;
-
- ::WSARecv( pPerHandle->s, &buf, 1, &dwRecv, &dwFlags, &pPerIo->ol, NULL );
- }
- break;
- case OP_WRITE:
- case OP_ACCEPT:
- break;
- }
- }
- return 0;
- }
