Adds reporting of total header block bytes to SpdyHeadersBlockParser.

net/spdy/spdy_headers_block_parser.h

 // Copyright 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.
 
 #ifndef NET_SPDY_SPDY_HEADERS_BLOCK_PARSER_H_
 #define NET_SPDY_SPDY_HEADERS_BLOCK_PARSER_H_
 
 #include "base/logging.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/strings/string_piece.h"
@@ skipping 13 matching lines @@
   // the block.
   virtual void OnHeaderBlock(SpdyStreamId stream_id,
                              uint32_t num_of_headers) = 0;
 
   // A callback method which notifies on a SPDY header key value pair.
   virtual void OnHeader(SpdyStreamId stream_id,
                         base::StringPiece key,
                         base::StringPiece value) = 0;
 
   // A callback method which notifies when the parser finishes handling a SPDY
-  // headers block.
-  virtual void OnHeaderBlockEnd(SpdyStreamId stream_id) = 0;
+  // headers block. Also notifies on the total number of bytes in this block.
+  virtual void OnHeaderBlockEnd(SpdyStreamId stream_id,
+                                size_t header_bytes_parsed) = 0;
 };
 
 namespace test {
 
 class SpdyHeadersBlockParserPeer;
 
 }  // namespace test
 
 // This class handles SPDY headers block bytes and parses out key-value pairs
 // as they arrive. This class is not thread-safe, and assumes that all headers
@@ skipping 63 matching lines @@
     FINISHED_HEADER
   };
   ParserState state_;
 
   // Size in bytes of a length field in the spdy header.
   const size_t length_field_size_;
 
   // The maximal number of headers in a SPDY headers block.
   const size_t max_headers_in_block_;
 
+  // A running total of the bytes parsed since the last call to Reset().
+  size_t total_bytes_received_;
+
   // Number of key-value pairs until we complete handling the current
   // headers block.
   uint32_t remaining_key_value_pairs_for_frame_;
 
   // The length of the next header field to be read (either key or value).
   uint32_t next_field_length_;
 
   // Handles key-value pairs as we parse them.
   SpdyHeadersHandlerInterface* handler_;
 
   // Holds unprocessed buffer remainders between calls to
   // |HandleControlFrameHeadersData|.
   SpdyPinnableBufferPiece headers_block_prefix_;
 
   // Holds the key of a partially processed header between calls to
   // |HandleControlFrameHeadersData|.
   SpdyPinnableBufferPiece key_;
 
   // The current header block stream identifier.
   SpdyStreamId stream_id_;
 
   ParserError error_;
 };
 
 }  // namespace net
 
 #endif  // NET_SPDY_SPDY_HEADERS_BLOCK_PARSER_H_