select模式学习（二）之：客户端

#include "stdafx.h"

////////////////////////////////////////////
//
// TCP Client 非阻塞模式设定
//
////////////////////////////////////////////

#pragma comment(lib,"ws2_32.lib")

#include <WinSock2.h>

#define CONNECT_IP "127.0.0.1"
#define CONNECT_PORT 1207
#define DEFAULT_BUFF_LEN 256
int _tmain(int argc, _TCHAR* argv[])
{
WSAData wsData;
SOCKET sServer;
SOCKADDR_IN addrServer;
SOCKADDR_IN addrLocal;
char sendBuff[DEFAULT_BUFF_LEN]={0};
char recvBuff[DEFAULT_BUFF_LEN]={0};
int nError;
printf(">>>>>TCP 客户端启动<<<<<<\n");
WSAStartup( MAKEWORD(2,2), &wsData );
printf("-创建客户端用SOCKET\n");
sServer = socket( AF_INET, SOCK_STREAM, IPPROTO_TCP );
if( sServer == INVALID_SOCKET )
{
printf("!!! socket failed: %d\n", WSAGetLastError());
WSACleanup();
return -1;
}
addrServer.sin_family = AF_INET;
addrServer.sin_addr.S_un.S_addr = inet_addr( CONNECT_IP );
addrServer.sin_port = htons(CONNECT_PORT);
printf("-设定服务器地址信息: %s:%d\n",inet_ntoa(addrServer.sin_addr), ntohs(addrServer.sin_port));

//设定本地用地址和端口
//addrLocal.sin_family = AF_INET;
//addrLocal.sin_addr.S_un.S_addr = inet_addr( CONNECT_IP );
//addrLocal.sin_port = htons( 2701 );
//printf("-绑定本地用地址和端口: %s:%d\n", inet_ntoa(addrLocal.sin_addr), ntohs(addrLocal.sin_port));
////nError = bind( sServer, (const sockaddr*)&addrLocal, sizeof(addrLocal) );
//if( nError == SOCKET_ERROR )
//{
//printf("!!! bind failed: %d\n", WSAGetLastError());
//WSACleanup();
//return -1;
//}

printf("-连接指定服务器\n");
nError = connect( sServer, (const sockaddr*)&addrServer, sizeof(addrServer) );
if( nError == SOCKET_ERROR )
{
printf("!!! connect failed: %d\n", WSAGetLastError());
closesocket( sServer );
WSACleanup();
return -1;
}
///////////////////////////////////
// 客户端 SOCKET为非阻塞模式
//
///////////////////////////////////
DWORD nMode = 1;
nError = ioctlsocket( sServer, FIONBIO, &nMode );
if( nError == SOCKET_ERROR )
{
printf("!!! ioctlsocket failed: %d\n", WSAGetLastError());
closesocket( sServer );
WSACleanup();
return -1;
}
printf("-SOCKET模式设定: %s\n", (nMode==0?"阻塞模式设定":"非阻塞模式设定"));

printf("-开始准备送信受信\n");
int nInputLen = 0;
int nIndex = 0;
int nLeft = 0;
fd_set fdRead;
fd_set fdWrite;
timeval tv={10,0};
while(true)
{
FD_ZERO( &fdRead );
FD_ZERO( &fdWrite );
FD_SET( sServer, &fdRead );
FD_SET( sServer, &fdWrite );
nError = select( 0, &fdRead, &fdWrite, NULL, &tv );
if( nError == 0 )
{
printf("!!! select time out\n");
continue;
}
else if( nError < 0 )
{
printf("!!! select failed: %d\n", WSAGetLastError());
break;
}

//发现SOCKET可读/可写
if( FD_ISSET( sServer, &fdRead) )
{
memset( recvBuff, 0, sizeof(recvBuff) );
nError = recv( sServer, recvBuff, sizeof(recvBuff)-1, 0 );
;;;;;if( nError == SOCKET_ERROR )
{
printf("!!! recv failed: %d\n", WSAGetLastError());
break;
}
else if( nError == 0 )
{
printf("-Server Has Closed.\n");
break;
}
recvBuff[nError] = '\0';
printf("-Received: %s\n\n", recvBuff);
}
if( FD_ISSET(sServer, &fdWrite) )
{
printf("Input Message You Want to Send( Quit ):\n");
fflush( stdin );
memset( sendBuff, 0, sizeof(sendBuff) );
fgets( sendBuff, sizeof(sendBuff)-1, stdin );
nInputLen = strlen( sendBuff );
sendBuff[nInputLen-1] = '\0';//去掉回车符
--nInputLen;
nIndex = 0;
nLeft = nInputLen;
if(strcmp(sendBuff, "Quit")==0)
{
break;
}
//将输入的数据发送过去
while( nLeft > 0)
{
nError = send( sServer, &sendBuff[nIndex], nLeft, 0 );
;;;;if( nError == SOCKET_ERROR )
{
printf("!!! send failed: %d\n",WSAGetLastError());
closesocket( sServer );
WSACleanup();
return -1;
}
else if( nError == 0 )
{
printf("-Send OK.\n");
break;
}

nLeft -= nError;
nIndex += nError;
}
printf("-Has send: %s\n",sendBuff);
}
}//while(true)

printf("-关闭SOCKET\n");
closesocket(sServer);
WSACleanup();
return 0;
}