I/O多路复用之poll

最新推荐文章于 2024-07-19 20:53:43 发布

TangguTae

最新推荐文章于 2024-07-19 20:53:43 发布

阅读量453

点赞数 1

分类专栏： linux 文章标签： c++ I/O 网络 linux poll

本文链接：https://blog.csdn.net/weixin_43164548/article/details/125311260

版权

linux 专栏收录该内容

22 篇文章 0 订阅

订阅专栏

之前有说过I/O多路复用中的select，这次的poll其实和select很相似。它们之间的区别在于我们要如何指定待检查的文件描述符。在select中，提供三个集合readfds，writefds和exceptfds用来标明所感兴趣的文件描述符集合。而在poll中提供一列文件描述符，并在每个文件描述符添加所感兴趣的事件。

poll

poll的参数

第一个参数struct pollfd*类型的参数，也就是该类型的一个数组。

struct pollfd结构体总共有三个成员

对于该结构体，第一个参数是所关心的文件描述符，第二个参数为文件描述符所关心的事件，第三个参数是该文件描述符实际发生的事件。

第二个参数nfds是前面数组对应的大小

第三个参数用来设置poll的阻塞行为。当timeout为-1时表示阻塞等待。为0时表示非阻塞等待，大于0时表示等待的时间。

poll的返回值

返回值大于0时，表示有文件描述符已经就绪。

返回值等于-1时，表示有错误发生。

返回值等于0时，表示poll的等待超时。

poll的测试

#include<iostream>                                                       
#include<poll.h>
#include<cstring>
#include<unistd.h>
#include<sys/socket.h>
#include<sys/types.h>
#include<netinet/in.h>
#include<arpa/inet.h>
using namespace std;

const int NUM = 64;
class PollServer
{
  private:
    int lsock;
    int port;
  public:
    PollServer(int _port)
    {
      port = _port;
    }
    ~PollServer()
    {
      close(lsock);
    }
    void InitServer()//初始化服务器
    {
      lsock = socket(AF_INET,SOCK_STREAM,0);
      if(lsock < 0)
      {
        cerr<<"sock error!"<<endl;
        exit(-1);
      }
      sockaddr_in local;
      local.sin_family = AF_INET;
      local.sin_port = htons(port);
      local.sin_addr.s_addr = INADDR_ANY;
      if(bind(lsock,(struct sockaddr*)&local,sizeof(local)) < 0)
      {
        cerr<<"bind error!"<<endl;
        exit(-1);
      }
      if(listen(lsock,5) < 0)
      {
        cerr<<"listen error!"<<endl;
        exit(-1);
      }
    }
    void ServerStart()
    {
      struct pollfd rfds[NUM];//观测64个文件描述符
      for(int i=0;i < NUM;i++)//先初始化一下
      {
        rfds[i].fd = -1;
        rfds[i].events = 0;
        rfds[i].revents = 0;
      }
      rfds[0].fd = lsock;//首先将监听套接字添加进去
      rfds[0].events |= POLLIN;//关心监听套接字上面的输入事件
      rfds[0].revents = 0;
      while(true)
      {
        switch(poll(rfds,NUM,-1))//以阻塞的方式等待
        {
          case 0:
            cout<<"time out..."<<endl;
            break;
          case -1:
            cout<<"poll error!"<<endl;
            break;
          default:
            for(int i=0; i < NUM; i++)//返回值大于0，遍历所有的套接字
            {
              if(rfds[i].fd == -1)
                continue;
              if(rfds[i].revents & POLLIN)
              {
                if(rfds[i].fd == lsock)
                {  
                  cout<<"get a new link..."<<endl;
                  //新的连接到来
                  //to do
                }
                else 
                {    
                  //其它套接字上面的读事件触发
                  //to do
                }
              }
            }
            break;
        }
      }
    }
};

int main(int argc,char* argv[])
{
  if(argc!=2)
  {
    cout<<"parameter error!!!"<<endl;
    exit(-1);
  }
  PollServer* ps = new PollServer(atoi(argv[1]));
  ps->InitServer();
  ps->ServerStart();
  return 0;
}