梦应逍遥 发表于 2024-7-30 13:28:45

【Linux网络】poll{初识poll / poll接口 / poll vs select / poll开辟多客

1.初识poll

https://i-blog.csdnimg.cn/direct/ce44de6e6aae404b87e6d91bbb3f3c05.png
https://i-blog.csdnimg.cn/direct/ccb54716a9c145f29b98bb53f3073fbf.png
poll是Linux体系中的一个体系调用,它用于监控多个文件形貌符(file descriptors,如套接字、管道、文件等)的状态变革。通过poll,步伐可以同时等待多个文件形貌符上发生的特定事件(如可读、可写、错误等),而无需为每个文件形貌符创建单独的线程或进程。这使得poll成为处理多个并发连接和I/O操作的有效手段。
#include <poll.h>
int poll(struct pollfd *fds, nfds_t nfds, int timeout);
fds:是一个指向struct pollfd结构体数组的指针,每个元素指定了一个要检查的文件形貌符及其感爱好的事件。
nfds:指定了fds数组中元素的数量。
timeout:指定了poll调用阻塞的时间,以毫秒为单位。如果设置为-1,poll将无穷期地等待;如果设置为0,poll将立即返回;如果设置为正数,poll将在指定的毫秒数后超时。
poll与select的主要联系与区别


[*]联系
多路复用:poll和select都是I/O多路复用机制,允许步伐同时监控多个文件形貌符,提高了步伐处理并发I/O的效率。
事件通知:两者都提供了一种机制,通过事件通知步伐哪些文件形貌符已经准备好进行读取、写入或有异常环境。
非阻塞I/O:它们都可以与非阻塞I/O联合使用,使得步伐可以在没有数据可读或写入时继续实行其他任务。
[*]区别


[*] 文件形貌符数量限制:
select:通常受到体系文件形貌符数量限制的限制,一般在1024个左右。
poll:没有文件形貌符数量的硬限制,理论上可以监控更多的文件形貌符(但实际上受限于体系资源和内存限制)。
[*] 数据结构:
select:使用三个独立的位图(bitmap)来跟踪文件形貌符的状态,这些位图在内核空间维护。
poll:使用一个struct pollfd结构体数组来存储要监控的文件形貌符及其事件,这个数组在用户空间维护。
[*] 事件信息丰富度:
select:需要步伐遍历位图来查找事件,对于每个文件形貌符的状态变革,select提供的信息相对较少。
poll:struct pollfd结构体中的revents字段在事件发生时会被内核设置,提供了更丰富的信息关于每个文件形貌符的状态变革。
[*] 性能:
select:在文件形貌符数量较多时,性能会下降,由于需要遍历整个位图来查找停当的文件形貌符。
poll:虽然也需要在用户空间和内核空间之间复制文件形貌符集合,但由于其数据结构的计划,在处理大量文件形貌符时大概具有更好的性能。然而,如果监控的文件形貌符数量非常大,仍然大概碰到性能瓶颈。
[*] 超时精度:
select:超时参数是一个整数值,表现调用应该等待的秒数,精度较低。
poll:超时参数是一个毫秒值,提供了更高的时间精度。
[*] 移植性:
两者都具有良好的可移植性,可以在差别的Unix-like体系中使用。但需要注意的是,在某些特定体系或环境下,poll的支持大概不如select广泛。
综上所述,poll和select在Linux体系中都扮演着重要的角色,但在具体使用时需要根据应用场景、文件形貌符数量、性能要求等因素进行选择。
poll的原理

poll函数是Linux体系中的一个重要体系调用,用于监控多个文件形貌符(file descriptors)的状态变革。下面从参数和底层原理两个方面对poll函数进行简要叙述。
#include <poll.h>
int poll(struct pollfd fds[], nfds_t nfds, int timeout);
fds:这是一个指向struct pollfd结构体数组的指针。每个pollfd结构体代表了一个要监控的文件形貌符及其感爱好的事件。pollfd结构体的界说通常如下:
c
struct pollfd {
    int fd;          // 文件描述符
    short events;    // 等待的事件
    short revents;   // 实际发生了的事件
};
fd:要监控的文件形貌符。
events:请求监控的事件,可以是读、写、异常等多种事件的组合。
revents:函数返回时,由内核设置,表现实际发生的事件。
nfds:指定了fds数组中元素的数量,即要监控的文件形貌符的总数。
timeout:指定了poll调用阻塞的时间,以毫秒为单位。
如果timeout为正数,poll将等待指定的毫秒数。
如果timeout为0,poll将立即返回,不阻塞。
如果timeout为-1,poll将无穷期地等待,直到有事件发生。
   poll函数的底层原理
poll函数的底层实现原理主要基于等待队列。当调用poll函数时,它会遍历传入的pollfd结构体数组,为每个感爱好的文件形貌符注册一个等待事件。这些等待事件会被挂接到内核中相应的等待队列上。
等待队列:内核中的每个文件形貌符都大概关联有一个或多个等待队列,用于存放等待该文件形貌符上特定事件发生的进程或线程。
轮询与阻塞:poll函数会轮询检查每个文件形貌符的等待队列,检察是否有事件发生。
如果有文件形貌符上发生了感爱好的事件,poll函数会立即返回,并将这些事件记录在对应pollfd结构体的revents字段中。
如果没有任何文件形貌符上发生事件,而且timeout参数指定了非零值,poll函数会进入阻塞状态,直到超时时间到达或至少有一个文件形貌符上发生了事件。
   返回与错误处理:
当有文件形貌符上发生事件或超时时间到达时,poll函数会返回。返回值表现发生了事件的文件形貌符数量(如果大于0),大概在超时时返回0,大概在发生错误时返回-1并设置errno以指示错误缘故原由。
效率与限制:虽然poll函数没有像select那样有文件形貌符数量的硬限制,但在处理大量文件形貌符时,仍然需要将整个pollfd数组在用户空间和内核空间之间复制,这大概会导致性能下降。此外,poll函数在内部仍然需要遍历全部要监控的文件形貌符,因此在大规模并发场景下大概不是最高效的办理方案。
综上所述,poll函数通过轮询和等待队列机制实现了对多个文件形貌符状态的监控,是Linux体系中处理I/O多路复用的重要手段之一。然而,在处理大量文件形貌符时,大概需要考虑其性能限制并探索更高效的办理方案(如epoll)。
   源码
#include <poll.h>
int poll(struct pollfd *fds, nfds_t nfds, int timeout);
// pollfd结构struct pollfd { int fd; /* file descriptor */ short events; /* requested events */ short revents; /* returned events */}; 多路转接包括:用户告诉内核你需要关心什么 && 内核告诉用户你让我关心的fd有哪些停当了。select 用位图,poll 用结构体数组。poll 在用户传给内核的时间,告诉内核需要关心 struct pollfd 结构体中的 fd 中的 events 事件;返回时,内核设置struct pollfd 结构体中的 revents 事件,表现该fd的该事件停当。poll 最大的特点:将输入和输出事件进行分离!
内核怎么知道是关心读事件还是写事件还是其他事件呢?当内核返回用户也一样。 events 和 revents 都是 short 类型,都是 16 个比特位,在 Linux 中,使用比特位传参!把事件设置成位图的情势。
https://i-blog.csdnimg.cn/direct/dfffd4c93dbd47948662f04a30fe4f01.png
poll 的本质是将读写事件分离,传入用户定的数组元素的大小,通过 events 和 revents 以位图的方式来传递停当和关心标记位的办理方案!
poll的优点

poll 也是多路转接方案的一种,它主要办理的就是 select 中的等待 fd 有上限的问题,以及每次都要对关心的 fd 进行事件重置的问题。
差别与select使用三个位图来表现三个fdset的方式,poll使用一个pollfd的指针实现.
pollfd结构包罗了要监视的event和发生的event,不再使用select“参数-值”传递的方式. 接口使用比select更方便。
poll并没有最大数量限制 (但是数量过大后性能也是会下降).
poll的缺点

poll中监听的文件形貌符数目增多时和select函数一样,poll返回后,需要轮询pollfd来获取停当的形貌符.
每次调用poll都需要把大量的pollfd结构从用户态拷贝到内核中.
同时连接的大量客户端在一时候大概只有很少的处于停当状态, 因此随着监视的形貌符数量的增长, 其效率也会线性下降.
poll vs select

poll 本质上是通过一个结构体数组来等待 fd 的,它办理了 select 等待 fd 有上限的问题,怎样办理?_event_fds 这个数组的大小是自己定的,可以定的非常大,大到内存扛不住,此时就是操作体系/内存 软件或硬件的问题了,不是 poll 接口自己的问题。select 等待 fd 有上限的问题,本质上是接口自己的问题,poll 本质上是办理了 select 等待 fd 有上限的问题。
poll 与 select 都需要遍历检测有哪些文件形貌符停当,poll 在内核中需要遍历检测有哪些文件形貌符停当;在用户层需要遍历检测有哪些事件已经停当。
poll 和 select 都避免不开遍历的问题,当fd过多,效率提升不明显。
2.poll开辟多客户端echo服务器

封装套接字接口

#pragma once

#include <iostream>
#include <string>
#include <unistd.h>
#include <cstring>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include "Log.hpp"

enum
{
    SocketErr = 2,
    BindErr,
    ListenErr,
};

const int g_backlog = 10;

class Sock
{
private:
    int _sockfd;

public:
    Sock()
    {
    }
    ~Sock()
    {
    }

public:
    void Socket()
    {
      _sockfd = socket(AF_INET, SOCK_STREAM, 0);
      if (_sockfd < 0)
      {
            lg(Fatal, "socker error, %s: %d", strerror(errno), errno);
            exit(SocketErr);
      }
    }

    void Bind(uint16_t port)
    {
      struct sockaddr_in local;
      memset(&local, 0, sizeof(local));
      local.sin_family = AF_INET;
      local.sin_port = htons(port);
      local.sin_addr.s_addr = INADDR_ANY;

      if (bind(_sockfd, (struct sockaddr *)&local, sizeof(local)) < 0)
      {
            lg(Fatal, "bind error, %s: %d", strerror(errno), errno);
            exit(BindErr);
      }
    }

    void Listen()
    {
      if (listen(_sockfd, g_backlog) < 0)
      {
            lg(Fatal, "listen error, %s: %d", strerror(errno), errno);
            exit(ListenErr);
      }
    }

    int Accept(std::string *clientip, uint16_t *clientport)
    {
      struct sockaddr_in peer;
      socklen_t len = sizeof(peer);
      int newfd = accept(_sockfd, (struct sockaddr *)&peer, &len);
      if (newfd < 0)
      {
            lg(Warning, "accept error, %s: %d", strerror(errno), errno);
            return -1;
      }

      char ipstr;
      inet_ntop(AF_INET, &peer.sin_addr, ipstr, sizeof(ipstr));

      *clientip = ipstr;
      *clientport = ntohs(peer.sin_port);

      return newfd;
    }

    bool Connect(const std::string &ip, const uint16_t &port)
    {
      struct sockaddr_in peer;
      memset(&peer, 0, sizeof(peer));
      peer.sin_family = AF_INET;
      peer.sin_port = htons(port);
      inet_pton(AF_INET, ip.c_str(), &(peer.sin_addr));

      int n = connect(_sockfd, (struct sockaddr *)&peer, sizeof(peer));
      if (n == -1)
      {
            std::cerr << "connect to " << ip << ":" << port << " error" << std::endl;
            return false;
      }
      return true;
    }

    void CloseFd()
    {
      close(_sockfd);
    }

    int getSocketFd()
    {
      return _sockfd;
    }
};
Makefile

poll_server:Main.cc
        g++ -o $@ $^ -std=c++11
.PHONY:clean
clean:
        rm -f poll_server
主函数

#include "PollServer.hpp"
#include <memory>

int main()
{
    std::unique_ptr<PollServer> svr(new PollServer());
    svr->Init();
    svr->Start();

    return 0;
}
日志服务

#pragma once

#include <iostream>
#include <time.h>
#include <stdarg.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>
#include <stdlib.h>

#define SIZE 1024

#define Info 0
#define Debug 1
#define Warning 2
#define Error 3
#define Fatal 4

#define Screen 1
#define Onefile 2
#define Classfile 3

#define LogFile "log.txt"

class Log
{
private:
    int printMethod;
    std::string path;

public:
    Log()
    {
      printMethod = Screen;
      path = "./";
    }

    void Enable(int method)
    {
      printMethod = method;
    }

    std::string levelToString(int level)
    {
      switch (level)
      {
      case Info:
            return "Info";
      case Debug:
            return "Debug";
      case Warning:
            return "Warning";
      case Error:
            return "Error";
      case Fatal:
            return "Fatal";
      default:
            return "None";
      }
    }

    /*
   void logmessage(int level, const char *format, ...)
      {
            time_t t = time(nullptr);
            struct tm *ctime = localtime(&t);
            char leftbuffer;
            snprintf(leftbuffer, sizeof(leftbuffer), "[%s][%d-%d-%d %d:%d:%d]", levelToString(level).c_str(),
                     ctime->tm_year + 1900, ctime->tm_mon + 1, ctime->tm_mday,
                     ctime->tm_hour, ctime->tm_min, ctime->tm_sec);

            va_list s;
            va_start(s, format);
            char rightbuffer;
            vsnprintf(rightbuffer, sizeof(rightbuffer), format, s);
            va_end(s);

            // 格式:默认部分+自定义部分
            char logtxt;
            snprintf(logtxt, sizeof(logtxt), "%s %s\n", leftbuffer, rightbuffer);

            // printf("%s", logtxt);
            printLog(level, logtxt);
      }
    */

   // lg(Warning, "accept error, %s: %d", strerror(errno), errno);
    void operator()(int level, const char *msg_format, ...)
    {
      time_t timestamp = time(nullptr);
      struct tm *ctime = localtime(&timestamp);
      //level 年月日
      char leftbuffer;
      snprintf(leftbuffer, sizeof(leftbuffer), "[%s][%d-%d-%d %d:%d:%d]", levelToString(level).c_str(),
               ctime->tm_year + 1900, ctime->tm_mon + 1, ctime->tm_mday,
               ctime->tm_hour, ctime->tm_min, ctime->tm_sec);

      //自定义msg
      va_list arg_list;//存储可变参数列表信息
      va_start(arg_list, msg_format);//初始化 使其指向函数参数列表中format参数之后的第一个可变参数
      char rightbuffer;
      vsnprintf(rightbuffer, sizeof(rightbuffer), msg_format, arg_list);
      va_end(arg_list);//清理va_list变量

      // 格式:默认部分+自定义部分
      char log_content;
      snprintf(log_content, sizeof(log_content), "%s %s", leftbuffer, rightbuffer);

      // printf("%s", logtxt); // 暂时打印
      printLog(level, log_content);
    }

    void printLog(int level, const std::string &log_content)
    {
      switch (printMethod)
      {
      case Screen:
            std::cout << log_content << std::endl;
            break;
      case Onefile:
            printOneFile(LogFile, log_content);
            break;
      case Classfile:
            printClassFile(level, log_content);
            break;
      default:
            break;
      }
    }

    void printOneFile(const std::string &log_filename, const std::string &log_content)
    {
      //path = "./"; #define LogFile "log.txt"
      std::string _logFilename = path + log_filename;
      int fd = open(_logFilename.c_str(), O_WRONLY | O_CREAT | O_APPEND, 0666); // "log.txt"
      if (fd < 0)
            return;
      write(fd, log_content.c_str(), log_content.size());
      close(fd);
    }

    void printClassFile(int level, const std::string &log_content)
    {
      //#define LogFile "log.txt"
      std::string filename = LogFile;
      filename += ".";
      filename += levelToString(level); // "log.txt.Debug"
      printOneFile(filename, log_content);
    }

    ~Log()
    {
    }
};

Log lg;

/*
int sum(int n, ...)
{
    va_list s; // char*
    va_start(s, n);

    int sum = 0;
    while(n)
    {
      sum += va_arg(s, int); // printf("hello %d, hello %s, hello %c, hello %d,", 1, "hello", 'c', 123);
      n--;
    }

    va_end(s); //s = NULL
    return sum;
}
*/

聊天服务器

#pragma once

#include <iostream>
#include <poll.h>
#include <sys/time.h>
#include "Socket.hpp"

using namespace std;

static const uint16_t defaultport = 8888;
static const int fd_num_max = 64;
int defaultFd = -1;
int non_event = 0;

class PollServer
{
private:
    Sock _listensock;
    uint16_t _port;
    struct pollfd _event_fds; // 结构体数组
    // struct pollfd *_event_fds; 动态数组 可扩容

    // int fd_array;
    // int wfd_array;
public:
    PollServer(uint16_t port = defaultport)
      : _port(port)
    {
      for (int i = 0; i < fd_num_max; i++)
      {
            _event_fds.fd = defaultFd;
            _event_fds.events = non_event;
            _event_fds.revents = non_event;
      }
    }
    bool Init()
    {
      _listensock.Socket();
      _listensock.Bind(_port);
      _listensock.Listen();

      return true;
    }
    void Accepter()
    {
      // 连接事件就绪
      std::string clientip;
      uint16_t clientport = 0;
      int sock = _listensock.Accept(&clientip, &clientport); // 会不会阻塞在这里?不会
      if (sock < 0)
            return;
      lg(Info, "accept success, %s: %d, sock fd: %d", clientip.c_str(), clientport, sock);

      // sock -> _event_fds[]
      int pos = 1;
      for (; pos < fd_num_max; pos++) // 第二个循环
      {
            if (_event_fds.fd != defaultFd)
                continue;
            else
                break;
      }
      if (pos == fd_num_max)
      {
            lg(Warning, "server is full, close %d now!", sock);
            close(sock);
            // 也可以不关闭fd 扩容存fd
      }
      else
      {
             _event_fds.fd = sock;
            _event_fds.events = POLLIN;
            _event_fds.revents = non_event;
            PrintFd();
      }
    }
    void Recver(int fd, int pos)
    {
      char buffer;
      ssize_t n = read(fd, buffer, sizeof(buffer) - 1); // 需要考虑是否是完整数据包问题(此处忽略)
      if (n > 0)
      {
            buffer = 0;
            cout << "get a messge: " << buffer << endl;
      }
      else if (n == 0)
      {
            lg(Info, "client quit, me too, close fd is : %d", fd);
            close(fd);
            _event_fds.fd = defaultFd;
      }
      else
      {
            lg(Warning, "recv error: fd is : %d", fd);
            close(fd);
            _event_fds.fd = defaultFd;
      }
    }
    void Dispatcher()
    {
      for (int i = 0; i < fd_num_max; i++) // 这是第三个循环
      {
            int fd = _event_fds.fd;
            if (fd == defaultFd)
                continue;

            if (_event_fds.revents & POLLIN)
            {
                if (fd == _listensock.getSocketFd())
                  Accepter(); // 连接管理器
                else
                  Recver(fd, i); // non listenfd
            }
      }
    }
    void Start()
    {
      _event_fds.fd = _listensock.getSocketFd();
      _event_fds.events = POLLIN;
      int timeout = 3000; // 3s
      for (;;)
      {
            int n = poll(_event_fds, fd_num_max, timeout);
            switch (n)
            {
            case 0:
                cout << "time out... " << endl;
                break;
            case -1:
                cerr << "poll error" << endl;
                break;
            default:
                // 有事件就绪了
                cout << "get a new link!!!!!" << endl;
                Dispatcher();
                break;
            }
      }
    }
    void PrintFd()
    {
      cout << "online fd list: ";
      for (int i = 0; i < fd_num_max; i++)
      {
            if (_event_fds.fd == defaultFd)
                continue;
            cout << _event_fds.fd << " ";
      }
      cout << endl;
    }
    ~PollServer()
    {
      _listensock.CloseFd();
    }
};

免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!更多信息从访问主页:qidao123.com:ToB企服之家,中国第一个企服评测及商务社交产业平台。
页: [1]
查看完整版本: 【Linux网络】poll{初识poll / poll接口 / poll vs select / poll开辟多客