Многоклиентный сервер: выход из функции - PullRequest
Новая
       22

Многоклиентный сервер: выход из функции

0 голосов
/ 04 мая 2018

У меня проблемы с сервером / клиентом. Если несколько человек подключаются к моему серверу, и один из них завершает соединение с сервером, все остальные, которые также подключены к серверу, теряют соединение с. 

void *connection_handler(void *);
extern int check_err_buffer(int sock);
extern error_t *error_buffer;
extern invalid_user_input_t invalid_user_input;
regex_t regex;

int main(int argc, char *argv[]) {
    //init KVS
    init();

    int socket_desc, client_sock, c, *new_sock;
    struct sockaddr_in server, client;

    //Create socket
    socket_desc = socket(AF_INET, SOCK_STREAM | O_NONBLOCK, 0);
    int opts;
    opts = fcntl(socket_desc,F_GETFL);
    opts = opts & (~O_NONBLOCK);
    fcntl(socket_desc,F_SETFL,opts);

    if (socket_desc == -1) {
        printf("Could not create socket");
    }
    puts("Socket created");

    //Prepare the sockaddr_in structure
    server.sin_family = AF_INET;
    server.sin_addr.s_addr = INADDR_ANY;
    server.sin_port = htons(8503);

    //Bind
    if (bind(socket_desc, (struct sockaddr *) &server, sizeof(server)) < 0) {
        //print the error message
        perror("bind failed. Error");
        return 1;
    }
    puts("bind done");

    //Listen
    listen(socket_desc, 3);

    //Accept and incoming connection
    puts("Waiting for incoming connections...");
    c = sizeof(struct sockaddr_in);//Accept and incoming connection
    puts("Waiting for incoming connections...");
    c = sizeof(struct sockaddr_in);
    while (1) {
        puts("waiting for client...");
        client_sock = accept(socket_desc, (struct sockaddr *) &client, (socklen_t * ) & c);
        puts("Connection accepted");
        pthread_t sniffer_thread;
        new_sock = malloc(1);
        *new_sock = client_sock;

        pthread_create(&sniffer_thread, NULL, connection_handler, (void *) new_sock);


        //Now join the thread , so that we dont terminate before the thread
        //pthread_join(sniffer_thread, NULL);
        puts("Handler assigned");
    }

    if (client_sock < 0) {
        perror("accept failed");
        return 1;
    }

    return 0;
}

В основной функции я принимаю разные подключения от клиентов. У меня есть больше функций, которые я не вставил, потому что они не так актуальны для этой проблемы. Во второй части я рассмотрю некоторые случаи. Я смотрю, если клиент введет букву 'q', если это правда, я закрою соединение с close(). Но во всех случаях он закрывает сервер, и все соединения теряются. 

/*
* This will handle connection for each client
* */
void *connection_handler(void *socket_desc) {
    //Get the socket descriptor
    int sock = *(int *) socket_desc;
    char *message, client_message[3000];
    puts("sending welcome msg to client...");
    write(sock, message, 3000);

    int reti;
    //compile a regexp that checks for valid user input
    regcomp(&regex, "([dg] [0-9]+)|([v] .+)|(p [0-9]+ .+)|q", REG_EXTENDED);
    //Receive a message from client
    char *input[3];
    while(1){
        //wait for some client input
        recv(sock, client_message, 2000, 0);
        //check the input with the regexp
        reti = regexec(&regex, client_message, 0, NULL, REG_EXTENDED);
        if (!reti) {
            //tokenize the client input with \n as a delim char
            char *token = strtok(client_message, " ");
            int i = 0;
            while (token) {
                input[i] = token;
                token = strtok(NULL, " ");
                i++;
            }
                //quit operation
            if (strcmp(input[0], "q") == 0) {
                //if a client quits, the server dies. TODO: fix
                puts("quitting");
                //free(socket_desc);
                close(socket_desc);
                return 0;
            }
            //user input does not match regexp, try again..
        } else if (reti == REG_NOMATCH) {
            error_buffer = invalid_user_input;
            check_err_buffer(sock);
            usleep(100);
            write(sock, " ", 2);
        }
    }
    //Free the socket pointer
    free(socket_desc);

    return 0;
}

На стороне клиента я проверяю, если message[0] == 'q', и закрываю сокет. Я был бы очень рад, если бы я получил помощь или подсказки, чтобы решить эту проблему. Спасибо

1 Ответ

0 голосов
/ 05 мая 2018

Как здесь 

      new_sock = malloc(1);

код выделяет только 1 байт, следующая строка 

      *new_sock = client_sock;

записывает в недействительную память, так как назначено int, а sizeof int байтов записывается туда, куда указывает new_sock. По определению sizeof int равно по крайней мере 2, возможно, больше, при этом вызывается это неопределенное поведение. С этого момента может случиться что угодно.

Чтобы это исправить, сделайте 

      new_sock = malloc(sizeof *new_sock);

вместо.

Проверка возвращаемого значения функций является отладкой бесплатно. close(socket_desc); - нонсенс, и он обязательно вернет -1 и установит errno в EBADFD.

Заменить 

    close(socket_desc);

от 

    if (-1 == close(socket_desc))
    {
      perror("close() falied");
    }

чтобы увидеть, как код не работает.

...