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

Optimize code 

Signed-off-by: Jianhui Zhao <jianhuizhao329@gmail.com>
main
Jianhui Zhao 2017-11-13 21:31:58 +08:00
parent bd8ec5f31d
commit 6f9da0f7ce
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/uhttp.c
View File

@ -168,9 +168,10 @@ static int on_headers_complete(http_parser *parser)
struct uh_connection *con = container_of(parser, struct uh_connection, parser); struct uh_connection *con = container_of(parser, struct uh_connection, parser);
if (parser->method != HTTP_GET && parser->method != HTTP_POST) { if (parser->method != HTTP_GET && parser->method != HTTP_POST) {
uh_send_error(con, UH_STATUS_METHOD_NOT_ALLOWED, NULL); uh_send_error(con, UH_STATUS_NOT_IMPLEMENTED, NULL);
return -1; return -1;
} }
return 0; return 0;
} }
@ -182,6 +183,12 @@ static int on_body(http_parser *parser, const char *at, size_t len)
con->req.body.at = at; con->req.body.at = at;
con->req.body.len += len; con->req.body.len += len;
if (con->req.body.len > UH_BODY_SIZE_LIMIT) {
uh_send_error(con, UH_STATUS_PAYLOAD_TOO_LARGE, NULL);
return -1;
}
return 0; return 0;
} }