Added a debug info UI for Blimp

blimp/net/stream_packet_reader.cc

 // Copyright 2015 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 "blimp/net/stream_packet_reader.h"
 
 #include <iostream>
 
 #include "base/callback_helpers.h"
 #include "base/logging.h"
 #include "base/memory/weak_ptr.h"
 #include "base/message_loop/message_loop.h"
 #include "base/sys_byteorder.h"
+#include "blimp/net/blimp_connection_statistics.h"
 #include "blimp/net/common.h"
 #include "net/base/io_buffer.h"
 #include "net/base/net_errors.h"
 #include "net/socket/stream_socket.h"
 
 namespace blimp {
 
 std::ostream& operator<<(std::ostream& out,
                          const StreamPacketReader::ReadState state) {
   switch (state) {
     case StreamPacketReader::ReadState::HEADER:
       out << "HEADER";
       break;
     case StreamPacketReader::ReadState::PAYLOAD:
       out << "PAYLOAD";
       break;
     case StreamPacketReader::ReadState::IDLE:
       out << "IDLE";
       break;
   }
   return out;
 }
 
-StreamPacketReader::StreamPacketReader(net::StreamSocket* socket)
-    : read_state_(ReadState::IDLE), socket_(socket), weak_factory_(this) {
+StreamPacketReader::StreamPacketReader(net::StreamSocket* socket,
+                                       BlimpConnectionStatistics* statistics)
+    : read_state_(ReadState::IDLE),
+      socket_(socket),
+      statistics_(statistics),
+      weak_factory_(this) {
   DCHECK(socket_);
   header_buffer_ = new net::GrowableIOBuffer;
   header_buffer_->SetCapacity(kPacketHeaderSizeBytes);
 }
 
 StreamPacketReader::~StreamPacketReader() {}
 
 void StreamPacketReader::ReadPacket(
     const scoped_refptr<net::GrowableIOBuffer>& buf,
     const net::CompletionCallback& callback) {
@@ skipping 82 matching lines @@
   DCHECK_EQ(ReadState::PAYLOAD, read_state_);
   DCHECK_GE(result, 0);
 
   payload_buffer_->set_offset(payload_buffer_->offset() + result);
   if (static_cast<size_t>(payload_buffer_->offset()) < payload_size_) {
     return socket_->Read(payload_buffer_.get(),
                          payload_size_ - payload_buffer_->offset(),
                          base::Bind(&StreamPacketReader::OnReadComplete,
                                     weak_factory_.GetWeakPtr()));
   }
+  statistics_->Add(BlimpConnectionStatistics::BYTES_RECEIVED, payload_size_);
 
   // Finished reading the payload.
   read_state_ = ReadState::IDLE;
   payload_buffer_->set_offset(0);
   return payload_size_;
 }
 
 void StreamPacketReader::OnReadComplete(int result) {
   DCHECK_NE(net::ERR_IO_PENDING, result);
 
@@ skipping 11 matching lines @@
   // If all reading completed, either successfully or by error, inform the
   // caller.
   if (result != net::ERR_IO_PENDING) {
     payload_buffer_ = nullptr;
     base::ResetAndReturn(&callback_)
         .Run(result == net::OK ? payload_size_ : result);
   }
 }
 
 }  // namespace blimp