参考《TCP/IP网络编程第17章》
epoll函数
-
epoll函数的优点:
- 无需编写以监视状态变化为目的的针对所有文件描述符的循环语句;
- 调用对应于 select 函数的 epoll_wait 函数时无需每次传递监视对象信息。
-
epoll 函数的功能:
- epoll_create:创建保存 epoll 文件描述符的空间;
- epoll_ctl:向空间注册并注销文件描述符;
- epoll_wait:与 select 函数类似,等待文件描述符发生变化;
-
epoll_event 结构体:
- 用于保存事件的文件描述符结合;
-
epoll_event 的成员 events 中可以保存的常量及所指的事件类型:
- EPOLLIN:需要读取数据的情况
- EPOLLOUT:输出缓冲为空,可以立即发送数据的情况
- EPOLLPRI:收到 OOB 数据的情况
- EPOLLRDHUP:断开连接或半关闭的情况,这在边缘触发方式下非常有用
- EPOLLERR:发生错误的情况
实现基于epoll函数的服务器和客户端通信
-
服务器端代码
:
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <sys/socket.h>
#include <sys/unistd.h>
#include <sys/types.h>
#include <sys/errno.h>
#include <netinet/in.h>
#include <signal.h>
#include <sys/epoll.h>
#include <iostream>
#include <string>
using std::cout;
using std::cin;
using std::string;
using std::endl;
#define BUF_SIZE 1024
#define EPOLL_SIZE 50
#define SERVER_IP "127.0.0.1" // 指定服务端的IP
#define SERVER_PORT 9190 // 指定服务端的port
int main()
{
int serv_sock, clnt_sock;
struct sockaddr_in serv_adr, clnt_adr;
socklen_t adr_sz;
int str_len, i;
char buf[BUF_SIZE];
struct epoll_event *ep_events;
struct epoll_event event;
int epfd, event_cnt;
serv_sock = socket(PF_INET, SOCK_STREAM, 0);
bzero(&serv_adr, sizeof(serv_adr));
serv_adr.sin_family = AF_INET;
serv_adr.sin_addr.s_addr = htonl(INADDR_ANY);
serv_adr.sin_port = htons(SERVER_PORT);
if(bind(serv_sock, (struct sockaddr*)&serv_adr, sizeof(serv_adr)) == -1)
cout << "Bind error" << errno << ": " << strerror(errno) << endl;
if(listen(serv_sock, 5) == -1)
cout << "Listen error" << errno << ": " << strerror(errno) << endl;
epfd = epoll_create(EPOLL_SIZE);//可以忽略这个参数,填入的参数为操作系统参考
ep_events = (struct epoll_event*)malloc(sizeof(struct epoll_event) * EPOLL_SIZE);
event.events = EPOLLIN;//需要读取数据的情况
event.data.fd = serv_sock;
//例程epfd 中添加文件描述符 serv_sock,目的是监听 enevt 中的事件
epoll_ctl(epfd, EPOLL_CTL_ADD,serv_sock, &event);
while(1)
{
event_cnt = epoll_wait(epfd, ep_events, EPOLL_SIZE, - 1);//获取改变了的文件描述符,返回数量
if(event_cnt == -1) {
cout << "epoll_wait() error" << endl;
break;
}
for(i = 0; i < event_cnt; ++i)
{
if(ep_events[i].data.fd == serv_sock)//客户端请求连接时
{
adr_sz = sizeof(clnt_adr);
clnt_sock = accept(serv_sock, (struct sockaddr*)&clnt_adr, &adr_sz);
event.events = EPOLLIN;
event.data.fd = clnt_sock;//把客户端套接字添加进去
epoll_ctl(epfd, EPOLL_CTL_ADD, clnt_sock, &event);
cout << "connected client:" << clnt_sock << endl;
}
else //是客户端套接字时
{
str_len = read(ep_events[i].data.fd,buf, BUF_SIZE);
if(str_len == 0)
{
epoll_ctl(epfd, EPOLL_CTL_DEL, ep_events[i].data.fd, &event);
close(ep_events[i].data.fd);
cout << "closed client:" << ep_events[i].data.fd << endl;
}
else
{
write(ep_events[i].data.fd,buf, str_len);
}
}
}
}
close(serv_sock);
close(epfd);
return 0;
}
-
客户端代码
:
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <sys/socket.h>
#include <sys/unistd.h>
#include <sys/types.h>
#include <sys/errno.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <iostream>
#include <string>
using std::cout;
using std::cin;
using std::string;
using std::endl;
#define BUFFSIZE 1024
#define SERVER_IP "127.0.0.1" // 指定服务端的IP
#define SERVER_PORT 9190 // 指定服务端的port
int main()
{
int sock;
char message[BUFFSIZE];
int str_len;
struct sockaddr_in serv_adr;
sock = socket(PF_INET, SOCK_STREAM, 0);
if(sock == -1) cout << "socket() error" << endl;
bzero(&serv_adr, sizeof(serv_adr));
serv_adr.sin_family = AF_INET;
serv_adr.sin_addr.s_addr = inet_addr(SERVER_IP);
serv_adr.sin_port = htons(SERVER_PORT);
if(connect(sock, (struct sockaddr *)&serv_adr, sizeof(serv_adr)) == -1)
cout << "connect() error: " << errno << " " << strerror(errno) << endl;
else
cout << "connected............" << endl;
while(1)
{
cout << "Input message(Q to quit): " << endl;
cin >> message;
if(!strcmp(message, "q") || !strcmp(message, "Q"))
break;
write(sock, message, strlen(message));
str_len = read(sock, message, BUFFSIZE - 1);
message[str_len] = 0;
cout << "Message from server:" << message << endl;
}
close(sock);
return 0;
}
版权声明:本文为weixin_42659457原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。