int select(int nfds, fd_set *readfds, fd_set *writefds,fd_set *exceptfds, struct timeva l *timeout) |
功能：轮询机制，可以监听文件描述符集中文件描述符的状态。（可读,可写，异常状态）
参数：nfds : 最大文件描述符+ 1
readfds : 读文件描述集合
writefds : 写文件描述符集合
execeptfds 其他文件描述符集合
timeout： 超时时间
超时时间:
struct timeva l {
long tv_sec;
long tv_usec;
};
返回值：成功：返回值大于0:监听到文件描述符的值有变化（可读可写）返回值为0 :超时,失败：< 0的值。
先说明两个结构体：
select机制中提供了一个数据结构 struct fd_set ，可以理解为一个集合，实际上是一个位图，每一个特定为来标志相应大小文件描述符，这个集合中存放的是文件描述符(file descriptor)，即文件句柄(也就是位图上的每一位都能与一个打开的文件句柄(文件描述符)建立联系，这个工作由程序员来完成)，这可以是我们所说的普通意义的文件，当然Unix下任何设备、管道、FIFO等都是文件形式，全部包括在内，所以毫无疑问一个socket就是一个文件，socket句柄就是一个文件描述符。fd_set集合可以通过一些宏由人为来操作，程序员通过操作4类宏，来完成最fd_set的操作：
void FD_CLR(int fd, fd_set *set);
功能：将文件描述符fd 从文件描述符符集合当中清除
int FD_ISSET(int fd, fd_set *set);
功能：判断文件描述符集合中的文件描述符时候可读可写
void FD_SET(int fd, fd_set *set);
功能：将fd 添加到文件描述符集合当中
void FD_ZERO(fd_set *set);
功能：清空整个文件描述符集合。
fd_set ： 设置文件描述符符集合
#include <stdio.h> #include <sys/types.h> #include <sys/socket.h> #include <netinet/in.h> #include <string.h> #include <unistd.h> #include <sys/stat.h> #include <fcntl.h> #include <string.h> #include <stdlib.h> #include <sys/time.h> #include <sys/select.h>
#define IP "192.168.11.111" #define PROT 10086 #define N 1024
//tcp 服务器 int main( ) { int file = 0; int sendfile = 0; int readsize = 0; int fd = 0; int filesize = 0; off_t filelength = 0; int newid; int ret; int retsend; int socketfd; struct sockaddr_in addr; socklen_t addrlen; char buf[10] = {'0'};
//建立socket socketfd = socket(AF_INET, SOCK_STREAM,0); if(socketfd < 0) { printf("socket is error\r\n"); return -1; }
memset(&addr,0,sizeof(addr));
addrlen = sizeof(addr); addr.sin_family = AF_INET; addr.sin_port = htons(PROT); //存放端口 addr.sin_addr.s_addr = inet_addr(IP);// 0.0.0. 填充任意的IP
//bind 端口与 IP ret = bind(socketfd,(struct sockaddr *)&addr,addrlen); if(ret < 0) { printf("bind is error\r\n"); return -1; }
printf("bind is ok!\r\n");
//listen ret = listen(socketfd,15); if(ret < 0) { printf("listen is error\r\n"); return -1; } printf("listen.............\r\n");
memset(&addr,0,sizeof(addr)); //接收客户端的文件描述符 如果服务与客户端链接成功会产生一个新的文件描述符
//设置文件描述符的集合 fd_set readfds;
//清空文件描述符集合 FD_ZERO(&readfds);
//添加文件描述符到集合当中 FD_SET(socketfd,&readfds);
// 设置最大文件描述符 + 1 int MAXFDS = socketfd;
while(1) {
fd_set tmp = readfds; ret = select(MAXFDS+1,&tmp,NULL,NULL,NULL); if(ret < 0) { printf("select is error\r\n"); return -1; }else if(ret == 0) { printf("time out\r\n"); break; }else {
int i; for(i=1;i<MAXFDS+1;i++) {
if(FD_ISSET(socketfd,&tmp)&&socketfd==i) {
newid = accept(socketfd,(struct sockaddr *)&addr,&addrlen); if(newid < 0) { printf("accept is error\r\n"); return -1; }else { // 将新产生的文件描述符添加到 read文件描述符集合当中 FD_SET(newid,& readfds); //更新最大文件描述 MAXFDS = (newid > socketfd ) ? newid :socketfd; printf("client:%d\r\n",newid); } } else if(FD_ISSET(i,&tmp)) { retsend = recv(i,buf,sizeof(buf),0); if(retsend > 0) { printf("buf:%s\r\n",buf); } else if(retsend <= 0) { FD_CLR(newid, &readfds); close(newid); printf("close:%d\r\n",newid); break;
}
} } } } close(socketfd); close(newid);
return 0; }
|