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

View File

@ -240,6 +240,16 @@ static const char *conn_get_body(struct uh_connection *conn, int *len)
return at; 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) static int on_url_cb(struct http_parser *parser, const char *at, size_t length)
{ {
struct uh_connection *conn = (struct uh_connection *)parser->data; 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_connection *conn = (struct uh_connection *)parser->data;
struct uh_request *req = &conn->req; struct uh_request *req = &conn->req;
int n = req->header_num;
if (n == UHTTPD_MAX_HEADER_NUM) { if (req->last_was_header_value) {
uh_log_err("Header too more\n"); req->last_was_header_value = false;
return 1; 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[req->header_num - 1].name_len += length;
req->headers_info[n].name_len = length;
return 0; 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_connection *conn = (struct uh_connection *)parser->data;
struct uh_request *req = &conn->req; struct uh_request *req = &conn->req;
int n = req->header_num;
req->headers_info[n].value_offset = ROF(conn, at); if (!req->last_was_header_value) {
req->headers_info[n].value_len = length; 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; 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_connection *conn = (struct uh_connection *)parser->data;
struct uh_request *req = &conn->req; struct uh_request *req = &conn->req;
req->body.offset = ROF(conn, at); if (req->body.offset == 0)
req->body.len = length; req->body.offset = ROF(conn, at);
req->body.len += length;
return 0; return 0;
} }
@ -347,6 +364,7 @@ static int on_message_complete_cb(struct http_parser *parser)
} }
static struct http_parser_settings settings = { static struct http_parser_settings settings = {
.on_message_begin = on_message_begin_cb,
.on_url = on_url_cb, .on_url = on_url_cb,
.on_header_field = on_header_field_cb, .on_header_field = on_header_field_cb,
.on_header_value = on_header_value_cb, .on_header_value = on_header_value_cb,

View File

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