Check for equality of the URL's origin in replaceState/pushState

third_party/WebKit/Source/core/frame/History.cpp

 /*
  * Copyright (C) 2007 Apple 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
  *    notice, this list of conditions and the following disclaimer in the
@@ skipping 25 matching lines @@
 #include "core/loader/HistoryItem.h"
 #include "core/loader/NavigationScheduler.h"
 #include "core/page/Page.h"
 #include "platform/RuntimeEnabledFeatures.h"
 #include "platform/weborigin/KURL.h"
 #include "platform/weborigin/SecurityOrigin.h"
 #include "wtf/MainThread.h"
 
 namespace blink {
 
+namespace {
+
+bool equalIgnoringPathQueryAndFragment(const KURL& a, const KURL& b)
+{
+    int aLength = a.pathStart();
+    int bLength = b.pathStart();
+
+    if (aLength != bLength)
+        return false;
+
+    const String& aString = a.string();
+    const String& bString = b.string();
+    for (int i = 0; i < aLength; ++i) {
+        if (aString[i] != bString[i])
+            return false;
+    }
+    return true;
+}
+
+}  // namespace
+
 History::History(LocalFrame* frame)
     : DOMWindowProperty(frame)
     , m_lastStateObjectRequested(nullptr)
 {
 }
 
 DEFINE_TRACE(History)
 {
     DOMWindowProperty::trace(visitor);
 }
@@ skipping 99 matching lines @@
     Document* document = m_frame->document();
 
     if (urlString.isNull())
         return document->url();
     if (urlString.isEmpty())
         return document->baseURL();
 
     return KURL(document->baseURL(), urlString);
 }
 
+bool History::canChangeToUrl(const KURL& url)
+{
+    if (!url.isValid())

[Mike West, 2015/12/08 17:34:59]

Nit: Can you assert that you have `m_frame` and `m_frame->document()`?

[robwu, 2015/12/08 20:31:47]

This is already implied by the existing code. `m_frame` is checked at the start
of History::stateObjectAdded, and `m_frame->document()` is used in
History::urlFor. The assert would be a bit too paranoid / pointless.

+        return false;
+
+    Document* document = m_frame->document();
+    SecurityOrigin* origin = document->securityOrigin();
+    if (origin->isGrantedUniversalAccess())
+        return true;
+
+    if (origin->isUnique())
+        return false;
+
+    if (!equalIgnoringPathQueryAndFragment(url, document->url()))
+        return false;
+
+    RefPtr<SecurityOrigin> requestedOrigin = SecurityOrigin::create(url);
+    if (requestedOrigin->isUnique() || !requestedOrigin->isSameSchemeHostPort(origin))
+        return false;
+
+    return true;
+}
+
 void History::stateObjectAdded(PassRefPtr<SerializedScriptValue> data, const String& /* title */, const String& urlString, HistoryScrollRestorationType restorationType, FrameLoadType type, ExceptionState& exceptionState)
 {
     if (!m_frame || !m_frame->page() || !m_frame->loader().documentLoader())
         return;
 
     KURL fullURL = urlForState(urlString);
-    if (!fullURL.isValid() || !m_frame->document()->securityOrigin()->canRequest(fullURL)) {
+    if (!canChangeToUrl(fullURL)) {
         // We can safely expose the URL to JavaScript, as a) no redirection takes place: JavaScript already had this URL, b) JavaScript can only access a same-origin History object.
-        exceptionState.throwSecurityError("A history state object with URL '" + fullURL.elidedString() + "' cannot be created in a document with origin '" + m_frame->document()->securityOrigin()->toString() + "'.");
+        exceptionState.throwSecurityError("A history state object with URL '" + fullURL.elidedString() + "' cannot be created in a document with origin '" + m_frame->document()->securityOrigin()->toString() + "' and URL '" + m_frame->document()->url().elidedString() + "'.");
         return;
     }
 
     m_frame->loader().updateForSameDocumentNavigation(fullURL, SameDocumentNavigationHistoryApi, data, restorationType, type);
 }
 
 } // namespace blink