blob: 50a0d931eace9ddbc17e9015a6a08706c1485171 [file] [log] [blame] [edit]
// Copyright (c) 2012 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/spdy/buffered_spdy_framer.h"
#include "base/logging.h"
namespace net {
BufferedSpdyFramer::BufferedSpdyFramer(int version, bool enable_compression)
: spdy_framer_(version),
visitor_(NULL),
header_buffer_used_(0),
header_buffer_valid_(false),
header_stream_id_(SpdyFramer::kInvalidStream),
frames_received_(0) {
spdy_framer_.set_enable_compression(enable_compression);
memset(header_buffer_, 0, sizeof(header_buffer_));
}
BufferedSpdyFramer::~BufferedSpdyFramer() {
}
void BufferedSpdyFramer::set_visitor(
BufferedSpdyFramerVisitorInterface* visitor) {
visitor_ = visitor;
spdy_framer_.set_visitor(this);
}
void BufferedSpdyFramer::OnError(SpdyFramer* spdy_framer) {
DCHECK(spdy_framer);
visitor_->OnError(spdy_framer->error_code());
}
void BufferedSpdyFramer::OnSynStream(SpdyStreamId stream_id,
SpdyStreamId associated_stream_id,
SpdyPriority priority,
uint8 credential_slot,
bool fin,
bool unidirectional) {
frames_received_++;
DCHECK(!control_frame_fields_.get());
control_frame_fields_.reset(new ControlFrameFields());
control_frame_fields_->type = SYN_STREAM;
control_frame_fields_->stream_id = stream_id;
control_frame_fields_->associated_stream_id = associated_stream_id;
control_frame_fields_->priority = priority;
control_frame_fields_->credential_slot = credential_slot;
control_frame_fields_->fin = fin;
control_frame_fields_->unidirectional = unidirectional;
InitHeaderStreaming(stream_id);
}
void BufferedSpdyFramer::OnHeaders(SpdyStreamId stream_id,
bool fin) {
frames_received_++;
DCHECK(!control_frame_fields_.get());
control_frame_fields_.reset(new ControlFrameFields());
control_frame_fields_->type = HEADERS;
control_frame_fields_->stream_id = stream_id;
control_frame_fields_->fin = fin;
InitHeaderStreaming(stream_id);
}
void BufferedSpdyFramer::OnSynReply(SpdyStreamId stream_id,
bool fin) {
frames_received_++;
DCHECK(!control_frame_fields_.get());
control_frame_fields_.reset(new ControlFrameFields());
control_frame_fields_->type = SYN_REPLY;
control_frame_fields_->stream_id = stream_id;
control_frame_fields_->fin = fin;
InitHeaderStreaming(stream_id);
}
bool BufferedSpdyFramer::OnCredentialFrameData(const char* frame_data,
size_t len) {
DCHECK(false);
return false;
}
bool BufferedSpdyFramer::OnControlFrameHeaderData(SpdyStreamId stream_id,
const char* header_data,
size_t len) {
CHECK_EQ(header_stream_id_, stream_id);
if (len == 0) {
// Indicates end-of-header-block.
CHECK(header_buffer_valid_);
SpdyHeaderBlock headers;
bool parsed_headers = spdy_framer_.ParseHeaderBlockInBuffer(
header_buffer_, header_buffer_used_, &headers);
if (!parsed_headers) {
visitor_->OnStreamError(
stream_id, "Could not parse Spdy Control Frame Header.");
return false;
}
DCHECK(control_frame_fields_.get());
switch (control_frame_fields_->type) {
case SYN_STREAM:
visitor_->OnSynStream(control_frame_fields_->stream_id,
control_frame_fields_->associated_stream_id,
control_frame_fields_->priority,
control_frame_fields_->credential_slot,
control_frame_fields_->fin,
control_frame_fields_->unidirectional,
headers);
break;
case SYN_REPLY:
visitor_->OnSynReply(control_frame_fields_->stream_id,
control_frame_fields_->fin,
headers);
break;
case HEADERS:
visitor_->OnHeaders(control_frame_fields_->stream_id,
control_frame_fields_->fin,
headers);
break;
default:
DCHECK(false) << "Unexpect control frame type: "
<< control_frame_fields_->type;
break;
}
control_frame_fields_.reset(NULL);
return true;
}
const size_t available = kHeaderBufferSize - header_buffer_used_;
if (len > available) {
header_buffer_valid_ = false;
visitor_->OnStreamError(
stream_id, "Received more data than the allocated size.");
return false;
}
memcpy(header_buffer_ + header_buffer_used_, header_data, len);
header_buffer_used_ += len;
return true;
}
void BufferedSpdyFramer::OnDataFrameHeader(const SpdyDataFrame* frame) {
frames_received_++;
header_stream_id_ = frame->stream_id();
}
void BufferedSpdyFramer::OnStreamFrameData(SpdyStreamId stream_id,
const char* data,
size_t len,
SpdyDataFlags flags) {
visitor_->OnStreamFrameData(stream_id, data, len, flags);
}
void BufferedSpdyFramer::OnSetting(SpdySettingsIds id,
uint8 flags,
uint32 value) {
visitor_->OnSetting(id, flags, value);
}
void BufferedSpdyFramer::OnPing(uint32 unique_id) {
visitor_->OnPing(unique_id);
}
void BufferedSpdyFramer::OnRstStream(SpdyStreamId stream_id,
SpdyStatusCodes status) {
visitor_->OnRstStream(stream_id, status);
}
void BufferedSpdyFramer::OnGoAway(SpdyStreamId last_accepted_stream_id,
SpdyGoAwayStatus status) {
visitor_->OnGoAway(last_accepted_stream_id, status);
}
void BufferedSpdyFramer::OnWindowUpdate(SpdyStreamId stream_id,
int delta_window_size) {
visitor_->OnWindowUpdate(stream_id, delta_window_size);
}
void BufferedSpdyFramer::OnControlFrameCompressed(
const SpdyControlFrame& uncompressed_frame,
const SpdyControlFrame& compressed_frame) {
visitor_->OnControlFrameCompressed(uncompressed_frame, compressed_frame);
}
int BufferedSpdyFramer::protocol_version() {
return spdy_framer_.protocol_version();
}
size_t BufferedSpdyFramer::ProcessInput(const char* data, size_t len) {
return spdy_framer_.ProcessInput(data, len);
}
void BufferedSpdyFramer::Reset() {
spdy_framer_.Reset();
}
SpdyFramer::SpdyError BufferedSpdyFramer::error_code() const {
return spdy_framer_.error_code();
}
SpdyFramer::SpdyState BufferedSpdyFramer::state() const {
return spdy_framer_.state();
}
bool BufferedSpdyFramer::MessageFullyRead() {
return spdy_framer_.MessageFullyRead();
}
bool BufferedSpdyFramer::HasError() {
return spdy_framer_.HasError();
}
SpdySynStreamControlFrame* BufferedSpdyFramer::CreateSynStream(
SpdyStreamId stream_id,
SpdyStreamId associated_stream_id,
SpdyPriority priority,
uint8 credential_slot,
SpdyControlFlags flags,
bool compressed,
const SpdyHeaderBlock* headers) {
return spdy_framer_.CreateSynStream(stream_id, associated_stream_id, priority,
credential_slot, flags, compressed,
headers);
}
SpdySynReplyControlFrame* BufferedSpdyFramer::CreateSynReply(
SpdyStreamId stream_id,
SpdyControlFlags flags,
bool compressed,
const SpdyHeaderBlock* headers) {
return spdy_framer_.CreateSynReply(stream_id, flags, compressed, headers);
}
SpdyRstStreamControlFrame* BufferedSpdyFramer::CreateRstStream(
SpdyStreamId stream_id,
SpdyStatusCodes status) const {
return spdy_framer_.CreateRstStream(stream_id, status);
}
SpdySettingsControlFrame* BufferedSpdyFramer::CreateSettings(
const SettingsMap& values) const {
return spdy_framer_.CreateSettings(values);
}
SpdyPingControlFrame* BufferedSpdyFramer::CreatePingFrame(
uint32 unique_id) const {
return spdy_framer_.CreatePingFrame(unique_id);
}
SpdyGoAwayControlFrame* BufferedSpdyFramer::CreateGoAway(
SpdyStreamId last_accepted_stream_id,
SpdyGoAwayStatus status) const {
return spdy_framer_.CreateGoAway(last_accepted_stream_id, status);
}
SpdyHeadersControlFrame* BufferedSpdyFramer::CreateHeaders(
SpdyStreamId stream_id,
SpdyControlFlags flags,
bool compressed,
const SpdyHeaderBlock* headers) {
return spdy_framer_.CreateHeaders(stream_id, flags, compressed, headers);
}
SpdyWindowUpdateControlFrame* BufferedSpdyFramer::CreateWindowUpdate(
SpdyStreamId stream_id,
uint32 delta_window_size) const {
return spdy_framer_.CreateWindowUpdate(stream_id, delta_window_size);
}
SpdyCredentialControlFrame* BufferedSpdyFramer::CreateCredentialFrame(
const SpdyCredential& credential) const {
return spdy_framer_.CreateCredentialFrame(credential);
}
SpdyDataFrame* BufferedSpdyFramer::CreateDataFrame(SpdyStreamId stream_id,
const char* data,
uint32 len,
SpdyDataFlags flags) {
return spdy_framer_.CreateDataFrame(stream_id, data, len, flags);
}
SpdyPriority BufferedSpdyFramer::GetHighestPriority() const {
return spdy_framer_.GetHighestPriority();
}
bool BufferedSpdyFramer::IsCompressible(const SpdyFrame& frame) const {
return spdy_framer_.IsCompressible(frame);
}
void BufferedSpdyFramer::InitHeaderStreaming(SpdyStreamId stream_id) {
memset(header_buffer_, 0, kHeaderBufferSize);
header_buffer_used_ = 0;
header_buffer_valid_ = true;
header_stream_id_ = stream_id;
DCHECK_NE(header_stream_id_, SpdyFramer::kInvalidStream);
}
} // namespace net