poll函数实现多路复用

结构体pollfd

struct pollfd
{
int fd;               //file descriptor
short event;   //event of interest on fd
short reven;  //event that occurred on fd
}

每一个pollfd结构体指定了一个被监视的文件描述符,可以传递多个结构体,指示poll()监视多个文件描述符。每个结构体的events域是监视该文件描述符的事件掩码,由用户来设置这个域。revents域是文件描述符的操作结果事件掩码。内核在调用返回时设置这个域。events域中请求的任何事件都可能在revents域中返回。

pollfd结构体合法的事件

POLLIN
有数据可读。
POLLRDNORM
有普通数据可读。
POLLRDBAND
有优先数据可读。
POLLPRI
有紧迫数据可读。
POLLOUT
写数据不会导致阻塞。
POLLWRNORM
写普通数据不会导致阻塞。
POLLWRBAND
写优先数据不会导致阻塞。
POLLMSG
SIGPOLL 消息可用。
此外,revents域中还可能返回下列事件:
POLLER
指定的文件描述符发生错误。
POLLHUP
指定的文件描述符挂起事件。
POLLNVAL
指定的文件描述符非法。
这些事件在events域中无意义,因为它们在合适的时候总是会从revents中返回。使用poll()和select()不一样,你不需要显式地请求异常情况报告。
POLLIN | POLLPRI等价于select()的读事件,POLLOUT |POLLWRBAND等价于select()的写事件。POLLIN等价于POLLRDNORM |POLLRDBAND,而POLLOUT则等价于POLLWRNORM。
例如,要同时监视一个文件描述符是否可读和可写,我们可以设置 events为POLLIN |POLLOUT。在poll返回时,我们可以检查revents中的标志,对应于文件描述符请求的events结构体。如果POLLIN事件被设置,则文件描述符可以被读取而不阻塞。如果POLLOUT被设置,则文件描述符可以写入而不导致阻塞。这些标志并不是互斥的:它们可能被同时设置,表示这个文件描述符的读取和写入操作都会正常返回而不阻塞。
timeout参数指定等待的毫秒数,无论I/O是否准备好,poll都会返回。timeout指定为负数值表示无限超时;timeout为0指示poll调用立即返回并列出准备好I/O的文件描述符,但并不等待其它的事件。这种情况下,poll()就像它的名字那样,一旦选举出来,立即返回。

函数原型

int poll(struct pollfd fdarray[], nfds_t nfds, int timeout);
poll返回值和错误代码
成功时,poll()返回结构体中revents域不为0的文件描述符个数;如果在超时前没有任何事件发生,poll()返回0;失败时,poll()返回-1,并设置errno为下列值之一:
EBADF
一个或多个结构体中指定的文件描述符无效。
EFAULT
fds指针指向的地址超出进程的地址空间。
EINTR
请求的事件之前产生一个信号,调用可以重新发起。
EINVAL
nfds参数超出PLIMIT_NOFILE值。
ENOMEM
可用内存不足,无法完成请求。

以上理论摘自http://www.cnblogs.com/nathan-1988/archive/2012/07/01/2571786.html 谢谢~~~

代码如下所示:

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <errno.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <poll.h>
#define MYPORT 6666    // the port users will be connecting to

#define BACKLOG 5     // how many pending connections queue will hold
#ifndef INFTIM
#define INFTIM -1
#endif
#define BUF_SIZE 1024
#define MAXLINE 100
#define MAXCLIENT 5
int fd_access[BACKLOG];    // accepted connection fd
int conn_amount;    // current connection amount

int main(void)
{
        int sock_fd, new_fd,len,conn_no;  // listen on sock_fd, new connection on new_fd
        struct sockaddr_in server_addr;    // server address information
        struct sockaddr_in client_addr; // connector's address information
        socklen_t sin_size;
        char buf[BUF_SIZE];
        int result;
        int i;
        int maxsock=0;
        int conn_fd;
        struct pollfd client[MAXCLIENT];
        if ((sock_fd = socket(AF_INET, SOCK_STREAM, 0)) == -1) 
        {
                perror("socket");
				exit(1);
        } 
        server_addr.sin_family = AF_INET;         // host byte order
        server_addr.sin_port = htons(MYPORT);     // short, network byte order
        server_addr.sin_addr.s_addr = INADDR_ANY; // automatically fill with my IP
        memset(server_addr.sin_zero, '\0', sizeof(server_addr.sin_zero));
        if (bind(sock_fd, (struct sockaddr *)&server_addr, sizeof(server_addr)) == -1) 
        {
                perror("bind");
                exit(1);
        }
        if (listen(sock_fd, BACKLOG) == -1) 
        {
                perror("listen");
                exit(1);
        }
        printf("listen port %d\n", MYPORT);

        sin_size = sizeof(client_addr);
        client[0].fd=sock_fd;
        client[0].events=POLLIN;
        for(i=1;i<MAXCLIENT;i++)
                client[i].fd=-1;
        while (1) 
        {
                printf("helo\n");
                result=poll(client,maxsock+1,INFTIM);
                if(client[0].revents&POLLIN)
                {
                        new_fd=accept(sock_fd,(struct sockaddr*)&client_addr,&sin_size);
                        for(i = 1; i < MAXCLIENT;++i)
                        {
                                if( client[i].fd < 0 )
                                {       
                                        client[i].fd = new_fd;
                                        client[i].events = POLLIN;//POLLRDNORM;
                                        break;
                                }
                        }
                        if( i == MAXCLIENT)
                        {
                                printf("too many clients"); 
                                exit(1);
                        }
                        if( i > maxsock )
                                maxsock = i;
                        if( --result<= 0 )
                                continue;
                }
                for(i = 1; i <= maxsock; i++)
                {
                        if( (conn_fd = client[i].fd) < 0)
                                continue;
                        if(client[i].revents & (POLLIN | POLLERR))
                        {
                                if( (len = read(conn_fd, buf, MAXLINE)) < 0) 
                                {
                                        if( errno == ECONNRESET)
                                        {
                                                close(conn_fd);
                                                client[i].fd = -1;
                                        }
                                        else
                                                perror("read error");
                                }
                                else if(len == 0)
                                {
                                        close(conn_fd);
                                        client[i].fd = -1;
                                }
                                else
                                        printf("%s\n",buf);
                                if(--conn_no <= 0)
                                        break;

                        }
                }
        }
        exit(0);
}

(END) 


以上是服务器端

客户端代码见http://blog.csdn.net/xluren/article/details/8043484#t15


  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值