Propagate network delegate events to ChromeDataUseAscriber.

components/data_use_measurement/core/data_use_ascriber.h

Index: components/data_use_measurement/core/data_use_ascriber.h
diff --git a/components/data_use_measurement/core/data_use_ascriber.h b/components/data_use_measurement/core/data_use_ascriber.h
index a4d6300731a5fcb2c2a6acdf23cbb95c90437042..64bf30cb0566add089d31925ac4b008a1468e0ad 100644
--- a/components/data_use_measurement/core/data_use_ascriber.h
+++ b/components/data_use_measurement/core/data_use_ascriber.h
@@ -5,7 +5,13 @@
 #ifndef COMPONENTS_DATA_USE_MEASUREMENT_CORE_DATA_USE_ASCRIBER_H_
 #define COMPONENTS_DATA_USE_MEASUREMENT_CORE_DATA_USE_ASCRIBER_H_
 
+#include <stdint.h>
+#include <memory>

[bengr, 2016/09/23 21:21:13]

Add a blank line above.

[Not at Google. Contact bengr, 2016/09/26 16:29:00]

Done.

+
+#include "url/gurl.h"
+
 namespace net {
+class NetworkDelegate;
 class URLRequest;
 }
 
@@ -22,11 +28,28 @@ class DataUseAscriber {
  public:
   virtual ~DataUseAscriber() {}
 
+  // Creates a network delegate that will be used to track data use.
+  std::unique_ptr<net::NetworkDelegate> CreateNetworkDelegate(
+      std::unique_ptr<net::NetworkDelegate> wrapped_network_delegate);
+
   // Returns the DataUseRecorder to which data usage for the given URL should
   // be ascribed. If no existing DataUseRecorder exists, a new one will be
   // created.
   virtual DataUseRecorder* GetDataUseRecorder(
       const net::URLRequest* request) = 0;
+
+  // Methods called by DataUseNetworkDelegate to propagate data use information:
+  virtual void OnBeforeUrlRequest(net::URLRequest* request);
+
+  virtual void OnBeforeRedirect(net::URLRequest* request,
+                                const GURL& new_location);
+
+  virtual void OnNetworkBytesSent(net::URLRequest* request, int64_t bytes_sent);
+
+  virtual void OnNetworkBytesReceived(net::URLRequest* request,
+                                      int64_t bytes_received);
+
+  virtual void OnUrlRequestCompleted(net::URLRequest* request, bool started);
 };
 
 }  // namespace data_use_measurement