C++ select函数的使用
时间: 2024-05-10 16:19:19 浏览: 153
`select()`函数是Unix/Linux网络编程中常用的一个函数,用于多路复用I/O操作,可以同时监听多个文件描述符的可读、可写等事件,并在事件触发时通知应用程序进行相应的处理。以下是`select()`函数的使用步骤:
1. 创建一个`fd_set`类型的变量,用于存储需要监听的文件描述符。
2. 将需要监听的文件描述符添加到`fd_set`变量中,使用`FD_SET()`宏函数实现。
3. 设置`select()`函数的超时时间,或者将其设置为阻塞模式等待事件的发生。
4. 调用`select()`函数开始监听,当有文件描述符满足条件时,`select()`函数返回。
5. 使用`FD_ISSET()`宏函数检查哪些文件描述符满足条件,并进行相应的处理。
下面是一个简单的示例代码,通过`select()`函数监听标准输入和socket连接,并处理可读事件。
```c++
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <sys/time.h>
#include <sys/types.h>
#define MAX_CLIENTS 10
#define BUFFER_SIZE 1024
int main(int argc, char *argv[]) {
int server_fd, client_fd[MAX_CLIENTS];
struct sockaddr_in server_addr, client_addr;
socklen_t client_len;
char buffer[BUFFER_SIZE];
fd_set read_fds;
int max_fd, activity, i, valread, sd;
// 创建server socket
server_fd = socket(AF_INET, SOCK_STREAM, 0);
if (server_fd < 0) {
perror("socket failed");
exit(EXIT_FAILURE);
}
// 绑定地址和端口
server_addr.sin_family = AF_INET;
server_addr.sin_addr.s_addr = INADDR_ANY;
server_addr.sin_port = htons(8888);
if (bind(server_fd, (struct sockaddr*)&server_addr, sizeof(server_addr)) < 0) {
perror("bind failed");
exit(EXIT_FAILURE);
}
// 监听连接
if (listen(server_fd, MAX_CLIENTS) < 0) {
perror("listen failed");
exit(EXIT_FAILURE);
}
// 初始化client_fd数组
for (i = 0; i < MAX_CLIENTS; i++) {
client_fd[i] = 0;
}
while (1) {
// 清空文件描述符集合
FD_ZERO(&read_fds);
// 添加标准输入和server socket到集合中
FD_SET(STDIN_FILENO, &read_fds);
FD_SET(server_fd, &read_fds);
max_fd = server_fd;
// 添加所有已连接的client socket到集合中
for (i = 0; i < MAX_CLIENTS; i++) {
sd = client_fd[i];
if (sd > 0) {
FD_SET(sd, &read_fds);
}
if (sd > max_fd) {
max_fd = sd;
}
}
// 等待事件发生
activity = select(max_fd + 1, &read_fds, NULL, NULL, NULL);
if (activity < 0) {
perror("select error");
continue;
}
// 如果标准输入有数据可读
if (FD_ISSET(STDIN_FILENO, &read_fds)) {
fgets(buffer, BUFFER_SIZE, stdin);
printf("stdin data: %s\n", buffer);
}
// 如果server socket有新连接请求
if (FD_ISSET(server_fd, &read_fds)) {
client_len = sizeof(client_addr);
client_fd[i] = accept(server_fd, (struct sockaddr*)&client_addr, &client_len);
if (client_fd[i] < 0) {
perror("accept failed");
continue;
}
printf("New connection, socket fd: %d, ip: %s, port: %d\n", client_fd[i],
inet_ntoa(client_addr.sin_addr), ntohs(client_addr.sin_port));
}
// 如果有client socket有数据可读
for (i = 0; i < MAX_CLIENTS; i++) {
sd = client_fd[i];
if (FD_ISSET(sd, &read_fds)) {
valread = read(sd, buffer, BUFFER_SIZE);
if (valread == 0) {
// client关闭连接
printf("Client disconnected, socket fd: %d\n", sd);
close(sd);
client_fd[i] = 0;
} else {
// 处理client数据
printf("Data received: %s\n", buffer);
}
}
}
}
return 0;
}
```
阅读全文