ThreadSafeDataBuffer.h [plain text]
#pragma once
#include <wtf/ThreadSafeRefCounted.h>
#include <wtf/Vector.h>
namespace WebCore {
class ThreadSafeDataBuffer;
class ThreadSafeDataBufferImpl : public ThreadSafeRefCounted<ThreadSafeDataBufferImpl> {
friend class ThreadSafeDataBuffer;
private:
ThreadSafeDataBufferImpl(Vector<uint8_t>&& data)
{
m_data = WTFMove(data);
}
ThreadSafeDataBufferImpl(const Vector<uint8_t>& data)
: m_data(data)
{
}
ThreadSafeDataBufferImpl(const void* data, unsigned length)
: m_data(length)
{
memcpy(m_data.data(), data, length);
}
Vector<uint8_t> m_data;
};
class ThreadSafeDataBuffer {
public:
static ThreadSafeDataBuffer create(Vector<uint8_t>&& data)
{
return ThreadSafeDataBuffer(WTFMove(data));
}
static ThreadSafeDataBuffer copyVector(const Vector<uint8_t>& data)
{
return ThreadSafeDataBuffer(data);
}
static ThreadSafeDataBuffer copyData(const void* data, unsigned length)
{
return ThreadSafeDataBuffer(data, length);
}
ThreadSafeDataBuffer()
{
}
const Vector<uint8_t>* data() const
{
return m_impl ? &m_impl->m_data : nullptr;
}
size_t size() const
{
return m_impl ? m_impl->m_data.size() : 0;
}
bool operator==(const ThreadSafeDataBuffer& other) const
{
if (!m_impl)
return !other.m_impl;
return m_impl->m_data == other.m_impl->m_data;
}
template<class Encoder> void encode(Encoder&) const;
template<class Decoder> static bool decode(Decoder&, ThreadSafeDataBuffer&);
private:
explicit ThreadSafeDataBuffer(Vector<uint8_t>&& data)
{
m_impl = adoptRef(new ThreadSafeDataBufferImpl(WTFMove(data)));
}
explicit ThreadSafeDataBuffer(const Vector<uint8_t>& data)
{
m_impl = adoptRef(new ThreadSafeDataBufferImpl(data));
}
explicit ThreadSafeDataBuffer(const void* data, unsigned length)
{
m_impl = adoptRef(new ThreadSafeDataBufferImpl(data, length));
}
RefPtr<ThreadSafeDataBufferImpl> m_impl;
};
template<class Encoder>
void ThreadSafeDataBuffer::encode(Encoder& encoder) const
{
bool hasData = m_impl;
encoder << hasData;
if (hasData)
encoder << m_impl->m_data;
}
template<class Decoder>
bool ThreadSafeDataBuffer::decode(Decoder& decoder, ThreadSafeDataBuffer& result)
{
bool hasData;
if (!decoder.decode(hasData))
return false;
if (hasData) {
Vector<uint8_t> data;
if (!decoder.decode(data))
return false;
result = ThreadSafeDataBuffer::create(WTFMove(data));
}
return true;
}
}