网络编程---多客户端服务器

写一个服务器和两个客户端

运行服务器和2个客户端，实现聊天功能

客户端1 和 客户端2 进行聊天

客户端1将聊天数据发送给服务器

服务器将聊天数据转发给客户端2

要求： 服务器使用 select 模型实现 客户端1使用 poll 模型实现 客户端2使用 多线程实现

服务器：

1 
2 #include "head.h"
3 
4 #define PORT 12345
5 #define MAX_CLIENTS 10
6 #define BUFFER_SIZE 1024
7 
8 void handle_client_message(int client_sock, int *clients, int num_clients) {
9     char buffer[BUFFER_SIZE];
0     int bytes_received = recv(client_sock, buffer, sizeof(buffer), 0);
1     if (bytes_received <= 0) {
2         close(client_sock);
3         for (int i = 0; i < num_clients; i++) {
4             if (clients[i] == client_sock) {
5                 clients[i] = clients[num_clients - 1];
6                 break;
7             }
8         }
9         return;
0     }
1     buffer[bytes_received] = '\0';
2     printf("Received message: %s\n", buffer);
3 
4     // 广播消息给所有其他客户端
5     for (int i = 0; i < num_clients; i++) {
6         if (clients[i] != client_sock) {
7             send(clients[i], buffer, strlen(buffer), 0);
8         }
9     }
0 }
1 
2 void server() {
3     int server_sock, client_sock, max_fd;
4     struct sockaddr_in server_addr, client_addr;
5     socklen_t client_len = sizeof(client_addr);
6     fd_set read_fds, temp_fds;
7     int clients[MAX_CLIENTS];
8     int num_clients = 0;
9 
0     // 创建服务器socket
1     server_sock = socket(AF_INET, SOCK_STREAM, 0);
2     if (server_sock < 0) {
3         perror("Socket creation failed");
4         exit(EXIT_FAILURE);
5     }
6 
7     // 绑定服务器地址
8     memset(&server_addr, 0, sizeof(server_addr));
9     server_addr.sin_family = AF_INET;
0     server_addr.sin_addr.s_addr = INADDR_ANY;
1     server_addr.sin_port = htons(PORT);
2 
3     if (bind(server_sock, (struct sockaddr *)&server_addr, sizeof(server_addr)) < 0) {
4         perror("Binding failed");
5         exit(EXIT_FAILURE);
6     }
7 
8     // 监听客户端连接
9     listen(server_sock, 5);
0     FD_ZERO(&read_fds);
1     FD_SET(server_sock, &read_fds);
2     max_fd = server_sock;
3 
4     printf("Server listening on port %d...\n", PORT);
5 
6     while (1) {
7         temp_fds = read_fds;
8 
9         // 使用select等待活动的socket
0         if (select(max_fd + 1, &temp_fds, NULL, NULL, NULL) < 0) {
1             perror("Select error");
2             exit(EXIT_FAILURE);
3         }
4 
5         // 检查所有客户端连接
6         for (int i = 0; i <= max_fd; i++) {
7             if (FD_ISSET(i, &temp_fds)) {
8                 if (i == server_sock) {
9                     // 有新的客户端连接
0                     client_sock = accept(server_sock, (struct sockaddr *)&client_addr, &client_len);
1                     if (client_sock < 0) {
2                         perror("Accept failed");
3                         continue;
4                     }
5                     printf("New connection: %d\n", client_sock);
6 
7                     FD_SET(client_sock, &read_fds);
8                     clients[num_clients++] = client_sock;
9                     if (client_sock > max_fd) {
0                         max_fd = client_sock;
1                     }
2                 } else {
3                     // 处理现有客户端的消息
4                     handle_client_message(i, clients, num_clients);
5                 }
6             }
7         }
8     }
9                                                                                                                                   
0     close(server_sock);
1 }
2 
3 int main() {
4     server();
5     return 0;
6 }
7 

客户端1：

#include "head.h"#define PORT 12345
#define BUFFER_SIZE 1024void client1() {int sock;struct sockaddr_in server_addr;struct pollfd fds[2];char buffer[BUFFER_SIZE];// 创建客户端socketsock = socket(AF_INET, SOCK_STREAM, 0);if (sock < 0) {perror("Socket creation failed");exit(EXIT_FAILURE);}// 设置服务器地址memset(&server_addr, 0, sizeof(server_addr));server_addr.sin_family = AF_INET;server_addr.sin_addr.s_addr = inet_addr("127.0.0.1");server_addr.sin_port = htons(PORT);// 连接到服务器if (connect(sock, (struct sockaddr *)&server_addr, sizeof(server_addr)) < 0) {perror("Connection failed");exit(EXIT_FAILURE);}printf("Connected to server. Type your messages.\n");// 设置poll结构fds[0].fd = sock; // 服务器socketfds[0].events = POLLIN;fds[1].fd = STDIN_FILENO; // 标准输入fds[1].events = POLLIN;while (1) {// 使用poll模型int ret = poll(fds, 2, -1);if (ret < 0) {perror("Poll failed");exit(EXIT_FAILURE);}if (fds[0].revents & POLLIN) {// 服务器发送消息int n = recv(sock, buffer, sizeof(buffer), 0);if (n <= 0) {printf("Server disconnected\n");break;}buffer[n] = '\0';printf("Server: %s\n", buffer);}if (fds[1].revents & POLLIN) {// 用户输入消息if (fgets(buffer, sizeof(buffer), stdin) != NULL) {send(sock, buffer, strlen(buffer), 0);}}}close(sock);
}int main() {                                                                                                      client1();return 0;
}

客户端2：

#include "head.h"#define PORT 12345
#define BUFFER_SIZE 1024void *receive_messages(void *socket_desc) {int sock = *(int *)socket_desc;char buffer[BUFFER_SIZE];while (1) {int n = recv(sock, buffer, sizeof(buffer), 0);if (n <= 0) {printf("Server disconnected\n");break;}buffer[n] = '\0';printf("Server: %s\n", buffer);}return NULL;
}void client2() {int sock;struct sockaddr_in server_addr;pthread_t thread_id;char buffer[BUFFER_SIZE];// 创建客户端socketsock = socket(AF_INET, SOCK_STREAM, 0);if (sock < 0) {perror("Socket creation failed");exit(EXIT_FAILURE);}// 设置服务器地址memset(&server_addr, 0, sizeof(server_addr));server_addr.sin_family = AF_INET;server_addr.sin_addr.s_addr = inet_addr("127.0.0.1");server_addr.sin_port = htons(PORT);// 连接到服务器if (connect(sock, (struct sockaddr *)&server_addr, sizeof(server_addr)) < 0) {perror("Connection failed");exit(EXIT_FAILURE);}printf("Connected to server. Type your messages.\n");// 创建接收消息线程if (pthread_create(&thread_id, NULL, receive_messages, (void *)&sock) < 0) {perror("Could not create thread");exit(EXIT_FAILURE);}// 发送消息while (1) {fgets(buffer, sizeof(buffer), stdin);send(sock, buffer, strlen(buffer), 0);}close(sock);
}int main() {client2();return 0;}

结果：