select 客户端

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

#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;
}