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

Fix bug: Parsing error for fragment data 

Signed-off-by: zhaojh329 <zhaojh329@gmail.com>
main
zhaojh329 2020-09-07 00:40:12 +08:00
parent fbcf45ac36
commit 30729cf9e7
2 changed files with 32 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
src/connection.c
View File

@ -240,6 +240,16 @@ static const char *conn_get_body(struct uh_connection *conn, int *len)
return at;
}
static int on_message_begin_cb(struct http_parser *parser)
{
struct uh_connection *conn = (struct uh_connection *)parser->data;
struct uh_request *req = &conn->req;
req->last_was_header_value = true;
return 0;
}
static int on_url_cb(struct http_parser *parser, const char *at, size_t length)
{
struct uh_connection *conn = (struct uh_connection *)parser->data;
@ -259,15 +269,20 @@ static int on_header_field_cb(struct http_parser *parser, const char *at, size_t
{
struct uh_connection *conn = (struct uh_connection *)parser->data;
struct uh_request *req = &conn->req;
int n = req->header_num;
if (n == UHTTPD_MAX_HEADER_NUM) {
uh_log_err("Header too more\n");
return 1;
if (req->last_was_header_value) {
req->last_was_header_value = false;
req->header_num++;
if (req->header_num == UHTTPD_MAX_HEADER_NUM) {
uh_log_err("Header too more\n");
return 1;
}
req->headers_info[req->header_num - 1].name_offset = ROF(conn, at);
}
req->headers_info[n].name_offset = ROF(conn, at);
req->headers_info[n].name_len = length;
req->headers_info[req->header_num - 1].name_len += length;
return 0;
}
@ -276,12 +291,13 @@ static int on_header_value_cb(struct http_parser *parser, const char *at, size_t
{
struct uh_connection *conn = (struct uh_connection *)parser->data;
struct uh_request *req = &conn->req;
int n = req->header_num;
req->headers_info[n].value_offset = ROF(conn, at);
req->headers_info[n].value_len = length;
if (!req->last_was_header_value) {
req->last_was_header_value = true;
req->headers_info[req->header_num - 1].value_offset = ROF(conn, at);
}
req->header_num++;
req->headers_info[req->header_num - 1].value_len += length;
return 0;
}
@ -291,8 +307,9 @@ static int on_body_cb(struct http_parser *parser, const char *at, size_t length)
struct uh_connection *conn = (struct uh_connection *)parser->data;
struct uh_request *req = &conn->req;
req->body.offset = ROF(conn, at);
req->body.len = length;
if (req->body.offset == 0)
req->body.offset = ROF(conn, at);
req->body.len += length;
return 0;
}
@ -347,6 +364,7 @@ static int on_message_complete_cb(struct http_parser *parser)
}
static struct http_parser_settings settings = {
.on_message_begin = on_message_begin_cb,
.on_url = on_url_cb,
.on_header_field = on_header_field_cb,
.on_header_value = on_header_value_cb,

2
src/connection.h
View File

@ -56,10 +56,12 @@ struct uh_request {
int value_len;
} headers_info[UHTTPD_MAX_HEADER_NUM];
int header_num;
bool last_was_header_value;
struct {
char *name;
char *value;
} headers[UHTTPD_MAX_HEADER_NUM];
struct {
int offset;
int len;