Sharing of SharedArrayBuffer via PostMessage transfer

Source/wtf/ArrayBufferContents.h

 /*
  * Copyright (C) 2009 Apple Inc. All rights reserved.
  * 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:
  * 1. Redistributions of source code must retain the above copyright
  *    notice, this list of conditions and the following disclaimer.
  * 2. Redistributions in binary form must reproduce the above copyright
@@ skipping 11 matching lines @@
  * 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.
  */
 
 #ifndef ArrayBufferContents_h
 #define ArrayBufferContents_h
 
 #include "wtf/ArrayBufferDeallocationObserver.h"
 #include "wtf/Noncopyable.h"
+#include "wtf/RefPtr.h"
+#include "wtf/ThreadSafeRefCounted.h"
 #include "wtf/WTFExport.h"
 
 namespace WTF {
 
 class WTF_EXPORT ArrayBufferContents {
     WTF_MAKE_NONCOPYABLE(ArrayBufferContents);
 public:
     enum InitializationPolicy {
         ZeroInitialize,
         DontInitialize
     };
 
+    enum SharingType {
+        NotShared,
+        Shared,
+    };
+
     ArrayBufferContents();
-    ArrayBufferContents(unsigned numElements, unsigned elementByteSize, ArrayBufferContents::InitializationPolicy);
+    ArrayBufferContents(unsigned numElements, unsigned elementByteSize, SharingType isShared, ArrayBufferContents::InitializationPolicy);
 
     // Use with care. data must be allocated with allocateMemory.
     // ArrayBufferContents will take ownership of the data and free it (using freeMemory)
     // upon destruction.
     // This constructor will not call observer->StartObserving(), so it is a responsibility
     // of the caller to make sure JS knows about external memory.
-    ArrayBufferContents(void* data, unsigned sizeInBytes, ArrayBufferDeallocationObserver*);
+    ArrayBufferContents(void* data, unsigned sizeInBytes, SharingType isShared, ArrayBufferDeallocationObserver*);
 
     ~ArrayBufferContents();
 
     void clear();
 
-    void* data() const { return m_data; }
-    unsigned sizeInBytes() const { return m_sizeInBytes; }
+    void* data() const { return m_holder ? m_holder->data() : nullptr; }

[haraken, 2015/06/11 05:59:29]

Is it a valid thing to access these fields after the m_holder is cleared? I
guess we can change this to ASSERT(m_holder).

+    unsigned sizeInBytes() const { return m_holder ? m_holder->sizeInBytes() : 0; }
+    bool isShared() const { return m_holder ? m_holder->isShared() : false; }
 
     void setDeallocationObserver(ArrayBufferDeallocationObserver& observer)
     {
-        if (!m_deallocationObserver) {
-            m_deallocationObserver = &observer;
-            m_deallocationObserver->blinkAllocatedMemory(m_sizeInBytes);
-        }
+        ASSERT(m_holder);
+        m_holder->setDeallocationObserver(observer);
     }
     void setDeallocationObserverWithoutAllocationNotification(ArrayBufferDeallocationObserver& observer)
     {
-        if (!m_deallocationObserver) {
-            m_deallocationObserver = &observer;
-        }
+        ASSERT(m_holder);
+        m_holder->setDeallocationObserverWithoutAllocationNotification(observer);
     }
 
     void transfer(ArrayBufferContents& other);
     void copyTo(ArrayBufferContents& other);
 
     static void allocateMemory(size_t, InitializationPolicy, void*&);
     static void freeMemory(void*, size_t);
 
 private:
-    void* m_data;
-    unsigned m_sizeInBytes;
-    ArrayBufferDeallocationObserver* m_deallocationObserver;
+    class DataHolder : public ThreadSafeRefCounted<DataHolder> {
+        WTF_MAKE_NONCOPYABLE(DataHolder);
+    public:
+        DataHolder();
+        ~DataHolder();
+
+        void allocateNew(unsigned sizeInBytes, SharingType isShared, InitializationPolicy, ArrayBufferDeallocationObserver*);
+        void adopt(void* data, unsigned sizeInBytes, SharingType isShared, ArrayBufferDeallocationObserver*);
+        void copyMemoryTo(DataHolder& other);
+
+        void* data() const { return m_data; }
+        unsigned sizeInBytes() const { return m_sizeInBytes; }
+        bool isShared() const { return m_isShared == Shared; }
+        void setDeallocationObserver(ArrayBufferDeallocationObserver&);
+        void setDeallocationObserverWithoutAllocationNotification(ArrayBufferDeallocationObserver&);
+
+    private:
+        void* m_data;
+        unsigned m_sizeInBytes;
+        SharingType m_isShared;
+        ArrayBufferDeallocationObserver* m_deallocationObserver;
+    };
+
+    RefPtr<DataHolder> m_holder;
 };
 
 } // namespace WTF
 
 #endif // ArrayBufferContents_h