add res:stop

1 files changed, 18 insertions, 0 deletions

diff --git a/src/net.c b/src/net.c
index b71b290..260657a 100644
--- a/src/net.c
+++ b/src/net.c
@@ -161,6 +161,8 @@ void* handle_client(void *_arg){
         luaI_tsetcf(L, res_idx, "sendfile", l_sendfile);

         luaI_tsetcf(L, res_idx, "write", l_write);

         luaI_tsetcf(L, res_idx, "close", l_close);

+        luaI_tsetcf(L, res_idx, "stop", l_stop);

+

 

         //values

         luaI_tseti(L, res_idx, "client_fd", client_fd);

@@ -210,10 +212,18 @@ void* handle_client(void *_arg){
 

                   //call the function

                   lua_call(L, 2, 0);

+

+                  //check if res:stop() was called

+                  lua_pushstring(L, "_stop");

+                  lua_gettable(L, res_idx);

+                  if(!lua_isnil(L, -1))

+                    goto net_end;

                  

             }

           }

         }

+

+net_end:

         larray_clear(params);

         parray_lclear(owo); //dont free the rest

 

@@ -289,6 +299,7 @@ int start_serv(lua_State* L, int port){
   if (pthread_mutex_init(&con_mutex, NULL) != 0)

     p_fatal("con_mutex init failed\n");

 

+  int count = 0;

   for(;;){

     struct sockaddr_in client_addr;

     socklen_t client_addr_len = sizeof(client_addr);

@@ -297,6 +308,13 @@ int start_serv(lua_State* L, int port){
     if((*client_fd = accept(server_fd, (struct sockaddr*)&client_addr, &client_addr_len)) < 0)

       p_fatal("failed to accept\n");

 

+    if(count >= max_con){

+      //deny request

+      free(client_fd);

+      continue;

+    }

+    count++;

+

     //open a state to call shit, should be somewhat thread safe

     thread_arg_struct* args = malloc(sizeof * args);