使用select需要注意的细节
在学校的时候就使用过select,但是在项目中使用的时候却犯了个错误。
select如何使用就不进行总结了,网上教程太多,以下是项目中我写的一小段代码,便于总结。
int TvsStateManager::handleProbeStreamMsg() { struct sockaddr_in addr; int fd, n,addrlen; struct ip_mreq mreq; char recvBuf[BUF_SIZE]; u_int flag = 1; /* create what looks like an ordinary UDP socket */ if ((fd=socket(AF_INET, SOCK_DGRAM, 0)) < 0){ LogE("creat socket failure\n"); return -1; } /* allow multiple sockets to use the same PORT number */ if (setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &flag, sizeof(flag)) < 0){ LogE("reusing addr failure\n"); return -2; } /* set up destination address */ memset(&addr,0,sizeof(addr)); addr.sin_family = AF_INET; addr.sin_addr.s_addr = htonl(INADDR_ANY); /* N.B.: differs from sender */ addr.sin_port = htons(mConfig->mMultiCastStreamPort); /* bind to receive address */ if (bind(fd, (struct sockaddr *) &addr, sizeof(addr)) < 0){ LogE("bind socket failure\n"); return -3; } /* use setsockopt() to request that the kernel join a multicast group */ mreq.imr_multiaddr.s_addr = inet_addr(mConfig->mMultiCastStreamIP.c_str()); mreq.imr_interface.s_addr = htonl(INADDR_ANY); if (setsockopt(fd, IPPROTO_IP, IP_ADD_MEMBERSHIP, &mreq, sizeof(mreq)) < 0){ LogE("setsockopt join multicast group failure\n"); return -4; } // if(setsockopt(fd, SOL_SOCKET, SO_RCVTIMEO, (char *)&timeout, sizeof(timeout)) < 0){ // LogE("set timeout failure\n"); // return -5; // } fd_set readfds; int maxfds = 0; struct timeval timeout; while (1){ /*这个超时设置很关键,必须设置在里面,因为select模式,timeout会随着检查文件描述符集合状态而减小,换句话说就是用剩余的时间来更新这个结构*/ timeout.tv_sec = 5; timeout.tv_usec = 0; addrlen=sizeof(addr); FD_ZERO(&readfds); FD_SET(fd,&readfds); maxfds = fd +1; if(select(maxfds, &readfds, NULL, NULL, &timeout) > 0){ if((n = recvfrom(fd, recvBuf, BUF_SIZE, 0, (struct sockaddr *)&addr, &addrlen)) > 0){ mProbeStream = true; mQtPanel->sendProbeStreamMsg(mProbeStream); } bzero(recvBuf,sizeof(recvBuf)); if(setsockopt(fd, IPPROTO_IP, IP_DROP_MEMBERSHIP, &mreq, sizeof(mreq)) < 0){ LogE("setsockopt quit multicast failure\n"); return -7; } close(fd); break; } } return 0; } 可以看到我在使用结构struct timeval,将timeout设置放到了while里面,这样才是正确的。可能也是由于自己以前理解的不够透彻,当时我把设置timeout放到了while外面,那么引起的结果就是程序只等待一次5秒,后面却一直显示timeout不在等待5秒,测试程序就不再弄了。重新翻阅了下《unix环境高级编程》这本书,有一段不起眼的话说的很详细,如下:POSIX.1允许实现修改timeval结构中的值,所以在select返回后,你不能指望该结构仍旧保持调用select之前它所包含的值。FreeBSD 8.0、Mac OS X 10.6.8和Solaris 10都保持该结构中的值不变。但是,若在超时时间尚未到期时,select就返回,那么Linux 3.2.0将用剩余时间值更新该结构。
这段话已经很明确了,select设置的时间是会随着改变的,另外如果想不让它改变,那么可以使用pselect函数,而且pselect函数超时更加精确,pselect使用的是timespec结构,timespec以秒和纳秒表示超时值,而select的timeval结构则是秒和微妙级别。另外pselect的超时值是被设置为const的,这也就保证了调用pselect不会改变此值。