sort networking code

1 files changed, 195 insertions, 0 deletions

diff --git a/src/net/luai.c b/src/net/luai.c
new file mode 100644
index 0000000..ea93285
--- /dev/null
+++ b/src/net/luai.c
@@ -0,0 +1,195 @@
+#include "luai.h"
+#include "common.h"
+
+void i_write_header(lua_State* L, int header_top, str** _resp, char* content, size_t len){
+  str* resp;
+  lua_pushvalue(L, header_top);
+
+  str* header_vs = str_init("");
+  lua_pushnil(L);
+    
+  for(;lua_next(L, header_top) != 0;){
+      char* key = (char*)luaL_checklstring(L, -2, NULL);
+      if(strcmp(key, "Code") != 0){
+        str_push(header_vs, key);
+        str_push(header_vs, ": ");
+        str_push(header_vs, (char*)luaL_checklstring(L, -1, NULL));
+        str_push(header_vs, "\r\n");
+      }
+      lua_pop(L, 1);
+  }
+
+  lua_pushvalue(L, header_top);
+  lua_pushstring(L, "Code");
+  lua_gettable(L, header_top);
+  int code = luaL_checkinteger(L, -1);
+    
+  char code_det[50] = {0};
+  http_code(code, code_det);
+  http_build(&resp, code,  code_det, header_vs->c, content, len);
+
+  str_free(header_vs);
+  *_resp = resp;
+}
+
+/**
+ * @brief parses all files in response buffer into a lua table
+ *
+ * @param {lua_State*} lua state to put table into
+ * @param {char*} response buffer
+ * @param {str*} response header Content-Type value
+ * @return {int} lua index of table
+*/
+int rolling_file_parse(lua_State* L, int* files_idx, int* body_idx, char* buffer, str* content_type, size_t blen, struct file_parse* _content){
+  struct file_parse content = *_content;
+  /*enum file_status* status = (enum file_status*)parray_get(content, "_status");
+  str* current = (str*)parray_get(content, "_current");
+  str* old = (str*)parray_get(content, "_old");
+  str* boundary = (str*)parray_get(content, "_boundary");
+  str* boundary_id = (str*)parray_get(content, "_boundary_id");
+  int* dash_count = (int*)parray_get(content, "_dash_count");
+  int* table_idx = (int*)parray_get(content, "_table_idx");*/
+  int override = 0;
+
+  //time_start(start)
+  if(content.status == _ignore){
+    content.boundary = str_init(""); //usually add + 2 to the length when using
+    int state = 0;
+    for(int i = 0; content_type->c[i] != '\0'; i++){
+      if(state == 2 && content_type->c[i] != '-') str_pushl(content.boundary, content_type->c + i, 1);
+      if(content_type->c[i] == ';') state = 1;
+      if(content_type->c[i] == '=' && state == 1) state = 2;
+    }
+    if(state == 2){
+      str_pushl(content.boundary, "\r\n\r\n", 4);
+    }
+
+    content.status = state==2?BARRIER_READ:NORMAL;//malloc(sizeof * status); content.status = state==2?BARRIER_READ:NORMAL;
+    content.dash_count = 0;//malloc(sizeof * dash_count); *dash_count = 0;
+    content.current = str_init("");
+
+    content.table_idx = lua_gettop(L);//malloc(sizeof * table_idx); *table_idx = lua_gettop(L);
+    //parray_set(content, "_table_idx", (void*)(table_idx));
+    //parray_set(content, "_status", (void*)(status));
+    //parray_set(content, "_dash_count", (void*)(dash_count));
+    //parray_set(content, "_current", (void*)(current));
+    
+    content.boundary_id = str_init("");
+    str_popb(content.boundary, 4);
+    //parray_set(content, "_boundary", (void*)boundary);
+    //parray_set(content, "_boundary_id", (void*)boundary_id);
+    
+  }
+  //time_end("start", start)
+  //printf("hi\n");
+  if(content.status == NORMAL){
+    //printf("normal\n");
+    //strnstr(buffer, )
+    //if(override) str_clear(current);
+    //str_pushl(current, buffer, blen);
+    //printf("%s\n",current->c);
+    lua_pushvalue(L, *body_idx);
+    lua_pushlstring(L, buffer, blen);
+    lua_concat(L, 2);
+    *body_idx = lua_gettop(L);
+  } else {
+  file_start:;
+    //time_start(barrier_read)
+    if(content.status == BARRIER_READ){
+      //printf("read %llu\n", blen);
+      for(int i = 0; i != blen; i++){
+        //printf("%c",buffer[i]);
+        //printf("\n");
+        if(*buffer == '\r'){
+          content.status = FILE_HEADER;
+          buffer+=2;
+          blen-=i+2;
+          
+          content.table_idx = lua_rawlen(L, *files_idx) + 1;
+          lua_pushinteger(L, content.table_idx);
+          lua_newtable(L);
+          lua_settable(L, *files_idx);
+          break;
+        }
+        str_pushl(content.boundary_id, buffer, 1);
+        buffer++;
+      }
+    }
+    //time_end("barrier_read", barrier_read)
+    lua_pushvalue(L, *files_idx);
+    lua_pushinteger(L, content.table_idx);
+    lua_gettable(L, -2);
+    int rfiles_idx = lua_gettop(L);
+    //time_start(file_header)
+    if(content.status == FILE_HEADER){
+      //printf("header\n");
+      for(int i = 0; i < blen; i++){
+
+        if(buffer[i] == ':'){
+          content.old = content.current;
+          content.current = str_init("");
+        } else if(buffer[i] == '\n'){
+          if(content.current->len == 0){
+            content.status = FILE_BODY;
+            buffer += i;
+            blen -= i;
+            content.old = NULL;
+            str_free(content.current);
+            content.current = str_init("");
+            break;
+          }
+          //printf("%i '%s' : '%s'\n",*table_idx, old->c, current->c);
+
+          luaI_tsets(L, rfiles_idx, content.old->c, content.current->c);
+          
+          str_free(content.old);
+          content.old = NULL;
+          str_clear(content.current);
+        } else if(buffer[i] != '\r' && !(buffer[i] == ' ' && content.current->len == 0)) str_pushl(content.current, buffer + i, 1);
+      }
+    } 
+    //time_end("file_header", file_header)
+    //time_start(file_body)
+    if(content.status == FILE_BODY){
+      //printf("body\n");
+      //if(content.old==NULL) content.old = str_init("");
+      char* barrier_end = strnstr(buffer, content.boundary->c, blen);
+      if(barrier_end == NULL){
+        str* temp = str_initl(content.current->c, content.current->len);
+        str_pushl(temp, buffer, blen);
+        barrier_end = strnstr(temp->c, content.boundary->c, temp->len);
+        if(barrier_end != NULL) abort(); // todo
+
+        str* temp2 = content.current;
+        content.current = temp;
+        str_free(temp2);
+
+      } else {
+        char* start = barrier_end, *end = barrier_end;
+        for(; *start != '\n'; start--);
+        for(; *end != '\n'; end++);
+        int clen = start - buffer;
+        str_pushl(content.current, buffer, clen);
+        luaI_tsetsl(L, rfiles_idx, "content", content.current->c, content.current->len);
+        str_clear(content.current);
+        blen-= end - buffer;
+        buffer = end;
+        content.status = BARRIER_READ;
+        goto file_start;
+        //printf("%s\n",content.current->c);
+      }
+      
+    }
+    //time_end("file_body", file_body)
+  }
+  /*parray_set(content, "_dash_count", dash_count);
+  parray_set(content, "_boundary_id", boundary_id);
+  parray_set(content, "_boundary", boundary);
+  parray_set(content, "_status", status);
+  parray_set(content, "_current", current);
+  parray_set(content, "_old", old);*/
+
+  *_content = content;
+
+  return 0;
+}