再玩下linux中的的epoll函数------可以用任何描述符号fd而不限于网络socket

972 篇文章 329 订阅
322 篇文章 11 订阅

      之前玩过epoll,  再来看下, 其实也很简单:

#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netdb.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <ctype.h>
#include <errno.h>
#include <malloc.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <sys/ioctl.h>
#include <stdarg.h>
#include <fcntl.h>
#include <fcntl.h>
#include <sys/poll.h>
#include <sys/epoll.h>

#define NUM 1
#define MAX 100

int main()  
{
	epoll_event ev;
	ev.data.fd = STDIN_FILENO;
	ev.events = EPOLLIN;
	int epollFD = epoll_create(MAX);                         // 告诉内核监测的数目的最大值, 返回的epollFD为epoll管理句柄
    epoll_ctl(epollFD, EPOLL_CTL_ADD, STDIN_FILENO, &ev);    // 将ev和对应的STDIN_FILENO添加到epoll句柄,用于被epollFD管理

    epoll_event all_events[MAX]; // "保活"的句柄直接从这里取
    while(1) 
	{
		int timeoutMS = 5000;    // 5000ms

		// epoll_wait和select/poll函数的作用类似
        int nfds = epoll_wait(epollFD, all_events, NUM, timeoutMS);   // all_events和nfds是输出值, nfds是"保活"的数据
		if(nfds < 0)
		{
			printf("epoll_wait error\n");
			break;
		}

		if(nfds == 0)
		{
			printf("epoll_wait timeout\n");
			continue;
		}
		
		printf("nfds is %d\n", nfds);
		
        for(int i = 0; i < nfds; i++)  // 此处不用NUM, 而用nfds, 而all_events[i]渠道的就是"保活"的信息
		{
			int iFD = all_events[i].data.fd; 
			printf("iFD is %d\n", iFD);
			
			char szBuf[1024] = {0};
			int recvLen = read(iFD, szBuf, sizeof(szBuf) - 1);
			if (recvLen > 0) 
			{  
				printf("recv data [%s]\n", szBuf);
			}
			else
			{
				epoll_ctl(epollFD, EPOLL_CTL_DEL, STDIN_FILENO, &ev);
				printf("recv error\n");
			}
		}
    }  
	
    close(epollFD);
    return 0;
}  
        建议有兴趣的同事实际玩一下上述程序。



  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
好的,以下是一个简单的 epoll 和 signal 的服务器-客户端代码示例: 服务器端代码: ```C++ #include <iostream> #include <cstring> #include <cstdlib> #include <unistd.h> #include <signal.h> #include <arpa/inet.h> #include <sys/types.h> #include <sys/socket.h> #include <sys/epoll.h> #include <netinet/in.h> #define MAX_EVENTS 10 #define BUF_SIZE 256 using namespace std; int main(int argc, char ** argv) { if (argc != 2) { cout << "Usage: " << argv[0] << " <port>" << endl; return 1; } int server_port = atoi(argv[1]); int server_sockfd; struct sockaddr_in server_addr; // 创建 Socket server_sockfd = socket(AF_INET, SOCK_STREAM, 0); // 设置地址重用 int reuse = 1; setsockopt(server_sockfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)); // 设置地址和端口 memset(&server_addr, 0, sizeof(server_addr)); server_addr.sin_family = AF_INET; server_addr.sin_addr.s_addr = htonl(INADDR_ANY); server_addr.sin_port = htons(server_port); // 绑定 Socket bind(server_sockfd, (struct sockaddr *)&server_addr, sizeof(server_addr)); // 监听 Socket listen(server_sockfd, 10); int epoll_fd, nfds; struct epoll_event ev, events[MAX_EVENTS]; char buffer[BUF_SIZE]; // 创建 epoll epoll_fd = epoll_create(10); // 添加 server_sockfdepoll ev.events = EPOLLIN; ev.data.fd = server_sockfd; epoll_ctl(epoll_fd, EPOLL_CTL_ADD, server_sockfd, &ev); while (1) { // 等待事件发生 nfds = epoll_wait(epoll_fd, events, MAX_EVENTS, -1); if (nfds == -1) { perror("epoll_wait"); break; } // 处理事件 for (int i = 0; i < nfds; i++) { if (events[i].data.fd == server_sockfd) { // 接受客户端连接请求 int client_sockfd; struct sockaddr_in client_addr; socklen_t client_len = sizeof(client_addr); client_sockfd = accept(server_sockfd, (struct sockaddr *)&client_addr, &client_len); // 添加 client_sockfdepoll ev.events = EPOLLIN | EPOLLRDHUP | EPOLLET; ev.data.fd = client_sockfd; epoll_ctl(epoll_fd, EPOLL_CTL_ADD, client_sockfd, &ev); } else if (events[i].events & EPOLLIN) { // 处理读事件 int sockfd = events[i].data.fd; int n = read(sockfd, buffer, BUF_SIZE); if (n == 0) { // 客户端关闭连接 close(sockfd); epoll_ctl(epoll_fd, EPOLL_CTL_DEL, sockfd, &ev); } else if (n > 0) { // 发送数据到客户端 write(sockfd, buffer, n); } else { perror("read"); } } else if (events[i].events & EPOLLRDHUP) { // 处理客户端关闭连接事件 int sockfd = events[i].data.fd; close(sockfd); epoll_ctl(epoll_fd, EPOLL_CTL_DEL, sockfd, &ev); } else { // 处理其他事件 } } } // 关闭 Socket close(server_sockfd); return 0; } ``` 客户端代码: ```C++ #include <iostream> #include <cstring> #include <cstdlib> #include <unistd.h> #include <arpa/inet.h> #include <sys/types.h> #include <sys/socket.h> #include <signal.h> #define BUF_SIZE 256 using namespace std; int running = 1; void sig_handle(int signo) { if (signo == SIGINT) { cout << "Terminating client..." << endl; running = 0; } } int main(int argc, char ** argv) { if (argc != 3) { cout << "Usage: " << argv[0] << " <server_ip> <server_port>" << endl; return 1; } int server_port = atoi(argv[2]); int client_sockfd; struct sockaddr_in server_addr; char buffer[BUF_SIZE]; // 创建 Socket client_sockfd = socket(AF_INET, SOCK_STREAM, 0); // 设置地址和端口 memset(&server_addr, 0, sizeof(server_addr)); server_addr.sin_family = AF_INET; server_addr.sin_addr.s_addr = inet_addr(argv[1]); server_addr.sin_port = htons(server_port); // 连接到服务器 connect(client_sockfd, (struct sockaddr *)&server_addr, sizeof(server_addr)); // 注册信处理函数 signal(SIGINT, sig_handle); // 循环读取用户输入并发送到服务器 while (running) { memset(buffer, 0, BUF_SIZE); fgets(buffer, BUF_SIZE, stdin); int n = write(client_sockfd, buffer, strlen(buffer)); if (n < 0) { perror("write"); break; } n = read(client_sockfd, buffer, BUF_SIZE); if (n < 0) { perror("read"); break; } buffer[n] = '\0'; cout << "Server reply:" << buffer; } // 关闭 Socket close(client_sockfd); return 0; } ``` 希望这个代码示例能对您有所帮助。如果您有任何疑问,请随时问我。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值