diff --git a/plugin/protocol/http/test/test_http_half.cpp b/plugin/protocol/http/test/test_http_half.cpp index 2dbec29..d09c693 100644 --- a/plugin/protocol/http/test/test_http_half.cpp +++ b/plugin/protocol/http/test/test_http_half.cpp @@ -1360,6 +1360,126 @@ TEST_F(HttpHalfResponseParser, DeflateResponseStream) EXPECT_EQ(ret, 0); } +unsigned char __240_Response_gzip_chunk_txt[] = { + 0x48, 0x54, 0x54, 0x50, 0x2f, 0x31, 0x2e, 0x31, 0x20, 0x34, 0x30, 0x31, + 0x20, 0x55, 0x6e, 0x61, 0x75, 0x74, 0x68, 0x6f, 0x72, 0x69, 0x7a, 0x65, + 0x64, 0x0d, 0x0a, 0x43, 0x6f, 0x6e, 0x74, 0x65, 0x6e, 0x74, 0x2d, 0x54, + 0x79, 0x70, 0x65, 0x3a, 0x20, 0x74, 0x65, 0x78, 0x74, 0x2f, 0x68, 0x74, + 0x6d, 0x6c, 0x3b, 0x20, 0x63, 0x68, 0x61, 0x72, 0x73, 0x65, 0x74, 0x3d, + 0x55, 0x54, 0x46, 0x2d, 0x38, 0x0d, 0x0a, 0x57, 0x57, 0x57, 0x2d, 0x41, + 0x75, 0x74, 0x68, 0x65, 0x6e, 0x74, 0x69, 0x63, 0x61, 0x74, 0x65, 0x3a, + 0x20, 0x42, 0x41, 0x53, 0x49, 0x43, 0x20, 0x72, 0x65, 0x61, 0x6c, 0x6d, + 0x3d, 0x22, 0x6d, 0x61, 0x69, 0x6c, 0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, + 0x65, 0x2e, 0x63, 0x6f, 0x6d, 0x22, 0x0d, 0x0a, 0x43, 0x6f, 0x6e, 0x74, + 0x65, 0x6e, 0x74, 0x2d, 0x45, 0x6e, 0x63, 0x6f, 0x64, 0x69, 0x6e, 0x67, + 0x3a, 0x20, 0x67, 0x7a, 0x69, 0x70, 0x0d, 0x0a, 0x44, 0x61, 0x74, 0x65, + 0x3a, 0x20, 0x54, 0x68, 0x75, 0x2c, 0x20, 0x32, 0x37, 0x20, 0x53, 0x65, + 0x70, 0x20, 0x32, 0x30, 0x31, 0x38, 0x20, 0x30, 0x37, 0x3a, 0x32, 0x35, + 0x3a, 0x33, 0x30, 0x20, 0x47, 0x4d, 0x54, 0x0d, 0x0a, 0x45, 0x78, 0x70, + 0x69, 0x72, 0x65, 0x73, 0x3a, 0x20, 0x54, 0x68, 0x75, 0x2c, 0x20, 0x32, + 0x37, 0x20, 0x53, 0x65, 0x70, 0x20, 0x32, 0x30, 0x31, 0x38, 0x20, 0x30, + 0x37, 0x3a, 0x32, 0x35, 0x3a, 0x33, 0x30, 0x20, 0x47, 0x4d, 0x54, 0x0d, + 0x0a, 0x43, 0x61, 0x63, 0x68, 0x65, 0x2d, 0x43, 0x6f, 0x6e, 0x74, 0x72, + 0x6f, 0x6c, 0x3a, 0x20, 0x70, 0x72, 0x69, 0x76, 0x61, 0x74, 0x65, 0x2c, + 0x20, 0x6d, 0x61, 0x78, 0x2d, 0x61, 0x67, 0x65, 0x3d, 0x30, 0x0d, 0x0a, + 0x58, 0x2d, 0x43, 0x6f, 0x6e, 0x74, 0x65, 0x6e, 0x74, 0x2d, 0x54, 0x79, + 0x70, 0x65, 0x2d, 0x4f, 0x70, 0x74, 0x69, 0x6f, 0x6e, 0x73, 0x3a, 0x20, + 0x6e, 0x6f, 0x73, 0x6e, 0x69, 0x66, 0x66, 0x0d, 0x0a, 0x58, 0x2d, 0x46, + 0x72, 0x61, 0x6d, 0x65, 0x2d, 0x4f, 0x70, 0x74, 0x69, 0x6f, 0x6e, 0x73, + 0x3a, 0x20, 0x53, 0x41, 0x4d, 0x45, 0x4f, 0x52, 0x49, 0x47, 0x49, 0x4e, + 0x0d, 0x0a, 0x58, 0x2d, 0x58, 0x53, 0x53, 0x2d, 0x50, 0x72, 0x6f, 0x74, + 0x65, 0x63, 0x74, 0x69, 0x6f, 0x6e, 0x3a, 0x20, 0x31, 0x3b, 0x20, 0x6d, + 0x6f, 0x64, 0x65, 0x3d, 0x62, 0x6c, 0x6f, 0x63, 0x6b, 0x0d, 0x0a, 0x53, + 0x65, 0x72, 0x76, 0x65, 0x72, 0x3a, 0x20, 0x47, 0x53, 0x45, 0x0d, 0x0a, + 0x54, 0x72, 0x61, 0x6e, 0x73, 0x66, 0x65, 0x72, 0x2d, 0x45, 0x6e, 0x63, + 0x6f, 0x64, 0x69, 0x6e, 0x67, 0x3a, 0x20, 0x63, 0x68, 0x75, 0x6e, 0x6b, + 0x65, 0x64, 0x0d, 0x0a, 0x0d, 0x0a, 0x30, 0x30, 0x30, 0x30, 0x30, 0x30, + 0x30, 0x31, 0x0d, 0x0a, 0x1f, 0x0d, 0x0a, 0x30, 0x30, 0x30, 0x30, 0x30, + 0x30, 0x30, 0x31, 0x0d, 0x0a, 0x8b, 0x0d, 0x0a, 0x30, 0x30, 0x30, 0x30, + 0x30, 0x30, 0x30, 0x31, 0x0d, 0x0a, 0x08, 0x0d, 0x0a, 0x30, 0x30, 0x30, + 0x30, 0x30, 0x30, 0x30, 0x31, 0x0d, 0x0a, 0x00, 0x0d, 0x0a, 0x30, 0x30, + 0x30, 0x30, 0x30, 0x30, 0x30, 0x31, 0x0d, 0x0a, 0x00, 0x0d, 0x0a, 0x30, + 0x30, 0x30, 0x30, 0x30, 0x30, 0x30, 0x31, 0x0d, 0x0a, 0x00, 0x0d, 0x0a, + 0x30, 0x30, 0x30, 0x30, 0x30, 0x30, 0x30, 0x31, 0x0d, 0x0a, 0x00, 0x0d, + 0x0a, 0x30, 0x30, 0x30, 0x30, 0x30, 0x30, 0x30, 0x31, 0x0d, 0x0a, 0x00, + 0x0d, 0x0a, 0x30, 0x30, 0x30, 0x30, 0x30, 0x30, 0x30, 0x31, 0x0d, 0x0a, + 0x00, 0x0d, 0x0a, 0x30, 0x31, 0x0d, 0x0a, 0x00, 0x0d, 0x0a, 0x36, 0x66, + 0x0d, 0x0a, 0xb3, 0xf1, 0x08, 0xf1, 0xf5, 0xb1, 0xe3, 0xb2, 0xf1, 0x70, + 0x75, 0x74, 0x01, 0x52, 0x21, 0x9e, 0x21, 0x3e, 0xae, 0x76, 0xa1, 0x79, + 0x89, 0xa5, 0x25, 0x19, 0xf9, 0x45, 0x99, 0x55, 0xa9, 0x29, 0x36, 0xfa, + 0x10, 0x31, 0x2e, 0x1b, 0x7d, 0xa8, 0x1a, 0x27, 0x7f, 0x97, 0x48, 0x05, + 0x27, 0x77, 0x67, 0x7f, 0x1f, 0xff, 0x20, 0x5b, 0x25, 0x65, 0x37, 0x30, + 0x50, 0x52, 0x08, 0x71, 0x8d, 0x08, 0x01, 0x72, 0x0d, 0xc0, 0x40, 0x09, + 0x64, 0xa4, 0x21, 0x9a, 0x41, 0x40, 0x01, 0xa0, 0xa8, 0x91, 0x9d, 0x6b, + 0x51, 0x51, 0x7e, 0x91, 0x82, 0x89, 0x81, 0x21, 0x50, 0xc8, 0x08, 0x64, + 0x30, 0xc8, 0x44, 0xb0, 0x05, 0x60, 0xb7, 0x00, 0x00, 0xd7, 0x9d, 0xfb, + 0x97, 0x93, 0x00, 0x00, 0x00, 0x0d, 0x0a, 0x30, 0x0d, 0x0a, 0x0d, 0x0a +}; +unsigned int __240_Response_gzip_chunk_txt_len = 636; + +unsigned char __240_body[] = { + 0x3c, 0x48, 0x54, 0x4d, 0x4c, 0x3e, 0x0a, 0x3c, 0x48, 0x45, 0x41, 0x44, + 0x3e, 0x0a, 0x3c, 0x54, 0x49, 0x54, 0x4c, 0x45, 0x3e, 0x55, 0x6e, 0x61, + 0x75, 0x74, 0x68, 0x6f, 0x72, 0x69, 0x7a, 0x65, 0x64, 0x3c, 0x2f, 0x54, + 0x49, 0x54, 0x4c, 0x45, 0x3e, 0x0a, 0x3c, 0x2f, 0x48, 0x45, 0x41, 0x44, + 0x3e, 0x0a, 0x3c, 0x42, 0x4f, 0x44, 0x59, 0x20, 0x42, 0x47, 0x43, 0x4f, + 0x4c, 0x4f, 0x52, 0x3d, 0x22, 0x23, 0x46, 0x46, 0x46, 0x46, 0x46, 0x46, + 0x22, 0x20, 0x54, 0x45, 0x58, 0x54, 0x3d, 0x22, 0x23, 0x30, 0x30, 0x30, + 0x30, 0x30, 0x30, 0x22, 0x3e, 0x0a, 0x3c, 0x48, 0x31, 0x3e, 0x55, 0x6e, + 0x61, 0x75, 0x74, 0x68, 0x6f, 0x72, 0x69, 0x7a, 0x65, 0x64, 0x3c, 0x2f, + 0x48, 0x31, 0x3e, 0x0a, 0x3c, 0x48, 0x32, 0x3e, 0x45, 0x72, 0x72, 0x6f, + 0x72, 0x20, 0x34, 0x30, 0x31, 0x3c, 0x2f, 0x48, 0x32, 0x3e, 0x0a, 0x3c, + 0x2f, 0x42, 0x4f, 0x44, 0x59, 0x3e, 0x0a, 0x3c, 0x2f, 0x48, 0x54, 0x4d, + 0x4c, 0x3e, 0x0a +}; +unsigned int __240_body_len = 147; + +int __hf_callback_on_240(struct http_half_private * hf_private, + tfe_http_event ev, const unsigned char * data, size_t len, void * user) +{ + auto * vec_raw_content = (std::vector *) user; + if (ev & EV_HTTP_REQ_BODY_CONT) + { + vec_raw_content->insert(vec_raw_content->end(), data, data + len); + } + + if(ev & EV_HTTP_REQ_BODY_END) + { + EXPECT_EQ(vec_raw_content->size(), __240_body_len); + EXPECT_EQ(memcmp(vec_raw_content, __240_body, __240_body_len), 0); + } + + return 0; +} + +TEST_F(HttpHalfResponseParser, GzipChunkResponse) +{ + auto * __data = (const unsigned char *) __240_Response_gzip_chunk_txt; + size_t __datalen = sizeof(__240_Response_gzip_chunk_txt); + + hf_private_set_callback(hf_private, __hf_callback_on_240, &__ctx, NULL); + + /* No.17 test case do not contains a 'Content-Length', so + * parser cannot indintify the boundary of HTTP body */ + int ret = hf_private_parse(hf_private, __data, __datalen); + EXPECT_EQ(ret, 1); +} + +TEST_F(HttpHalfResponseParser, GzipChunkResponseStream) +{ + auto * __data = (const unsigned char *) __240_Response_gzip_chunk_txt; + size_t __datalen = sizeof(__240_Response_gzip_chunk_txt); + + hf_private_set_callback(hf_private, __hf_callback_on_240, &__ctx, NULL); + + auto half_len = __datalen / 2; + int ret = hf_private_parse(hf_private, __data, half_len); + EXPECT_EQ(ret, 0); + + ret = hf_private_parse(hf_private, __data, __datalen); + EXPECT_EQ(ret, 1); +} + void tfe_stream_write_access_log(const struct tfe_stream * stream, int level, const char * fmt, ...) { return;