98 lines
2.9 KiB
C++
98 lines
2.9 KiB
C++
|
|
#include "Webserv.hpp"
|
|
|
|
#define MAX_EVENTS 42 // arbitrary
|
|
#define TIMEOUT 3000
|
|
|
|
void Webserv::run()
|
|
{
|
|
std::cerr << "Server started\n";
|
|
struct epoll_event events[MAX_EVENTS];
|
|
int nfds;
|
|
int i;
|
|
int count_loop = 0;
|
|
std::vector<listen_socket>::iterator it_lsocket;
|
|
Client *client_cgi_output = NULL;
|
|
Client *client_cgi_input = NULL;
|
|
|
|
g_run = true;
|
|
while (g_run)
|
|
{
|
|
std::cerr << ++count_loop << "----loop epoll() ";
|
|
nfds = ::epoll_wait(_epfd, events, MAX_EVENTS, TIMEOUT);
|
|
std::cerr << "(nfds=" << nfds << ")\n";
|
|
if (nfds == -1)
|
|
{
|
|
int errno_copy = errno;
|
|
std::perror("err epoll_wait()");
|
|
if (errno_copy == EINTR)
|
|
g_run = false;
|
|
else
|
|
throw std::runtime_error("Epoll wait");
|
|
}
|
|
else if (nfds == 0 && !_clients.empty())
|
|
_timeout();
|
|
i = 0;
|
|
while (i < nfds)
|
|
{
|
|
try {
|
|
it_lsocket = std::find(_listen_sockets.begin(), _listen_sockets.end(), events[i].data.fd);
|
|
client_cgi_output = _find_cgi_output_fd(events[i].data.fd);
|
|
client_cgi_input = _find_cgi_input_fd(events[i].data.fd);
|
|
if (it_lsocket != _listen_sockets.end())
|
|
{
|
|
if (events[i].events & EPOLLERR || events[i].events & EPOLLHUP)
|
|
_handle_epoll_error_lsocket(events[i].events, it_lsocket);
|
|
else if (events[i].events & EPOLLIN)
|
|
_accept_connection(*it_lsocket);
|
|
}
|
|
else if (client_cgi_input)
|
|
{
|
|
if (events[i].events & EPOLLERR || events[i].events & EPOLLHUP)
|
|
_handle_epollerr_cgi_input(events[i].events, client_cgi_input);
|
|
else if (events[i].events & EPOLLOUT)
|
|
_cgi_input_ready(client_cgi_input);
|
|
}
|
|
else if (client_cgi_output)
|
|
{
|
|
if (events[i].events & EPOLLERR)
|
|
_handle_epollerr_cgi_output(events[i].events, client_cgi_output);
|
|
else if (events[i].events & EPOLLIN)
|
|
_read_cgi_output(client_cgi_output);
|
|
else if ( (events[i].events & EPOLLHUP) && !(events[i].events & EPOLLIN) )
|
|
_handle_epollhup_cgi_output(events[i].events, client_cgi_output);
|
|
}
|
|
else if (std::find(_clients.begin(), _clients.end(), events[i].data.fd) != _clients.end()) // TODO: save the it in var to avoid multiples find()
|
|
{
|
|
if (events[i].events & EPOLLERR || events[i].events & EPOLLHUP)
|
|
_handle_epoll_error_client(events[i].events, events[i].data.fd);
|
|
else if (events[i].events & EPOLLIN)
|
|
_request( &(*std::find(_clients.begin(), _clients.end(), events[i].data.fd)) );
|
|
else if (events[i].events & EPOLLOUT)
|
|
_response( &(*std::find(_clients.begin(), _clients.end(), events[i].data.fd)) );
|
|
}
|
|
++i;
|
|
if (!g_run)
|
|
break;
|
|
}
|
|
catch (const std::bad_alloc& e) {
|
|
std::cerr << e.what() << '\n';
|
|
_close_all_clients();
|
|
/* Swap to free the memory
|
|
From : http://www.uml.org.cn/c%2B%2B/pdf/EffectiveSTL.pdf#page=66 */
|
|
std::vector<Client>().swap(_clients);
|
|
break;
|
|
}
|
|
catch (const Webserv::ExecFail& e)
|
|
{
|
|
std::cerr << e.what() << '\n';
|
|
throw;
|
|
}
|
|
catch (const std::exception& e) {
|
|
std::cerr << e.what() << '\n';
|
|
++i;
|
|
}
|
|
}
|
|
}
|
|
}
|