cashew: reset local counter on plan transition

src/device_impl.cc

 // Copyright (c) 2010, 2011 The Chromium OS 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 "src/device_impl.h"
 
 #include <glog/logging.h>
 
 #include "src/byte_counter.h"
 #include "src/service.h"
@@ skipping 126 matching lines @@
     DeleteByteCounter();
     byte_counter_running_ = false;
   }
   DCHECK(byte_counter_ == NULL);
 }
 
 bool DeviceImpl::ByteCounterRunning() const {
   return byte_counter_running_;
 }
 
+void DeviceImpl::ResetByteCounter(uint64 rx_bytes, uint64 tx_bytes) {
+  DLOG(INFO) << path_ << ": ResetByteCounter: rx_bytes = " << rx_bytes
+      << ", tx_bytes = " << tx_bytes;
+  DCHECK(byte_counter_running_);
+  DCHECK(byte_counter_ != NULL);
+  byte_counter_->SetRxBytes(rx_bytes);
+  byte_counter_->SetTxBytes(tx_bytes);
+}
+
 // ByteCounterDelegate methods
 
 void DeviceImpl::OnByteCounterUpdate(const ByteCounter *counter,
                                      uint64 rx_bytes, uint64 tx_bytes) {
   DCHECK(byte_counter_running_);
   DCHECK(byte_counter_ != NULL);
   DCHECK(counter == byte_counter_);
   DLOG(INFO) << path_ << ": OnByteCounterUpdate: rx_bytes = " << rx_bytes
       << ", tx_bytes = " << tx_bytes;
   parent_->OnByteCounterUpdate(rx_bytes, tx_bytes);
@@ skipping 190 matching lines @@
 
 void DeviceImpl::DeleteByteCounter() {
   DLOG(INFO) << path_ << ": DeleteByteCounter";
   if (byte_counter_ != NULL) {
     delete byte_counter_;
     byte_counter_ = NULL;
   }
 }
 
 }  // namespace cashew