#ifndef DataReference_h
#define DataReference_h
#include <WebCore/SharedBuffer.h>
#include <wtf/Vector.h>
namespace IPC {
class ArgumentDecoder;
class ArgumentEncoder;
class DataReference {
public:
DataReference()
: m_data(0)
, m_size(0)
{
}
DataReference(const uint8_t* data, size_t size)
: m_data(data)
, m_size(size)
{
}
template<size_t inlineCapacity>
DataReference(const Vector<uint8_t, inlineCapacity>& vector)
: m_data(vector.data())
, m_size(vector.size())
{
}
bool isEmpty() const { return size() == 0; }
size_t size() const { return m_size; }
const uint8_t* data() const
{
if (isEmpty())
return 0;
return m_data;
}
Vector<uint8_t> vector() const
{
Vector<uint8_t> result;
result.append(m_data, m_size);
return result;
}
virtual void encode(ArgumentEncoder&) const;
static bool decode(ArgumentDecoder&, DataReference&);
virtual ~DataReference() { }
private:
const uint8_t* m_data;
size_t m_size;
};
class SharedBufferDataReference: public DataReference {
public:
SharedBufferDataReference(WebCore::SharedBuffer* buffer)
{
m_buffer = buffer;
}
size_t size() const { return m_buffer->size(); }
const uint8_t* data() const = delete;
Vector<uint8_t> vector() const = delete;
void encode(ArgumentEncoder&) const override;
virtual ~SharedBufferDataReference() { m_buffer = 0; }
private:
RefPtr<WebCore::SharedBuffer> m_buffer;
};
}
#endif // DataReference_h