libcurl实现解析(3) - libcurl对select的使用

1.前言

在本系列的前一篇文章中。介绍了libcurl对poll()的使用。

參考"libcurl原理解析(2) - libcurl对poll的使用"。

本篇文章主要分析curl_poll()中对select()的封装使用。与前一篇类似，我们仅仅分离出与select相关的代码。

2.curl_poll函数分析

这个函数中使用到的一些其他的数据结构，能够參考前一篇文章中的介绍。本篇不再介绍。

/*
这个函数是对poll()的封装。假设poll()不存在，则使用select()替代。
假设使用的是select()，而且文件描写叙述符fd太大，超过了FD_SETSIZE，则返回error。
假设传入的timeout值是一个负数。则会无限的等待。直到没有有效的fd被提供。
当发生
这样的情况(没有有效的fd)时。则负数timeout值会被忽略，且函数会马上超时。

返回值：
-1 = 系统调用错误或fd>=FD_SETSIZE.
0 = timeout.
N = 返回的pollfd结构体的个数，且当中的revents成员不为0.
*/
int Curl_poll(struct pollfd ufds[], unsigned int nfds, int timeout_ms)
{
	struct timeval pending_tv;
	struct timeval *ptimeout;
	fd_set fds_read;
	fd_set fds_write;
	fd_set fds_err;
	curl_socket_t maxfd;

	struct timeval initial_tv = { 0, 0 };
	bool fds_none = TRUE;   //用于验证传入的ufds数组是否有效  
	unsigned int i;
	int pending_ms = 0;
	int error;    //保存错误码
	int r;

	//检測全部fd中是否存在有效的fd。  
	//假设至少存在一个有效的fd，则fds_none置为false。停止检測 
	if (ufds) 
	{
		for (i = 0; i < nfds; i++) 
		{
			if (ufds[i].fd != CURL_SOCKET_BAD) 
			{
				fds_none = FALSE;
				break;
			}
		}
	}

	//假设全部的fd都是无效的(即bad socket, -1)。则等待一段时间后。直接返回。
	if (fds_none)
	{
		r = Curl_wait_ms(timeout_ms);  //此函数会随后进行分析 
		return r;
	}

	//当传入的timeout值是一个负数(堵塞情形)或者0时。则无需衡量elapsed time.  
	//否则，获取当前时间。

	if (timeout_ms > 0) 
	{
		pending_ms = timeout_ms;
		initial_tv = curlx_tvnow();//调用gettimeofday()或time()获取当前时间  
	}

	//每次调用select()前都须要又一次初始化fdset,由于它们既是输入參数又是输出參数。
	FD_ZERO(&fds_read);
	FD_ZERO(&fds_write);
	FD_ZERO(&fds_err);
	maxfd = (curl_socket_t)-1;

	for (i = 0; i < nfds; i++) 
	{
		ufds[i].revents = 0;
		if (ufds[i].fd == CURL_SOCKET_BAD)  //跳过无效的fd
			continue;
		VERIFY_SOCK(ufds[i].fd);  //检測是否0<=fd<FD_SETSIZE.超出这个范围。则返回-1.

		if (ufds[i].events & (POLLIN | POLLOUT | POLLPRI |
			POLLRDNORM | POLLWRNORM | POLLRDBAND)) 
		{
			if (ufds[i].fd > maxfd)  //获取到最大的fd,做为select()的第一个參数。
				maxfd = ufds[i].fd;
			if (ufds[i].events & (POLLRDNORM | POLLIN))
				FD_SET(ufds[i].fd, &fds_read);
			if (ufds[i].events & (POLLWRNORM | POLLOUT))
				FD_SET(ufds[i].fd, &fds_write);
			if (ufds[i].events & (POLLRDBAND | POLLPRI))
				FD_SET(ufds[i].fd, &fds_err);
		}
	}

	//做为select()的timeout參数
	ptimeout = (timeout_ms < 0) ? NULL : &pending_tv;

	do 
	{
		if (timeout_ms > 0) 
		{
			pending_tv.tv_sec = pending_ms / 1000;
			pending_tv.tv_usec = (pending_ms % 1000) * 1000;
		}
		else if (!timeout_ms) 
		{
			pending_tv.tv_sec = 0;
			pending_tv.tv_usec = 0;
		}

		//真正调用select(). 第2。3，4參数已经在前面初始化(清空)过了。
		r = select((int)maxfd + 1, &fds_read, &fds_write, &fds_err, ptimeout);
		if (r != -1)  //select调用成功，结束循环
			break;

		//select调用失败。返回-1。通过errno能够获取到错误码。

		error = SOCKERRNO; //宏定义。
#define SOCKERRNO  (errno)  

		//以下的error_not_EINTR 是宏定义. 
		//#define error_not_EINTR (0 || error != EINTR) 
		if (error && error_not_EINTR) //检測是否存在error，且不是EINTR错误
			break;

		//没有出错或者存在EINTR错误。则推断是否继续运行select()
		if (timeout_ms > 0) 
		{
			//elapsed_ms是宏定义。
			//#define elapsed_ms  (int)curlx_tvdiff(curlx_tvnow(), initial_tv)  
			pending_ms = timeout_ms - elapsed_ms;
			if (pending_ms <= 0) 
			{
				r = 0;  //模拟select超时的情形 
				break;
			}
		}
	} while (r == -1);
	/*如今能够对上面的这个while循环总结一下：
	1.假设select调用成功(即返回值>=0)，则结束循环
	2.假设select调用失败(即返回值==-1)，则检測errno是否为EINTR错误。 
	假设不是EINTR，则结束循环。
	假设是EINTR，则检測是否已经超时。超时则结束循环，没有超时则继续select()。  
	*/
  
	if (r < 0)  //select()调用失败
		return -1;
	if (r == 0)  //select()超时
		return 0;

	//select()调用成功, 统计当中状态发生改变的fd的个数，保存至r.
	r = 0;
	for (i = 0; i < nfds; i++) 
	{
		ufds[i].revents = 0;
		if (ufds[i].fd == CURL_SOCKET_BAD)
			continue;
		if (FD_ISSET(ufds[i].fd, &fds_read))  //fd可读
			ufds[i].revents |= POLLIN;
		if (FD_ISSET(ufds[i].fd, &fds_write))  //fd可写
			ufds[i].revents |= POLLOUT;
		if (FD_ISSET(ufds[i].fd, &fds_err))  //fd出错
			ufds[i].revents |= POLLPRI;
		if (ufds[i].revents != 0)
			r++;
	}

	return r;
}

这个函数运行完毕后。第一个输入參数ufds中的成员revents可能会被改动。最后。函数返回select()的实际返回值。

3.curl_wait_ms函数分析

以下是curl_wait_ms()函数的详细实现。

也是基于poll或者select来实现的。这里也仅仅讨论它的select()实现版本号。

/*
这个函数用于等待特定的时间值。在函数Curl_socket_ready()以及Curl_poll()中被调用。
当没有提供不论什么fd来检測时。则仅仅是等待特定的一段时间。

假设是在windows平台下，则winsock中的poll()以及select()超时机制，须要一个有效的socket fd.
这个函数不同意无限等待，假设传入的值是0或者负数。则马上返回。
超时时间的精度以及最大值。取决于系统。


返回值：
-1 = 系统调用错误，或无效的输入值(timeout)，或被中断。
0 = 指定的时间已经超时
*/
int Curl_wait_ms(int timeout_ms)
{
	struct timeval pending_tv;
	struct timeval initial_tv;
	int pending_ms;
	int error;
	int r = 0;

	if (!timeout_ms)     //超时值为0，马上返回  
		return 0;
	if (timeout_ms < 0)  //不能为负数  
	{
		SET_SOCKERRNO(EINVAL);
		return -1;
	}

	pending_ms = timeout_ms;
	initial_tv = curlx_tvnow();
	do 
	{
		pending_tv.tv_sec = pending_ms / 1000;
		pending_tv.tv_usec = (pending_ms % 1000) * 1000;

		r = select(0, NULL, NULL, NULL, &pending_tv);

		if (r != -1)   //select()调用成功，则跳出循环   
			break;

		//select调用失败。返回-1。通过errno能够获取到错误码。  
		error = SOCKERRNO;    //宏定义。
#define SOCKERRNO  (errno)    

		//以下的error_not_EINTR 是宏定义. #define error_not_EINTR (0 || error != EINTR)   
		if (error && error_not_EINTR)  ////检測是否存在error，且不是EINTR错误  
			break;

		//elapsed_ms是宏定义:  
		//#define elapsed_ms  (int)curlx_tvdiff(curlx_tvnow(), initial_tv)   
		pending_ms = timeout_ms - elapsed_ms;
		if (pending_ms <= 0)
		{
			r = 0;  //模拟select超时的情形  
			break;
		}
	} while (r == -1);

	//确保返回值r仅仅能为-1(超时失败)或者0(超时成功)。
  
	//r不可能大于0，由于传入到select()函数的3个fdset数组所有都是NULL。假设select的返回值>0,则说明调用出问题了。  
	//故这里会将r置为-1,即调用超时失败。  
	if (r)
		r = -1;
	return r;
}