1 // Copyright 2016 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "quiche/http2/decoder/payload_decoders/priority_payload_decoder.h"
6
7 #include <stddef.h>
8
9 #include "quiche/http2/decoder/http2_frame_decoder_listener.h"
10 #include "quiche/http2/http2_constants.h"
11 #include "quiche/http2/test_tools/frame_parts.h"
12 #include "quiche/http2/test_tools/frame_parts_collector.h"
13 #include "quiche/http2/test_tools/http2_frame_builder.h"
14 #include "quiche/http2/test_tools/http2_random.h"
15 #include "quiche/http2/test_tools/http2_structures_test_util.h"
16 #include "quiche/http2/test_tools/payload_decoder_base_test_util.h"
17 #include "quiche/http2/test_tools/random_decoder_test_base.h"
18 #include "quiche/common/platform/api/quiche_logging.h"
19 #include "quiche/common/platform/api/quiche_test.h"
20
21 namespace http2 {
22 namespace test {
23
24 class PriorityPayloadDecoderPeer {
25 public:
FrameType()26 static constexpr Http2FrameType FrameType() {
27 return Http2FrameType::PRIORITY;
28 }
29
30 // Returns the mask of flags that affect the decoding of the payload (i.e.
31 // flags that that indicate the presence of certain fields or padding).
FlagsAffectingPayloadDecoding()32 static constexpr uint8_t FlagsAffectingPayloadDecoding() { return 0; }
33 };
34
35 namespace {
36
37 struct Listener : public FramePartsCollector {
OnPriorityFramehttp2::test::__anona8ab305d0111::Listener38 void OnPriorityFrame(const Http2FrameHeader& header,
39 const Http2PriorityFields& priority_fields) override {
40 QUICHE_VLOG(1) << "OnPriority: " << header << "; " << priority_fields;
41 StartAndEndFrame(header)->OnPriorityFrame(header, priority_fields);
42 }
43
OnFrameSizeErrorhttp2::test::__anona8ab305d0111::Listener44 void OnFrameSizeError(const Http2FrameHeader& header) override {
45 QUICHE_VLOG(1) << "OnFrameSizeError: " << header;
46 FrameError(header)->OnFrameSizeError(header);
47 }
48 };
49
50 class PriorityPayloadDecoderTest
51 : public AbstractPayloadDecoderTest<PriorityPayloadDecoder,
52 PriorityPayloadDecoderPeer, Listener> {
53 protected:
RandPriorityFields()54 Http2PriorityFields RandPriorityFields() {
55 Http2PriorityFields fields;
56 test::Randomize(&fields, RandomPtr());
57 return fields;
58 }
59 };
60
61 // Confirm we get an error if the payload is not the correct size to hold
62 // exactly one Http2PriorityFields.
TEST_F(PriorityPayloadDecoderTest,WrongSize)63 TEST_F(PriorityPayloadDecoderTest, WrongSize) {
64 auto approve_size = [](size_t size) {
65 return size != Http2PriorityFields::EncodedSize();
66 };
67 Http2FrameBuilder fb;
68 fb.Append(RandPriorityFields());
69 fb.Append(RandPriorityFields());
70 EXPECT_TRUE(VerifyDetectsFrameSizeError(0, fb.buffer(), approve_size));
71 }
72
TEST_F(PriorityPayloadDecoderTest,VariousPayloads)73 TEST_F(PriorityPayloadDecoderTest, VariousPayloads) {
74 for (int n = 0; n < 100; ++n) {
75 Http2PriorityFields fields = RandPriorityFields();
76 Http2FrameBuilder fb;
77 fb.Append(fields);
78 Http2FrameHeader header(fb.size(), Http2FrameType::PRIORITY, RandFlags(),
79 RandStreamId());
80 set_frame_header(header);
81 FrameParts expected(header);
82 expected.SetOptPriority(fields);
83 EXPECT_TRUE(DecodePayloadAndValidateSeveralWays(fb.buffer(), expected));
84 }
85 }
86
87 } // namespace
88 } // namespace test
89 } // namespace http2
90