#ifdef _WIN32 //add -lws2_32 #include //#define socklen_t __socklen_t //#define close closesocket typedef int socklen_t; #else #include #include #define closesocket close #endif #include #include #include #include #include #include #include #include #include "net.h" #include "lua.h" #include "io.h" #include "table.h" #include "types/str.h" #include "types/parray.h" #define max_con 200 //2^42 #define BUFFER_SIZE 1//20000 #define HTTP_BUFFER_SIZE 4098 #define max_content_length 20 static int ports[65535] = { 0 }; static parray_t* paths = NULL; struct lchar { char* c; int len; char req[20]; }; struct sarray_t { struct lchar** cs; int len; }; pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER; pthread_mutex_t lua_mutex = PTHREAD_MUTEX_INITIALIZER; /** * @brief calls recv into buffer until everything is read * * buffer is allocated in BUFFER_SIZE chunks * * @param {int} fd of the connection * @param {char**} pointer to a unallocated buffer * @param {int*} pointer to an int, will be where the header ends * @return {int64_t} bytes read, -1 if the body was damaged, -2 if the header was */ int64_t recv_full_buffer(int client_fd, char** _buffer, int* header_eof, int* state){ char* header, *buffer = malloc(BUFFER_SIZE * sizeof * buffer); memset(buffer, 0, BUFFER_SIZE); int64_t len = 0; *header_eof = -1; int n, content_len = -1; //printf("&_\n"); for(;;){ n = recv(client_fd, buffer + len, BUFFER_SIZE, 0); printf("hi"); if(n < 0){ *_buffer = buffer; printf("%s %i\n",strerror(errno),errno); if(*header_eof == -1) return -2; //dont even try w/ request, no header to read return -1; //well the header is fine atleast }; if(*header_eof == -1 && (header = strstr(buffer, "\r\n\r\n")) != NULL){ *header_eof = header - buffer; char* cont_len_raw = strstr(buffer, "Content-Length: "); if(cont_len_raw == NULL) { len += n; *_buffer = buffer; return len; } str* cont_len_str = str_init(""); if(cont_len_raw == NULL) abort(); //i is length of 'Content-Length: ' for(int i = 16; cont_len_raw[i] != '\r'; i++) str_pushl(cont_len_str, cont_len_raw + i, 1); content_len = strtol(cont_len_str->c, NULL, 10); str_free(cont_len_str); if(content_len > max_content_length) { *_buffer = buffer; *state = (len + n != content_len + *header_eof + 4); return len + n; } buffer = realloc(buffer, content_len + *header_eof + 4 + BUFFER_SIZE); if(buffer == NULL) p_fatal("unable to allocate"); } len += n; if(*header_eof == -1){ buffer = realloc(buffer, len + BUFFER_SIZE + 1); memset(buffer + len, 0, n + 1); } if(content_len != -1 && len - *header_eof - 4 >= content_len) break; } *_buffer = buffer; return len; } /** * @brief converts the request buffer into a parray_t * * @param {char*} request buffer * @param {int} where the header ends * @param {parray_t**} pointer to a unallocated parray_t * @return {int} returns 0 or -1 on failure */ int parse_header(char* buffer, int header_eof, parray_t** _table){ if(header_eof == -1) return -1; char add[] = {0,0}; int lines = 3; for(int i = 0; i != header_eof; i++) lines += buffer[i] == '\n'; parray_t* table = parray_init(); str* current = str_init(""); int oi = 0; int item = 0; for(; oi != header_eof; oi++){ if(buffer[oi] == ' ' || buffer[oi] == '\n'){ if(buffer[oi] == '\n') current->c[current->len - 1] = 0; parray_set(table, item == 0 ? "Request" : item == 1 ? "Path" : "Version", (void*)str_init(current->c)); str_clear(current); item++; if(buffer[oi] == '\n') break; } else str_pushl(current, buffer + oi, 1); } int key = 1; str* sw = NULL; for(int i = oi + 1; i != header_eof; i++){ if(buffer[i] == ' ' && strcmp(current->c, "") == 0) continue; if(key && buffer[i] == ':' || !key && buffer[i] == '\n'){ if(key){ sw = current; current = str_init(""); key = 0; } else { if(buffer[oi] == '\n') current->c[current->len - 1] = 0; parray_set(table, sw->c, (void*)str_init(current->c)); str_clear(current); sw = NULL; key = 1; } continue; } else str_pushl(current, buffer + i, 1); } //printf("'%.*s'\n",header_eof + 5, buffer); parray_set(table, sw->c, (void*)str_init(current->c)); //parray_set(table, "Body", (void*)str_initl(buffer + header_eof + 4, buffer_len - header_eof - 4)); str_free(current); *_table = table; return 0; } /** * @brief contructs an http request * * @param {str**} pointer to an unallocated destination string * @param {int} response code * @param {char*} string representation of the response code * @param {char*} all other header values * @param {char*} response content * @param {size_t} content length */ void http_build(str** _dest, int code, char* code_det, char* header_vs, char* content, size_t len){ char* dest = malloc(HTTP_BUFFER_SIZE); memset(dest, 0, HTTP_BUFFER_SIZE); sprintf(dest, "HTTP/1.1 %i %s\r\n" "%s" "\r\n" , code, code_det, header_vs); *_dest = str_init(dest); str_pushl(*_dest, content, len); free(dest); } typedef struct { int fd; int port; lua_State* L; struct sockaddr_in cli; } thread_arg_struct; /** * @brief gets a string representation of a http code * * @param {int} http response code * @param {char*} allocated destination string */ void http_code(int code, char* code_det){ //this was done with a script btw switch(code){ case 100: sprintf(code_det,"Continue"); break; case 101: sprintf(code_det,"Switching Protocols"); break; case 102: sprintf(code_det,"Processing"); break; case 103: sprintf(code_det,"Early Hints"); break; case 200: sprintf(code_det,"OK"); break; case 201: sprintf(code_det,"Created"); break; case 202: sprintf(code_det,"Accepted"); break; case 203: sprintf(code_det,"Non-Authoritative Information"); break; case 204: sprintf(code_det,"No Content"); break; case 205: sprintf(code_det,"Reset Content"); break; case 206: sprintf(code_det,"Partial Content"); break; case 207: sprintf(code_det,"Multi-Status"); break; case 208: sprintf(code_det,"Already Reported"); break; case 226: sprintf(code_det,"IM Used"); break; case 300: sprintf(code_det,"Multiple Choices"); break; case 301: sprintf(code_det,"Moved Permanently"); break; case 302: sprintf(code_det,"Found"); break; case 303: sprintf(code_det,"See Other"); break; case 304: sprintf(code_det,"Not Modified"); break; case 307: sprintf(code_det,"Temporary Redirect"); break; case 308: sprintf(code_det,"Permanent Redirect"); break; case 400: sprintf(code_det,"Bad Request"); break; case 401: sprintf(code_det,"Unauthorized"); break; case 402: sprintf(code_det,"Payment Required"); break; case 403: sprintf(code_det,"Forbidden"); break; case 404: sprintf(code_det,"Not Found"); break; case 405: sprintf(code_det,"Method Not Allowed"); break; case 406: sprintf(code_det,"Not Acceptable"); break; case 407: sprintf(code_det,"Proxy Authentication Required"); break; case 408: sprintf(code_det,"Request Timeout"); break; case 409: sprintf(code_det,"Conflict"); break; case 410: sprintf(code_det,"Gone"); break; case 411: sprintf(code_det,"Length Required"); break; case 412: sprintf(code_det,"Precondition Failed"); break; case 413: sprintf(code_det,"Content Too Large"); break; case 414: sprintf(code_det,"URI Too Long"); break; case 415: sprintf(code_det,"Unsupported Media Type"); break; case 416: sprintf(code_det,"Range Not Satisfiable"); break; case 417: sprintf(code_det,"Expectation Failed"); break; case 418: sprintf(code_det,"I'm a teapot"); break; case 421: sprintf(code_det,"Misdirected Request"); break; case 422: sprintf(code_det,"Unprocessable Content"); break; case 423: sprintf(code_det,"Locked"); break; case 424: sprintf(code_det,"Failed Dependency"); break; case 425: sprintf(code_det,"Too Early"); break; case 426: sprintf(code_det,"Upgrade Required"); break; case 428: sprintf(code_det,"Precondition Required"); break; case 429: sprintf(code_det,"Too Many Requests"); break; case 431: sprintf(code_det,"Request Header Fields Too Large"); break; case 451: sprintf(code_det,"Unavailable For Legal Reasons"); break; case 500: sprintf(code_det,"Internal Server Error"); break; case 501: sprintf(code_det,"Not Implemented"); break; case 502: sprintf(code_det,"Bad Gateway"); break; case 503: sprintf(code_det,"Service Unavailable"); break; case 504: sprintf(code_det,"Gateway Timeout"); break; case 505: sprintf(code_det,"HTTP Version Not Supported"); break; case 506: sprintf(code_det,"Variant Also Negotiates"); break; case 507: sprintf(code_det,"Insufficient Storage"); break; case 508: sprintf(code_det,"Loop Detected"); break; case 510: sprintf(code_det,"Not Extended"); break; case 511: sprintf(code_det,"Network Authentication Required"); break; default: sprintf(code_det,"unknown"); } } 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; } void client_fd_errors(int client_fd){ if(client_fd>=0) return; switch(client_fd){ case -1: p_fatal("client fd already closed\n"); case -2: p_fatal("request was partial\n"); default: p_fatal("unknown negative client_fd value"); } } int l_write(lua_State* L){ int res_idx = 1; lua_pushvalue(L, 1); lua_pushstring(L, "_request"); lua_gettable(L, -2); int head = strcmp(luaL_checkstring(L, -1), "HEAD") == 0; lua_pushvalue(L, res_idx); lua_pushstring(L, "client_fd"); lua_gettable(L, res_idx); int client_fd = luaL_checkinteger(L, -1); client_fd_errors(client_fd); size_t len; char* content = (char*)luaL_checklstring(L, 2, &len); lua_pushvalue(L, res_idx); lua_pushstring(L, "header"); lua_gettable(L, -2); int header_top = lua_gettop(L); lua_pushstring(L, "_sent"); lua_gettable(L, -2); str* resp; if(lua_isnil(L, -1)){ if(head) i_write_header(L, header_top, &resp, "", 0); else i_write_header(L, header_top, &resp, content, len); lua_pushvalue(L, header_top); lua_pushstring(L, "_sent"); lua_pushinteger(L, 1); lua_settable(L, -3); } else { if(head) return 0; resp = str_init(content); } send(client_fd, resp->c, resp->len, 0); str_free(resp); return 0; } int l_send(lua_State* L){ int res_idx = 1; lua_pushvalue(L, res_idx); lua_pushstring(L, "client_fd"); lua_gettable(L, res_idx); int client_fd = luaL_checkinteger(L, -1); client_fd_errors(client_fd); size_t len; char* content = (char*)luaL_checklstring(L, 2, &len); lua_pushvalue(L, res_idx); lua_pushstring(L, "header"); lua_gettable(L, -2); int header = lua_gettop(L); str* resp; lua_pushvalue(L, 1); lua_pushstring(L, "_request"); lua_gettable(L, -2); if(strcmp(luaL_checkstring(L, -1), "HEAD") == 0){ i_write_header(L, header, &resp, "", 0); } else i_write_header(L, header, &resp, content, len); int a = send(client_fd, resp->c, resp->len, 0); // lua_pushstring(L, "client_fd"); lua_pushinteger(L, -1); lua_settable(L, res_idx); closesocket(client_fd); //printf("%i | %i\n'%s'\n%i\n",client_fd,a,resp->c,resp->len); str_free(resp); return 0; } int l_close(lua_State* L){ int res_idx = 1; lua_pushvalue(L, res_idx); lua_pushstring(L, "client_fd"); lua_gettable(L, res_idx); int client_fd = luaL_checkinteger(L, -1); client_fd_errors(client_fd); lua_pushstring(L, "client_fd"); lua_pushinteger(L, -1); lua_settable(L, res_idx); closesocket(client_fd); return 0; } int l_serve(lua_State* L){ int res_idx = 1; lua_pushvalue(L, res_idx); lua_pushstring(L, "client_fd"); lua_gettable(L, res_idx); int client_fd = luaL_checkinteger(L, -1); client_fd_errors(client_fd); char* path = (char*)luaL_checkstring(L, 2); //continue here return 0; } int content_disposition(str* src, parray_t** _dest){ char* end = strnstr(src->c, ";", src->len); parray_t* dest = parray_init(); if(end == NULL){ parray_set(dest, "form-data", (void*)str_init(src->c)); return 0; } str* temp = str_init(""); str_pushl(temp, src->c, end - src->c); parray_set(dest, "form-data", (void*)temp); int len = end - src->c; char* buffer = end + 1; gen_parse(buffer, src->len - len, &dest); //printf("\n**\n"); //for(int i = 0; i != dest->len; i++){ // printf("'%s : %s'\n",((str*)dest->P[i].key)->c,((str*)dest->P[i].value)->c); //} *_dest = dest; return 1; } enum file_status { _ignore, BARRIER_READ, FILE_HEADER, FILE_BODY, NORMAL }; /** * @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, parray_t** _content){ parray_t* 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; if(status == NULL){ 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(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(boundary, "\r\n\r\n", 4); } status = malloc(sizeof * status); *status = state==2?BARRIER_READ:NORMAL; dash_count = malloc(sizeof * dash_count); *dash_count = 0; current = str_init(boundary->c); table_idx = 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)); boundary_id = str_init(""); str_popb(boundary, 4); parray_set(content, "_boundary", (void*)boundary); parray_set(content, "_boundary_id", (void*)boundary_id); } if(*status == NORMAL){ //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: if(*status == BARRIER_READ){ for(int i = 0; i != blen; i++){ if(*buffer == '\r'){ *status = FILE_HEADER; buffer+=2; blen-=i+2; str_clear(current); *table_idx = lua_rawlen(L, *files_idx) + 1; lua_pushinteger(L, *table_idx); lua_newtable(L); lua_settable(L, *files_idx); break; } str_pushl(boundary_id, buffer, 1); buffer++; } } lua_pushvalue(L, *files_idx); lua_pushinteger(L, *table_idx); lua_gettable(L, -2); int rfiles_idx = lua_gettop(L); if(*status == FILE_HEADER){ for(int i = 0; i < blen; i++){ if(buffer[i] == ':'){ old = current; current = str_init(""); } else if(buffer[i] == '\n'){ if(current->len == 0){ *status = FILE_BODY; buffer += i; blen -= i; old = NULL; str_free(current); current = str_init(""); break; } //printf("%i '%s' : '%s'\n",*table_idx, old->c, current->c); luaI_tsets(L, rfiles_idx, old->c, current->c); old = NULL; str_clear(current); } else if(buffer[i] != '\r' && !(buffer[i] == ' ' && current->len == 0)) str_pushl(current, buffer + i, 1); } } if(*status == FILE_BODY){ if(old==NULL) old = str_init(""); for(int i = 0; i < blen; i++){ if(boundary->c[old->len - *dash_count] == buffer[i] || buffer[i] == '-'){ str_pushl(old, buffer + i, 1); if(buffer[i] == '-') (*dash_count)++; if(old->len - *dash_count >= boundary->len){ luaI_tsets(L, rfiles_idx, "content", current->c); /*lua_pushinteger(L, lua_rawlen(L, *files_idx) + 1); lua_pushvalue(L, *table_idx); lua_settable(L, *files_idx);*/ for(; i < blen; i++) if(buffer[i] == '\n') break; str_clear(current); *status = BARRIER_READ; buffer+=i; blen-=i; *dash_count = 0; goto file_start; break; }; } else { if(old->len != 0 || *dash_count != 0){ str_pushl(current, old->c, old->len); str_clear(old); *dash_count = 0; } str_pushl(current, buffer + i, 1); } } } } 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; } int l_roll(lua_State* L){ lua_pushvalue(L, 1); lua_pushstring(L, "_data"); lua_gettable(L, -2); parray_t* data = (void*)lua_topointer(L, -1); lua_pushvalue(L, 1); lua_pushstring(L, "client_fd"); lua_gettable(L, -2); int client_fd = luaL_checkinteger(L, -1); client_fd_errors(client_fd); fd_set rfd; FD_ZERO(&rfd); FD_SET(client_fd, &rfd); if(select(client_fd+1, &rfd, NULL, NULL, &((struct timeval){.tv_sec = 0, .tv_usec = 0})) == 0){ lua_pushinteger(L, 0); return 1; } int alen = luaL_checkinteger(L, 2); char* buffer = malloc(alen * sizeof * buffer); int r = recv(client_fd, buffer, alen, 0); if(r <= 0){ lua_pushinteger(L, r); return 1; } lua_pushstring(L, "Body"); lua_gettable(L, 1); int body_idx = lua_gettop(L); lua_pushstring(L, "files"); lua_gettable(L, 1); int files_idx = lua_gettop(L); rolling_file_parse(L, &files_idx, &body_idx, buffer, NULL, r, &data); luaI_tsetv(L, 1, "Body", body_idx); luaI_tsetv(L, 1, "files", files_idx); free(buffer); lua_pushinteger(L, r); return 1; } volatile size_t threads = 0; void* handle_client(void *_arg){ //printf("--\n"); clock_t begin = clock(); //pthread_mutex_lock(&mutex); int read_state = 0; thread_arg_struct* args = (thread_arg_struct*)_arg; int client_fd = args->fd; char* buffer; char dummy[2] = {0, 0}; int header_eof; //sleep(1); //create state for this thread lua_State* L = luaL_newstate(); luaL_openlibs(L); pthread_mutex_lock(&mutex); int old_top = lua_gettop(args->L); lua_getglobal(args->L, "_G"); i_dcopy(args->L, L, NULL); lua_settop(args->L, old_top); //l_pprint(L); lua_setglobal(L, "_G"); pthread_mutex_unlock(&mutex); //printf("start: %f\n",(double)(clock() - begin) / CLOCKS_PER_SEC); begin = clock(); //read full request int64_t bytes_received = recv_full_buffer(client_fd, &buffer, &header_eof, &read_state); //printf("read\n"); //for(int i = 0; i != bytes_received; i++) putchar(buffer[i]); //putchar('\n'); //printf("read bytes: %li, %f\n",bytes_received,(double)(clock() - begin) / CLOCKS_PER_SEC); begin = clock(); //ignore if header is just fucked if(bytes_received >= -1){ parray_t* table; //checks for a valid header //printf("before header\n"); if(parse_header(buffer, header_eof, &table) != -1){ //printf("end of header\n"); //printf("parsed: %f\n",(double)(clock() - begin) / CLOCKS_PER_SEC); begin = clock(); str* sk = (str*)parray_get(table, "Path"); str* sR = (str*)parray_get(table, "Request"); str* sT = (str*)parray_get(table, "Content-Type"); str* sC = (str*)parray_get(table, "Cookie"); int some = bytes_received - header_eof - 10; parray_t* file_cont = parray_init(); //printf("'%s'\n\n",buffer); lua_newtable(L); int files_idx = lua_gettop(L); lua_pushstring(L, ""); int body_idx = lua_gettop(L); //printf("before\n"); if(sT != NULL) rolling_file_parse(L, &files_idx, &body_idx, buffer + header_eof + 4, sT, bytes_received - header_eof - 4, &file_cont); //printf("after\n"); //printf("&"); //rolling_file_parse(L, buffer + header_eof + 4 + 1200, sT, 300, &file_cont); //rolling_file_parse(L, buffer + header_eof + 4 + 900, sT, bytes_received - header_eof - 4 - 900, &file_cont); //rolling_file_parse(L, buffer + header_eof + 4 + 300, sT, 100, &file_cont); char portc[10] = {0}; sprintf(portc, "%i", args->port); str* aa = str_init(portc); str_push(aa, sk->c); void* v = parray_find(paths, aa->c); //printf("found: %f\n",(double)(clock() - begin) / CLOCKS_PER_SEC); begin = clock(); str_free(aa); if(v != NULL){ lua_newtable(L); int req_idx = lua_gettop(L); lua_newtable(L); int res_idx = lua_gettop(L); //handle cookies if(sC != NULL){ lua_newtable(L); int lcookie = lua_gettop(L); parray_t* cookie = parray_init(); //printf("%i\n",gen_parse(sC->c, sC->len, &cookie)); for(int i = 0; i != cookie->len; i++){ //printf("%s %s\n", cookie->P[i].key->c, ((str*)cookie->P[i].value)->c); luaI_tsetsl(L, lcookie, cookie->P[i].key->c, ((str*)cookie->P[i].value)->c, ((str*)cookie->P[i].value)->len); } luaI_tsetv(L, req_idx, "cookies", lcookie); parray_clear(cookie, STR); str_free(sC); parray_remove(table, "Cookie", NONE); } /* //handle files if(sT != NULL && bytes_received > 0){ int pf = file_parse(L, buffer + header_eof, sT, bytes_received - header_eof); if(pf >= 0){ luaI_tsetv(L, req_idx, "files", pf); parray_set(table, "Body", (void*)str_init("")); } }*/ lua_pushlightuserdata(L, file_cont); int ld = lua_gettop(L); luaI_tsetv(L, req_idx, "_data", ld); luaI_tsetv(L, req_idx, "files", files_idx); //printf("cookie and file: %f\n",(double)(clock() - begin) / CLOCKS_PER_SEC); //parray_set(table, "Body", (void*)str_initl(buffer + header_eof + 4, buffer_len - header_eof - 4)); //luaI_tsetsl(L, req_idx, "Body", buffer + header_eof + 4, bytes_received - header_eof - 4); luaI_tsetv(L, req_idx, "Body", body_idx); //printf("%s\n",buffer); begin = clock(); for(int i = 0; i != table->len; i+=1){ //printf("'%s' :: '%s'\n",table[i]->c, table[i+1]->c); luaI_tsets(L, req_idx, table->P[i].key->c, ((str*)table->P[i].value)->c); } luaI_tsets(L, req_idx, "ip", inet_ntoa(args->cli.sin_addr)); if(bytes_received == -1){ client_fd = -2; } luaI_tsetb(L, req_idx, "partial", read_state == 1); luaI_tseti(L, req_idx, "_bytes", bytes_received); luaI_tseti(L, req_idx, "client_fd", client_fd); luaI_tsetcf(L, req_idx, "roll", l_roll); //luaI_tsetcf(L, req_idx, "continue", l_continue); //functions luaI_tsetcf(L, res_idx, "send", l_send); //luaI_tsetcf(L, res_idx, "serve", l_serve); luaI_tsetcf(L, res_idx, "write", l_write); luaI_tsetcf(L, res_idx, "close", l_close); //values luaI_tseti(L, res_idx, "client_fd", client_fd); luaI_tsets(L, res_idx, "_request", sR->c); //header table lua_newtable(L); int header_idx = lua_gettop(L); luaI_tseti(L, header_idx, "Code", 200); luaI_tsets(L, header_idx, "Content-Type", "text/html"); luaI_tsetv(L, res_idx, "header", header_idx); //printf("wrote table: %f\n",(double)(clock() - begin) / CLOCKS_PER_SEC); begin = clock(); //the function(s) //get all function that kinda match parray_t* owo = (parray_t*)v; uint64_t passes = 0; for(int i = 0; i != owo->len; i++){ //though these are arrays of arrays we have to iterate *again* struct sarray_t* awa = (struct sarray_t*)owo->P[i].value; for(int z = 0; z != awa->len; z++){ char* path; struct lchar* wowa = awa->cs[z]; if(strcmp(wowa->req, "all") == 0 || strcmp(wowa->req, sR->c) == 0 || (strcmp(sR->c, "HEAD") && strcmp(wowa->req, "GET"))){ luaI_tseti(L, res_idx, "passes", passes); passes++; luaL_loadbuffer(L, wowa->c, wowa->len, wowa->c); int func = lua_gettop(L); lua_pushvalue(L, func); // push function call lua_pushvalue(L, res_idx); //push methods related to dealing with the request lua_pushvalue(L, req_idx); //push info about the request //call the function lua_call(L, 2, 0); } } } parray_lclear(owo); //dont free the rest //printf("out: %f\n",(double)(clock() - begin) / CLOCKS_PER_SEC); begin = clock(); lua_pushstring(L, "client_fd"); lua_gettable(L, res_idx); client_fd = luaL_checkinteger(L, -1); } void* awa; if((awa = parray_get(file_cont, "_current")) != NULL) str_free(awa); if((awa = parray_get(file_cont, "_boundary")) != NULL) str_free(awa); if((awa = parray_get(file_cont, "_boundary_id")) != NULL) str_free(awa); if((awa = parray_get(file_cont, "_table_idx")) != NULL) free(awa); if((awa = parray_get(file_cont, "_status")) != NULL) free(awa); if((awa = parray_get(file_cont, "_dash_count")) != NULL) free(awa); parray_clear(file_cont, NONE); } //printf("end anyways\n"); parray_clear(table, STR); } shutdown(client_fd, 2); close(client_fd); free(args); free(buffer); lua_close(L); //printf("closed: %f\n",(double)(clock() - begin) / CLOCKS_PER_SEC); pthread_mutex_lock(&mutex); threads--; pthread_mutex_unlock(&mutex); //printf("out\n"); return NULL; } int start_serv(lua_State* L, int port){ //need these on windows for sockets (stupid) #ifdef _WIN32 WSADATA Data; WSAStartup(MAKEWORD(2, 2), &Data); #endif int server_fd; struct sockaddr_in server_addr; //open the socket if((server_fd = socket(AF_INET, SOCK_STREAM, 0)) < 0) p_fatal("error opening socket\n"); server_addr.sin_family = AF_INET; server_addr.sin_addr.s_addr = INADDR_ANY; server_addr.sin_port = htons(port); //bind to port if(bind(server_fd, (struct sockaddr*)&server_addr, sizeof(server_addr)) < 0) p_fatal("failed to bind to port\n"); if(listen(server_fd, max_con) < 0) p_fatal("failed to listen\n"); if (pthread_mutex_init(&mutex, NULL) != 0) p_fatal("mutex init failed\n"); if (setsockopt(server_fd, SOL_SOCKET, SO_REUSEADDR, (const char*)&(int){1}, sizeof(int)) < 0) p_fatal("SO_REUSEADDR refused\n"); for(;;){ struct sockaddr_in client_addr; socklen_t client_addr_len = sizeof(client_addr); int* client_fd = malloc(sizeof(int)); if((*client_fd = accept(server_fd, (struct sockaddr*)&client_addr, &client_addr_len)) < 0) p_fatal("failed to accept\n"); //open a state to call shit, should be somewhat thread safe thread_arg_struct* args = malloc(sizeof * args); args->fd = *client_fd; args->port = port; args->cli = client_addr; args->L = L; pthread_mutex_lock(&mutex); threads++; pthread_mutex_unlock(&mutex); //send request to handle_client() pthread_t thread_id; pthread_create(&thread_id, NULL, handle_client, (void*)args); pthread_detach(thread_id); //handle_client((void*)args); free(client_fd); } } int l_req_com(lua_State* L, char* req){ lua_pushstring(L, "port"); lua_gettable(L, 1); int port = luaL_checkinteger(L, -1); char portc[10] = {0}; sprintf(portc, "%i", port);//, lua_tostring(L, 2)); str* portss = str_init(portc); str_push(portss, (char*)lua_tostring(L, 2)); struct lchar* awa; str* uwu = str_init(""); lua_pushvalue(L, 3); lua_dump(L, writer, (void*)uwu, 0); lua_pushlstring(L, uwu->c, uwu->len); str_free(uwu); size_t len; char* a = (char*)luaL_checklstring(L, -1, &len); awa = malloc(len + 1); awa->c = a; awa->len = len; strcpy(awa->req, req); if(paths == NULL) paths = parray_init(); //please free this void* v_old_paths = parray_get(paths, portss->c); struct sarray_t* old_paths; if(v_old_paths == NULL){ old_paths = malloc(sizeof * old_paths); old_paths->len = 0; old_paths->cs = malloc(sizeof * old_paths->cs); } else old_paths = (struct sarray_t*)v_old_paths; old_paths->len++; old_paths->cs = realloc(old_paths->cs, sizeof * old_paths->cs * old_paths->len); old_paths->cs[old_paths->len - 1] = awa; parray_set(paths, portss->c, (void*)old_paths); str_free(portss); return 1; } #define gen_reqs(T)\ int l_##T##q (lua_State* L){\ l_req_com(L, #T);\ return 1;\ } gen_reqs(GET); gen_reqs(HEAD); gen_reqs(POST); gen_reqs(PUT); gen_reqs(DELETE); gen_reqs(CONNECT); gen_reqs(OPTIONS); gen_reqs(TRACE); gen_reqs(PATCH); gen_reqs(all); //non standard lol, like expressjs 'use' keyword :3 int l_lock(lua_State* L){ pthread_mutex_lock(&lua_mutex); return 0; } int l_unlock(lua_State* L){ pthread_mutex_unlock(&lua_mutex); return 0; } int l_listen(lua_State* L){ lua_State* src = luaL_newstate(); lua_State* dest = luaL_newstate(); if(lua_gettop(L) != 2) { p_fatal("not enough args"); } if(lua_type(L, 1) != LUA_TFUNCTION) { p_fatal("(arg:1) expected a function"); } int port = luaL_checkinteger(L, 2); lua_newtable(L); int mt = lua_gettop(L); luaI_tsetcf(L, mt, "GET", l_GETq); luaI_tsetcf(L, mt, "HEAD", l_HEADq); luaI_tsetcf(L, mt, "POST", l_POSTq); luaI_tsetcf(L, mt, "PUT", l_PUTq); luaI_tsetcf(L, mt, "DELETE", l_DELETEq); luaI_tsetcf(L, mt, "CONNECT", l_CONNECTq); luaI_tsetcf(L, mt, "OPTIONS", l_OPTIONSq); luaI_tsetcf(L, mt, "TRACE", l_TRACEq); luaI_tsetcf(L, mt, "PATCH", l_PATCHq); luaI_tsetcf(L, mt, "all", l_allq); lua_pushstring(L, "lock"); lua_pushcfunction(L, l_lock); lua_settable(L, -3); lua_pushstring(L, "unlock"); lua_pushcfunction(L, l_unlock); lua_settable(L, -3); lua_pushstring(L, "port"); lua_pushvalue(L, 2); lua_settable(L, -3); lua_pushvalue(L, 1); //the function lua_pushvalue(L, -2); //the server table lua_pcall(L, 1, 0, 0); start_serv(L, port); return 0; }