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则是真正的超时。