Update stuff

This commit is contained in:
2026-05-26 00:54:34 -07:00
parent 5dfbdc43c0
commit cf8ddf14bf
2 changed files with 124 additions and 77 deletions
+45 -35
View File
@@ -5,10 +5,11 @@
#include <string.h>
#include <unistd.h>
static const char *EMPTY_STRING = "";
static const char* EMPTY_STRING = "";
HTTPHeaderList *http_header_list_push(HTTPHeaderList *list, const char *key, const char *value) {
HTTPHeaderList *new = malloc(sizeof(HTTPHeaderList));
HTTPHeaderList* http_header_list_push(HTTPHeaderList* list, const char* key, const char* value)
{
HTTPHeaderList* new = malloc(sizeof(HTTPHeaderList));
if (!new) {
return NULL;
}
@@ -31,9 +32,10 @@ HTTPHeaderList *http_header_list_push(HTTPHeaderList *list, const char *key, con
return new;
}
void free_http_header_list(HTTPHeaderList *list) {
void free_http_header_list(HTTPHeaderList* list)
{
while (list) {
HTTPHeaderList *next = list->next;
HTTPHeaderList* next = list->next;
free(list->key);
free(list->value);
free(list);
@@ -41,7 +43,8 @@ void free_http_header_list(HTTPHeaderList *list) {
}
}
const char *http_header_list_search(HTTPHeaderList *list, const char *key, const char *def) {
const char* http_header_list_search(HTTPHeaderList* list, const char* key, const char* def)
{
while (list) {
if (strcmp(list->key, key) == 0) {
return list->value;
@@ -51,21 +54,22 @@ const char *http_header_list_search(HTTPHeaderList *list, const char *key, const
return def;
}
#define MAX_REQUEST_LENGTH 16384
#define MAX_METHOD_LENGTH 16
#define MAX_URI_LENGTH 256
#define MAX_VERSION_LENGTH 3
#define MAX_HEADER_KEY_LENGTH 2048
#define MAX_REQUEST_LENGTH 16384
#define MAX_METHOD_LENGTH 16
#define MAX_URI_LENGTH 256
#define MAX_VERSION_LENGTH 3
#define MAX_HEADER_KEY_LENGTH 2048
#define MAX_HEADER_VALUE_LENGTH 2048
#define RETURN_IF_READ_ERROR(s) \
if (ferror((s))) { \
return HRPR_READ_FAILED; \
#define RETURN_IF_READ_ERROR(s) \
if (ferror((s))) { \
return HRPR_READ_FAILED; \
}
// if an error occured, req->uri is not allocated
static HTTPRequestParseResult parse_method_uri_line(
FILE *stream, size_t *restrict bytes_read, HTTPRequest *restrict req) {
FILE* stream, size_t* restrict bytes_read, HTTPRequest* restrict req)
{
// allow for some leeway in passing incorrect methods
char method[MAX_METHOD_LENGTH + 1];
char uri[MAX_URI_LENGTH + 1];
@@ -73,7 +77,7 @@ static HTTPRequestParseResult parse_method_uri_line(
char whitespace[4];
ssize_t signed_bytes_read;
#define S1(s) #s
#define S(s) S1(s)
#define S(s) S1(s)
int nconv = fscanf(stream,
// clang-format off
"%" S(MAX_METHOD_LENGTH) "[A-Z]"
@@ -109,8 +113,9 @@ static HTTPRequestParseResult parse_method_uri_line(
// return true if there are more headers and no error occurred, false otherwise
// this will *not* free LIST if an error occurs
static bool next_header(FILE *stream, size_t *restrict bytes_read,
HTTPRequestParseResult *restrict res, HTTPHeaderList *restrict *restrict list) {
static bool next_header(FILE* stream, size_t* restrict bytes_read,
HTTPRequestParseResult* restrict res, HTTPHeaderList* restrict* restrict list)
{
char c = fgetc(stream);
RETURN_IF_READ_ERROR(stream);
if (c == '\r') {
@@ -130,7 +135,7 @@ static bool next_header(FILE *stream, size_t *restrict bytes_read,
char whitespace[3];
ssize_t signed_bytes_read;
#define S1(s) #s
#define S(s) S1(s)
#define S(s) S1(s)
int nconv = fscanf(stream,
// clang-format off
"%" S(MAX_HEADER_KEY_LENGTH) "[a-zA-Z0-9.-]"
@@ -154,16 +159,17 @@ static bool next_header(FILE *stream, size_t *restrict bytes_read,
return true;
}
HTTPRequestParseResult parse_http_request(FILE *stream, HTTPRequest *restrict out) {
HTTPRequestParseResult parse_http_request(FILE* stream, HTTPRequest* restrict out)
{
out->uri = EMPTY_STRING;
out->path = EMPTY_STRING;
out->method = EMPTY_STRING;
out->headers = NULL;
size_t total_bytes;
HTTPRequestParseResult res;
if ((res = parse_method_uri_line(stream, &total_bytes, out)) != HRPR_OK) {
return res;
}
out->headers = NULL;
size_t bytes_read;
while (next_header(stream, &bytes_read, &res, &out->headers)) {
if ((total_bytes += bytes_read) > MAX_REQUEST_LENGTH) {
@@ -174,20 +180,22 @@ HTTPRequestParseResult parse_http_request(FILE *stream, HTTPRequest *restrict ou
return res;
}
void free_http_request(HTTPRequest *restrict req) {
void free_http_request(HTTPRequest* restrict req)
{
if (req->method != EMPTY_STRING) {
free((char *) req->method);
free((char*)req->method);
}
if (req->uri != EMPTY_STRING) {
free((char *) req->uri);
free((char*)req->uri);
}
free_http_header_list(req->headers);
}
const char *status_code_to_message(int status, size_t *restrict length) {
const char* status_code_to_message(int status, size_t* restrict length)
{
static const struct {
int code;
const char *msg;
const char* msg;
size_t size;
} CODES[] = {
#define P(s, m) { s, m, sizeof(m) - 1 }
@@ -213,15 +221,17 @@ const char *status_code_to_message(int status, size_t *restrict length) {
return NULL;
}
void format_http_response(FILE *stream, HTTPResponse *restrict resp) {
void format_http_response(FILE* stream, HTTPResponse* restrict resp)
{
assert(status_code_to_message(resp->status, NULL));
fprintf(stream, "HTTP/1.1 %d %s\r\n", resp->status, status_code_to_message(resp->status, NULL));
fprintf(stream, "Content-Length: %zu\r\n", resp->body_length);
for (HTTPHeaderList *h = resp->headers; h; h = h->next) {
fwrite(h->key, 1, h->key_length, stream);
fwrite(": ", 1, 2, stream);
fwrite(h->value, 1, h->value_length, stream);
fwrite("\r\n", 1, 2, stream);
dprintf(fileno(stream), "HTTP/1.1 %d %s\r\n", resp->status,
status_code_to_message(resp->status, NULL));
dprintf(fileno(stream), "Content-Length: %zu\r\n", resp->body_length);
for (HTTPHeaderList* h = resp->headers; h; h = h->next) {
write(fileno(stream), h->key, h->key_length);
write(fileno(stream), ": ", 2);
write(fileno(stream), h->value, h->value_length);
write(fileno(stream), "\r\n", 2);
}
fwrite("\r\n", 1, 2, stream);
write(fileno(stream), "\r\n", 2);
}