服务端:
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <fcntl.h>
#include <errno.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <sys/epoll.h>
#define PORT 8080
#define MAX_EVENTS 10
#define BUFFER_SIZE 1024
// 设置非阻塞套接字
int set_non_blocking(int sockfd) {
int flags = fcntl(sockfd, F_GETFL, 0);
if (flags == -1) {
perror("fcntl get");
return -1;
}
if (fcntl(sockfd, F_SETFL, flags | O_NONBLOCK) == -1) {
perror("fcntl set");
return -1;
}
return 0;
}
int main() {
int listen_sock, conn_sock, epoll_fd;
struct sockaddr_in server_addr, client_addr;
socklen_t addrlen = sizeof(client_addr);
struct epoll_event ev, events[MAX_EVENTS];
char buffer[BUFFER_SIZE];
// 创建监听套接字
if ((listen_sock = socket(AF_INET, SOCK_STREAM, 0)) == -1) {
perror("socket");
exit(EXIT_FAILURE);
}
// 设置地址结构
memset(&server_addr, 0, sizeof(server_addr));
server_addr.sin_family = AF_INET;
server_addr.sin_addr.s_addr = INADDR_ANY;
server_addr.sin_port = htons(PORT);
// 绑定套接字
if (bind(listen_sock, (struct sockaddr *)&server_addr, sizeof(server_addr)) == -1) {
perror("bind");
close(listen_sock);
exit(EXIT_FAILURE);
}
// 开始监听
if (listen(listen_sock, SOMAXCONN) == -1) {
perror("listen");
close(listen_sock);
exit(EXIT_FAILURE);
}
// 设置监听套接字为非阻塞
set_non_blocking(listen_sock);
// 创建 epoll 实例
epoll_fd = epoll_create1(0);
if (epoll_fd == -1) {
perror("epoll_create1");
close(listen_sock);
exit(EXIT_FAILURE);
}
// 将监听套接字添加到 epoll 中
ev.events = EPOLLIN;
ev.data.fd = listen_sock;
if (epoll_ctl(epoll_fd, EPOLL_CTL_ADD, listen_sock, &ev) == -1) {
perror("epoll_ctl: listen_sock");
close(epoll_fd);
close(listen_sock);
exit(EXIT_FAILURE);
}
printf("服务器正在运行,等待连接...\n");
// 主循环
while (1) {
int nfds = epoll_wait(epoll_fd, events, MAX_EVENTS, -1);
if (nfds == -1) {
perror("epoll_wait");
break;
}
for (int n = 0; n < nfds; n++) {
if (events[n].data.fd == listen_sock) {
// 有新的连接
conn_sock = accept(listen_sock, (struct sockaddr *)&client_addr, &addrlen);
if (conn_sock == -1) {
perror("accept");
continue;
}
printf("新连接:%d\n", conn_sock);
set_non_blocking(conn_sock);
ev.events = EPOLLIN;
ev.data.fd = conn_sock;
if (epoll_ctl(epoll_fd, EPOLL_CTL_ADD, conn_sock, &ev) == -1) {
perror("epoll_ctl: conn_sock");
close(conn_sock);
}
} else {
// 有数据可读
conn_sock = events[n].data.fd;
ssize_t count = read(conn_sock, buffer, BUFFER_SIZE);
if (count == -1) {
perror("read");
close(conn_sock);
} else if (count == 0) {
// 客户端关闭连接
printf("关闭连接:%d\n", conn_sock);
close(conn_sock);
} else {
// 回显数据
write(conn_sock, buffer, count);
}
}
}
}
close(epoll_fd);
close(listen_sock);
return 0;
}
客户端
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <errno.h>
#define SERVER_ADDR "127.0.0.1" // 服务器地址
#define PORT 8080 // 服务器端口
#define BUFFER_SIZE 1024 // 缓冲区大小
int main() {
int sockfd;
struct sockaddr_in server_addr;
char buffer[BUFFER_SIZE];
// 创建套接字
if ((sockfd = socket(AF_INET, SOCK_STREAM, 0)) < 0) {
perror("socket");
exit(EXIT_FAILURE);
}
// 设置服务器地址结构
memset(&server_addr, 0, sizeof(server_addr));
server_addr.sin_family = AF_INET;
server_addr.sin_port = htons(PORT);
// 将点分十进制的字符串 IP 地址转换为网络字节顺序的地址
if (inet_pton(AF_INET, SERVER_ADDR, &server_addr.sin_addr) <= 0) {
perror("inet_pton");
close(sockfd);
exit(EXIT_FAILURE);
}
// 连接到服务器
if (connect(sockfd, (struct sockaddr *)&server_addr, sizeof(server_addr)) < 0) {
perror("connect");
close(sockfd);
exit(EXIT_FAILURE);
}
printf("连接到服务器 %s:%d\n", SERVER_ADDR, PORT);
while (1) {
// 从标准输入读取数据
printf("请输入要发送的消息(输入 'exit' 退出):");
fgets(buffer, BUFFER_SIZE, stdin);
// 去除末尾换行符
buffer[strcspn(buffer, "\n")] = 0;
// 如果输入 "exit",则退出循环
if (strcmp(buffer, "exit") == 0) {
break;
}
// 发送消息到服务器
send(sockfd, buffer, strlen(buffer), 0);
// 接收服务器的回复
ssize_t count = recv(sockfd, buffer, BUFFER_SIZE, 0);
if (count <= 0) {
if (count == 0) {
printf("服务器关闭连接\n");
} else {
perror("recv");
}
break;
}
// 打印服务器返回的消息
buffer[count] = '\0'; // 确保以 NULL 结尾
printf("从服务器收到消息: %s\n", buffer);
}
// 关闭套接字
close(sockfd);
return 0;
}
编译运行即可