OLD | NEW |
(Empty) | |
| 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 "net/http2/decoder/payload_decoders/altsvc_payload_decoder.h" |
| 6 |
| 7 #include <stddef.h> |
| 8 |
| 9 #include "base/logging.h" |
| 10 #include "base/macros.h" |
| 11 #include "net/http2/decoder/decode_buffer.h" |
| 12 #include "net/http2/decoder/http2_frame_decoder_listener.h" |
| 13 #include "net/http2/http2_constants.h" |
| 14 #include "net/http2/http2_structures.h" |
| 15 #include "net/http2/tools/http2_bug_tracker.h" |
| 16 |
| 17 namespace net { |
| 18 |
| 19 std::ostream& operator<<(std::ostream& out, |
| 20 AltSvcPayloadDecoder::PayloadState v) { |
| 21 switch (v) { |
| 22 case AltSvcPayloadDecoder::PayloadState::kStartDecodingStruct: |
| 23 return out << "kStartDecodingStruct"; |
| 24 case AltSvcPayloadDecoder::PayloadState::kMaybeDecodedStruct: |
| 25 return out << "kMaybeDecodedStruct"; |
| 26 case AltSvcPayloadDecoder::PayloadState::kDecodingStrings: |
| 27 return out << "kDecodingStrings"; |
| 28 case AltSvcPayloadDecoder::PayloadState::kResumeDecodingStruct: |
| 29 return out << "kResumeDecodingStruct"; |
| 30 } |
| 31 return out << static_cast<int>(v); |
| 32 } |
| 33 |
| 34 DecodeStatus AltSvcPayloadDecoder::StartDecodingPayload( |
| 35 FrameDecoderState* state, |
| 36 DecodeBuffer* db) { |
| 37 DVLOG(2) << "AltSvcPayloadDecoder::StartDecodingPayload: " |
| 38 << state->frame_header(); |
| 39 DCHECK_EQ(Http2FrameType::ALTSVC, state->frame_header().type); |
| 40 DCHECK_LE(db->Remaining(), state->frame_header().payload_length); |
| 41 DCHECK_EQ(0, state->frame_header().flags); |
| 42 |
| 43 state->InitializeRemainders(); |
| 44 payload_state_ = PayloadState::kStartDecodingStruct; |
| 45 |
| 46 return ResumeDecodingPayload(state, db); |
| 47 } |
| 48 |
| 49 DecodeStatus AltSvcPayloadDecoder::ResumeDecodingPayload( |
| 50 FrameDecoderState* state, |
| 51 DecodeBuffer* db) { |
| 52 const Http2FrameHeader& frame_header = state->frame_header(); |
| 53 DVLOG(2) << "AltSvcPayloadDecoder::ResumeDecodingPayload: " << frame_header; |
| 54 DCHECK_EQ(Http2FrameType::ALTSVC, frame_header.type); |
| 55 DCHECK_LE(state->remaining_payload(), frame_header.payload_length); |
| 56 DCHECK_LE(db->Remaining(), state->remaining_payload()); |
| 57 DCHECK_NE(PayloadState::kMaybeDecodedStruct, payload_state_); |
| 58 // |status| has to be initialized to some value to avoid compiler error in |
| 59 // case PayloadState::kMaybeDecodedStruct below, but value does not matter, |
| 60 // see DCHECK_NE above. |
| 61 DecodeStatus status = DecodeStatus::kDecodeError; |
| 62 while (true) { |
| 63 DVLOG(2) << "AltSvcPayloadDecoder::ResumeDecodingPayload payload_state_=" |
| 64 << payload_state_; |
| 65 switch (payload_state_) { |
| 66 case PayloadState::kStartDecodingStruct: |
| 67 status = state->StartDecodingStructureInPayload(&altsvc_fields_, db); |
| 68 // FALLTHROUGH_INTENDED |
| 69 |
| 70 case PayloadState::kMaybeDecodedStruct: |
| 71 if (status == DecodeStatus::kDecodeDone && |
| 72 altsvc_fields_.origin_length <= state->remaining_payload()) { |
| 73 size_t origin_length = altsvc_fields_.origin_length; |
| 74 size_t value_length = state->remaining_payload() - origin_length; |
| 75 state->listener()->OnAltSvcStart(frame_header, origin_length, |
| 76 value_length); |
| 77 } else if (status != DecodeStatus::kDecodeDone) { |
| 78 DCHECK(state->remaining_payload() > 0 || |
| 79 status == DecodeStatus::kDecodeError) |
| 80 << "\nremaining_payload: " << state->remaining_payload() |
| 81 << "\nstatus: " << status << "\nheader: " << frame_header; |
| 82 // Assume in progress. |
| 83 payload_state_ = PayloadState::kResumeDecodingStruct; |
| 84 return status; |
| 85 } else { |
| 86 // The origin's length is longer than the remaining payload. |
| 87 DCHECK_GT(altsvc_fields_.origin_length, state->remaining_payload()); |
| 88 return state->ReportFrameSizeError(); |
| 89 } |
| 90 // FALLTHROUGH_INTENDED |
| 91 |
| 92 case PayloadState::kDecodingStrings: |
| 93 return DecodeStrings(state, db); |
| 94 |
| 95 case PayloadState::kResumeDecodingStruct: |
| 96 status = state->ResumeDecodingStructureInPayload(&altsvc_fields_, db); |
| 97 payload_state_ = PayloadState::kMaybeDecodedStruct; |
| 98 continue; |
| 99 } |
| 100 HTTP2_BUG << "PayloadState: " << payload_state_; |
| 101 } |
| 102 } |
| 103 |
| 104 DecodeStatus AltSvcPayloadDecoder::DecodeStrings(FrameDecoderState* state, |
| 105 DecodeBuffer* db) { |
| 106 DVLOG(2) << "AltSvcPayloadDecoder::DecodeStrings remaining_payload=" |
| 107 << state->remaining_payload() |
| 108 << ", db->Remaining=" << db->Remaining(); |
| 109 // Note that we don't explicitly keep track of exactly how far through the |
| 110 // origin; instead we compute it from how much is left of the original |
| 111 // payload length and the decoded total length of the origin. |
| 112 size_t origin_length = altsvc_fields_.origin_length; |
| 113 size_t value_length = state->frame_header().payload_length - origin_length - |
| 114 Http2AltSvcFields::EncodedSize(); |
| 115 if (state->remaining_payload() > value_length) { |
| 116 size_t remaining_origin_length = state->remaining_payload() - value_length; |
| 117 size_t avail = db->MinLengthRemaining(remaining_origin_length); |
| 118 state->listener()->OnAltSvcOriginData(db->cursor(), avail); |
| 119 db->AdvanceCursor(avail); |
| 120 state->ConsumePayload(avail); |
| 121 if (remaining_origin_length > avail) { |
| 122 payload_state_ = PayloadState::kDecodingStrings; |
| 123 return DecodeStatus::kDecodeInProgress; |
| 124 } |
| 125 } |
| 126 // All that is left is the value string. |
| 127 DCHECK_LE(state->remaining_payload(), value_length); |
| 128 DCHECK_LE(db->Remaining(), state->remaining_payload()); |
| 129 if (db->HasData()) { |
| 130 size_t avail = db->Remaining(); |
| 131 state->listener()->OnAltSvcValueData(db->cursor(), avail); |
| 132 db->AdvanceCursor(avail); |
| 133 state->ConsumePayload(avail); |
| 134 } |
| 135 if (state->remaining_payload() == 0) { |
| 136 state->listener()->OnAltSvcEnd(); |
| 137 return DecodeStatus::kDecodeDone; |
| 138 } |
| 139 payload_state_ = PayloadState::kDecodingStrings; |
| 140 return DecodeStatus::kDecodeInProgress; |
| 141 } |
| 142 |
| 143 } // namespace net |
OLD | NEW |