Home | History | Annotate | Download | only in weborigin
      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