1 /* 2 * Copyright (C) 2013 Google Inc. All rights reserved. 3 * 4 * Redistribution and use in source and binary forms, with or without 5 * modification, are permitted provided that the following conditions are 6 * met: 7 * 8 * * Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * * Redistributions in binary form must reproduce the above 11 * copyright notice, this list of conditions and the following disclaimer 12 * in the documentation and/or other materials provided with the 13 * distribution. 14 * * Neither the name of Google Inc. nor the names of its 15 * contributors may be used to endorse or promote products derived from 16 * this software without specific prior written permission. 17 * 18 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 19 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 20 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 21 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 22 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 23 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 24 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 25 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 26 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 27 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 28 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 29 */ 30 31 #include "config.h" 32 #include "platform/weborigin/OriginAccessEntry.h" 33 34 #include "platform/weborigin/KURL.h" 35 #include "platform/weborigin/SecurityOrigin.h" 36 #include "public/platform/Platform.h" 37 #include "public/platform/WebPublicSuffixList.h" 38 #include <gtest/gtest.h> 39 40 using blink::SecurityOrigin; 41 using blink::OriginAccessEntry; 42 43 namespace { 44 45 class OriginAccessEntryTestSuffixList : public blink::WebPublicSuffixList { 46 public: 47 virtual size_t getPublicSuffixLength(const blink::WebString&) 48 { 49 return 3; // e.g. "com" 50 } 51 }; 52 53 class OriginAccessEntryTestPlatform : public blink::Platform { 54 public: 55 virtual blink::WebPublicSuffixList* publicSuffixList() 56 { 57 return &m_suffixList; 58 } 59 60 // Stub for pure virtual method. 61 virtual void cryptographicallyRandomValues(unsigned char*, size_t) { ASSERT_NOT_REACHED(); } 62 63 private: 64 OriginAccessEntryTestSuffixList m_suffixList; 65 }; 66 67 TEST(OriginAccessEntryTest, PublicSuffixListTest) 68 { 69 OriginAccessEntryTestPlatform platform; 70 blink::Platform::initialize(&platform); 71 72 RefPtr<SecurityOrigin> origin = SecurityOrigin::createFromString("http://www.google.com"); 73 OriginAccessEntry entry1("http", "google.com", OriginAccessEntry::AllowSubdomains, OriginAccessEntry::TreatIPAddressAsIPAddress); 74 OriginAccessEntry entry2("http", "hamster.com", OriginAccessEntry::AllowSubdomains, OriginAccessEntry::TreatIPAddressAsIPAddress); 75 OriginAccessEntry entry3("http", "com", OriginAccessEntry::AllowSubdomains, OriginAccessEntry::TreatIPAddressAsIPAddress); 76 EXPECT_EQ(OriginAccessEntry::MatchesOrigin, entry1.matchesOrigin(*origin)); 77 EXPECT_EQ(OriginAccessEntry::DoesNotMatchOrigin, entry2.matchesOrigin(*origin)); 78 EXPECT_EQ(OriginAccessEntry::MatchesOriginButIsPublicSuffix, entry3.matchesOrigin(*origin)); 79 80 blink::Platform::shutdown(); 81 } 82 83 } // namespace 84 85