WebKitBlobBuilder.cpp [plain text]
#include "config.h"
#include "WebKitBlobBuilder.h"
#include "Blob.h"
#include "Document.h"
#include "ExceptionCode.h"
#include "File.h"
#include "HistogramSupport.h"
#include "LineEnding.h"
#include "ScriptCallStack.h"
#include "TextEncoding.h"
#include <wtf/ArrayBuffer.h>
#include <wtf/ArrayBufferView.h>
#include <wtf/PassRefPtr.h>
#include <wtf/Vector.h>
#include <wtf/text/AtomicString.h>
#include <wtf/text/CString.h>
namespace WebCore {
enum BlobConstructorArrayBufferOrView {
BlobConstructorArrayBuffer,
BlobConstructorArrayBufferView,
BlobConstructorArrayBufferOrViewMax,
};
BlobBuilder::BlobBuilder()
: m_size(0)
{
}
Vector<char>& BlobBuilder::getBuffer()
{
if (m_items.isEmpty() || m_items[m_items.size() - 1].type != BlobDataItem::Data)
m_items.append(BlobDataItem(RawData::create()));
return *m_items[m_items.size() - 1].data->mutableData();
}
void BlobBuilder::append(const String& text, const String& endingType)
{
CString utf8Text = UTF8Encoding().encode(text.characters(), text.length(), EntitiesForUnencodables);
Vector<char>& buffer = getBuffer();
size_t oldSize = buffer.size();
if (endingType == "native")
normalizeLineEndingsToNative(utf8Text, buffer);
else {
ASSERT(endingType == "transparent");
buffer.append(utf8Text.data(), utf8Text.length());
}
m_size += buffer.size() - oldSize;
}
#if ENABLE(BLOB)
void BlobBuilder::append(ArrayBuffer* arrayBuffer)
{
HistogramSupport::histogramEnumeration("WebCore.Blob.constructor.ArrayBufferOrView", BlobConstructorArrayBuffer, BlobConstructorArrayBufferOrViewMax);
if (!arrayBuffer)
return;
appendBytesData(arrayBuffer->data(), arrayBuffer->byteLength());
}
void BlobBuilder::append(ArrayBufferView* arrayBufferView)
{
HistogramSupport::histogramEnumeration("WebCore.Blob.constructor.ArrayBufferOrView", BlobConstructorArrayBufferView, BlobConstructorArrayBufferOrViewMax);
if (!arrayBufferView)
return;
appendBytesData(arrayBufferView->baseAddress(), arrayBufferView->byteLength());
}
#endif
void BlobBuilder::append(Blob* blob)
{
if (!blob)
return;
if (blob->isFile()) {
File* file = toFile(blob);
long long snapshotSize;
double snapshotModificationTime;
file->captureSnapshot(snapshotSize, snapshotModificationTime);
m_size += snapshotSize;
#if ENABLE(FILE_SYSTEM)
if (!file->fileSystemURL().isEmpty())
m_items.append(BlobDataItem(file->fileSystemURL(), 0, snapshotSize, snapshotModificationTime));
else
#endif
m_items.append(BlobDataItem(file->path(), 0, snapshotSize, snapshotModificationTime));
} else {
long long blobSize = static_cast<long long>(blob->size());
m_size += blobSize;
m_items.append(BlobDataItem(blob->url(), 0, blobSize));
}
}
void BlobBuilder::appendBytesData(const void* data, size_t length)
{
Vector<char>& buffer = getBuffer();
size_t oldSize = buffer.size();
buffer.append(static_cast<const char*>(data), length);
m_size += buffer.size() - oldSize;
}
PassRefPtr<Blob> BlobBuilder::getBlob(const String& contentType)
{
OwnPtr<BlobData> blobData = BlobData::create();
blobData->setContentType(Blob::normalizedContentType(contentType));
blobData->swapItems(m_items);
RefPtr<Blob> blob = Blob::create(blobData.release(), m_size);
m_items.append(BlobDataItem(blob->url(), 0, m_size));
return blob;
}
}