[PresentationAPI] on-session-message handler for binary messages (Blink side).

Source/modules/presentation/PresentationSession.cpp

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "config.h"
 #include "modules/presentation/PresentationSession.h"
 
 #include "core/dom/DOMArrayBuffer.h"
 #include "core/dom/DOMArrayBufferView.h"
 #include "core/dom/Document.h"
@@ skipping 73 matching lines @@
 private:
     Member<PresentationSession> m_presentationSession;
     FileReaderLoader m_loader;
 };
 
 PresentationSession::PresentationSession(LocalFrame* frame, const String& id, const String& url)
     : DOMWindowProperty(frame)
     , m_id(id)
     , m_url(url)
     , m_state(WebPresentationSessionState::Connected)
+    , m_binaryType(BinaryTypeBlob)
 {
 }
 
 PresentationSession::~PresentationSession()
 {
     ASSERT(!m_blobLoader);
 }
 
 // static
 PresentationSession* PresentationSession::take(WebPresentationSessionClient* clientRaw, Presentation* presentation)
@@ skipping 108 matching lines @@
             m_messages.removeFirst();
             break;
         case MessageTypeBlob:
             ASSERT(!m_blobLoader);
             m_blobLoader = new BlobLoader(message->blobDataHandle, this);
             break;
         }
     }
 }
 
+String PresentationSession::binaryType() const
+{
+    switch (m_binaryType) {
+    case BinaryTypeBlob:
+        return "blob";
+    case BinaryTypeArrayBuffer:
+        return "arraybuffer";
+    }
+    ASSERT_NOT_REACHED();
+    return String();
+}
+
+void PresentationSession::setBinaryType(const String& binaryType)
+{
+    if (binaryType == "blob") {
+        m_binaryType = BinaryTypeBlob;
+        return;
+    }
+    if (binaryType == "arraybuffer") {
+        m_binaryType = BinaryTypeArrayBuffer;
+        return;
+    }
+    ASSERT_NOT_REACHED();
+}
+
 void PresentationSession::didReceiveTextMessage(const String& message)
 {
+    if (m_state == WebPresentationSessionState::Disconnected)
+        return;
+
     dispatchEvent(MessageEvent::create(message));
 }
 
+void PresentationSession::didReceiveBinaryMessage(const uint8_t* data, size_t length)
+{
+    if (m_state == WebPresentationSessionState::Disconnected)
+        return;
+
+    switch (m_binaryType) {
+    case BinaryTypeBlob: {
+        RefPtr<RawData> rawData = RawData::create();
+        rawData->mutableData()->resize(length);
+        memcpy(rawData->mutableData()->data(), data, length);
+        OwnPtr<BlobData> blobData = BlobData::create();
+        blobData->appendData(rawData.release(), 0, BlobDataItem::toEndOfFile);
+        Blob* blob = Blob::create(BlobDataHandle::create(blobData.release(), length));
+        dispatchEvent(MessageEvent::create(blob));
+        return;
+    }
+    case BinaryTypeArrayBuffer:
+        RefPtr<DOMArrayBuffer> buffer = DOMArrayBuffer::create(data, length);
+        dispatchEvent(MessageEvent::create(buffer.release()));
+        return;
+    }
+    ASSERT_NOT_REACHED();
+}
+
 void PresentationSession::close()
 {
     if (m_state != WebPresentationSessionState::Connected)
         return;
     PresentationController* controller = presentationController();
     if (controller)
         controller->closeSession(m_url, m_id);
 
     // Cancel current Blob loading if any.
     if (m_blobLoader) {
@@ skipping 45 matching lines @@
 {
     ASSERT(!m_messages.isEmpty() && m_messages.first()->type == MessageTypeBlob);
     // FIXME: generate error message?
     // Ignore the current failed blob item and continue with next items.
     m_messages.removeFirst();
     m_blobLoader.clear();
     handleMessageQueue();
 }
 
 } // namespace blink