非阻塞

来源：互联网发布：赛尔网络河北分公司编辑：程序博客网时间：2024/04/28 01:14

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

0 0