[Origin Trials] Support updates of disabled token list

chrome/common/origin_trials/chrome_origin_trial_policy.cc

 // 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 "chrome/common/origin_trials/chrome_origin_trial_policy.h"
 
 #include <stdint.h>
 
 #include <vector>
 
@@ skipping 21 matching lines @@
   if (base::CommandLine::InitializedForCurrentProcess()) {
     base::CommandLine* command_line = base::CommandLine::ForCurrentProcess();
     if (command_line->HasSwitch(switches::kOriginTrialPublicKey)) {
       SetPublicKeyFromASCIIString(
         command_line->GetSwitchValueASCII(switches::kOriginTrialPublicKey));
     }
     if (command_line->HasSwitch(switches::kOriginTrialDisabledFeatures)) {
       SetDisabledFeatures(command_line->GetSwitchValueASCII(
           switches::kOriginTrialDisabledFeatures));
     }
+    if (command_line->HasSwitch(switches::kOriginTrialDisabledTokens)) {
+      SetDisabledTokens(command_line->GetSwitchValueASCII(
+          switches::kOriginTrialDisabledTokens));
+    }
   }
 }
 
 ChromeOriginTrialPolicy::~ChromeOriginTrialPolicy() {}
 
 base::StringPiece ChromeOriginTrialPolicy::GetPublicKey() const {
   return base::StringPiece(public_key_);
 }
 
 bool ChromeOriginTrialPolicy::IsFeatureDisabled(
     base::StringPiece feature) const {
   return disabled_features_.count(feature.as_string()) > 0;
 }
 
+bool ChromeOriginTrialPolicy::IsTokenDisabled(
+    base::StringPiece token_signature) const {
+  return disabled_tokens_.count(token_signature.as_string()) > 0;
+}
+
 bool ChromeOriginTrialPolicy::SetPublicKeyFromASCIIString(
     const std::string& ascii_public_key) {
   // Base64-decode the incoming string. Set the key if it is correctly formatted
   std::string new_public_key;
   if (!base::Base64Decode(ascii_public_key, &new_public_key))
     return false;
   if (new_public_key.size() != 32)
     return false;
   public_key_.swap(new_public_key);
   return true;
 }
 
 bool ChromeOriginTrialPolicy::SetDisabledFeatures(
     const std::string& disabled_feature_list) {
   std::set<std::string> new_disabled_features;
   const std::vector<std::string> features =
       base::SplitString(disabled_feature_list, "|", base::TRIM_WHITESPACE,
                         base::SPLIT_WANT_NONEMPTY);
   for (const std::string& feature : features)
     new_disabled_features.insert(feature);
   disabled_features_.swap(new_disabled_features);
   return true;
 }
+
+bool ChromeOriginTrialPolicy::SetDisabledTokens(
+    const std::string& disabled_token_list) {
+  std::set<std::string> new_disabled_tokens;
+  const std::vector<std::string> tokens =
+      base::SplitString(disabled_token_list, "|", base::TRIM_WHITESPACE,
+                        base::SPLIT_WANT_NONEMPTY);
+  for (const std::string& ascii_token : tokens) {
+    std::string token_signature;
+    if (!base::Base64Decode(ascii_token, &token_signature))
+      continue;
+    if (token_signature.size() != 64)
+      continue;
+    new_disabled_tokens.insert(token_signature);
+  }
+  disabled_tokens_.swap(new_disabled_tokens);
+  return true;
+}