diff --git a/http_parser.c b/http_parser.c index 5e0a950..542a875 100644 --- a/http_parser.c +++ b/http_parser.c @@ -2175,3 +2175,8 @@ http_parser_pause(http_parser *parser, int paused) { assert(0 && "Attempting to pause parser in error state"); } } + +int +http_body_is_final(const struct http_parser *parser) { + return parser->state == s_message_done; +} diff --git a/http_parser.h b/http_parser.h index 7908ad0..d3f60f9 100644 --- a/http_parser.h +++ b/http_parser.h @@ -313,6 +313,9 @@ int http_parser_parse_url(const char *buf, size_t buflen, /* Pause or un-pause the parser; a nonzero value pauses */ void http_parser_pause(http_parser *parser, int paused); +/* Checks if this is the final chunk of the body. */ +int http_body_is_final(const http_parser *parser); + #ifdef __cplusplus } #endif diff --git a/test.c b/test.c index 81e0c3b..d205c0d 100644 --- a/test.c +++ b/test.c @@ -67,6 +67,7 @@ struct message { int headers_complete_cb_called; int message_complete_cb_called; int message_complete_on_eof; + int body_is_final; }; static int currently_parsing_eof; @@ -1449,12 +1450,26 @@ header_value_cb (http_parser *p, const char *buf, size_t len) return 0; } +void +check_body_is_final (const http_parser *p) +{ + if (messages[num_messages].body_is_final) { + fprintf(stderr, "\n\n *** Error http_body_is_final() should return 1 " + "on last on_body callback call " + "but it doesn't! ***\n\n"); + assert(0); + abort(); + } + messages[num_messages].body_is_final = http_body_is_final(p); +} + int body_cb (http_parser *p, const char *buf, size_t len) { assert(p == parser); strncat(messages[num_messages].body, buf, len); messages[num_messages].body_size += len; + check_body_is_final(p); // printf("body_cb: '%s'\n", requests[num_messages].body); return 0; } @@ -1465,6 +1480,7 @@ count_body_cb (http_parser *p, const char *buf, size_t len) assert(p == parser); assert(buf); messages[num_messages].body_size += len; + check_body_is_final(p); return 0; } @@ -1501,6 +1517,18 @@ message_complete_cb (http_parser *p) assert(0); abort(); } + + if (messages[num_messages].body_size && + http_body_is_final(p) && + !messages[num_messages].body_is_final) + { + fprintf(stderr, "\n\n *** Error http_body_is_final() should return 1 " + "on last on_body callback call " + "but it doesn't! ***\n\n"); + assert(0); + abort(); + } + messages[num_messages].message_complete_cb_called = TRUE; messages[num_messages].message_complete_on_eof = currently_parsing_eof;