Set file tokens for NaCl main nexe if available to enable validation caching.

chrome/test/nacl/nacl_browsertest_uma.cc

Index: chrome/test/nacl/nacl_browsertest_uma.cc
diff --git a/chrome/test/nacl/nacl_browsertest_uma.cc b/chrome/test/nacl/nacl_browsertest_uma.cc
index c6a3fba6a24f899c7f8eb64a96b6b212e97c1795..97ff5db82786928a3a29de1c276da8dc9d3bc7b5 100644
--- a/chrome/test/nacl/nacl_browsertest_uma.cc
+++ b/chrome/test/nacl/nacl_browsertest_uma.cc
@@ -4,6 +4,7 @@
 
 #include "chrome/test/base/uma_histogram_helper.h"
 #include "chrome/test/nacl/nacl_browsertest_util.h"
+#include "components/nacl/browser/nacl_browser.h"
 #include "native_client/src/trusted/service_runtime/nacl_error_code.h"
 #include "ppapi/c/private/ppb_nacl_private.h"
 
@@ -26,6 +27,17 @@ NACL_BROWSER_TEST_F(NaClBrowserTest, SuccessfulLoadUMA, {
   histograms.ExpectUniqueSample("NaCl.LoadStatus.SelLdr",
                                 LOAD_OK, 1);
 
+  // Check validation cache usage:
+  // For the open-web, only the IRT is considered a "safe" and
+  // identity-cachable file. The nexes and .so files are not.
+  // Should have one cache query for the IRT.
+  histograms.ExpectBucketCount("NaCl.ValidationCache.Query",
+                               nacl::NaClBrowser::CACHE_MISS, 1);
+  // TOTAL should then be 1 query so far.
+  histograms.ExpectTotalCount("NaCl.ValidationCache.Query", 1);
+  // Should have received a cache setting afterwards for IRT.
+  histograms.ExpectTotalCount("NaCl.ValidationCache.Set", 1);
+
   // Make sure we have other important histograms.
   if (!IsAPnaclTest()) {
     histograms.ExpectTotalCount("NaCl.Perf.StartupTime.LoadModule", 1);
@@ -52,6 +64,52 @@ NACL_BROWSER_TEST_F(NaClBrowserTest, SuccessfulLoadUMA, {
   }
 })
 
+class NaClBrowserTestVcacheExtension:
+      public NaClBrowserTestNewlibExtension {
+ public:
+  virtual base::FilePath::StringType Variant() OVERRIDE {
+    return FILE_PATH_LITERAL("extension_vcache_test/newlib");
+  }
+};
+
+IN_PROC_BROWSER_TEST_F(NaClBrowserTestVcacheExtension,
+                       ValidationCacheOfMainNexe) {
+  // Hardcoded extension AppID that corresponds to the hardcoded
+  // public key in the manifest.json file. We need to load the extension
+  // nexe from the same origin, so we can't just try to load the extension
+  // nexe as a mime-type handler from a non-extension URL.
+  base::FilePath::StringType full_url = FILE_PATH_LITERAL(
+      "chrome-extension://cbcdidchbppangcjoddlpdjlenngjldk/"
+      "extension_validation_cache.html");
+  RunNaClIntegrationTest(full_url, true);
+
+  // Make sure histograms from child processes have been accumulated in the
+  // browser brocess.
+  UMAHistogramHelper histograms;
+  histograms.Fetch();
+  // Should have received 2 validation queries (one for IRT and one for NEXE),
+  // and responded with a miss.
+  histograms.ExpectBucketCount("NaCl.ValidationCache.Query",
+                               nacl::NaClBrowser::CACHE_MISS, 2);
+  // TOTAL should then be 2 queries so far.
+  histograms.ExpectTotalCount("NaCl.ValidationCache.Query", 2);
+  // Should have received a cache setting afterwards for IRT and nexe.
+  histograms.ExpectBucketCount("NaCl.ValidationCache.Set",
+                               nacl::NaClBrowser::CACHE_HIT, 2);
+
+  // Load it again to hit the cache.
+  RunNaClIntegrationTest(full_url, true);
+  histograms.Fetch();
+  // Should have received 2 more validation queries later (IRT and NEXE),
+  // and responded with a hit.
+  histograms.ExpectBucketCount("NaCl.ValidationCache.Query",
+                               nacl::NaClBrowser::CACHE_HIT, 2);
+  // TOTAL should then be 4 queries now.
+  histograms.ExpectTotalCount("NaCl.ValidationCache.Query", 4);
+  // Still only 2 settings.
+  histograms.ExpectTotalCount("NaCl.ValidationCache.Set", 2);
+}
+
 // TODO(ncbray) convert the rest of nacl_uma.py (currently in the NaCl repo.)
 // Test validation failures and crashes.