select、poll和epoll编写简单高并发服务器

本文详细介绍了Linux下三种I/O多路复用函数:select、poll和epoll的工作原理、使用方法及优缺点。select函数适用于小型系统,但文件描述符数量有限制;poll函数提高了文件描述符数量的限制,但仍有性能问题;epoll是Linux特有的,提供更高效和灵活的I/O管理,支持边缘触发和水平触发。通过对这三种函数的比较,可以看出epoll在处理大量并发连接时具有更高的效率。
摘要由CSDN通过智能技术生成

1. select()函数

select是一个计算机函数,位于头文件#include <sys/select.h> 。该函数用于监视文件描述符的变化情况——读写或是异常。

  • 头文件
#include <sys/select.h>
//使用select可能会用到的头文件
#include <sys/time.h>
#include <sys/types.h>
#include <unistd.h>
  • 函数原型
    int select (int maxfd + 1,fd_set *readset,fd_set *writeset,fd_set *exceptset,const struct timeval *timeout);
  • 参数
    参数一:是一个整数值,是指集合中所有文件描述符的范围,即所有文件描述符的最大值加1。
    参数二:是指向fd_set结构的指针,这个集合中应该包括文件描述符,我们是要监视这些文件描述符的读变化的。
    参数三:同上,但是监视的是这些文件描述符的写变化
    参数四:同上面两个参数的意图,用来监视文件错误异常
    参数五:一个指向timeval结构的指针,用于决定select等待I/O的最长时间。如果为空将一直等待。
    timeval结构的定义:
struct timeval{
long tv_sec; 		// seconds秒
long tv_usec; 		// microseconds微妙
}
  • 返回值
    >0:集合中有文件可读,select就会返回一个大于0的值,表示有文件可读
    <0:出错
    =0:超出timeout的时间,select返回0

fd_set集合通常有以下四个宏来操作:

void FD_ZERO(fd_set *fdset);  	//清空fdset中所有文件描述符
void FD_SET(int fd,fd_set *fdset);  //添加文件描述符fd到集合fdset中
void FD_CLR(int fd,fd_set *fdset);   //将文件描述符fd从集合fdset中去除
int FD_ISSET(int fd,fd_set *fdset); 	//判断描述符fd是否在给定的描述符集fdset中

FD_ISSET()判断描述符fd是否在给定的描述符集fdset中,通常配合select函数使用,由于select函数成功返回时会将未准备好(无事件发生)的描述符位清零。通常我们使用FD_ISSET是为了检查在select函数返回后,某个描述符是否准备好,以便进行接下来的处理操作。
在这里插入图片描述

select的优缺点:

优点:
(1) select的可移植性较好,可以跨平台;
(2) select可设置的监听时间timeout精度更好,可精确到微秒,而poll为毫秒。
缺点:
(1) select支持的文件描述符数量上限为1024,不能根据用户需求进行更改;
(2) select每次调用时都要将文件描述符集合从用户态拷贝到内核态,开销较大;
(3) select返回的就绪文件描述符集合,需要用户循环遍历所监听的所有文件描述符是否在该集合中,当监听描述符数量很大时效率较低。

下面select、poll和epoll代码里声明的int socket_server_init(char *ip, int port);函数在最后有给定义。

select代码:

#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <errno.h>
#include <stdlib.h>
#include <sys/socket.h>
#include <getopt.h>
#include <libgen.h>
#include <ctype.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <sys/time.h>
#include <sys/select.h>

#define ARRAY_SIZE(x)   (sizeof(x)/sizeof(x[0]))

int socket_server_init(char *ip, int port);		//定义的一个获取listen_fd的函数
void print_usage(char *program)	//简单的参数使用说明
{
        printf("%s usage:\n", program);
        printf("-d(--daemon): no argument (optional)\n");
        printf("-p(--port  ): sepcify argument port\n");
        printf("-h(--help  ): no argument (optional)\n");
}
int main(int argc, char *argv[])
{
        int                     port = 0;
        int                     ch;
        int                     len = 16;
        int                     i, j;
        int                     rv;
        int                     listen_fd, client_fd;
        int                     found;
        int                     daemon_run = 0;
        int                     maxfd;
        fd_set                  rdset;
        char                    *IP = NULL;
        char                    *progname;
        int                     limit[1024];
        char                    buf[256];
        struct  sockaddr_in     cli_addr;
        struct option           opts[] = {
                {"daemon", required_argument, NULL, 'd'},
                {"port", required_argument, NULL, 'p'},
                {"help", required_argument, NULL, 'h'},
                {NULL, 0, NULL, 0}
        };
        while((ch = getopt_long(argc, argv, "dp:h", opts, NULL)) != -1)
        {
                switch(ch)
                {
                        case 'd':
                                daemon_run = 1;
                                break;
                        case 'p':
                                port = atoi(optarg);
                                break;
                        case 'h':
                                print_usage(progname);
                                break;
                        default:
                                printf("Unknown return val: %d\n", ch);
                }
        }
        progname = basename(argv[0]);
        if(!port)
        {
                print_usage(progname);
                return -1;
        }
        if(daemon_run)
        {
                daemon(0,0);
        }
        if((listen_fd = socket_server_init(NULL, port)) < 0)
        {
                printf("ERROR: %s server listen port[%d] failure.\n", progname, port);
                goto cleanup;
        }
        for(i = 0; i < ARRAY_SIZE(limit); i++)
        {
                limit[i] = -1;		//表示该位置为空,可以存放套接字
        }
        limit[0] = listen_fd;
        while(1)
        {
                FD_ZERO(&rdset);		//清空rdset中所有文件描述符
                for(i = 0; i < ARRAY_SIZE(limit); i++)
                {
                        if(limit[i] < 0)
                                continue;
                        maxfd = limit[i] > maxfd ? limit[i]:maxfd;
                        FD_SET(limit[i], &rdset);		//将数组中存放的文件描述符都添加到集合rdset中
                }
                rv = select(maxfd+1, &rdset, NULL, NULL, NULL);
                if(rv < 0)		//出错
                {
                        printf("select failure: %s\n", strerror(errno));
                        break;
                }
                else if(rv == 0)		//超时
                {
                        printf("select timeout\n");
                        break;
                }
                //else   rv > 0		//有文件可读
                printf("rv return: %d\n", rv);  	//这里rv的值测试出来都是 1,表示有文件可读
                if(FD_ISSET(listen_fd, &rdset))		//select检测出来是listen_fd发生的事件
                {
                        client_fd = accept(listen_fd, (struct sockaddr*)&cli_addr, &len);
                        //不在意客户端的信息可以把后面两个参数设置为NULL  
                        if(client_fd < 0)
                        {
                                printf("accept new client failure: %s\n", strerror(errno));
                                return -1;
                        }
                        printf("Accept new client[%d][%s:%d] scuess!\n", client_fd, inet_ntoa(cli_addr.sin_addr), ntohs(cli_addr.sin_port));
                        found = 0;
                        for(i = 0; i < ARRAY_SIZE(limit); i++)
                        {
                                if(limit[i] < 0)
                                {
                                        limit[i] = client_fd;	//把新接收的client_fd存到数组中
                                        found = 1;
                                        break;
                                }
                        }
                        if(!found)  //found == 0,limit满了
                        {
                                printf("accept new client scuess,but full...\n");
                                close(client_fd);
                        }
                }
                else		//select检测出来发生的事件不是listen_fd,是其他事件
                {
                        for(i = 0; i < ARRAY_SIZE(limit); i++)
                        {
                                if(limit[i] < 0 || !FD_ISSET(limit[i], &rdset))
                                        continue;
                                printf("client_fd: %d\n", limit[i]);
                                memset(buf, 0, sizeof(buf));
                                rv = read(limit[i], buf, sizeof(buf));
                                if(rv <= 0)
                                {
                                        printf("read failure or get disconnected...\n");
                                        close(limit[i]);
                                        limit[i] = -1;
                                }
                                else    //rv > 0
                                {
                                        printf("read %d bytes data from client[%d]: %s\n", rv, limit[i], buf);
                                        for(j = 0; j < rv; j++)
                                        {
                                                buf[j] = toupper(buf[j]);
                                        }
                                        if(write(limit[i], buf, rv) < 0)
                                        {
                                                printf("client[%d] write failure: %s\n", limit[i], strerror(errno));
                                                close(limit[i]);
                                                limit[i] = -1;
                                        }
                                }
                        }
                }
        }
cleanup:
        close(listen_fd);
        return 0;
}

2. poll()函数

poll()函数:这个函数是某些Unix系统提供的用于执行与select()函数同等功能的函数,下面是这个函数的声明

  • 头文件
    #include <poll.h>
  • 函数原型
    int poll(struct pollfd *fds, nfds_t nfds, int timeout);
  • 参数说明
    fds:是一个struct pollfd结构类型的数组,用于存放需要检测其状态的socket描述符;
struct pollfd {
	int fd; 		/*文件描述符*/
	short events; 	/* 等待的需要测试事件 */
	short revents; 	/* 实际发生了的事件,也就是返回结果 */
};

nfds:nfds_t类型的参数,是描述符的个数,结构体pollfd数组元素的个数;
timeout:是poll函数调用阻塞的时间,单位:毫秒。参数设置为-1时,表示永远阻塞等待,直到一个指定事件发生poll()才返回。0表示立即返回,不阻塞。大于0时,表示等待指定数目的毫秒数后返回。

  • 返回值
    • >0时:数组fds中准备好读、写或出错状态的那些socket描述符的个数;
    • =0时:超时前,数组fds中没有任何socket描述符准备好读、写,或出错;如果timeout==0,那么poll() 函数立即返回而不阻塞;如果timeout==INFTIM,而感兴趣的事件永远不发生,那么poll()就会永远不返回阻塞下去;
    • <0时: poll函数调用失败,同时会自动设置全局变量errno,可能情况如下:
EBADF         一个或多个结构体中指定的文件描述符无效。
EFAULTfds     指针指向的地址超出进程的地址空间。
EINTR         请求的事件之前产生一个信号,调用可以重新发起。
EINVALnfds    参数超出PLIMIT_NOFILE值。
ENOMEM        可用内存不足,无法完成请求。

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

常量说明是否可以作为输入是否可以作为输出
POLLIN普通或优先级带数据可读
POLLRDNORM普通数据可读
POLLRDBAND优先级带数据可读
POLLPRI高优先级数据可读
POLLOUT普通数据可写
POLLWRNORM普通数据可写
POLLWRBAND优先级数据可写
POLLERR发生错误
POLLHUP挂起
POLLNVAL描述字不是一个打开的文件

poll代码:

#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <errno.h>
#include <stdlib.h>
#include <sys/socket.h>
#include <getopt.h>
#include <libgen.h>
#include <ctype.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <sys/time.h>
#include <poll.h>

#define ARRAY_SIZE(x)   (sizeof(x)/sizeof(x[0]))

int socket_server_init(char *ip, int port);
void print_usage(char *program)
{
        printf("%s usage:\n", program);
        printf("-d(--daemon): no argument \n");
        printf("-p(--port  ): sepcify argument port\n");
        printf("-h(--help  ): no argument\n");
}
int main(int argc, char *argv[])
{
        int                     port = 0;
        int                     ch;
        int                     len = 16;
        int                     i, j;
        int                     rv;
        int                     listen_fd, client_fd;
        int                     found;
        int                     daemon;
        int                     maxfd;
        char                    *IP = NULL;
        char                    *progname;
        char                    buf[256];
        struct sockaddr_in      cli_addr;
        struct pollfd           fds[1024];
        struct option           opts[] = {
                {"daemon", required_argument, NULL, 'd'},
                {"port", required_argument, NULL, 'p'},
                {"help", required_argument, NULL, 'h'},
                {NULL, 0, NULL, 0}
        };
        while((ch = getopt_long(argc, argv, "dp:h", opts, NULL)) != -1)
        {
                switch(ch)
                {
                        case 'd':
                                daemon = 1;
                                break;
                        case 'p':
                                port = atoi(optarg);
                                break;
                        case 'h':
                                print_usage(progname);
                                break;
                        default:
                                printf("Unknown return val: %d\n", ch);
                }
        }
        progname = basename(argv[0]);
        if(!port)
        {
                print_usage(progname);
                return -1;
        }
        if((listen_fd = socket_server_init(NULL, port)) < 0)
        {
                printf("ERROR: %s server listen port[%d] failure.\n", progname, port);
                goto cleanup;
        }
        for(i = 0; i < ARRAY_SIZE(fds); i++)
        {
                fds[i].fd = -1;
        }
        fds[0].fd = listen_fd;
        fds[0].events = POLLIN;
        maxfd = 0;
        while(1)
        {
                rv = poll(fds, maxfd+1, -1);    //设置为永不超时
                if(rv  < 0)
                {
                        printf("poll() failure: %s\n", strerror(errno));
                        break;
                }
                else if(0 == rv)        //常量写左边,不然容易犯错(已犯错)
                {
                        printf("poll() get timeout...\n");
                        break;
                }
                //else (rv > 0)
                if(fds[0].revents & POLLIN)
                {
                        printf("poll rv return: %d\n", rv);
                        client_fd = accept(listen_fd, (struct sockaddr*)NULL, NULL);
                        if(client_fd < 0)
                        {
                                printf("accept() failure: %s\n", strerror(errno));
                                continue;
                        }
                        printf("accept client[%d] scuess.\n", client_fd);
                        found = 0;
                        for(i = 1; i < ARRAY_SIZE(fds); i++)
                        {
                                if(fds[i].fd < 0)
                                {
                                        fds[i].fd = client_fd;
                                        fds[i].events = POLLIN;
                                        found = 1;
                                        break;
                                }
                        }
                        if(!found)
                        {
                                printf("accept new client scuess, but full...\n");
                                close(client_fd);
                                continue;
                        }
                        maxfd = maxfd > i ? maxfd : i;
                        printf("maxfd: %d\n", maxfd);
                }
                else
                {
                        for(i = 1; i < ARRAY_SIZE(fds); i++)
                        {
                                if(fds[i].fd < 0 || !(fds[i].revents & POLLIN))
                                        continue;
                                printf("i: %d\t fd: %d\n", i, fds[i].fd);
                                memset(buf, 0, sizeof(buf));
                                rv = read(fds[i].fd, buf, sizeof(buf));
                                if(rv <= 0)
                                {
                                        printf("read failure or socket get disconnected: %s\n", strerror(errno));
                                        printf("rv return: %d\n", rv);
                                        close(fds[i].fd);
                                        fds[i].fd = -1;
                                }
                                else
                                {
                                        printf("read %d bytes data from client: %s\n", rv, buf);
                                        printf("rv return: %d\n", rv);
                                        for(j = 0; j < rv; j++)
                                        {
                                                buf[j] = toupper(buf[j]);
                                        }
                                        if(write(fds[i].fd, buf, rv) < 0)
                                        {
                                                printf("write failure: %s\n", strerror(errno));
                                                close(fds[i].fd);
                                                fds[i].fd = -1;
                                        }
                                }
                        }
                }
        }
cleanup:
        close(listen_fd);
        return 0;
}

3. epoll()函数

以下许多相关知识点来源于百度百科。

epoll是Linux内核为处理大批量文件描述符而作了改进的poll,是Linux下多路复用IO接口select/poll的增强版本,它能显著提高程序在大量并发连接中只有少量活跃的情况下的系统CPU利用率。另一点原因就是获取事件的时候,它无须遍历整个被侦听的描述符集,只要遍历那些被内核IO事件异步唤醒而加入Ready队列的描述符集合就行了。epoll除了提供select/poll那种IO事件的水平触发(Level Triggered)外,还提供了边缘触发(Edge Triggered),这就使得用户空间程序有可能缓存IO状态,减少epoll_wait/epoll_pwait的调用,提高应用程序效率。

唯一有点麻烦是epoll有2种工作方式:LT和ET。

  • LT(level triggered)是缺省的工作方式,并且同时支持block和no-block socket.在这种做法中,内核告诉你一个文件描述符是否就绪了,然后你可以对这个就绪的fd进行IO操作。如果你不作任何操作,内核还是会继续通知你的,所以,这种模式编程出错误可能性要小一点。传统的select/poll都是这种模型的代表。
  • ET (edge-triggered)是高速工作方式,只支持non-block socket。在这种模式下,当描述符从未就绪变为就绪时,内核通过epoll告诉你。然后它会假设你知道文件描述符已经就绪,并且不会再为那个文件描述符发送更多的就绪通知,直到你做了某些操作导致那个文件描述符不再为就绪状态了(比如,你在发送,接收或者接收请求,或者发送接收的数据少于一定量时导致了一个EWOULDBLOCK 错误)。但是请注意,如果一直不对这个fd作IO操作(从而导致它再次变成未就绪),内核不会发送更多的通知(only once),不过在TCP协议中,ET模式的加速效用仍需要更多的benchmark确认。

ET和LT的区别就在这里体现,LT事件不会丢弃,而是只要读buffer里面有数据可以让用户读,则不断的通知你。而ET则只在事件发生之时通知。可以简单理解为LT是水平触发,而ET则为边缘触发。LT模式只要有事件未处理就会触发,而ET则只在高低电平变换时(即状态从1到0或者0到1)触发。

  • int epoll_create(int size);
    创建一个epoll的句柄,size用来告诉内核这个监听的数目一共有多大。这个参数不同于select()中的第一个参数,给出最大监听的fd+1的值。需要注意的是,当创建好epoll句柄后,它就是会占用一个fd值,在linux下如果查看/proc/进程id/fd/,是能够看到这个fd的,所以在使用完epoll后,必须调用close()关闭,否则可能导致fd被耗尽。

注意:size参数只是告诉内核这个 epoll对象会处理的事件大致数目,而不是能够处理的事件的最大个数。在 Linux最新的一些内核版本的实现中,这个 size参数没有任何意义。

  • int epoll_ctl(int epfd, int op, int fd, struct epoll_event *event);
    epoll的事件注册函数,epoll_ctl向 epoll对象中添加、修改或者删除感兴趣的事件,返回0表示成功,否则返回–1。
    此时需要根据errno错误码判断错误类型。
    它不同与select()是在监听事件时告诉内核要监听什么类型的事件,而是在这里先注册要监听的事件类型。
    epoll_wait方法返回的事件必然是通过 epoll_ctl添加到 epoll中的。

    • 第一个参数是epoll_create()的返回值,
    • 第二个参数op表示动作,用三个宏来表示:
      EPOLL_CTL_ADD:注册新的fd到epfd中;
      EPOLL_CTL_MOD:修改已经注册的fd的监听事件;
      EPOLL_CTL_DEL:从epfd中删除一个fd;
    • 第三个参数是需要监听的fd,
    • 第四个参数是告诉内核需要监听什么事,struct epoll_event结构如下:
typedef union epoll_data {
void *ptr;
int fd;
uint32_t u32;
uint64_t u64;
} epoll_data_t;
struct epoll_event {
uint32_t events; / Epoll 事件 /
epoll_data_t data; / 用户数据 /
};

events可以是以下几个宏的集合:

EPOLLIN :表示对应的文件描述符可以读(包括对端SOCKET正常关闭);
EPOLLOUT:表示对应的文件描述符可以写;
EPOLLPRI:表示对应的文件描述符有紧急的数据可读(这里应该表示有带外数据到来);
EPOLLERR:表示对应的文件描述符发生错误;
EPOLLHUP:表示对应的文件描述符被挂断;
EPOLLET: 将EPOLL设为边缘触发(Edge Triggered)模式,这是相对于水平触发(Level Triggered)来说的。
EPOLLONESHOT:只监听一次事件,当监听完这次事件之后,如果还需要继续监听这个socket的话,需要再次把这个socket加入到EPOLL队列里
  • int epoll_wait(int epfd, struct epoll_event * events, int maxevents, int timeout);
    等待事件的产生,类似于select()调用。参数events用来从内核得到事件的集合,maxevents告之内核这个events有多大,这个 maxevents的值不能大于创建epoll_create()时的size,参数timeout是超时时间(毫秒,0会立即返回,-1将不确定,也有说法说是永久阻塞)。该函数返回需要处理的事件数目,如返回0表示已超时。如果返回–1,则表示出现错误,需要检查 errno错误码判断错误类型。

    • 第1个参数 epfd是 epoll的描述符。
    • 第2个参数 events则是分配好的 epoll_event结构体数组,epoll将会把发生的事件复制到 events数组中(events不可以是空指针,内核只负责把数据复制到这个 events数组中,不会去帮助我们在用户态中分配内存。内核这种做法效率很高)。
    • 第3个参数 maxevents表示本次可以返回的最大事件数目,通常 maxevents参数与预分配的events数组的大小是相等的。
    • 第4个参数 timeout表示在没有检测到事件发生时最多等待的时间(单位为毫秒),如果 timeout为0,则表示 epoll_wait在 rdllist链表中为空,立刻返回,不会等待。

epoll代码:

#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <errno.h>
#include <stdlib.h>
#include <sys/socket.h>
#include <getopt.h>
#include <libgen.h>
#include <ctype.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <sys/time.h>
#include <sys/epoll.h>

#define ARRAY_SIZE(x)   (sizeof(x)/sizeof(x[0]))
#define MAX_EVENTS      1024

int socket_server_init(char *ip, int port);
void print_usage(char *program)
{
        printf("%s usage:\n", program);
        printf("-d(--daemon): no argument \n");
        printf("-p(--port  ): sepcify argument port\n");
        printf("-h(--help  ): no argument\n");
}
int main(int argc, char *argv[])
{
        int                     port = 0;
        int                     ch;
        int                     i, j;
        int                     rv;
        int                     listen_fd, client_fd;
        int                     daemon;
        int                     epollfd;
        int                     events;
        char                    *IP = NULL;
        char                    *progname;
        char                    buf[256];
        struct sockaddr_in      cli_addr;
        struct epoll_event              event;
        struct epoll_event              event_array[MAX_EVENTS];
        struct option           opts[] = {
                {"daemon", required_argument, NULL, 'd'},
                {"port", required_argument, NULL, 'p'},
                {"help", required_argument, NULL, 'h'},
                {NULL, 0, NULL, 0}
        };
        while((ch = getopt_long(argc, argv, "dp:h", opts, NULL)) != -1)
        {
                switch(ch)
                {
                        case 'd':
                                daemon = 1;
                                break;
                        case 'p':
                                port = atoi(optarg);
                                break;
                        case 'h':
                                print_usage(progname);
                                break;
                        default:
                                printf("Unknown return val: %d\n", ch);
                }
        }
        progname = basename(argv[0]);
        if(!port)
        {
                print_usage(progname);
                return -1;
        }
        if((listen_fd = socket_server_init(NULL, port)) < 0)
        {
                printf("ERROR: %s server listen port[%d] failure.\n", progname, port);
                goto cleanup;
        }
        if((epollfd = epoll_create(MAX_EVENTS)) < 0)
        {
                printf("epoll_create() failure: %s\n", strerror(errno));
                return -1;
        }
        event.data.fd = listen_fd;
        event.events = EPOLLIN | EPOLLET;
        if(epoll_ctl(epollfd, EPOLL_CTL_ADD, listen_fd, &event) < 0)
        {
                printf("epoll_ctl() failure: %s\n", strerror(errno));
                return -2;
        }
        while(1)
        {
                if((events = epoll_wait(epollfd, event_array, MAX_EVENTS, -1)) < 0) 	
                //等待事件发生,将发生的事件放到结构体数组event_array中去,并返回发生事件的数目
                {
                        printf("epoll_wait() failure: %s\n", strerror(errno));
                        break;
                }
                else if(0 == events)
                {
                        printf("epoll_wait() get timeout...(it's impossible)\n");
                        continue;
                }
                //else events > 0
                for(i = 0; i < events; i++)		//events表示发生事件套接字的数目
                {
                        if((event_array[i].events & EPOLLERR) || (event_array[i].events & EPOLLHUP))
                        {
                                printf("epoll_wait get error on fd[%d]: %s\n", event_array[i].data.fd, strerror(errno));
                                epoll_ctl(epollfd, EPOLL_CTL_ADD, event_array[i].data.fd, &event);
                                close(event_array[i].data.fd);
                        }
                        if(event_array[i].data.fd == listen_fd)	//监听套接字有事件发生
                        {
                                if((client_fd = accept(listen_fd, (struct sockaddr*)NULL, NULL)) < 0)
                                {
                                        printf("accept new client failure: %s\n", strerror(errno));
                                        continue;
                                } 
                                event.data.fd = client_fd;
                                event.events = EPOLLIN|EPOLLET;
                                if(epoll_ctl(epollfd, EPOLL_CTL_ADD, client_fd, &event) < 0)
                                {
                                        printf("epoll add client socket failure: %s\n", strerror(errno));
                                        close(event_array[i].data.fd);
                                        continue;
                                }
                                printf("epoll add new client socket[%d] ok.\n", client_fd);
                        }
                        else		//其它套接字有事件发生
                        {
                                memset(buf, 0, sizeof(buf));
                                rv = read(event_array[i].data.fd, buf, sizeof(buf));
                                if(rv <= 0)
                                {
                                        printf("read() failure or client get disconnected: %s\n", strerror(errno));
                                        close(event_array[i].data.fd);
                                        continue;
                                }
                                printf("read %d bytes data from client: %s\n", rv, buf);
                                for(j = 0; j < rv; j++)
                                {
                                        buf[j] = toupper(buf[j]);
                                }
                                if(write(event_array[i].data.fd, buf, rv) < 0)
                                {
                                        printf("write failure: %s\n", strerror(errno));
                                        close(event_array[i].data.fd);
                                }
                                printf("write back to client: %s\n", buf);
                        }
                }
        }
cleanup:
        close(listen_fd);
        return 0;
}

int socket_server_init(char *ip, int port);

int socket_server_init(char *ip, int port)
{
        struct  sockaddr_in     serv_addr;
        int                     listen_fd;
        int                     on = 1;
        listen_fd = socket(AF_INET, SOCK_STREAM, 0);
        if(listen_fd < 0)
        {
                printf("create socket failure: %s\n", strerror(errno));
                return -1;
        }
        printf("create socket[fd:%d] scuess\n", listen_fd);
        setsockopt(listen_fd, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on));
        memset(&serv_addr, 0, sizeof(serv_addr));
        serv_addr.sin_family = AF_INET;
        serv_addr.sin_port = htons(port);
        if(!ip)
        {
                serv_addr.sin_addr.s_addr = htonl(INADDR_ANY);
        }
        else            //只监听指定的一个IP
        {
                inet_pton(AF_INET, ip, &serv_addr.sin_addr);
        }
        if(bind(listen_fd, (struct sockaddr *)&serv_addr, sizeof(serv_addr)) < 0)
        {
                printf("bind() port failure: %s\n", strerror(errno));
                return -1;
        }
        if(listen(listen_fd, 13) < 0)
        {
                printf("listen error: %s\n", strerror(errno));
                return -1;
        }
        return listen_fd;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值