OLD | NEW |
1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 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 "base/command_line.h" | 5 #include "base/command_line.h" |
| 6 #include "base/test/scoped_command_line.h" |
6 #include "chrome/common/chrome_switches.h" | 7 #include "chrome/common/chrome_switches.h" |
7 #include "content/public/common/origin_util.h" | 8 #include "content/public/common/origin_util.h" |
8 #include "testing/gtest/include/gtest/gtest.h" | 9 #include "testing/gtest/include/gtest/gtest.h" |
9 #include "url/gurl.h" | 10 #include "url/gurl.h" |
10 | 11 |
11 using content::IsOriginSecure; | 12 using content::IsOriginSecure; |
12 | 13 |
13 namespace chrome { | 14 namespace chrome { |
14 | 15 |
15 TEST(SecureOriginWhiteList, UnsafelyTreatInsecureOriginAsSecure) { | 16 class SecureOriginWhiteListTest : public testing::Test { |
| 17 void TearDown() override { |
| 18 // Ensure that we reset the whitelisted origins without any flags applied. |
| 19 content::ResetSchemesAndOriginsWhitelistForTesting(); |
| 20 }; |
| 21 }; |
| 22 |
| 23 TEST_F(SecureOriginWhiteListTest, UnsafelyTreatInsecureOriginAsSecure) { |
16 EXPECT_FALSE(content::IsOriginSecure(GURL("http://example.com/a.html"))); | 24 EXPECT_FALSE(content::IsOriginSecure(GURL("http://example.com/a.html"))); |
17 EXPECT_FALSE( | 25 EXPECT_FALSE( |
18 content::IsOriginSecure(GURL("http://127.example.com/a.html"))); | 26 content::IsOriginSecure(GURL("http://127.example.com/a.html"))); |
19 | |
20 // Add http://example.com and http://127.example.com to whitelist by | 27 // Add http://example.com and http://127.example.com to whitelist by |
21 // command-line and see if they are now considered secure origins. | 28 // command-line and see if they are now considered secure origins. |
22 // (The command line is applied via | 29 // (The command line is applied via |
23 // ChromeContentClient::AddSecureSchemesAndOrigins) | 30 // ChromeContentClient::AddSecureSchemesAndOrigins) |
24 base::CommandLine* command_line = base::CommandLine::ForCurrentProcess(); | 31 base::test::ScopedCommandLine scoped_command_line; |
| 32 base::CommandLine* command_line = scoped_command_line.GetProcessCommandLine(); |
25 command_line->AppendSwitchASCII( | 33 command_line->AppendSwitchASCII( |
26 switches::kUnsafelyTreatInsecureOriginAsSecure, | 34 switches::kUnsafelyTreatInsecureOriginAsSecure, |
27 "http://example.com,http://127.example.com"); | 35 "http://example.com,http://127.example.com"); |
28 command_line->AppendSwitch(switches::kUserDataDir); | 36 command_line->AppendSwitch(switches::kUserDataDir); |
29 content::ResetSchemesAndOriginsWhitelistForTesting(); | 37 content::ResetSchemesAndOriginsWhitelistForTesting(); |
30 | 38 |
31 // They should be now white-listed. | 39 // They should be now white-listed. |
32 EXPECT_TRUE(content::IsOriginSecure(GURL("http://example.com/a.html"))); | 40 EXPECT_TRUE(content::IsOriginSecure(GURL("http://example.com/a.html"))); |
33 EXPECT_TRUE(content::IsOriginSecure(GURL("http://127.example.com/a.html"))); | 41 EXPECT_TRUE(content::IsOriginSecure(GURL("http://127.example.com/a.html"))); |
34 | 42 |
35 // Check that similarly named sites are not considered secure. | 43 // Check that similarly named sites are not considered secure. |
36 EXPECT_FALSE(content::IsOriginSecure(GURL("http://128.example.com/a.html"))); | 44 EXPECT_FALSE(content::IsOriginSecure(GURL("http://128.example.com/a.html"))); |
37 EXPECT_FALSE(content::IsOriginSecure( | 45 EXPECT_FALSE(content::IsOriginSecure( |
38 GURL("http://foobar.127.example.com/a.html"))); | 46 GURL("http://foobar.127.example.com/a.html"))); |
39 } | 47 } |
40 | 48 |
41 } // namespace chrome | 49 } // namespace chrome |
OLD | NEW |