作业5.26

定义一些常量和数据结构来存储用户信息和消息。 

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <sys/socket.h>#define MAX_USERS 100 // 最大用户数
#define MAX_MESSAGE_LENGTH 256 // 最大消息长度
#define SERVER_PORT 8888 // 服务器端口// 用户信息结构体
typedef struct {int id;struct sockaddr_in addr;int is_online;
} User;// 消息结构体
typedef struct {User user;char message[MAX_MESSAGE_LENGTH];
} Message;// 用户数组
User users[MAX_USERS];
int user_count = 0;// 添加用户
void add_user(User user) {users[user_count++] = user;
}// 移除用户
void remove_user(int id) {int i;for (i = 0; i < user_count; i++) {if (users[i].id == id) {users[i].is_online = 0;break;}}
}// 发送消息给所有用户（除了发送者）
void send_message_to_all(Message msg, int sender_id) {int i;for (i = 0; i < user_count; i++) {if (users[i].id != sender_id && users[i].is_online) {sendto(sockfd, &msg, sizeof(msg), 0, (struct sockaddr*)&users[i].addr, sizeof(users[i].addr));}}
}

 服务器端的代码：

int main() {int sockfd;struct sockaddr_in server_addr, client_addr;socklen_t addr_len = sizeof(client_addr);Message msg;// 创建UDP socketif ((sockfd = socket(AF_INET, SOCK_DGRAM, 0)) == -1) {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 = INADDR_ANY;server_addr.sin_port = htons(SERVER_PORT);// 绑定socket到服务器地址和端口if (bind(sockfd, (struct sockaddr*)&server_addr, sizeof(server_addr)) == -1) {perror("bind failed");exit(EXIT_FAILURE);}printf("Server is listening on port %d\n", SERVER_PORT);// 主循环接收消息while (1) {// 接收客户端消息if (recvfrom(sockfd, &msg, sizeof(msg), 0, (struct sockaddr*)&client_addr, &addr_len) == -1) {perror("recvfrom failed");continue;}// 处理消息// ...// 发送消息给所有用户（除了发送者）send_message_to_all(msg, msg.user.id);}close(sockfd);return 0;
}

 客户端的代码：

int main() {int sockfd;struct sockaddr_in server_addr;Message msg;// 创建UDP socketif ((sockfd = socket(AF_INET, SOCK_DGRAM, 0)) == -1) {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 = INADDR_ANY;server_addr.sin_port = htons(SERVER_PORT);// 用户登录，发送登录消息给服务器// ...// 创建接收消息的线程// ...// 主循环发送消息while (1) {// 读取用户输入的消息// ...// 发送消息给服务器sendto(sockfd, &msg, sizeof(msg), 0, (struct sockaddr*)&server_addr, sizeof(server_addr));}close(sockfd);return 0;
}