Add google_update::GetUntrustedDataValueFromTag()

chrome/installer/util/google_update_util.cc

 // Copyright (c) 2012 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/installer/util/google_update_util.h"
 
 #include <algorithm>
 #include <map>
 #include <utility>
 #include <vector>
@@ skipping 16 matching lines @@
 
 using base::win::RegKey;
 
 namespace google_update {
 
 namespace {
 
 const int kGoogleUpdateTimeoutMs = 20 * 1000;
 
 const char kEnvVariableUntrustedData[] = "GoogleUpdateUntrustedData";
-const int kEnvVariableUntrustedDataMaxLength = 4096;
+const int kUntrustedDataMaxLength = 4096;
 
 // Returns true if Google Update is present at the given level.
 bool IsGoogleUpdatePresent(bool system_install) {
   // Using the existence of version key in the registry to decide.
   return GoogleUpdateSettings::GetGoogleUpdateVersion(system_install).IsValid();
 }
 
 // Returns GoogleUpdateSetup.exe's executable path at specified level.
 // or an empty path if none is found.
 base::FilePath GetGoogleUpdateSetupExe(bool system_install) {
@@ skipping 77 matching lines @@
   return !(c >= 'A' && c <= 'Z' || c >= 'a' && c <= 'z' ||
            c >= '0' && c <= '9' || c == '-' || c == '_' || c == '$');
 }
 
 // Returns true if |key| from untrusted data is valid.
 bool IsUntrustedDataKeyValid(const std::string& key) {
   return std::find_if(key.begin(), key.end(), IsIllegalUntrustedDataKeyChar)
       == key.end();
 }
 
-// Reads and parses untrusted data passed from Google Update as key-value
-// pairs, then overwrites |untrusted_data_map| with the result.
-// Returns true if data are successfully read.
-bool GetGoogleUpdateUntrustedData(
+// Parses |data_string| as key-value pairs and overwrites |untrusted_data| with
+// the result. Returns true if the data could be parsed.
+bool ParseUntrustedData(
+    const std::string& data_string,
     std::map<std::string, std::string>* untrusted_data) {
   DCHECK(untrusted_data);
-  scoped_ptr<base::Environment> env(base::Environment::Create());
-  std::string data_string;
-  if (env == NULL || !env->GetVar(kEnvVariableUntrustedData, &data_string))
-    return false;
-
-  if (data_string.length() > kEnvVariableUntrustedDataMaxLength ||
-      !IsStringPrintable(data_string)) {
-    LOG(ERROR) << "Invalid value in " << kEnvVariableUntrustedData;
+  if (data_string.length() > kUntrustedDataMaxLength) {
+    LOG(ERROR) << "Untrusted data string is too long.";
     return false;
   }
 
-  VLOG(1) << kEnvVariableUntrustedData << ": " << data_string;
+  if (!IsStringPrintable(data_string)) {
+    LOG(ERROR) << "Invalid value in untrusted data string.";

[grt (UTC plus 2), 2014/03/04 15:55:59]

i don't think we need two log statements for the conditions on line 141 and 146.
please lump them back together as they once were.

[jackhou1, 2014/03/05 05:29:22]

Done.

+    return false;
+  }
+
+  VLOG(1) << "Untrusted data string: " << data_string;
 
   std::vector<std::pair<std::string, std::string> > kv_pairs;
   if (!base::SplitStringIntoKeyValuePairs(data_string, '=', '&', &kv_pairs)) {
     LOG(ERROR) << "Failed to parse untrusted data: " << data_string;
     return false;
   }
 
   untrusted_data->clear();
   std::vector<std::pair<std::string, std::string> >::const_iterator it;
   for (it = kv_pairs.begin(); it != kv_pairs.end(); ++it) {
     const std::string& key(it->first);
     // TODO(huangs): URL unescape |value|.
     const std::string& value(it->second);
     if (IsUntrustedDataKeyValid(key) && IsStringPrintable(value))
       (*untrusted_data)[key] = value;
     else
       LOG(ERROR) << "Illegal character found in untrusted data.";
   }
   return true;
 }
 
+// Reads and parses untrusted data passed from Google Update as key-value
+// pairs, then overwrites |untrusted_data_map| with the result.
+// Returns true if data are successfully read.
+bool GetGoogleUpdateUntrustedData(
+    std::map<std::string, std::string>* untrusted_data) {
+  scoped_ptr<base::Environment> env(base::Environment::Create());
+  std::string data_string;
+  if (!env || !env->GetVar(kEnvVariableUntrustedData, &data_string))
+    return false;
+
+  return ParseUntrustedData(data_string, untrusted_data);
+}
+
 }  // namespace
 
 bool EnsureUserLevelGoogleUpdatePresent() {
   VLOG(0) << "Ensuring Google Update is present at user-level.";
 
   bool success = false;
   if (IsGoogleUpdatePresent(false)) {
     success = true;
   } else {
     base::string16 cmd_string;
@@ skipping 30 matching lines @@
   if (GetGoogleUpdateUntrustedData(&untrusted_data)) {
     std::map<std::string, std::string>::const_iterator data_it(
         untrusted_data.find(key));
     if (data_it != untrusted_data.end())
       return data_it->second;
   }
 
   return std::string();
 }
 
+std::string GetUntrustedDataValueFromTag(const std::string& tag,
+                                         const std::string& key) {
+  std::map<std::string, std::string> untrusted_data;
+  if (ParseUntrustedData(tag, &untrusted_data))
+    return untrusted_data[key];
+
+  return std::string();
+}
+
 }  // namespace google_update