656565

#include <stdio.h>

#include <stdlib.h>

#include <string.h>

#include <unistd.h>

#include <arpa/inet.h>

#define BUFFER_SIZE 1024

#define EXIT_MESSAGE "exit"

void error(const char *message) {

    perror(message);

    exit(1);

}

int main(int argc, char *argv[]) {

    if (argc != 2) {

        fprintf(stderr, "Usage: %s <port>\n", argv[0]);

        exit(1);

    }

    int server_socket, client_socket;

    char buffer[BUFFER_SIZE];

    struct sockaddr_in server_addr, client_addr;

    socklen_t client_addr_len;

    // 创建服务器套接字

    server_socket = socket(AF_INET, SOCK_STREAM, 0);

    if (server_socket < 0) {

        error("Error opening socket");

    }

    // 设置服务器地址结构

    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(atoi(argv[1]));

    // 绑定服务器套接字到指定端口

    if (bind(server_socket, (struct sockaddr *)&server_addr, sizeof(server_addr)) < 0) {

        error("Error binding socket");

    }

    // 监听连接请求

    if (listen(server_socket, 5) < 0) {

        error("Error listening");

    }

    printf("Waiting for connections...\n");

    int client_count = 1;

    while (1) {

        // 接受客户端连接

        client_addr_len = sizeof(client_addr);

        client_socket = accept(server_socket, (struct sockaddr *)&client_addr, &client_addr_len);

        if (client_socket < 0) {

            error("Error accepting connection");

        }

        printf("Client %d connected\n", client_count);

        // 创建子进程来处理客户端通信

        pid_t pid = fork();

        if (pid < 0) {

            error("Error creating child process");

        }

        if (pid == 0) {

            // 子进程中处理与客户端的通信

            close(server_socket);  // 子进程不需要监听连接请求的套接字

            // 接收和发送消息，直到收到退出消息

            while (1) {

                // 接收客户端消息

                memset(buffer, 0, sizeof(buffer));

                if (recv(client_socket, buffer, sizeof(buffer), 0) < 0) {

                    error("Error receiving data");

                }

                printf("Client %d: %s\n", client_count, buffer);

                // 去除换行符

                buffer[strcspn(buffer, "\n")] = '\0';

                // 检查退出消息

                if (strcmp(buffer, EXIT_MESSAGE) == 0) {

                    printf("Client %d exited\n", client_count);

                    break;

                }

                // 发送消息给客户端

                printf("Server: ");

                fgets(buffer, sizeof(buffer), stdin);

                // 去除换行符

                buffer[strcspn(buffer, "\n")] = '\0';

                // 检查退出消息

                if (strcmp(buffer, EXIT_MESSAGE) == 0) {

                    if (send(client_socket, buffer, strlen(buffer), 0) < 0) {

                        error("Error sending data");

                    }

                    printf("You exited\n");

                    break;

                }

                if (send(client_socket, buffer, strlen(buffer), 0) < 0) {

                    error("Error sending data");

                }

            }

            // 关闭客户端套接字

            close(client_socket);

            exit(0);  // 子进程退出

        } else {

            // 父进程中继续监听连接请求

            close(client_socket);  // 父进程不需要与客户端通信的套接字

            client_count++;

        }

    }

    // 关闭服务器套接字

    close(server_socket);

    return 0;

}