| Index: net/http2/decoder/payload_decoders/ping_payload_decoder.cc
|
| diff --git a/net/http2/decoder/payload_decoders/ping_payload_decoder.cc b/net/http2/decoder/payload_decoders/ping_payload_decoder.cc
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..abe43d0a6a2d633afdaabd0523ee3b1f80e30202
|
| --- /dev/null
|
| +++ b/net/http2/decoder/payload_decoders/ping_payload_decoder.cc
|
| @@ -0,0 +1,89 @@
|
| +// Copyright 2016 The Chromium Authors. All rights reserved.
|
| +// Use of this source code is governed by a BSD-style license that can be
|
| +// found in the LICENSE file.
|
| +
|
| +#include "net/http2/decoder/payload_decoders/ping_payload_decoder.h"
|
| +
|
| +#include "base/logging.h"
|
| +#include "net/http2/decoder/http2_frame_decoder_listener.h"
|
| +#include "net/http2/http2_constants.h"
|
| +
|
| +namespace net {
|
| +namespace {
|
| +constexpr auto kOpaqueSize = Http2PingFields::EncodedSize();
|
| +}
|
| +
|
| +DecodeStatus PingPayloadDecoder::StartDecodingPayload(FrameDecoderState* state,
|
| + DecodeBuffer* db) {
|
| + const Http2FrameHeader& frame_header = state->frame_header();
|
| + const uint32_t total_length = frame_header.payload_length;
|
| +
|
| + DVLOG(2) << "PingPayloadDecoder::StartDecodingPayload: " << frame_header;
|
| + DCHECK_EQ(Http2FrameType::PING, frame_header.type);
|
| + DCHECK_LE(db->Remaining(), total_length);
|
| + DCHECK_EQ(0, frame_header.flags & ~(Http2FrameFlag::FLAG_ACK));
|
| +
|
| + // Is the payload entirely in the decode buffer and is it the correct size?
|
| + // Given the size of the header and payload (17 bytes total), this is most
|
| + // likely the case the vast majority of the time.
|
| + if (db->Remaining() == kOpaqueSize && total_length == kOpaqueSize) {
|
| + // Special case this situation as it allows us to avoid any copying;
|
| + // the other path makes two copies, first into the buffer in
|
| + // Http2StructureDecoder as it accumulates the 8 bytes of opaque data,
|
| + // and a second copy into the Http2PingFields member of in this class.
|
| + // This supports the claim that this decoder is (mostly) non-buffering.
|
| + static_assert(sizeof(Http2PingFields) == kOpaqueSize,
|
| + "If not, then can't enter this block!");
|
| + auto ping = reinterpret_cast<const Http2PingFields*>(db->cursor());
|
| + if (frame_header.IsAck()) {
|
| + state->listener()->OnPingAck(frame_header, *ping);
|
| + } else {
|
| + state->listener()->OnPing(frame_header, *ping);
|
| + }
|
| + db->AdvanceCursor(kOpaqueSize);
|
| + return DecodeStatus::kDecodeDone;
|
| + }
|
| + state->InitializeRemainders();
|
| + return HandleStatus(
|
| + state, state->StartDecodingStructureInPayload(&ping_fields_, db));
|
| +}
|
| +
|
| +DecodeStatus PingPayloadDecoder::ResumeDecodingPayload(FrameDecoderState* state,
|
| + DecodeBuffer* db) {
|
| + DVLOG(2) << "ResumeDecodingPayload: remaining_payload="
|
| + << state->remaining_payload();
|
| + DCHECK_EQ(Http2FrameType::PING, state->frame_header().type);
|
| + DCHECK_LE(db->Remaining(), state->frame_header().payload_length);
|
| + return HandleStatus(
|
| + state, state->ResumeDecodingStructureInPayload(&ping_fields_, db));
|
| +}
|
| +
|
| +DecodeStatus PingPayloadDecoder::HandleStatus(FrameDecoderState* state,
|
| + DecodeStatus status) {
|
| + DVLOG(2) << "HandleStatus: status=" << status
|
| + << "; remaining_payload=" << state->remaining_payload();
|
| + if (status == DecodeStatus::kDecodeDone) {
|
| + if (state->remaining_payload() == 0) {
|
| + const Http2FrameHeader& frame_header = state->frame_header();
|
| + if (frame_header.IsAck()) {
|
| + state->listener()->OnPingAck(frame_header, ping_fields_);
|
| + } else {
|
| + state->listener()->OnPing(frame_header, ping_fields_);
|
| + }
|
| + return DecodeStatus::kDecodeDone;
|
| + }
|
| + // Payload is too long.
|
| + return state->ReportFrameSizeError();
|
| + }
|
| + // Not done decoding the structure. Either we've got more payload to decode,
|
| + // or we've run out because the payload is too short.
|
| + DCHECK(
|
| + (status == DecodeStatus::kDecodeInProgress &&
|
| + state->remaining_payload() > 0) ||
|
| + (status == DecodeStatus::kDecodeError && state->remaining_payload() == 0))
|
| + << "\n status=" << status
|
| + << "; remaining_payload=" << state->remaining_payload();
|
| + return status;
|
| +}
|
| +
|
| +} // namespace net
|
|
|