1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
|
#include "luai.h"
#include "common.h"
#include "util.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);
const char* code_det = http_code(code);
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 http_body_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;
if(content.status == _ignore){
parray_t* par = parray_init();
gen_parse(content_type->c, content_type->len, &par);
content.boundary = parray_pop(par, "boundary");
content.status = content.boundary != NULL?BARRIER_START:NORMAL;
content.current = str_init("");
content.line = str_init("");
parray_clear(par, STR);
}
if(content.status == NORMAL){
lua_pushvalue(L, *body_idx);
lua_pushlstring(L, buffer, blen);
lua_concat(L, 2);
*body_idx = lua_gettop(L);
} else {
file_start:;
if(content.status == BARRIER_START){
for(; blen > 0; buffer++, blen--){
if(*buffer == '\n'){
content.status = BARRIER_END;
blen--;
buffer++;
break;
}
}
}
if(content.status == BARRIER_END){
int check = 80;
if(content.current->len < check) check = content.current->len;
ssize_t backtrack = content.current->len - check;
str_pushl(content.current, buffer, blen);
char* end = memmem(content.current->c + backtrack, content.current->len - backtrack, content.boundary->c, content.boundary->len);
if(end != NULL){
ssize_t size = content.current->len - (end - content.current->c + 4);
str_popb(content.current, content.current->len - (end - content.current->c) + 4);
size_t header_len = (char*)memmem(content.current->c, content.current->len, "\r\n\r\n", 4) - content.current->c;
parray_t* header = parray_init();
parse_header_kv(content.current->c, header_len, header);
str* cd = (str*)parray_get(header, "content-disposition");
if(cd != NULL){
lua_newtable(L);
int newfile_idx = lua_gettop(L);
parray_t* cd_parse = parray_init();
gen_parse(cd->c, cd->len, &cd_parse);
lua_newtable(L);
int cdidx = lua_gettop(L);
luaI_fromparray(L, cdidx, cd_parse, 1);
luaI_fromparray(L, newfile_idx, header, 1);
luaI_tsetv(L, newfile_idx, "content-disposition", cdidx);
str* name = (str*)parray_get(cd_parse, "name");
if(name == NULL) name = (str*)parray_get(cd_parse, "filename");
if(name == NULL){
int ind = lua_objlen(L, *files_idx) + 1;
lua_pushinteger(L, ind);
lua_pushvalue(L, newfile_idx);
lua_settable(L, *files_idx);
}
else {
luaI_tsetv(L, *files_idx, name->c, newfile_idx);
}
parray_clear(cd_parse, STR);
lua_pushlstring(L, content.current->c + header_len + 4, content.current->len - header_len - 4);
lua_setfield(L, newfile_idx, "body");
}
str_clear(content.current);
parray_clear(header, STR);
content.status = BARRIER_START;
ssize_t oblen = blen;
blen = oblen - (oblen - size);
buffer += oblen - blen;
goto file_start;
}
}
}
*_content = content;
return 0;
}
|
