Added new callbacks to SpdyHeadersHandlerInterface.

net/spdy/spdy_headers_block_parser.cc

 // Copyright (c) 2014 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/spdy_headers_block_parser.h"
 
 #include <memory>
 
 #include "base/sys_byteorder.h"
 
@@ skipping 46 matching lines @@
 
 std::vector<char> SpdyHeadersBlockParserReader::Remainder() {
   std::vector<char> remainder(Available(), '\0');
   if (remainder.size()) {
     ReadN(remainder.size(), &remainder[0]);
   }
   DCHECK_EQ(0u, Available());
   return remainder;
 }
 
-SpdyHeadersBlockParser::SpdyHeadersBlockParser(KeyValueHandler* handler) :
-    state_(READING_HEADER_BLOCK_LEN),
-    remaining_key_value_pairs_for_frame_(0),
-    next_field_len_(0),
+SpdyHeadersBlockParser::SpdyHeadersBlockParser(
+    SpdyHeadersHandlerInterface* handler) : state_(READING_HEADER_BLOCK_LEN),
+    remaining_key_value_pairs_for_frame_(0), next_field_len_(0),
     handler_(handler) {
 }
 
 SpdyHeadersBlockParser::~SpdyHeadersBlockParser() {}
 
 bool SpdyHeadersBlockParser::ParseUInt32(Reader* reader,
                                          uint32_t* parsed_value) {
   // Are there enough bytes available?
   if (reader->Available() < sizeof(uint32_t)) {
     return false;
@@ skipping 24 matching lines @@
   base::StringPiece prefix;
   if (headers_block_prefix_.size()) {
     prefix.set(&headers_block_prefix_[0], headers_block_prefix_.size());
   }
   Reader reader(prefix, base::StringPiece(headers_data, len));
 
   // If we didn't parse out yet the number of key-value pairs in this
   // headers block, try to do it now (succeeds if we received enough bytes).
   if (state_ == READING_HEADER_BLOCK_LEN) {
     if (ParseUInt32(&reader, &remaining_key_value_pairs_for_frame_)) {
+      handler_->OnHeaderBlock(remaining_key_value_pairs_for_frame_);
       state_ = READING_KEY_LEN;
     } else {
       headers_block_prefix_ = reader.Remainder();
       return;
     }
   }
 
   // Parse out and handle the key-value pairs.
   while (remaining_key_value_pairs_for_frame_ > 0) {
     // Parse the key-value length, in case we don't already have it.
@@ skipping 42 matching lines @@
       // Did not receive enough bytes.
       break;
     }
   }
 
   // Unread suffix becomes the prefix upon next invocation.
   headers_block_prefix_ = reader.Remainder();
 
   // Did we finish handling the current block?
   if (remaining_key_value_pairs_for_frame_ == 0) {
+    handler_->OnHeaderBlockEnd();
     Reset();
   }
 }
 
 }  // namespace net