Блог

Это моя реализация будущего HTTP-сервера.

 void    Webserv::resetFdSets()
{

    int         fildes;

    FD_ZERO(amp;to_read);
    FD_ZERO(amp;to_write);
    max_fd = -1;
    Server_map::iterator it;
    for (it = servers.begin(); it != servers.end();   it) //set server sockets to be read
    {
        fildes = it->second.getFd();
        FD_SET(fildes, amp;to_read);
        if (fildes > max_fd)
            max_fd = fildes;
    }
    std::list<int>::iterator iter;
    for (iter = accepted.begin(); iter != accepted.end();   iter) // set client sockets if any
    {
        fildes = (*iter);
        FD_SET(fildes, amp;to_read);
        FD_SET(fildes, amp;to_write);
        if (fildes > max_fd)
            max_fd = fildes;
    }
}
 

принять ()

 void    Webserv::acceptConnections()
{

   int         client_fd;
   sockaddr_in cli_addr;
   socklen_t   cli_len = sizeof(sockaddr_in);

   Server_map::iterator    it;
   for (it = servers.begin(); it != servers.end();   it)
   {
       ft_bzero(amp;cli_addr, sizeof(cli_addr));
       if (FD_ISSET(it->second.getFd(), amp;to_read)) // if theres data in server socket
       {
           client_fd = accept(it->second.getFd(), reinterpret_cast<sockaddr*>(amp;cli_addr), amp;cli_len);
           if (client_fd > 0) // accept and add client
           {
               fcntl(client_fd, F_SETFL, O_NONBLOCK);
               accepted.push_back(client_fd);
           }
           else
               throw (std::runtime_error("accept() failed"));
       }
    }
}
 

recv()

 void    Webserv::checkReadSockets()
{

    char        buf[4096];
    std::string raw_request;
    ssize_t     bytes;
    static int  connections;

    std::list<int>::iterator    it;
    for (it = accepted.begin(); it != accepted.end();   it)
    {
        if (FD_ISSET(*it, amp;to_read))
        {
              connections;
            std::cout << "Connection counter : " << connections << std::endl;
            while ((bytes = recv(*it, buf, sizeof(buf) - 1, MSG_DONTWAIT)) > 0)
            {
                buf[bytes] = '';
                raw_request  = buf;
            }
            std::cout << raw_request << std::endl;
        }
    }
}
 

отправить ()

 void    Webserv::checkWriteSockets()
{

   char            buf[8096];
   char            http_success[] = {
       "HTTP/1.1 200 OKrn"
       "Server: This op serverrn"
       "Content-Length: 580rn"
       "Content-Type: text/htmlrn"
       "Connection: Closedrnrn" 
   };

   std::list<int>::iterator it = accepted.begin();
   while (it != accepted.end())
   {
       int cliSock = (*it);
       if (FD_ISSET(cliSock, amp;to_write))
       {
           int     response_fd = open("hello.html", O_RDONLY);

           int bytes = read(response_fd, buf, sizeof(buf));
           send(cliSock, http_success, sizeof(http_success) - 1, MSG_DONTWAIT); // header
           send(cliSock, buf, bytes, MSG_DONTWAIT); // hello world

           close(cliSock);
           close(response_fd);
        
           it = accepted.erase(it);
       }
       else
             it;
   }
}
 

main loop:

 void    Webserv::operate()
{

   int     rc;

   while (true)
   {
       resetFdSets(); // add server and client fd for select()
       if ((rc = select(max_fd   1, amp;to_read, amp;to_write, NULL, NULL)) > 0) // if any of the ports are active
       {
           acceptConnections(); // create new client sockets with accept()
           checkReadSockets(); // parse and route client requests recv() ...
           checkWriteSockets(); // send() response and delete client
       }
       else if (rc < 0)
       {
           std::cerr << "select() failed" << std::endl;
           exit(1);
       }
   }
}
 

Тот же код выделил определение класса Webserv в pastebin: https://pastebin.com/9B8uYumF

На данный момент алгоритм таков:

• сбросьте настройки fd_sets для чтения и записи.
• если какой-либо из файловых дескрипторов запускает select, мы проверяем, установлены ли FD сервера, принимаем соединения и сохраняем их в списке целых чисел.
• если какой-либо из клиентских FD является FD_ISSET() для чтения — мы считываем их запрос и печатаем его.
• наконец, если какой-либо из клиентских FD готов к приему — мы помещаем туда html-страницу и закрываем соединение.

При проверке всех дескрипторов клиента FD_ISSET() возвращает false в 80% случаев, когда этого не должно быть. Следовательно, я не могу получать запросы клиентов последовательно. Как ни странно, он возвращает true с помощью fd_set для сокетов записи гораздо чаще. Вот демонстрационная версия с 2 из 10 успешных чтений с использованием приведенного выше кода: https://imgur.com/a/nzc21LV

Я знаю, что новые клиенты всегда принимаются, потому что он всегда вводит условие if в acceptConnections() . Другими словами, listen FD инициализируются правильно.

редактировать: Вот как он инициализируется:

 void    Server::init()
{
    if ((listen_fd = socket(AF_INET, SOCK_STREAM, 0)) < 0)
    {
        throw (std::runtime_error("socket() failed"));
    }
    fcntl(listen_fd, F_SETFL, O_NONBLOCK);

    int yes = 1;
    setsockopt(listen_fd, SOL_SOCKET, SO_REUSEADDR, amp;yes, sizeof(int));
    if (bind(listen_fd, reinterpret_cast<sockaddr*>(amp;server_addr), sizeof(server_addr)))
    {
        throw (std::runtime_error("bind() failed"));
    }

    if (listen(listen_fd, 0))
        throw (std::runtime_error("listen() failed"));
}