重叠ioserver.cpp

// 重叠ioserver.cpp : Defines the entry point for the console application.
//

#include "stdafx.h"
#include "重叠ioserver.h"
#include "winsock2.h"

#define DATA_BUFSIZE 256

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

#pragma comment(lib,"WS2_32")
/////////////////////////////////////////////////////////////////////////////
// The one and only application object

CWinApp theApp;

using namespace std;

int _tmain(int argc, TCHAR* argv[], TCHAR* envp[])
{
int nRetCode = 0;

// initialize MFC and print and error on failure
if (!AfxWinInit(::GetModuleHandle(NULL), NULL, ::GetCommandLine(), 0))
{
// TODO: change error code to suit your needs
cerr << _T("Fatal Error: MFC initialization failed") << endl;
nRetCode = 1;
}
else
{
// TODO: code your application's behavior here.
WSADATA wsaData;
WSABUF wsaBuf;
char buffer[DATA_BUFSIZE];
DWORD dwEventTotal;
DWORD dwRecvBytes;
DWORD dwFlag;
DWORD dwTransferred;
WSAEVENT wsaEvent[WSA_MAXIMUM_WAIT_EVENTS];
WSAOVERLAPPED AcceptOverlapped;
SOCKET hSocketServer,hSocketAccept;
SOCKADDR_IN hAddrServer,hAddrClient;
int saRemoteLen;
int nRet;

nRet=WSAStartup(MAKEWORD(2,2),&wsaData);
if(nRet!=0)
{
cout<<"WSAStartup Failed!"<<endl;
return 0;
}

hAddrServer.sin_family=AF_INET;
hAddrServer.sin_port=htons(8888);
hAddrServer.sin_addr.s_addr=INADDR_ANY;

hSocketServer=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);
if(hSocketServer==SOCKET_ERROR)
{
cout<<"create socket failed"<<endl;
return 0;
}

nRet=::bind(hSocketServer,(SOCKADDR*)&hAddrServer,sizeof(hAddrServer));
if(nRet!=0)
{
cout<<"bind failed!"<<endl;
return 0;
}
nRet=::listen(hSocketServer,5);
if(nRet!=0)
{
cout<<"listen failed"<<endl;
return 0;
}

saRemoteLen=sizeof(hAddrClient);

hSocketAccept=::WSAAccept(hSocketServer,(SOCKADDR*)&hAddrClient,&saRemoteLen,0,NULL);

cout<<"Client :"<<::inet_ntoa(hAddrClient.sin_addr)<<"connect"<<endl;

dwEventTotal=0;

wsaEvent[dwEventTotal]=::WSACreateEvent();

ZeroMemory(&AcceptOverlapped,sizeof(AcceptOverlapped));

wsaBuf.len=DATA_BUFSIZE;
wsaBuf.buf=buffer;
dwFlag=0;

dwEventTotal++;

if(WSARecv(hSocketAccept,&wsaBuf,1,&dwRecvBytes,&dwFlag,&AcceptOverlapped,NULL)==SOCKET_ERROR)
{
if(WSAGetLastError()!=WSA_IO_PENDING)
{
cout<<"error1"<<endl;
}

}

while(true)
{
DWORD dwIndex;
dwIndex=::WSAWaitForMultipleEvents(dwEventTotal,wsaEvent,FALSE,WSA_INFINITE,FALSE);
::WSAResetEvent(wsaEvent[dwIndex-WSA_WAIT_EVENT_0]);
::WSAGetOverlappedResult(hSocketAccept,&AcceptOverlapped,&dwTransferred,FALSE,&dwFlag);
if(dwTransferred==0)
{
cout<<"closing socket"<<hSocketAccept<<endl;
closesocket(hSocketAccept);
WSACloseEvent(wsaEvent[dwIndex-WSA_WAIT_EVENT_0]);
return 0;
}
cout<<"recv data from "<<hSocketAccept<<endl;

dwFlag=0;
ZeroMemory(&AcceptOverlapped,sizeof(WSAOVERLAPPED));
AcceptOverlapped.hEvent=wsaEvent[dwIndex-WSA_WAIT_EVENT_0];
wsaBuf.len=DATA_BUFSIZE;
wsaBuf.buf=buffer;
if(::WSARecv(hSocketAccept,&wsaBuf,1,&dwRecvBytes,&dwFlag,&AcceptOverlapped,NULL)==SOCKET_ERROR)
{
if(WSAGetLastError()!=WSA_IO_PENDING)
{
cout<<"error2"<<endl;
break;
}
}
}
::WSACleanup();
}

return nRetCode;
}