一个可靠的发送和接受数据函数

转载 2015年07月08日 18:12:03

 在广域网条件下,客户端与服务器打交道,如果仅仅是简单的send-recv方式,那么潜在的bug是无法避免的,尤其是在一个指令序列不断交互,问题更大了。

例如,你预期的情况是这样的:

C->S: Hello,server.  \r\n

S->C:Hello,client \r\n

C->S:Tell me your name \r\n

S->C:My name is Jimmy \r\n

而实际却得到这样的结果:

C->S: Hello,server \r\n

S->C:Hello,cli

C->S:Tell me your name \r\n

S->C:eint  \r\n My name is Jimmy \r\n

为此,必须充分考虑广域网的阻塞,延迟等非常规情况出现。

  1. #include <iostream>  
  2. #include <assert.h>  
  3. #include <winsock2.h>  
  4.   
  5. #pragma comment(lib, "ws2_32.lib")  
  6.   
  7. using namespace std;  
  8.   
  9. class CMySocket  
  10. {  
  11. public:  
  12.     CMySocket(){};  
  13.     ~CMySocket(){};  
  14.     bool  SendData();  
  15.     bool  ReceiveData();  
  16. private:  
  17.     SOCKET  m_socket;  
  18.     char*   m_dataBuf;  
  19.     int     m_dataLen;  
  20.     int     m_waitSeconds;  
  21. };  
  22.   
  23.   
  24. bool CMySocket::SendData()  
  25. {  
  26.     int idx = 0,res;  
  27.     int nLeft=m_dataLen;  
  28.   
  29.     if(m_dataLen==0){  
  30.         return false;  
  31.     }  
  32.   
  33.     fd_set  fdwrite;  
  34.     timeval time;  
  35.   
  36.     time.tv_sec = m_waitSeconds;  
  37.     time.tv_usec = 0;  
  38.   
  39.     while(1)  
  40.     {  
  41.         FD_ZERO(&fdwrite);  
  42.   
  43.         FD_SET(m_socket,&fdwrite);  
  44.   
  45.         if((res = select(m_socket+1,NULL,&fdwrite,NULL,&time)) == SOCKET_ERROR)  
  46.         {  
  47.             FD_CLR(m_socket,&fdwrite);  
  48.             return false;  
  49.         }  
  50.   
  51.         if(!res)//timeout  
  52.         {  
  53.             FD_CLR(m_socket,&fdwrite);  
  54.             return false;  
  55.         }  
  56.   
  57.         if(res && FD_ISSET(m_socket,&fdwrite))  
  58.         {  
  59.             if(nLeft > 0)  
  60.             {  
  61.                 if((res = send(m_socket,&m_dataBuf[idx],nLeft,0)) == SOCKET_ERROR)  
  62.                 {  
  63.                     FD_CLR(m_socket,&fdwrite);  
  64.                 }  
  65.                 if(!res)  
  66.                     break;  
  67.                 nLeft -= res;  
  68.                 idx += res;  
  69.             }  
  70.             else  
  71.                 break;  
  72.         }  
  73.     }  
  74.   
  75.     FD_CLR(m_socket,&fdwrite);  
  76.   
  77.     return true;  
  78. }  
  79.   
  80.   
  81. bool CMySocket::ReceiveData()  
  82. {  
  83.     int res,i = 0;  
  84.     fd_set fdread;  
  85.     timeval time;  
  86.   
  87.     time.tv_sec = m_waitSeconds;  
  88.     time.tv_usec = 0;  
  89.   
  90.     assert(m_dataBuf);  
  91.       
  92.     if(m_dataBuf == NULL)  
  93.         return false;  
  94.   
  95.     while(1)  
  96.     {  
  97.         FD_ZERO(&fdread);  
  98.   
  99.         FD_SET(m_socket,&fdread);  
  100.   
  101.         if((res = select(m_socket+1, &fdread, NULL, NULL, &time)) == SOCKET_ERROR)  
  102.         {  
  103.             FD_CLR(m_socket,&fdread);  
  104.             return false;  
  105.         }  
  106.   
  107.         if(!res)//timeout  
  108.         {  
  109.             FD_CLR(m_socket,&fdread);  
  110.             return false;  
  111.         }  
  112.   
  113.         if(res && FD_ISSET(m_socket,&fdread))  
  114.         {  
  115.             if(i >= m_dataLen)  
  116.             {  
  117.                 FD_CLR(m_socket,&fdread);  
  118.                 return false;//lack of memory  
  119.             }  
  120.             if(recv(m_socket,&m_dataBuf[i++],1,0) == SOCKET_ERROR)  
  121.             {  
  122.                 FD_CLR(m_socket,&fdread);  
  123.                 return false;  
  124.             }  
  125.             if(m_dataBuf[i-1]=='\n')  
  126.             {  
  127.                 m_dataBuf[i] = '\0';  
  128.                 break;  
  129.             }  
  130.         }  
  131.     }  
  132.   
  133.     FD_CLR(m_socket,&fdread);  
  134.     return true;  
  135. }  


注解:

select函数用于确定一个或多个套接口的状态。对每一个套接口,调用者可查询它的可读性、可写性及错误状态信息。用fd_set结构来表示一组等待检查的套接口。

原型

int select(
int nfds,
fd_set* readfds,
fd_set* writefds,
fd_set* exceptfds,
const struct timeval* timeout
);

nfds:本参数忽略,仅起到兼容作用。
readfds:(可选)指针,指向一组等待可读性检查的套接口。
writefds:(可选)指针,指向一组等待可写性检查的套接口。
exceptfds:(可选)指针,指向一组等待错误检查的套接口。
timeout:select()最多等待时间,对阻塞操作则为NULL。

   FD_CLR(s,*set):从集合set中删除描述字s。
   FD_ISSET(s,*set):若s为集合中一员,非零;否则为零。
   FD_SET(s,*set):向集合添加描述字s。
   FD_ZERO(*set):将set初始化为空集NULL。

 

返回值:


   select()调用返回处于就绪状态并且已经包含在fd_set结构中的描述字总数;如果超时则返回0;否则的话,返回SOCKET_ERROR错误,应用程序可通过WSAGetLastError()获取相应错误代码。

错误代码:
   WSANOTINITIALISED:在使用此API之前应首先成功地调用WSAStartup()。
   WSAENETDOWN:WINDOWS套接口实现检测到网络子系统失效。
   WSAEINVAL:超时时间值非法。
   WSAEINTR:通过一个WSACancelBlockingCall()来取消一个(阻塞的)调用。
   WSAEINPROGRESS:一个阻塞的WINDOWS套接口调用正在运行中。
   WSAENOTSOCK:描述字集合中包含有非套接口的元素。


附带别人的一篇文章:

  1. /* 
  2.  
  3. Socket中如何设置连接超时【转】 
  4.  
  5. http://fcxz.blogbus.com/logs/1564481.html 
  6.  
  7.  
  8. 设置connect的超时很简单,CSDN上也有人提到过使用select,但却没有一个令人满意与完整的答案。 
  9. 偶所讲的也正是select函数,此函数集成在winsock1.1中, 
  10. 简单点讲,"作用使那些想避免在套接字调用过程中被锁定的应用程序,采取一种有序的方式,同时对多个套接字进行管理"(《Windows网络编程技术》原话)。 
  11. 使用方法与解释请见《Windows网络编程技术》。 
  12.    
  13. 在使用此函数前,需先将socket设置为非锁定模式,这样,在connect时,才会立马跳过, 
  14. 同时,通常也会产生一个WSAEWOULDBLOCK错误,这个错误没关系。再执行select则是真正的超时。 
  15.  
  16. */  
  17.   
  18. void main()  
  19. {  
  20.   
  21.     WSADATA wsd;  
  22.     SOCKET cClient;  
  23.     int ret;  
  24.     struct sockaddr_in server;  
  25.     hostent *host=NULL;  
  26.   
  27.     if(WSAStartup(MAKEWORD(2,0),&wsd)){return 0;}  
  28.     cClient=socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);  
  29.       
  30.     if(cClient==INVALID_SOCKET){  
  31.         return 0;  
  32.     }  
  33.   
  34.     //set Recv and Send time out  
  35.     int TimeOut=6000; //设置发送超时6秒  
  36.       
  37.     if(::setsockopt(cClient,SOL_SOCKET,SO_SNDTIMEO,(char *)&TimeOut,sizeof(TimeOut))==SOCKET_ERROR){  
  38.         return 0;  
  39.     }  
  40.   
  41.     TimeOut=6000;//设置接收超时6秒  
  42.     if(::setsockopt(cClient,SOL_SOCKET,SO_RCVTIMEO,(char *)&TimeOut,sizeof(TimeOut))==SOCKET_ERROR){  
  43.         return 0;  
  44.     }  
  45.   
  46.     //设置非阻塞方式连接  
  47.     unsigned long ul = 1;  
  48.     ret = ioctlsocket(cClient, FIONBIO, (unsigned long*)&ul);  
  49.       
  50.     if(ret==SOCKET_ERROR)  
  51.         return 0;  
  52.   
  53.     //连接  
  54.     server.sin_family = AF_INET;  
  55.     server.sin_port = htons(25);  
  56.     server.sin_addr .s_addr = inet_addr((LPCSTR)pSmtp);  
  57.     if(server.sin_addr.s_addr == INADDR_NONE){  
  58.         return 0;  
  59.     }  
  60.   
  61.     connect(cClient,(const struct sockaddr *)&server,sizeof(server));  
  62.   
  63.     //select模型,即设置超时  
  64.     struct timeval timeout ;  
  65.     fd_set r;  
  66.   
  67.     FD_ZERO(&r);  
  68.     FD_SET(cClient,&r);  
  69.     timeout.tv_sec = 15; //连接超时15秒  
  70.     timeout.tv_usec =0;  
  71.     ret = select(0, 0,&r, 0,&timeout);  
  72.     if ( ret<= 0 )  
  73.     {  
  74.         ::closesocket(cClient);  
  75.         return 0;  
  76.     }  
  77.   
  78.     //一般非锁定模式套接比较难控制,可以根据实际情况考虑 再设回阻塞模式  
  79.     unsigned long ul1= 0 ;  
  80.     ret = ioctlsocket(cClient, FIONBIO, (unsigned long*)&ul1);  
  81.     if(ret==SOCKET_ERROR){  
  82.         ::closesocket (cClient);  
  83.         return 0;  
  84.     }  
  85. }  
  86. http://blog.csdn.net/wangjiannuaa/article/details/6914162

相关文章推荐

一个基于UDP的可靠数据传输库

A UDP-based Reliable Data Transfer Library 一个基于UDP的可靠数据传输库   Introduces an open source UDP-based ...

一个比NPM更快更安全可靠的JavaScript包管理工具——Yarn

上一篇简单介绍了NPM包管理工具 在这篇文章中 我想给大家介绍一个取代使用NPM的包管理工具——Yarn 以及它与NPM之间的差异简介这个就是yarn的logo,应该是一只猫 Yarn的官网:...

编写一个安全可靠的C++11类:引入移动语义

右值引用和移动语义 在C++11(即C++0x)中,引入了右值引用的概念,同时在STL中提供std::move函数。这个机制完善了C++中关于左值、右值、以及引用的概念,优雅完美地解决了临时变量效率的...

基于Kafka如何部署一个稳定可靠的zookeeper集群服务

背景一个月左右zk挂了2次,在此期间 Mafka和其他服务不可用,因为业务(使用不当)原因导致zk挂掉,间接影响其他服务(包括mafka)可用性,目前线上多个业务和多种服务共同一套mobile-zk...
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:深度学习:神经网络中的前向传播和反向传播算法推导
举报原因:
原因补充:

(最多只允许输入30个字)