io多路复用例子c语言
时间: 2023-08-24 09:07:29 浏览: 104
下面是一个使用IO多路复用的简单例子,使用C语言编写:
```c
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#define MAX_CLIENTS 5
#define BUFFER_SIZE 1024
int main() {
int server_fd, new_socket, activity, i, valread, sd;
int max_sd;
struct sockaddr_in address;
char buffer[BUFFER_SIZE];
fd_set readfds;
// 创建服务器套接字
if ((server_fd = socket(AF_INET, SOCK_STREAM, 0)) == 0) {
perror("socket failed");
exit(EXIT_FAILURE);
}
// 设置服务器地址和端口
address.sin_family = AF_INET;
address.sin_addr.s_addr = INADDR_ANY;
address.sin_port = htons(8080);
// 将服务器套接字绑定到指定端口
if (bind(server_fd, (struct sockaddr *)&address, sizeof(address)) < 0) {
perror("bind failed");
exit(EXIT_FAILURE);
}
// 监听传入的连接
if (listen(server_fd, MAX_CLIENTS) < 0) {
perror("listen failed");
exit(EXIT_FAILURE);
}
printf("Waiting for connections...\n");
while (1) {
FD_ZERO(&readfds); // 清空文件描述符集合
FD_SET(server_fd, &readfds); // 添加服务器套接字到集合
max_sd = server_fd;
// 添加活跃的客户端套接字到集合
for (i = 0; i < MAX_CLIENTS; i++) {
sd = client_socket[i];
if (sd > 0)
FD_SET(sd, &readfds);
if (sd > max_sd)
max_sd = sd;
}
// 等待任何活跃的套接字准备就绪
activity = select(max_sd + 1, &readfds, NULL, NULL, NULL);
if ((activity < 0) && (errno != EINTR)) {
printf("select error");
}
// 如果服务器套接字准备就绪,表示有新的连接
if (FD_ISSET(server_fd, &readfds)) {
if ((new_socket = accept(server_fd, (struct sockaddr *)&address, (socklen_t*)&addrlen)) < 0) {
perror("accept failed");
exit(EXIT_FAILURE);
}
printf("New connection, socket fd is %d, IP is : %s, port : %d\n",
new_socket, inet_ntoa(address.sin_addr), ntohs(address.sin_port));
// 将新的套接字添加到客户端套接字数组
for (i = 0; i < MAX_CLIENTS; i++) {
if (client_socket[i] == 0) {
client_socket[i] = new_socket;
printf("Adding to list of sockets as %d\n", i);
break;
}
}
}
// 处理客户端发送的数据
for (i = 0; i < MAX_CLIENTS; i++) {
sd = client_socket[i];
if (FD_ISSET(sd, &readfds)) {
if ((valread = read(sd, buffer, BUFFER_SIZE)) == 0) {
// 客户端关闭连接
getpeername(sd, (struct sockaddr*)&address, \
(socklen_t*)&addrlen);
printf("Host disconnected, ip %s, port %d\n",
inet_ntoa(address.sin_addr), ntohs(address.sin_port));
close(sd);
client_socket[i] = 0;
} else {
// 发送回复给客户端
buffer[valread] = '\0';
send(sd, buffer, strlen(buffer), 0);
}
}
}
}
return 0;
}
```
这个例子展示了如何使用`select`函数来实现IO多路复用。服务器在循环中等待连接和处理客户端发送的数据。当有新的连接时,会将新的套接字添加到客户端套接字数组中,并打印连接的信息。当客户端发送数据时,服务器会将数据回复给客户端。当客户端关闭连接时,服务器会关闭套接字并从客户端套接字数组中移除。
请注意,这只是一个简单的示例,实际应用中可能需要更复杂的逻辑来处理不同的事件和错误。此外,代码中省略了一些错误处理和必要的变量声明,请根据实际需要进行修改和完善。
阅读全文