Add a cache counter for iOS.

ios/chrome/browser/browsing_data/cache_counter.cc

Index: ios/chrome/browser/browsing_data/cache_counter.cc
diff --git a/ios/chrome/browser/browsing_data/cache_counter.cc b/ios/chrome/browser/browsing_data/cache_counter.cc
new file mode 100644
index 0000000000000000000000000000000000000000..6de72d42c771ac6b7325bb9d044f2fad2e0cc368
--- /dev/null
+++ b/ios/chrome/browser/browsing_data/cache_counter.cc
@@ -0,0 +1,136 @@
+// Copyright 2016 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 "base/bind.h"
+#include "base/threading/thread_task_runner_handle.h"
+#include "components/browsing_data/core/pref_names.h"
+#include "ios/chrome/browser/browsing_data/cache_counter.h"
+#include "ios/web/public/browser_state.h"
+#include "ios/web/public/web_thread.h"
+#include "net/base/completion_callback.h"
+#include "net/base/net_errors.h"
+#include "net/disk_cache/disk_cache.h"
+#include "net/http/http_cache.h"
+#include "net/http/http_transaction_factory.h"
+#include "net/url_request/url_request_context.h"
+#include "net/url_request/url_request_context_getter.h"
+
+namespace {
+
+class IOThreadCacheCounter {
+ public:
+  IOThreadCacheCounter(
+      const scoped_refptr<net::URLRequestContextGetter>& context_getter,
+      const net::CompletionCallback& result_callback)
+      : next_step_(STEP_GET_BACKEND),
+        context_getter_(context_getter),
+        result_callback_(result_callback),
+        result_(0),
+        backend_(nullptr) {}
+
+  void Count() {
+    web::WebThread::PostTask(web::WebThread::IO, FROM_HERE,
+                             base::Bind(&IOThreadCacheCounter::CountInternal,
+                                        base::Unretained(this), net::OK));
+  }
+
+ private:
+  enum Step {
+    STEP_GET_BACKEND,  // Get the disk_cache::Backend instance.
+    STEP_COUNT,        // Run CalculateSizeOfAllEntries() on it.
+    STEP_CALLBACK      // Respond on the UI thread.
+  };
+
+  void CountInternal(int rv) {

[droger, 2016/09/22 14:13:01]

Add DCHECK_CURRENTLY_ON(IO);

[msramek, 2016/09/22 17:02:40]

Done.

+    while (rv != net::ERR_IO_PENDING) {
+      // In case of an error, skip to the last step.
+      if (rv < 0)
+        next_step_ = STEP_CALLBACK;
+
+      // Process the counting in three steps: STEP_GET_BACKEND -> STEP_COUNT ->
+      // -> STEP_CALLBACK.
+      switch (next_step_) {
+        case STEP_GET_BACKEND: {
+          next_step_ = STEP_COUNT;
+
+          net::HttpCache* http_cache = context_getter_->GetURLRequestContext()
+                                           ->http_transaction_factory()
+                                           ->GetCache();
+
+          rv = http_cache->GetBackend(
+              &backend_, base::Bind(&IOThreadCacheCounter::CountInternal,
+                                    base::Unretained(this)));
+          break;

[droger, 2016/09/22 14:13:01]

Sometimes it's return, sometimes break.

Maybe use return everywhere?

[msramek, 2016/09/22 17:02:40]

That is intentional. All disk_cache::Backend methods have the same interface -
they either return a value immediately, or they return net::ERR_IO_PENDING and
then later call a callback.

Here, "break;" is used to jump out of the switch(), but not out of while(), so
if |rv| != ERR_IO_PENDING, the next step is executed immediately. Using
"return;" would exit the method and wait for a callback that never comes.

In STEP_CALLBACK, we use "return;", because it's the last step. Using "break;"
would execute that step ad infinitum.

I added a sentinel STEP_DONE to represent the finished state, so that we can use
"break;" here as well.

+        }
+
+        case STEP_COUNT: {
+          next_step_ = STEP_CALLBACK;
+
+          DCHECK(backend_);
+          rv = backend_->CalculateSizeOfAllEntries(base::Bind(
+              &IOThreadCacheCounter::CountInternal, base::Unretained(this)));
+          break;
+        }
+
+        case STEP_CALLBACK: {
+          result_ = rv;
+
+          web::WebThread::PostTask(
+              web::WebThread::UI, FROM_HERE,
+              base::Bind(&IOThreadCacheCounter::OnCountingFinished,
+                         base::Unretained(this)));
+
+          return;
+        }
+      }
+    }
+  }
+
+  void OnCountingFinished() {

[droger, 2016/09/22 14:13:01]

DCHECK_CURRENTLY_ON(UI);
Or a base::ThreadChecker if you prefer.

[msramek, 2016/09/22 17:02:40]

Done.

+    result_callback_.Run(result_);
+    base::ThreadTaskRunnerHandle::Get()->DeleteSoon(FROM_HERE, this);

[droger, 2016/09/22 14:13:01]

Just curious: why not
delete this;

[msramek, 2016/09/22 17:02:40]

Changed to "delete this;". No particular reason - it just feels weird to me that
I cannot assume the object exists after I just called a method on it :)

+  }
+
+  Step next_step_;
+  scoped_refptr<net::URLRequestContextGetter> context_getter_;
+  net::CompletionCallback result_callback_;
+  int result_;
+  disk_cache::Backend* backend_;
+};
+
+}  // namespace
+
+CacheCounter::CacheCounter(web::BrowserState* browser_state)
+    : pending_(false), browser_state_(browser_state), weak_ptr_factory_(this) {}
+
+CacheCounter::~CacheCounter() {}
+
+const char* CacheCounter::GetPrefName() const {
+  return browsing_data::prefs::kDeleteCache;
+}
+
+void CacheCounter::Count() {
+  // TODO(msramek): disk_cache::Backend currently does not implement counting
+  // for subsets of cache, only for the entire cache. Thus, we ignore the time
+  // period setting and always request counting for the unbounded time interval.
+  // It is up to the UI to interpret the results for finite time intervals as
+  // upper estimates.
+  pending_ = true;
+
+  // IOThreadCacheCounter deletes itself when done.
+  (new IOThreadCacheCounter(browser_state_->GetRequestContext(),
+                            base::Bind(&CacheCounter::OnCacheSizeCalculated,
+                                       weak_ptr_factory_.GetWeakPtr())))
+      ->Count();
+}
+
+void CacheCounter::OnCacheSizeCalculated(int result_bytes) {
+  pending_ = false;
+
+  // A value less than 0 means a net error code.
+  if (result_bytes < 0)
+    return;
+
+  ReportResult(result_bytes);
+}