backend: fixies

This commit is contained in:
Simon 2023-02-13 01:47:36 +01:00
parent 28a91d9f6b
commit d9ee82597e
2 changed files with 132 additions and 127 deletions

View File

@ -1,9 +1,11 @@
#include "tcp.h" #include "tcp.h"
#include <arpa/inet.h> #include <arpa/inet.h>
#include <errno.h>
#include <netinet/in.h> #include <netinet/in.h>
#include <stdint.h> #include <stdint.h>
#include <stdio.h> #include <stdio.h>
#include <stdlib.h> #include <stdlib.h>
#include <string.h>
#include <sys/socket.h> #include <sys/socket.h>
#include <sys/types.h> #include <sys/types.h>
#include <unistd.h> #include <unistd.h>
@ -36,10 +38,12 @@ TcpServer* tcp_server_create(const char* ip, uint16_t port)
server_address.sin_port = htons(port); server_address.sin_port = htons(port);
if (bind(server_socket, (struct sockaddr*)&server_address, sizeof(server_address)) < 0) { if (bind(server_socket, (struct sockaddr*)&server_address, sizeof(server_address)) < 0) {
printf("error: tcp: could not bind socket\n"); printf("error: tcp: could not bind socket\n");
close(server_socket);
return NULL; return NULL;
} }
if (listen(server_socket, SOMAXCONN) < 0) { if (listen(server_socket, SOMAXCONN - 1) < 0) {
printf("error: tcp: could not listen on server\n"); printf("error: tcp: could not listen on server\n");
close(server_socket);
return NULL; return NULL;
} }
@ -59,12 +63,13 @@ void tcp_server_destroy(TcpServer* server)
TcpConnection* tcp_server_accept(TcpServer* server) TcpConnection* tcp_server_accept(TcpServer* server)
{ {
struct sockaddr_in client_address; struct sockaddr_in client_address = { 0 };
socklen_t client_size; socklen_t client_size = 0;
int client_socket int client_socket
= accept(server->server_socket, (struct sockaddr*)&client_address, &client_size); = accept(server->server_socket, (struct sockaddr*)&client_address, &client_size);
if (client_socket < 0) { if (client_socket < 0) {
printf("error: tcp: could not accept connection\n"); printf("error: tcp: could not accept connection\n");
printf("errno: %d %s\n", errno, strerror(errno));
return NULL; return NULL;
} }

View File

@ -7,8 +7,6 @@
#include <stdlib.h> #include <stdlib.h>
#include <string.h> #include <string.h>
const uint16_t port = 8000;
TcpServer* server = NULL; TcpServer* server = NULL;
void interrupt_handler(int a) void interrupt_handler(int a)
@ -20,12 +18,132 @@ void interrupt_handler(int a)
exit(1); exit(1);
} }
void handle_client_connection(TcpConnection* connection)
{
uint8_t buffer[8192] = { 0 };
ssize_t recieved = tcp_recieve(connection, buffer, 8192);
if (recieved < 0) {
printf("error: could not recieve\n");
return;
} else if (recieved == 0) {
printf("client disconnected\n");
return;
}
HttpRequestHeader header = parse_http_request_header((char*)buffer, strlen((char*)buffer));
char* path = calloc(header.path_length + 1, sizeof(char));
strncpy(path, (char*)&buffer[header.path_index], header.path_length);
if (strncmp(path, "/api", 4) == 0) {
// something something api
} else {
if (strstr(path, "..") != NULL) {
uint8_t send_buffer[]
= "HTTP/1.1 400 BAD\r\n"
"Content-Type: text/html\r\n"
"\r\n"
"<!DOCTYPE html><html><head><meta charset=\"utf-8\"><title>Bad "
"request</title></head><body><h1>Fuck you!</h1></body></html>\r\n";
ssize_t written = tcp_send(connection, send_buffer, sizeof(send_buffer));
if (written < 0) {
printf("error: could not write\n");
return;
}
} else if (header.path_length == 0 || strncmp(path, "/", header.path_length) == 0) {
FILE* file = fopen("../frontend/index.html", "r");
if (file == NULL) {
printf("error: could not open file\n");
return;
}
uint8_t send_buffer[] = "HTTP/1.1 200 OK\r\n"
"Content-Type: text/html\r\n"
"\r\n";
ssize_t written = tcp_send(connection, send_buffer, sizeof(send_buffer));
if (written < 0) {
printf("error: could not write\n");
return;
}
char char_read;
while ((char_read = (char)fgetc(file)) != EOF) {
tcp_send(connection, (uint8_t*)&char_read, sizeof(char));
}
} else {
char rootpath[] = "../frontend";
size_t filepath_size = sizeof(rootpath) + header.path_length + 1;
char* filepath = calloc(filepath_size, sizeof(char));
snprintf(filepath, filepath_size, "%s%s", rootpath, path);
char* dot = strrchr(path, '.');
char mime_type[20] = { 0 };
char file_flag[3] = { 'r', 0 };
if (dot != NULL && strncmp(dot, ".html", 5) == 0) {
snprintf(mime_type, 20, "text/html");
} else if (dot != NULL && strncmp(dot, ".css", 4) == 0) {
snprintf(mime_type, 20, "text/css");
} else if (dot != NULL && strncmp(dot, ".js", 3) == 0) {
snprintf(mime_type, 20, "text/javascript");
} else if (dot != NULL && strncmp(dot, ".map", 4) == 0) {
snprintf(mime_type, 20, "application/json");
} else if (dot != NULL && strncmp(dot, ".ico", 4) == 0) {
snprintf(mime_type, 20, "image/x-icon");
} else if (dot != NULL && strncmp(dot, ".jpg", 4) == 0) {
snprintf(mime_type, 20, "image/jpeg");
file_flag[1] = 'b';
} else if (dot != NULL && strncmp(dot, ".png", 4) == 0) {
snprintf(mime_type, 20, "image/png");
file_flag[1] = 'b';
} else if (dot != NULL && strncmp(dot, ".woff2", 6) == 0) {
snprintf(mime_type, 20, "font/woff2");
file_flag[1] = 'b';
} else {
printf("error: unknown file type\n");
return;
}
FILE* file = fopen(filepath, file_flag);
if (file == NULL) {
printf("error: could not open file\n");
return;
}
char send_buffer_1[] = "HTTP/1.1 200 OK\r\n"
"Content-Type: ";
char send_buffer_2[] = "\r\n"
"\r\n";
ssize_t written = tcp_send(connection, (uint8_t*)send_buffer_1, strlen(send_buffer_1));
if (written < 0) {
printf("error: could not write\n");
return;
}
written = tcp_send(connection, (uint8_t*)mime_type, strlen(mime_type));
if (written < 0) {
printf("error: could not write\n");
return;
}
written = tcp_send(connection, (uint8_t*)send_buffer_2, strlen(send_buffer_2));
if (written < 0) {
printf("error: could not write\n");
return;
}
int char_read;
while ((char_read = fgetc(file)) != EOF) {
tcp_send(connection, (uint8_t*)&char_read, sizeof(char));
}
free(filepath);
fclose(file);
}
}
free(path);
}
int main(void) int main(void)
{ {
tcp_global_initialize_sockets(); tcp_global_initialize_sockets();
signal(SIGINT, &interrupt_handler); signal(SIGINT, &interrupt_handler);
const uint16_t port = 8000;
printf("starting server...\n"); printf("starting server...\n");
server = tcp_server_create("127.0.0.1", port); server = tcp_server_create("127.0.0.1", port);
if (server == NULL) if (server == NULL)
@ -34,133 +152,15 @@ int main(void)
while (true) { while (true) {
printf("waiting for client...\n"); printf("waiting for client...\n");
TcpConnection* client_connection = tcp_server_accept(server); TcpConnection* connection = tcp_server_accept(server);
if (client_connection == NULL) { if (connection == NULL) {
printf("error: could not accept client\n"); printf("error: could not accept client\n");
continue; continue;
} }
printf("client connected\n"); printf("client connected\n");
while (true) { handle_client_connection(connection);
uint8_t buffer[8192] = { 0 };
ssize_t recieved = tcp_recieve(client_connection, buffer, 8192);
if (recieved < 0) {
printf("error: could not recieve\n");
break;
} else if (recieved == 0) {
printf("client disconnected\n");
break;
}
HttpRequestHeader header
= parse_http_request_header((char*)buffer, strlen((char*)buffer));
char* path = calloc(header.path_length + 1, sizeof(char));
strncpy(path, (char*)&buffer[header.path_index], header.path_length);
if (strncmp(path, "/api", 4) == 0) {
// something something api
} else {
if (strstr(path, "..") != NULL) {
uint8_t send_buffer[]
= "HTTP/1.1 400 BAD\r\n"
"Content-Type: text/html\r\n"
"\r\n"
"<!DOCTYPE html><html><head><meta charset=\"utf-8\"><title>Bad "
"request</title></head><body><h1>Fuck you!</h1></body></html>\r\n";
ssize_t written = tcp_send(client_connection, send_buffer, sizeof(send_buffer));
if (written < 0) {
printf("error: could not write\n");
break;
}
} else if (header.path_length == 0 || strncmp(path, "/", header.path_length) == 0) {
FILE* file = fopen("../frontend/index.html", "r");
if (file == NULL) {
printf("error: could not open file\n");
break;
}
uint8_t send_buffer[] = "HTTP/1.1 200 OK\r\n"
"Content-Type: text/html\r\n"
"\r\n";
ssize_t written = tcp_send(client_connection, send_buffer, sizeof(send_buffer));
if (written < 0) {
printf("error: could not write\n");
break;
}
char char_read;
while ((char_read = (char)fgetc(file)) != EOF) {
tcp_send(client_connection, (uint8_t*)&char_read, sizeof(char));
}
} else {
char rootpath[] = "../frontend";
size_t filepath_size = sizeof(rootpath) + header.path_length + 1;
char* filepath = calloc(filepath_size, sizeof(char));
snprintf(filepath, filepath_size, "%s%s", rootpath, path);
char* dot = strrchr(path, '.');
char mime_type[20] = { 0 };
char file_flag[3] = { 'r', 0 };
if (dot != NULL && strncmp(dot, ".html", 5) == 0) {
snprintf(mime_type, 20, "text/html");
} else if (dot != NULL && strncmp(dot, ".css", 4) == 0) {
snprintf(mime_type, 20, "text/css");
} else if (dot != NULL && strncmp(dot, ".js", 3) == 0) {
snprintf(mime_type, 20, "text/javascript");
} else if (dot != NULL && strncmp(dot, ".map", 4) == 0) {
snprintf(mime_type, 20, "application/json");
} else if (dot != NULL && strncmp(dot, ".ico", 4) == 0) {
snprintf(mime_type, 20, "image/x-icon");
} else if (dot != NULL && strncmp(dot, ".jpg", 4) == 0) {
snprintf(mime_type, 20, "image/jpeg");
file_flag[1] = 'b';
} else if (dot != NULL && strncmp(dot, ".png", 4) == 0) {
snprintf(mime_type, 20, "image/png");
file_flag[1] = 'b';
} else if (dot != NULL && strncmp(dot, ".woff2", 6) == 0) {
snprintf(mime_type, 20, "font/woff2");
file_flag[1] = 'b';
} else {
printf("error: unknown file type\n");
break;
}
FILE* file = fopen(filepath, file_flag);
if (file == NULL) {
printf("error: could not open file\n");
break;
}
char send_buffer_1[] = "HTTP/1.1 200 OK\r\n"
"Content-Type: ";
char send_buffer_2[] = "\r\n"
"\r\n";
ssize_t written = tcp_send(
client_connection, (uint8_t*)send_buffer_1, strlen(send_buffer_1));
if (written < 0) {
printf("error: could not write\n");
break;
}
written = tcp_send(client_connection, (uint8_t*)mime_type, strlen(mime_type));
if (written < 0) {
printf("error: could not write\n");
break;
}
written = tcp_send(
client_connection, (uint8_t*)send_buffer_2, strlen(send_buffer_2));
if (written < 0) {
printf("error: could not write\n");
break;
}
int char_read;
while ((char_read = fgetc(file)) != EOF) {
tcp_send(client_connection, (uint8_t*)&char_read, sizeof(char));
}
free(filepath);
fclose(file);
}
}
free(path);
break;
}
printf("disconnecting client\n"); printf("disconnecting client\n");
tcp_connection_destroy(client_connection); tcp_connection_destroy(connection);
} }
tcp_server_destroy(server); tcp_server_destroy(server);
return 0; return 0;