86 lines
2.0 KiB
C++
86 lines
2.0 KiB
C++
|
|
#include "Webserv.hpp"
|
|
|
|
// Arbitrary values
|
|
#define BUFSIZE 8192 // (8Ko)
|
|
#define MAX_HEADER_SIZE 16384 // (16Ko)
|
|
|
|
enum read_return
|
|
{
|
|
READ_IN_PROGRESS,
|
|
READ_COMPLETE,
|
|
READ_CLOSE,
|
|
};
|
|
|
|
void Webserv::_request(Client *client)
|
|
{
|
|
int ret = _read_request(client);
|
|
|
|
if (g_last_signal)
|
|
_handle_last_signal();
|
|
|
|
if (ret == READ_CLOSE)
|
|
{
|
|
_close_client(client->get_cl_fd());
|
|
}
|
|
else if (ret == READ_COMPLETE)
|
|
{
|
|
if (client->body_complete)
|
|
std::cerr << "______BODY\n" << client->get_rq_body() << "\n______\n"; // DEBUG
|
|
_epoll_update(client->get_cl_fd(), EPOLLOUT, EPOLL_CTL_MOD);
|
|
client->request_complete = true;
|
|
}
|
|
}
|
|
|
|
int Webserv::_read_request(Client *client)
|
|
{
|
|
char buf[BUFSIZE];
|
|
ssize_t ret;
|
|
|
|
std::cerr << "call recv()" << "\n" ;
|
|
ret = ::recv(client->get_cl_fd(), buf, BUFSIZE, 0);
|
|
std::cerr << "recv() on fd(" << client->get_cl_fd() << ") returned = " << ret << "\n" ;
|
|
if (ret == -1)
|
|
{
|
|
std::perror("err recv()");
|
|
return READ_CLOSE;
|
|
}
|
|
if (ret == 0)
|
|
{
|
|
std::cerr << "recv() read 0, then close client" << "\n"; // DEBUG
|
|
return READ_CLOSE;
|
|
}
|
|
client->raw_request.append(buf, ret);
|
|
|
|
// print_special(client->raw_request);
|
|
|
|
if (!client->header_complete)
|
|
{
|
|
client->parse_request_headers(_servers);
|
|
if (client->status)
|
|
return READ_COMPLETE;
|
|
if (client->header_complete)
|
|
{
|
|
if (client->get_rq_headers("Content-Type").empty()
|
|
&& client->get_rq_headers("Content-Length").empty()
|
|
// && client->get_rq_headers("Transfer-Encoding").empty()
|
|
)
|
|
return READ_COMPLETE; // No body case
|
|
}
|
|
else if (client->raw_request.size() > MAX_HEADER_SIZE)
|
|
{ // 413 or 400 ? 413 seems common among http server, but don't fit perfectly.
|
|
client->status = 413;
|
|
return READ_COMPLETE;
|
|
}
|
|
}
|
|
else if (client->header_complete)
|
|
{
|
|
// client->read_body_size += ret; // Not accurate, part of body could have been read with headers, unused for now
|
|
client->parse_request_body();
|
|
if (client->status || client->body_complete)
|
|
return READ_COMPLETE;
|
|
}
|
|
|
|
return READ_IN_PROGRESS;
|
|
}
|