diff --git a/common/include/tfe_http.h b/common/include/tfe_http.h index 14f8342..0a96344 100644 --- a/common/include/tfe_http.h +++ b/common/include/tfe_http.h @@ -96,10 +96,12 @@ enum http_ev_bit_number REQ_BODY_BEGIN_BITNUM, REQ_BODY_CONT_BITNUM, REQ_BODY_END_BITNUM, + REQ_END_BITNUM, RESP_HDR_BITNUM, RESP_BODY_BEGIN_BITNUM, RESP_BODY_CONT_BITNUM, - RESP_BODY_END_BITNUM + RESP_BODY_END_BITNUM, + RESP_END_BITNUM }; enum tfe_http_event @@ -108,13 +110,12 @@ enum tfe_http_event EV_HTTP_REQ_BODY_BEGIN = 1ULL << REQ_BODY_BEGIN_BITNUM, EV_HTTP_REQ_BODY_CONT = 1ULL << REQ_BODY_CONT_BITNUM, EV_HTTP_REQ_BODY_END = 1ULL << REQ_BODY_END_BITNUM, - EV_HTTP_REQ_BODY_FULL = EV_HTTP_REQ_BODY_BEGIN | EV_HTTP_REQ_BODY_CONT | EV_HTTP_REQ_BODY_END, - EV_HTTP_REQ_END = EV_HTTP_REQ_BODY_END, + EV_HTTP_REQ_END = 1ULL << REQ_END_BITNUM, EV_HTTP_RESP_HDR = 1ULL << RESP_HDR_BITNUM, EV_HTTP_RESP_BODY_BEGIN = 1ULL << RESP_BODY_BEGIN_BITNUM, EV_HTTP_RESP_BODY_CONT = 1ULL << RESP_BODY_CONT_BITNUM, EV_HTTP_RESP_BODY_END = 1ULL << RESP_BODY_END_BITNUM, - EV_HTTP_RESP_END = EV_HTTP_RESP_BODY_END, + EV_HTTP_RESP_END = 1ULL << RESP_END_BITNUM, }; struct tfe_http_req_spec diff --git a/plugin/protocol/http/src/http_half.cpp b/plugin/protocol/http/src/http_half.cpp index 836ff18..949b2fe 100644 --- a/plugin/protocol/http/src/http_half.cpp +++ b/plugin/protocol/http/src/http_half.cpp @@ -268,7 +268,7 @@ static int __parser_callback_on_body(struct http_parser * parser, const char * a hf_private->event_cb(hf_private, ev_body_begin, NULL, parser->content_length, hf_private->event_cb_user); } - if (hf_private->event_cb) + if (hf_private->event_cb && length != 0) { hf_private->event_cb(hf_private, ev_body_cont, (const unsigned char *) at, length, hf_private->event_cb_user); } @@ -281,20 +281,29 @@ static int __parser_callback_on_message_complete(http_parser * parser) { struct http_half_private * hf_private = __PARSER_TO_HF_PRIVATE(parser); enum tfe_http_direction hf_direction = hf_private->hf_public.direction; + + enum tfe_http_event ev_message_end; enum tfe_http_event ev_body_end; if (hf_direction == TFE_HTTP_REQUEST) { + ev_message_end = EV_HTTP_REQ_END; ev_body_end = EV_HTTP_REQ_BODY_END; } else { + ev_message_end = EV_HTTP_RESP_END; ev_body_end = EV_HTTP_RESP_BODY_END; } + if (hf_private->event_cb && hf_private->body_status == STATUS_READING) + { + hf_private->event_cb(hf_private, ev_body_end, NULL, 0, hf_private->event_cb_user); + } + if (hf_private->event_cb) { - hf_private->event_cb(hf_private, ev_body_end, NULL, 0, hf_private->event_cb_user); + hf_private->event_cb(hf_private, ev_message_end, NULL, 0, hf_private->event_cb_user); } hf_private->body_status = STATUS_COMPLETE; diff --git a/plugin/protocol/http/test/test_http_half.cpp b/plugin/protocol/http/test/test_http_half.cpp index a0cb1e7..dd991f0 100644 --- a/plugin/protocol/http/test/test_http_half.cpp +++ b/plugin/protocol/http/test/test_http_half.cpp @@ -40,7 +40,7 @@ TEST(HttpHalfConstruct, ProtoIdentifyNotHTTP) EXPECT_EQ(ret, -1); } -static const char * __get_http_request_01 = +static const char * __get_http_request_no_body = "GET /gfwlist/gfwlist/master/gfwlist.txt HTTP/1.1\r\n" "Host: raw.githubusercontent.com\r\n" "Connection: close\r\n" @@ -51,17 +51,80 @@ static const char * __get_http_request_01 = "If-None-Match: \"023aeae5eafc12082067c36031888adb3bafa797\"\r\n" "\r\n"; -void __get_http_request_01_verify_helper(struct http_half_private * hf_private) +void __get_http_request_header_verify_helper(struct http_half_private * hf_private) { auto * hf_public = &hf_private->hf_public; EXPECT_EQ(hf_public->major_version, 1); EXPECT_EQ(hf_public->minor_version, 1); auto * hf_public_request = &hf_public->req_spec; + + /* PUBLIC FIELD */ + EXPECT_EQ(hf_public_request->method, TFE_HTTP_GET); EXPECT_STREQ(hf_public_request->uri, "/gfwlist/gfwlist/master/gfwlist.txt"); EXPECT_STREQ(hf_public_request->url, "/gfwlist/gfwlist/master/gfwlist.txt"); EXPECT_STREQ(hf_public_request->host, "raw.githubusercontent.com"); - EXPECT_EQ(hf_public_request->method, TFE_HTTP_GET); + + /* Header Field */ + struct http_field_name field_name; + void * __iterator = NULL; + const char * hdr_value = NULL; + + /* HOST */ + hdr_value = tfe_http_field_iterate(hf_public, &__iterator, &field_name); + EXPECT_EQ(field_name.field_id, TFE_HTTP_HOST); + EXPECT_STREQ(hdr_value, "raw.githubusercontent.com"); + + /* Connection */ + hdr_value = tfe_http_field_iterate(hf_public, &__iterator, &field_name); + EXPECT_EQ(field_name.field_id, TFE_HTTP_CONNECTION); + EXPECT_STREQ(hdr_value, "close"); + + /* User-Agent */ + hdr_value = tfe_http_field_iterate(hf_public, &__iterator, &field_name); + EXPECT_EQ(field_name.field_id, TFE_HTTP_USER_AGENT); + EXPECT_STREQ(hdr_value, + "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/69.0.3497.100 Safari/537.36"); + + /* Accept */ + hdr_value = tfe_http_field_iterate(hf_public, &__iterator, &field_name); + EXPECT_EQ(field_name.field_id, TFE_HTTP_UNKNOWN_FIELD); + EXPECT_STREQ(field_name.field_name, "Accept"); + EXPECT_STREQ(hdr_value, "*/*"); + + /* Accept-Encoding */ + hdr_value = tfe_http_field_iterate(hf_public, &__iterator, &field_name); + EXPECT_EQ(field_name.field_id, TFE_HTTP_ACCEPT_ENCODING); + EXPECT_STREQ(hdr_value, "gzip, deflate"); + + /* Accept-Language */ + hdr_value = tfe_http_field_iterate(hf_public, &__iterator, &field_name); + EXPECT_EQ(field_name.field_id, TFE_HTTP_UNKNOWN_FIELD); + EXPECT_STREQ(field_name.field_name, "Accept-Language"); + EXPECT_STREQ(hdr_value, "zh-CN,zh;q=0.9,en-US;q=0.8,en;q=0.7"); + + /* If-None-Match */ + hdr_value = tfe_http_field_iterate(hf_public, &__iterator, &field_name); + EXPECT_EQ(field_name.field_id, TFE_HTTP_UNKNOWN_FIELD); + EXPECT_STREQ(field_name.field_name, "If-None-Match"); + EXPECT_STREQ(hdr_value, "\"023aeae5eafc12082067c36031888adb3bafa797\""); +} + +int __get_http_request_01_callback(struct http_half_private * hf_private, + tfe_http_event ev, const unsigned char * data, size_t len, void * user) +{ + const static tfe_http_event event_call_order[] = + { + EV_HTTP_REQ_HDR, + EV_HTTP_REQ_END, + }; + + auto * calling_seq = (unsigned int *) user; + EXPECT_LE(*calling_seq, TFE_DIM(event_call_order)); + EXPECT_EQ(event_call_order[*calling_seq], ev); + + (*calling_seq)++; + return 0; } TEST(HttpHalfConstruct, GetWithNoBody) @@ -69,11 +132,316 @@ TEST(HttpHalfConstruct, GetWithNoBody) auto * hf_request = hf_private_create(TFE_HTTP_REQUEST, 0, 0); ASSERT_TRUE(hf_request != NULL); - int ret = - hf_private_parse(hf_request, (const unsigned char *) __get_http_request_01, strlen(__get_http_request_01)); + int ret = hf_private_parse(hf_request, (const unsigned char *) __get_http_request_no_body, + strlen(__get_http_request_no_body)); ASSERT_EQ(ret, 0); - __get_http_request_01_verify_helper(hf_request); + __get_http_request_header_verify_helper(hf_request); +} + +TEST(HttpHalfConstructCallback, GetWithNoBody) +{ + auto * hf_request = hf_private_create(TFE_HTTP_REQUEST, 0, 0); + ASSERT_TRUE(hf_request != NULL); + + auto * __data = (const unsigned char *) __get_http_request_no_body; + size_t __datalen = strlen(__get_http_request_no_body); + unsigned int __expect_id = 0; + + hf_private_set_callback(hf_request, __get_http_request_01_callback, &__expect_id, NULL); + hf_private_parse(hf_request, __data, __datalen); + __get_http_request_header_verify_helper(hf_request); +} + +/* +POST / HTTP/1.1 +User-Agent: Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1) QQBrowser/6.0 +Host: qbwup.imtt.qq.com +Content-Length: 726 +Cache-Control: no-cache + +......,..... +......,..............h......V.c..O..-o.{..'._J"9Y...yM...XR....$......L..*r..3..Q..&.......B......' + .....K.(.2..1...}..nep..lI..Ds.5d....E. .Y...9..zr.........H$.....n...GaX.....k.. +.I8...4|..5~..j0u5..>.'...........F16M./...........W..... +7.]xu..2.J.|..A.p`@a.Nq....!v.$...Z.....W..,....<.&:...@..........Y.x.A + .<..r.d8.(.te.....%....=.....xnx..0....*.../.. lJz.>9.....>>..,....9. + ....$./......k........;T.6#{...t...............c....xv.M.mw.......w... + rg..........-d.Z.../f..st....v...8}..p..a.5N...:.......s/U.......o.." + .}U...S:~.......TC.....X...}2..+&>.+............T?>........T.. -..]r + n.....]co..!.......}.0.'...lgR...L.`f +.^..2;M.F]..........A.>...`sJ,.>J.....*/ + +static const unsigned char __post_http_request[] = + { /* Packet 4060 */ + 0x50, 0x4f, 0x53, 0x54, 0x20, 0x2f, 0x20, 0x48, + 0x54, 0x54, 0x50, 0x2f, 0x31, 0x2e, 0x31, 0x0d, + 0x0a, 0x55, 0x73, 0x65, 0x72, 0x2d, 0x41, 0x67, + 0x65, 0x6e, 0x74, 0x3a, 0x20, 0x4d, 0x6f, 0x7a, + 0x69, 0x6c, 0x6c, 0x61, 0x2f, 0x34, 0x2e, 0x30, + 0x20, 0x28, 0x63, 0x6f, 0x6d, 0x70, 0x61, 0x74, + 0x69, 0x62, 0x6c, 0x65, 0x3b, 0x20, 0x4d, 0x53, + 0x49, 0x45, 0x20, 0x37, 0x2e, 0x30, 0x3b, 0x20, + 0x57, 0x69, 0x6e, 0x64, 0x6f, 0x77, 0x73, 0x20, + 0x4e, 0x54, 0x20, 0x35, 0x2e, 0x31, 0x29, 0x20, + 0x51, 0x51, 0x42, 0x72, 0x6f, 0x77, 0x73, 0x65, + 0x72, 0x2f, 0x36, 0x2e, 0x30, 0x0d, 0x0a, 0x48, + 0x6f, 0x73, 0x74, 0x3a, 0x20, 0x71, 0x62, 0x77, + 0x75, 0x70, 0x2e, 0x69, 0x6d, 0x74, 0x74, 0x2e, + 0x71, 0x71, 0x2e, 0x63, 0x6f, 0x6d, 0x0d, 0x0a, + 0x43, 0x6f, 0x6e, 0x74, 0x65, 0x6e, 0x74, 0x2d, + 0x4c, 0x65, 0x6e, 0x67, 0x74, 0x68, 0x3a, 0x20, + 0x37, 0x32, 0x36, 0x0d, 0x0a, 0x43, 0x61, 0x63, + 0x68, 0x65, 0x2d, 0x43, 0x6f, 0x6e, 0x74, 0x72, + 0x6f, 0x6c, 0x3a, 0x20, 0x6e, 0x6f, 0x2d, 0x63, + 0x61, 0x63, 0x68, 0x65, 0x0d, 0x0a, 0x0d, 0x0a, + 0x00, 0x00, 0x02, 0xd6, 0x10, 0x02, 0x2c, 0x3c, + 0x4c, 0x56, 0x08, 0x71, 0x62, 0x70, 0x63, 0x73, + 0x74, 0x61, 0x74, 0x66, 0x04, 0x73, 0x74, 0x61, + 0x74, 0x7d, 0x00, 0x01, 0x02, 0xb3, 0x08, 0x00, + 0x01, 0x06, 0x05, 0x63, 0x72, 0x79, 0x70, 0x74, + 0x18, 0x00, 0x01, 0x06, 0x0a, 0x6c, 0x69, 0x73, + 0x74, 0x3c, 0x63, 0x68, 0x61, 0x72, 0x3e, 0x1d, + 0x00, 0x01, 0x02, 0x95, 0x0d, 0x00, 0x01, 0x02, + 0x90, 0xc9, 0xf0, 0x2c, 0x12, 0xa4, 0x8d, 0x1d, + 0xdd, 0xa4, 0xb6, 0xd6, 0xde, 0xb0, 0x9b, 0xe8, + 0xf6, 0x13, 0x68, 0x9e, 0x95, 0xf8, 0xf9, 0xcb, + 0xe8, 0x56, 0xa3, 0x63, 0x0e, 0x8b, 0x4f, 0xb8, + 0x1c, 0x2d, 0x6f, 0xb9, 0x7b, 0x0c, 0x13, 0x27, + 0x82, 0x5f, 0x4a, 0x22, 0x39, 0x59, 0x12, 0xb8, + 0xfa, 0x79, 0x4d, 0x80, 0x1b, 0xce, 0x58, 0x52, + 0x81, 0xf6, 0xb4, 0x83, 0x24, 0x99, 0x18, 0x1a, + 0x85, 0x01, 0x91, 0x4c, 0xa3, 0xa5, 0x2a, 0x72, + 0xe1, 0xea, 0x33, 0xb9, 0xae, 0x51, 0xfa, 0xca, + 0x26, 0x91, 0x1c, 0xed, 0xba, 0x9d, 0x14, 0xe1, + 0x42, 0x9f, 0x15, 0xd3, 0xf0, 0x15, 0xc4, 0x27, + 0x00, 0xa7, 0xdd, 0xfc, 0xa4, 0x4b, 0xeb, 0x28, + 0x1e, 0x32, 0x82, 0xba, 0x31, 0xdd, 0x03, 0x80, + 0x7d, 0x7f, 0xc5, 0x6e, 0x65, 0x70, 0x9c, 0xd8, + 0x6c, 0x49, 0xd6, 0xe3, 0x44, 0x73, 0x97, 0x35, + 0x64, 0x81, 0xf0, 0x98, 0xfa, 0x45, 0xeb, 0x09, + 0xaf, 0x59, 0x17, 0xf2, 0xe7, 0x39, 0x1c, 0xbc, + 0x7a, 0x72, 0x81, 0xae, 0xce, 0xd1, 0xe8, 0xfa, + 0x96, 0xdd, 0xa0, 0x48, 0x24, 0xe7, 0x17, 0x0b, + 0xad, 0x16, 0x6e, 0xfd, 0xc5, 0xd7, 0x47, 0x61, + 0x58, 0xc0, 0xdc, 0xa0, 0xf1, 0xe6, 0x6b, 0xa9, + 0xb4, 0x0d, 0x8c, 0x49, 0x38, 0x14, 0xe4, 0xc5, + 0x34, 0x7c, 0x03, 0xf8, 0x35, 0x7e, 0xad, 0xed, + 0x6a, 0x30, 0x75, 0x35, 0x12, 0xe8, 0x3e, 0x91, + 0x27, 0xe6, 0x89, 0xe4, 0x18, 0xaa, 0xeb, 0x80, + 0x15, 0x97, 0x87, 0xa2, 0x46, 0x31, 0x36, 0x4d, + 0xc7, 0x2f, 0xdf, 0x85, 0xd7, 0xe2, 0x89, 0xcb, + 0x83, 0xa1, 0xd7, 0xb3, 0xa8, 0x57, 0x82, 0xf5, + 0x8b, 0xe2, 0x0c, 0x0a, 0x37, 0x89, 0x5d, 0x78, + 0x75, 0x1d, 0xdf, 0x32, 0xf4, 0x4a, 0xc9, 0x7c, + 0xd9, 0xa3, 0x41, 0xae, 0x70, 0x60, 0x40, 0x61, + 0x9e, 0x4e, 0x71, 0xa8, 0x85, 0x8a, 0xb2, 0x21, + 0x76, 0x1f, 0x24, 0xcb, 0xfa, 0xe0, 0x5a, 0x08, + 0xfe, 0xcb, 0x0c, 0xa4, 0x57, 0xf1, 0x9d, 0x2c, + 0xe2, 0xab, 0x2e, 0x02, 0x3c, 0xbd, 0x26, 0x3a, + 0x8e, 0x87, 0xec, 0x40, 0xb8, 0xba, 0xc2, 0xf5, + 0x08, 0xd4, 0xb4, 0xc4, 0xa1, 0x9d, 0x59, 0xa7, + 0x78, 0xdb, 0x41, 0x09, 0x09, 0xc9, 0x3c, 0x93, + 0x81, 0x72, 0xef, 0x64, 0x38, 0x06, 0x28, 0x0b, + 0x74, 0x65, 0xfe, 0x14, 0xa6, 0x8f, 0x1f, 0x25, + 0x1b, 0xd6, 0xa8, 0x9f, 0x3d, 0x13, 0xfa, 0xf0, + 0xf1, 0x2e, 0x78, 0x6e, 0x78, 0xf8, 0x99, 0x30, + 0x1f, 0x94, 0xca, 0xe8, 0x2a, 0xa1, 0xb8, 0x97, + 0x2f, 0xfa, 0xa0, 0x20, 0x6c, 0x4a, 0x7a, 0xca, + 0x3e, 0x39, 0x88, 0xc5, 0xfa, 0x8d, 0xcd, 0x3e, + 0x3e, 0x03, 0xa3, 0x2c, 0xa0, 0x80, 0x05, 0x9d, + 0x39, 0xc1, 0x20, 0xc4, 0xd4, 0x9c, 0xd7, 0x24, + 0x1e, 0x2f, 0xa1, 0xb9, 0x98, 0x86, 0xa0, 0xf9, + 0x6b, 0xf3, 0xb9, 0xe7, 0x1d, 0xff, 0xe7, 0xba, + 0x0e, 0x3b, 0x54, 0xa5, 0x36, 0x23, 0x7b, 0xc1, + 0xa3, 0xd5, 0x74, 0x1b, 0x18, 0xc6, 0xa3, 0x80, + 0x19, 0xd1, 0xa7, 0x03, 0x9f, 0xc1, 0xa9, 0x84, + 0x90, 0xab, 0x63, 0x93, 0xb5, 0xaa, 0xac, 0x78, + 0x76, 0xcd, 0x4d, 0x94, 0x6d, 0x77, 0xb8, 0xee, + 0xde, 0xc7, 0xd2, 0x94, 0xb1, 0x77, 0xc2, 0x2e, + 0x9a, 0x72, 0x67, 0x19, 0xe2, 0x18, 0xbb, 0xc4, + 0x91, 0x13, 0xb9, 0x9e, 0xa8, 0x2d, 0x64, 0xa6, + 0x5a, 0xb9, 0x12, 0xfb, 0x2f, 0x66, 0x12, 0xef, + 0x73, 0x74, 0x9e, 0x95, 0xdb, 0xe1, 0x76, 0xa1, + 0xbc, 0x8d, 0x38, 0x7d, 0x95, 0xa6, 0x70, 0x08, + 0xb5, 0x61, 0x08, 0x35, 0x4e, 0x0c, 0xbe, 0xf4, + 0x3a, 0x0c, 0xf0, 0xa5, 0xbf, 0xb0, 0x14, 0xd8, + 0x73, 0x2f, 0x55, 0xf3, 0x00, 0xac, 0xe4, 0x02, + 0xed, 0xd8, 0x6f, 0xa1, 0xa1, 0x22, 0xd8, 0x7d, + 0x55, 0xb7, 0xc7, 0x05, 0x53, 0x3a, 0x7e, 0x97, + 0xb9, 0xbe, 0xc1, 0xcb, 0x1a, 0x9c, 0x54, 0x43, + 0xfa, 0xfd, 0xb6, 0xba, 0xd8, 0x58, 0xf2, 0x87, + 0x04, 0x7d, 0x32, 0xdc, 0xd4, 0x2b, 0x26, 0x3e, + 0x9e, 0x2b, 0xd0, 0xa3, 0xc3, 0x84, 0x9e, 0x06, + 0xab, 0xa1, 0xf6, 0xd5, 0xb6, 0x18, 0x54, 0x3f, + 0x3e, 0xc5, 0x9a, 0xe3, 0xea, 0xfb, 0xb4, 0x90, + 0x91, 0x54, 0x90, 0xe1, 0x09, 0x2d, 0xf0, 0x03, + 0x5d, 0x72, 0x6e, 0x2e, 0xdc, 0xcc, 0xdc, 0xb3, + 0x5d, 0x63, 0x6f, 0xad, 0xfc, 0x21, 0x98, 0xad, + 0xd7, 0x00, 0xb0, 0x1c, 0xef, 0x7d, 0x00, 0x30, + 0xf8, 0x27, 0x83, 0x92, 0x98, 0x6c, 0x67, 0x52, + 0x9a, 0x98, 0xf5, 0x4c, 0x2e, 0x60, 0x66, 0x0a, + 0xb8, 0x5e, 0xe1, 0x90, 0x32, 0x3b, 0x4d, 0x0b, + 0x46, 0x5d, 0x17, 0x8e, 0xb1, 0xf3, 0x81, 0xc0, + 0x14, 0xd5, 0xbd, 0xf8, 0x41, 0xcf, 0x3e, 0x99, + 0x9e, 0xbd, 0x60, 0x73, 0x4a, 0x2c, 0xab, 0x3e, + 0x4a, 0x8c, 0x98, 0x0c, 0xa8, 0x0c}; + +/* +POST / HTTP/1.1 +User-Agent: Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1) QQBrowser/6.0 +Host: qbwup.imtt.qq.com +Content-Length: 726 +Cache-Control: no-cache + +......,..... +......,..............h......V.c..O..-o.{..'._J"9Y...yM...XR....$......L..*r..3..Q..&.......B......' + .....K.(.2..1...}..nep..lI..Ds.5d....E. .Y...9..zr.........H$.....n...GaX.....k.. +.I8...4|..5~..j0u5..>.'...........F16M./...........W..... +7.]xu..2.J.|..A.p`@a.Nq....!v.$...Z.....W..,....<.&:...@..........Y.x.A + .<..r.d8.(.te.....%....=.....xnx..0....*.../.. lJz.>9.....>>..,....9. + ....$./......k........;T.6#{...t...............c....xv.M.mw.......w... + rg..........-d.Z.../f..st....v...8}..p..a.5N...:.......s/U.......o.." + .}U...S:~.......TC.....X...}2..+&>.+............T?>........T.. -..]r + n.....]co..!.......}.0.'...lgR...L.`f +.^..2;M.F]..........A.>...`sJ,.>J.....*/ + +void __http_post_header_verify_helper(struct http_half_private * hf_private) +{ + auto * hf_public = &hf_private->hf_public; + EXPECT_EQ(hf_public->major_version, 1); + EXPECT_EQ(hf_public->minor_version, 1); + + auto * hf_public_request = &hf_public->req_spec; + + /* PUBLIC FIELD */ + EXPECT_EQ(hf_public_request->method, TFE_HTTP_POST); + EXPECT_STREQ(hf_public_request->uri, "/"); + EXPECT_STREQ(hf_public_request->url, "/"); + EXPECT_STREQ(hf_public_request->host, "qbwup.imtt.qq.com"); + + /* Header Field */ + struct http_field_name field_name; + void * __iterator = NULL; + const char * hdr_value = NULL; + + /* User-Agent */ + hdr_value = tfe_http_field_iterate(hf_public, &__iterator, &field_name); + EXPECT_EQ(field_name.field_id, TFE_HTTP_USER_AGENT); + EXPECT_STREQ(hdr_value, "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1) QQBrowser/6.0"); + + /* HOST */ + hdr_value = tfe_http_field_iterate(hf_public, &__iterator, &field_name); + EXPECT_EQ(field_name.field_id, TFE_HTTP_HOST); + EXPECT_STREQ(hdr_value, "qbwup.imtt.qq.com"); + + /* Content-Length */ + hdr_value = tfe_http_field_iterate(hf_public, &__iterator, &field_name); + EXPECT_EQ(field_name.field_id, TFE_HTTP_CONT_LENGTH); + EXPECT_STREQ(hdr_value, "726"); + + /* Accept */ + hdr_value = tfe_http_field_iterate(hf_public, &__iterator, &field_name); + EXPECT_EQ(field_name.field_id, TFE_HTTP_UNKNOWN_FIELD); + EXPECT_STREQ(field_name.field_name, "Cache-Control"); + EXPECT_STREQ(hdr_value, "no-cache"); +} + +struct __post_http_req_ctx +{ + unsigned int calling_seq{0}; + size_t total_length{0}; + + bool req_hdr_called{false}; + bool req_body_begin_called{false}; + bool req_body_cont_called{false}; + bool req_body_end_called{false}; + bool req_end_called{false}; +}; + +int __post_http_request_callback(struct http_half_private * hf_private, + tfe_http_event ev, const unsigned char * data, size_t len, void * user) +{ + const static tfe_http_event event_call_order[] = + { + EV_HTTP_REQ_HDR, + EV_HTTP_REQ_BODY_BEGIN, + EV_HTTP_REQ_BODY_CONT, + EV_HTTP_REQ_BODY_END, + EV_HTTP_REQ_END + }; + + auto * ctx = (struct __post_http_req_ctx *) user; + EXPECT_LE(ctx->calling_seq, TFE_DIM(event_call_order)); + EXPECT_EQ(event_call_order[ctx->calling_seq], ev); + ctx->calling_seq++; + + if (ev & EV_HTTP_REQ_HDR) + { + __http_post_header_verify_helper(hf_private); + ctx->req_hdr_called = true; + } + + if (ev & EV_HTTP_REQ_BODY_BEGIN) + { + ctx->req_body_begin_called = true; + } + + if (ev & EV_HTTP_REQ_BODY_CONT) + { + ctx->total_length += len; + ctx->req_body_cont_called = true; + } + + if (ev & EV_HTTP_REQ_BODY_END) + { + EXPECT_EQ(ctx->total_length, 726); + ctx->req_body_end_called = true; + } + + if (ev & EV_HTTP_REQ_END) + { + ctx->req_end_called = true; + } + + return 0; +} + +TEST(HttpHalfConstruct, PostWithBody) +{ + auto * hf_request = hf_private_create(TFE_HTTP_REQUEST, 0, 0); + ASSERT_TRUE(hf_request != NULL); + + auto * __data = (const unsigned char *) __post_http_request; + size_t __datalen = sizeof(__post_http_request); + + int ret = hf_private_parse(hf_request, __data, __datalen); + ASSERT_EQ(ret, 0); + + __http_post_header_verify_helper(hf_request); +} + +TEST(HttpHalfConstructCallback, PostWithBody) +{ + auto * hf_request = hf_private_create(TFE_HTTP_REQUEST, 0, 0); + ASSERT_TRUE(hf_request != NULL); + + auto * __data = (const unsigned char *) __post_http_request; + size_t __datalen = sizeof(__post_http_request); + + auto * ctx = new struct __post_http_req_ctx; + hf_private_set_callback(hf_request, __post_http_request_callback, ctx, NULL); + + hf_private_parse(hf_request, __data, __datalen); + __http_post_header_verify_helper(hf_request); + + EXPECT_TRUE(ctx->req_hdr_called); + EXPECT_TRUE(ctx->req_body_begin_called); + EXPECT_TRUE(ctx->req_body_cont_called); + EXPECT_TRUE(ctx->req_body_end_called); + EXPECT_TRUE(ctx->req_end_called); } int main(int argc, char ** argv)