[ABANDONED] WebSocket Blob receive in the browser process: renderer changes

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

 /*
  * Copyright (C) 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:
  *
  *     * 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 23 matching lines @@
 #include "bindings/core/v8/ScriptController.h"
 #include "bindings/modules/v8/UnionTypesModules.h"
 #include "core/dom/DOMArrayBuffer.h"
 #include "core/dom/DOMArrayBufferView.h"
 #include "core/dom/Document.h"
 #include "core/dom/ExceptionCode.h"
 #include "core/dom/ExecutionContext.h"
 #include "core/dom/SecurityContext.h"
 #include "core/events/MessageEvent.h"
 #include "core/fileapi/Blob.h"
+#include "core/fileapi/FileReaderLoader.h"
 #include "core/frame/ConsoleTypes.h"
 #include "core/frame/LocalDOMWindow.h"
 #include "core/frame/LocalFrame.h"
 #include "core/frame/UseCounter.h"
 #include "core/frame/csp/ContentSecurityPolicy.h"
 #include "core/inspector/ConsoleMessage.h"
 #include "core/inspector/ScriptCallStack.h"
 #include "modules/websockets/CloseEvent.h"
 #include "platform/Logging.h"
 #include "platform/blob/BlobData.h"
@@ skipping 491 matching lines @@
     switch (m_binaryType) {
     case BinaryTypeBlob:
         return "blob";
     case BinaryTypeArrayBuffer:
         return "arraybuffer";
     }
     ASSERT_NOT_REACHED();
     return String();
 }
 
+// static
+WebSocketChannel::BinaryType DOMWebSocket::toChannelBinaryType(BinaryType binaryType)
+{
+    static_assert(static_cast<WebSocketChannel::BinaryType>(BinaryTypeBlob) == WebSocketChannel::BinaryTypeBlob, "enums must match");
+    static_assert(static_cast<WebSocketChannel::BinaryType>(BinaryTypeArrayBuffer) == WebSocketChannel::BinaryTypeArrayBuffer, "enums must match");
+    ASSERT(binaryType == BinaryTypeBlob || binaryType == BinaryTypeArrayBuffer);
+    return static_cast<WebSocketChannel::BinaryType>(binaryType);
+}
+
 void DOMWebSocket::setBinaryType(const String& binaryType)
 {
     if (binaryType == "blob") {
-        m_binaryType = BinaryTypeBlob;
+        if (m_binaryType != BinaryTypeBlob) {
+            m_binaryType = BinaryTypeBlob;
+            if (m_channel)
+                m_channel->changeBinaryType(toChannelBinaryType(m_binaryType));
+        }
         return;
     }
     if (binaryType == "arraybuffer") {
-        m_binaryType = BinaryTypeArrayBuffer;
+        if (m_binaryType != BinaryTypeArrayBuffer) {
+            m_binaryType = BinaryTypeArrayBuffer;
+            if (m_channel)
+                m_channel->changeBinaryType(toChannelBinaryType(m_binaryType));
+        }
         return;
     }
     ASSERT_NOT_REACHED();
 }
 
 const AtomicString& DOMWebSocket::interfaceName() const
 {
     return EventTargetNames::DOMWebSocket;
 }
 
@@ skipping 48 matching lines @@
 
 void DOMWebSocket::didReceiveTextMessage(const String& msg)
 {
     WTF_LOG(Network, "WebSocket %p didReceiveTextMessage() Text message '%s'", this, msg.utf8().data());
     if (m_state != OPEN)
         return;
     Platform::current()->histogramEnumeration("WebCore.WebSocket.ReceiveType", WebSocketReceiveTypeString, WebSocketReceiveTypeMax);
     m_eventQueue->dispatch(MessageEvent::create(msg, SecurityOrigin::create(m_url)->toString()));
 }
 
-void DOMWebSocket::didReceiveBinaryMessage(PassOwnPtr<Vector<char>> binaryData)
+void DOMWebSocket::didReceiveArrayBuffer(PassOwnPtr<Vector<char>> binaryData)
 {
-    WTF_LOG(Network, "WebSocket %p didReceiveBinaryMessage() %lu byte binary message", this, static_cast<unsigned long>(binaryData->size()));
+    WTF_LOG(Network, "WebSocket %p didReceiveArrayBuffer() %lu byte binary message", this, static_cast<unsigned long>(binaryData->size()));
     switch (m_binaryType) {
     case BinaryTypeBlob: {
         size_t size = binaryData->size();
         RefPtr<RawData> rawData = RawData::create();
         binaryData->swap(*rawData->mutableData());
         OwnPtr<BlobData> blobData = BlobData::create();
         blobData->appendData(rawData.release(), 0, BlobDataItem::toEndOfFile);
         Blob* blob = Blob::create(BlobDataHandle::create(blobData.release(), size));
         Platform::current()->histogramEnumeration("WebCore.WebSocket.ReceiveType", WebSocketReceiveTypeBlob, WebSocketReceiveTypeMax);
         m_eventQueue->dispatch(MessageEvent::create(blob, SecurityOrigin::create(m_url)->toString()));
         break;
     }
 
     case BinaryTypeArrayBuffer:
         RefPtr<DOMArrayBuffer> arrayBuffer = DOMArrayBuffer::create(binaryData->data(), binaryData->size());
         Platform::current()->histogramEnumeration("WebCore.WebSocket.ReceiveType", WebSocketReceiveTypeArrayBuffer, WebSocketReceiveTypeMax);
         m_eventQueue->dispatch(MessageEvent::create(arrayBuffer.release(), SecurityOrigin::create(m_url)->toString()));
         break;
     }
 }
 
+void DOMWebSocket::didReceiveBlob(PassRefPtr<BlobDataHandle> blobDataHandle)
+{
+    WTF_LOG(Network, "WebSocket %p didReceiveBlob() %llu byte binary message", this, blobDataHandle->size());
+    switch (m_binaryType) {
+    case BinaryTypeBlob: {
+        Blob* blob = Blob::create(blobDataHandle);
+        Platform::current()->histogramEnumeration("WebCore.WebSocket.ReceiveType", WebSocketReceiveTypeBlob, WebSocketReceiveTypeMax);
+        m_eventQueue->dispatch(MessageEvent::create(blob, SecurityOrigin::create(m_url)->toString()));
+        break;
+    }
+
+    case BinaryTypeArrayBuffer: {
+        // FIXME: Make this async.
+        FileReaderLoader fileReaderLoader(FileReaderLoader::ReadAsArrayBuffer, nullptr);
+        fileReaderLoader.start(executionContext(), blobDataHandle);
+        RefPtr<DOMArrayBuffer> arrayBuffer = fileReaderLoader.arrayBufferResult();
+        Platform::current()->histogramEnumeration("WebCore.WebSocket.ReceiveType", WebSocketReceiveTypeArrayBuffer, WebSocketReceiveTypeMax);
+        m_eventQueue->dispatch(MessageEvent::create(arrayBuffer.release(), SecurityOrigin::create(m_url)->toString()));
+        break;
+    }
+    }
+}
+
 void DOMWebSocket::didError()
 {
     WTF_LOG(Network, "WebSocket %p didError()", this);
     m_state = CLOSED;
     m_eventQueue->dispatch(Event::create(EventTypeNames::error));
 }
 
 void DOMWebSocket::didConsumeBufferedAmount(uint64_t consumed)
 {
     ASSERT(m_bufferedAmount >= consumed + m_consumedBufferedAmount);
@@ skipping 29 matching lines @@
 DEFINE_TRACE(DOMWebSocket)
 {
     visitor->trace(m_channel);
     visitor->trace(m_eventQueue);
     WebSocketChannelClient::trace(visitor);
     RefCountedGarbageCollectedEventTargetWithInlineData<DOMWebSocket>::trace(visitor);
     ActiveDOMObject::trace(visitor);
 }
 
 } // namespace blink