epoll 客户端 收发
时间: 2023-07-27 08:32:49 浏览: 96
好的,你的问题是关于epoll客户端的收发,我会尽力回答。以下是一个基于epoll的客户端程序,可以实现收发功能:
```c
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <sys/epoll.h>
#define MAX_EVENTS 10
#define MAX_BUFFER_SIZE 1024
int main(int argc, char* argv[]) {
if (argc != 3) {
printf("usage: %s ip port\n", argv[0]);
exit(1);
}
char* ip = argv[1];
int port = atoi(argv[2]);
// 创建socket
int sockfd = socket(AF_INET, SOCK_STREAM, 0);
if (sockfd < 0) {
perror("socket error");
exit(1);
}
// 连接服务器
struct sockaddr_in server_addr;
memset(&server_addr, 0, sizeof(server_addr));
server_addr.sin_family = AF_INET;
server_addr.sin_addr.s_addr = inet_addr(ip);
server_addr.sin_port = htons(port);
if (connect(sockfd, (struct sockaddr*)&server_addr, sizeof(server_addr)) < 0) {
perror("connect error");
exit(1);
}
// 创建epoll句柄
int epfd = epoll_create(MAX_EVENTS);
if (epfd < 0) {
perror("epoll_create error");
exit(1);
}
// 注册sockfd到epoll句柄中
struct epoll_event ev;
ev.events = EPOLLIN | EPOLLET;
ev.data.fd = sockfd;
if (epoll_ctl(epfd, EPOLL_CTL_ADD, sockfd, &ev) < 0) {
perror("epoll_ctl error");
exit(1);
}
// 创建缓冲区
char buffer[MAX_BUFFER_SIZE];
while (1) {
// 等待事件的发生
struct epoll_event events[MAX_EVENTS];
int n = epoll_wait(epfd, events, MAX_EVENTS, -1);
if (n < 0) {
perror("epoll_wait error");
exit(1);
}
// 处理发生的事件
for (int i = 0; i < n; i++) {
if (events[i].data.fd == sockfd) {
if (events[i].events & EPOLLIN) {
// 接收数据
int len = recv(sockfd, buffer, MAX_BUFFER_SIZE, 0);
if (len < 0) {
perror("recv error");
exit(1);
}
else if (len == 0) {
printf("server closed\n");
exit(1);
}
else {
buffer[len] = '\0';
printf("received: %s", buffer);
}
}
else if (events[i].events & EPOLLOUT) {
// 发送数据
printf("input message: ");
fgets(buffer, MAX_BUFFER_SIZE, stdin);
int len = send(sockfd, buffer, strlen(buffer), 0);
if (len < 0) {
perror("send error");
exit(1);
}
}
}
}
}
// 关闭socket
close(sockfd);
return 0;
}
```
该程序使用了epoll实现IO多路复用,可以同时处理接收和发送数据的操作。在程序中,先创建了一个socket并连接到服务器,然后创建了一个epoll句柄,并将socket注册到epoll句柄中。接下来进入循环,等待事件的发生,并处理发生的事件。当socket可读时,接收数据并输出;当socket可写时,从标准输入中读取数据并发送。当收到服务器关闭的消息时,程序退出。
阅读全文