Store Origin-Trial tokens to ServiceWorkerDataBase

content/browser/service_worker/service_worker_database.cc

 // Copyright 2014 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 "content/browser/service_worker/service_worker_database.h"
 
 #include <utility>
 
 #include "base/files/file_util.h"
 #include "base/lazy_instance.h"
@@ skipping 1205 matching lines @@
   }
   for (int i = 0; i < data.foreign_fetch_origin_size(); ++i) {
     url::Origin parsed_origin(GURL(data.foreign_fetch_origin(i)));
     if (parsed_origin.unique()) {
       DLOG(ERROR) << "Foreign fetch origin '" << data.foreign_fetch_origin(i)
                   << "' is not valid.";
       return ServiceWorkerDatabase::STATUS_ERROR_CORRUPTED;
     }
     out->foreign_fetch_origins.push_back(parsed_origin);
   }
+  if (data.has_origin_trial_tokens()) {
+    const ServiceWorkerOriginTrialInfo& info = data.origin_trial_tokens();
+    TrialTokenValidator::FeatureToTokensMap origin_trial_tokens;
+    for (int i = 0; i < info.features_size(); ++i) {
+      const auto& feature = info.features(i);
+      for (int j = 0; j < feature.tokens_size(); ++j)
+        origin_trial_tokens[feature.name()].push_back(feature.tokens(j));
+    }
+    out->origin_trial_tokens = origin_trial_tokens;
+  }
 
   return ServiceWorkerDatabase::STATUS_OK;
 }
 
 void ServiceWorkerDatabase::WriteRegistrationDataInBatch(
     const RegistrationData& registration,
     leveldb::WriteBatch* batch) {
   DCHECK(batch);
 
   // The registration id and version id should be bumped before this.
@@ skipping 13 matching lines @@
   data.set_resources_total_size_bytes(registration.resources_total_size_bytes);
   for (const GURL& url : registration.foreign_fetch_scopes) {
     DCHECK(ServiceWorkerUtils::ScopeMatches(registration.scope, url))
         << "Foreign fetch scope '" << url
         << "' does not match service worker scope '" << registration.scope
         << "'.";
     data.add_foreign_fetch_scope(url.spec());
   }
   for (const url::Origin& origin : registration.foreign_fetch_origins)
     data.add_foreign_fetch_origin(origin.Serialize());
+  if (registration.origin_trial_tokens) {
+    ServiceWorkerOriginTrialInfo* info = data.mutable_origin_trial_tokens();
+    for (const auto& feature : *registration.origin_trial_tokens) {
+      ServiceWorkerOriginTrialFeature* feature_out = info->add_features();
+      feature_out->set_name(feature.first);
+      for (const auto& token : feature.second)
+        feature_out->add_tokens(token);
+    }
+  }
 
   std::string value;
   bool success = data.SerializeToString(&value);
   DCHECK(success);
   GURL origin = registration.scope.GetOrigin();
   batch->Put(CreateRegistrationKey(data.registration_id(), origin), value);
 }
 
 ServiceWorkerDatabase::Status ServiceWorkerDatabase::ReadResourceRecords(
     const RegistrationData& registration,
@@ skipping 365 matching lines @@
   if (status != STATUS_OK)
     Disable(from_here, status);
   ServiceWorkerMetrics::CountWriteDatabaseResult(status);
 }
 
 bool ServiceWorkerDatabase::IsDatabaseInMemory() const {
   return path_.empty();
 }
 
 }  // namespace content