[scheduler] Plumb websocket information to scheduler

third_party/WebKit/Source/modules/websockets/DocumentWebSocketChannel.cpp

 /*
  * Copyright (C) 2013 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
@@ skipping 12 matching lines @@
  * 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 "modules/websockets/DocumentWebSocketChannel.h"
 
+#include <memory>
 #include "core/dom/DOMArrayBuffer.h"
 #include "core/dom/Document.h"
 #include "core/dom/ExecutionContext.h"
 #include "core/fileapi/FileReaderLoader.h"
 #include "core/fileapi/FileReaderLoaderClient.h"
 #include "core/frame/LocalFrame.h"
 #include "core/inspector/ConsoleMessage.h"
 #include "core/inspector/InspectorInstrumentation.h"
 #include "core/loader/FrameLoader.h"
 #include "core/loader/FrameLoaderClient.h"
 #include "core/loader/MixedContentChecker.h"
 #include "modules/websockets/InspectorWebSocketEvents.h"
 #include "modules/websockets/WebSocketChannelClient.h"
 #include "modules/websockets/WebSocketFrame.h"
 #include "modules/websockets/WebSocketHandleImpl.h"
+#include "platform/WebFrameScheduler.h"
 #include "platform/loader/fetch/UniqueIdentifier.h"
 #include "platform/network/NetworkLog.h"
 #include "platform/network/WebSocketHandshakeRequest.h"
 #include "platform/weborigin/SecurityOrigin.h"
 #include "public/platform/InterfaceProvider.h"
 #include "public/platform/Platform.h"
 #include "wtf/PtrUtil.h"
-#include <memory>
 
 namespace blink {
 
 class DocumentWebSocketChannel::BlobLoader final
     : public GarbageCollectedFinalized<DocumentWebSocketChannel::BlobLoader>,
       public FileReaderLoaderClient {
  public:
   BlobLoader(PassRefPtr<BlobDataHandle>, DocumentWebSocketChannel*);
   ~BlobLoader() override {}
 
@@ skipping 92 matching lines @@
   }
   if (MixedContentChecker::isMixedContent(document()->getSecurityOrigin(),
                                           url)) {
     String message =
         "Connecting to a non-secure WebSocket server from a secure origin is "
         "deprecated.";
     document()->addConsoleMessage(
         ConsoleMessage::create(JSMessageSource, WarningMessageLevel, message));
   }
 
+  if (document()->frame()) {
+    connection_handle_for_scheduler_ =
+        document()->frame()->frameScheduler()->onActiveConnectionCreated();
+  }
+
   m_url = url;
   Vector<String> protocols;
   // Avoid placing an empty token in the Vector when the protocol string is
   // empty.
   if (!protocol.isEmpty()) {
     // Since protocol is already verified and escaped, we can simply split
     // it.
     protocol.split(", ", true, protocols);
   }
 
@@ skipping 103 matching lines @@
   m_messages.append(new Message(codeToSend, reason));
   processSendQueue();
 }
 
 void DocumentWebSocketChannel::fail(const String& reason,
                                     MessageLevel level,
                                     std::unique_ptr<SourceLocation> location) {
   NETWORK_DVLOG(1) << this << " fail(" << reason << ")";
   // m_handle and m_client can be null here.
 
+  connection_handle_for_scheduler_.reset();
+
   InspectorInstrumentation::didReceiveWebSocketFrameError(document(),
                                                           m_identifier, reason);
   const String message = "WebSocket connection to '" + m_url.elidedString() +
                          "' failed: " + reason;
   document()->addConsoleMessage(ConsoleMessage::create(
       JSMessageSource, level, message, std::move(location)));
 
   if (m_client)
     m_client->didError();
   // |reason| is only for logging and should not be provided for scripts,
   // hence close reason must be empty.
   handleDidClose(false, CloseEventCodeAbnormalClosure, String());
   // handleDidClose may delete this object.
 }
 
 void DocumentWebSocketChannel::disconnect() {
   NETWORK_DVLOG(1) << this << " disconnect()";
   if (m_identifier) {
     TRACE_EVENT_INSTANT1(
         "devtools.timeline", "WebSocketDestroy", TRACE_EVENT_SCOPE_THREAD,
         "data", InspectorWebSocketEvent::data(document(), m_identifier));
     InspectorInstrumentation::didCloseWebSocket(document(), m_identifier);
   }
+  connection_handle_for_scheduler_.reset();
   abortAsyncOperations();
   m_handle.reset();
   m_client = nullptr;
   m_identifier = 0;
 }
 
 DocumentWebSocketChannel::Message::Message(const CString& text)
     : type(MessageTypeText), text(text) {}
 
 DocumentWebSocketChannel::Message::Message(
@@ skipping 172 matching lines @@
   InspectorInstrumentation::didReceiveWebSocketHandshakeResponse(
       document(), m_identifier, m_handshakeRequest.get(), response);
   m_handshakeRequest.clear();
 }
 
 void DocumentWebSocketChannel::didFail(WebSocketHandle* handle,
                                        const String& message) {
   NETWORK_DVLOG(1) << this << " didFail(" << handle << ", " << String(message)
                    << ")";
 
+  connection_handle_for_scheduler_.reset();
+
   DCHECK(m_handle);
   DCHECK_EQ(handle, m_handle.get());
 
   // This function is called when the browser is required to fail the
   // WebSocketConnection. Hence we fail this channel by calling
   // |this->failAsError| function.
   failAsError(message);
   // |this| may be deleted.
 }
 
@@ skipping 62 matching lines @@
   }
 }
 
 void DocumentWebSocketChannel::didClose(WebSocketHandle* handle,
                                         bool wasClean,
                                         unsigned short code,
                                         const String& reason) {
   NETWORK_DVLOG(1) << this << " didClose(" << handle << ", " << wasClean << ", "
                    << code << ", " << String(reason) << ")";
 
+  connection_handle_for_scheduler_.reset();
+
   DCHECK(m_handle);
   DCHECK_EQ(handle, m_handle.get());
 
   m_handle.reset();
 
   if (m_identifier) {
     TRACE_EVENT_INSTANT1(
         "devtools.timeline", "WebSocketDestroy", TRACE_EVENT_SCOPE_THREAD,
         "data", InspectorWebSocketEvent::data(document(), m_identifier));
     InspectorInstrumentation::didCloseWebSocket(document(), m_identifier);
@@ skipping 59 matching lines @@
   WebSocketChannel::trace(visitor);
 }
 
 std::ostream& operator<<(std::ostream& ostream,
                          const DocumentWebSocketChannel* channel) {
   return ostream << "DocumentWebSocketChannel "
                  << static_cast<const void*>(channel);
 }
 
 }  // namespace blink