Linux下如何清空socket的接收缓冲区的数据

=================================================
本文为HeYuanHui原作

转载必须确保本文完整并完整保留原作者信息及本文原始链接!

NN:      khler
E-mail: khler@163.com
QQ:     23381103
MSN:   pragmac@hotmail.com
=================================================

 

最近碰到一个问题，对于阻塞模式的socket通讯，如果要实现设备的命令控制，那么进入命令流前，缓冲区不能存有上次通讯没有取回的信息，否则一旦命令发出，然后读取缓冲区，很显然会读到上一次的剩余数据。做法当然很简单，就是先清除接收区的缓冲数据，可是如何清除？

网上有很多这样的问题，但都没什么规范的解决办法，有的甚至为了达到清空的目的，建议先close一下socket，这个太大手笔了，为了解决一个小问题而大动干戈，根本不是个合理的解决办法。

还有就是用recv读取，但是由于不知道缓存里有多少数据，如果是阻塞模式，到最后必然等到超时才知道数据已经读取完毕，这是个问题。

另一个是用fgetc，通过返回判断是否是feof：

whlie (1) {
a=fgetc(f);
if (feof(f)) break;
//…
b=fgetc(f);
if (feof(f)) break;
//…
}

当然，我不知道读取完毕后最后一次调用fgetc会不会堵塞，需要测试。

在非阻塞模式下，我们用recv就可以轻松搞定了，但是阻塞模式下，由于我们不知道缓冲区有多少数据，不能直接调用recv尝试清除。

使用一个小小的技巧，利用select函数，我们可以轻松搞定这个问题：

select函数用于监视一个文件描述符集合，如果集合中的描述符没有变化，则一直阻塞在这里，直到超时时间到达；在超时时间内，一旦某个描述符触发了你所关心的事件，select立即返回，通过检索文件描述符集合处理相应事件；select函数出错则返回小于零的值，如果有事件触发，则返回触发事件的描述符个数；如果超时，返回0，即没有数据可读。

重点在于：我们可以用select的超时特性，将超时时间设置为0，通过检测select的返回值，就可以判断缓冲是否被清空。通过这个技巧，使一个阻塞的socket成了‘非阻塞’socket。

现在就可以得出解决方案了：使用select函数来监视要清空的socket描述符，并把超时时间设置为0，每次读取一个字节然后丢弃（或者按照业务需要进行处理，随你便了），一旦select返回0，说明缓冲区没数据了（“超时”了）。

struct timeval tmOut;      
tmOut.tv_sec = 0;
tmOut.tv_usec = 0;
fd_set         fds;
FD_ZEROS(&fds);
FD_SET(skt, &fds);

int            nRet;

char tmp[2];

memset(tmp, 0, sizeof(tmp));

while(1)
{           nRet= select(FD_SETSIZE, &fds, NULL, NULL, &tmOut);
           if(nRet== 0)  break;
           recv(skt, tmp, 1,0);
}

 这种方式的好处是，不再需要用recv、recvfrom等阻塞函数直接去读取，而是使用select，利用其超时特性检测缓冲区是否为空来判断是否有数据，有数据时才调用recv进行清除。

有人说同样可以用recv和socket的超时设置去清空啊，这个没错，但是你需要直接对socket描述符设置超时时间，而为了清空数据而直接修改socket描述符的属性，可能会影响到其他地方的使用，造成系统奇奇怪怪的问题，所以，不推荐使用。