Merge PluginGroups for Adobe Reader

webkit/glue/plugins/plugin_group.cc

 // Copyright (c) 2010 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 "webkit/glue/plugins/plugin_group.h"
 
 #include "base/linked_ptr.h"
 #include "base/string_util.h"
 #include "base/sys_string_conversions.h"
 #include "base/utf_string_conversions.h"
 #include "base/values.h"
 #include "base/version.h"
 #include "webkit/glue/plugins/plugin_list.h"
 #include "webkit/glue/plugins/webplugininfo.h"
 
-const char* PluginGroup::kAdobeReader8GroupName = "Adobe Reader 8";
-const char* PluginGroup::kAdobeReader9GroupName = "Adobe Reader 9";
+const char* PluginGroup::kAdobeReaderGroupName = "Adobe Reader";
 
 /*static*/
 std::set<string16>* PluginGroup::policy_disabled_plugin_patterns_;
 
 /*static*/
 void PluginGroup::SetPolicyDisabledPluginPatterns(
     const std::set<string16>& set) {
   if (!policy_disabled_plugin_patterns_)
     policy_disabled_plugin_patterns_ = new std::set<string16>(set);
   else
@@ skipping 24 matching lines @@
        it != plugins.end();
        ++it) {
     if (FilePath::CompareEqualIgnoreCase(it->path.value(),
         plugin_path.value()) && IsPluginNameDisabledByPolicy(it->name)) {
       return true;
     }
   }
   return false;
 }
 
+VersionRange::VersionRange(VersionRangeDefinition definition)
+    : low_str(definition.version_matcher_low),
+      high_str(definition.version_matcher_high),
+      min_str(definition.min_version) {
+  if (!low_str.empty())
+    low.reset(Version::GetVersionFromString(low_str));
+  if (!high_str.empty())
+    high.reset(Version::GetVersionFromString(high_str));
+  if (!min_str.empty())
+    min.reset(Version::GetVersionFromString(min_str));
+}
+
+VersionRange::VersionRange(const VersionRange& other) {
+  InitFrom(other);
+}
+
+VersionRange& VersionRange::operator=(const VersionRange& other) {
+  InitFrom(other);
+  return *this;
+}
+
+void VersionRange::InitFrom(const VersionRange& other) {
+  low_str = other.low_str;
+  high_str = other.high_str;
+  min_str = other.min_str;
+  low.reset(Version::GetVersionFromString(other.low_str));
+  high.reset(Version::GetVersionFromString(other.high_str));
+  min.reset(Version::GetVersionFromString(other.min_str));
+}
+
 PluginGroup::PluginGroup(const string16& group_name,
                          const string16& name_matcher,
-                         const std::string& version_range_low,
-                         const std::string& version_range_high,
-                         const std::string& min_version,
                          const std::string& update_url,
                          const std::string& identifier)
     : identifier_(identifier),
       group_name_(group_name),
       name_matcher_(name_matcher),
-      version_range_low_str_(version_range_low),
-      version_range_high_str_(version_range_high),
       update_url_(update_url),
       enabled_(false),
-      min_version_str_(min_version),
       version_(Version::GetVersionFromString("0")) {
-  if (!version_range_low.empty())
-    version_range_low_.reset(Version::GetVersionFromString(version_range_low));
-  if (!version_range_high.empty()) {
-    version_range_high_.reset(
-        Version::GetVersionFromString(version_range_high));
-  }
-  if (!min_version.empty())
-    min_version_.reset(Version::GetVersionFromString(min_version));
 }
 
 void PluginGroup::InitFrom(const PluginGroup& other) {
   identifier_ = other.identifier_;
   group_name_ = other.group_name_;
   name_matcher_ = other.name_matcher_;
-  version_range_low_str_ = other.version_range_low_str_;
-  version_range_high_str_ = other.version_range_high_str_;
-  version_range_low_.reset(
-      Version::GetVersionFromString(version_range_low_str_));
-  version_range_high_.reset(
-      Version::GetVersionFromString(version_range_high_str_));
   description_ = other.description_;
   update_url_ = other.update_url_;
   enabled_ = other.enabled_;
-  min_version_str_ = other.min_version_str_;
-  min_version_.reset(Version::GetVersionFromString(min_version_str_));
+  for (size_t i = 0; i < other.version_ranges_.size(); ++i)
+    version_ranges_.push_back(other.version_ranges_[i]);
   DCHECK_EQ(other.web_plugin_infos_.size(), other.web_plugin_positions_.size());
   for (size_t i = 0; i < other.web_plugin_infos_.size(); ++i)
     AddPlugin(other.web_plugin_infos_[i], other.web_plugin_positions_[i]);
   if (!version_.get())
     version_.reset(Version::GetVersionFromString("0"));
 }
 
 PluginGroup::PluginGroup(const PluginGroup& other) {
   InitFrom(other);
 }
 
 PluginGroup& PluginGroup::operator=(const PluginGroup& other) {
+  version_ranges_.clear();
   InitFrom(other);
   return *this;
 }
 
 /*static*/
 PluginGroup* PluginGroup::FromPluginGroupDefinition(
     const PluginGroupDefinition& definition) {
-  return new PluginGroup(ASCIIToUTF16(definition.name),
-                         ASCIIToUTF16(definition.name_matcher),
-                         definition.version_matcher_low,
-                         definition.version_matcher_high,
-                         definition.min_version,
-                         definition.update_url,
-                         definition.identifier);
+  PluginGroup* group = new PluginGroup(ASCIIToUTF16(definition.name),
+                                       ASCIIToUTF16(definition.name_matcher),
+                                       definition.update_url,
+                                       definition.identifier);
+  for (size_t i = 0; i < definition.num_versions; ++i)
+    group->version_ranges_.push_back(VersionRange(definition.versions[i]));
+  return group;
 }
 
 PluginGroup::~PluginGroup() { }
 
 /*static*/
 std::string PluginGroup::GetIdentifier(const WebPluginInfo& wpi) {
 #if defined(OS_POSIX)
   return wpi.path.BaseName().value();
 #elif defined(OS_WIN)
   return base::SysWideToUTF8(wpi.path.BaseName().value());
 #endif
 }
 
 /*static*/
 std::string PluginGroup::GetLongIdentifier(const WebPluginInfo& wpi) {
 #if defined(OS_POSIX)
   return wpi.path.value();
 #elif defined(OS_WIN)
   return base::SysWideToUTF8(wpi.path.value());
 #endif
 }
 
 /*static*/
 PluginGroup* PluginGroup::FromWebPluginInfo(const WebPluginInfo& wpi) {
   // Create a matcher from the name of this plugin.
-  return new PluginGroup(wpi.name, wpi.name, std::string(), std::string(),
-                         std::string(), std::string(),
+  return new PluginGroup(wpi.name, wpi.name, std::string(),
                          GetIdentifier(wpi));
 }
 
 bool PluginGroup::Match(const WebPluginInfo& plugin) const {
   if (name_matcher_.empty()) {
     return false;
   }
 
   // Look for the name matcher anywhere in the plugin name.
   if (plugin.name.find(name_matcher_) == string16::npos) {
     return false;
   }
 
-  if (version_range_low_.get() == NULL ||
-      version_range_high_.get() == NULL) {
+  if (version_ranges_.empty()) {
     return true;
   }
 
-  // There's a version range, we must be in it.
+  // There's at least one version range, we must be in it.
   scoped_ptr<Version> plugin_version(
       Version::GetVersionFromString(UTF16ToWide(plugin.version)));
   if (plugin_version.get() == NULL) {
     // No version could be extracted, assume we don't match the range.
     return false;
   }
 
-  // We match if we are in the range: [low, high)
-  return (version_range_low_->CompareTo(*plugin_version) <= 0 &&
-          version_range_high_->CompareTo(*plugin_version) > 0);
+  // We match if we are in the range: [low, high) for any of the defined
+  // VersionRanges.
+  for (size_t i = 0; i < version_ranges_.size(); ++i) {
+    const VersionRange* range = &version_ranges_[i];

[Bernhard Bauer, 2010/12/08 13:36:30]

I think you could just use a const VersionRange& here.

[Jakob Kummerow, 2010/12/09 08:58:17]

Done.

+    // If the VersionRange defines neither a high nor a low version, it matches
+    // everything.
+    if (range->low.get() == NULL &&
+        range->high.get() == NULL) {
+      return true;
+    }

[Bernhard Bauer, 2010/12/08 13:36:30]

I think we could even get rid of this rule. If someone wants to define a plugin
group matching all versions, they can just define no version ranges.

[Jakob Kummerow, 2010/12/09 08:58:17]

No, we can't, because they might want to define a minimum version. See
plugin_list.cc for existing examples.

+    DCHECK(range->low.get() != NULL) << "Lower bound of version range must be "
+                                     << "defined.";
+    // An undefined upper bound is treated as infinity.
+    if (range->high.get() == NULL) {
+      if (range->low->CompareTo(*plugin_version) <= 0) {
+        return true;
+      }
+      continue;
+    }
+
+    if (range->low->CompareTo(*plugin_version) <= 0 &&
+        range->high->CompareTo(*plugin_version) > 0) {
+      return true;
+    }

[Bernhard Bauer, 2010/12/08 13:36:30]

You could move these checks together: 

if (range->low->CompareTo(*plugin_version) <= 0 &&
    (range->high.get() == NULL) || range->high->CompareTo(*plugin_version) > 0))

[Jakob Kummerow, 2010/12/09 08:58:17]

Done.

+  }
+  // None of the VersionRanges matched.
+  return false;
 }
 
 Version* PluginGroup::CreateVersionFromString(const string16& version_string) {
   // Remove spaces and ')' from the version string,
   // Replace any instances of 'r', ',' or '(' with a dot.
   std::wstring version = UTF16ToWide(version_string);
   RemoveChars(version, L") ", &version);
   std::replace(version.begin(), version.end(), 'r', '.');
   std::replace(version.begin(), version.end(), ',', '.');
   std::replace(version.begin(), version.end(), '(', '.');
@@ skipping 120 matching lines @@
   if (group_disabled_by_policy || all_plugins_disabled_by_policy) {
     result->SetString("enabledMode", "disabledByPolicy");
   } else {
     result->SetString("enabledMode", enabled_ ? "enabled" : "disabledByUser");
   }
   result->Set("plugin_files", plugin_files);
 
   return result;
 }
 
+/*static*/
+bool PluginGroup::IsPluginOutdated(const Version& plugin_version,
+                                   const VersionRange& version_range) {
+  if ((version_range.low.get() == NULL &&
+          version_range.high.get() == NULL) ||
+      (version_range.low->CompareTo(plugin_version) <= 0 &&
+          version_range.high.get() == NULL) ||
+      (version_range.low->CompareTo(plugin_version) <= 0 &&
+          version_range.high->CompareTo(plugin_version) > 0)) {

[Bernhard Bauer, 2010/12/08 13:36:30]

Could you reuse part of the logic in Match here?

[Jakob Kummerow, 2010/12/09 08:58:17]

Done (by refactoring shared code out into a separate method).

+    // This range is responsible for this plugin.
+    if (version_range.min.get() &&
+        plugin_version.CompareTo(*version_range.min) < 0) {
+      return true;
+    }
+  }
+  return false;
+}
+
 // Returns true if the latest version of this plugin group is vulnerable.
 bool PluginGroup::IsVulnerable() const {
-  if (min_version_.get() == NULL || version_->GetString() == "0") {
-    return false;
+  for (size_t i = 0; i < version_ranges_.size(); ++i) {
+    if (IsPluginOutdated(*version_, version_ranges_[i]))
+      return true;
   }
-  return version_->CompareTo(*min_version_) < 0;
+  return false;
 }
 
 void PluginGroup::DisableOutdatedPlugins() {
-  if (!min_version_.get())
-    return;
-
   description_ = string16();
   enabled_ = false;
 
   for (std::vector<WebPluginInfo>::iterator it =
            web_plugin_infos_.begin();
        it != web_plugin_infos_.end(); ++it) {
     scoped_ptr<Version> version(CreateVersionFromString(it->version));
-    if (version.get() && version->CompareTo(*min_version_) < 0) {
-      it->enabled = false;
-      NPAPI::PluginList::Singleton()->DisablePlugin(it->path);
+    if (version.get()) {
+      for (size_t i = 0; i < version_ranges_.size(); ++i) {
+        if (IsPluginOutdated(*version, version_ranges_[i])) {
+          it->enabled = false;
+          NPAPI::PluginList::Singleton()->DisablePlugin(it->path);
+        }
+      }
     }
     UpdateActivePlugin(*it);
   }
 }
 
 void PluginGroup::Enable(bool enable) {
   bool enabled_plugin_exists = false;
   for (std::vector<WebPluginInfo>::iterator it =
        web_plugin_infos_.begin();
        it != web_plugin_infos_.end(); ++it) {
     if (enable && !IsPluginNameDisabledByPolicy(it->name)) {
       NPAPI::PluginList::Singleton()->EnablePlugin(it->path);
       it->enabled = true;
       enabled_plugin_exists = true;
     } else {
       it->enabled = false;
       NPAPI::PluginList::Singleton()->DisablePlugin(it->path);
     }
   }
   enabled_ = enabled_plugin_exists;
 }