160 lines
3.4 KiB
C++
160 lines
3.4 KiB
C++
|
|
#include "Webserv.hpp"
|
|
|
|
Webserv::Webserv()
|
|
{
|
|
std::cout << "Server init\n";
|
|
_socket_fd = ::socket(AF_INET, SOCK_STREAM | SOCK_NONBLOCK, 0);
|
|
//_socket_fd = ::socket(AF_INET, SOCK_STREAM, 0);
|
|
if (_socket_fd == -1)
|
|
{
|
|
std::perror("err socket(): ");
|
|
throw std::runtime_error("Socket init");
|
|
}
|
|
}
|
|
|
|
/* Webserv::Webserv(Webserv const &src)
|
|
{
|
|
|
|
} */
|
|
|
|
Webserv::~Webserv()
|
|
{
|
|
std::cout << "Server destroyed\n";
|
|
}
|
|
|
|
/* Webserv & Webserv::operator=(Webserv const &rhs)
|
|
{
|
|
|
|
} */
|
|
|
|
///////////////
|
|
// Functions //
|
|
|
|
void Webserv::bind(in_port_t port)
|
|
{
|
|
// cast invalid ? how to ?
|
|
// const struct sockaddr* cast_test = static_cast<const struct sockaddr*>(addr);
|
|
|
|
struct sockaddr_in addr;
|
|
std::memset(&addr, 0, sizeof addr);
|
|
addr.sin_family = AF_INET;
|
|
addr.sin_port = ::htons(port);
|
|
addr.sin_addr.s_addr = ::htonl(INADDR_ANY); // htonl useless with 0 value (INADDR_ANY) ?
|
|
|
|
if (::bind(_socket_fd, (const sockaddr*)&addr, sizeof addr) == -1)
|
|
{
|
|
std::perror("err bind(): ");
|
|
throw std::runtime_error("Socket bind");
|
|
}
|
|
}
|
|
|
|
void Webserv::listen(unsigned int max_connections)
|
|
{
|
|
if (::listen(_socket_fd, max_connections) == -1)
|
|
{
|
|
std::perror("err listen(): ");
|
|
throw std::runtime_error("Socket listen");
|
|
}
|
|
}
|
|
|
|
#define BUFSIZE 8192
|
|
#define MSG_TEST "Le Webserv / 20 =D\n"
|
|
#define MSG_BOUNCE "bounced properly ;)\n" // placeholder
|
|
|
|
void Webserv::start()
|
|
{
|
|
struct sockaddr_in addr;
|
|
socklen_t addr_len;
|
|
int accepted_fd;
|
|
char buf[BUFSIZE+1]; // WIP buffer. need to try with std::vector or std::string.
|
|
int ret;
|
|
|
|
int timeout = 3000;
|
|
int i = 1;
|
|
int nfds = 1;
|
|
int ret_nfds;
|
|
|
|
_poll_s[0].fd = _socket_fd;
|
|
_poll_s[0].events = POLLIN;
|
|
|
|
std::cout << "Server started\n";
|
|
int i_loop = 0;
|
|
++i_loop;
|
|
std::cout << i_loop << "----------\n";
|
|
std::cout << "loop poll()\n";
|
|
while ( (ret_nfds = ::poll(_poll_s, nfds, timeout)) != -1)
|
|
{
|
|
i = 1;
|
|
while (i < nfds)
|
|
{
|
|
if (ret_nfds == 0)
|
|
{
|
|
std::cout << "timeout\n";
|
|
::close(_poll_s[i].fd);
|
|
_poll_s[i].fd = -1;
|
|
--nfds;
|
|
}
|
|
|
|
else if (_poll_s[i].revents & POLLIN) // READ
|
|
{
|
|
std::cout << "recv()\n";
|
|
ret = ::recv(_poll_s[i].fd, buf, BUFSIZE, 0);
|
|
if (ret == -1)
|
|
{
|
|
std::perror("err recv(): ");
|
|
if (::send(_poll_s[i].fd, MSG_BOUNCE, sizeof MSG_BOUNCE - 1, 0) == -1)
|
|
std::perror("err send(): ");
|
|
|
|
::close(_poll_s[i].fd);
|
|
_poll_s[i].fd = -1;
|
|
--nfds;
|
|
continue;
|
|
}
|
|
/*
|
|
if (ret == BUFSIZE)
|
|
// send error like "request too long" to client
|
|
*/
|
|
buf[ret] = '\0';
|
|
_poll_s[i].events = POLLOUT;
|
|
}
|
|
|
|
else if (_poll_s[i].revents & POLLOUT) // WRITE
|
|
{
|
|
std::cout << "send()\n";
|
|
if (::send(_poll_s[i].fd, buf, ret, 0) == -1) // echo the read
|
|
std::perror("err send(): ");
|
|
if (::send(_poll_s[i].fd, MSG_TEST, sizeof MSG_TEST - 1, 0) == -1)
|
|
std::perror("err send(): ");
|
|
|
|
::close(_poll_s[i].fd);
|
|
_poll_s[i].fd = -1;
|
|
--nfds;
|
|
}
|
|
|
|
++i;
|
|
}
|
|
|
|
|
|
if ((_poll_s[0].fd == _socket_fd) && (_poll_s[0].revents & POLLIN)) // ACCEPT
|
|
{
|
|
std::cout << "accept()\n";
|
|
addr_len = sizeof addr;
|
|
accepted_fd = ::accept(_socket_fd, (sockaddr*)&addr, &addr_len);
|
|
if (accepted_fd == -1)
|
|
{
|
|
std::perror("err accept(): ");
|
|
continue;
|
|
}
|
|
::fcntl(accepted_fd, F_SETFL, O_NONBLOCK);
|
|
_poll_s[nfds].fd = accepted_fd;
|
|
_poll_s[nfds].events = POLLIN;
|
|
++nfds;
|
|
}
|
|
|
|
++i_loop;
|
|
std::cout << i_loop << "----------\n";
|
|
std::cout << "loop poll()\n";
|
|
}
|
|
}
|