Add basic Referrer-Policy support for Documents

third_party/WebKit/Source/core/dom/ExecutionContext.cpp

 /*
  * Copyright (C) 2008 Apple Inc. All Rights Reserved.
  * Copyright (C) 2012 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:
  * 1. Redistributions of source code must retain the above copyright
  *    notice, this list of conditions and the following disclaimer.
  * 2. Redistributions in binary form must reproduce the above copyright
@@ skipping 16 matching lines @@
 
 #include "core/dom/ExecutionContext.h"
 
 #include "bindings/core/v8/SourceLocation.h"
 #include "core/dom/ExecutionContextTask.h"
 #include "core/events/ErrorEvent.h"
 #include "core/events/EventTarget.h"
 #include "core/fetch/MemoryCache.h"
 #include "core/frame/UseCounter.h"
 #include "core/html/PublicURLManager.h"
+#include "core/inspector/ConsoleMessage.h"
 #include "core/inspector/InspectorInstrumentation.h"
 #include "core/workers/WorkerGlobalScope.h"
 #include "core/workers/WorkerThread.h"
+#include "platform/weborigin/SecurityPolicy.h"
 #include "wtf/PtrUtil.h"
 #include <memory>
 
 namespace blink {
 
 class ExecutionContext::PendingException {
     WTF_MAKE_NONCOPYABLE(PendingException);
 public:
     PendingException(const String& errorMessage, std::unique_ptr<SourceLocation> location)
         : m_errorMessage(errorMessage)
@@ skipping 195 matching lines @@
 {
     String unusedErrorMessage;
     return isSecureContext(unusedErrorMessage, privilegeContextCheck);
 }
 
 String ExecutionContext::outgoingReferrer() const
 {
     return url().strippedForUseAsReferrer();
 }
 
+void ExecutionContext::parseAndSetReferrerPolicy(const String& policies)
+{
+    ReferrerPolicy referrerPolicy = ReferrerPolicyDefault;
+
+    Vector<String> tokens;
+    policies.split(',', true, tokens);
+    for (const auto& token : tokens) {
+        ReferrerPolicy currentResult;
+        if (SecurityPolicy::referrerPolicyFromString(token, &currentResult)) {
+            referrerPolicy = currentResult;
+        }
+    }
+
+    if (referrerPolicy == ReferrerPolicyDefault) {
+        addConsoleMessage(ConsoleMessage::create(RenderingMessageSource, ErrorMessageLevel, "Failed to set referrer policy: The value '" + policies + "' is not one of 'always', 'default', 'never', 'no-referrer', 'no-referrer-when-downgrade', 'origin', 'origin-when-crossorigin', or 'unsafe-url'. The referrer policy has been left unchanged."));
+        return;
+    }
+
+    setReferrerPolicy(referrerPolicy);
+}
+
 void ExecutionContext::setReferrerPolicy(ReferrerPolicy referrerPolicy)
 {
     // When a referrer policy has already been set, the latest value takes precedence.
     UseCounter::count(this, UseCounter::SetReferrerPolicy);
     if (m_referrerPolicy != ReferrerPolicyDefault)
         UseCounter::count(this, UseCounter::ResetReferrerPolicy);
 
     m_referrerPolicy = referrerPolicy;
 }
 
 void ExecutionContext::removeURLFromMemoryCache(const KURL& url)
 {
     memoryCache()->removeURLFromCache(url);
 }
 
 DEFINE_TRACE(ExecutionContext)
 {
     visitor->trace(m_publicURLManager);
     ContextLifecycleNotifier::trace(visitor);
     Supplementable<ExecutionContext>::trace(visitor);
 }
 
 } // namespace blink