read_request.cpp

/* ************************************************************************** */
/*                                                                            */
/*                                                        :::      ::::::::   */
/*   read_request.cpp                                   :+:      :+:    :+:   */
/*                                                    +:+ +:+         +:+     */
/*   By: ychen2 <ychen2@student.42.fr>              +#+  +:+       +#+        */
/*                                                +#+#+#+#+#+   +#+           */
/*   Created: 2024/08/20 18:32:29 by ychen2            #+#    #+#             */
/*   Updated: 2024/09/09 21:23:31 by ychen2           ###   ########.fr       */
/*                                                                            */
/* ************************************************************************** */

class Server;

#include "MiddleStages.hpp"
#include "Server_helper.hpp"
#include "constants.hpp"
#include "handle_error.hpp"

static void finishReadingHeaders(std::list<State>::iterator &state,
                                 Server &server) {
  // std::cout << "Finished reading headers." << std::endl;

  state->req = Request(state->request_buff);
  server.getServerConfig(*state);
  std::map<std::string, LocationConfig> location = state->server.getLocations();

  state->loc = compare_location(*state, location);
  std::map<std::string, std::string> headers = state->req.getHeaders();
  std::map<std::string, std::string>::iterator CL =
      headers.find("Content-Length");
  if (CL == headers.end()) {
    if (state->req.getMethod() == POST) {
      return handle_error(*state, BAD_REQUEST, MISSING_CONTENT_LENGTH, server);
    } else if (state->req.getMethod() == GET ||
               state->req.getMethod() == DELETE)
      state->contentLength = 0;
  } else {
    if (state->req.getMethod() == GET || state->req.getMethod() == DELETE)
      return handle_error(*state, BAD_REQUEST, EXTRA_CONTENT_LENGTH, server);
    state->contentLength = std::strtol(CL->second.c_str(), NULL, 10);
    if (errno == ERANGE) {
      return handle_error(*state, BAD_REQUEST, INVALID_CONTENT_LENGTH, server);
    }
  }
  if (state->contentLength > state->loc.getClientBodyBufferSize())
    return handle_error(*state, BAD_REQUEST, CONTENT_LENGTH_TOO_LARGE, server);
}

void read_request(std::list<State>::iterator &state, const struct pollfd &pfd,
                  Server &server) {
  if (!(pfd.revents & POLLIN))
    return;

  char buf[BUFFER_SIZE];
  ssize_t rc = recv(state->conn_fd, buf, BUFFER_SIZE, MSG_DONTWAIT);

  // < 0 ..> an error occurs, = 0 client closes the connection
  if (rc <= 0) {
    // std::cerr << "close connection from read_request." << state->event_ct <<
    // std::endl;
    server.close_conn(state);
    return;
  }

  state->request_buff.append(buf, rc);

  state->bodyPos = state->request_buff.find("\r\n\r\n");
  if (state->bodyPos != std::string::npos && !state->isFinishHeaders) {
    finishReadingHeaders(state, server);
    state->isFinishHeaders = true;
  } else if (state->isFinishHeaders) {
    state->req = Request(state->request_buff);
  }

  if (state->contentLength == state->req.getBody().size()) {
    int status_code = state->req.checkRequest();
    if (status_code != OK) {
      return handle_error(*state, status_code, INVALID_REQUEST_FORMAT, server);
    }

    handle_request(*state, server);
  }

}