164 lines
2.9 KiB
C++
164 lines
2.9 KiB
C++
|
|
#include "hugo_Webserv.hpp"
|
|
|
|
Webserv::Webserv()
|
|
{
|
|
int on = 1;
|
|
|
|
std::cout << "Server init\n";
|
|
|
|
// create socket descriptor
|
|
_socket_fd = ::socket(AF_INET, SOCK_STREAM | SOCK_NONBLOCK, 0);
|
|
if (_socket_fd == -1)
|
|
{
|
|
::perror("err socket()");
|
|
throw std::runtime_error("Socket init");
|
|
}
|
|
|
|
// allow socket descriptor to be reuseable
|
|
if (setsockopt(_socket_fd, SOL_SOCKET, SO_REUSEADDR, (char *)&on, sizeof(on)) < 0)
|
|
{
|
|
::perror("setsockopt() failed");
|
|
throw std::runtime_error("Socket init");
|
|
}
|
|
|
|
}
|
|
|
|
Webserv::~Webserv()
|
|
{
|
|
std::cout << "Server destroyed\n";
|
|
}
|
|
|
|
void Webserv::bind(in_port_t port)
|
|
{
|
|
struct sockaddr_in addr;
|
|
|
|
addr.sin_family = AF_INET;
|
|
addr.sin_port = ::htons(port);
|
|
addr.sin_addr.s_addr = ::htonl(INADDR_ANY);
|
|
if (::bind(_socket_fd, (const sockaddr*)&addr, sizeof addr) == -1)
|
|
{
|
|
::perror("err bind()");
|
|
throw std::runtime_error("Socket bind");
|
|
}
|
|
}
|
|
|
|
void Webserv::listen(unsigned int max_connections)
|
|
{
|
|
if (::listen(_socket_fd, max_connections) == -1)
|
|
{
|
|
::perror("err listen(): ");
|
|
throw std::runtime_error("Socket listen");
|
|
}
|
|
}
|
|
|
|
void Webserv::_add_fd(int sd, short event)
|
|
{
|
|
struct pollfd new_poll_fd;
|
|
|
|
new_poll_fd.fd = sd;
|
|
new_poll_fd.events = event;
|
|
new_poll_fd.revents = 0;
|
|
_fds.push_back(new_poll_fd);
|
|
}
|
|
|
|
void Webserv::_close_fd(int i)
|
|
{
|
|
close(_fds[i].fd);
|
|
_fds[i].fd = -1;
|
|
_fds.erase(_fds.begin() + i);
|
|
}
|
|
|
|
void Webserv::_accept_connection()
|
|
{
|
|
int new_sd = 0;
|
|
|
|
printf(" Listening socket is readable\n");
|
|
while (new_sd != -1)
|
|
{
|
|
new_sd = accept(_socket_fd, NULL, NULL);
|
|
if (new_sd < 0)
|
|
{
|
|
if (errno != EWOULDBLOCK && errno != EAGAIN)
|
|
{
|
|
::perror("err accept()");
|
|
throw std::runtime_error("Socket accept");
|
|
}
|
|
break;
|
|
}
|
|
printf(" New incoming connection - %d\n", new_sd);
|
|
_add_fd(new_sd, POLLIN);
|
|
}
|
|
}
|
|
|
|
void Webserv::_connect_socket(int i, int bufsize)
|
|
{
|
|
int len;
|
|
int ret;
|
|
std::vector<char> buffer(bufsize);
|
|
|
|
printf(" Descriptor %d is readable\n", _fds[i].fd);
|
|
while (TRUE)
|
|
{
|
|
ret = recv(_fds[i].fd, &buffer[0], buffer.size(), 0);
|
|
if (ret < 0)
|
|
{
|
|
if (errno != EWOULDBLOCK)
|
|
{
|
|
::perror(" recv() failed");
|
|
_close_fd(i);
|
|
}
|
|
break;
|
|
}
|
|
|
|
if (ret == 0)
|
|
{
|
|
printf(" Connection closed\n");
|
|
_close_fd(i);
|
|
break;
|
|
}
|
|
|
|
len = ret;
|
|
printf(" %d bytes received\n", len);
|
|
|
|
ret = send(_fds[i].fd, &buffer[0], len, 0);
|
|
if (ret < 0)
|
|
{
|
|
::perror(" send() failed");
|
|
_close_fd(i);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
void Webserv::start(int timeout, int bufsize)
|
|
{
|
|
_add_fd(_socket_fd, POLLIN);
|
|
|
|
std::cout << "Server started\n";
|
|
while (TRUE)
|
|
{
|
|
printf("-----------\npoll\n");
|
|
poll(&_fds[0], _fds.size(), timeout);
|
|
|
|
for (unsigned int i = 0; i < _fds.size(); i++)
|
|
{
|
|
if(_fds[i].revents == 0)
|
|
continue;
|
|
|
|
if(_fds[i].revents != POLLIN)
|
|
{
|
|
::perror("err revents");
|
|
throw std::runtime_error("Socket poll");
|
|
}
|
|
|
|
if (_fds[i].fd == _socket_fd)
|
|
_accept_connection();
|
|
|
|
else
|
|
_connect_socket(i, bufsize);
|
|
}
|
|
}
|
|
}
|
|
|