最近学习网络编程相关,学习资源在b站搜索【linux网络编程】。本文章记录相关学习心得。
相关API
- 创建一个epoll句柄,参数size用来告诉内核监听的文件描述符的个数,跟内存大小有关。此步骤创建一个建议大小为size的红黑树(二叉平衡树),返回值为根结点的fd
#include <sys/epoll.h>
int epoll_create(int size) size:监听数目
- 控制某个epoll监控的文件描述符上的事件:注册、修改、删除。此步骤向根结点增加,删除修改节点等
#include <sys/epoll.h>
int epoll_ctl(int epfd, int op, int fd, struct epoll_event *event)
epfd: 为epoll_creat的句柄
op: 表示动作,用3个宏来表示:
EPOLL_CTL_ADD (注册新的fd到epfd),
EPOLL_CTL_MOD (修改已经注册的fd的监听事件),
EPOLL_CTL_DEL (从epfd删除一个fd);
event: 告诉内核需要监听的事件
struct epoll_event {
__uint32_t events; /* Epoll events */
epoll_data_t data; /* User data variable */
};
typedef union epoll_data {
void *ptr;
int fd;
uint32_t u32;
uint64_t u64;
} epoll_data_t;
EPOLLIN : 表示对应的文件描述符可以读(包括对端SOCKET正常关闭)
EPOLLOUT: 表示对应的文件描述符可以写
EPOLLPRI: 表示对应的文件描述符有紧急的数据可读(这里应该表示有带外数据到来)
EPOLLERR: 表示对应的文件描述符发生错误
EPOLLHUP: 表示对应的文件描述符被挂断;
EPOLLET: 将EPOLL设为边缘触发(Edge Triggered)模式,这是相对于水平触发(Level Triggered)而言的
EPOLLONESHOT:只监听一次事件,当监听完这次事件之后,如果还需要继续监听这个socket的话,需要再次把这个socket加入到EPOLL队列里
- 等待所监控文件描述符上有事件的产生,类似于select()调用。
#include <sys/epoll.h>
int epoll_wait(int epfd, struct epoll_event *events, int maxevents, int timeout)
events: 用来存内核得到事件的集合,
maxevents: 告之内核这个events有多大,这个maxevents的值不能大于创建epoll_create()时的size,
timeout: 是超时时间
-1:阻塞
0: 立即返回,非阻塞
> 0 :指定毫秒
返回值: 成功返回有多少文件描述符就绪,时间到时返回0,出错返回-1
代码
#include <sys/time.h>
#include <sys/types.h>
#include <unistd.h>
#include <sys/socket.h>
#include <ctype.h>
#include <arpa/inet.h> // sockaddr_in
#include <stdio.h>
#include <stdlib.h>
#include <strings.h>
#include <sys/epoll.h>
#include <errno.h>
#define PORT 6666
#define OPEN_MAX 1024
int main() {
int listen_fd = socket(AF_INET, SOCK_STREAM, 0);
struct sockaddr_in ser_addr;
bzero(&ser_addr, sizeof(ser_addr));
ser_addr.sin_family = AF_INET;
ser_addr.sin_port = htons(PORT);
ser_addr.sin_addr.s_addr = htonl(INADDR_ANY);
//端口复用
int opt = 1;
setsockopt(listen_fd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));
bind(listen_fd, (struct sockaddr *)&ser_addr, sizeof(ser_addr));
listen(listen_fd, 128);
struct epoll_event temp_event, epfd_event[OPEN_MAX];
temp_event.events = EPOLLIN;
temp_event.data.fd = listen_fd;
int epfd = epoll_create(OPEN_MAX); // root
epoll_ctl(epfd, EPOLL_CTL_ADD, listen_fd, &temp_event);
int nready;
int connect_fd;
struct sockaddr_in cli_addr;
bzero(&cli_addr, sizeof(cli_addr));
socklen_t cli_len = sizeof(cli_addr);
int socket_fd;
int n;
char buf[BUFSIZ];
while(1) {
nready = epoll_wait(epfd, epfd_event, OPEN_MAX, -1);
for (int i = 0; i < nready; i++) {
if(epfd_event[i].events == EPOLLIN) {
if (epfd_event[i].data.fd == listen_fd) { //监听新的客户端
connect_fd = accept(listen_fd,
(struct sockaddr*)&cli_addr,
&cli_len);
if (connect_fd < 0) {
printf("accept error!\n");
exit(1);
}
char ip[BUFSIZ];
printf("client ip: %s, port : %d\n", inet_ntop(AF_INET, &cli_addr.sin_addr.s_addr, ip, sizeof(ip)), ntohs(cli_addr.sin_port));
temp_event.events = EPOLLIN;
temp_event.data.fd = connect_fd;
epoll_ctl(epfd, EPOLL_CTL_ADD, //向树上新增节点
connect_fd, &temp_event);
} else { //数据传送
socket_fd = epfd_event[i].data.fd;
n = read(socket_fd, buf, sizeof(buf));
if (n < 0) {
printf("read error!\n");
} else if (n == 0) {
printf("cli[%d] is close!\n", i);
epoll_ctl(epfd, EPOLL_CTL_DEL,
socket_fd, NULL);
close(socket_fd);
} else {
for (int j = 0; j < n; j++) {
buf[j] = toupper(buf[j]);
}
write(socket_fd, buf, n);
write(STDOUT_FILENO, buf, n);
}
}
}
}
}
close(listen_fd);
return 0;
}