server.c

#include <sys/socket.h>
#include <sys/types.h>
#include <errno.h>
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <stdlib.h>
#include <poll.h>
#include "utils.h"
#include <signal.h>
#include "cmd.h"

#define BUF_SIZE 256

poll_fds fds;
clients clients_base;

void int_signal()
{
    int i;
    char s[] = "### Сервер закрывается\n";
    mass_send(fds, s, sizeof(s));
    for(i = 0; i < get_fds_size(); i++)
    {
        if(fds[i].fd != -1)
            close(fds[i].fd);
    }
    cleanup(fds, clients_base);
    exit(0);
}

int main(int argc, char * argv[])
{
    int main_socket, port, events, temp_socket, i;
    ssize_t n_read;
    char buf[BUF_SIZE + 1];
    char *msg;

    init_signals();

    signal(SIGINT, int_signal);
    signal(SIGSTOP, int_signal);
    signal(SIGTERM, int_signal);

    if(argc < 2)
    {
        fprintf(stderr,"Необходимо указать номер порта в параметрах\n");
        return 1;
    }

    port = atoi(argv[1]);

    set_pswrd();

    main_socket = init_socket(port);

    fds = init_fds();
    clients_base = init_clients();
    ban_init();

    if(fds == NULL)
    {
        fprintf(stderr, "%s (%d): Структура не была создана: %s\n",
                __FILE__, __LINE__ - 3,  strerror(errno));
        exit(1);
    }
    fds = add_fds(fds, main_socket);

    for(;;)
    {
        events = poll(fds, get_fds_size(), 100);
        if(events == -1)
        {
            fprintf(stderr, "%s (%d): Проблемы с poll: %s\n",
                    __FILE__, __LINE__ - 3,  strerror(errno));
            exit(1);
        }

        if(events == 0)
            continue;

        printf("Events = %d\n",events);

        if(fds[0].revents)
        {
            temp_socket = accept(main_socket, NULL, NULL);
            if(temp_socket == -1)
            {
                fprintf(stderr, "%s (%d): Не удалось принять: %s\n",
                        __FILE__, __LINE__ - 3,  strerror(errno));
                exit(1);
            }
            printf("Клиент %d подсоединился\n", get_fds_size());

            fds = add_fds(fds, temp_socket);
            clients_base = add_client(clients_base);


            write(temp_socket,"server-MZH\n",strlen("server-MZH\n"));
            auth(temp_socket);
            fds[0].revents = 0;
        }

        for(i = 1; i < get_fds_size(); i++)
        {
            if(fds[i].revents)
            {
                n_read = read(fds[i].fd, buf, BUF_SIZE + 1);
                if(n_read == -1)
                {
                    fprintf(stderr, "%s (%d): Ошибка при чтении из сокета: %s\n",
                            __FILE__, __LINE__ - 3,  strerror(errno));
                    close(fds[i].fd);
                    fds[i].fd = -1;
                }
                if(n_read == 0)
                {
                    printf("клиент %d отсоединился\n", i);
                    i = disconnect(&fds, &clients_base, i);
                }
                if(n_read > 0)
                {
                    buf[n_read] = '\0';
                    strip_beg(buf);
                    if(strcmp(clients_base[i].name, "\0") == 0)
                        auth2(fds, clients_base, i, buf, fds[i].fd);
                    else
                    {
                        strip_beg(buf);
                        msg = add_to_buf(clients_base, i, buf);
                        if(msg == NULL)
                            continue;
                        if(msg[0] != '\0')
                        {
                            if(cmds(&fds, &clients_base, i, msg))
                            {
                                /* Все сделается в cmd */;
                            }
                            else
                            {
                                msg_everyone(fds, clients_base, i, msg);
                            }
                        }
                        free(msg);

                    }
                }

            }
            fds[i].revents = 0;
        }
    }
    return 100;
}