/*
 * Copyright (c) 2015, Google Inc. All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are
 * met:
 *
 *     * Redistributions of source code must retain the above copyright
 * notice, this list of conditions and the following disclaimer.
 *     * Redistributions in binary form must reproduce the above
 * copyright notice, this list of conditions and the following disclaimer
 * in the documentation and/or other materials provided with the
 * distribution.
 *     * Neither the name of Google Inc. nor the names of its
 * contributors may be used to endorse or promote products derived from
 * this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include "core/loader/BaseFetchContext.h"

#include "core/testing/NullExecutionContext.h"
#include "platform/RuntimeEnabledFeatures.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace blink {

class MockBaseFetchContext final : public BaseFetchContext {
 public:
  explicit MockBaseFetchContext(ExecutionContext* execution_context)
      : execution_context_(execution_context) {}
  ~MockBaseFetchContext() override {}

  // BaseFetchContext overrides:
  ContentSettingsClient* GetContentSettingsClient() const override {
    return nullptr;
  }
  Settings* GetSettings() const override { return nullptr; }
  SubresourceFilter* GetSubresourceFilter() const override { return nullptr; }
  bool ShouldBlockRequestByInspector(const ResourceRequest&) const override {
    return false;
  }
  void DispatchDidBlockRequest(const ResourceRequest&,
                               const FetchInitiatorInfo&,
                               ResourceRequestBlockedReason) const override {}
  bool ShouldBypassMainWorldCSP() const override { return false; }
  bool IsSVGImageChromeClient() const override { return false; }
  void CountUsage(WebFeature) const override {}
  void CountDeprecation(WebFeature) const override {}
  bool ShouldBlockFetchByMixedContentCheck(
      const ResourceRequest&,
      const KURL&,
      SecurityViolationReportingPolicy) const override {
    return false;
  }
  bool ShouldBlockFetchAsCredentialedSubresource(const ResourceRequest&,
                                                 const KURL&) const override {
    return false;
  }
  ReferrerPolicy GetReferrerPolicy() const override {
    return execution_context_->GetReferrerPolicy();
  }
  String GetOutgoingReferrer() const override {
    return execution_context_->OutgoingReferrer();
  }
  const KURL& Url() const override { return execution_context_->Url(); }

  const SecurityOrigin* GetParentSecurityOrigin() const override {
    return nullptr;
  }
  Optional<WebAddressSpace> GetAddressSpace() const override {
    return WTF::make_optional(
        execution_context_->GetSecurityContext().AddressSpace());
  }
  const ContentSecurityPolicy* GetContentSecurityPolicy() const override {
    return execution_context_->GetContentSecurityPolicy();
  }
  void AddConsoleMessage(ConsoleMessage*) const override {}

  DEFINE_INLINE_TRACE() {
    visitor->Trace(execution_context_);
    BaseFetchContext::Trace(visitor);
  }

 private:
  Member<ExecutionContext> execution_context_;
};

class BaseFetchContextTest : public ::testing::Test {
 protected:
  void SetUp() override {
    execution_context_ = new NullExecutionContext();
    static_cast<NullExecutionContext*>(execution_context_.Get())
        ->SetUpSecurityContext();
    fetch_context_ = new MockBaseFetchContext(execution_context_);
  }

  Persistent<ExecutionContext> execution_context_;
  Persistent<BaseFetchContext> fetch_context_;
};

TEST_F(BaseFetchContextTest, SetIsExternalRequestForPublicContext) {
  EXPECT_EQ(kWebAddressSpacePublic,
            execution_context_->GetSecurityContext().AddressSpace());

  struct TestCase {
    const char* url;
    bool is_external_expectation;
  } cases[] = {
      {"data:text/html,whatever", false},  {"file:///etc/passwd", false},
      {"blob:http://example.com/", false},

      {"http://example.com/", false},      {"https://example.com/", false},

      {"http://192.168.1.1:8000/", true},  {"http://10.1.1.1:8000/", true},

      {"http://localhost/", true},         {"http://127.0.0.1/", true},
      {"http://127.0.0.1:8000/", true}};
  RuntimeEnabledFeatures::SetCorsRFC1918Enabled(false);
  for (const auto& test : cases) {
    SCOPED_TRACE(test.url);
    ResourceRequest main_request(test.url);
    fetch_context_->AddAdditionalRequestHeaders(main_request,
                                                kFetchMainResource);
    EXPECT_FALSE(main_request.IsExternalRequest());

    ResourceRequest sub_request(test.url);
    fetch_context_->AddAdditionalRequestHeaders(sub_request, kFetchSubresource);
    EXPECT_FALSE(sub_request.IsExternalRequest());
  }

  RuntimeEnabledFeatures::SetCorsRFC1918Enabled(true);
  for (const auto& test : cases) {
    SCOPED_TRACE(test.url);
    ResourceRequest main_request(test.url);
    fetch_context_->AddAdditionalRequestHeaders(main_request,
                                                kFetchMainResource);
    EXPECT_EQ(test.is_external_expectation, main_request.IsExternalRequest());

    ResourceRequest sub_request(test.url);
    fetch_context_->AddAdditionalRequestHeaders(sub_request, kFetchSubresource);
    EXPECT_EQ(test.is_external_expectation, sub_request.IsExternalRequest());
  }
}

TEST_F(BaseFetchContextTest, SetIsExternalRequestForPrivateContext) {
  execution_context_->GetSecurityContext().SetAddressSpace(
      kWebAddressSpacePrivate);
  EXPECT_EQ(kWebAddressSpacePrivate,
            execution_context_->GetSecurityContext().AddressSpace());

  struct TestCase {
    const char* url;
    bool is_external_expectation;
  } cases[] = {
      {"data:text/html,whatever", false},  {"file:///etc/passwd", false},
      {"blob:http://example.com/", false},

      {"http://example.com/", false},      {"https://example.com/", false},

      {"http://192.168.1.1:8000/", false}, {"http://10.1.1.1:8000/", false},

      {"http://localhost/", true},         {"http://127.0.0.1/", true},
      {"http://127.0.0.1:8000/", true}};
  RuntimeEnabledFeatures::SetCorsRFC1918Enabled(false);
  for (const auto& test : cases) {
    SCOPED_TRACE(test.url);
    ResourceRequest main_request(test.url);
    fetch_context_->AddAdditionalRequestHeaders(main_request,
                                                kFetchMainResource);
    EXPECT_FALSE(main_request.IsExternalRequest());

    ResourceRequest sub_request(test.url);
    fetch_context_->AddAdditionalRequestHeaders(sub_request, kFetchSubresource);
    EXPECT_FALSE(sub_request.IsExternalRequest());
  }

  RuntimeEnabledFeatures::SetCorsRFC1918Enabled(true);
  for (const auto& test : cases) {
    SCOPED_TRACE(test.url);
    ResourceRequest main_request(test.url);
    fetch_context_->AddAdditionalRequestHeaders(main_request,
                                                kFetchMainResource);
    EXPECT_EQ(test.is_external_expectation, main_request.IsExternalRequest());

    ResourceRequest sub_request(test.url);
    fetch_context_->AddAdditionalRequestHeaders(sub_request, kFetchSubresource);
    EXPECT_EQ(test.is_external_expectation, sub_request.IsExternalRequest());
  }
}

TEST_F(BaseFetchContextTest, SetIsExternalRequestForLocalContext) {
  execution_context_->GetSecurityContext().SetAddressSpace(
      kWebAddressSpaceLocal);
  EXPECT_EQ(kWebAddressSpaceLocal,
            execution_context_->GetSecurityContext().AddressSpace());

  struct TestCase {
    const char* url;
    bool is_external_expectation;
  } cases[] = {
      {"data:text/html,whatever", false},  {"file:///etc/passwd", false},
      {"blob:http://example.com/", false},

      {"http://example.com/", false},      {"https://example.com/", false},

      {"http://192.168.1.1:8000/", false}, {"http://10.1.1.1:8000/", false},

      {"http://localhost/", false},        {"http://127.0.0.1/", false},
      {"http://127.0.0.1:8000/", false}};

  RuntimeEnabledFeatures::SetCorsRFC1918Enabled(false);
  for (const auto& test : cases) {
    ResourceRequest main_request(test.url);
    fetch_context_->AddAdditionalRequestHeaders(main_request,
                                                kFetchMainResource);
    EXPECT_FALSE(main_request.IsExternalRequest());

    ResourceRequest sub_request(test.url);
    fetch_context_->AddAdditionalRequestHeaders(sub_request, kFetchSubresource);
    EXPECT_FALSE(sub_request.IsExternalRequest());
  }

  RuntimeEnabledFeatures::SetCorsRFC1918Enabled(true);
  for (const auto& test : cases) {
    ResourceRequest main_request(test.url);
    fetch_context_->AddAdditionalRequestHeaders(main_request,
                                                kFetchMainResource);
    EXPECT_EQ(test.is_external_expectation, main_request.IsExternalRequest());

    ResourceRequest sub_request(test.url);
    fetch_context_->AddAdditionalRequestHeaders(sub_request, kFetchSubresource);
    EXPECT_EQ(test.is_external_expectation, sub_request.IsExternalRequest());
  }
}

// Tests that CanFollowRedirect() checks both report-only and enforced CSP
// headers.
TEST_F(BaseFetchContextTest, RedirectChecksReportedAndEnforcedCSP) {
  ContentSecurityPolicy* policy =
      execution_context_->GetContentSecurityPolicy();
  policy->DidReceiveHeader("script-src https://foo.test",
                           kContentSecurityPolicyHeaderTypeEnforce,
                           kContentSecurityPolicyHeaderSourceHTTP);
  policy->DidReceiveHeader("script-src https://bar.test",
                           kContentSecurityPolicyHeaderTypeReport,
                           kContentSecurityPolicyHeaderSourceHTTP);
  KURL url(KURL(), "http://baz.test");
  ResourceRequest resource_request(url);
  resource_request.SetRequestContext(WebURLRequest::kRequestContextScript);
  ResourceLoaderOptions options(kDoNotAllowStoredCredentials,
                                kClientDidNotRequestCredentials);
  EXPECT_EQ(ResourceRequestBlockedReason::CSP,
            fetch_context_->CanFollowRedirect(
                Resource::kScript, resource_request, url, options,
                SecurityViolationReportingPolicy::kReport,
                FetchParameters::kUseDefaultOriginRestrictionForType));
  EXPECT_EQ(2u, policy->violation_reports_sent_.size());
}

// Tests that AllowResponse() checks both report-only and enforced CSP headers.
TEST_F(BaseFetchContextTest, AllowResponseChecksReportedAndEnforcedCSP) {
  ContentSecurityPolicy* policy =
      execution_context_->GetContentSecurityPolicy();
  policy->DidReceiveHeader("script-src https://foo.test",
                           kContentSecurityPolicyHeaderTypeEnforce,
                           kContentSecurityPolicyHeaderSourceHTTP);
  policy->DidReceiveHeader("script-src https://bar.test",
                           kContentSecurityPolicyHeaderTypeReport,
                           kContentSecurityPolicyHeaderSourceHTTP);
  KURL url(KURL(), "http://baz.test");
  ResourceRequest resource_request(url);
  resource_request.SetRequestContext(WebURLRequest::kRequestContextScript);
  ResourceLoaderOptions options(kDoNotAllowStoredCredentials,
                                kClientDidNotRequestCredentials);
  EXPECT_EQ(ResourceRequestBlockedReason::CSP,
            fetch_context_->AllowResponse(Resource::kScript, resource_request,
                                          url, options));
  EXPECT_EQ(2u, policy->violation_reports_sent_.size());
}

}  // namespace blink