Lab
/
libuhttpd
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

example: Improve code 

Signed-off-by: Jianhui Zhao <zhaojh329@gmail.com>
main^2
Jianhui Zhao 2021-01-15 15:35:35 +08:00
parent afeee7ec2f
commit b0d35ed4fb
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
example/handler.c
View File

@ -61,8 +61,6 @@ void echo_handler(struct uh_connection *conn, int event)
void upload_handler(struct uh_connection *conn, int event) void upload_handler(struct uh_connection *conn, int event)
{ {
static int fd = -1;
if (event == UH_EV_HEAD_COMPLETE) { if (event == UH_EV_HEAD_COMPLETE) {
struct uh_str str = conn->get_header(conn, "Content-Length"); struct uh_str str = conn->get_header(conn, "Content-Length");
int content_length; int content_length;
@ -77,8 +75,11 @@ void upload_handler(struct uh_connection *conn, int event)
return; return;
} }
conn->userdata = (void *)(intptr_t)-1;
} if (event == UH_EV_BODY) { } if (event == UH_EV_BODY) {
struct uh_str body = conn->extract_body(conn); struct uh_str body = conn->extract_body(conn);
int fd = (intptr_t)conn->userdata;
if (fd < 0) { if (fd < 0) {
fd = open("upload.bin", O_RDWR | O_CREAT | O_TRUNC, 0644); fd = open("upload.bin", O_RDWR | O_CREAT | O_TRUNC, 0644);
@ -93,7 +94,10 @@ void upload_handler(struct uh_connection *conn, int event)
close(fd); close(fd);
return; return;
} }
conn->userdata = (void *)(intptr_t)fd;
} else if (event == UH_EV_COMPLETE) { } else if (event == UH_EV_COMPLETE) {
int fd = (intptr_t)conn->userdata;
struct stat st; struct stat st;
size_t size = 0; size_t size = 0;