Socket实现非阻塞连接

#include <stdio.h>
#include <winsock.h>
#include <string.h>
#include <Windows.h>
#pragma comment(lib, "ws2_32.lib")

#define TIME_OUT_TIME 20 //connect超时时间20秒

void geturl(char *url)
{
WSADATA WSAData={0};
SOCKET sockfd;
struct sockaddr_in addr;
struct hostent *pURL;
char myurl[BUFSIZ];
char *pHost = 0, *pGET = 0;
char host[BUFSIZ], GET[BUFSIZ];
char header[BUFSIZ] = "";
static char text[BUFSIZ];
int i;
int TimeOut;

int error=-1, len;
len = sizeof(int);

int ret;

timeval tm;
fd_set set;

/*
* windows下使用socket必须用WSAStartup初始化，否则不能调用
*/
if(WSAStartup(MAKEWORD(2,2), &WSAData))
{
printf("WSA failed\n");
return;
}

/*
* 分离url中的主机地址和相对路径
*/
strcpy(myurl, url);
for (pHost = myurl; *pHost != '/' && *pHost != '\0'; ++pHost);
if ( (int)(pHost - myurl) == strlen(myurl) )
strcpy(GET, "/");
else
strcpy(GET, pHost);
*pHost = '\0';
strcpy(host, myurl);
printf("%s\n%s\n", host, GET);

/*
* 设定socket参数,并未真正初始化
*/
sockfd = socket(PF_INET, SOCK_STREAM, IPPROTO_TCP);
if (sockfd == INVALID_SOCKET)
{
printf("建立套接字失败\r\n");
return;
}

TimeOut = 6000; //设置发送超时6秒

if (setsockopt(sockfd, SOL_SOCKET, SO_SNDTIMEO, (char*)&TimeOut, sizeof(TimeOut)) == SOCKET_ERROR)
{
printf("设置发送超时失败\r\n");
return;
}

TimeOut = 6000; //设置接收超时6秒

if (setsockopt(sockfd, SOL_SOCKET, SO_RCVTIMEO, (char*)&TimeOut, sizeof(TimeOut)) == SOCKET_ERROR)
{
printf("设置接收超时失败\r\n");
return;
}

//设置非阻塞连接
unsigned long ul = 1;
ret = ioctlsocket(sockfd, FIONBIO, (unsigned long*)&ul);
if (ret == SOCKET_ERROR)
{
printf("非阻塞连接失败\r\n");
return;
}

pURL = gethostbyname(host);
addr.sin_family = AF_INET;
addr.sin_addr.s_addr = *((unsigned long*)pURL->h_addr);
addr.sin_port = htons(80);

/*
* 组织发送到web服务器的信息
* 为何要发送下面的信息请参考HTTP协议的约定
*/
strcat(header, "GET ");
strcat(header, GET);
strcat(header, " HTTP/1.1\r\n");
strcat(header, "HOST: ");
strcat(header, host);
strcat(header, "\r\nConnection: Close\r\n\r\n");

/*
* 连接到服务器，发送请求header，并接受反馈（即网页源代码）
*/
connect(sockfd,(SOCKADDR *)&addr,sizeof(addr));
//printf("%d", WSAGetLastError());

//select 模型，即设置超时
struct timeval timeout ;
fd_set r;

FD_ZERO(&r);
FD_SET(sockfd, &r);
timeout.tv_sec = 15; //连接超时15秒
timeout.tv_usec =0;
ret = select(0, 0, &r, 0, &timeout);
if ( ret <= 0 )
{
closesocket(sockfd);
printf("连接超时\r\n");
return;
}

send(sockfd, header, strlen(header), 0);
//Sleep(5000);
while ( (recv(sockfd, text, BUFSIZ, 0) > 0) || (WSAGetLastError() == WSAEWOULDBLOCK))
{
//printf("%s", text);
strnset(text, '\0', BUFSIZ);
}
printf("%s", "完成！\n");

closesocket(sockfd);

WSACleanup();
}

int main()
{
char url[256];
printf("http://");
scanf("%s", url);
for (int i = 0; i < 10; i++)
{
geturl(url);
}
return 0;
}

以上这段代码中，有一个geturl函数，该函数的作用是发送并接收数据包。

在该函数进行了socket超时连接设置，所以在connect时会返回WSAEWOULDBLOCK（10035）错误，若在recv时不加上WSAEWOULDBLOCK判断，就接收不到数据。

备注：

以下转自http://www.moon-soft.com/doc/6652.htm

把CSDN与中文yahoo翻了底朝天，也没找到如何设置socket的连接超时的满意方法，问此问题的兄弟已有一大堆，这里偶就讲一下win下如何设置socket的connect超时。
设
置connect的超时很简单，CSDN上也有人提到过使用select，但却没有一个令人满意与完整的答案。偶所讲的也正是select函数，此函数集
成在winsock1.1中，简单点讲，"作用使那些想避免在套接字调用过程中被锁定的应用程序，采取一种有序的方式，同时对多个套接字进行管理"
(《Windows网络编程技术》原话)。使用方法与解释请见《Windows网络编程技术》。
在使用此函数前，需先将socket设置为非锁定模式，这样，在connect时，才会立马跳过，同时，通常也会产生一个WSAEWOULDBLOCK错误，这个错误没关系。再执行select则是真正的超时。