Fix WebSocket to set first party for cookies

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

 /*
  * Copyright (C) 1999 Lars Knoll (knoll@kde.org)
  *           (C) 1999 Antti Koivisto (koivisto@kde.org)
  *           (C) 2001 Dirk Mueller (mueller@kde.org)
  *           (C) 2006 Alexey Proskuryakov (ap@webkit.org)
  * Copyright (C) 2004, 2005, 2006, 2007, 2008, 2009, 2011, 2012 Apple Inc. All rights reserved.
  * Copyright (C) 2008, 2009 Torch Mobile Inc. All rights reserved. (http://www.torchmobile.com/)
  * Copyright (C) 2008, 2009, 2011, 2012 Google Inc. All rights reserved.
  * Copyright (C) 2010 Nokia Corporation and/or its subsidiary(-ies)
  * Copyright (C) Research In Motion Limited 2010-2011. All rights reserved.
@@ skipping 4149 matching lines @@
         date.setMillisecondsSinceEpochForDateTime(convertToLocalTime(currentTimeMS()));
     return String::format("%02d/%02d/%04d %02d:%02d:%02d", date.month() + 1, date.monthDay(), date.fullYear(), date.hour(), date.minute(), date.second());
 }
 
 const KURL Document::firstPartyForCookies() const
 {
     // If this is an imported document, grab its master document's first-party:
     if (importsController() && importsController()->master() && importsController()->master() != this)
         return importsController()->master()->firstPartyForCookies();
 
+    if (!frame()) {
+        return SecurityOrigin::urlWithUniqueSecurityOrigin();
+    }

[Mike West, 2016/07/05 08:45:10]

Nit: No {} for single line bodies.

Would you mind adding a `// TODO(mkwst): This doesn't properly handle HTML
Import documents.` I have another CL floating around somewhere that addresses
this case, but it's not ready to land yet.

[tyoshino (SeeGerritForStatus), 2016/07/05 14:41:27]

Fixed
Added to the top of the function

+
     // TODO(mkwst): This doesn't correctly handle sandboxed documents; we want to look at their URL,
     // but we can't because we don't know what it is.
     Frame* top = frame()->tree().top();
     KURL topDocumentURL = top->isLocalFrame()
         ? toLocalFrame(top)->document()->url()
         : KURL(KURL(), top->securityContext()->getSecurityOrigin()->toString());
     if (SchemeRegistry::shouldTreatURLSchemeAsFirstPartyWhenTopLevel(topDocumentURL.protocol()))
         return topDocumentURL;
 
     // We're intentionally using the URL of each document rather than the document's SecurityOrigin.
@@ skipping 1850 matching lines @@
 }
 
 void showLiveDocumentInstances()
 {
     WeakDocumentSet& set = liveDocumentSet();
     fprintf(stderr, "There are %u documents currently alive:\n", set.size());
     for (Document* document : set)
         fprintf(stderr, "- Document %p URL: %s\n", document, document->url().getString().utf8().data());
 }
 #endif