我们知道, 对于阻塞的socket而言, connect函数也是阻塞的, 我在Windows上测试过, 对于阻塞的socket而言, connect的阻塞时间约为25s(linux上是75s吧, 各个平台都不一样). 也就是说, 很多时候, 客户端需要等25s才继续往下执行。 我们想象一下, 用户肯定会不满意啊, 得罪了用户, 那就糟糕了。 那能不能搞个自己设置超时时间的connect函数呢? 完全可以! 在本文中, 我们来学习一下非阻塞connect函数的实现---让程序员自定义connect函数的超时时间。
说明: 两年后, 当我再次审视这些程序的时候, 我发现, select后, 强烈建议做FD_ISSET检查。
直接上客户端的代码:
#include <stdio.h>
#include <winsock2.h>
#pragma comment(lib, "ws2_32.lib")
int main()
{
// 网络初始化
WORD wVersionRequested;
WSADATA wsaData;
wVersionRequested = MAKEWORD(2, 2);
WSAStartup( wVersionRequested, &wsaData );
// 创建客户端socket(默认为是阻塞socket)
SOCKET sockClient = socket(AF_INET, SOCK_STREAM, 0);
// 设置为非阻塞的socket
int iMode = 1;
ioctlsocket(sockClient, FIONBIO, (u_long FAR*)&iMode);
// 定义服务端
SOCKADDR_IN addrSrv;
addrSrv.sin_addr.S_un.S_addr = inet_addr("127.0.0.1");
addrSrv.sin_family = AF_INET;
addrSrv.sin_port = htons(8888);
// 超时时间
struct timeval tm;
tm.tv_sec = 5;
tm.tv_usec = 0;
int ret = -1;
// 尝试去连接服务端
if (-1 != connect(sockClient, (SOCKADDR*)&addrSrv, sizeof(SOCKADDR)))
{
ret = 1; // 连接成功
}
else
{
fd_set set;
FD_ZERO(&set);
FD_SET(sockClient, &set);
if (select(-1, NULL, &set, NULL, &tm) <= 0)
{
ret = -1; // 有错误(select错误或者超时)
}
else
{
int error = -1;
int optLen = sizeof(int);
getsockopt(sockClient, SOL_SOCKET, SO_ERROR, (char*)&error, &optLen);
// 之所以下面的程序不写成三目运算符的形式, 是为了更直观, 便于注释
if (0 != error)
{
ret = -1; // 有错误
}
else
{
ret = 1; // 无错误
}
}
}
// 设回为阻塞socket
iMode = 0;
ioctlsocket(sockClient, FIONBIO, (u_long FAR*)&iMode); //设置为阻塞模式
// connect状态
printf("ret is %d\n", ret);
// 发送数据到服务端测试以下
if(1 == ret)
{
send(sockClient, "hello world", strlen("hello world") + 1, 0);
}
// 释放网络连接
closesocket(sockClient);
WSACleanup();
return 0;
}
我们先不管服务端, 直接运行上面的程序, 过5s后, 程序结果为:ret is -1
好, 我们关掉当面的客户端。 并启用下面的服务端:
#include <stdio.h>
#include <winsock2.h> // winsock接口
#pragma comment(lib, "ws2_32.lib") // winsock实现
int main()
{
WORD wVersionRequested; // 双字节,winsock库的版本
WSADATA wsaData; // winsock库版本的相关信息
wVersionRequested = MAKEWORD(1, 1); // 0x0101 即:257
// 加载winsock库并确定winsock版本,系统会把数据填入wsaData中
WSAStartup( wVersionRequested, &wsaData );
// AF_INET 表示采用TCP/IP协议族
// SOCK_STREAM 表示采用TCP协议
// 0是通常的默认情况
unsigned int sockSrv = socket(AF_INET, SOCK_STREAM, 0);
SOCKADDR_IN addrSrv;
addrSrv.sin_family = AF_INET; // TCP/IP协议族
addrSrv.sin_addr.S_un.S_addr = inet_addr("0.0.0.0"); // socket对应的IP地址
addrSrv.sin_port = htons(8888); // socket对应的端口
// 将socket绑定到某个IP和端口(IP标识主机,端口标识通信进程)
bind(sockSrv,(SOCKADDR*)&addrSrv, sizeof(SOCKADDR));
// 将socket设置为监听模式,5表示等待连接队列的最大长度
listen(sockSrv, 5);
SOCKADDR_IN addrClient;
int len = sizeof(SOCKADDR);
unsigned int sockConn = accept(sockSrv,(SOCKADDR*)&addrClient, &len);
printf("To receive...\n");
char recvBuf[100] = {0};
recv(sockConn, recvBuf, 100, 0); // 接收客户端数据,最后一个参数一般设置为0
printf("recv is %s\n", recvBuf);
while(1);
closesocket(sockConn);
closesocket(sockSrv);
WSACleanup();
return 0;
}
然后呢, 我们再启动客户端, 发现客户端立即出现:ret is 1, 服务端对应的结果为:
To receive...
recv is hello world
由此可见, 上面的客户端程序实现了非阻塞的connect, 也就是用, 程序员可以自定义超时时间。 ok, 先这样。