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