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