您的位置:首页 > 其它

fd_set 用法 socket

2015-08-12 11:58 260 查看
以下为Linux的winsock用法。

一、winsock中

#include <winsock.h>

原型

int select(

int
nfds,

fd_set*
readfds,

fd_set*
writefds,

fd_set*
exceptfds,

const struct timeval*
timeout

);


nfds:本参数忽略,仅起到兼容作用。

readfds:(可选)指针,指向一组等待可读性检查的套接口。

writefds:(可选)指针,指向一组等待可写性检查的套接口。

exceptfds:(可选)指针,指向一组等待错误检查的套接口。

timeout:select()最多等待时间,对阻塞操作则为NULL。

注释:

本函数用于确定一个或多个套接口的状态。对每一个套接口,调用者可查询它的可读性、可写性及错误状态信息。用fd_set结构来表示一组等待检查的套接口。在调用返回时,这个结构存有满足一定条件的套接口组的子集,并且select()返回满足条件的套接口的数目。有一组宏可用于对fd_set的操作,这些宏与Berkeley Unix软件中的兼容,但内部的表达是完全不同的。

readfds参数标识等待可读性检查的套接口。如果该套接口正处于监听listen()状态,则若有连接请求到达,该套接口便被标识为可读,这样一个accept()调用保证可以无阻塞完成。对其他套接口而言,可读性意味着有排队数据供读取。或者对于SOCK_STREAM类型套接口来说,相对于该套接口的虚套接口已关闭,于是recv()或recvfrom()操作均能无阻塞完成。如果虚电路被“优雅地”中止,则recv()不读取数据立即返回;如果虚电路被强制复位,则recv()将以WSAECONNRESET错误立即返回。如果SO_OOBINLINE选项被设置,则将检查带外数据是否存在(参见setsockopt())。

writefds参数标识等待可写性检查的套接口。如果一个套接口正在connect()连接(非阻塞),可写性意味着连接顺利建立。如果套接口并未处于connect()调用中,可写性意味着send()和sendto()调用将无阻塞完成。〔但并未指出这个保证在多长时间内有效,特别是在多线程环境中〕。

exceptfds参数标识等待带外数据存在性或意味错误条件检查的套接口。请注意如果设置了SO_OOBINLINE选项为假FALSE,则只能用这种方法来检查带外数据的存在与否。对于SO_STREAM类型套接口,远端造成的连接中止和KEEPALIVE错误都将被作为意味出错。如果套接口正在进行连接connect()(非阻塞方式),则连接试图的失败将会表现在exceptfds参数中。

如果对readfds、writefds或exceptfds中任一个组类不感兴趣,可将它置为空NULL。

在winsock.h头文件中共定义了四个宏来操作描述字集。FD_SETSIZE变量用于确定一个集合中最多有多少描述字(FD_SETSIZE缺省值为64,可在包含winsock.h前用#define FD_SETSIZE来改变该值)。对于内部表示,fd_set被表示成一个套接口的队列,最后一个有效元素的后续元素为INVAL_SOCKET。宏为:

FD_CLR(s,*set):从集合set中删除描述字s。

FD_ISSET(s,*set):若s为集合中一员,非零;否则为零。

FD_SET(s,*set):向集合添加描述字s。

FD_ZERO(*set):将set初始化为空集NULL。

timeout参数控制select()完成的时间。若timeout参数为空指针,则select()将一直阻塞到有一个描述字满足条件。否则的话,timeout指向一个timeval结构,其中指定了select()调用在返回前等待多长时间。如果timeval为{0,0},则select()立即返回,这可用于探询所选套接口的状态。如果处于这种状态,则select()调用可认为是非阻塞的,且一切适用于非阻塞调用的假设都适用于它。举例来说,阻塞钩子函数不应被调用,且WINDOWS套接口实现不应yield。

返回值:

select()调用返回处于就绪状态并且已经包含在fd_set结构中的描述字总数;如果超时则返回0;否则的话,返回SOCKET_ERROR错误,应用程序可通过WSAGetLastError()获取相应错误代码。

错误代码:

WSANOTINITIALISED:在使用此API之前应首先成功地调用WSAStartup()。

WSAENETDOWN:WINDOWS套接口实现检测到网络子系统失效。

WSAEINVAL:超时时间值非法。

WSAEINTR:通过一个WSACancelBlockingCall()来取消一个(阻塞的)调用。

WSAEINPROGRESS:一个阻塞的WINDOWS套接口调用正在运行中。

WSAENOTSOCK:描述字集合中包含有非套接口的元素。
范例 :

sock= socket(AF_INET,SOCK_STREAM,0);



struct sockaddr_in addr; //告诉sock 应该再什么地方licence

memset(&addr,0,sizeof(addr));

addr.sin_family=AF_INET;

addr.sin_port=htons(11111); //端口啦

addr.sin_addr.s_addr=htonl(INADDR_ANY); //在本机的所有ip上开始监听

bind (sock,(sockaddr *)&addr,sizeof(addr));//bind....

listen(sock,5); //最大5个队列

SOCKET socka; //这个用来接受一个连接

fd_set rfd; // 描述符集 这个将用来测试有没有一个可用的连接

struct timeval timeout;

FD_ZERO(&rfd); //总是这样先清空一个描述符集

timeout.tv_sec=60; //等下select用到这个

timeout.tv_usec=0;

u_long ul=1;

ioctlsocket(sock,FIONBIO,&ul); //用非阻塞的连接

//现在开始用select

FD_SET(sock,&rfd); //把sock放入要测试的描述符集 就是说把sock放入了rfd里面 这样下一步调用select对rfd进行测试的时候就会测试sock了(因为我们将sock放入的rdf) 一个描述符集可以包含多个被测试的描述符,

if(select(sock+1,&rfd,0,0, &timeout)==0)

{ //这个大括号接上面的,返回0那么就超过了timeout预定的时间

//处理....

}

if(FD_ISSET(sock,&rfd))

{ //有一个描述符准备好了

socka=accept(sock,0,0); //一个用来测试读 一个用来测试写

FD_ZERO(&rfd);

FD_ZERO(&wfd);

FD_SET(socka,&rfd);//把socka放入读描述符集

FD_SET(sockb,&rfd);//把sockb放入读描述符集

FD_SET(socka,&wfd);把socka放入写描述符集

FD_SET(sockb,&wfd);把sockb放入写描述符集

if(SOCKET_ERROR!=select(0,&rfd,&wfd,0,0)) //测试这两个描述符集,永不超时 其中rfd只用来测试读 wfd只用来测试写

{ //没有错误

if(FD_ISSET(socka,&rfd)) //socka可读

{...}

if(FD_ISSET(sockb,&rfd) //sockb可读

{...}

if(FD_ISSET(socka,&wfd) //socka 可写

{...}

if(FD_ISSET(sockb,&wfd) //sockb可写

{...}

}


select()函数主要是建立在fd_set类型的基础上的。fd_set(它比较重要所以先介绍一下)是一组文件描述字(fd)的集合,它用一位来表示一个fd(下面会仔细介绍),对于fd_set类型通过下面四个宏来操作:

fd_set set;

FD_ZERO(&set); /* 将set清零使集合中不含任何fd*/

FD_SET(fd, &set); /* 将fd加入set集合 */

FD_CLR(fd, &set); /* 将fd从set集合中清除 */

FD_ISSET(fd, &set); /* 测试fd是否在set集合中*/

过去,一个fd_set通常只能包含<32的fd(文件描述字),因为fd_set其 实只用了一个32位矢量来表示fd;现在,UNIX系统通常会在头文件<sys/select.h>中定义常量FD_SETSIZE,它是数 据类型fd_set的描述字数量,其值通常是1024,这样就能表示<1024的fd。根据fd_set的位矢量实现,我们可以重新理解操作 fd_set的四个宏:

fd_set set;

FD_ZERO(&set); /*将set的所有位置0,如set在内存中占8位则将set置为

00000000*/

FD_SET(0, &set); /* 将set的第0位置1,如set原来是00000000,则现在变为10000000,这样fd==1的文件描述字就被加进set中了 */

FD_CLR(4, &set); /*将set的第4位置0,如set原来是10001000,则现在变为10000000,这样fd==4的文件描述字就被从set中清除了 */

FD_ISSET(5, &set); /* 测试set的第5位是否为1,如果set原来是10000100,则返回非零,表明fd==5的文件描述字在set中;否则返回0*/

―――――――――――――――――――――――――――――――――――――――

注意fd的最大值必须<FD_SETSIZE。

―――――――――――――――――――――――――――――――――――――――

select函数的接口比较简单:

int select(int nfds, fd_set *readset, fd_set *writeset,

fd_set* exceptset, struct timeval *timeout);

功能:

测试指定的fd可读?可写?有异常条件待处理?

参数:

nfds

需要检查的文件描述字个数(即检查到fd_set的第几位),数值应该比三组fd_set中 所含的最大fd值更大,一般设为三组fd_set中所含的最大fd值加1(如在readset,writeset,exceptset中所含最大的fd为 5,则nfds=6,因为fd是从0开始的)。设这个值是为提高效率,使函数不必检查fd_set的所有1024位。

readset

用来检查可读性的一组文件描述字。

writeset

用来检查可写性的一组文件描述字。

exceptset

用来检查是否有异常条件出现的文件描述字。(注:错误不包括在异常条件之内)

timeout

有三种可能:

1. timeout=NULL(阻塞:直到有一个fd位被置为1函数才返回)

2. timeout所指向的结构设为非零时间(等待固定时间:有一个fd位被置为1或者时间耗尽,函数均返回)

3. timeout所指向的结构,时间设为0(非阻塞:函数检查完每个fd后立即返回)

返回值:

返回对应位仍然为1的fd的总数。

Remarks:

三组fd_set均将某些fd位置0,只有那些可读,可写以及有异常条件待处理的fd位仍然为1。

使用select函数的过程一般是:

先调用宏FD_ZERO将指定的fd_set清零,然后调用宏FD_SET将需要测试的fd加入fd_set,接着调用函数select测试fd_set中的所有fd,最后用宏FD_ISSET检查某个fd在函数select调用后,相应位是否仍然为1。

在过去,一个fd_set通常只能包含少于等于32个文件描述符,因为fd_set其实只用了一个int的比特矢量来实现,在大多数情况下,检查fd_set能包括任意值的文件描述符是系统的责任,但确定你的fd_set到底能放多少有时你应该检查/修改宏FD_SETSIZE的值。*这个值是系统相关的*,同时检查你的系统中的select() 的man手册。有一些系统对多于1024个文件描述符的支持有问题

#ifndef FD_SETSIZE

#define FD_SETSIZE 64

#endif /* FD_SETSIZE */

typedef struct fd_set {

u_int fd_count; /* how many are SET? */

SOCKET fd_array[FD_SETSIZE]; /* an array of SOCKETs */

} fd_set;

extern int PASCAL FAR __WSAFDIsSet(SOCKET, fd_set FAR *);

#define FD_CLR(fd, set) do { /

u_int __i; /

for (__i = 0; __i < ((fd_set FAR *)(set))->;fd_count ; __i++) { /

if (((fd_set FAR *)(set))->;fd_array[__i] == fd) { /

while (__i < ((fd_set FAR *)(set))->;fd_count-1) { /

((fd_set FAR *)(set))->;fd_array[__i] = /

((fd_set FAR *)(set))->;fd_array[__i+1]; /

__i++; /

} /

((fd_set FAR *)(set))->;fd_count--; /

break; /

} /

} /

} while(0)

#define FD_SET(fd, set) do { /

u_int __i; /

for (__i = 0; __i < ((fd_set FAR *)(set))->;fd_count; __i++) { /

if (((fd_set FAR *)(set))->;fd_array[__i] == (fd)) { /

break; /

} /

} /

if (__i == ((fd_set FAR *)(set))->;fd_count) { /

if (((fd_set FAR *)(set))->;fd_count < FD_SETSIZE) { /

((fd_set FAR *)(set))->;fd_array[__i] = (fd); /

((fd_set FAR *)(set))->;fd_count++; /

} /

} /

} while(0)

#define FD_ZERO(set) (((fd_set FAR *)(set))->;fd_count=0)

#define FD_ISSET(fd, set) __WSAFDIsSet((SOCKET)(fd), (fd_set FAR *)(set))

typedef int32_t __fd_mask;

#define _NFDBITS (sizeof(__fd_mask) * 8) /* 8 bits per byte */

#define __howmany(x,y) (((x)+((y)-1))/(y))

#ifndef _FD_SET

# define _FD_SET

typedef struct __fd_set {

long fds_bits[__howmany(FD_SETSIZE, (sizeof(long) * 8))];

} fd_set;

# ifndef _KERNEL

# ifdef __cplusplus

extern "C" {

# endif /* __cplusplus */

#ifdef _INCLUDE_HPUX_SOURCE

# define FD_SET(n,p) (((__fd_mask *)((p)->;fds_bits))[(n)/_NFDBITS] |= (1 <<

((n) % _NFDBITS)))

# define FD_CLR(n,p) (((__fd_mask *)((p)->;fds_bits))[(n)/_NFDBITS] &= ~(1 <<

((n) % _NFDBITS)))

# define FD_ISSET(n,p) (((__fd_mask *)((p)->;fds_bits))[(n)/_NFDBITS] & (1 <<

((n) % _NFDBITS)))

# define FD_ZERO(p) memset((void *)(p), (int) 0, sizeof(*(p)))

#else

# define FD_SET(n,p) (__fd_set1(n, p))

# define FD_CLR(n,p) (__fd_clr(n, p))

# define FD_ISSET(n,p) (__fd_isset(n, p))

# define FD_ZERO(p) memset((void *)(p), (int) 0, sizeof(fd_set))

以下是一个测试单个文件描述字可读性的例子:

int isready(int fd)

{

int rc;

fd_set fds;

struct timeval tv;

FD_ZERO(&fds);

FD_SET(fd,&fds);

tv.tv_sec = tv.tv_usec = 0;

rc = select(fd+1, &fds, NULL, NULL, &tv);

if (rc < 0) //error

return -1;

return FD_ISSET(fd,&fds) ? 1 : 0;

}

下面还有一个复杂一些的应用:

//这段代码将指定测试Socket的描述字的可读可写性,因为Socket使用的也是fd

uint32 SocketWait(TSocket *s,bool rd,bool wr,uint32 timems)

{

fd_set rfds,wfds;

#ifdef _WIN32

TIMEVAL tv;

#else

struct timeval tv;

#endif /* _WIN32 */

FD_ZERO(&rfds);

FD_ZERO(&wfds);

if (rd) //TRUE

FD_SET(*s,&rfds); //添加要测试的描述字

if (wr) //FALSE

FD_SET(*s,&wfds);

tv.tv_sec=timems/1000; //second

tv.tv_usec=timems%1000; //ms

for (;;) //如果errno==EINTR,反复测试缓冲区的可读性

switch(select((*s)+1,&rfds,&wfds,NULL,

(timems==TIME_INFINITE?NULL:&tv))) //测试在规定的时间内套接口接收缓冲区中是否有数据可读

{ //0--超时,-1--出错

case 0: /* time out */

return 0;

case (-1): /* socket error */

if (SocketError()==EINTR)

break;

return 0; //有错但不是EINTR

default:

if (FD_ISSET(*s,&rfds)) //如果s是fds中的一员返回非0,否则返回0

return 1;

if (FD_ISSET(*s,&wfds))

return 2;

return 0;

};

}

现在说说windows和Linux的socket的区别:

在win下面,实现是(VS):

typedef struct fd_set {

u_int fd_count; /* how many are SET? */

SOCKET fd_array[FD_SETSIZE]; /* an array of SOCKETs */

} fd_set;

很明了,一个计数的fd_count,另一个就是SOCKET数组。其中,FD_SETSIZE是64.(具体可以去查看vs的代码)

win下面FD_SET就是检查SOCKET在数组中是否存在,如果不存在,那么就插入到数组最后。而FD_CLR(fd, set)是把fd后面的东西往前拷贝,然后计数减1.FD_ZERO仅仅是把计数置为0(这个要注意!!)!FD_ISSET就看不到实现了。

下面是fd_set在linux下的实现:

typedef struct

{

/* XPG4.2 requires this member name. Otherwise avoid the name

from the global namespace. */

#ifdef __USE_XOPEN

__fd_mask fds_bits[__FD_SETSIZE / __NFDBITS];

# define __FDS_BITS(set) ((set)->fds_bits)

#else

__fd_mask __fds_bits[__FD_SETSIZE / __NFDBITS];

# define __FDS_BITS(set) ((set)->__fds_bits)

#endif

} fd_set;

根据UNIX网络编程对fd_set的介绍,fd_set是个整数数组,用每个bit位来表示fd的。比如,一个32位整数,则数组第一个整数表示0-31的fd,以此类推,第二个整数表示32-63.。。。

查看linux的FD_SET、FD_CLR是用汇编实现的。根据说明可以知道,就是给bit置位。

(在我的系统中,__FD_SETSIZE是1024,__NFDBITS是sizeof(long int)即32,就是说,fd_set是4个long int 数组,最多放128个文件描述符(是不是太少了啊???))

由于fd_set实现不同,导致selet也不太一样,其他第一个参数在windows下就没有意义了(MSDN: The nfds parameter is included only for compatibility with Berkeley sockets),而在linux下面,是fd_set种的最大的一个描述符+1.具体可以参考其他资料。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: