关于对多平台兼容的TCP连接的开发思路

在做嵌入式SDK产品的开发过程中，由于产品本身需要兼容多种平台，所以对代码质量有很高的要求，今天我们主要介绍的就是针对于TCP开发过程中的一些增强代码兼容性的小技巧。

做过网络开发的小伙伴可能遇到过这样类似的问题，在最频繁的TCP连接的过程中，偶然性的会出现connect连接死锁，好一点的情况可能是几十秒能够从连接请求中退出，坏一点的情况可能一直不能返回，导致程序无法正常运行，下面给出一种通过将connect设置为非阻塞模式进行TCP连接方式，避免connect死锁问题，代码如下：

int error = -1, len;
len = sizeof(int);
timeval tm;
fd_set set;
unsigned long ul = 1;
ioctlsocket(serverSock, FIONBIO, &ul);  //设置为非阻塞模式
BOOL ret = false;
if (connect(serverSock, (struct sockaddr *)&serverAddr, sizeof(serverAddr)) < 0)
{
 	tm.tv_sec  = timeout;
    	tm.tv_usec = 0;
    	FD_ZERO(&set);
     FD_SET(serverSock, &set);
 	if (select(serverSock+1, NULL, &set, NULL, &tm) > 0)
  	{
   		getsockopt(serverSock, SOL_SOCKET, SO_ERROR, (char*)&error, (int*)&len);
   		if (error == 0)
    			ret = TRUE;
   		else
    			ret = FALSE;
  	}
  	else
   	ret = FALSE;
}
else ret = TRUE;
ul = 0;
ioctlsocket(serverSock, FIONBIO, &ul);  //设置为阻塞模式

• 思路分析：
• 获取套接字

int sock = socket(AF_INET, SOCK_STREAM, 0);

• 非阻塞配置

//RTOS平台
unsigned long ul = 1;
ioctlsocket(serverSock, FIONBIO, &ul); 
//linux平台
if(block_flag = fcntl(sockfd, F_GETFL) < 0)
           {
          		LOG_ERR("F_GETFL error \r\n");
          }
           block_flag |= O_NONBLOCK;
           if(fcntl(sockfd, F_SETFL, block_flag) < 0)
           {
               LOG_ERR("F_SETFL error \r\n");
           }

• 连接服务器

connect(serverSock, (struct sockaddr *)&serverAddr, sizeof(serverAddr)；

• 使用select等待连接成功

select(serverSock+1, NULL, &set, NULL, &tm);

• 恢复阻塞模式

ul = 0;
ioctlsocket(serverSock, FIONBIO, &ul);

在进行连接之前，先将socket_fd设置为阻塞模式，调用连接后connet立即返回，如果成功，则直接连接成功，进行后续处理，如果失败，通过select监听socket_fd是否可写，是，则证明tcp连接成功，否则连接超时，则判定连接失败，具体超时时间可根据自己需求进行修改；最后在连接成功后再将状态修改为阻塞模式。

这种方法不仅能够避免connect卡死的问题，同时又可以让connect的超时时间变得可控，再对多平台的开发时，可以通过这种方式增强代码的兼容性，提高产品质量。

• 非阻塞模式有3种用途

1.三次握手同时做其他的处理。connect要花一个往返时间完成，从几毫秒的局域网到几百毫秒或几秒的广域网。这段时间可能有一些其他的处理要执行，比如数据准备，预处理等。

2.用这种技术建立多个连接。这在web浏览器中很普遍.

3.由于程序用select等待连接完成，可以设置一个select等待时间限制，从而缩短connect超时时间。多数实现中，connect的超时时间在75秒到几分钟之间。有时程序希望在等待一定时间内结束，使用非阻塞connect可以防止阻塞75秒，在多线程网络编程中，尤其必要。 例如有一个通过建立线程与其他主机进行socket通信的应用程序，如果建立的线程使用阻塞connect与远程通信，当有几百个线程并发的时候，由于网络延迟而全部阻塞，阻塞的线程不会释放系统的资源，同一时刻阻塞线程超过一定数量时候，系统就不再允许建立新的线程（每个进程由于进程空间的原因能产生的线程有限），如果使用非阻塞的connect，连接失败使用select等待很短时间，如果还没有连接后，线程立刻结束释放资源，防止大量线程阻塞而使程序崩溃。