[KURL] Avoid re-hashing input if is already canonicalized

third_party/WebKit/Source/platform/weborigin/KURL.cpp

 /*
  * Copyright (C) 2004, 2007, 2008, 2011, 2012 Apple Inc. All rights reserved.
  * Copyright (C) 2012 Research In Motion Limited. All rights reserved.
  * Copyright (C) 2008, 2009, 2011 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.
@@ skipping 11 matching lines @@
  * 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 "platform/weborigin/KURL.h"
 
 #include "platform/weborigin/KnownPorts.h"
 #include "url/url_util.h"
+#include "wtf/MathExtras.h"
 #include "wtf/PtrUtil.h"
 #include "wtf/StdLibExtras.h"
 #include "wtf/text/CString.h"
 #include "wtf/text/StringHash.h"
 #include "wtf/text/StringUTF8Adaptor.h"
 #include "wtf/text/TextEncoding.h"
 #include <algorithm>
 #ifndef NDEBUG
 #include <stdio.h>
 #endif
@@ skipping 668 matching lines @@
   if (url.is8Bit())
     return url::FindAndCompareScheme(asURLChar8Subtle(url), url.length(),
                                      protocol, 0);
   return url::FindAndCompareScheme(url.characters16(), url.length(), protocol,
                                    0);
 }
 
 void KURL::init(const KURL& base,
                 const String& relative,
                 const WTF::TextEncoding* queryEncoding) {
-  if (!relative.isNull() && relative.is8Bit()) {
-    StringUTF8Adaptor relativeUTF8(relative);
-    init(base, relativeUTF8.data(), relativeUTF8.length(), queryEncoding);
-  } else
-    init(base, relative.characters16(), relative.length(), queryEncoding);
-  initProtocolIsInHTTPFamily();
-  initInnerURL();
-  DCHECK_EQ(protocol(), protocol().lower());
-}
-
-template <typename CHAR>
-void KURL::init(const KURL& base,
-                const CHAR* relative,
-                int relativeLength,
-                const WTF::TextEncoding* queryEncoding) {
   // As a performance optimization, we do not use the charset converter
   // if encoding is UTF-8 or other Unicode encodings. Note that this is
   // per HTML5 2.5.3 (resolving URL). The URL canonicalizer will be more
   // efficient with no charset converter object because it can do UTF-8
   // internally with no extra copies.
 
+  StringUTF8Adaptor baseUTF8(base.getString());
+
   // We feel free to make the charset converter object every time since it's
   // just a wrapper around a reference.
   KURLCharsetConverter charsetConverterObject(queryEncoding);
   KURLCharsetConverter* charsetConverter =
       (!queryEncoding || isUnicodeEncoding(queryEncoding))
           ? 0
           : &charsetConverterObject;
 
-  StringUTF8Adaptor baseUTF8(base.getString());
+  // Clamp to int max to avoid overflow.
+  url::RawCanonOutputT<char> output;
+  if (!relative.isNull() && relative.is8Bit()) {
+    StringUTF8Adaptor relativeUTF8(relative);
+    m_isValid = url::ResolveRelative(baseUTF8.data(), baseUTF8.length(),
+                                     base.m_parsed, relativeUTF8.data(),
+                                     clampTo<int>(relativeUTF8.length()),
+                                     charsetConverter, &output, &m_parsed);
+  } else {
+    m_isValid = url::ResolveRelative(baseUTF8.data(), baseUTF8.length(),
+                                     base.m_parsed, relative.characters16(),
+                                     clampTo<int>(relative.length()),
+                                     charsetConverter, &output, &m_parsed);
+  }
 
-  url::RawCanonOutputT<char> output;
-  m_isValid = url::ResolveRelative(baseUTF8.data(), baseUTF8.length(),
-                                   base.m_parsed, relative, relativeLength,
-                                   charsetConverter, &output, &m_parsed);
+  // AtomicString::fromUTF8 will re-hash the raw output and check the
+  // AtomicStringTable (addWithTranslator) for the string. This can be very
+  // expensive for large URLs. However, since many URLs are generated from
+  // existing AtomicStrings (which already have their hashes computed), this
+  // fast path is used if the input string is already canonicalized.
+  //
+  // Because this optimization does not apply to non-AtomicStrings, explicitly
+  // check that the input is Atomic before moving forward with it. If we mark
+  // non-Atomic input as Atomic here, we will render the (const) input string
+  // thread unsafe.
+  if (!relative.isNull() && relative.impl()->isAtomic() &&
+      StringView(output.data(), static_cast<unsigned>(output.length())) ==
+          relative) {
+    m_string = relative;

[esprehn, 2016/11/05 02:33:14]

btw why does this compile?

explicit AtomicString(const String& s)

I thought you had to be explicit and write AtomicString(relative).

C++ is so confusing. :/

[Charlie Harrison, 2016/11/05 03:20:16]

m_string is just a String on KURL, not an explicit AtomicString (for KURL::copy
we sometimes have it be non atomic).

I just tried to go through some of the string constructors and got lost finding
how each case automatically converts though. It is quite confusing. I can
certainly add AtomicString(relative) here to be a bit clearer. I believe the
semantics are identical.

+  } else {
+    m_string = AtomicString::fromUTF8(output.data(), output.length());
+  }
 
-  // See FIXME in KURLPrivate in the header. If canonicalization has not
-  // changed the string, we can avoid an extra allocation by using assignment.
-  m_string = AtomicString::fromUTF8(output.data(), output.length());
+  initProtocolIsInHTTPFamily();
+  initInnerURL();
+  DCHECK_EQ(protocol(), protocol().lower());
 }
 
 void KURL::initInnerURL() {
   if (!m_isValid) {
     m_innerURL.reset();
     return;
   }
   if (url::Parsed* innerParsed = m_parsed.inner_parsed())
     m_innerURL = wrapUnique(new KURL(
         ParsedURLString,
@@ skipping 100 matching lines @@
   m_parsed = newParsed;
   m_string = AtomicString::fromUTF8(output.data(), output.length());
 }
 
 bool KURL::isSafeToSendToAnotherThread() const {
   return m_string.isSafeToSendToAnotherThread() &&
          (!m_innerURL || m_innerURL->isSafeToSendToAnotherThread());
 }
 
 }  // namespace blink