Winsock Select模型范例
服务器端:
#include <Winsock2.h>
#include <stdio.h>
void InitSocket()
{
WORD wVersionRequested;
WSADATA wsaData;
int err;
wVersionRequested = MAKEWORD( 2, 2 );
err = WSAStartup( wVersionRequested, &wsaData );
if ( err != 0 ) {
/* Tell the user that we could not find a usable */
/* WinSock DLL. */
return;
}
/* Confirm that the WinSock DLL supports 2.2.*/
/* Note that if the DLL supports versions greater */
/* than 2.2 in addition to 2.2, it will still return */
/* 2.2 in wVersion since that is the version we */
/* requested. */
if ( LOBYTE( wsaData.wVersion ) != 2 ||
HIBYTE( wsaData.wVersion ) != 2 ) {
/* Tell the user that we could not find a usable */
/* WinSock DLL. */
WSACleanup( );
return;
}
}
int main(int argc, char* argv[])
{
InitSocket();
SOCKET sock=::socket(AF_INET,SOCK_STREAM,0);
if(sock==INVALID_SOCKET)
{
printf("创建socket失败");
WSACleanup( );
return 0;
}
//::sockaddr_in serAddr;
SOCKADDR_IN serAddr;
serAddr.sin_family=AF_INET;
serAddr.sin_port=::htons(5150);
serAddr.sin_addr.s_addr = htonl(INADDR_ANY);
if(SOCKET_ERROR==bind(sock,(sockaddr*)&serAddr,sizeof(sockaddr)))
{
printf("绑定失败");
WSACleanup( );
return 0;
}
listen(sock,5);
fd_set fdSocket;
FD_ZERO(&fdSocket);
FD_SET(sock,&fdSocket);
while(true)
{
fd_set fdRead=fdSocket;
int nRet=::select(0,&fdRead,NULL,NULL,NULL);
if(nRet>0)
{
for(int i=0;i<(int)fdSocket.fd_count;i++)
{
if(FD_ISSET(fdSocket.fd_array[i],&fdRead))
{
//有新的连接到达
if(fdSocket.fd_array[i]==sock)
{
if(fdSocket.fd_count<FD_SETSIZE)
{
::sockaddr_in addr;
int len=sizeof(sockaddr);
SOCKET client=accept(sock,(sockaddr*)&addr,&len);
printf("接收到新的连接 : %s/n",::inet_ntoa(addr.sin_addr));
FD_SET(client,&fdSocket);
}
else
{
printf("too many connections/n");
}
}
else
{
char*buf=new char[1024];
int nRecv=::recv(fdSocket.fd_array[i],buf,1024,0);
if(nRecv>0)
{
buf[nRecv]='/0';
printf("收到数据: %s/n",buf);
}
else
{
::closesocket(fdSocket.fd_array[i]);
FD_CLR(fdSocket.fd_array[i],&fdSocket);
printf("一客户端断开连接/n");
}
delete []buf;
}
}
}
}
}
return 0;
}
客户端:
#include <winsock2.h>
#include <stdio.h>
#include <memory.h>
#include <iostream>
#include <string>
using namespace std;
void main(int argc, char **argv)
{
WSADATA wsaData;
SOCKET s;
SOCKADDR_IN ServerAddr;
int Port = 5150;
int Ret, Ret1;
string Data;
char DataServe[8];
memset(DataServe, 0, 8);
char pause;
if (argc <= 1)
{
printf("USAGE: tcpclient <Server IP address>./n");
return;
}
// Initialize Winsock version 2.2
if ((Ret = WSAStartup(MAKEWORD(2,2), &wsaData)) != 0)
{
// NOTE: Since Winsock failed to load we cannot use WSAGetLastError
// to determine the error code as is normally done when a Winsock
// API fails. We have to report the return status of the function.
printf("WSAStartup failed with error %d/n", Ret);
return;
}
// Create a new socket to make a client connection.
if ((s = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP))
== INVALID_SOCKET)
{
printf("socket failed with error %d/n", WSAGetLastError());
WSACleanup();
return;
}
// Setup a SOCKADDR_IN structure that will be used to connect
// to a listening server on port 5150. For demonstration
// purposes, we required the user to supply an IP address
// on the command line and we filled this field in with the
// data from the user.
ServerAddr.sin_family = AF_INET;
ServerAddr.sin_port = htons(Port);
ServerAddr.sin_addr.s_addr = inet_addr(argv[1]);
// Make a connection to the server with socket s.
printf("We are trying to connect to %s:%d.../n",
inet_ntoa(ServerAddr.sin_addr), htons(ServerAddr.sin_port));
if (connect(s, (SOCKADDR *) &ServerAddr, sizeof(ServerAddr))
== SOCKET_ERROR)
{
printf("connect failed with error %d/n", WSAGetLastError());
closesocket(s);
WSACleanup();
return;
}
printf("Our connection succeeded./n");
// At this point you can start sending or receiving data on
// the socket s. We will just send a hello message to the server.
printf("We will now try to send a hello message./n");
for(int i = 0; i<3; i++)
{
cin>>Data;
if ((Ret = send(s, Data.c_str()/*"Hello"*/, Data.size(), 0)) == SOCKET_ERROR)
{
printf("send failed with error %d/n", WSAGetLastError());
closesocket(s);
WSACleanup();
return;
}
printf("We successfully sent %d byte(s)./n", Ret);
}
// When you are finished sending and receiving data on socket s,
// you should close the socket.
printf("We are closing the connection./n");
closesocket(s);
// When your application is finished handling the connection, call
// WSACleanup.
WSACleanup();
cin>>pause;
}