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 |