From 7c920f7f22a26e377aa9eab0f56ca45caddf7217 Mon Sep 17 00:00:00 2001
From: amelia squires <ameliasquires@disroot.org>
Date: Wed, 2 Oct 2024 11:54:33 -0500
Subject: server errors

---
 src/net.c | 22 +++++++++++++++++++---
 1 file changed, 19 insertions(+), 3 deletions(-)

(limited to 'src/net.c')

diff --git a/src/net.c b/src/net.c
index 9128d89..7f94047 100644
--- a/src/net.c
+++ b/src/net.c
@@ -3,6 +3,8 @@
 #include "net/lua.h"
 #include "net/luai.h"
 
+#define max_uri_size 2048
+
 volatile size_t threads = 0;
 void* handle_client(void *_arg){
   //printf("--\n");
@@ -48,13 +50,19 @@ void* handle_client(void *_arg){
     putchar(buffer[i]);
   putchar('\n');*/
   //printf("hi %li:%i\n", bytes_received,header_eof);
+  if(bytes_received == -2) net_error(client_fd, 431);
+
   
-  //ignore if header is just fucked
+  //ignore if header is just fucked  
   if(bytes_received >= -1){
     parray_t* table;
 
     //checks for a valid header
-    if(parse_header(buffer, header_eof, &table) != -1){
+    int val = parse_header(buffer, header_eof, &table);
+
+    if(val == -2) net_error(client_fd, 414);
+
+    if(val >= 0){
 
       str* sk = (str*)parray_get(table, "Path");
       str* sR = (str*)parray_get(table, "Request");
@@ -211,7 +219,12 @@ void* handle_client(void *_arg){
                   lua_pushvalue(L, req_idx); //push info about the request
 
                   //call the function
-                  lua_call(L, 2, 0);
+                  if(lua_pcall(L, 2, 0, 0) != 0){
+                    //send an error message if send has not been called
+                    if(client_fd >= 0) net_error(client_fd, 500);
+                    
+                    goto net_end;
+                  }
 
                   //check if res:stop() was called
                   lua_pushstring(L, "_stop");
@@ -311,7 +324,10 @@ int start_serv(lua_State* L, int port){
 
     if(count >= max_con){
       //deny request
+      net_error(*client_fd, 503);
+      close(*client_fd);
       free(client_fd);
+
       continue;
     }
     count++;
-- 
cgit v1.2.3