This commit is contained in:
Nicholas Orlowsky 2023-10-20 00:27:35 -04:00
parent f1868ceea5
commit dea773366e
No known key found for this signature in database
GPG key ID: BE7DF0188A405E2B
8 changed files with 103 additions and 24 deletions

View file

@ -1,10 +1,10 @@
.PHONY: format lint build build-release build-docker run debug
build:
g++ main.cpp -g -o ./anthracite
g++ main.cpp --std=c++20 -g -o ./anthracite
build-release:
g++ main.cpp -O3 -march=native -o ./anthracite
g++ main.cpp --std=c++20 -O3 -march=native -o ./anthracite
build-docker:
docker build . -t anthracite

View file

@ -5,5 +5,11 @@
class backend {
public:
backend() = default;
virtual ~backend() = default;
backend(backend const&) = delete;
backend& operator = (backend const&) = delete;
backend(backend&&) = delete;
backend& operator=(backend&&) = delete;
virtual std::unique_ptr<http_response> handle_request(http_request& req) = 0;
};

View file

@ -11,9 +11,9 @@ private:
filename = file_dir + filename;
auto file_info = file_cache.find(filename);
int status = 200;
int status = http_status_codes::OK;
if (file_info == file_cache.end()) {
status = 404;
status = http_status_codes::NOT_FOUND;
filename = "./error_pages/404.html";
file_info = file_cache.find(filename);
}
@ -43,10 +43,12 @@ private:
}
public:
file_backend(std::string dir = "./www") : file_dir(dir) {
file_backend(std::string dir = "./www") : file_dir(std::move(dir)) {
populate_cache();
}
~file_backend() = default;
std::unique_ptr<http_response> handle_request(http_request& req) override {
return handle_request_cache(req);
}

View file

@ -17,6 +17,7 @@
#include <bits/stdc++.h>
constexpr int benchmark_loops = 100;
constexpr int SEED = 570;
template <typename KeyType, typename ValueType>
class smart_map {
@ -57,7 +58,7 @@ class smart_map {
void assess_datastructure() {
std::vector<std::pair<KeyType, ValueType>> vals(hmap.begin(), hmap.end());
std::shuffle(vals.begin(), vals.end(), std::default_random_engine(570));
std::shuffle(vals.begin(), vals.end(), std::default_random_engine(SEED));
use_hmap = assess_hmap(vals) > assess_vmap(vals);
}

View file

@ -24,6 +24,72 @@ enum http_method {
UNKNOWN
};
enum http_status_codes {
CONTINUE = 100,
SWITCHING_PROTOCOLS = 101,
PROCESSING = 102,
EARLY_HINTS = 103,
OK = 200,
CREATED = 201,
ACCEPTED = 202,
NON_AUTHORITATIVE_INFORMATION = 203,
NO_CONTENT = 204,
RESET_CONTENT = 205,
PARTIAL_CONTENT = 206,
MULTI_STATUS = 207,
ALREADY_REPORTED = 208,
IM_USED = 226,
MULTIPLE_CHOICES = 300,
MOVED_PERMANENTLY = 301,
FOUND = 302,
SEE_OTHER = 303,
NOT_MODIFIED = 304,
USE_PROXY = 305,
TEMPORARY_REDIRECT = 307,
PERMANENT_REDIRECT = 308,
BAD_REQUEST = 400,
UNAUTHORIZED = 401,
PAYMENT_REQUIRED = 402,
FORBIDDEN = 403,
NOT_FOUND = 404,
METHOD_NOT_ALLOWED = 405,
NOT_ACCEPTABLE = 406,
PROXY_AUTHENTICATION_REQUIRED = 407,
REQUEST_TIMEOUT = 408,
CONFLICT = 409,
GONE = 410,
LENGTH_REQUIRED = 411,
PRECONDITION_FAILED = 412,
PAYLOAD_TOO_LARGE = 413,
URI_TOO_LONG = 414,
UNSUPPORTED_MEDIA_TYPE = 415,
RANGE_NOT_SATISFIABLE = 416,
EXPECTATION_FAILED = 417,
I_AM_A_TEAPOT = 418,
ENHANCE_YOUR_CALM = 420,
MISDIRECTED_REQUEST = 421,
UNPROCESSABLE_ENTITY = 422,
LOCKED = 423,
FAILED_DEPENDENCY = 424,
TOO_EARLY = 425,
UPGRADE_REQUIRED = 426,
PRECONDITION_REQUIRED = 428,
TOO_MANY_REQUESTS = 429,
REQUEST_HEADER_FIELDS_TOO_LARGE = 431,
UNAVAILABLE_FOR_LEGAL_REASONS = 451,
INTERNAL_SERVER_ERROR = 500,
NOT_IMPLEMENTED = 501,
BAD_GATEWAY = 502,
SERVICE_UNAVAILABLE = 503,
GATEWAY_TIMEOUT = 504,
HTTP_VERSION_NOT_SUPPORTED = 505,
VARIANT_ALSO_NEGOTIATES = 506,
INSUFFICIENT_STORAGE = 507,
LOOP_DETECTED = 508,
NOT_EXTENDED = 510,
NETWORK_AUTHENTICATION_REQUIRED = 511
};
static std::unordered_map<std::string, http_method> const http_method_map = {
{ "GET", http_method::GET },
{ "POST", http_method::POST },

View file

@ -8,7 +8,7 @@ private:
std::unordered_map<std::string, http_header> _headers; // kinda goofy, whatever
public:
http_response(std::string& content, std::string filename, int status_code = 200)
http_response(std::string& content, std::string filename, int status_code = http_status_codes::OK)
: _content(content)
, _status_code(status_code)
, _filename(std::move(filename))
@ -32,7 +32,7 @@ public:
std::string header_to_string()
{
std::string response = "";
response += "HTTP/1.1 " + std::to_string(_status_code) + " " + http_status_map.find(_status_code)->second + "\r\n";
response += "HTTP/1.0 " + std::to_string(_status_code) + " " + http_status_map.find(_status_code)->second + "\r\n";
std::string content_type = "text/html";
std::string file_extension = _filename.substr(_filename.rfind('.') + 1);
auto mime_type = mime_types.find(file_extension);

View file

@ -5,6 +5,7 @@
#include <iostream>
#include <mutex>
#include <netinet/in.h>
#include <span>
#include <sstream>
#include <sys/socket.h>
#include <thread>
@ -16,14 +17,10 @@ void log_request_and_response(http_request& req, std::unique_ptr<http_response>&
constexpr int default_port = 80;
constexpr int max_worker_threads = 128;
int active_threads = 0;
std::mutex mtx;
std::condition_variable cv;
void handle_client(anthracite_socket s, file_backend& fb)
void handle_client(anthracite_socket s, backend& b, std::mutex& thread_wait_mutex, std::condition_variable& thread_wait_condvar, int& active_threads)
{
http_request req(s);
std::unique_ptr<http_response> resp = fb.handle_request(req);
std::unique_ptr<http_response> resp = b.handle_request(req);
log_request_and_response(req, resp);
std::string header = resp->header_to_string();
s.send_message(header);
@ -31,32 +28,37 @@ void handle_client(anthracite_socket s, file_backend& fb)
resp.reset();
s.close_conn();
{
std::lock_guard<std::mutex> lock(mtx);
std::lock_guard<std::mutex> lock(thread_wait_mutex);
active_threads--;
}
cv.notify_one();
thread_wait_condvar.notify_one();
}
int main(int argc, char** argv)
{
auto args = std::span(argv, size_t(argc));
int port_number = default_port;
if (argc > 1) {
port_number = atoi(argv[1]);
port_number = atoi(args[1]);
}
std::cout << "Initializing Anthracite" << std::endl;
anthracite_socket s(port_number);
file_backend fb(argc > 2 ? argv[2] : "./www");
file_backend fb(argc > 2 ? args[2] : "./www");
std::cout << "Initialization Complete" << std::endl;
std::cout << "Listening for HTTP connections on port " << port_number << std::endl;
int active_threads = 0;
std::mutex thread_wait_mutex;
std::condition_variable thread_wait_condvar;
while (true) {
s.wait_for_conn();
std::unique_lock<std::mutex> lock(mtx);
cv.wait(lock, [] { return active_threads < max_worker_threads; });
std::unique_lock<std::mutex> lock(thread_wait_mutex);
thread_wait_condvar.wait(lock, [active_threads] { return active_threads < max_worker_threads; });
active_threads++;
std::thread(handle_client, s, std::ref(fb)).detach();
std::thread(handle_client, s, std::ref(fb), std::ref(thread_wait_mutex), std::ref(thread_wait_condvar), std::ref(active_threads)).detach();
}
exit(0);

View file

@ -9,6 +9,8 @@
#include <unistd.h>
#include <unordered_map>
constexpr int MAX_QUEUE_LENGTH = 100;
class anthracite_socket {
private:
int server_socket;
@ -18,7 +20,7 @@ private:
socklen_t client_addr_len {};
public:
anthracite_socket(int port, int max_queue = 10)
anthracite_socket(int port, int max_queue = MAX_QUEUE_LENGTH)
: server_socket(socket(AF_INET, SOCK_STREAM, 0))
, client_ip("")
{
@ -62,7 +64,7 @@ public:
send(client_socket, &msg[0], msg.length(), 0);
}
std::string recv_message(int buffer_size = 1024)
std::string recv_message(int buffer_size)
{
if (client_socket == -1) {
return "";