requests.cpp

#include <stdlib.h>     /* exit, atoi, malloc, free */
#include <stdio.h>
#include <unistd.h>     /* read, write, close */
#include <string.h>     /* memcpy, memset */
#include <sys/socket.h> /* socket, connect */
#include <netinet/in.h> /* struct sockaddr_in, struct sockaddr */
#include <netdb.h>      /* struct hostent, gethostbyname */
#include <arpa/inet.h>
#include "client_lib.hpp"
#include "helpers.hpp"
#include "requests.hpp"

char *compute_get_request(const char *host, const char *url, unordered_map<string, string> cookies, char *jwt)
{
    char *message = (char *)calloc(BUFLEN, sizeof(char));
    char *line = (char *)calloc(LINELEN, sizeof(char));

    // Method type + url + HTTP version
    sprintf(line, "GET %s HTTP/1.1", url);
    compute_message(message, line);

    // Add the host
    sprintf(line, "Host: %s", host);
    compute_message(message, line);

    // If we have a JWT, add it
    if (jwt != NULL) {
        sprintf(line, "Authorization: Bearer %s", jwt);
        compute_message(message, line);
    }

    // Connection header to close it every time
    compute_message(message, "Connection: close");

    // Add all the cookies
    if (cookies.size() > 0) {
        bool first = true;
        for (auto &pair : cookies) {
            if (first) {
                sprintf(line, "Cookie: %s=%s", pair.first.c_str(), pair.second.c_str());
                strcat(message, line);
                first = false;
            } else {
                sprintf(line, "; %s=%s", pair.first.c_str(), pair.second.c_str());
                strcat(message, line);
            }
        }
        compute_message(message, "");
    }
    
    // Add the final newline
    compute_message(message, "");
    return message;
}

char *compute_post_request(const char *host, const char *url, const char* content_type, const char *body_data,
                            int body_len, unordered_map<string, string> cookies, char *jwt)
{
    char *message = (char *)calloc(BUFLEN, sizeof(char));
    char *line = (char *)calloc(LINELEN, sizeof(char));

    // Method + url + HTTP version
    sprintf(line, "POST %s HTTP/1.1", url);
    compute_message(message, line);

    // Add the host
    sprintf(line, "Host: %s", host);
    compute_message(message, line);

    // Add Content-Type header for POST method
    sprintf(line, "Content-Type: %s", content_type);
    compute_message(message, line);

    // Add Content-Length header for Content-Type
    sprintf(line, "Content-Length: %d", body_len);
    compute_message(message, line);

    // If we have a JWT add it
    if (jwt != NULL) {
        sprintf(line, "Authorization: Bearer %s", jwt);
        compute_message(message, line);
    }

    // Connection header to close it every time
    compute_message(message, "Connection: close");

    memset(line, '\0', LINELEN);
    // Add all the cookies
    bool first = true;
    for (auto &pair : cookies) {
        if (first) {
            sprintf(line, "Cookie: %s=%s", pair.first.c_str(), pair.second.c_str());
            strcat(message, line);
            first = false;
        } else {
            sprintf(line, "; %s=%s", pair.first.c_str(), pair.second.c_str());
            strcat(message, line);
        }
    }
    if (cookies.size() > 0) {
        compute_message(message, "");
    }

    // Add final newline
    compute_message(message, "");

    // Add the payload
    strcat(message, body_data);

    free(line);
    return message;
}

char *compute_put_request(const char *host, const char *url, const char* content_type, const char *body_data,
                            int body_len, unordered_map<string, string> cookies, char *jwt) {
        char *message = (char *)calloc(BUFLEN, sizeof(char));
    char *line = (char *)calloc(LINELEN, sizeof(char));

    // Method + url + HTTP version
    sprintf(line, "PUT %s HTTP/1.1", url);
    compute_message(message, line);

    // Add the host
    sprintf(line, "Host: %s", host);
    compute_message(message, line);

    // Add Content-Type and Content-Length for PUT method
    sprintf(line, "Content-Type: %s", content_type);
    compute_message(message, line);

    sprintf(line, "Content-Length: %d", body_len);
    compute_message(message, line);

    // If we have a JWT we add it
    if (jwt != NULL) {
        sprintf(line, "Authorization: Bearer %s", jwt);
        compute_message(message, line);
    }

    // Close connection every time
    compute_message(message, "Connection: close");

    memset(line, '\0', LINELEN);
    // Add all the cookies
    bool first = true;
    for (auto &pair : cookies) {
        if (first) {
            sprintf(line, "Cookie: %s=%s", pair.first.c_str(), pair.second.c_str());
            strcat(message, line);
            first = false;
        } else {
            sprintf(line, "; %s=%s", pair.first.c_str(), pair.second.c_str());
            strcat(message, line);
        }
    }
    if (cookies.size() > 0) {
        compute_message(message, "");
    }

    // Add the final endline
    compute_message(message, "");

    // Add the payload
    strcat(message, body_data);

    free(line);
    return message;
}

char *compute_delete_request(const char *host, const char *url, unordered_map<string, string> cookies, char *jwt, char *body) {
        char *message = (char *)calloc(BUFLEN, sizeof(char));
    char *line = (char *)calloc(LINELEN, sizeof(char));

    // Method + url + HTTP version
    sprintf(line, "DELETE %s HTTP/1.1", url);
    compute_message(message, line);

    // Add the host
    sprintf(line, "Host: %s", host);
    compute_message(message, line);

    // Add JWT if we have it
    if (jwt != NULL) {
        sprintf(line, "Authorization: Bearer %s", jwt);
        compute_message(message, line);
    }

    // Close the connection
    compute_message(message, "Connection: close");

    // Add all the cookies
    if (cookies.size() > 0) {
        bool first = true;
        for (auto &pair : cookies) {
            if (first) {
                sprintf(line, "Cookie: %s=%s", pair.first.c_str(), pair.second.c_str());
                strcat(message, line);
                first = false;
            } else {
                sprintf(line, "; %s=%s", pair.first.c_str(), pair.second.c_str());
                strcat(message, line);
            }
        }
        compute_message(message, "");
    }

    // Add the final newline
    compute_message(message, "");

    // If we have a body append it
    if (body) strcat(message, body);

    return message;
}