epoll和select的例程代码

epoll.c

​
#include<stdio.h>
#include<stdlib.h>
#include<fcntl.h>
#include<sys/stat.h>
#include<sys/types.h>
#include<sys/epoll.h>

int main()
{
    int epollfd = epoll_create(512);

    int fd_key = open("/dev/input/event1", O_RDONLY|O_NONBLOCK);
    int fd_mice = open("/dev/input/mice", O_RDONLY|O_NONBLOCK);

    struct epoll_event ev;
    ev.events = EPOLLIN; // 监控可读

    ev.data.fd = fd_key; 
    epoll_ctl(epollfd, EPOLL_CTL_ADD, fd_key,  &ev);

    ev.data.fd = fd_mice;
    epoll_ctl(epollfd, EPOLL_CTL_ADD, fd_mice, &ev);

    struct epoll_event evout[2];
    char buf[1024];
    while(1)
    {
        int ret = epoll_wait(epollfd, evout, 2, 5000);
        if(ret == 0) continue; // 超时
        if(ret < 0 && errno == EINTR) continue; // 被信号打断
        if(ret < 0) break; // 错误发生了

        // ret > 0情况
        int i;
        for(i=0; i<ret; ++i)
        {
            int fd = evout[i].data.fd;
            if(read(fd, buf, sizeof(buf)) < 0)
            {
                // close自动将它从epoll中移除
                close(fd); 
            }

            if(fd == fd_key) printf("键盘有消息\n");
            else if(fd == fd_mice) printf("鼠标有消息\n");
        }
    }
}

​

 

select.c

#include<stdio.h>
#include<stdlib.h>
#include<fcntl.h>
#include<sys/stat.h>
#include<sys/types.h>
#include<sys/select.h>

int main()
{
    int fd_mice = open("/dev/input/mice", O_RDONLY);
    int fd_key = opeb("/dev/input/event1",O_RDONLY);
    char buf[1024];
    fd_set readfds;  //申请一个文件描述符集合

    while(1)
    {   
        FD_ZERO(&readfds);  //清空集合
        FD_SET(fd_mice,&readfds);  //把fd_mice添加到集合
        FD_SET(fd_key,&readfds);

        struct timeval timeout;
        timeout.tv_sec = 5;
        timeout.tv_usec = 5;

        int ret = select(1024, &readfds, NULL, NULL, &timeout);
        if(ret==0) continue; //阻塞超时,无就绪
        if(ret<0 && errno == EINTR) continue; //被信号打断
        if(ret <0) break; //错误

        if(FD_ISSET(fd_key, &readfds))
        {
            read(fd_key, buf, sizeof(buf));
            printf("keyboard event");
        }
        if(FD_ISSET(fd_mice, &readfds))
        {
            read(fd_mice, buf, sizeof(buf));
            printf("keyboard event");
        }
    }
}

 

sever.c

 #include <stdio.h>
   #include <stdlib.h>
   #include <string.h>
   #include <errno.h>
   
   #include <netinet/in.h>
   #include <sys/socket.h>
   #include <arpa/inet.h>
   #include <sys/epoll.h>
  #include <unistd.h>
  #include <sys/types.h>
  
  #define IPADDRESS   "127.0.0.1"
  #define PORT        8787
  #define MAXSIZE     1024
  #define LISTENQ     5
  #define FDSIZE      1000
  #define EPOLLEVENTS 100
  
  //函数声明
  //创建套接字并进行绑定
  static int socket_bind(const char* ip,int port);
  //IO多路复用epoll
  static void do_epoll(int listenfd);
  //事件处理函数
  static void
  handle_events(int epollfd,struct epoll_event *events,int num,int listenfd,char *buf);
  //处理接收到的连接
  static void handle_accpet(int epollfd,int listenfd);
  //读处理
  static void do_read(int epollfd,int fd,char *buf);
  //写处理
  static void do_write(int epollfd,int fd,char *buf);
  //添加事件
  static void add_event(int epollfd,int fd,int state);
  //修改事件
  static void modify_event(int epollfd,int fd,int state);
  //删除事件
  static void delete_event(int epollfd,int fd,int state);
  
  int main(int argc,char *argv[])
  {
      int  listenfd;
      listenfd = socket_bind(IPADDRESS,PORT);  //创建套接字并进行绑定
      listen(listenfd,LISTENQ);
      do_epoll(listenfd);
      return 0;
  }
  
  static int socket_bind(const char* ip,int port)
  {
      int  listenfd;
      struct sockaddr_in servaddr;
      listenfd = socket(AF_INET,SOCK_STREAM,0);
      if (listenfd == -1)
      {
          perror("socket error:");
          exit(1);
      }
      bzero(&servaddr,sizeof(servaddr));
      servaddr.sin_family = AF_INET;
      inet_pton(AF_INET,ip,&servaddr.sin_addr);
      servaddr.sin_port = htons(port);
      if (bind(listenfd,(struct sockaddr*)&servaddr,sizeof(servaddr)) == -1)
      {
          perror("bind error: ");
          exit(1);
      }
      return listenfd;
  }
  
  static void do_epoll(int listenfd)
  {
      int epollfd;
      struct epoll_event events[EPOLLEVENTS];
      int ret;
      char buf[MAXSIZE];
      memset(buf,0,MAXSIZE);
      //创建一个描述符
      epollfd = epoll_create(FDSIZE);
      //添加监听描述符事件
      add_event(epollfd,listenfd,EPOLLIN);
      for ( ; ; )
      {
          //获取已经准备好的描述符事件
          ret = epoll_wait(epollfd,events,EPOLLEVENTS,-1);
          handle_events(epollfd,events,ret,listenfd,buf);
      }
      close(epollfd);
  }
  
  static void
  handle_events(int epollfd,struct epoll_event *events,int num,int listenfd,char *buf)
  {
      int i;
      int fd;
      //进行选好遍历
      for (i = 0;i < num;i++)
      {
         fd = events[i].data.fd;
         //根据描述符的类型和事件类型进行处理
         if ((fd == listenfd) &&(events[i].events & EPOLLIN))
             handle_accpet(epollfd,listenfd);
         else if (events[i].events & EPOLLIN)
             do_read(epollfd,fd,buf);
         else if (events[i].events & EPOLLOUT)
             do_write(epollfd,fd,buf);
     }
 }
 static void handle_accpet(int epollfd,int listenfd)
 {
     int clifd;
     struct sockaddr_in cliaddr;
     socklen_t  cliaddrlen;
     clifd = accept(listenfd,(struct sockaddr*)&cliaddr,&cliaddrlen);
     if (clifd == -1)
         perror("accpet error:");
     else
     {
         printf("accept a new client: %s:%d\n",inet_ntoa(cliaddr.sin_addr),cliaddr.sin_port);
         //添加一个客户描述符和事件
         add_event(epollfd,clifd,EPOLLIN);
     }
 }
 
 static void do_read(int epollfd,int fd,char *buf)
 {
     int nread;
     nread = read(fd,buf,MAXSIZE);
     if (nread == -1)
     {
         perror("read error:");
         close(fd);
         delete_event(epollfd,fd,EPOLLIN);
     }
     else if (nread == 0)
     {
         fprintf(stderr,"client close.\n");
         close(fd);
         delete_event(epollfd,fd,EPOLLIN);
     }
     else
     {
         printf("read message is : %s",buf);
         //修改描述符对应的事件,由读改为写
         modify_event(epollfd,fd,EPOLLOUT);
     }
 }
 
 static void do_write(int epollfd,int fd,char *buf)
 {
     int nwrite;
     nwrite = write(fd,buf,strlen(buf));
     if (nwrite == -1)
     {
         perror("write error:");
         close(fd);
         delete_event(epollfd,fd,EPOLLOUT);
     }
     else
         modify_event(epollfd,fd,EPOLLIN);
     memset(buf,0,MAXSIZE);
 }
 
 static void add_event(int epollfd,int fd,int state)
 {
     struct epoll_event ev;
     ev.events = state;
     ev.data.fd = fd;
     epoll_ctl(epollfd,EPOLL_CTL_ADD,fd,&ev);
 }
 
 static void delete_event(int epollfd,int fd,int state)
 {
     struct epoll_event ev;
     ev.events = state;
     ev.data.fd = fd;
     epoll_ctl(epollfd,EPOLL_CTL_DEL,fd,&ev);
 }
 
 static void modify_event(int epollfd,int fd,int state)
 {
     struct epoll_event ev;
     ev.events = state;
     ev.data.fd = fd;
     epoll_ctl(epollfd,EPOLL_CTL_MOD,fd,&ev);
 }

clien.c

    #include <netinet/in.h>
   #include <sys/socket.h>
    #include <stdio.h>
    #include <string.h>
    #include <stdlib.h>
    #include <sys/epoll.h>
    #include <time.h>
    #include <unistd.h>
    #include <sys/types.h>
   #include <arpa/inet.h>
   
   #define MAXSIZE     1024
   #define IPADDRESS   "127.0.0.1"
   #define SERV_PORT   8787
   #define FDSIZE        1024
   #define EPOLLEVENTS 20
   
   static void handle_connection(int sockfd);
   static void
   handle_events(int epollfd,struct epoll_event *events,int num,int sockfd,char *buf);
   static void do_read(int epollfd,int fd,int sockfd,char *buf);
   static void do_read(int epollfd,int fd,int sockfd,char *buf);
   static void do_write(int epollfd,int fd,int sockfd,char *buf);
   static void add_event(int epollfd,int fd,int state);
   static void delete_event(int epollfd,int fd,int state);
   static void modify_event(int epollfd,int fd,int state);
   
   int main(int argc,char *argv[])
   {
       int                 sockfd;
       struct sockaddr_in  servaddr;
       sockfd = socket(AF_INET,SOCK_STREAM,0);
       bzero(&servaddr,sizeof(servaddr));
       servaddr.sin_family = AF_INET;
       servaddr.sin_port = htons(SERV_PORT);
       inet_pton(AF_INET,IPADDRESS,&servaddr.sin_addr);
       connect(sockfd,(struct sockaddr*)&servaddr,sizeof(servaddr));
       //处理连接
       handle_connection(sockfd);
       close(sockfd);
       return 0;
   }
   
   
   static void handle_connection(int sockfd)
   {
       int epollfd;
       struct epoll_event events[EPOLLEVENTS];
       char buf[MAXSIZE];
       int ret;
       epollfd = epoll_create(FDSIZE);
       add_event(epollfd,STDIN_FILENO,EPOLLIN);
       for ( ; ; )
       {
           ret = epoll_wait(epollfd,events,EPOLLEVENTS,-1);
           handle_events(epollfd,events,ret,sockfd,buf);
       }
       close(epollfd);
   }
   
   static void
   handle_events(int epollfd,struct epoll_event *events,int num,int sockfd,char *buf)
   {
       int fd;
       int i;
       for (i = 0;i < num;i++)
       {
           fd = events[i].data.fd;
           if (events[i].events & EPOLLIN)
               do_read(epollfd,fd,sockfd,buf);
           else if (events[i].events & EPOLLOUT)
               do_write(epollfd,fd,sockfd,buf);
       }
   }
   
   static void do_read(int epollfd,int fd,int sockfd,char *buf)
   {
       int nread;
       nread = read(fd,buf,MAXSIZE);
           if (nread == -1)
       {
           perror("read error:");
           close(fd);
       }
       else if (nread == 0)
       {
           fprintf(stderr,"server close.\n");
           close(fd);
       }
       else
       {
           if (fd == STDIN_FILENO)
               add_event(epollfd,sockfd,EPOLLOUT);
           else
           {
               delete_event(epollfd,sockfd,EPOLLIN);
               add_event(epollfd,STDOUT_FILENO,EPOLLOUT);
           }
       }
  }
  
  static void do_write(int epollfd,int fd,int sockfd,char *buf)
  {
      int nwrite;
      nwrite = write(fd,buf,strlen(buf));
      if (nwrite == -1)
      {
          perror("write error:");
          close(fd);
      }
      else
      {
          if (fd == STDOUT_FILENO)
              delete_event(epollfd,fd,EPOLLOUT);
          else
              modify_event(epollfd,fd,EPOLLIN);
      }
      memset(buf,0,MAXSIZE);
  }
  
  static void add_event(int epollfd,int fd,int state)
  {
      struct epoll_event ev;
      ev.events = state;
      ev.data.fd = fd;
      epoll_ctl(epollfd,EPOLL_CTL_ADD,fd,&ev);
  }
  
  static void delete_event(int epollfd,int fd,int state)
  {
      struct epoll_event ev;
      ev.events = state;
      ev.data.fd = fd;
      epoll_ctl(epollfd,EPOLL_CTL_DEL,fd,&ev);
  }
  
  static void modify_event(int epollfd,int fd,int state)
  {
      struct epoll_event ev;
      ev.events = state;
      ev.data.fd = fd;
      epoll_ctl(epollfd,EPOLL_CTL_MOD,fd,&ev);
  }

 

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值