| OLD | NEW |
| 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "webkit/glue/plugins/plugin_group.h" | 5 #include "webkit/glue/plugins/plugin_group.h" |
| 6 | 6 |
| 7 #include <string> | 7 #include <string> |
| 8 #include <vector> | 8 #include <vector> |
| 9 | 9 |
| 10 #include "base/scoped_ptr.h" | 10 #include "base/scoped_ptr.h" |
| 11 #include "base/string_util.h" | 11 #include "base/string_util.h" |
| 12 #include "base/utf_string_conversions.h" | 12 #include "base/utf_string_conversions.h" |
| 13 #include "base/values.h" | 13 #include "base/values.h" |
| 14 #include "base/version.h" | 14 #include "base/version.h" |
| 15 #include "testing/gtest/include/gtest/gtest.h" | 15 #include "testing/gtest/include/gtest/gtest.h" |
| 16 #include "webkit/glue/plugins/webplugininfo.h" | 16 #include "webkit/glue/plugins/webplugininfo.h" |
| 17 #include "webkit/glue/plugins/plugin_list.h" |
| 17 | 18 |
| 18 static const PluginGroupDefinition kPluginDef = { | 19 static const PluginGroupDefinition kPluginDef = { |
| 19 "myplugin", "MyPlugin", "MyPlugin", "", "", "3.0.44", "http://latest/" }; | 20 "myplugin", "MyPlugin", "MyPlugin", "", "", "3.0.44", "http://latest/" }; |
| 20 static const PluginGroupDefinition kPluginDef3 = { | 21 static const PluginGroupDefinition kPluginDef3 = { |
| 21 "myplugin-3", "MyPlugin 3", "MyPlugin", "0", "4", "3.0.44", "http://latest"
}; | 22 "myplugin-3", "MyPlugin 3", "MyPlugin", "0", "4", "3.0.44", |
| 23 "http://latest" }; |
| 22 static const PluginGroupDefinition kPluginDef4 = { | 24 static const PluginGroupDefinition kPluginDef4 = { |
| 23 "myplugin-4", "MyPlugin 4", "MyPlugin", "4", "5", "4.0.44", "http://latest"
}; | 25 "myplugin-4", "MyPlugin 4", "MyPlugin", "4", "5", "4.0.44", |
| 26 "http://latest" }; |
| 24 static const PluginGroupDefinition kPluginDefNotVulnerable = { | 27 static const PluginGroupDefinition kPluginDefNotVulnerable = { |
| 25 "myplugin-latest", "MyPlugin", "MyPlugin", "", "", "", "http://latest" }; | 28 "myplugin-latest", "MyPlugin", "MyPlugin", "", "", "", "http://latest" }; |
| 26 | 29 |
| 27 // name, path, version, desc, mime_types, enabled. | 30 // name, path, version, desc, mime_types, enabled. |
| 28 static WebPluginInfo kPlugin2043 = WebPluginInfo( | 31 static WebPluginInfo kPlugin2043 = WebPluginInfo( |
| 29 ASCIIToUTF16("MyPlugin"), ASCIIToUTF16("2.0.43"), | 32 ASCIIToUTF16("MyPlugin"), ASCIIToUTF16("2.0.43"), |
| 30 ASCIIToUTF16("MyPlugin version 2.0.43")); | 33 ASCIIToUTF16("MyPlugin version 2.0.43")); |
| 31 static WebPluginInfo kPlugin3043 = WebPluginInfo( | 34 static WebPluginInfo kPlugin3043 = WebPluginInfo( |
| 32 ASCIIToUTF16("MyPlugin"), ASCIIToUTF16("3.0.43"), | 35 ASCIIToUTF16("MyPlugin"), ASCIIToUTF16("3.0.43"), |
| 33 ASCIIToUTF16("MyPlugin version 3.0.43")); | 36 ASCIIToUTF16("MyPlugin version 3.0.43")); |
| 34 static WebPluginInfo kPlugin3044 = WebPluginInfo( | 37 static WebPluginInfo kPlugin3044 = WebPluginInfo( |
| 35 ASCIIToUTF16("MyPlugin"), ASCIIToUTF16("3.0.44"), | 38 ASCIIToUTF16("MyPlugin"), ASCIIToUTF16("3.0.44"), |
| 36 ASCIIToUTF16("MyPlugin version 3.0.44")); | 39 ASCIIToUTF16("MyPlugin version 3.0.44")); |
| 37 static WebPluginInfo kPlugin3045 = WebPluginInfo( | 40 static WebPluginInfo kPlugin3045 = WebPluginInfo( |
| 38 ASCIIToUTF16("MyPlugin"), ASCIIToUTF16("3.0.45"), | 41 ASCIIToUTF16("MyPlugin"), ASCIIToUTF16("3.0.45"), |
| 39 ASCIIToUTF16("MyPlugin version 3.0.45")); | 42 ASCIIToUTF16("MyPlugin version 3.0.45")); |
| 40 static WebPluginInfo kPlugin4043 = WebPluginInfo( | 43 static WebPluginInfo kPlugin4043 = WebPluginInfo( |
| 41 ASCIIToUTF16("MyPlugin"), ASCIIToUTF16("4.0.43"), | 44 ASCIIToUTF16("MyPlugin"), ASCIIToUTF16("4.0.43"), |
| 42 ASCIIToUTF16("MyPlugin version 4.0.43")); | 45 ASCIIToUTF16("MyPlugin version 4.0.43")); |
| 43 | 46 |
| 44 class PluginGroupTest : public testing::Test { | 47 class PluginGroupTest : public testing::Test { |
| 48 public: |
| 49 static PluginGroup* CreatePluginGroup( |
| 50 const PluginGroupDefinition& definition) { |
| 51 return PluginGroup::FromPluginGroupDefinition(definition); |
| 52 } |
| 53 static PluginGroup* CreatePluginGroup(const WebPluginInfo& wpi) { |
| 54 return PluginGroup::FromWebPluginInfo(wpi); |
| 55 } |
| 45 protected: | 56 protected: |
| 46 virtual void TearDown() { | 57 virtual void TearDown() { |
| 47 PluginGroup::SetPolicyDisabledPluginPatterns(std::set<string16>()); | 58 PluginGroup::SetPolicyDisabledPluginPatterns(std::set<string16>()); |
| 48 } | 59 } |
| 49 }; | 60 }; |
| 50 | 61 |
| 51 TEST(PluginGroupTest, PluginGroupMatch) { | 62 TEST(PluginGroupTest, PluginGroupMatch) { |
| 52 scoped_ptr<PluginGroup> group(PluginGroup::FromPluginGroupDefinition( | 63 scoped_ptr<PluginGroup> group(PluginGroupTest::CreatePluginGroup( |
| 53 kPluginDef3)); | 64 kPluginDef3)); |
| 54 EXPECT_TRUE(group->Match(kPlugin3045)); | 65 EXPECT_TRUE(group->Match(kPlugin3045)); |
| 55 group->AddPlugin(kPlugin3045, 0); | 66 group->AddPlugin(kPlugin3045, 0); |
| 56 EXPECT_FALSE(group->IsVulnerable()); | 67 EXPECT_FALSE(group->IsVulnerable()); |
| 57 } | 68 } |
| 58 | 69 |
| 59 TEST(PluginGroupTest, PluginGroupMatchCorrectVersion) { | 70 TEST(PluginGroupTest, PluginGroupMatchCorrectVersion) { |
| 60 scoped_ptr<PluginGroup> group(PluginGroup::FromPluginGroupDefinition( | 71 scoped_ptr<PluginGroup> group(PluginGroupTest::CreatePluginGroup( |
| 61 kPluginDef3)); | 72 kPluginDef3)); |
| 62 EXPECT_TRUE(group->Match(kPlugin2043)); | 73 EXPECT_TRUE(group->Match(kPlugin2043)); |
| 63 EXPECT_TRUE(group->Match(kPlugin3043)); | 74 EXPECT_TRUE(group->Match(kPlugin3043)); |
| 64 EXPECT_FALSE(group->Match(kPlugin4043)); | 75 EXPECT_FALSE(group->Match(kPlugin4043)); |
| 65 | 76 |
| 66 group.reset(PluginGroup::FromPluginGroupDefinition(kPluginDef4)); | 77 group.reset(PluginGroupTest::CreatePluginGroup(kPluginDef4)); |
| 67 EXPECT_FALSE(group->Match(kPlugin2043)); | 78 EXPECT_FALSE(group->Match(kPlugin2043)); |
| 68 EXPECT_FALSE(group->Match(kPlugin3043)); | 79 EXPECT_FALSE(group->Match(kPlugin3043)); |
| 69 EXPECT_TRUE(group->Match(kPlugin4043)); | 80 EXPECT_TRUE(group->Match(kPlugin4043)); |
| 70 } | 81 } |
| 71 | 82 |
| 72 TEST(PluginGroupTest, PluginGroupDescription) { | 83 TEST(PluginGroupTest, PluginGroupDescription) { |
| 73 string16 desc3043(ASCIIToUTF16("MyPlugin version 3.0.43")); | 84 string16 desc3043(ASCIIToUTF16("MyPlugin version 3.0.43")); |
| 74 string16 desc3045(ASCIIToUTF16("MyPlugin version 3.0.45")); | 85 string16 desc3045(ASCIIToUTF16("MyPlugin version 3.0.45")); |
| 75 WebPluginInfo plugin3043(kPlugin3043); | 86 WebPluginInfo plugin3043(kPlugin3043); |
| 76 WebPluginInfo plugin3045(kPlugin3045); | 87 WebPluginInfo plugin3045(kPlugin3045); |
| 77 | 88 |
| 78 { | 89 { |
| 79 scoped_ptr<PluginGroup> group(PluginGroup::FromPluginGroupDefinition( | 90 scoped_ptr<PluginGroup> group(PluginGroupTest::CreatePluginGroup( |
| 80 kPluginDef3)); | 91 kPluginDef3)); |
| 81 EXPECT_TRUE(group->Match(plugin3043)); | 92 EXPECT_TRUE(group->Match(plugin3043)); |
| 82 group->AddPlugin(plugin3043, 0); | 93 group->AddPlugin(plugin3043, 0); |
| 83 EXPECT_EQ(desc3043, group->description()); | 94 EXPECT_EQ(desc3043, group->description()); |
| 84 EXPECT_TRUE(group->IsVulnerable()); | 95 EXPECT_TRUE(group->IsVulnerable()); |
| 85 EXPECT_TRUE(group->Match(plugin3045)); | 96 EXPECT_TRUE(group->Match(plugin3045)); |
| 86 group->AddPlugin(plugin3045, 1); | 97 group->AddPlugin(plugin3045, 1); |
| 87 EXPECT_EQ(desc3043, group->description()); | 98 EXPECT_EQ(desc3043, group->description()); |
| 88 EXPECT_TRUE(group->IsVulnerable()); | 99 EXPECT_TRUE(group->IsVulnerable()); |
| 89 } | 100 } |
| 90 | 101 |
| 91 { | 102 { |
| 92 // Disable the first plugin. | 103 // Disable the first plugin. |
| 93 plugin3043.enabled = false; | 104 plugin3043.enabled = false; |
| 94 scoped_ptr<PluginGroup> group(PluginGroup::FromPluginGroupDefinition( | 105 scoped_ptr<PluginGroup> group(PluginGroupTest::CreatePluginGroup( |
| 95 kPluginDef3)); | 106 kPluginDef3)); |
| 96 EXPECT_TRUE(group->Match(plugin3043)); | 107 EXPECT_TRUE(group->Match(plugin3043)); |
| 97 group->AddPlugin(plugin3043, 0); | 108 group->AddPlugin(plugin3043, 0); |
| 98 EXPECT_EQ(desc3043, group->description()); | 109 EXPECT_EQ(desc3043, group->description()); |
| 99 EXPECT_TRUE(group->IsVulnerable()); | 110 EXPECT_TRUE(group->IsVulnerable()); |
| 100 EXPECT_TRUE(group->Match(plugin3045)); | 111 EXPECT_TRUE(group->Match(plugin3045)); |
| 101 group->AddPlugin(plugin3045, 1); | 112 group->AddPlugin(plugin3045, 1); |
| 102 EXPECT_EQ(desc3045, group->description()); | 113 EXPECT_EQ(desc3045, group->description()); |
| 103 EXPECT_FALSE(group->IsVulnerable()); | 114 EXPECT_FALSE(group->IsVulnerable()); |
| 104 } | 115 } |
| 105 | 116 |
| 106 { | 117 { |
| 107 // Disable the second plugin. | 118 // Disable the second plugin. |
| 108 plugin3045.enabled = false; | 119 plugin3045.enabled = false; |
| 109 scoped_ptr<PluginGroup> group(PluginGroup::FromPluginGroupDefinition( | 120 scoped_ptr<PluginGroup> group(PluginGroupTest::CreatePluginGroup( |
| 110 kPluginDef3)); | 121 kPluginDef3)); |
| 111 EXPECT_TRUE(group->Match(plugin3043)); | 122 EXPECT_TRUE(group->Match(plugin3043)); |
| 112 group->AddPlugin(plugin3043, 1); | 123 group->AddPlugin(plugin3043, 1); |
| 113 EXPECT_EQ(desc3043, group->description()); | 124 EXPECT_EQ(desc3043, group->description()); |
| 114 EXPECT_TRUE(group->IsVulnerable()); | 125 EXPECT_TRUE(group->IsVulnerable()); |
| 115 EXPECT_TRUE(group->Match(plugin3045)); | 126 EXPECT_TRUE(group->Match(plugin3045)); |
| 116 group->AddPlugin(plugin3045, 0); | 127 group->AddPlugin(plugin3045, 0); |
| 117 EXPECT_EQ(desc3043, group->description()); | 128 EXPECT_EQ(desc3043, group->description()); |
| 118 EXPECT_TRUE(group->IsVulnerable()); | 129 EXPECT_TRUE(group->IsVulnerable()); |
| 119 } | 130 } |
| 120 } | 131 } |
| 121 | 132 |
| 122 TEST(PluginGroupTest, PluginGroupDefinition) { | 133 TEST(PluginGroupTest, PluginGroupDefinition) { |
| 123 const PluginGroupDefinition* definitions = | 134 const PluginGroupDefinition* definitions = |
| 124 PluginGroup::GetPluginGroupDefinitions(); | 135 NPAPI::PluginList::GetPluginGroupDefinitions(); |
| 125 for (size_t i = 0; i < PluginGroup::GetPluginGroupDefinitionsSize(); ++i) { | 136 for (size_t i = 0; |
| 137 i < NPAPI::PluginList::GetPluginGroupDefinitionsSize(); |
| 138 ++i) { |
| 126 scoped_ptr<PluginGroup> def_group( | 139 scoped_ptr<PluginGroup> def_group( |
| 127 PluginGroup::FromPluginGroupDefinition(definitions[i])); | 140 PluginGroupTest::CreatePluginGroup(definitions[i])); |
| 128 ASSERT_TRUE(def_group.get() != NULL); | 141 ASSERT_TRUE(def_group.get() != NULL); |
| 129 EXPECT_FALSE(def_group->Match(kPlugin2043)); | 142 EXPECT_FALSE(def_group->Match(kPlugin2043)); |
| 130 } | 143 } |
| 131 } | 144 } |
| 132 | 145 |
| 133 TEST(PluginGroupTest, DisableOutdated) { | 146 TEST(PluginGroupTest, DisableOutdated) { |
| 134 scoped_ptr<PluginGroup> group(PluginGroup::FromPluginGroupDefinition( | 147 scoped_ptr<PluginGroup> group(PluginGroupTest::CreatePluginGroup( |
| 135 kPluginDef3)); | 148 kPluginDef3)); |
| 136 group->AddPlugin(kPlugin3043, 0); | 149 group->AddPlugin(kPlugin3043, 0); |
| 137 group->AddPlugin(kPlugin3045, 1); | 150 group->AddPlugin(kPlugin3045, 1); |
| 138 EXPECT_EQ(ASCIIToUTF16("MyPlugin version 3.0.43"), group->description()); | 151 EXPECT_EQ(ASCIIToUTF16("MyPlugin version 3.0.43"), group->description()); |
| 139 EXPECT_TRUE(group->IsVulnerable()); | 152 EXPECT_TRUE(group->IsVulnerable()); |
| 140 | 153 |
| 141 group->DisableOutdatedPlugins(); | 154 group->DisableOutdatedPlugins(); |
| 142 EXPECT_EQ(ASCIIToUTF16("MyPlugin version 3.0.45"), group->description()); | 155 EXPECT_EQ(ASCIIToUTF16("MyPlugin version 3.0.45"), group->description()); |
| 143 EXPECT_FALSE(group->IsVulnerable()); | 156 EXPECT_FALSE(group->IsVulnerable()); |
| 144 } | 157 } |
| 145 | 158 |
| 146 TEST(PluginGroupTest, VersionExtraction) { | 159 TEST(PluginGroupTest, VersionExtraction) { |
| 147 // Some real-world plugin versions (spaces, commata, parentheses, 'r', oh my) | 160 // Some real-world plugin versions (spaces, commata, parentheses, 'r', oh my) |
| 148 const char* versions[][2] = { | 161 const char* versions[][2] = { |
| 149 { "7.6.6 (1671)", "7.6.6.1671" }, // Quicktime | 162 { "7.6.6 (1671)", "7.6.6.1671" }, // Quicktime |
| 150 { "2, 0, 0, 254", "2.0.0.254" }, // DivX | 163 { "2, 0, 0, 254", "2.0.0.254" }, // DivX |
| 151 { "3, 0, 0, 0", "3.0.0.0" }, // Picasa | 164 { "3, 0, 0, 0", "3.0.0.0" }, // Picasa |
| 152 { "1, 0, 0, 1", "1.0.0.1" }, // Earth | 165 { "1, 0, 0, 1", "1.0.0.1" }, // Earth |
| 153 { "10,0,45,2", "10.0.45.2" }, // Flash | 166 { "10,0,45,2", "10.0.45.2" }, // Flash |
| 154 { "11.5.7r609", "11.5.7.609"} // Shockwave | 167 { "11.5.7r609", "11.5.7.609"} // Shockwave |
| 155 }; | 168 }; |
| 156 | 169 |
| 157 for (size_t i = 0; i < arraysize(versions); i++) { | 170 for (size_t i = 0; i < arraysize(versions); i++) { |
| 158 const WebPluginInfo plugin = WebPluginInfo( | 171 const WebPluginInfo plugin = WebPluginInfo( |
| 159 ASCIIToUTF16("Blah Plugin"), ASCIIToUTF16(versions[i][0]), string16()); | 172 ASCIIToUTF16("Blah Plugin"), ASCIIToUTF16(versions[i][0]), string16()); |
| 160 scoped_ptr<PluginGroup> group(PluginGroup::FromWebPluginInfo(plugin)); | 173 scoped_ptr<PluginGroup> group(PluginGroupTest::CreatePluginGroup(plugin)); |
| 161 EXPECT_TRUE(group->Match(plugin)); | 174 EXPECT_TRUE(group->Match(plugin)); |
| 162 group->AddPlugin(plugin, 0); | 175 group->AddPlugin(plugin, 0); |
| 163 scoped_ptr<DictionaryValue> data(group->GetDataForUI()); | 176 scoped_ptr<DictionaryValue> data(group->GetDataForUI()); |
| 164 std::string version; | 177 std::string version; |
| 165 data->GetString("version", &version); | 178 data->GetString("version", &version); |
| 166 EXPECT_EQ(versions[i][1], version); | 179 EXPECT_EQ(versions[i][1], version); |
| 167 } | 180 } |
| 168 } | 181 } |
| 169 | 182 |
| 170 TEST(PluginGroupTest, DisabledByPolicy) { | 183 TEST(PluginGroupTest, DisabledByPolicy) { |
| 171 std::set<string16> disabled_plugins; | 184 std::set<string16> disabled_plugins; |
| 172 disabled_plugins.insert(ASCIIToUTF16("Disable this!")); | 185 disabled_plugins.insert(ASCIIToUTF16("Disable this!")); |
| 173 disabled_plugins.insert(ASCIIToUTF16("*Google*")); | 186 disabled_plugins.insert(ASCIIToUTF16("*Google*")); |
| 174 PluginGroup::SetPolicyDisabledPluginPatterns(disabled_plugins); | 187 PluginGroup::SetPolicyDisabledPluginPatterns(disabled_plugins); |
| 175 | 188 |
| 176 EXPECT_FALSE(PluginGroup::IsPluginNameDisabledByPolicy(ASCIIToUTF16("42"))); | 189 EXPECT_FALSE(PluginGroup::IsPluginNameDisabledByPolicy(ASCIIToUTF16("42"))); |
| 177 EXPECT_TRUE(PluginGroup::IsPluginNameDisabledByPolicy( | 190 EXPECT_TRUE(PluginGroup::IsPluginNameDisabledByPolicy( |
| 178 ASCIIToUTF16("Disable this!"))); | 191 ASCIIToUTF16("Disable this!"))); |
| 179 EXPECT_TRUE(PluginGroup::IsPluginNameDisabledByPolicy( | 192 EXPECT_TRUE(PluginGroup::IsPluginNameDisabledByPolicy( |
| 180 ASCIIToUTF16("Google Earth"))); | 193 ASCIIToUTF16("Google Earth"))); |
| 181 } | 194 } |
| OLD | NEW |